Implemented hashing all recovered artifacts

Imanol-Mikel Barba Sabariego
1 parent 1d2034c1
Showing 1 changed file with 24 additions and 4 deletions
digger.py
@@ -16,6 +16,7 @@ import traceback
 import logger
 from consolelogger import ConsoleLogger
 from filelogger import FileLogger
+from hashlib import sha256
 def getModules():
     choiceList = []
@@ -44,7 +45,26 @@ def prepareModule(moduleName):
     except Exception as e:
         raise
-def showFinishDialog(allSuccessful):
+def sha256sum(file):
+    f = open(file, 'rb')
+    result = sha256(f.read()).hexdigest()
+    f.close()
+    return result
+
+def bagAndTag():
+    tombPath = tomb.getPath()
+    hashes = open(tombPath + "hashes.txt",'a+')
+
+    for root, dirnames, filenames in os.walk(tombPath):
+        for dir in dirnames:
+            print(dir)
+            for subroot, subdirnames, subfilenames in os.walk(tombPath + dir):
+                print(subfilenames)
+                for filename in subfilenames:
+                    hashes.write(sha256sum(subroot + "/" + filename) + " " + subroot + "/" + filename + "\n")
+
+def finish(allSuccessful):
+    bagAndTag()
     msg = ""
     if(allSuccessful):
         msg = "All modules finished execution"
@@ -103,7 +123,7 @@ if __name__ == &quot;__main__&quot;:
                             #logger.msgLog(traceback.format_exc(), module[0],  logger.TYPE_ERROR)
                             ans = showContinueDialog(d,msg)
                             if ans == "abort":
-                                showFinishDialog(False)
+                                finish(False)
                             elif ans == "continue":
                                 logger.msgLog("Skipping module " + module[0], "digger", logger.TYPE_INFO)
                                 break
@@ -121,7 +141,7 @@ if __name__ == &quot;__main__&quot;:
                             #logger.msgLog(traceback.format_exc(), tag logger.TYPE_ERROR)
                             ans = showContinueDialog(d, msg)
                             if ans == "abort":
-                                showFinishDialog(False)
+                                finish(False)
                             elif ans == "continue":
                                 logger.msgLog("Skipping module " + tag, "digger", logger.TYPE_INFO)
                                 break
@@ -136,4 +156,4 @@ if __name__ == &quot;__main__&quot;:
                     #logger.msgLog(traceback.format_exc(), module.name, logger.TYPE_ERROR)
-            showFinishDialog(True)
+            finish(True)