Merge pull request 'Ignore system drives as live image' (#71) from bugfix/systemdrive_live into master

Reviewed-on: #71

.gitignore

@@ -41,7 +41,10 @@
 
 reHDD
 
-reHDD.log
+*.log
+*.ods
+*.txt
 
+.vscode/
 
 ignoreDrives.conf

.gitmodules

@@ -0,0 +1,3 @@
+[submodule "tfnoisegen"]
+	path = tfnoisegen
+	url = https://git.mosad.xyz/localhorst/tfnoisegen.git

README.md

@@ -1,18 +1,33 @@
 # reHDD
 
-## Useful for:
+## Features:
-* checking new drives for the first time
+* show S.M.A.R.T values of attached drives
-* checking used drives for their next live
+* checking used drives for their next live based on threshold limits
+* delete a drive instant with wipefs
 * deleting a drive securely via overwriting
+* only needs a display and keyboard
+* process multiple drives at once
+
+## Download USB Image ##
+See reHDD-Bootable how the live image created: https://git.mosad.xyz/localhorst/reHDD-Bootable
+
+Use [Etcher](https://www.balena.io/etcher/#download) or `dd` to create an bootable USB drive .
 
 ## Screenshot 
-![alt text](https://git.mosad.xyz/localhorst/reHDD/raw/commit/42bc26eac95429e20c0f0d59f684dfec0d600e75/doc/screenshot.png "Screenshot")
+![Screenshot of reHDD with multiple drives in different states](https://git.mosad.xyz/localhorst/reHDD/raw/commit/c40dfe2cbb8f86490b49caf82db70a10015f06f9/doc/screenshot.png "Screenshot")
 
 ## Debian Build Notes
 
 * `apt-get install ncurses-dev git make g++`
+* `git submodule init`
+* `git submodule update`
 * `make release`
 
+## Enable Label Printer ##
+
+Just install [reHDDPrinter](https://git.mosad.xyz/localhorst/reHDDPrinter).
+No further settings needed.
+
 ## Create Standalone with Debian 11
 
 Instructions how to create a standalone machine that boots directly to reHDD. This is aimed for production use, like several drives a day shredding.
@@ -22,13 +37,17 @@ Instructions how to create a standalone machine that boots directly to reHDD. Th
 * Upload reHDD log every 12h if wanted
 
 ### Software requirements
-* `apt-get install hwinfo smartmontools curl`
+* `apt-get install hwinfo smartmontools curl htop sudo`
-
 
 ### Installation
 
 clone this repo into /root/
 
+```
+git submodule init
+git submodule update
+```
+
 `cd /root/reHDD/`
 
 `make release`

astyle.sh

@@ -1,17 +0,0 @@
-#! /bin/bash
-
-echo starting astyle for $PWD
-
-astyle --style=gnu src/*.cpp
-rm -f src/*.orig
-
-astyle --style=gnu src/logger/*.cpp
-rm -f src/logger/*.orig
-
-astyle --style=gnu include/*.h
-rm -f include/*.orig
-
-astyle --style=gnu include/logger/*.h
-rm -f include//logger/*.orig
-
-echo finished astyle for $PWD

ignoreDrives.conf

@@ -1,2 +1,3 @@
 4673974d
 2cb3dea4
+8ffbc421

include/delete.h

@@ -13,13 +13,11 @@
 class Delete
 {
 protected:
-
 public:
     static void deleteDrive(Drive *drive);
 
 private:
     Delete(void);
-
 };
 
 #endif // DELETE_H_

include/drive.h

@@ -14,21 +14,41 @@ class Drive
 {
 
 public:
-    enum TaskState {NONE,
+    enum TaskState
+    {
+        NONE,
         SHRED_SELECTED,
-                    SHRED_ACTIVE,
+        SHRED_ACTIVE, // shred iterations active
+        CHECK_ACTIVE, // optional checking active
         DELETE_SELECTED,
         DELETE_ACTIVE,
         FROZEN
     } state;
 
-    bool bWasShredded = false;
+    struct
-    bool bWasDeleteted = false;
+    {
+        time_t u32ShredTimeDelta;
+        std::chrono::time_point<std::chrono::system_clock> chronoShredTimestamp;
+        unsigned long ulWrittenBytes;
+        unsigned long ulSpeedMetricBytesWritten;
+    } sShredSpeed;
+
+    bool bWasShredded = false; // all shred iterations done
+    bool bWasShredStarted = false; // shred was atleast once started
+    bool bWasChecked = false;  // all shred iterations and optional checking done
+    bool bWasDeleted = false;
     bool bIsOffline = false;
-    uint32_t u32DriveChecksumAferShredding = 0U;
+    uint32_t u32DriveChecksumAfterShredding = 0U;
 
 private:
     string sPath;
+    time_t u32Timestamp = 0U;          // unix timestamp for detecting a frozen drive
+    double d32TaskPercentage = 0U;     // in percent for Shred (1 to 100)
+    time_t u32TimestampTaskStart = 0U; // unix timestamp for duration of an action
+    time_t u32TaskDuration = 0U;       // time needed to complete the task
+
+    struct
+    {
         string sModelFamily;
         string sModelName;
         string sSerial;
@@ -36,16 +56,13 @@ private:
         uint32_t u32ErrorCount = 0U;
         uint32_t u32PowerOnHours = 0U; // in hours
         uint32_t u32PowerCycles = 0U;
-    time_t u32Timestamp = 0U; //unix timestamp for detecting a frozen drive
+        uint32_t u32Temperature = 0U; // in Fahrenheit, just kidding: degree Celsius
-    double d32TaskPercentage = 0U; //in percent for Shred (1 to 100)
+    } sSmartData;
-    time_t u32TimestampTaskStart = 0U; //unix timestamp for duration of an action
-    time_t u32TaskDuration = 0U; //time needed to complete the task
 
 private:
     void setTimestamp();
 
 protected:
-
 public:
     Drive(string path)
     {
@@ -60,6 +77,7 @@ public:
     uint32_t getErrorCount(void);
     uint32_t getPowerOnHours(void); // in hours
     uint32_t getPowerCycles(void);
+    uint32_t getTemperature(void); // in Fahrenheit, just kidding: degree Celsius
     void checkFrozenDrive(void);
 
     void setDriveSMARTData(string modelFamily,
@@ -68,12 +86,14 @@ public:
                            uint64_t capacity,
                            uint32_t errorCount,
                            uint32_t powerOnHours,
-                            uint32_t powerCycles);
+                           uint32_t powerCycles,
+                           uint32_t temperature);
 
     string sCapacityToText();
     string sErrorCountToText();
     string sPowerOnHoursToText();
     string sPowerCyclesToText();
+    string sTemperatureToText();
 
     void setTaskPercentage(double d32TaskPercentage);
     double getTaskPercentage(void);
@@ -83,7 +103,6 @@ public:
 
     void calculateTaskDuration();
     time_t getTaskDuration();
-
 };
 
 #endif // DRIVE_H_

include/logger/logger.h

@@ -68,7 +68,6 @@ private:
     ~Logger();
 
 public:
-
     void info(string s);
     void warning(string s);
     void error(string s);

include/printer.h

@@ -0,0 +1,56 @@
+/**
+ * @file    printer.h
+ * @brief   Send drive data to printer service using ipc msg queue
+ * @author  Hendrik Schutter
+ * @date    24.11.2022
+ */
+
+#ifndef PRINTER_H_
+#define PRINTER_H_
+
+#include "reHDD.h"
+
+#include <sys/ipc.h>
+#include <sys/msg.h>
+
+#define STR_BUFFER_SIZE 64U
+#define IPC_MSG_QUEUE_KEY 0x1B11193C0
+
+typedef struct
+{
+    char caDriveIndex[STR_BUFFER_SIZE];
+    char caDriveHours[STR_BUFFER_SIZE];
+    char caDriveCycles[STR_BUFFER_SIZE];
+    char caDriveErrors[STR_BUFFER_SIZE];
+    char caDriveShredTimestamp[STR_BUFFER_SIZE];
+    char caDriveShredDuration[STR_BUFFER_SIZE];
+    char caDriveCapacity[STR_BUFFER_SIZE];
+    char caDriveState[STR_BUFFER_SIZE];
+    char caDriveModelFamily[STR_BUFFER_SIZE];
+    char caDriveModelName[STR_BUFFER_SIZE];
+    char caDriveSerialnumber[STR_BUFFER_SIZE];
+    char caDriveReHddVersion[STR_BUFFER_SIZE];
+
+} t_driveData;
+
+typedef struct
+{
+    long msg_queue_type;
+    t_driveData driveData;
+} t_msgQueueData;
+
+class Printer
+{
+protected:
+public:
+    static Printer *getPrinter();
+    void print(Drive *drive);
+
+private:
+    static bool instanceFlag;
+    static Printer *single;
+    int msqid;
+    Printer();
+    ~Printer();
+};
+#endif // PRINTER_H_

include/reHDD.h

@@ -8,19 +8,21 @@
 #ifndef REHDD_H_
 #define REHDD_H_
 
-#define REHDD_VERSION "bV0.2.2"
+#define REHDD_VERSION "V1.1.3"
 
 // Drive handling Settings
 #define WORSE_HOURS 19200    // mark drive if at this limit or beyond
 #define WORSE_POWERUP 10000  // mark drive if at this limit or beyond
+#define WORSE_TEMPERATURE 55 // mark drive if at this limit or beyond
 #define SHRED_ITERATIONS 3U
-#define FROZEN_TIMEOUT 10 //After this timeout (minutes) the drive will be marked as frozen
+#define FROZEN_TIMEOUT 20                           // After this timeout (minutes) the drive will be marked as frozen, if no progress
+#define METRIC_THRESHOLD 3L * 1000L * 1000L * 1000L // calc shred speed with this minimum of time delta
 
 // Logger Settings
 #define LOG_PATH "./reHDD.log"
-#define DESCRIPTION "reHDD - Copyright Hendrik Schutter 2022"
+#define DESCRIPTION "reHDD - Copyright Hendrik Schutter 2024"
 #define DEVICE_ID "generic"
-#define SOFTWARE_VERSION  "alpha"
+#define SOFTWARE_VERSION REHDD_VERSION
 #define HARDWARE_VERSION "generic"
 
 // #define LOG_LEVEL_HIGH  //log everything, like drive scan thread
@@ -29,9 +31,9 @@
 #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_ALERT //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
 
 // IPC pipes
 #define READ 0
@@ -62,11 +64,13 @@ using namespace std;
 #include "shred.h"
 #include "delete.h"
 #include "tui.h"
+#include "printer.h"
 #include "logger/logger.h"
 
 extern Logger *logging;
 
-template <typename T, typename I> T* iterator_to_pointer(I i)
+template <typename T, typename I>
+T *iterator_to_pointer(I i)
 {
     return (&(*i));
 }
@@ -74,21 +78,22 @@ template <typename T, typename I> T* iterator_to_pointer(I i)
 class reHDD
 {
 protected:
-
 public:
     reHDD(void);
     static void app_logic();
 
 private:
-
     static void searchDrives(list<Drive> *plistDrives);
     static void printDrives(list<Drive> *plistDrives);
+    static void startShredAllDrives(list<Drive> *plistDrives);
+    static void updateShredMetrics(list<Drive> *plistDrives);
     static void filterIgnoredDrives(list<Drive> *plistDrives);
+    static void filterInvalidDrives(list<Drive> *plistDrives);
     static void filterNewDrives(list<Drive> *plistOldDrives, list<Drive> *plistNewDrives);
     static void addSMARTData(list<Drive> *plistDrives);
-    static void ThreadScannDevices();
+    static void ThreadScanDevices();
     static void ThreadUserInput();
-    static void ThreadShred();
+    static void ThreadShred(Drive *const pDrive);
     static void ThreadDelete();
     static void ThreadCheckFrozenDrives();
     static void handleArrowKey(TUI::UserInput userInput);
@@ -96,6 +101,7 @@ private:
     static void handleESC();
     static void handleAbort();
     static Drive *getSelectedDrive();
+    static bool getSystemDrive(string &systemDrive);
 };
 
 #endif // REHDD_H_

include/shred.h

@@ -17,11 +17,12 @@
 #include <unistd.h>
 #include <string.h>
 
-#define CHUNK_SIZE 1024*1024*2 //amount of bytes that are overwritten at once --> 2MB
+#define CHUNK_SIZE 1024 * 1024 * 32 // amount of bytes that are overwritten at once --> 32MB
-#define CHUNK_DIMENSION 100U //amount of chunks are read at once from random source
+#define TFNG_DATA_SIZE CHUNK_SIZE   // amount of bytes used by tfng
 
 // #define DEMO_DRIVE_SIZE 1024*1024*256L // 256MB
 // #define DEMO_DRIVE_SIZE 1024*1024*1024L // 1GB
+// #define DEMO_DRIVE_SIZE 5*1024*1024*1024L // 5GB
 // #define DEMO_DRIVE_SIZE 1024*1024*1024*10L // 10GB
 
 typedef int fileDescriptor;
@@ -29,9 +30,7 @@ typedef int fileDescriptor;
 class Shred
 {
 protected:
-
 public:
-
     Shred();
     ~Shred();
     int shredDrive(Drive *drive, int *ipSignalFd);
@@ -39,7 +38,8 @@ public:
 private:
     fileDescriptor randomSrcFileDiscr;
     fileDescriptor driveFileDiscr;
-    unsigned char caChunk[CHUNK_DIMENSION][CHUNK_SIZE];
+    unsigned char caTfngData[TFNG_DATA_SIZE];
+    unsigned char caReadBuffer[CHUNK_SIZE];
     unsigned long ulDriveByteSize;
     unsigned long ulDriveByteOverallCount = 0; // all bytes shredded in all iterations + checking -> used for progress calculation
     double d32Percent = 0.0;
@@ -50,7 +50,6 @@ private:
     unsigned long getDriveSizeInBytes(fileDescriptor file);
     unsigned int uiCalcChecksum(fileDescriptor file, Drive *drive, int *ipSignalFd);
     void cleanup();
-
 };
 
 #endif // SHRED_H_

include/smart.h

@@ -13,28 +13,21 @@
 class SMART
 {
 protected:
-
 public:
     static void readSMARTData(Drive *drive);
 
 private:
     SMART(void);
 
-    static void parseModelFamily(string sLine);
+    static bool parseExitStatus(string sLine, uint8_t &status);
-    static void parseModelName(string sLine);
+    static bool parseModelFamily(string sLine, string &modelFamily);
-    static void parseSerial(string sLine);
+    static bool parseModelName(string sLine, string &modelName);
-    static void parseCapacity(string sLine);
+    static bool parseSerial(string sLine, string &serial);
-    static void parseErrorCount(string sLine);
+    static bool parseCapacity(string sLine, uint64_t &capacity);
-    static void parsePowerOnHours(string sLine);
+    static bool parseErrorCount(string sLine, uint32_t &errorCount);
-    static void parsePowerCycle(string sLine);
+    static bool parsePowerOnHours(string sLine, uint32_t &powerOnHours);
-
+    static bool parsePowerCycles(string sLine, uint32_t &powerCycles);
-    static  string modelFamily;
+    static bool parseTemperature(string sLine, uint32_t &temperature);
-    static  string modelName;
-    static  string serial;
-    static  uint64_t capacity;
-    static  uint32_t errorCount;
-    static  uint32_t powerOnHours;
-    static  uint32_t powerCycle;
 };
 
 #endif // SMART_H_

include/tui.h

@@ -12,17 +12,27 @@
 
 #define COLOR_AREA_STDSCR 1
 #define COLOR_AREA_OVERVIEW 2
-#define COLOR_AREA_ENTRY 3
+#define COLOR_AREA_ENTRY_EVEN 3
-#define COLOR_AREA_ENTRY_SELECTED 4
+#define COLOR_AREA_ENTRY_ODD 4
-#define COLOR_AREA_DETAIL 5
+#define COLOR_AREA_ENTRY_SELECTED 5
+#define COLOR_AREA_DETAIL 6
 
 class TUI
 {
 protected:
-
 public:
-
+    enum UserInput
-    enum UserInput { UpKey, DownKey, Abort, Shred, Delete, Enter, ESC, Undefined};
+    {
+        UpKey,
+        DownKey,
+        Abort,
+        Shred,
+        ShredAll,
+        Delete,
+        Enter,
+        ESC,
+        Undefined
+    };
     struct MenuState
     {
         bool bAbort;
@@ -56,16 +66,17 @@ private:
     static WINDOW *createOverViewWindow(int iXSize, int iYSize);
     static WINDOW *createDetailViewWindow(int iXSize, int iYSize, int iXStart, Drive drive);
     static WINDOW *overwriteDetailViewWindow(int iXSize, int iYSize, int iXStart);
-    static WINDOW *createEntryWindow(int iXSize, int iYSize, int iXStart, int iYStart, string sModelFamily, string sModelName, string sCapacity, string sState, string sTime, bool bSelected);
+    static WINDOW *createEntryWindow(int iXSize, int iYSize, int iXStart, int iYStart, int iListIndex, string sModelFamily, string sSerial, string sCapacity, string sState, string sTime, string sSpeed, string sTemp, bool bSelected);
     static WINDOW *createSystemStats(int iXSize, int iYSize, int iXStart, int iYStart);
     static WINDOW *createMenuView(int iXSize, int iYSize, int iXStart, int iYStart, struct MenuState menustate);
     static WINDOW *createDialog(int iXSize, int iYSize, int iXStart, int iYStart, string selectedTask, string optionA, string optionB);
     static WINDOW *createFrozenWarning(int iXSize, int iYSize, int iXStart, int iYStart, string sPath, string sModelFamily, string sModelName, string sSerial, string sProgress);
-    static WINDOW* createSmartWarning(int iXSize, int iYSize, int iXStart, int iYStart, string sPath, uint32_t u32PowerOnHours, uint32_t u32PowerCycles,  uint32_t u32ErrorCount);
+    static WINDOW *createSmartWarning(int iXSize, int iYSize, int iXStart, int iYStart, string sPath, uint32_t u32PowerOnHours, uint32_t u32PowerCycles, uint32_t u32ErrorCount, uint32_t u32Temperature);
     static WINDOW *createZeroChecksumWarning(int iXSize, int iYSize, int iXStart, int iYStart, string sPath, string sModelFamily, string sModelName, string sSerial, uint32_t u32Checksum);
 
     void displaySelectedDrive(Drive drive, int stdscrX, int stdscrY);
     string formatTimeDuration(time_t u32Duration);
-
+    string formatSpeed(time_t u32ShredTimeDelta, unsigned long ulWrittenBytes);
+    static void vTruncateText(string *psText, uint16_t u16MaxLenght);
 };
 #endif // TUI_H_

makefile

@@ -8,19 +8,22 @@ SRC_EXT = cpp
 # Path to the source directory, relative to the makefile
 SRC_PATH = src
 # Space-separated pkg-config libraries used by this project
-LIBS =
+LIBS = lib
 # General compiler flags
-COMPILE_FLAGS = -std=c++17 -Wall -Wextra -g
+COMPILE_FLAGS = -std=c++23 -Wall -Wextra -g
 # Additional release-specific flags
-RCOMPILE_FLAGS = -D NDEBUG
+RCOMPILE_FLAGS = -D NDEBUG -Ofast
 # Additional debug-specific flags
 DCOMPILE_FLAGS = -D DEBUG
 # Add additional include paths
 INCLUDES = include
 # General linker settings
-LINK_FLAGS = -lpthread -lncurses
+LINK_FLAGS = -Llib -lpthread -lncurses -ltfng
+
 # Doc
 DOCDIR = doc
+TFRANDDIR = tfnoisegen
+TFRANDLIB = libtfng.a
 #### END PROJECT SETTINGS ####
 
 # Optionally you may move the section above to a separate config.mk file, and
@@ -158,6 +161,7 @@ dirs:
 	@echo "Creating directories"
 	@mkdir -p $(dir $(OBJECTS))
 	@mkdir -p $(BIN_PATH)
+	@mkdir -p $(LIBS)
 
 # Removes all build files
 .PHONY: clean
@@ -167,16 +171,22 @@ clean:
 	@echo "Deleting directories"
 	@$(RM) -r build
 	@$(RM) -r bin
+	@$(RM) -r $(LIBS)
 	@$(RM) -f reHDD.log
+	$(MAKE) clean -C tfnoisegen 
 
 # Main rule, checks the executable and symlinks to the output
 all: $(BIN_PATH)/$(BIN_NAME)
+	$(MAKE) libtfng.a -C tfnoisegen
+	@cp $(TFRANDDIR)/$(TFRANDLIB) $(LIBS)
 	@echo "Making symlink: $(BIN_NAME) -> $<"
 	@$(RM) $(BIN_NAME)
 	@ln -s $(BIN_PATH)/$(BIN_NAME) $(BIN_NAME)
 
 # Link the executable
 $(BIN_PATH)/$(BIN_NAME): $(OBJECTS)
+	$(MAKE) libtfng.a -C tfnoisegen
+	@cp $(TFRANDDIR)/$(TFRANDLIB) $(LIBS)
 	@echo "Linking: $@"
 	@$(START_TIME)
 	$(CMD_PREFIX)$(CXX) $(OBJECTS) $(LDFLAGS) -o $@

scripts/getty@tty2.service.d_override.conf

@@ -4,7 +4,7 @@ Description=dmesg on tty2
 [Service]
 WorkingDirectory=/usr/bin/
 ExecStart=
-ExecStart=-/usr/bin/dmesg -wH
+ExecStart=-/usr/bin/dmesg -wHT
 StandardInput=tty
 StandardOutput=tty
 Restart=always

scripts/update.sh

@@ -2,14 +2,16 @@
 
 echo starting update
 
-FILE=./ignoreDrives.conf
+systemctl stop /lib/systemd/system/getty@tty1.service.d
-if test -f "$FILE"; then
-   echo backup exits
-else
-    cp reHDD/ignoreDrives.conf ./ignoreDrives.conf
-fi
 
-cd  reHDD
+cd /root/reHDD/
+
+FILE=../ignoreDrives.conf
+if test -f "$FILE"; then
+   echo "backup exits already"
+else
+    cp /root/reHDD/ignoreDrives.conf /root/ignoreDrives.conf
+fi
 
 git reset
 
@@ -21,6 +23,14 @@ git checkout master
 
 git pull
 
+git submodule init
+
+git submodule update
+
+make clean
+
 make release
 
-cp ../ignoreDrives.conf ./
+cp  /root/ignoreDrives.conf /root/reHDD/ignoreDrives.conf
+
+systemctl start /lib/systemd/system/getty@tty1.service.d

src/delete.cpp

@@ -32,6 +32,9 @@ void Delete::deleteDrive(Drive* drive)
     const char *cpComand = sCMD.c_str();
     // cout << "delete: " << cpComand << endl;
 
+    if (drive->bWasShredStarted == false)
+    {
+        //only start delete if the drive was not shredded before
         FILE *deleteCmdOutput = popen(cpComand, "r");
 
         while ((getline(&cLine, &len, deleteCmdOutput)) != -1)
@@ -40,3 +43,4 @@ void Delete::deleteDrive(Drive* drive)
         }
         pclose(deleteCmdOutput);
     }
+}

src/drive.cpp

@@ -14,36 +14,41 @@ string Drive::getPath(void)
 
 string Drive::getModelFamily(void)
 {
-    return sModelFamily;
+    return sSmartData.sModelFamily;
 }
 
 string Drive::getModelName(void)
 {
-    return sModelName;
+    return sSmartData.sModelName;
 }
 
 string Drive::getSerial(void)
 {
-    return sSerial;
+    return sSmartData.sSerial;
 }
 
 uint64_t Drive::getCapacity(void)
 {
-    return u64Capacity;
+    return sSmartData.u64Capacity;
 }
 
 uint32_t Drive::getErrorCount(void)
 {
-    return u32ErrorCount;
+    return sSmartData.u32ErrorCount;
 }
 uint32_t Drive::getPowerOnHours(void)
 {
-    return u32PowerOnHours;
+    return sSmartData.u32PowerOnHours;
 }
 
 uint32_t Drive::getPowerCycles(void)
 {
-    return u32PowerCycles;
+    return sSmartData.u32PowerCycles;
+}
+
+uint32_t Drive::getTemperature(void)
+{
+    return sSmartData.u32Temperature;
 }
 
 string Drive::sCapacityToText()
@@ -67,7 +72,6 @@ string Drive::sErrorCountToText()
     return to_string(getErrorCount());
 }
 
-
 string Drive::sPowerOnHoursToText()
 {
     double dDays = 0U;
@@ -92,6 +96,12 @@ string Drive::sPowerCyclesToText()
     return to_string(getPowerCycles());
 }
 
+string Drive::sTemperatureToText()
+{
+    return to_string(getTemperature()) + " C";
+    ;
+}
+
 void Drive::setTaskPercentage(double d32TaskPercentage)
 {
     if (d32TaskPercentage <= 100)
@@ -105,7 +115,6 @@ double Drive::getTaskPercentage(void)
     return this->d32TaskPercentage;
 }
 
-
 /**
  * \brief   set S.M.A.R.T. values in model
  * \param	string modelFamily
@@ -115,6 +124,7 @@ double Drive::getTaskPercentage(void)
  * \param   uint32_t errorCount
  * \param   uint32_t powerOnHours
  * \param   uint32_t powerCycle
+ * \param   uint32_t temperature
  * \return  void
  */
 void Drive::setDriveSMARTData(string modelFamily,
@@ -123,16 +133,17 @@ void Drive::setDriveSMARTData( string modelFamily,
                               uint64_t capacity,
                               uint32_t errorCount,
                               uint32_t powerOnHours,
-                               uint32_t powerCycle)
+                              uint32_t powerCycle,
+                              uint32_t temperature)
 {
-    this->sModelFamily = modelFamily;
+    this->sSmartData.sModelFamily = modelFamily;
-    sModelName = modelName;
+    this->sSmartData.sModelName = modelName;
-    sSerial = serial;
+    this->sSmartData.sSerial = serial;
-    u64Capacity = capacity;
+    this->sSmartData.u64Capacity = capacity;
-    u32ErrorCount = errorCount;
+    this->sSmartData.u32ErrorCount = errorCount;
-    u32PowerOnHours = powerOnHours;
+    this->sSmartData.u32PowerOnHours = powerOnHours;
-    u32PowerCycles = powerCycle;
+    this->sSmartData.u32PowerCycles = powerCycle;
-
+    this->sSmartData.u32Temperature = temperature;
 }
 
 void Drive::setTimestamp()
@@ -168,10 +179,10 @@ void Drive::checkFrozenDrive(void)
     time_t u32localtime;
     time(&u32localtime);
 
-    if((u32localtime - this->u32Timestamp) >= (FROZEN_TIMEOUT*60) && (this->u32Timestamp > 0))
+    if ((u32localtime - this->u32Timestamp) >= (FROZEN_TIMEOUT * 60) && (this->u32Timestamp > 0) && (this->getTaskPercentage() < 100.0))
     {
         Logger::logThis()->warning("Drive Frozen: " + this->getModelName() + " " + this->getSerial());
-            this->bWasDeleteted = false;
+        this->bWasDeleted = false;
         this->bWasShredded = false;
         this->state = Drive::FROZEN;
     }

src/logger/logger.cpp

@@ -5,7 +5,6 @@
  * @date    04.09.2020
  */
 
-
 #include "../../include/reHDD.h" //for logger settings
 #include "../../include/logger/logger.h"
 
@@ -18,8 +17,6 @@ Logger* Logger::single = NULL;
 
 /**
  * \brief   create new logger instance
- * \param	path to log file
- * \param	struct with data
  * \return  instance of Logger
  */
 Logger::Logger()
@@ -155,7 +152,6 @@ string Logger::getMacAddress()
     if (ioctl(s, SIOCGIFHWADDR, &ifr) < 0)
     {
         strcpy(ifr.ifr_name, "eno1");
-
     }
 
     unsigned char *hwaddr = (unsigned char *)ifr.ifr_hwaddr.sa_data;
@@ -231,5 +227,3 @@ Logger* Logger::logThis()
         return single; // return existing obj
     }
 }
-
-

src/printer.cpp

@@ -0,0 +1,84 @@
+/**
+ * @file    printer.cpp
+ * @brief   Send drive data to printer service using ipc msg queue
+ * @author  Hendrik Schutter
+ * @date    24.11.2022
+ */
+
+#include "../include/reHDD.h"
+
+bool Printer::instanceFlag = false;
+Printer *Printer::single = NULL;
+
+/**
+ * \brief   create new Printer instance
+ * \param	path to log file
+ * \param	struct with data
+ * \return  instance of Printer
+ */
+Printer::Printer()
+{
+    if (-1 == (this->msqid = msgget((key_t)IPC_MSG_QUEUE_KEY, IPC_CREAT | 0666)))
+    {
+        Logger::logThis()->error("Printer: Create mgs queue failed!");
+    }
+}
+
+/**
+ * \brief   deconstructor
+ * \return  void
+ */
+Printer::~Printer()
+{
+    instanceFlag = false;
+}
+
+/**
+ * \brief   send data to msg queue
+ * \return  void
+ */
+void Printer::print(Drive *drive)
+{
+    t_msgQueueData msgQueueData;
+    msgQueueData.msg_queue_type = 1;
+
+    sprintf(msgQueueData.driveData.caDriveIndex, "%i", 42); // TODO: get from tui
+    sprintf(msgQueueData.driveData.caDriveState, "shredded");
+    strcpy(msgQueueData.driveData.caDriveModelFamily, drive->getModelFamily().c_str());
+    strcpy(msgQueueData.driveData.caDriveModelName, drive->getModelName().c_str());
+    sprintf(msgQueueData.driveData.caDriveCapacity, "%li", drive->getCapacity());
+    strcpy(msgQueueData.driveData.caDriveSerialnumber, drive->getSerial().c_str());
+    sprintf(msgQueueData.driveData.caDriveHours, "%i", drive->getPowerOnHours());
+    sprintf(msgQueueData.driveData.caDriveCycles, "%i", drive->getPowerCycles());
+    sprintf(msgQueueData.driveData.caDriveErrors, "%i", drive->getErrorCount());
+    sprintf(msgQueueData.driveData.caDriveShredTimestamp, "%li", drive->getActionStartTimestamp());
+    sprintf(msgQueueData.driveData.caDriveShredDuration, "%li", drive->getTaskDuration());
+    sprintf(msgQueueData.driveData.caDriveReHddVersion, REHDD_VERSION);
+
+    if (-1 == msgsnd(this->msqid, &msgQueueData, sizeof(t_msgQueueData) - sizeof(long), 0))
+    {
+        Logger::logThis()->error("Printer: Send mgs queue failed!");
+    }
+    else
+    {
+        Logger::logThis()->info("Printer: print triggered - Drive: " + drive->getSerial());
+    }
+}
+
+/**
+ * \brief   return a instance of the printer
+ * \return  printer obj
+ */
+Printer *Printer::getPrinter()
+{
+    if (!instanceFlag)
+    {
+        single = new Printer(); // create new obj
+        instanceFlag = true;
+        return single;
+    }
+    else
+    {
+        return single; // return existing obj
+    }
+}

src/reHDD.cpp

@@ -11,11 +11,11 @@ static int fdNewDrivesInformPipe[2];//File descriptor for pipe that informs if n
 
 static int fdShredInformPipe[2]; // File descriptor for pipe that informs if a wipe thread signals
 
-static std::mutex mxScannDrives;
+static std::mutex mxDrives;
 
 list<Drive> listNewDrives; // store found drives that are updated every 5sec
 
-static list <Drive> listDrives; //stores all drive data from scann thread
+static list<Drive> listDrives; // stores all drive data from scan thread
 
 TUI *ui;
 
@@ -46,7 +46,7 @@ void reHDD::app_logic(void)
     pipe(fdNewDrivesInformPipe);
     pipe(fdShredInformPipe);
 
-    thread thDevices(ThreadScannDevices); //start thread that scanns for drives
+    thread thDevices(ThreadScanDevices);                 // start thread that scans for drives
     thread thUserInput(ThreadUserInput);                 // start thread that reads user input
     thread thCheckFrozenDrives(ThreadCheckFrozenDrives); // start thread that checks timeout for drives
 
@@ -60,17 +60,18 @@ void reHDD::app_logic(void)
 
         if (FD_ISSET(fdNewDrivesInformPipe[0], &selectSet))
         {
-                    mxScannDrives.lock();
+            mxDrives.lock();
             char dummy;
             read(fdNewDrivesInformPipe[0], &dummy, 1);
             filterNewDrives(&listDrives, &listNewDrives); // filter and copy to app logic vector
             printDrives(&listDrives);
-                    mxScannDrives.unlock();
+            mxDrives.unlock();
         }
         if (FD_ISSET(fdShredInformPipe[0], &selectSet))
         {
             char dummy;
             read(fdShredInformPipe[0], &dummy, 1);
+            updateShredMetrics(&listDrives);
 #ifdef LOG_LEVEL_HIGH
             Logger::logThis()->info("got progress signal from a shred task");
 #endif
@@ -97,16 +98,17 @@ Drive* reHDD::getSelectedDrive()
     }
 }
 
-void reHDD::ThreadScannDevices()
+void reHDD::ThreadScanDevices()
 {
     while (true)
     {
-            mxScannDrives.lock();
+        mxDrives.lock();
         listNewDrives.clear();
         searchDrives(&listNewDrives);        // search for new drives and store them in list
         filterIgnoredDrives(&listNewDrives); // filter out ignored drives
         addSMARTData(&listNewDrives);        // add S.M.A.R.T. Data to the drives
-            mxScannDrives.unlock();
+        filterInvalidDrives(&listNewDrives); // filter out drives that report zero capacity
+        mxDrives.unlock();
         write(fdNewDrivesInformPipe[1], "A", 1);
         sleep(5); // sleep 5 sec
     }
@@ -116,7 +118,7 @@ void reHDD::ThreadCheckFrozenDrives()
 {
     while (true)
     {
-            mxScannDrives.lock();
+        mxDrives.lock();
         for (auto it = begin(listDrives); it != end(listDrives); ++it)
         {
             if (it->state == Drive::SHRED_ACTIVE)
@@ -124,7 +126,7 @@ void reHDD::ThreadCheckFrozenDrives()
                 it->checkFrozenDrive();
             }
         }
-            mxScannDrives.unlock();
+        mxDrives.unlock();
         sleep(13); // sleep 13 sec
     }
 }
@@ -164,6 +166,7 @@ void reHDD::ThreadUserInput()
                     getSelectedDrive()->state = Drive::DELETE_SELECTED;
                 }
             }
+
             ui->updateTUI(&listDrives, u8SelectedEntry);
             break;
         case TUI::UserInput::Shred:
@@ -176,6 +179,12 @@ void reHDD::ThreadUserInput()
                     getSelectedDrive()->state = Drive::SHRED_SELECTED;
                 }
             }
+
+            ui->updateTUI(&listDrives, u8SelectedEntry);
+            break;
+        case TUI::UserInput::ShredAll:
+            // cout << "ShredAll" << endl;
+            startShredAllDrives(&listDrives);
             ui->updateTUI(&listDrives, u8SelectedEntry);
             break;
         case TUI::UserInput::Enter:
@@ -194,13 +203,13 @@ void reHDD::ThreadUserInput()
     }
 }
 
-void reHDD::ThreadShred()
+void reHDD::ThreadShred(Drive *const pDrive)
 {
-    if (getSelectedDrive() != nullptr)
+    if (pDrive != nullptr)
     {
-            getSelectedDrive()->setActionStartTimestamp(); //save timestamp at start of shredding
+        pDrive->setActionStartTimestamp();                     // save timestamp at start of shredding
         Shred *pShredTask = new Shred();                       // create new shred task
-            pShredTask->shredDrive(getSelectedDrive(), &fdShredInformPipe[1]); //start new shred task
+        pShredTask->shredDrive(pDrive, &fdShredInformPipe[1]); // start new shred task
         delete pShredTask;                                     // delete shred task
         ui->updateTUI(&listDrives, u8SelectedEntry);
     }
@@ -213,7 +222,7 @@ void reHDD::ThreadDelete()
         getSelectedDrive()->setActionStartTimestamp();      // save timestamp at start of deleting
         Delete::deleteDrive(getSelectedDrive());            // blocking, no thread
         getSelectedDrive()->state = Drive::TaskState::NONE; // delete finished
-            getSelectedDrive()->bWasDeleteted = true;
+        getSelectedDrive()->bWasDeleted = true;
         Logger::logThis()->info("Finished delete for: " + getSelectedDrive()->getModelName() + "-" + getSelectedDrive()->getSerial());
         ui->updateTUI(&listDrives, u8SelectedEntry);
     }
@@ -222,7 +231,7 @@ void reHDD::ThreadDelete()
 void reHDD::filterNewDrives(list<Drive> *plistOldDrives, list<Drive> *plistNewDrives)
 {
     list<Drive>::iterator itOld; // Iterator for current (old) drive list
-    list <Drive>::iterator itNew; //Iterator for new drive list that was created from to scann thread
+    list<Drive>::iterator itNew; // Iterator for new drive list that was created from to scan thread
 
     // remove offline old drives from previously run
     for (itOld = plistOldDrives->begin(); itOld != plistOldDrives->end();)
@@ -246,16 +255,18 @@ void reHDD::filterNewDrives(list <Drive>* plistOldDrives, list <Drive>* plistNew
     // search offline drives and mark them
     for (itOld = plistOldDrives->begin(); itOld != plistOldDrives->end(); ++itOld)
     {
-            itOld->bIsOffline = true; //set offline befor seachring in the new list
+        itOld->bIsOffline = true; // set offline before searching in the new list
         for (itNew = plistNewDrives->begin(); itNew != plistNewDrives->end();)
         {
             if ((itOld->getSerial() == itNew->getSerial()) || (itOld->getPath() == itNew->getPath()))
             {
                 itOld->bIsOffline = false; // drive is still attached
+                // copy new smart data to existing drive
+                itOld->setDriveSMARTData(itNew->getModelFamily(), itNew->getModelName(), itNew->getSerial(), itNew->getCapacity(), itNew->getErrorCount(), itNew->getPowerOnHours(), itNew->getPowerCycles(), itNew->getTemperature());
 #ifdef LOG_LEVEL_HIGH
-                            Logger::logThis()->info("Delete new drive, because allready attached: " + itNew->getModelName());
+                Logger::logThis()->info("Delete new drive, because already attached: " + itNew->getModelName());
 #endif
-                            itNew = plistNewDrives->erase(itNew); //This drive is allready attached, remove from new list
+                itNew = plistNewDrives->erase(itNew); // This drive is already attached, remove from new list
             }
             else
             {
@@ -279,7 +290,7 @@ void reHDD::filterNewDrives(list <Drive>* plistOldDrives, list <Drive>* plistNew
     for (itNew = plistNewDrives->begin(); itNew != plistNewDrives->end(); ++itNew)
     {
         plistOldDrives->push_back(*itNew);
-            Logger::logThis()->info("Add new drive: " + itNew->getModelName());
+        // Logger::logThis()->info("Add new drive: " + itNew->getModelName());
     }
     plistNewDrives->clear();
 }
@@ -291,11 +302,11 @@ void reHDD::filterNewDrives(list <Drive>* plistOldDrives, list <Drive>* plistNew
  */
 void reHDD::searchDrives(list<Drive> *plistDrives)
 {
-    Logger::logThis()->info("-->    search drives   <--");
+    // Logger::logThis()->info("-->    search drives   <--");
     char *cLine = NULL;
     size_t len = 0;
 
-    FILE* outputfileHwinfo = popen("lsblk -I 8 -d -o NAME", "r");
+    FILE *outputfileHwinfo = popen("lsblk -e 11 -d -o NAME", "r");
 
     if (outputfileHwinfo == NULL)
     {
@@ -311,7 +322,16 @@ void reHDD::searchDrives(list <Drive>* plistDrives)
             tmpDrive->state = Drive::NONE;
             tmpDrive->bIsOffline = false;
             plistDrives->push_back(*tmpDrive);
-                    //Logger::logThis()->info("drive found: " + tmpDrive->getPath());
+            // Logger::logThis()->info("SATA drive found: " + tmpDrive->getPath());
+        }
+
+        if (string(cLine).length() == 8)
+        {
+            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);
@@ -324,15 +344,35 @@ void reHDD::searchDrives(list <Drive>* plistDrives)
  */
 void reHDD::filterIgnoredDrives(list<Drive> *plistDrives)
 {
-    list<tuple<string>> vtlIgnoredDevices; //store drives from ingnore file
+    string systemDrivePath;
-    ifstream input( "ignoreDrives.conf" ); //read ingnore file
+    if (getSystemDrive(systemDrivePath))
+    {
+        // Logger::logThis()->info("Found system drive: " + systemDrivePath);
+
+        list<Drive>::iterator it;
+        for (it = plistDrives->begin(); it != plistDrives->end(); ++it)
+        {
+            if (it->getPath().find(systemDrivePath) != std::string::npos) // compare found system drive and current drive
+            {
+                // system drive found --> ignore this drive
+#ifdef LOG_LEVEL_HIGH
+                Logger::logThis()->info("system drive found --> ignore this drive: " + it->getPath());
+#endif
+                it = plistDrives->erase(it);
+                it--;
+            }
+        }
+    }
+
+    list<tuple<string>> vtlIgnoredDevices; // store drives from ignore file
+    ifstream input("ignoreDrives.conf");   // read ignore file
 
     for (string sLine; getline(input, sLine);)
     {
-            Logger::logThis()->info("read uuid: " + sLine);
+        // Logger::logThis()->info("read uuid: " + sLine);
         vtlIgnoredDevices.emplace_back(sLine); // add found path and uuid from ignore file to vector
     }
-    //loop through found entries in ingnore file
+    // loop through found entries in ignore file
     for (auto row : vtlIgnoredDevices)
     {
         list<Drive>::iterator it;
@@ -377,6 +417,53 @@ void reHDD::filterIgnoredDrives(list <Drive>* plistDrives)
     }
 }
 
+/**
+ * \brief   filter out drives that are not indented for processing
+ * \param	pointer of list <Drive>* plistDrives
+ * \return  void
+ */
+void reHDD::filterInvalidDrives(list<Drive> *plistDrives)
+{
+    list<Drive>::iterator it;
+    for (it = plistDrives->begin(); it != plistDrives->end(); ++it)
+    {
+        if (it->getCapacity() == 0U)
+        {
+#ifdef LOG_LEVEL_HIGH
+            Logger::logThis()->info("Drive reports zero capacity --> ignore this drive: " + it->getPath());
+#endif
+            it = plistDrives->erase(it);
+            it--;
+        }
+    }
+}
+
+/**
+ * \brief   start shred for all drives
+ * \param	pointer of list <Drive>* plistDrives
+ * \return  void
+ */
+void reHDD::startShredAllDrives(list<Drive> *plistDrives)
+{
+    list<Drive>::iterator it;
+    mxDrives.lock();
+    for (it = plistDrives->begin(); it != plistDrives->end(); ++it)
+    {
+        if (it->state == Drive::NONE)
+        {
+            Drive *pTmpDrive = iterator_to_pointer<Drive, std::list<Drive>::iterator>(it);
+#ifdef LOG_LEVEL_HIGH
+            ostringstream address;
+            address << (void const *)&(*pTmpDrive);
+            Logger::logThis()->info("Started shred (all) for: " + pTmpDrive->getModelName() + "-" + pTmpDrive->getSerial() + " @" + address.str());
+#endif
+            pTmpDrive->state = Drive::TaskState::SHRED_ACTIVE;
+            thread(ThreadShred, pTmpDrive).detach();
+        }
+    }
+    mxDrives.unlock();
+}
+
 /**
  * \brief   print drives with all information
  * \param	pointer of list <Drive>* plistDrives
@@ -412,6 +499,33 @@ void reHDD::printDrives(list <Drive>* plistDrives)
 #endif
 }
 
+/**
+ * \brief   update shred metrics for all drives
+ * \param	pointer of list <Drive>* plistDrives
+ * \return  void
+ */
+void reHDD::updateShredMetrics(list<Drive> *plistDrives)
+{
+    list<Drive>::iterator it;
+    for (it = plistDrives->begin(); it != plistDrives->end(); ++it)
+    {
+        if (it->state == Drive::SHRED_ACTIVE)
+        {
+            Drive *pTmpDrive = iterator_to_pointer<Drive, std::list<Drive>::iterator>(it);
+            // set metrics for calculating shred speed
+            std::chrono::time_point<std::chrono::system_clock> chronoCurrentTimestamp = std::chrono::system_clock::now();
+            time_t u32ShredTimeDelta = (chronoCurrentTimestamp - pTmpDrive->sShredSpeed.chronoShredTimestamp).count();
+            if (u32ShredTimeDelta > METRIC_THRESHOLD)
+            {
+                pTmpDrive->sShredSpeed.u32ShredTimeDelta = u32ShredTimeDelta;
+                pTmpDrive->sShredSpeed.chronoShredTimestamp = std::chrono::system_clock::now();
+                pTmpDrive->sShredSpeed.ulWrittenBytes = pTmpDrive->sShredSpeed.ulSpeedMetricBytesWritten;
+                pTmpDrive->sShredSpeed.ulSpeedMetricBytesWritten = 0U;
+            }
+        }
+    }
+}
+
 /**
  * \brief   add S.M.A.R.T data from SMART
  * \param	pointer of list <Drive>* plistDrives
@@ -454,26 +568,28 @@ void reHDD::handleArrowKey(TUI::UserInput userInput)
         break;
     }
 
-    Logger::logThis()->info("ArrowKey - selected drive: " + to_string(u8SelectedEntry));
+    // Logger::logThis()->info("ArrowKey - selected drive: " + to_string(u8SelectedEntry));
 }
 
 void reHDD::handleEnter()
 {
+
     if (getSelectedDrive() != nullptr)
     {
         if (getSelectedDrive()->state == Drive::TaskState::SHRED_SELECTED)
         {
-                    Logger::logThis()->info("Started shred for: " + getSelectedDrive()->getModelName() + "-" +  getSelectedDrive()->getSerial());
+            Logger::logThis()->info("Started shred/check for: " + getSelectedDrive()->getModelName() + "-" + getSelectedDrive()->getSerial());
             getSelectedDrive()->state = Drive::TaskState::SHRED_ACTIVE;
-                    //task for drive is running --> don´t show  more task options
+            // task for drive is running --> don't show  more task options
-                    thread(ThreadShred).detach();
+            Drive *pTmpDrive = getSelectedDrive();
+            thread(ThreadShred, pTmpDrive).detach();
         }
 
         if (getSelectedDrive()->state == Drive::TaskState::DELETE_SELECTED)
         {
             Logger::logThis()->info("Started delete for: " + getSelectedDrive()->getModelName() + "-" + getSelectedDrive()->getSerial());
             getSelectedDrive()->state = Drive::TaskState::DELETE_ACTIVE;
-                    //task for drive is running --> don´t show  more task options
+            // task for drive is running --> don't show  more task options
             thread(ThreadDelete).detach();
         }
     }
@@ -510,5 +626,58 @@ void reHDD::handleAbort()
     }
 }
 
+bool reHDD::getSystemDrive(string &systemDrive)
+{
+    char *cLine = NULL;
+    size_t len = 0;
+    bool systemDriveFound = false;
 
+    FILE *outputfileHwinfo = popen("lsblk -e 11 -o NAME,MOUNTPOINT", "r");
 
+    if (outputfileHwinfo == NULL)
+    {
+        Logger::logThis()->error("Unable to scan attached drives for system drive");
+        exit(EXIT_FAILURE);
+    }
+
+    while ((getline(&cLine, &len, outputfileHwinfo)) != -1)
+    {
+        string currentLine = cLine;
+
+        if (currentLine.find("NAME") != std::string::npos)
+        {
+            continue;
+        }
+
+        // Logger::logThis()->info(currentLine);
+
+        if ((cLine[0U] != '|') && (cLine[0U] != '`'))
+        {
+            systemDrive = currentLine;
+            systemDrive.erase(std::remove(systemDrive.begin(), systemDrive.end(), '\n'), systemDrive.end()); // remove newline
+            systemDrive.erase(std::remove(systemDrive.begin(), systemDrive.end(), ' '), systemDrive.end());  // remove spaces
+            // Logger::logThis()->info("Drive found: " + systemDrive);
+        }
+
+        if (currentLine.ends_with(" /boot/efi\n"s))
+        {
+            systemDriveFound = true;
+            break;
+        }
+
+        if (currentLine.ends_with(" /run/overlay/live\n"s))
+        {
+            systemDriveFound = true;
+            break;
+        }
+
+        if (currentLine.ends_with(" /\n"s))
+        {
+            systemDriveFound = true;
+            break;
+        }
+    }
+    pclose(outputfileHwinfo);
+
+    return systemDriveFound;
+}

src/shred.cpp

@@ -2,11 +2,20 @@
  * @file    shred.cpp
  * @brief   shred drive
  * @author  hendrik schutter
- * @date    03.05.2020
+ * @date    22.08.2022
  */
 
 #include "../include/reHDD.h"
 
+#ifdef __cplusplus
+extern "C"
+{
+#endif
+#include "../tfnoisegen/tfprng.h"
+#ifdef __cplusplus
+}
+#endif
+
 const static char *randomsrc = (char *)"/dev/urandom";
 
 Shred::Shred()
@@ -24,6 +33,10 @@ Shred::~Shred()
  */
 int Shred::shredDrive(Drive *drive, int *ipSignalFd)
 {
+    ostringstream address;
+    address << (void const *)&(*drive);
+    Logger::logThis()->info("Shred-Task started - Drive: " + drive->getModelName() + "-" + drive->getSerial() + " @" + address.str());
+    drive->bWasShredStarted = true; //Mark drive as partly shredded
 
 #ifdef DRYRUN
     for (int i = 0; i <= 500; i++)
@@ -36,10 +49,12 @@ int Shred::shredDrive(Drive* drive, int* ipSignalFd)
         write(*ipSignalFd, "A", 1);
         usleep(20000);
     }
+    drive->bWasShredded = true;
 #endif
 
 #ifndef DRYRUN
     const char *cpDrivePath = drive->getPath().c_str();
+    unsigned char ucKey[TFNG_KEY_SIZE];
 
     // open random source
     randomSrcFileDiscr = open(randomsrc, O_RDONLY | O_LARGEFILE);
@@ -63,7 +78,22 @@ int Shred::shredDrive(Drive* drive, int* ipSignalFd)
         return -1;
     }
 
+    // read key for random generator
+    ssize_t readRet = read(randomSrcFileDiscr, ucKey, sizeof(ucKey));
+    if (readRet <= 0)
+    {
+        std::string errorMsg(strerror(readRet));
+        Logger::logThis()->error("Shred-Task: Read random key failed! " + errorMsg + " - Drive: " + drive->getSerial());
+        perror(randomsrc);
+        cleanup();
+        return -1;
+    }
+
+    tfng_prng_seedkey(ucKey);
+
     this->ulDriveByteSize = getDriveSizeInBytes(driveFileDiscr);
+    drive->sShredSpeed.chronoShredTimestamp = std::chrono::system_clock::now(); // set inital timestamp for speed metric
+    drive->sShredSpeed.ulSpeedMetricBytesWritten = 0U; // uses to calculate speed metric
 
 #ifdef LOG_LEVEL_HIGH
     Logger::logThis()->info("Shred-Task: Bytes-Size of Drive: " + to_string(this->ulDriveByteSize) + " - Drive: " + drive->getSerial());
@@ -72,44 +102,21 @@ int Shred::shredDrive(Drive* drive, int* ipSignalFd)
     for (unsigned int uiShredIterationCounter = 0U; uiShredIterationCounter < SHRED_ITERATIONS; uiShredIterationCounter++)
     {
         unsigned long ulDriveByteCounter = 0U; // used for one shred-iteration to keep track of the current drive position
-            uint32_t u32ChunkDimensionIndex = 0U;
 
         if (uiShredIterationCounter == (SHRED_ITERATIONS - 1))
         {
-                    //last shred iteration --> overwrite with zeros instead with random data
+            // last shred iteration --> overwrite (just the write chunk) bytes with zeros instead with random data
-                    memset(caChunk, 0U, CHUNK_DIMENSION*CHUNK_SIZE);
+            memset(caTfngData, 0U, CHUNK_SIZE);
         }
 
         while (ulDriveByteCounter < ulDriveByteSize)
         {
             int iBytesToShred = 0; // Bytes that will be overwritten in this chunk-iteration
 
-                    if((u32ChunkDimensionIndex == 0U) && (uiShredIterationCounter != (SHRED_ITERATIONS-1)))
+            if (uiShredIterationCounter != (SHRED_ITERATIONS - 1))
             {
-                            //read new chunks from random source if needed and this is NOT the last shred iteration
+                // NOT last shred iteration --> generate new random data
-                            unsigned long ulBytesInChunkBuffer = 0U;
+                tfng_prng_genrandom(caTfngData, TFNG_DATA_SIZE);
-
-                            while (ulBytesInChunkBuffer < CHUNK_DIMENSION*CHUNK_SIZE)
-                                {
-                                    //read new random bytes
-                                    int iReadBytes = read(randomSrcFileDiscr, caChunk, ((CHUNK_DIMENSION*CHUNK_SIZE)-ulBytesInChunkBuffer));
-                                    if (iReadBytes > 0)
-                                        {
-                                            ulBytesInChunkBuffer += iReadBytes;
-                                        }
-                                    else
-                                        {
-                                            std::string errorMsg(strerror(iReadBytes));
-                                            Logger::logThis()->error("Shred-Task: Read from random source failed! " + errorMsg + " - Drive: " + drive->getSerial());
-                                            perror("unable to read random data");
-                                            cleanup();
-                                            return -1;;
-                                        }
-                                } //end chunk read
-#ifdef LOG_LEVEL_HIGH
-                            Logger::logThis()->info("Shred-Task: Read new random data - Drive: " + drive->getSerial());
-#endif
-
             }
 
             if ((ulDriveByteSize - ulDriveByteCounter) < CHUNK_SIZE)
@@ -121,7 +128,7 @@ int Shred::shredDrive(Drive* drive, int* ipSignalFd)
                 iBytesToShred = CHUNK_SIZE;
             }
 
-                    int iByteShredded = write(driveFileDiscr, caChunk[u32ChunkDimensionIndex], iBytesToShred);
+            int iByteShredded = write(driveFileDiscr, caTfngData, iBytesToShred);
 
             if (iByteShredded <= 0)
             {
@@ -132,18 +139,20 @@ int Shred::shredDrive(Drive* drive, int* ipSignalFd)
                 return -1;
             }
 
-                    u32ChunkDimensionIndex = (u32ChunkDimensionIndex+1)%CHUNK_DIMENSION;
             ulDriveByteCounter += iByteShredded;
             ulDriveByteOverallCount += iByteShredded;
             d32Percent = this->calcProgress();
+            drive->sShredSpeed.ulSpeedMetricBytesWritten += iByteShredded;
 
 #ifdef LOG_LEVEL_HIGH
             Logger::logThis()->info("Shred-Task: ByteCount: " + to_string(ulDriveByteCounter) + " - iteration: " + to_string((uiShredIterationCounter + 1)) + " - progress: " + to_string(d32Percent) + " - Drive: " + drive->getSerial());
 #endif
             if ((d32Percent - d32TmpPercent) >= 0.01)
             {
+                // set shred percantage
                 drive->setTaskPercentage(d32TmpPercent);
                 d32TmpPercent = d32Percent;
+                // signal process in shreding
                 write(*ipSignalFd, "A", 1);
             }
 
@@ -157,24 +166,34 @@ int Shred::shredDrive(Drive* drive, int* ipSignalFd)
                 cleanup();
                 return -1;
             }
-                }//end one chunk write
+            // end one chunk write
+        }
         if (0 != iRewindDrive(driveFileDiscr))
         {
+            Logger::logThis()->error("Shred-Task: Unable to rewind drive! - Drive: " + drive->getSerial());
             cleanup();
             return -1;
         }
-        } //end one shred iteration
+        // end one shred iteration
+    }
+    // end of all shred iteratio
 
-#ifdef ZERO_CHECK_ALERT
+    tfng_prng_seedkey(NULL); // reset random generator
-    drive->u32DriveChecksumAferShredding = uiCalcChecksum(driveFileDiscr, drive, ipSignalFd);
+
+    drive->bWasShredded = true;
+    Logger::logThis()->info("Shred-Task finished - Drive: " + drive->getModelName() + "-" + drive->getSerial() + " @" + address.str());
+#ifdef ZERO_CHECK
+    drive->state = Drive::CHECK_ACTIVE;
+    Logger::logThis()->info("Check-Task started - Drive: " + drive->getModelName() + "-" + drive->getSerial() + " @" + address.str());
+    drive->u32DriveChecksumAfterShredding = uiCalcChecksum(driveFileDiscr, drive, ipSignalFd);
 #ifdef LOG_LEVEL_HIGH
     if (drive->u32DriveChecksumAferShredding != 0)
     {
-            Logger::logThis()->info("Shred-Task: Checksum not zero: " + to_string(drive->u32DriveChecksumAferShredding) + " - Drive: " + drive->getSerial());
+        Logger::logThis()->info("Shred-Task: Checksum not zero: " + to_string(drive->u32DriveChecksumAfterShredding) + " - Drive: " + drive->getSerial());
     }
     else
     {
-            Logger::logThis()->info("Shred-Task: Checksum zero: " + to_string(drive->u32DriveChecksumAferShredding) + " - Drive: " + drive->getSerial());
+        Logger::logThis()->info("Shred-Task: Checksum zero: " + to_string(drive->u32DriveChecksumAfterShredding) + " - Drive: " + drive->getSerial());
     }
 #endif
 #endif
@@ -182,12 +201,12 @@ int Shred::shredDrive(Drive* drive, int* ipSignalFd)
 
     cleanup();
 
-    if(drive->state == Drive::SHRED_ACTIVE)
+    if ((drive->state == Drive::SHRED_ACTIVE) || (drive->state == Drive::CHECK_ACTIVE))
     {
-            drive->bWasShredded = true;
         drive->state = Drive::NONE;
         drive->setTaskPercentage(0.0);
-            Logger::logThis()->info("Finished shred for: " + drive->getModelName() + "-" +  drive->getSerial());
+        Printer::getPrinter()->print(drive);
+        Logger::logThis()->info("Finished shred/check for: " + drive->getModelName() + "-" + drive->getSerial());
     }
     return 0;
 }
@@ -201,7 +220,7 @@ double Shred::calcProgress()
 {
     unsigned int uiMaxShredIteration = SHRED_ITERATIONS;
 
-#ifdef ZERO_CHECK_ALERT
+#ifdef ZERO_CHECK
     uiMaxShredIteration++; // increment because we will check after SHRED_ITERATIONS the drive for non-zero bytes
 #endif
     return (double)(((double)ulDriveByteOverallCount) / ((double)this->ulDriveByteSize * uiMaxShredIteration)) * 100.0f;
@@ -250,24 +269,31 @@ unsigned int Shred::uiCalcChecksum(fileDescriptor file,Drive* drive, int* ipSign
         {
             iBytesToCheck = CHUNK_SIZE;
         }
-            int iReadBytes = read(file, caChunk, iBytesToCheck);
+        int iReadBytes = read(file, caReadBuffer, iBytesToCheck);
         for (int iReadBytesCounter = 0U; iReadBytesCounter < iReadBytes; iReadBytesCounter++)
         {
-                    uiChecksum += caChunk[0][iReadBytesCounter];
+            uiChecksum += caReadBuffer[iReadBytesCounter];
         }
         ulDriveByteCounter += iReadBytes;
         ulDriveByteOverallCount += iReadBytes;
         d32Percent = this->calcProgress();
+        drive->sShredSpeed.ulSpeedMetricBytesWritten += iReadBytes;
+
 #ifdef LOG_LEVEL_HIGH
         Logger::logThis()->info("Shred-Task (Checksum): ByteCount: " + to_string(ulDriveByteCounter) + " - progress: " + to_string(d32Percent) + " - Drive: " + drive->getSerial());
 #endif
-            if((d32Percent-d32TmpPercent) >= 0.9)
+
+        if (((d32Percent - d32TmpPercent) >= 0.01) || (d32Percent == 100.0))
         {
             drive->setTaskPercentage(d32TmpPercent);
             d32TmpPercent = d32Percent;
+#ifdef LOG_LEVEL_HIGH
+            Logger::logThis()->info("send progress signal to main loop (check)");
+#endif
             write(*ipSignalFd, "A", 1);
         }
     }
+    drive->bWasChecked = true;
     return uiChecksum;
 }
 

src/smart.cpp

@@ -7,14 +7,6 @@
 
 #include "../include/reHDD.h"
 
-string SMART::modelFamily;
-string SMART::modelName;
-string SMART::serial;
-uint64_t SMART::capacity = 0U;
-uint32_t SMART::errorCount = 0U;
-uint32_t SMART::powerOnHours = 0U;
-uint32_t SMART::powerCycle = 0U;
-
 /**
  * \brief   get and set S.M.A.R.T. values in Drive
  * \param	pointer of Drive instance
@@ -22,79 +14,138 @@ uint32_t SMART::powerCycle = 0U;
  */
 void SMART::readSMARTData(Drive *drive)
 {
+    string modelFamily;
+    string modelName;
+    string serial;
+    uint64_t capacity = 0U;
+    uint32_t errorCount = 0U;
+    uint32_t powerOnHours = 0U;
+    uint32_t powerCycles = 0U;
+    uint32_t temperature = 0U;
+
     modelFamily.clear();
     modelName.clear();
     serial.clear();
-    capacity = 0U;
-    errorCount = 0U;
-    powerOnHours = 0U;
-    powerCycle = 0U;
 
-    size_t len = 0; //lenght of found line
+    string sSmartctlCommands[] = {" --json -a ", " --json -d sntjmicron -a ", " --json -d sntasmedia -a ", " --json -d sntrealtek -a ", " --json -d sat -a "};
-    char* cLine = NULL; //found line
 
-    string sCMD = ("smartctl --json -a ");
+    for (string sSmartctlCommand : sSmartctlCommands)
+    {
+        string sCMD = ("smartctl");
+        sCMD.append(sSmartctlCommand);
         sCMD.append(drive->getPath());
         const char *cpComand = sCMD.c_str();
 
+        //Logger::logThis()->info(cpComand);
+
         FILE *outputfileSmart = popen(cpComand, "r");
+        size_t len = 0U;     // length of found line
+        char *cLine = NULL; // found line
+        uint8_t status = 255U;
 
         while ((getline(&cLine, &len, outputfileSmart)) != -1)
         {
             string sLine = string(cLine);
 
-            SMART::parseModelFamily(sLine);
+            SMART::parseExitStatus(sLine, status);
-            SMART::parseModelName(sLine);
+            SMART::parseModelFamily(sLine, modelFamily);
-            SMART::parseSerial(sLine);
+            SMART::parseModelName(sLine, modelName);
-            SMART::parseCapacity(sLine);
+            SMART::parseSerial(sLine, serial);
-            SMART::parseErrorCount(sLine);
+            SMART::parseCapacity(sLine, capacity);
-            SMART::parsePowerOnHours(sLine);
+            SMART::parseErrorCount(sLine, errorCount);
-            SMART::parsePowerCycle(sLine);
+            SMART::parsePowerOnHours(sLine, powerOnHours);
+            SMART::parsePowerCycles(sLine, powerCycles);
+            SMART::parseTemperature(sLine, temperature);
         }
+
         pclose(outputfileSmart);
-    drive->setDriveSMARTData(modelFamily, modelName, serial, capacity, errorCount, powerOnHours, powerCycle); //wirte data in drive
+
+        if (status == 0U)
+        {
+            // Found S.M.A.R.T. data with this command
+            //Logger::logThis()->info("Found S.M.A.R.T. data with this command");
+            break;
+        }
+    }
+
+    drive->setDriveSMARTData(modelFamily, modelName, serial, capacity, errorCount, powerOnHours, powerCycles, temperature); // write data in drive
 }
 
 /**
- * \brief   parse ModelFamiliy
+ * \brief parse ExitStatus
  * \param string output line of smartctl
- * \return  void
+ * \param uint8_t parsed status
+ * \return bool if parsing was possible
  */
-void SMART::parseModelFamily(string sLine)
+bool SMART::parseExitStatus(string sLine, uint8_t &status)
+{
+    string search("\"exit_status\": ");
+    size_t found = sLine.find(search);
+    if (found != string::npos)
+    {
+        sLine.erase(0U, sLine.find(": ") + 1U);
+        status = stol(sLine);
+        return true;
+    }
+    else
+    {
+        return false;
+    }
+}
+
+/**
+ * \brief parse ModelFamily
+ * \param string output line of smartctl
+ * \param string parsed model family
+ * \return bool if parsing was possible
+ */
+bool SMART::parseModelFamily(string sLine, string &modelFamily)
 {
     string search("\"model_family\": ");
     size_t found = sLine.find(search);
     if (found != string::npos)
     {
-            sLine.erase(0, sLine.find(": ") + 3);
+        sLine.erase(0U, sLine.find(": ") + 3U);
-            sLine.erase(sLine.length()-3, 3);
+        sLine.erase(sLine.length() - 3U, 3U);
         modelFamily = sLine;
+        return true;
+    }
+    else
+    {
+        return false;
     }
 }
 
 /**
  * \brief parse ModelName
  * \param string output line of smartctl
- * \return  void
+ * \param string parsed model name
+ * \return bool if parsing was possible
  */
-void SMART::parseModelName(string sLine)
+bool SMART::parseModelName(string sLine, string &modelName)
 {
     string search("\"model_name\": ");
     size_t found = sLine.find(search);
     if (found != string::npos)
     {
-            sLine.erase(0, sLine.find(": ") + 3);
+        sLine.erase(0U, sLine.find(": ") + 3U);
-            sLine.erase(sLine.length()-3, 3);
+        sLine.erase(sLine.length() - 3U, 3U);
         modelName = sLine;
+        return true;
+    }
+    else
+    {
+        return false;
     }
 }
 
 /**
  * \brief parse Serial
  * \param string output line of smartctl
- * \return  void
+ * \param string parsed serial
+ * \return bool if parsing was possible
  */
-void SMART::parseSerial(string sLine)
+bool SMART::parseSerial(string sLine, string &serial)
 {
     string search("\"serial_number\": ");
     size_t found = sLine.find(search);
@@ -103,15 +154,21 @@ void SMART::parseSerial(string sLine)
         sLine.erase(0, sLine.find(": ") + 3);
         sLine.erase(sLine.length() - 3, 3);
         serial = sLine;
+        return true;
+    }
+    else
+    {
+        return false;
     }
 }
 
 /**
  * \brief parse Capacity
  * \param string output line of smartctl
- * \return  void
+ *  \param string parsed capacity
+ * \return bool if parsing was possible
  */
-void SMART::parseCapacity(string sLine)
+bool SMART::parseCapacity(string sLine, uint64_t &capacity)
 {
     string search("\"bytes\": ");
     size_t found = sLine.find(search);
@@ -120,50 +177,67 @@ void SMART::parseCapacity(string sLine)
         sLine.erase(0, sLine.find(": ") + 2);
         sLine.erase(sLine.length() - 1, 1);
         capacity = stol(sLine);
+        return true;
+    }
+    else
+    {
+        return false;
     }
 }
 
 /**
  * \brief parse ErrorCount
  * \param string output line of smartctl
- * \return  void
+ * \param uint32_t parsed error count
+ * \return bool if parsing was possible
  */
-void SMART::parseErrorCount(string sLine)
+bool SMART::parseErrorCount(string sLine, uint32_t &errorCount)
 {
     string search("\"error_count_total\": ");
     size_t found = sLine.find(search);
     if (found != string::npos)
     {
-            sLine.erase(0, sLine.find(": ")+2);
+        sLine.erase(0U, sLine.find(": ") + 2U);
-            sLine.erase(sLine.length()-2, 2);
+        sLine.erase(sLine.length() - 2U, 2U);
         errorCount = stol(sLine);
+        return true;
+    }
+    else
+    {
+        return false;
     }
 }
 
 /**
  * \brief parse PowerOnHours
- * \param	string output line of smartctl
+ * \param string output line of smartctl\
- * \return  void
+ * \param uint32_t parsed power on hours
+ * \return bool if parsing was possible
  */
-void SMART::parsePowerOnHours(string sLine)
+bool SMART::parsePowerOnHours(string sLine, uint32_t &powerOnHours)
 {
     string search("\"hours\": ");
     size_t found = sLine.find(search);
     if (found != string::npos)
     {
-            sLine.erase(0, sLine.find(": ") + 2);
+        sLine.erase(0U, sLine.find(": ") + 2U);
-            sLine.erase(sLine.length()-1, 1);
+        sLine.erase(sLine.length() - 1U, 1U);
         powerOnHours = stol(sLine);
-
+        return true;
+    }
+    else
+    {
+        return false;
     }
 }
 
 /**
  * \brief parse PowerCycle
  * \param string output line of smartctl
- * \return  void
+ * \param uint32_t parsed power cycles
+ * \return bool if parsing was possible
  */
-void SMART::parsePowerCycle(string sLine)
+bool SMART::parsePowerCycles(string sLine, uint32_t &powerCycles)
 {
     string search("\"power_cycle_count\": ");
     size_t found = sLine.find(search);
@@ -171,8 +245,41 @@ void SMART::parsePowerCycle(string sLine)
     {
         sLine.erase(0, sLine.find(": ") + 2);
         sLine.erase(sLine.length() - 2, 2);
-            powerCycle = stol(sLine);
+        powerCycles = stol(sLine);
-
+        return true;
+    }
+    else
+    {
+        return false;
     }
 }
 
+/**
+ * \brief parse temperature
+ * \param string output line of smartctl
+ * \param uint32_t parsed temperature
+ * \return bool if parsing was possible
+ */
+bool SMART::parseTemperature(string sLine, uint32_t &temperature)
+{
+    string search("\"current\": ");
+    size_t found = sLine.find(search);
+    if (found != string::npos)
+    {
+        sLine.erase(0U, sLine.find(": ") + 2U);
+        sLine.erase(sLine.length() - 1U, 2U);
+        if (sLine == "{")
+        {
+            temperature = 0U; // this drive doesn't support temperature
+        }
+        else
+        {
+            temperature = stol(sLine);
+        }
+        return true;
+    }
+    else
+    {
+        return false;
+    }
+}

src/tui.cpp

@@ -41,12 +41,23 @@ void TUI::initTUI()
     init_pair(COLOR_AREA_STDSCR, COLOR_WHITE, COLOR_BLUE);
     wbkgd(stdscr, COLOR_PAIR(COLOR_AREA_STDSCR));
 
-    init_pair(COLOR_AREA_ENTRY, COLOR_BLACK, COLOR_WHITE);
+    init_pair(COLOR_AREA_ENTRY_EVEN, COLOR_BLACK, COLOR_WHITE);
-    init_pair(COLOR_AREA_ENTRY_SELECTED, COLOR_BLACK, COLOR_RED);
+    init_pair(COLOR_AREA_ENTRY_ODD, COLOR_BLUE, COLOR_WHITE);
+
+#ifdef DRYRUN
+    init_pair(COLOR_AREA_ENTRY_SELECTED, COLOR_WHITE, COLOR_GREEN);
+#else
+    init_pair(COLOR_AREA_ENTRY_SELECTED, COLOR_WHITE, COLOR_RED);
+#endif
+
     init_pair(COLOR_AREA_OVERVIEW, COLOR_BLACK, COLOR_WHITE);
     init_pair(COLOR_AREA_DETAIL, COLOR_BLACK, COLOR_WHITE);
 
+#ifdef DRYRUN
+    mvprintw(0, 2, "reHDD - HDD refurbishing tool - GPL 3.0    DRYRUN is active! Don't use in production!");
+#else
     mvprintw(0, 2, "reHDD - HDD refurbishing tool - GPL 3.0 ");
+#endif
     Logger::logThis()->info("UI successfully initialized");
 }
 
@@ -61,10 +72,12 @@ void TUI::updateTUI(list <Drive>* plistDrives, uint8_t u8SelectedEntry)
 
     refresh();
 
-    overview=createOverViewWindow((int)(u16StdscrX/3), (u16StdscrY-3));
+    // overview window is 3/7 of the x-size
+    overview = createOverViewWindow((int)(u16StdscrX * (float)(3.0 / 7.0)), (u16StdscrY - 1));
     wrefresh(overview);
 
-    systemview=createSystemStats((int)(u16StdscrX/3), 10, u16StdscrX-(int)(u16StdscrX/3)-2, (u16StdscrY-11 ));
+    // system stat window is 2/7 of the x-size
+    systemview = createSystemStats(((int)(u16StdscrX * (float)(2.0 / 7.0))) - 6, 12, (int)(u16StdscrX * (float)(5.0 / 7.0) + 4), (u16StdscrY - 13));
     wrefresh(systemview);
 
     delwin(detailview);
@@ -74,11 +87,12 @@ void TUI::updateTUI(list <Drive>* plistDrives, uint8_t u8SelectedEntry)
     for (it = plistDrives->begin(); it != plistDrives->end(); ++it)
     {
         string sModelFamily = it->getModelFamily();
-            string sModelName = it->getModelName();
+        string sSerial = "SN: " + it->getSerial();
         string sCapacity = it->sCapacityToText();
         string sState = " ";
+        string sSpeed = " ";
         string sTime = " ";
-
+        string sTemp = it->sTemperatureToText();
 
         bool bSelectedEntry = false;
 
@@ -87,17 +101,16 @@ void TUI::updateTUI(list <Drive>* plistDrives, uint8_t u8SelectedEntry)
             bSelectedEntry = true; // mark this drive in entries list
             displaySelectedDrive(*it, u16StdscrX, u16StdscrY);
 
-                    if((it->getPowerOnHours() >= WORSE_HOURS) || (it->getPowerCycles() >= WORSE_POWERUP) || (it->getErrorCount() > 0))
+            if ((it->getPowerOnHours() >= WORSE_HOURS) || (it->getPowerCycles() >= WORSE_POWERUP) || (it->getErrorCount() > 0) || (it->getTemperature() >= WORSE_TEMPERATURE))
             {
                 // smart values are bad --> show warning
-                            smartWarning=createSmartWarning(50, 10, ((u16StdscrX)-(int)(u16StdscrX/2)+35),(int)(u16StdscrY/2)-5, it->getPath(), it->getPowerOnHours(), it->getPowerCycles(), it->getErrorCount());
+                smartWarning = createSmartWarning(50, 10, ((u16StdscrX) - (int)(u16StdscrX / 2) + 35), (int)(u16StdscrY / 2) - 5, it->getPath(), it->getPowerOnHours(), it->getPowerCycles(), it->getErrorCount(), it->getTemperature());
                 wrefresh(smartWarning);
             }
         }
 
         stringstream stream;
 
-
         switch (it->state)
         {
         case Drive::SHRED_ACTIVE:
@@ -107,30 +120,46 @@ void TUI::updateTUI(list <Drive>* plistDrives, uint8_t u8SelectedEntry)
 
             it->calculateTaskDuration();
             sTime = this->formatTimeDuration(it->getTaskDuration());
+            sSpeed = this->formatSpeed(it->sShredSpeed.u32ShredTimeDelta, it->sShredSpeed.ulWrittenBytes);
+            break;
+        case Drive::CHECK_ACTIVE:
+            stream << fixed << setprecision(3) << (it->getTaskPercentage());
+            sState = "Checking: " + stream.str() + "%";
+            it->calculateTaskDuration();
+            sTime = this->formatTimeDuration(it->getTaskDuration());
+            sSpeed = this->formatSpeed(it->sShredSpeed.u32ShredTimeDelta, it->sShredSpeed.ulWrittenBytes);
             break;
         case Drive::DELETE_ACTIVE:
             sState = "Deleting ...";
             it->calculateTaskDuration();
             sTime = this->formatTimeDuration(it->getTaskDuration());
             break;
-
         case Drive::NONE:
         case Drive::SHRED_SELECTED:
         case Drive::DELETE_SELECTED:
-                    if (it->bWasDeleteted)
+            if (it->bWasDeleted)
             {
                 sState = "DELETED"; // mark drive as deleted previously
             }
             if (it->bWasShredded)
             {
-                            sState = "SHREDDED"; //mark drive as shreded previously, overwrite if deleted
+                if (it->bWasChecked)
+                {
+                    // drive was also checked after shredding
+                    sState = "SHREDDED & CHECKED"; // mark drive as shredded previously and optional checked
+                }
+                else
+                {
+                    // shredded and not checked yet
+                    sState = "SHREDDED"; // mark drive as shredded previously
+                }
                 sTime = this->formatTimeDuration(it->getTaskDuration());
             }
 
-#ifdef ZERO_CHECK_ALERT
+#ifdef ZERO_CHECK
-                    if(bSelectedEntry && it->bWasShredded && (it->u32DriveChecksumAferShredding != 0U))
+            if (bSelectedEntry && it->bWasChecked && (it->u32DriveChecksumAfterShredding != 0U))
             {
-                            dialog=createZeroChecksumWarning(70, 16, ((u16StdscrX)-(int)(u16StdscrX/2)-20),(int)(u16StdscrY/2)-8, it->getPath(), it->getModelFamily(), it->getModelName(), it->getSerial(), it->u32DriveChecksumAferShredding);
+                dialog = createZeroChecksumWarning(70, 16, ((u16StdscrX) - (int)(u16StdscrX / 2) - 20), (int)(u16StdscrY / 2) - 8, it->getPath(), it->getModelFamily(), it->getModelName(), it->getSerial(), it->u32DriveChecksumAfterShredding);
                 wrefresh(dialog);
             }
 #endif
@@ -151,7 +180,9 @@ void TUI::updateTUI(list <Drive>* plistDrives, uint8_t u8SelectedEntry)
             break;
         }
 
-            WINDOW * tmp = createEntryWindow( ((int)(u16StdscrX/3) - 2), 5, 3, (5* (u8Index) )+3, sModelFamily, sModelName, sCapacity, sState, sTime, bSelectedEntry);
+        uint16_t u16StartOffsetY = (2 * (u8Index));
+
+        WINDOW *tmp = createEntryWindow((int)(u16StdscrX * (float)(3.0 / 7.0) - 2), 2, 3, u16StartOffsetY + 2, (distance(plistDrives->begin(), it) + 1), sModelFamily, sSerial, sCapacity, sState, sTime, sSpeed, sTemp, bSelectedEntry);
         wrefresh(tmp);
         u8Index++;
     } // end loop though drives
@@ -167,11 +198,11 @@ void TUI::updateTUI(list <Drive>* plistDrives, uint8_t u8SelectedEntry)
         menustate.bDelete = false;
         menustate.bShred = false;
 
-            menuview=createMenuView(((int)(u16StdscrX/3)-10 ), 10, (int)(u16StdscrX/3)+5,(u16StdscrY-11), menustate);
+        detailview = overwriteDetailViewWindow((u16StdscrX) - ((int)(u16StdscrX * (float)(3.0 / 7.0))) - 7, (u16StdscrY - 15), (int)(u16StdscrX * (float)(3.0 / 7.0) + 5));
-            wrefresh(menuview);
-
-            detailview=overwriteDetailViewWindow(((u16StdscrX)-(int)(u16StdscrX/3)-7), (u16StdscrY-15), (int)(u16StdscrX/3)+5);
         wrefresh(detailview);
+
+        menuview = createMenuView(((int)(u16StdscrX * (float)(2.0 / 7.0))) - 3, 12, (int)(u16StdscrX * (float)(3.0 / 7.0) + 5), (u16StdscrY - 13), menustate);
+        wrefresh(menuview);
     }
 
     mxUIrefresh.unlock();
@@ -203,6 +234,9 @@ enum TUI::UserInput TUI::readUserInput()
     case 's':
         return TUI::UserInput::Shred;
         break;
+    case 'S':
+        return TUI::UserInput::ShredAll;
+        break;
     default:
         return TUI::UserInput::Undefined;
         break;
@@ -223,10 +257,13 @@ void TUI::centerTitle(WINDOW *pwin, const char * title)
     waddch(pwin, ACS_LTEE);
 }
 
+/*
+left window that contains the drive entries
+*/
 WINDOW *TUI::createOverViewWindow(int iXSize, int iYSize)
 {
     WINDOW *newWindow;
-    newWindow = newwin(iYSize, iXSize, 2, 2);
+    newWindow = newwin(iYSize, iXSize, 1, 2);
 
     wbkgd(newWindow, COLOR_PAIR(COLOR_AREA_OVERVIEW));
     box(newWindow, ACS_VLINE, ACS_HLINE);
@@ -239,7 +276,7 @@ WINDOW* TUI::createOverViewWindow( int iXSize, int iYSize)
 WINDOW *TUI::createDetailViewWindow(int iXSize, int iYSize, int iXStart, Drive drive)
 {
     WINDOW *newWindow;
-    newWindow = newwin(iYSize, iXSize, 2, iXStart);
+    newWindow = newwin(iYSize, iXSize, 1, iXStart);
     wbkgd(newWindow, COLOR_PAIR(COLOR_AREA_DETAIL));
     box(newWindow, ACS_VLINE, ACS_HLINE);
 
@@ -247,7 +284,7 @@ WINDOW* TUI::createDetailViewWindow( int iXSize, int iYSize, int iXStart, Drive
     centerTitle(newWindow, title.c_str());
 
     string sPath = "Path:          " + drive.getPath();
-    string sModelFamlily = "ModelFamily:   " + drive.getModelFamily();
+    string sModelFamily = "ModelFamily:   " + drive.getModelFamily();
     string sModelName = "ModelName:     " + drive.getModelName();
     string sCapacity = "Capacity:      " + drive.sCapacityToText();
     string sSerial = "Serial:        " + drive.getSerial();
@@ -258,7 +295,7 @@ WINDOW* TUI::createDetailViewWindow( int iXSize, int iYSize, int iXStart, Drive
     uint16_t u16Line = 2;
 
     mvwaddstr(newWindow, u16Line++, 3, sPath.c_str());
-    mvwaddstr(newWindow,u16Line++, 3, sModelFamlily.c_str());
+    mvwaddstr(newWindow, u16Line++, 3, sModelFamily.c_str());
     mvwaddstr(newWindow, u16Line++, 3, sModelName.c_str());
     mvwaddstr(newWindow, u16Line++, 3, sCapacity.c_str());
     mvwaddstr(newWindow, u16Line++, 3, sSerial.c_str());
@@ -275,7 +312,7 @@ WINDOW* TUI::createDetailViewWindow( int iXSize, int iYSize, int iXStart, Drive
 WINDOW *TUI::overwriteDetailViewWindow(int iXSize, int iYSize, int iXStart)
 {
     WINDOW *newWindow;
-    newWindow = newwin(iYSize, iXSize, 2, iXStart);
+    newWindow = newwin(iYSize, iXSize, 1, iXStart);
     wbkgd(newWindow, COLOR_PAIR(COLOR_AREA_DETAIL));
     box(newWindow, ACS_VLINE, ACS_HLINE);
 
@@ -287,7 +324,7 @@ WINDOW* TUI::overwriteDetailViewWindow( int iXSize, int iYSize, int iXStart)
     string sLine03 = "Available under GPL 3.0";
     string sLine04 = "https://git.mosad.xyz/localhorst/reHDD";
     string sLine05 = "Delete: Wipe format table - this is NOT secure";
-    string sLine06 =  "Shred: Overwite drive " + to_string(SHRED_ITERATIONS) + " iterations - this is secure";
+    string sLine06 = "Shred: Overwrite drive " + to_string(SHRED_ITERATIONS) + " iterations - this is secure";
 
     uint16_t u16Line = 5;
 
@@ -305,7 +342,7 @@ WINDOW* TUI::overwriteDetailViewWindow( int iXSize, int iYSize, int iXStart)
     return newWindow;
 }
 
-WINDOW* TUI::createEntryWindow(int iXSize, int iYSize, int iXStart, int iYStart, string sModelFamily, string sModelName, string sCapacity, string sState, string sTime, bool bSelected)
+WINDOW *TUI::createEntryWindow(int iXSize, int iYSize, int iXStart, int iYStart, int iListIndex, string sModelFamily, string sSerial, string sCapacity, string sState, string sTime, string sSpeed, string sTemp, bool bSelected)
 {
     WINDOW *newWindow;
     newWindow = newwin(iYSize, iXSize, iYStart, iXStart);
@@ -313,24 +350,65 @@ WINDOW* TUI::createEntryWindow(int iXSize, int iYSize, int iXStart, int iYStart,
     if (!bSelected)
     {
         // entry is NOT selected
-            attron(COLOR_PAIR(COLOR_AREA_ENTRY));
+        if (iListIndex % 2 == 0)
-            wbkgd(newWindow, COLOR_PAIR(COLOR_AREA_ENTRY));
+        {
+            // even
+            attron(COLOR_PAIR(COLOR_AREA_ENTRY_EVEN));
+            wbkgd(newWindow, COLOR_PAIR(COLOR_AREA_ENTRY_EVEN));
+        }
+        else
+        {
+            // odd
+            attron(COLOR_PAIR(COLOR_AREA_ENTRY_ODD));
+            wbkgd(newWindow, COLOR_PAIR(COLOR_AREA_ENTRY_ODD));
+        }
     }
     else
     {
         // entry IS selected
-            attron(COLOR_PAIR(COLOR_AREA_ENTRY));
+        attron(COLOR_PAIR(COLOR_AREA_ENTRY_SELECTED));
         wbkgd(newWindow, COLOR_PAIR(COLOR_AREA_ENTRY_SELECTED));
     }
 
-    box(newWindow, ACS_VLINE, ACS_HLINE);
+    // box(newWindow, ACS_VLINE, ACS_HLINE);
 
-    mvwaddstr(newWindow,1, 1, sModelFamily.c_str());
+    // index number
-    mvwaddstr(newWindow,2, 1, sModelName.c_str());
+    mvwaddstr(newWindow, 0, 1, to_string(iListIndex).c_str());
-    mvwaddstr(newWindow,3, 1, sCapacity.c_str());
 
-    mvwaddstr(newWindow,2, iXSize-sState.length()-5, sState.c_str());
+    /*
-    mvwaddstr(newWindow,3, iXSize-sState.length()-5, sTime.c_str());
+    70 chars in x-axis
+
+    line:01
+        0:      space
+        1:      index number
+        2:      space
+        3-35:   ModelFamily
+        36:     space
+        37-43:  Capacity
+        44:     space
+        47-49:  Temp
+
+    line:02
+        0-2:    space
+        3-31:   Serial
+        32:     space
+        33-45:  Speed
+        46:     space
+        47-58:  Time
+        59:     space
+        60-70:  State (but right side aligned)
+    */
+
+    vTruncateText(&sModelFamily, 32);
+    mvwaddstr(newWindow, 0, 3, sModelFamily.c_str());
+    mvwaddstr(newWindow, 0, 37, sCapacity.c_str());
+    mvwaddstr(newWindow, 0, 47, sTemp.c_str());
+
+    vTruncateText(&sSerial, 28);
+    mvwaddstr(newWindow, 1, 3, sSerial.c_str());
+    mvwaddstr(newWindow, 1, 33, sSpeed.c_str());
+    mvwaddstr(newWindow, 1, 47, sTime.c_str());
+    mvwaddstr(newWindow, 1, iXSize - sState.length() - 2, sState.c_str());
 
     return newWindow;
 }
@@ -355,15 +433,21 @@ WINDOW* TUI::createSystemStats(int iXSize, int iYSize, int iXStart, int iYStart)
 
     string sLine01 = "reHDD - hard drive refurbishing tool";
     string sLine02 = "Version: " + string(REHDD_VERSION);
-    string sLine03 =  "Available under GPL 3.0";
+    string sLine03 = "Build time: ";
-    string sLine04 =  "https://git.mosad.xyz/localhorst/reHDD";
+    sLine03.append(__DATE__);
+    sLine03.append(" ");
+    sLine03.append(__TIME__);
+    string sLine04 = "Available under GPL 3.0";
+    string sLine05 = "https://git.mosad.xyz/localhorst/reHDD";
 
     uint16_t u16Line = 2;
 
     mvwaddstr(newWindow, u16Line++, (iXSize / 2) - (sLine01.size() / 2), sLine01.c_str());
+    u16Line++;
     mvwaddstr(newWindow, u16Line++, (iXSize / 2) - (sLine01.size() / 2), sLine02.c_str());
     mvwaddstr(newWindow, u16Line++, (iXSize / 2) - (sLine01.size() / 2), sLine03.c_str());
     mvwaddstr(newWindow, u16Line++, (iXSize / 2) - (sLine01.size() / 2), sLine04.c_str());
+    mvwaddstr(newWindow, u16Line++, (iXSize / 2) - (sLine01.size() / 2), sLine05.c_str());
     u16Line++;
     mvwaddstr(newWindow, u16Line++, (iXSize / 2) - (sLine01.size() / 2), time.c_str());
 
@@ -380,23 +464,23 @@ WINDOW* TUI::createMenuView(int iXSize, int iYSize, int iXStart, int iYStart, st
 
     centerTitle(newWindow, "Controls");
 
-    uint16_t u16Line = 2;
+    uint16_t u16Line = 4;
 
     if (menustate.bAbort)
     {
-            string sLineTmp = "Press A for Abort";
+        string sLineTmp = "Press a for Abort";
         mvwaddstr(newWindow, u16Line++, (iXSize / 2) - (sLineTmp.size() / 2), sLineTmp.c_str());
         u16Line++;
     }
     if (menustate.bShred)
     {
-            string sLineTmp = "Press S for Shred ";
+        string sLineTmp = "Press s for Shred (S for all drives)";
         mvwaddstr(newWindow, u16Line++, (iXSize / 2) - (sLineTmp.size() / 2), sLineTmp.c_str());
         u16Line++;
     }
     if (menustate.bDelete)
     {
-            string sLineTmp = "Press D for Delete";
+        string sLineTmp = "Press d for Delete";
         mvwaddstr(newWindow, u16Line++, (iXSize / 2) - (sLineTmp.size() / 2), sLineTmp.c_str());
     }
 
@@ -469,7 +553,7 @@ WINDOW* TUI::createZeroChecksumWarning(int iXSize, int iYSize, int iXStart, int
     string sLine01 = "Please detach this drive and check it manually:";
     string sShredChecksum = "After shredding the checksum was: " + to_string(u32Checksum);
     string sLinePath = "Path:          " + sPath;
-    string sLineModelFamlily = "ModelFamily:   " + sModelFamily;
+    string sLineModelFamily = "ModelFamily:   " + sModelFamily;
     string sLineModelName = "ModelName:     " + sModelName;
     string sLineSerial = "Serial:        " + sSerial;
 
@@ -482,7 +566,7 @@ WINDOW* TUI::createZeroChecksumWarning(int iXSize, int iYSize, int iXStart, int
     mvwaddstr(newWindow, u16Line++, 3, sLine01.c_str());
     u16Line++;
     mvwaddstr(newWindow, u16Line++, 3, sLinePath.c_str());
-    mvwaddstr(newWindow,u16Line++, 3, sLineModelFamlily.c_str());
+    mvwaddstr(newWindow, u16Line++, 3, sLineModelFamily.c_str());
     mvwaddstr(newWindow, u16Line++, 3, sLineModelName.c_str());
     mvwaddstr(newWindow, u16Line++, 3, sLineSerial.c_str());
     u16Line++;
@@ -506,9 +590,30 @@ string TUI::formatTimeDuration(time_t u32Duration)
     return out.str();
 }
 
+string TUI::formatSpeed(time_t u32ShredTimeDelta, unsigned long ulWrittenBytes)
+{
+    std::ostringstream out;
+    double dDeltaSec = ((double)((u32ShredTimeDelta) / 1000000000.0)); // convert nano in sec
+    double speed = ((ulWrittenBytes / 1000000.0) / dDeltaSec);
+    char s[25];
+    sprintf(s, "%0.2lf MB/s", speed);
+    out << s;
+    return out.str();
+}
+
+void TUI::vTruncateText(string *psText, uint16_t u16MaxLenght)
+{
+    if (psText->length() > u16MaxLenght)
+    {
+        psText->resize(u16MaxLenght - 3);
+        *psText = *psText + "...";
+    }
+}
+
 void TUI::displaySelectedDrive(Drive drive, int stdscrX, int stdscrY)
 {
     struct MenuState menustate;
+    static bool dialogIsActive;
     menustate.bAbort = false;
     menustate.bConfirmDelete = false;
     menustate.bConfirmShred = false;
@@ -530,6 +635,10 @@ void TUI::displaySelectedDrive(Drive drive, int stdscrX, int stdscrY)
         menustate.bAbort = true;
         break;
 
+    case Drive::CHECK_ACTIVE: // check task running for this drive
+        menustate.bAbort = true;
+        break;
+
     case Drive::DELETE_SELECTED: // delete task selected for this drive
         menustate.bConfirmDelete = true;
         break;
@@ -541,29 +650,35 @@ void TUI::displaySelectedDrive(Drive drive, int stdscrX, int stdscrY)
         break;
     }
 
-    detailview=createDetailViewWindow(((stdscrX)-(int)(stdscrX/3)-7), (stdscrY-15), (int)(stdscrX/3)+5, drive);
+    detailview = createDetailViewWindow((stdscrX) - ((int)(stdscrX * (float)(3.0 / 7.0))) - 7, (stdscrY - 15), (int)(stdscrX * (float)(3.0 / 7.0) + 5), drive);
     wrefresh(detailview);
 
-    menuview=createMenuView(((int)(stdscrX/3)-10 ), 10, (int)(stdscrX/3)+5,(stdscrY-11), menustate);
+    menuview = createMenuView(((int)(stdscrX * (float)(2.0 / 7.0))) - 3, 12, (int)(stdscrX * (float)(3.0 / 7.0) + 5), (stdscrY - 13), menustate);
     wrefresh(menuview);
 
     if (menustate.bConfirmShred == true)
     {
         dialog = createDialog(40, 10, ((stdscrX) - (int)(stdscrX / 3) - 7) - (int)((stdscrX / 3) + 5) / 2, (int)(stdscrY / 2) - 5, "Confirm SHRED", "Press ENTER for SHRED", "Press ESC for cancel");
         wrefresh(dialog);
+        dialogIsActive = true;
     }
     else if (menustate.bConfirmDelete == true)
     {
         dialog = createDialog(40, 10, ((stdscrX) - (int)(stdscrX / 3) - 7) - (int)((stdscrX / 3) + 5) / 2, (int)(stdscrY / 2) - 5, "Confirm DELETE", "Press ENTER for DELETE", "Press ESC for cancel");
         wrefresh(dialog);
+        dialogIsActive = true;
     }
     else
+    {
+        if (dialogIsActive)
         {
             delwin(dialog);
+            dialogIsActive = false;
+        }
     }
 }
 
-WINDOW* TUI::createSmartWarning(int iXSize, int iYSize, int iXStart, int iYStart, string sPath, uint32_t u32PowerOnHours, uint32_t u32PowerCycles,  uint32_t u32ErrorCount)
+WINDOW *TUI::createSmartWarning(int iXSize, int iYSize, int iXStart, int iYStart, string sPath, uint32_t u32PowerOnHours, uint32_t u32PowerCycles, uint32_t u32ErrorCount, uint32_t u32Temperature)
 {
     WINDOW *newWindow;
     newWindow = newwin(iYSize, iXSize, iYStart, iXStart);
@@ -596,7 +711,14 @@ WINDOW* TUI::createSmartWarning(int iXSize, int iYSize, int iXStart, int iYStart
 
     if (u32ErrorCount > 0)
     {
-            string sLineTmp = "S.M.A.R.T. erros detected: " + to_string(u32ErrorCount);
+        string sLineTmp = "S.M.A.R.T. errors detected: " + to_string(u32ErrorCount);
+        mvwaddstr(newWindow, u16Line++, (iXSize / 2) - (sLine01.size() / 2), sLineTmp.c_str());
+        u16Line++;
+    }
+
+    if (u32Temperature >= WORSE_TEMPERATURE)
+    {
+        string sLineTmp = "Drive too hot: " + to_string(u32Temperature) + " C";
         mvwaddstr(newWindow, u16Line++, (iXSize / 2) - (sLine01.size() / 2), sLineTmp.c_str());
     }
     return newWindow;