105 lines
2.8 KiB
Python
105 lines
2.8 KiB
Python
#
|
|
#
|
|
#
|
|
import argparse
|
|
import time
|
|
from libs import fingerprint
|
|
from libs import toolbox
|
|
|
|
|
|
def main():
|
|
parseArgs()
|
|
|
|
# db = fingerprint.DBSchema()
|
|
|
|
# if (filein and filejson):
|
|
# db.scanDBFile(filein)
|
|
# db.compareDB(filejson)
|
|
# elif (filein):
|
|
# retVal = db.scanDBFile(filein)
|
|
# if (retVal > 0):
|
|
# fh = open(fileout, "w")
|
|
# if verbose:
|
|
# db.debugFingerprint()
|
|
# if title:
|
|
# db.setTitle(title)
|
|
# if notes:
|
|
# db.setNotes(notes)
|
|
# db.writeFingerprint(fh)
|
|
# fh.close()
|
|
# else:
|
|
# print db.getErrorString(retVal)
|
|
# elif (filejson):
|
|
# db.importJson(filejson)
|
|
# db.debugFingerprint()
|
|
|
|
#
|
|
def compareFingerprintDir(filein, filejson, fpdir):
|
|
db = fingerprint.DBSchema()
|
|
|
|
#
|
|
def compareFingerprint(filein, filejson):
|
|
db = fingerprint.DBSchema()
|
|
db.scanDBFile(filein)
|
|
db.compareDB(filejson)
|
|
|
|
#
|
|
def createFingerprint(filein, fileout, verbose, app_name, app_ver, notes):
|
|
db = fingerprint.DBSchema()
|
|
retVal = db.scanDBFile(filein)
|
|
if (retVal > 0):
|
|
fh = open(fileout, "w")
|
|
if verbose:
|
|
db.debugFingerprint()
|
|
if app_name:
|
|
db.setAppName(app_name)
|
|
if app_ver:
|
|
db.setAppVer(app_ver)
|
|
if notes:
|
|
db.setNotes(notes)
|
|
db.writeFingerprint(fh)
|
|
fh.close()
|
|
else:
|
|
print db.getErrorString(retVal)
|
|
|
|
#
|
|
def parseArgs():
|
|
print '***** ***** ***** *****'
|
|
print ' DB Fingerprint'
|
|
print '***** ***** ***** *****\n'
|
|
fileout = ''
|
|
timestr = time.strftime('%Y-%m-%d_%H%M%S', time.localtime(time.time()))
|
|
parser = argparse.ArgumentParser(description='Fingerprint a sqlite database based on its schema')
|
|
parser.add_argument('-f', '--file', required=False)
|
|
parser.add_argument('-fd', '--fpdir', required=False)
|
|
parser.add_argument('-j', '--json', required=False)
|
|
parser.add_argument('-an', '--app_name', required=False)
|
|
parser.add_argument('-av', '--app_version', required=False)
|
|
parser.add_argument('-n', '--notes', required=False)
|
|
parser.add_argument('-v', '--verbose', action='store_true')
|
|
# parser.add_argument('-t', '--title', required=False)
|
|
args = parser.parse_args()
|
|
|
|
if (args.file is None) and (args.json is None):
|
|
parser.print_help()
|
|
exit()
|
|
|
|
|
|
# compare a sqlite database file to all fingerprints
|
|
if (args.json and args.fpdir):
|
|
compareFingerprints(args.file, args.json, args.fpdir)
|
|
# compare a sqlite database file to a fingerprint
|
|
if (args.json):
|
|
compareFingerprint(args.file, args.json)
|
|
# create a fingerprint from the sqlite file
|
|
filename = toolbox.ToolBox.parseFilename(args.file)
|
|
fileout = filename + "_" + timestr + '.json'
|
|
createFingerprint(args.file, fileout, args.verbose, args.app_name, args.app_version, args.notes)
|
|
|
|
# return (args.file, fileout, args.json, args.verbose, args.title, args.notes,
|
|
# args.fpdir)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|