Add sanity check on DB path

The current error message is a bit cryptic
devoteam-cybertrust · Nov 9, 2013 · 70415aa · 70415aa
1 parent 97a16df
commit 70415aa
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/analyzer/DBParser.py b/analyzer/DBParser.py
@@ -32,7 +32,7 @@ def __init__(self, dbPath):
                     argsAndReturnValue = self._sanitize_args_dict(plistlib.readPlistFromString(row[2].encode('utf-8')))))
                 rowid += 1
         except sqlite3.Error as e:
-            print "Fatal error: %s" % e
+            #print "Fatal error: %s" % e
             raise
         finally:
             if SqlConn:

diff --git a/analyzer/introspy.py b/analyzer/introspy.py
@@ -8,6 +8,7 @@
 __copyright__ = "Copyright 2013, iSEC Partners, Inc."
 
 from sys import argv
+import os
 from argparse import ArgumentParser
 from re import match
 from DBAnalyzer import DBAnalyzer
@@ -68,6 +69,10 @@ def main(argv):
 
 
     # Process the DB
+    # Make sure it's there
+    if not os.path.exists(db_path): # Nice race condition
+        print 'Error: Could not find the DB file.'
+        return
     analyzedDB = DBAnalyzer(db_path)