switch from homegrown code to disk-free-space
According to https://github.com/redneb/disk-free-space/issues/3 , disk-free-space should be at least as portable as my homegrown code was. One change I noticed is, getDiskSize was not implemented for windows in the old code, and should work now.
This commit is contained in:
parent
46fe686ba0
commit
a665f92b91
4 changed files with 10 additions and 141 deletions
|
@ -1,70 +1,23 @@
|
||||||
{- disk free space checking
|
{- disk free space checking shim
|
||||||
-
|
-
|
||||||
- Copyright 2012, 2014 Joey Hess <id@joeyh.name>
|
- Copyright 2016 Joey Hess <id@joeyh.name>
|
||||||
-
|
-
|
||||||
- License: BSD-2-clause
|
- License: BSD-2-clause
|
||||||
-}
|
-}
|
||||||
|
|
||||||
{-# LANGUAGE ForeignFunctionInterface, CPP #-}
|
{-# OPTIONS_GHC -fno-warn-tabs #-}
|
||||||
|
|
||||||
module Utility.DiskFree (
|
module Utility.DiskFree (
|
||||||
getDiskFree,
|
getDiskFree,
|
||||||
getDiskSize
|
getDiskSize
|
||||||
) where
|
) where
|
||||||
|
|
||||||
#ifdef WITH_CLIBS
|
import System.DiskSpace
|
||||||
|
import Utility.Applicative
|
||||||
import Common
|
import Utility.Exception
|
||||||
|
|
||||||
import Foreign.C.Types
|
|
||||||
import Foreign.C.String
|
|
||||||
import Foreign.C.Error
|
|
||||||
|
|
||||||
foreign import ccall safe "libdiskfree.h diskfree" c_diskfree
|
|
||||||
:: CString -> IO CULLong
|
|
||||||
|
|
||||||
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
|
|
||||||
)
|
|
||||||
where
|
|
||||||
safeErrno (Errno v) = v == 0
|
|
||||||
|
|
||||||
getDiskFree :: FilePath -> IO (Maybe Integer)
|
getDiskFree :: FilePath -> IO (Maybe Integer)
|
||||||
getDiskFree = getVal c_diskfree
|
getDiskFree = catchMaybeIO . getAvailSpace
|
||||||
|
|
||||||
getDiskSize :: FilePath -> IO (Maybe Integer)
|
getDiskSize :: FilePath -> IO (Maybe Integer)
|
||||||
getDiskSize = getVal c_disksize
|
getDiskSize = fmap diskTotal <$$> catchMaybeIO . getDiskUsage
|
||||||
|
|
||||||
#else
|
|
||||||
#ifdef mingw32_HOST_OS
|
|
||||||
|
|
||||||
import Common
|
|
||||||
|
|
||||||
import System.Win32.File
|
|
||||||
|
|
||||||
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
|
|
||||||
|
|
||||||
getDiskFree :: FilePath -> IO (Maybe Integer)
|
|
||||||
getDiskFree _ = return Nothing
|
|
||||||
|
|
||||||
getDiskSize :: FilePath -> IO (Maybe Integer)
|
|
||||||
getDiskSize _ = return Nothing
|
|
||||||
|
|
||||||
#endif
|
|
||||||
#endif
|
|
||||||
|
|
|
@ -1,84 +0,0 @@
|
||||||
/* disk free space checking, C mini-library
|
|
||||||
*
|
|
||||||
* Copyright 2012, 2014 Joey Hess <id@joeyh.name>
|
|
||||||
*
|
|
||||||
* License: BSD-2-clause
|
|
||||||
*/
|
|
||||||
|
|
||||||
/* Include appropriate headers for the OS, and define what will be used to
|
|
||||||
* check the free space. */
|
|
||||||
#if defined (__FreeBSD__)
|
|
||||||
# include <sys/param.h>
|
|
||||||
# include <sys/mount.h>
|
|
||||||
# define STATCALL statfs /* statfs64 not yet tested on a real FreeBSD machine */
|
|
||||||
# define STATSTRUCT statfs
|
|
||||||
# define BSIZE f_bsize
|
|
||||||
#else
|
|
||||||
#if defined __ANDROID__
|
|
||||||
# warning free space checking code not available for Android
|
|
||||||
# define UNKNOWN
|
|
||||||
#else
|
|
||||||
#if defined (__linux__) || defined (__APPLE__) || defined (__FreeBSD_kernel__) || (defined (__SVR4) && defined (__sun))
|
|
||||||
/* Linux or OSX or Debian kFreeBSD or Solaris */
|
|
||||||
/* This is a POSIX standard, so might also work elsewhere too. */
|
|
||||||
# include <sys/statvfs.h>
|
|
||||||
# define STATCALL statvfs
|
|
||||||
# define STATSTRUCT statvfs
|
|
||||||
# define BSIZE f_frsize
|
|
||||||
#else
|
|
||||||
# warning free space checking code not available for this OS
|
|
||||||
# define UNKNOWN
|
|
||||||
#endif
|
|
||||||
#endif
|
|
||||||
#endif
|
|
||||||
|
|
||||||
#include <errno.h>
|
|
||||||
#include <stdio.h>
|
|
||||||
|
|
||||||
unsigned long long int get(const char *path, int req) {
|
|
||||||
#ifdef UNKNOWN
|
|
||||||
errno = 1;
|
|
||||||
return 0;
|
|
||||||
#else
|
|
||||||
unsigned long long int v, blocksize;
|
|
||||||
struct STATSTRUCT buf;
|
|
||||||
|
|
||||||
if (STATCALL(path, &buf) != 0)
|
|
||||||
return 0; /* errno is set */
|
|
||||||
else
|
|
||||||
errno = 0;
|
|
||||||
|
|
||||||
switch (req) {
|
|
||||||
case 0:
|
|
||||||
v = buf.f_blocks;
|
|
||||||
break;
|
|
||||||
case 1:
|
|
||||||
v = buf.f_bavail;
|
|
||||||
break;
|
|
||||||
default:
|
|
||||||
v = 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
blocksize = buf.BSIZE;
|
|
||||||
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("."));
|
|
||||||
}
|
|
||||||
*/
|
|
|
@ -1 +0,0 @@
|
||||||
unsigned long long int diskfree(const char *path);
|
|
|
@ -111,7 +111,8 @@ Executable git-annex
|
||||||
esqueleto, persistent-sqlite, persistent, persistent-template,
|
esqueleto, persistent-sqlite, persistent, persistent-template,
|
||||||
aeson,
|
aeson,
|
||||||
feed,
|
feed,
|
||||||
regex-tdfa
|
regex-tdfa,
|
||||||
|
disk-free-space
|
||||||
CC-Options: -Wall
|
CC-Options: -Wall
|
||||||
GHC-Options: -Wall -fno-warn-tabs
|
GHC-Options: -Wall -fno-warn-tabs
|
||||||
Extensions: PackageImports
|
Extensions: PackageImports
|
||||||
|
|
Loading…
Reference in a new issue