some more speed ups for hash cracking

stamparm · stamparm · commit ea125d820d98 · 2011-11-02T09:57:42.000Z
diff --git a/lib/utils/hash.py b/lib/utils/hash.py
@@ -630,10 +630,14 @@ def dictionaryAttack(attack_dict):
                         process.terminate()
                         process.join()
 
-                while not retVal.empty():
-                    _, hash_, word = item = retVal.get(block=False)
-                    conf.hashDB.write(hash_, word)
-                    results.append(item)
+                finally:
+                    if retVal:
+                        conf.hashDB.beginTransaction()
+                        while not retVal.empty():
+                            _, hash_, word = item = retVal.get(block=False)
+                            conf.hashDB.write(hash_, word)
+                            results.append(item)
+                        conf.hashDB.endTransaction()
 
             clearConsoleLine()
 
@@ -706,10 +710,14 @@ class Value():
                             process.terminate()
                             process.join()
 
-                    while not retVal.empty():
-                        _, hash_, word = item = retVal.get(block=False)
-                        conf.hashDB.write(hash_, word)
-                        results.append(item)
+                    finally:
+                        if retVal:
+                            conf.hashDB.beginTransaction()
+                            while not retVal.empty():
+                                _, hash_, word = item = retVal.get(block=False)
+                                conf.hashDB.write(hash_, word)
+                                results.append(item)
+                            conf.hashDB.endTransaction()
 
                 clearConsoleLine()
 
diff --git a/lib/utils/hashdb.py b/lib/utils/hashdb.py
@@ -75,3 +75,9 @@ def write(self, key, value):
                         raise
                 else:
                     break
+
+    def beginTransaction(self):
+        self.cursor.execute('BEGIN TRANSACTION')
+
+    def endTransaction(self):
+        self.cursor.execute('END TRANSACTION')
