mirror of
https://github.com/otakuto/crazydiskinfo.git
synced 2025-07-25 10:28:41 +00:00
add main.cpp
This commit is contained in:
otakuto
parent
20f41c206e
commit
ce3a3302e6
1 changed files with 414 additions and 0 deletions
414
main.cpp
Normal file
414
main.cpp
Normal file
|
|
@ -0,0 +1,414 @@
|
|||
#include <iostream>
|
||||
#include <array>
|
||||
#include <vector>
|
||||
#include <algorithm>
|
||||
#include <functional>
|
||||
#include <locale.h>
|
||||
#include <ncurses.h>
|
||||
#include <atasmart.h>
|
||||
#include <sys/signal.h>
|
||||
#include <dirent.h>
|
||||
|
||||
constexpr int const STATUS_WIDTH = 80;
|
||||
|
||||
constexpr int const DEVICE_BAR_HEIGHT = 4;
|
||||
|
||||
constexpr int const VERSION_HEIGHT = 1;
|
||||
|
||||
int width;
|
||||
int height;
|
||||
|
||||
enum class Health
|
||||
{
|
||||
Good,
|
||||
Caution,
|
||||
Bad
|
||||
};
|
||||
|
||||
class Attribute
|
||||
{
|
||||
public:
|
||||
uint8_t id;
|
||||
std::string name;
|
||||
uint8_t current;
|
||||
uint8_t worst;
|
||||
uint8_t threshold;
|
||||
uint64_t raw;
|
||||
};
|
||||
|
||||
class SMART
|
||||
{
|
||||
public:
|
||||
std::string deviceName;
|
||||
std::string model;
|
||||
std::string firmware;
|
||||
std::string serial;
|
||||
uint64_t size;
|
||||
uint64_t powerOnCount;
|
||||
uint64_t powerOnHour;
|
||||
double temperature;
|
||||
|
||||
std::vector<Attribute> attribute;
|
||||
|
||||
SMART(std::string deviceName)
|
||||
:
|
||||
deviceName(deviceName)
|
||||
{
|
||||
SkDisk * skdisk;
|
||||
sk_disk_open(deviceName.c_str(), &skdisk);
|
||||
sk_disk_smart_read_data(skdisk);
|
||||
|
||||
const SkIdentifyParsedData * data;
|
||||
sk_disk_identify_parse(skdisk, &data);
|
||||
model = data->model;
|
||||
firmware = data->firmware;
|
||||
serial = data->serial;
|
||||
|
||||
uint64_t value;
|
||||
sk_disk_get_size(skdisk, &value);
|
||||
size = value;
|
||||
sk_disk_smart_get_power_cycle(skdisk, &value);
|
||||
powerOnCount = value;
|
||||
sk_disk_smart_get_power_on(skdisk, &value);
|
||||
powerOnHour = value / (1000llu * 60llu * 60llu);
|
||||
sk_disk_smart_get_temperature(skdisk, &value);
|
||||
temperature = (double)(value - 273150llu) / 1000.0;
|
||||
|
||||
sk_disk_smart_parse_attributes(skdisk, [](SkDisk * skdisk, SkSmartAttributeParsedData const * data, void * userdata)
|
||||
{
|
||||
auto attribute = reinterpret_cast<std::vector<Attribute> *>(userdata);
|
||||
Attribute attr = {};
|
||||
attr.id = data->id;
|
||||
attr.name = data->name;
|
||||
attr.current = data->current_value;
|
||||
attr.worst = data->worst_value;
|
||||
attr.threshold = data->threshold;
|
||||
for (int i = 0; i < 6; ++i)
|
||||
{
|
||||
attr.raw += data->raw[i] << (8 * i);
|
||||
}
|
||||
attribute->push_back(attr);
|
||||
}, &attribute);
|
||||
|
||||
sk_disk_free(skdisk);
|
||||
}
|
||||
};
|
||||
|
||||
Health temperatureToHealth(double temperature)
|
||||
{
|
||||
if (temperature < 50)
|
||||
{
|
||||
return Health::Good;
|
||||
}
|
||||
else if (temperature < 55)
|
||||
{
|
||||
return Health::Caution;
|
||||
}
|
||||
else
|
||||
{
|
||||
return Health::Bad;
|
||||
}
|
||||
}
|
||||
|
||||
Health attributeToHealth(Attribute const & attribute)
|
||||
{
|
||||
if ((attribute.threshold != 0) && (attribute.current < attribute.threshold))
|
||||
{
|
||||
return Health::Bad;
|
||||
}
|
||||
else if (((attribute.id == 0x05) || (attribute.id == 0xC5) || (attribute.id == 0xC6)) && (attribute.raw != 0))
|
||||
{
|
||||
return Health::Caution;
|
||||
}
|
||||
else
|
||||
{
|
||||
return Health::Good;
|
||||
}
|
||||
}
|
||||
|
||||
Health smartToHealth(SMART const & smart)
|
||||
{
|
||||
return attributeToHealth(*std::max_element(smart.attribute.cbegin(), smart.attribute.cend(), [](auto lhs, auto rhs)
|
||||
{
|
||||
return static_cast<int>(attributeToHealth(lhs)) < static_cast<int>(attributeToHealth(rhs));
|
||||
}));
|
||||
}
|
||||
|
||||
std::string healthToString(Health health)
|
||||
{
|
||||
switch (health)
|
||||
{
|
||||
case Health::Good:
|
||||
return "Good";
|
||||
case Health::Caution:
|
||||
return "Caution";
|
||||
case Health::Bad:
|
||||
return "Bad";
|
||||
default:
|
||||
return "Unknown";
|
||||
}
|
||||
}
|
||||
|
||||
void drawVersion(WINDOW * window)
|
||||
{
|
||||
wresize(window, VERSION_HEIGHT, width);
|
||||
|
||||
wattrset(window, COLOR_PAIR(4));
|
||||
mvwhline(window, 0, 0, '-', width);
|
||||
wattroff(window, COLOR_PAIR(4));
|
||||
|
||||
wattrset(window, COLOR_PAIR(8));
|
||||
mvwprintw(window, 0, (width - sizeof(" CrazyDiskInfo-1.0.0 ")) / 2, " CrazyDiskInfo-1.0.0 ");
|
||||
wattroff(window, COLOR_PAIR(8));
|
||||
|
||||
wnoutrefresh(window);
|
||||
}
|
||||
|
||||
void drawDeviceBar(WINDOW * window, std::vector<SMART> const & smartList, int select)
|
||||
{
|
||||
int x = 0;
|
||||
for (int i = 0; i < static_cast<int>(smartList.size()); ++i)
|
||||
{
|
||||
wattrset(window, COLOR_PAIR(1 + static_cast<int>(smartToHealth(smartList[i]))));
|
||||
mvwprintw(window, 0, x, "%-7s", healthToString(smartToHealth(smartList[i])).c_str());
|
||||
wattroff(window, COLOR_PAIR(1 + static_cast<int>(smartToHealth(smartList[i]))));
|
||||
|
||||
wattrset(window, COLOR_PAIR(1 + static_cast<int>(temperatureToHealth(smartList[i].temperature))));
|
||||
mvwprintw(window, 1, x, "%.1f ", smartList[i].temperature);
|
||||
waddch(window, ACS_DEGREE);
|
||||
waddstr(window, "C");
|
||||
wattroff(window, COLOR_PAIR(1 + static_cast<int>(temperatureToHealth(smartList[i].temperature))));
|
||||
|
||||
if (i == select)
|
||||
{
|
||||
wattrset(window, COLOR_PAIR(4) | A_BOLD);
|
||||
mvwprintw(window, 2, x, smartList[i].deviceName.c_str());
|
||||
wattroff(window, COLOR_PAIR(4) | A_BOLD);
|
||||
|
||||
wattrset(window, COLOR_PAIR(4));
|
||||
mvwhline(window, 3, x, '-', smartList[i].deviceName.length());
|
||||
wattroff(window, COLOR_PAIR(4));
|
||||
}
|
||||
else
|
||||
{
|
||||
mvwprintw(window, 2, x, smartList[i].deviceName.c_str());
|
||||
mvwhline(window, 3, x, ' ', smartList[i].deviceName.length());
|
||||
}
|
||||
x += smartList[i].deviceName.length() + 1;
|
||||
}
|
||||
pnoutrefresh(window, 0, 0, 1, 0, DEVICE_BAR_HEIGHT, width - 1);
|
||||
}
|
||||
|
||||
void drawStatus(WINDOW * window, SMART const & smart)
|
||||
{
|
||||
wresize(window, 10 + smart.attribute.size(), STATUS_WIDTH);
|
||||
wborder(window, '|', '|', '-', '-', '+', '+', '+', '+');
|
||||
{
|
||||
std::vector<std::string> unit = {{"Byte", "KiB", "MiB", "GiB", "TiB", "PiB", "EiB", "ZiB", "YiB"}};
|
||||
int u = 0;
|
||||
double size = smart.size;
|
||||
while (true)
|
||||
{
|
||||
double old = size;
|
||||
size /= 1024;
|
||||
if (size < 1.0)
|
||||
{
|
||||
size = old;
|
||||
break;
|
||||
}
|
||||
++u;
|
||||
}
|
||||
char s[STATUS_WIDTH];
|
||||
int len = snprintf(s, STATUS_WIDTH, " %s [%.1f %s] ", smart.model.c_str(), size, unit[u].c_str());
|
||||
|
||||
wattrset(window, COLOR_PAIR(4) | A_BOLD);
|
||||
mvwprintw(window, 0, (STATUS_WIDTH - len) / 2, "%s", s);
|
||||
wattroff(window, COLOR_PAIR(4) | A_BOLD);
|
||||
}
|
||||
|
||||
wattrset(window, COLOR_PAIR(4));
|
||||
mvwprintw(window, 2, (int)(STATUS_WIDTH * (1.0 / 5)), "Firmware:");
|
||||
wattroff(window, COLOR_PAIR(4));
|
||||
wattrset(window, COLOR_PAIR(4) | A_BOLD);
|
||||
wprintw(window, " %s", smart.firmware.c_str());
|
||||
wattroff(window, COLOR_PAIR(4) | A_BOLD);
|
||||
|
||||
wattrset(window, COLOR_PAIR(4));
|
||||
mvwprintw(window, 3, (int)(STATUS_WIDTH * (1.0 / 5)), "Serial: ");
|
||||
wattroff(window, COLOR_PAIR(4));
|
||||
wattrset(window, COLOR_PAIR(4) | A_BOLD);
|
||||
wprintw(window, " %s", smart.serial.c_str());
|
||||
wattroff(window, COLOR_PAIR(4) | A_BOLD);
|
||||
|
||||
wattrset(window, COLOR_PAIR(4));
|
||||
mvwprintw(window, 1, 1, "Status");
|
||||
wattroff(window, COLOR_PAIR(4));
|
||||
wattrset(window, COLOR_PAIR(1 + static_cast<int>(smartToHealth(smart))));
|
||||
mvwprintw(window, 2, 2, "+--------+");
|
||||
mvwprintw(window, 3, 2, "| |");
|
||||
mvwprintw(window, 4, 2, "+--------+");
|
||||
mvwprintw(window, 3, 2 + ((sizeof("| |") - healthToString(smartToHealth(smart)).length()) / 2), "%s", healthToString(smartToHealth(smart)).c_str());
|
||||
wattroff(window, COLOR_PAIR(1 + static_cast<int>(smartToHealth(smart))));
|
||||
|
||||
wattrset(window, COLOR_PAIR(4));
|
||||
mvwprintw(window, 5, 1, "Temperature");
|
||||
wattroff(window, COLOR_PAIR(4));
|
||||
wattrset(window, COLOR_PAIR(1 + static_cast<int>(temperatureToHealth(smart.temperature))));
|
||||
mvwprintw(window, 6, 2, " %0.1f ", smart.temperature);
|
||||
waddch(window, ACS_DEGREE);
|
||||
waddstr(window, "C ");
|
||||
wattroff(window, COLOR_PAIR(1 + static_cast<int>(temperatureToHealth(smart.temperature))));
|
||||
|
||||
wattrset(window, COLOR_PAIR(4));
|
||||
mvwprintw(window, 2, (int)(STATUS_WIDTH * (3.0 / 5)), "Power On Count:");
|
||||
wattrset(window, COLOR_PAIR(4) | A_BOLD);
|
||||
wprintw(window, " %llu ", smart.powerOnCount);
|
||||
wattroff(window, COLOR_PAIR(4) | A_BOLD);
|
||||
wattrset(window, COLOR_PAIR(4));
|
||||
wprintw(window, "count");
|
||||
|
||||
wattrset(window, COLOR_PAIR(4));
|
||||
mvwprintw(window, 3, (int)(STATUS_WIDTH * (3.0 / 5)), "Power On Hours:");
|
||||
wattroff(window, COLOR_PAIR(4));
|
||||
wattrset(window, COLOR_PAIR(4) | A_BOLD);
|
||||
wprintw(window, " %llu ", smart.powerOnHour);
|
||||
wattroff(window, COLOR_PAIR(4) | A_BOLD);
|
||||
wattrset(window, COLOR_PAIR(4));
|
||||
wprintw(window, "hours");
|
||||
wattroff(window, COLOR_PAIR(4));
|
||||
|
||||
wattrset(window, COLOR_PAIR(7));
|
||||
mvwprintw(window, 8, 1, " Status ID AttributeName Current Worst Threshold Raw Values ");
|
||||
wattroff(window, COLOR_PAIR(7));
|
||||
for (int i = 0; i < static_cast<int>(smart.attribute.size()); ++i)
|
||||
{
|
||||
wattrset(window, COLOR_PAIR(4 + static_cast<int>(attributeToHealth(smart.attribute[i]))));
|
||||
mvwprintw(window, 9 + i, 1, " %-7s %02X %-28s %7d %5d %9d %012X ", healthToString(attributeToHealth(smart.attribute[i])).c_str(), smart.attribute[i].id, smart.attribute[i].name.c_str(), smart.attribute[i].current, smart.attribute[i].worst, smart.attribute[i].threshold, smart.attribute[i].raw);
|
||||
wattroff(window, COLOR_PAIR(4 + static_cast<int>(attributeToHealth(smart.attribute[i]))));
|
||||
}
|
||||
pnoutrefresh(window, 0, 0,
|
||||
5, std::max(0, (width - STATUS_WIDTH) / 2),
|
||||
std::min(height - 1, 5 + 10 + static_cast<int>(smart.attribute.size())), std::min(width - 1, std::max(0, (width - STATUS_WIDTH) / 2) + STATUS_WIDTH));
|
||||
}
|
||||
|
||||
std::function<void(void)> update;
|
||||
void actionWINCH(int)
|
||||
{
|
||||
clear();
|
||||
endwin();
|
||||
refresh();
|
||||
update();
|
||||
}
|
||||
|
||||
int main()
|
||||
{
|
||||
setlocale(LC_ALL, "");
|
||||
initscr();
|
||||
cbreak();
|
||||
noecho();
|
||||
curs_set(0);
|
||||
getmaxyx(stdscr, height, width);
|
||||
|
||||
start_color();
|
||||
init_pair(1, COLOR_BLACK, COLOR_CYAN);
|
||||
init_pair(2, COLOR_BLACK, COLOR_YELLOW);
|
||||
init_pair(3, COLOR_WHITE, COLOR_RED);
|
||||
init_pair(4, COLOR_CYAN, COLOR_BLACK);
|
||||
init_pair(5, COLOR_BLACK, COLOR_YELLOW);
|
||||
init_pair(6, COLOR_WHITE, COLOR_RED);
|
||||
init_pair(7, COLOR_BLACK, COLOR_GREEN);
|
||||
init_pair(8, COLOR_YELLOW, COLOR_BLACK);
|
||||
|
||||
int select = 0;
|
||||
std::vector<SMART> smartList;
|
||||
auto dir = opendir("/sys/block");
|
||||
while (auto e = readdir(dir))
|
||||
{
|
||||
if (std::string(".") != std::string(e->d_name) &&
|
||||
std::string("..") != std::string(e->d_name) &&
|
||||
std::string("ram") != std::string(e->d_name).substr(0,3) &&
|
||||
std::string("loop") != std::string(e->d_name).substr(0,4))
|
||||
{
|
||||
SkDisk * skdisk;
|
||||
SkBool b;
|
||||
int f = sk_disk_open((std::string("/dev/") + std::string(e->d_name)).c_str(), &skdisk);
|
||||
if (f < 0)
|
||||
{
|
||||
continue;
|
||||
}
|
||||
sk_disk_smart_is_available(skdisk, &b);
|
||||
sk_disk_free(skdisk);
|
||||
if (b)
|
||||
{
|
||||
smartList.push_back(SMART(std::string("/dev/") + std::string(e->d_name)));
|
||||
}
|
||||
}
|
||||
}
|
||||
std::sort(smartList.begin(), smartList.end(), [](auto lhs, auto rhs){return lhs.deviceName < rhs.deviceName;});
|
||||
|
||||
WINDOW * windowVersion;
|
||||
windowVersion = newwin(1, width, 0, 0);
|
||||
|
||||
WINDOW * windowDeviceBar;
|
||||
{
|
||||
int x = 0;
|
||||
for (auto && e : smartList)
|
||||
{
|
||||
x += e.deviceName.length() + 1;
|
||||
}
|
||||
windowDeviceBar = newpad(DEVICE_BAR_HEIGHT, x);
|
||||
keypad(windowDeviceBar, true);
|
||||
}
|
||||
|
||||
WINDOW * windowDeviceStatus;
|
||||
windowDeviceStatus = newpad(10 + smartList[select].attribute.size(), STATUS_WIDTH);
|
||||
|
||||
update = [&]()
|
||||
{
|
||||
getmaxyx(stdscr, height, width);
|
||||
drawVersion(windowVersion);
|
||||
drawDeviceBar(windowDeviceBar, smartList, select);
|
||||
drawStatus(windowDeviceStatus, smartList[select]);
|
||||
doupdate();
|
||||
};
|
||||
update();
|
||||
{
|
||||
struct sigaction s = {{actionWINCH}};
|
||||
sigaction(SIGWINCH, &s, nullptr);
|
||||
}
|
||||
while(true)
|
||||
{
|
||||
switch (wgetch(windowDeviceBar))
|
||||
{
|
||||
case KEY_HOME:
|
||||
select = 0;
|
||||
update();
|
||||
break;
|
||||
|
||||
case KEY_END:
|
||||
select = static_cast<int>(smartList.size()) - 1;
|
||||
update();
|
||||
break;
|
||||
|
||||
case KEY_LEFT:
|
||||
case 'h':
|
||||
select = std::max(select - 1, 0);
|
||||
update();
|
||||
break;
|
||||
|
||||
case KEY_RIGHT:
|
||||
case 'l':
|
||||
select = std::min(select + 1, static_cast<int>(smartList.size()) - 1);
|
||||
update();
|
||||
break;
|
||||
|
||||
case 'q':
|
||||
endwin();
|
||||
return 0;
|
||||
|
||||
default:
|
||||
break;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
Loading…
Add table
Add a link
Reference in a new issue