reHDD/src/reHDD.cpp

/**
 * @file    reHDD.cpp
 * @brief   app logic
 * @author  hendrik schutter
 * @date    01.05.2020
 */

#include "../include/reHDD.h"

/**
 * \brief   app constructor
 * \param	void
 * \return  instance of App
 */
reHDD::reHDD(void)
{
    cout << "created app" << endl;
}

/**
 * \brief   app logic
 * \param	void
 * \return  void
 */
void reHDD::app_logic(void)
{
    cout << "app logic" << endl;

    searchDrives(&vecDrives);           //search for new drives and store them in list
    filterIgnoredDrives(&vecDrives);    //filter out ignored drives
    addSMARTData(&vecDrives);           //add S.M.A.R.T. Data to the drives
    printDrives(&vecDrives);            //print currently attached drives

    size_t u64SelectedDriveIndex = 0U;
    size_t u64DriveVecSize =  (vecDrives.size());

    cout << "Select drive to wipe:" << endl;
    cin >> u64SelectedDriveIndex;
    cout << "Selected drive index: " << u64SelectedDriveIndex << endl;

    if(u64SelectedDriveIndex < (u64DriveVecSize)) {
        Wipe::wipeDrive(&vecDrives[u64SelectedDriveIndex]);
    }

}

/**
 * \brief   search attached drives on /dev/sd*
 * \param	pointer of vector <Drive>* pvecDrives
 * \return  void
 */
void reHDD::searchDrives(vector <Drive>* pvecDrives)
{
    cout << "search drives ..." << endl;
    char * cLine = NULL;
    size_t len = 0;

    FILE* outputfileHwinfo = popen("./bin_util/hwinfo --short --disk", "r");

    if (outputfileHwinfo == NULL)
    {
        exit(EXIT_FAILURE);
    }

    while ((getline(&cLine, &len, outputfileHwinfo)) != -1)
    {
        if (string(cLine).find("/dev/sd") != string::npos)
        {
            Drive* tmpDrive = new Drive(string(cLine).substr (2,8));
            pvecDrives->push_back(*tmpDrive);
        }
    }
    fclose(outputfileHwinfo);
}

/**
 * \brief   filter out drives that are listed in "ignoreDrives.conf"
 * \param	pointer of vector <Drive>* pvecDrives
 * \return  void
 */
void reHDD::filterIgnoredDrives(vector <Drive>* pvecDrives)
{
    string sDelimiter = ":";
    string sIgnoredDrivePath;
    string sIgnoredDriveUUID;

    vector<tuple<string, string>> vtlIgnoredDevices; //store drives from ingnore file

    ifstream input( "ignoreDrives.conf" ); //read ingnore file

    for(string sLine; getline( input, sLine );)
    {
        if (string(sLine).find("/dev/sd") != string::npos)
        {
            size_t pos = 0;
            string token;
            while ((pos = sLine.find(sDelimiter)) != string::npos)
            {
                token = sLine.substr(0, pos);
                sIgnoredDrivePath = token;
                sLine.erase(0, pos + sDelimiter.length());
                sIgnoredDriveUUID = sLine;
            } //end while
            //cout << "Path: " << sIgnoredDrivePath << std::endl;
            //cout << "UUID: " << sIgnoredDriveUUID << std::endl;
            vtlIgnoredDevices.emplace_back(sIgnoredDrivePath, sIgnoredDriveUUID); //add found path and uuid from ingnore file to vector
        }
    }

    //loop through found entries in ingnore file
    for(auto row : vtlIgnoredDevices)
    {
        auto it = pvecDrives->begin();
        while (it != pvecDrives->end())
        {
            it++;
            string sUUID;
            if (!get<0>(row).compare(it->getPath())) //find same drive based on path
            {
                // cout << "Same drive path found" << endl;
                char * cLine = NULL;
                size_t len = 0;
                string sCMD = "./bin_util/blkid ";
                sCMD.append(it->getPath());
                //  cout << "cmd: " << sCMD << endl;
                FILE* outputfileBlkid = popen(sCMD.c_str(), "r"); //get UUID from drive
                if (outputfileBlkid == NULL)
                {
                    exit(EXIT_FAILURE);
                }

                while ((getline(&cLine, &len, outputfileBlkid)) != -1) //parse UUID from blkid
                {
                    if (string(cLine).find("PTUUID") != string::npos)
                    {
                        string sBlkidOut = string(cLine);
                        sBlkidOut.erase(0, 18);
                        sBlkidOut.erase(36, sBlkidOut.length() - 36);
                        sUUID = sBlkidOut;
                        //cout << "blkid uuid:" << sUUID << endl;
                    }
                }
                fclose(outputfileBlkid);
                //  cout << "blkid uuid:" << sUUID << endl;

                if (get<1>(row).compare(sUUID)) //compare uuid from ignore file and uuid from drive
                {
                    cout << "[ERROR] different uuid found than in ignore file:" << it->getPath() << endl;
                    exit(EXIT_FAILURE); // exit to prevent accidentally shred a system drive
                }
                else
                {
                    // same uuid found than in ignore file --> ignore this drive
                    it = pvecDrives->erase(it);
                }

            }
        }
    }
}

/**
 * \brief   print drives with all information
 * \param	pointer of vector <Drive>* pvecDrives
 * \return  void
 */
void reHDD::printDrives(vector <Drive>* pvecDrives)
{
    cout << "------------DRIVES---------------" << endl;
    vector <Drive>::iterator it;
    for (it = pvecDrives->begin(); it != pvecDrives->end(); ++it)
    {
        cout << "        Drive: " << distance(pvecDrives->begin(), it) << endl;
        cout << "Path:          " << it->getPath() << endl;
        cout << "ModelFamily:   " << it->getModelFamily() << endl;
        cout << "ModelName:     " << it->getModelName() << endl;
        cout << "Capacity:      " << it->getCapacity() << endl;
        cout << "Serial:        " << it->getSerial() << endl;
        cout << "PowerOnHours:  " << it->getPowerOnHours() << endl;
        cout << "PowerCycle:    " << it->getPowerCycles() << endl;
        cout << "ErrorCount:    " << it->getErrorCount() << endl;
        cout << endl;
    }
    cout << "---------------------------------" << endl;
}

/**
 * \brief   add S.M.A.R.T data from SMART
 * \param	pointer of vector <Drive>* pvecDrives
 * \return  void
 */
void reHDD::addSMARTData(vector <Drive>* pvecDrives)
{
    vector <Drive>::iterator it;
    for (it = pvecDrives->begin(); it != pvecDrives->end(); ++it)
    {
        Drive* pTmpDrive = iterator_to_pointer<Drive, std::vector<Drive>::iterator > (it);
        SMART::readSMARTData(pTmpDrive);
    }
}