localhorst/reHDD
1
0
Fork 0
Code Issues 5 Pull Requests Releases 10 Wiki Activity

get connection type

Browse Source
This commit is contained in:
Hendrik Schutter 2025-06-21 20:19:30 +02:00
parent d8584486b9
commit d76cb1a195
3 changed files with 48 additions and 30 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
include/drive.h
View File

@ -25,6 +25,14 @@ public:
FROZEN FROZEN
} state; } state;
enum ConnectionType
{
UNKNOWN,
USB,
SATA,
NVME
} connectionType;
struct struct
{ {
time_t u32ShredTimeDelta; time_t u32ShredTimeDelta;

2
include/reHDD.h
View File

@ -31,7 +31,7 @@
#endif #endif
// Logic // Logic
// #define DRYRUN // don't touch the drives #define DRYRUN // don't touch the drives
#define FROZEN_ALERT // show alert if drive is frozen #define FROZEN_ALERT // show alert if drive is frozen
#define ZERO_CHECK // check drive after shred if all bytes are zero, show alert if this fails #define ZERO_CHECK // check drive after shred if all bytes are zero, show alert if this fails

58
src/reHDD.cpp
View File

@ -306,41 +306,51 @@ void reHDD::filterNewDrives(list<Drive> *plistOldDrives, list<Drive> *plistNewDr
* \param pointer of list <Drive>* plistDrives * \param pointer of list <Drive>* plistDrives
* \return void * \return void
*/ */
void reHDD::searchDrives(list<Drive> *plistDrives) void reHDD::searchDrives(std::list<Drive> *plistDrives)
{ {
// Logger::logThis()->info("--> search drives <--"); FILE *fp = popen("lsblk -d -n -o NAME,TRAN", "r");
char *cLine = NULL; if (!fp)
size_t len = 0;
FILE *outputfileHwinfo = popen("lsblk -e 11 -d -o NAME", "r");
if (outputfileHwinfo == NULL)
{ {
Logger::logThis()->error("Unable to scan attached drives"); Logger::logThis()->error("Unable to execute lsblk to scan drives");
exit(EXIT_FAILURE); exit(EXIT_FAILURE);
} }
while ((getline(&cLine, &len, outputfileHwinfo)) != -1) char line[256];
while (fgets(line, sizeof(line), fp))
{ {
if (string(cLine).length() == 4) std::string devName, transport;
{ std::istringstream iss(line);
Drive *tmpDrive = new Drive("/dev/" + string(cLine).substr(0, 3)); iss >> devName >> transport;
if (devName.empty())
continue;
Drive *tmpDrive = new Drive("/dev/" + devName);
tmpDrive->state = Drive::NONE; tmpDrive->state = Drive::NONE;
tmpDrive->bIsOffline = false; tmpDrive->bIsOffline = false;
// Set connection type
if (transport == "sata")
tmpDrive->connectionType = Drive::SATA;
else if (transport == "usb")
tmpDrive->connectionType = Drive::USB;
else if (transport == "nvme")
tmpDrive->connectionType = Drive::NVME;
else
tmpDrive->connectionType = Drive::UNKNOWN;
plistDrives->push_back(*tmpDrive); plistDrives->push_back(*tmpDrive);
// Logger::logThis()->info("SATA drive found: " + tmpDrive->getPath());
Logger::logThis()->info(
"Drive found: " + tmpDrive->getPath() +
" (type: " +
(tmpDrive->connectionType == Drive::USB ? "USB" : tmpDrive->connectionType == Drive::SATA ? "SATA"
: tmpDrive->connectionType == Drive::NVME ? "NVME"
: "UNKNOWN") +
")");
} }
if (string(cLine).length() == 8) pclose(fp);
{
Drive *tmpDrive = new Drive("/dev/" + string(cLine).substr(0, 7));
tmpDrive->state = Drive::NONE;
tmpDrive->bIsOffline = false;
plistDrives->push_back(*tmpDrive);
// Logger::logThis()->info("NVME drive found: " + tmpDrive->getPath());
}
}
pclose(outputfileHwinfo);
} }
/** /**