diff --git a/Utility/DiskFree.hs b/Utility/DiskFree.hs index f04f636ecc..2f296e2cb9 100644 --- a/Utility/DiskFree.hs +++ b/Utility/DiskFree.hs @@ -1,13 +1,16 @@ {- disk free space checking - - - Copyright 2012 Joey Hess + - Copyright 2012, 2014 Joey Hess - - Licensed under the GNU GPL version 3 or higher. -} {-# LANGUAGE ForeignFunctionInterface, CPP #-} -module Utility.DiskFree ( getDiskFree ) where +module Utility.DiskFree ( + getDiskFree, + getDiskSize +) where #ifdef WITH_CLIBS @@ -20,9 +23,12 @@ import Foreign.C.Error foreign import ccall safe "libdiskfree.h diskfree" c_diskfree :: CString -> IO CULLong -getDiskFree :: FilePath -> IO (Maybe Integer) -getDiskFree path = withFilePath path $ \c_path -> do - free <- c_diskfree c_path +foreign import ccall safe "libdiskfree.h disksize" c_disksize + :: CString -> IO CULLong + +getVal :: (CString -> IO CULLong) -> FilePath -> IO (Maybe Integer) +getVal getter path = withFilePath path $ \c_path -> do + free <- getter c_path ifM (safeErrno <$> getErrno) ( return $ Just $ toInteger free , return Nothing @@ -30,6 +36,12 @@ getDiskFree path = withFilePath path $ \c_path -> do where safeErrno (Errno v) = v == 0 +getDiskFree :: FilePath -> IO (Maybe Integer) +getDiskFree = getVal c_diskfree + +getDiskSize :: FilePath -> IO (Maybe Integer) +getDiskSize = getVal c_disksize + #else #ifdef mingw32_HOST_OS @@ -41,6 +53,9 @@ getDiskFree :: FilePath -> IO (Maybe Integer) getDiskFree path = catchMaybeIO $ do (sectors, bytes, nfree, _ntotal) <- getDiskFreeSpace (Just path) return $ toInteger sectors * toInteger bytes * toInteger nfree + +getDiskSize :: FilePath -> IO (Maybe Integer) +getDiskSize _ = return Nothing #else #warning Building without disk free space checking support @@ -48,5 +63,8 @@ getDiskFree path = catchMaybeIO $ do getDiskFree :: FilePath -> IO (Maybe Integer) getDiskFree _ = return Nothing +getDiskSize :: FilePath -> IO (Maybe Integer) +getDiskSize _ = return Nothing + #endif #endif diff --git a/Utility/libdiskfree.c b/Utility/libdiskfree.c index d2843ed203..8c9ab6145b 100644 --- a/Utility/libdiskfree.c +++ b/Utility/libdiskfree.c @@ -1,6 +1,6 @@ /* disk free space checking, C mini-library * - * Copyright 2012 Joey Hess + * Copyright 2012, 2014 Joey Hess * * Licensed under the GNU GPL version 3 or higher. */ @@ -43,16 +43,12 @@ #include #include -/* Checks the amount of disk that is available to regular (non-root) users. - * (If there's an error, or this is not supported, - * returns 0 and sets errno to nonzero.) - */ -unsigned long long int diskfree(const char *path) { +unsigned long long int get(const char *path, int req) { #ifdef UNKNOWN errno = 1; return 0; #else - unsigned long long int available, blocksize; + unsigned long long int v, blocksize; struct STATSTRUCT buf; if (STATCALL(path, &buf) != 0) @@ -60,12 +56,35 @@ unsigned long long int diskfree(const char *path) { else errno = 0; - available = buf.f_bavail; + switch (req) { + case 0: + v = buf.f_blocks; + break; + case 1: + v = buf.f_bavail; + break; + default: + v = 0; + } + blocksize = buf.f_bsize; - return available * blocksize; + return v * blocksize; #endif } +/* Checks the amount of disk that is available to regular (non-root) users. + * (If there's an error, or this is not supported, + * returns 0 and sets errno to nonzero.) + */ +unsigned long long int diskfree(const char *path) { + return get(path, 1); +} + +/* Gets the total size of the disk. */ +unsigned long long int disksize(const char *path) { + return get(path, 0); +} + /* main () { printf("%lli\n", diskfree("."));