53 lines
1.1 KiB
Python
53 lines
1.1 KiB
Python
|
#!/usr/bin/env python
|
||
|
|
#
|
||
|
|
# Copyright 2012, 2013 Nick Galbreath
|
||
|
|
# nickg@client9.com
|
||
|
|
# BSD License -- see COPYING.txt for details
|
||
|
|
#
|
||
|
|
|
||
|
|
"""
|
||
|
|
Converts a libinjection JSON data file to python dict
|
||
|
|
"""
|
||
|
|
|
||
|
|
def toc(obj):
|
||
|
|
""" main routine """
|
||
|
|
|
||
|
|
print """
|
||
|
|
import libinjection
|
||
|
|
|
||
|
|
def lookup(state, stype, keyword):
|
||
|
|
keyword = keyword.upper()
|
||
|
|
if stype == libinjection.LOOKUP_FINGERPRINT:
|
||
|
|
if keyword in fingerprints and libinjection.sqli_not_whitelist(state):
|
||
|
|
return 'F'
|
||
|
|
else:
|
||
|
|
return chr(0)
|
||
|
|
return words.get(keyword, chr(0))
|
||
|
|
|
||
|
|
"""
|
||
|
|
|
||
|
|
words = {}
|
||
|
|
keywords = obj['keywords']
|
||
|
|
for k,v in keywords.iteritems():
|
||
|
|
words[str(k)] = str(v)
|
||
|
|
|
||
|
|
print 'words = {'
|
||
|
|
for k in sorted(words.keys()):
|
||
|
|
print "'{0}': '{1}',".format(k, words[k])
|
||
|
|
print '}\n'
|
||
|
|
|
||
|
|
|
||
|
|
keywords = obj['fingerprints']
|
||
|
|
print 'fingerprints = set(['
|
||
|
|
for k in sorted(keywords):
|
||
|
|
print "'{0}',".format(k.upper())
|
||
|
|
print '])'
|
||
|
|
|
||
|
|
return 0
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
import sys
|
||
|
|
import json
|
||
|
|
sys.exit(toc(json.load(sys.stdin)))
|
||
|
|
|