display shred duration after completion

src/drive.cpp

@@ -134,7 +134,6 @@ void Drive::setDriveSMARTData( string modelFamily,
     u32PowerCycles = powerCycle;
 }
 
-
 void Drive::setTimestamp()
 {
     time(&this->u32Timestamp);
@@ -150,6 +149,19 @@ time_t Drive::getActionStartTimestamp()
     return this->u32TimestampTaskStart;
 }
 
+void Drive::calculateTaskDuration()
+{
+    time_t u32localtime;
+    time(&u32localtime);
+
+    this->u32TaskDuration = u32localtime - this->u32TimestampTaskStart;
+}
+
+time_t Drive::getTaskDuration()
+{
+    return this->u32TaskDuration;
+}
+
 void Drive::checkFrozenDrive(void)
 {
     time_t u32localtime;