add getDiskSize

Couldn't find anything that exposed this for Windows.

2 files changed, 51 insertions, 14 deletions

diff --git a/Utility/DiskFree.hs b/Utility/DiskFree.hs
index f04f636ec..2f296e2cb 100644
--- a/Utility/DiskFree.hs
+++ b/Utility/DiskFree.hs
@@ -1,13 +1,16 @@
 {- disk free space checking 
  -
- - Copyright 2012 Joey Hess <joey@kitenet.net>
+ - Copyright 2012, 2014 Joey Hess <joey@kitenet.net>
  -
  - 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 d2843ed20..8c9ab6145 100644
--- a/Utility/libdiskfree.c
+++ b/Utility/libdiskfree.c
@@ -1,6 +1,6 @@
 /* disk free space checking, C mini-library
  *
- * Copyright 2012 Joey Hess <joey@kitenet.net>
+ * Copyright 2012, 2014 Joey Hess <joey@kitenet.net>
  *
  * Licensed under the GNU GPL version 3 or higher.
  */
@@ -43,16 +43,12 @@
 #include <errno.h>
 #include <stdio.h>
 
-/* 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("."));