diff --git a/android_class_shadowing_scanner/__init__.py b/android_class_shadowing_scanner/__init__.py index df2df0d..9e45989 100644 --- a/android_class_shadowing_scanner/__init__.py +++ b/android_class_shadowing_scanner/__init__.py @@ -311,11 +311,11 @@ def check_smali(): with (d / "app.apk").open("wb") as fp: fp.write(apk_bin) with zipfile.ZipFile(io.BytesIO(apk_bin)) as apk: - json_data[sha256] = {} - entry = analyze(apk, sha256, json_out=json_data[sha256]) + data[sha256] = {} + entry = analyze(apk, sha256, json_out=data[sha256]) subprocess.run(["apktool", "d", "app.apk", "-o", "apktool_out"], cwd=d) smalli_dirs = [] - for dex in json_data[sha256]["class_dex"]: + for dex in data[sha256]["class_dex"]: if dex == "classes.dex": smalli_dirs.append(out / "apktool_out" / "smali") else: @@ -323,7 +323,7 @@ def check_smali(): out / "apktool_out" / "smali_" + dex.removesuffix(".dex") ) dist_dup_classes = set() - for cl in json_data[sha256]["duplicated_classes"]: + for cl in data[sha256]["duplicated_classes"]: cl_f = cl.removesuffix(";").removeprefix("L") + ".smali" smali = None for cdir in smalli_dirs: @@ -334,5 +334,5 @@ def check_smali(): smali = smali_new elif smali != smali_new: dist_dup_classes.add(cl) - json_data[sha256]["redef_classes"] = list(dist_dup_classes) - print(json.dumps(json_data)) + data[sha256]["redef_classes"] = list(dist_dup_classes) + print(json.dumps(data))