39 lines
936 B
Python
39 lines
936 B
Python
"""
|
|
taken from https://raw.githubusercontent.com/tree-sitter/tree-sitter/master/test/fuzz/gen-dict.py
|
|
"""
|
|
|
|
|
|
import json
|
|
import sys
|
|
|
|
def find_literals(literals, node):
|
|
'''Recursively find STRING literals in the grammar definition'''
|
|
|
|
if type(node) is dict:
|
|
if 'type' in node and node['type'] == 'STRING' and 'value' in node:
|
|
literals.add(node['value'])
|
|
|
|
for key, value in node.iteritems():
|
|
find_literals(literals, value)
|
|
|
|
elif type(node) is list:
|
|
for item in node:
|
|
find_literals(literals, item)
|
|
|
|
def main():
|
|
'''Generate a libFuzzer / AFL dictionary from a tree-sitter grammar.json'''
|
|
with open(sys.argv[1]) as f:
|
|
grammar = json.load(f)
|
|
|
|
literals = set()
|
|
find_literals(literals, grammar)
|
|
|
|
for lit in sorted(literals):
|
|
if lit:
|
|
print '// %s' % lit
|
|
print '"%s"' % ''.join(['\\x%02x' % ord(b) for b in lit.encode('utf-8')])
|
|
print
|
|
|
|
if __name__ == '__main__':
|
|
main()
|