def analyze_binary_thr(binary_id):
db = DB()
binary = db.get_binary_by_id(binary_id)
models = db.get_available_models()
job = db.add(Job("Analyzing binary " + binary.name))
bin_path = os.path.join(FILE_UPLOAD_FOLDER, binary.filename + ".bin")
diss_path = os.path.join(DISS_BINARY_FOLDER, binary.filename + ".json")
hash = get_file_sha_hash(bin_path)
db.binary_add_hash(binary.id, hash)
Disassembler(bin_path).disassemble_to_file(diss_path)
for i, m in enumerate(models):
db.job_add_log(
job.id,
"Testing against model {}, {}/{}".format(m.name, len(models),
i + 1))
model_path = os.path.join(MODEL_FOLDER, m.filename)
map_path = os.path.join(model_path, "map.json")
model_file_path = os.path.join(model_path, "model.npz")
rating = ModelComparator(diss_path, map_path,
model_file_path).get_rating()
db.add(Result(binary.id, m.id, rating))
db.job_add_log(job.id, "Cleaning up the file mess")
db.set_binary_analyzed(binary.id)
os.system("rm {}".format(bin_path))
db.set_job_ended(job.id)
def create_model_thr(model_id):
db = DB()
model = db.get_model_by_id(model_id)
job = db.add(Job("Generating model " + model.name))
bin_path = os.path.join(FILE_UPLOAD_FOLDER, model.filename)
model_path = os.path.join(MODEL_FOLDER, model.filename)
diss_path = os.path.join(model_path, "diss")
map_path = os.path.join(model_path, "map.json")
model_file_path = os.path.join(model_path, "model")
os.system("mkdir {}".format(bin_path))
os.system("mkdir {}".format(model_path))
os.system("mkdir {}".format(diss_path))
os.system("unzip {} -d {} > /dev/null".format(bin_path + ".zip", bin_path))
bin_files = os.listdir(bin_path)
for i, binf in enumerate(bin_files):
db.job_add_log(
job.id, "Disassembling file {}/{}".format(len(bin_files), i + 1))
Disassembler(os.path.join(bin_path, binf)).disassemble_to_file(
os.path.join(diss_path,
str(i) + "-" + binf + ".json"))
db.job_add_log(job.id, "Creating map")
MapCreator(diss_path, map_path).create_map()
db.job_add_log(job.id, "Generating model")
ModelCreator(diss_path, map_path, model_file_path).create_model()
db.job_add_log(job.id, "Model generated")
db.set_model_analyzed(model.id)
db.job_add_log(job.id, "Cleaning up the file mess")
os.system("rm {}".format(bin_path + ".zip"))
os.system("rm -rf {}".format(bin_path))
db.set_job_ended(job.id)
def create_binary():
try:
db = DB()
data = request.get_json(silent=True) # name, desc, benign
binary = db.add(Binary(data["name"]))
if binary:
return jsonify({"binary_id": binary.id}), 200
else:
return jsonify({"error": "binary entry couldn't be created"}), 400
except Exception as e:
print(str(e))
abort(500)
def create_model():
try:
db = DB()
data = request.get_json(silent=True) # name, desc, benign
model = db.add(Model(data["name"], data["desc"], data["benign"]))
if model:
return jsonify({"model_id": model.id}), 200
else:
return jsonify({"error": "model couldn't be created"}), 400
except Exception as e:
print(e)
abort(500)
def new_binary(name, file):
db = DB()
binary = db.add(Binary(name))
cpdest = os.path.join(FILE_UPLOAD_FOLDER, binary.filename + ".bin")
os.system("cp {} {}".format(file, cpdest))
hm.analyze_binary_thr(binary.id)
def new_model(name, desc, benign, zipfile):
db = DB()
model = db.add(Model(name, desc, benign))
cpdest = os.path.join(FILE_UPLOAD_FOLDER, model.filename + ".zip")
os.system("cp {} {}".format(zipfile, cpdest))
hm.create_model_thr(model.id)