106 lines
2.7 KiB
C++
106 lines
2.7 KiB
C++
/**
|
||
* @file reHDD.h
|
||
* @brief app logic header
|
||
* @author hendrik schutter
|
||
* @date 01.05.2020
|
||
*/
|
||
|
||
#ifndef REHDD_H_
|
||
#define REHDD_H_
|
||
|
||
#define REHDD_VERSION "bV0.3.0"
|
||
|
||
// 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 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 DEVICE_ID "generic"
|
||
#define SOFTWARE_VERSION "alpha"
|
||
#define HARDWARE_VERSION "generic"
|
||
|
||
//#define LOG_LEVEL_HIGH //log everything, like drive scan thread
|
||
#ifndef LOG_LEVEL_HIGH
|
||
#define LOG_LEVEL_LOW //log only user actions and tasks
|
||
#endif
|
||
|
||
// Logic
|
||
//#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
|
||
|
||
//IPC pipes
|
||
#define READ 0
|
||
#define WRITE 1
|
||
|
||
#include <iostream>
|
||
#include <string>
|
||
#include <fstream>
|
||
#include <tuple>
|
||
#include <list>
|
||
#include <time.h>
|
||
#include <chrono>
|
||
#include <curses.h>
|
||
#include <thread>
|
||
#include <unistd.h>
|
||
#include <mutex>
|
||
#include <sys/select.h>
|
||
#include <algorithm>
|
||
#include <cstring>
|
||
#include <sstream>
|
||
#include <iomanip>
|
||
#include <signal.h>
|
||
|
||
using namespace std;
|
||
|
||
#include "drive.h"
|
||
#include "smart.h"
|
||
#include "shred.h"
|
||
#include "delete.h"
|
||
#include "tui.h"
|
||
#include "logger/logger.h"
|
||
|
||
extern Logger* logging;
|
||
|
||
template <typename T, typename I> T* iterator_to_pointer(I i)
|
||
{
|
||
return (&(*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 filterNewDrives(list <Drive>* plistOldDrives, list <Drive>* plistNewDrives);
|
||
static void addSMARTData(list <Drive>* plistDrives);
|
||
static void ThreadScannDevices();
|
||
static void ThreadUserInput();
|
||
static void ThreadShred(Drive* const pDrive);
|
||
static void ThreadDelete();
|
||
static void ThreadCheckFrozenDrives();
|
||
static void handleArrowKey(TUI::UserInput userInput);
|
||
static void handleEnter();
|
||
static void handleESC();
|
||
static void handleAbort();
|
||
static Drive* getSelectedDrive();
|
||
};
|
||
|
||
#endif // REHDD_H_
|