include/drive.h

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

include/reHDD.h

@@ -31,7 +31,7 @@
 #endif
 
 // 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 ZERO_CHECK   // check drive after shred if all bytes are zero, show alert if this fails
 

src/reHDD.cpp

@@ -306,41 +306,51 @@ void reHDD::filterNewDrives(list<Drive> *plistOldDrives, list<Drive> *plistNewDr
  * \param	pointer of list <Drive>* plistDrives
  * \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);
     }
 
-    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->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);
-            // 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);
 }
 
 /**