SystemHelper.cpp

Go to the documentation of this file.

#include <SystemHelper.h>

#ifdef HAVE_WINDOWS
#include <Windows.h>
#endif
#ifdef HAVE_LINUX
#include <sys/statvfs.h>
#endif

SystemHelper::SystemHelper() {

}

SystemHelper::~SystemHelper() {

}

DiskStatus SystemHelper::getDiskStatus(const std::string & path) const {
    DiskStatus out = {0,0};
#ifdef HAVE_WINDOWS
    ULARGE_INTEGER freeBytesAvailable, totalNumberOfBytes, totalFreeBytes;
    /* for conversion macros see: http://msdn.microsoft.com/en-us/library/87zae4a3%28VS.71%29.aspx
     * CA2CT should be const ascii tring to const tchar string
     */
#ifdef _UNICODE
    int wchars_num = MultiByteToWideChar( CP_UTF8 , 0 , path.c_str() , -1, NULL , 0 );
    wchar_t* wstr = new wchar_t[wchars_num];
    MultiByteToWideChar( CP_UTF8 , 0 , path.c_str() , -1, wstr , wchars_num );
    if(GetDiskFreeSpaceEx(wstr, &freeBytesAvailable, &totalNumberOfBytes, &totalFreeBytes)) {
        out.totalSize = totalNumberOfBytes.QuadPart;
        out.freeSize = freeBytesAvailable.QuadPart;
    }
    delete wstr;
#else
    if(GetDiskFreeSpaceEx(path.c_str(), &freeBytesAvailable, &totalNumberOfBytes, &totalFreeBytes)) {
        out.totalSize = totalNumberOfBytes.QuadPart;
        out.freeSize = freeBytesAvailable.QuadPart;
    }
#endif
#endif
#ifdef HAVE_LINUX
    struct statvfs fiData;
    if(statvfs(path.c_str(), &fiData) >= 0) {
        out.totalSize = fiData.f_blocks;
        out.totalSize *= fiData.f_frsize;
        out.freeSize = fiData.f_bavail;
        out.freeSize *= fiData.f_frsize;
    }
#endif
    return out;
}

double SystemHelper::getDiskFreePercent(const std::string & path) const {
    DiskStatus status = getDiskStatus(path);
    if(status.totalSize == 0) {
        return 0;
    }
    return ((double)(status.freeSize)) / ((double)(status.totalSize));
}