#!/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 is passed as bytes from C; decode to str for dict lookup

if isinstance(keyword, bytes):

keyword = keyword.decode('latin-1')

keyword = keyword.upper()

if stype == libinjection.LOOKUP_FINGERPRINT:

# sqli_check_fingerprint calls sqli_blacklist (fingerprint membership

# check) and sqli_not_whitelist (false-positive reduction) internally.

if libinjection.sqli_check_fingerprint(state):

return 'F'

else:

return chr(0)

return words.get(keyword, chr(0))

""")

words = {}

keywords = obj['keywords']

for k,v in keywords.items():

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)))