DeviceDisk Class Reference

#include <devicedisk.h>

Collaboration diagram for DeviceDisk:

Classes
struct	info

Public Member Functions
	DeviceDisk ()
vector< info >	getInfo ()
double	getAll (string path)
uint64_t	getSize (string path)
uint64_t	getUsed (string path)
uint64_t	getFree (string path)
double	getAll ()
uint64_t	getSize ()
uint64_t	getUsed ()
uint64_t	getFree ()
double	getFreeGiB (string path)
double	getUsedGiB (string path)
double	getSizeGiB (string path)
double	getFreeGiB ()
double	getUsedGiB ()
double	getSizeGiB ()

Public Attributes
uint64_t	Size
uint64_t	Used
uint64_t	Free
double	SizeGiB
double	UsedGiB
double	FreeGiB
double	FreePercent

Constructor & Destructor Documentation

DeviceDisk::DeviceDisk

(

)

{

}

Member Function Documentation

vector< DeviceDisk::info > DeviceDisk::getInfo

(

)

{
    vector <info> result;
    info tinfo;
#ifdef COSMOS_LINUX_OS
    string tdata;
    uint64_t tsize;

    int32_t iretn = data_execute("lsblk -fbl -o SIZE,MOUNTPOINT", tdata);
    vector<string> lines;
    if(iretn<0) {
        return result;
    } else  {
        lines = string_split(tdata, "\n");
    }
    for (string line : lines)
    {
        char tmount[50];
        if (sscanf(line.c_str(), "%lu %s\n", &tsize, tmount) == 2 && tmount[0] == '/')
        {
            tinfo.mount = tmount;
            tinfo.size = tsize;
            tinfo.used = getUsed(tinfo.mount);
            tinfo.free = tinfo.size - tinfo.used;
            result.push_back(tinfo);
        }
    }
//    FILE *fp = popen("/bin/lsblk -fbl -o SIZE,MOUNTPOINT", "r");
//    char tdata[100];
//    uint64_t tsize;
//    while ((fgets(tdata, 100, fp)) == tdata)
//    {
//        char tmount[50];
//        if (sscanf(tdata, "%lu %s\n", &tsize, tmount) == 2 && tmount[0] == '/')
//        {
//            tinfo.mount = tmount;
//            tinfo.size = tsize;
//            tinfo.used = getUsed(tinfo.mount);
//            tinfo.free = tinfo.size - tinfo.used;
//            result.push_back(tinfo);
//        }
//    }
#endif
#ifdef COSMOS_WIN_OS
    getAll();
    tinfo.size = Size;
    tinfo.free = Free;
    tinfo.used = Used;
    tinfo.mount = "c:/";
    result.push_back(tinfo);
#endif
#ifdef COSMOS_MAC_OS
    getAll();
    tinfo.size = Size;
    tinfo.mount = "/";
    result.push_back(tinfo);
#endif
    return result;
}

double DeviceDisk::getAll

(

string

path

)

{
    Size = 0;
    Free = 0;
    Used = 0;
    FreePercent = 0;

#if defined COSMOS_LINUX_OS
    struct statvfs buf;

    statvfs(path.c_str(),&buf);

    // disk size in bytes
    // must add the casting otherwise there will be problems
    Size = (uint64_t)buf.f_frsize * (uint64_t)buf.f_blocks;
    Free = (uint64_t)buf.f_frsize * (uint64_t)buf.f_bfree;

#endif

#if defined COSMOS_MAC_OS
    // Open disk
     uint32_t dev = open("/dev/disk0", O_RDONLY);

     if (dev == -1) {
         perror("Failed to open disk");
         return -1;
     }

     uint64_t sector_count = 0;
     // Query the number of sectors on the disk
     ioctl(dev, DKIOCGETBLOCKCOUNT, &sector_count);

     uint32_t sector_size = 0;
     // Query the size of each sector
     ioctl(dev, DKIOCGETBLOCKSIZE, &sector_size);

     uint64_t disk_size = sector_count * sector_size;
     printf("%ld", disk_size);

     Size = disk_size;

     return 0;
#endif

#if defined COSMOS_WIN_OS
    uint64_t freeSpace;

    GetDiskFreeSpaceExA( (LPCSTR) path.c_str(),
                        (PULARGE_INTEGER)&freeSpace,
                        (PULARGE_INTEGER)&Size,
                        (PULARGE_INTEGER)&Free);

#endif

    Used = Size - Free;
    FreePercent = (double)Free / Used;

    // convert to GiB
    SizeGiB = Size/GiB;
    FreeGiB = Free/GiB;
    UsedGiB = Used/GiB;

    return FreePercent;
}

uint64_t DeviceDisk::getSize

(

string

path

)

{
    getAll(path);
    return Size;
}

uint64_t DeviceDisk::getUsed

(

string

path

)

{
    getAll(path);
    return (Used);
}

uint64_t DeviceDisk::getFree

(

string

path

)

{
    getAll(path);

    return Free;
}

double DeviceDisk::getAll

(

)

{
#if defined COSMOS_LINUX_OS
    return getAll("/");
#endif

#if defined COSMOS_WIN_OS
    return getAll("C:");
#endif
}

uint64_t DeviceDisk::getSize

(

)

{
#if defined COSMOS_LINUX_OS
    return getSize("/");
#endif

#if defined COSMOS_WIN_OS
    return getSize("C:");
#endif
}

uint64_t DeviceDisk::getUsed

(

)

{
#if defined COSMOS_LINUX_OS
    return getUsed("/");
#endif

#if defined COSMOS_WIN_OS
    return getUsed("C:");
#endif
}

uint64_t DeviceDisk::getFree

(

)

{
#if defined COSMOS_LINUX_OS
    return getFree("/");
#endif

#if defined COSMOS_WIN_OS
    return getFree("C:");
#endif
}

double DeviceDisk::getFreeGiB

(

string

path

)

{
    // convert from Byte to GiB
    return (double)getFree(path)/GiB;
}

double DeviceDisk::getUsedGiB

(

string

path

)

{
    // convert from Byte to GiB
    return (double)getUsed(path)/GiB;
}

double DeviceDisk::getSizeGiB

(

string

path

)

{
    // convert from Byte to GiB
    return (double)getSize(path)/GiB;
}

double DeviceDisk::getFreeGiB

(

)

{
    // convert from Byte to GiB
    return (double)getFree()/GiB;
}

double DeviceDisk::getUsedGiB

(

)

{
    // convert from Byte to GiB
    return (double)getUsed()/GiB;
}

double DeviceDisk::getSizeGiB

(

)

{
    // convert from Byte to GiB
    return (double)getSize()/GiB;
}

Member Data Documentation

uint64_t DeviceDisk::Size

uint64_t DeviceDisk::Used

uint64_t DeviceDisk::Free

double DeviceDisk::SizeGiB

double DeviceDisk::UsedGiB

double DeviceDisk::FreeGiB

double DeviceDisk::FreePercent

---

The documentation for this class was generated from the following files:

• devicedisk.h
• devicedisk.cpp