40 lines
831 B
C++
40 lines
831 B
C++
#include "diskspace.h"
|
|
|
|
namespace bf = boost::filesystem;
|
|
|
|
#if !defined(_MSC_VER)
|
|
|
|
#include <sys/statvfs.h>
|
|
#include <cerrno>
|
|
|
|
namespace cpputils {
|
|
|
|
uint64_t free_disk_space_in_bytes(const bf::path& location) {
|
|
struct statvfs stat {};
|
|
int result = ::statvfs(location.string().c_str(), &stat);
|
|
if (0 != result) {
|
|
throw std::runtime_error("Error calling statvfs(). Errno: " + std::to_string(errno));
|
|
}
|
|
return stat.f_frsize*stat.f_bavail;
|
|
}
|
|
|
|
}
|
|
|
|
#else
|
|
|
|
#include <Windows.h>
|
|
|
|
namespace cpputils {
|
|
|
|
uint64_t free_disk_space_in_bytes(const bf::path& location) {
|
|
ULARGE_INTEGER freeBytes;
|
|
if (!GetDiskFreeSpaceEx(location.string().c_str(), &freeBytes, nullptr, nullptr)) {
|
|
throw std::runtime_error("Error calling GetDiskFreeSpaceEx(). Error code: " + std::to_string(GetLastError()));
|
|
}
|
|
return freeBytes.QuadPart;
|
|
}
|
|
|
|
}
|
|
|
|
#endif
|