mirror of
https://github.com/github/codeql.git
synced 2025-12-16 16:53:25 +01:00
54 lines
1.3 KiB
Python
54 lines
1.3 KiB
Python
import json
|
|
import os
|
|
import shutil
|
|
import subprocess
|
|
|
|
# Shared strings.
|
|
summaryModelPredicate = "summaryModel"
|
|
sinkModelPredicate = "sinkModel"
|
|
sourceModelPredicate = "sourceModel"
|
|
neutralModelPredicate = "neutralModel"
|
|
addsToTemplate = """ - addsTo:
|
|
pack: {0}
|
|
extensible: {1}
|
|
data:
|
|
{2}"""
|
|
|
|
def remove_dir(dirName):
|
|
if os.path.isdir(dirName):
|
|
shutil.rmtree(dirName)
|
|
print("Removed directory:", dirName)
|
|
|
|
def run_cmd(cmd, msg="Failed to run command"):
|
|
print('Running ' + ' '.join(cmd))
|
|
if subprocess.check_call(cmd):
|
|
print(msg)
|
|
exit(1)
|
|
|
|
def readData(workDir, bqrsFile):
|
|
generatedJson = os.path.join(workDir, "out.json")
|
|
print('Decoding BQRS to JSON.')
|
|
run_cmd(['codeql', 'bqrs', 'decode', bqrsFile, '--output', generatedJson, '--format=json'], "Failed to decode BQRS.")
|
|
|
|
with open(generatedJson) as f:
|
|
results = json.load(f)
|
|
|
|
try:
|
|
return results['#select']['tuples']
|
|
except KeyError:
|
|
print('Unexpected JSON output - no tuples found')
|
|
exit(1)
|
|
|
|
def insert_update(rows, key, value):
|
|
if key in rows:
|
|
rows[key] += value
|
|
else:
|
|
rows[key] = value
|
|
|
|
def merge(*dicts):
|
|
merged = {}
|
|
for d in dicts:
|
|
for entry in d:
|
|
insert_update(merged, entry, d[entry])
|
|
return merged
|