Modify script for bat-ana and bat-dis

Dockerfile

@@ -29,6 +29,4 @@ WORKDIR $HOME/app
 # Copy the current directory contents into the container at $HOME/app setting the owner to the user
 COPY --chown=user . $HOME/app
 
-RUN wget https://huggingface.co/datasets/ejschwartz/oo-method-test/raw/main/scripts/gen-training.py -O gen_training.py
-
 CMD ["python3", "app.py"]

app.py

@@ -15,8 +15,47 @@ model = gr.load("ejschwartz/oo-method-test-model-bylibrary", src="models")
 
 import gen_training
 
+from tempfile import NamedTemporaryFile
+anafile = NamedTemporaryFile(prefix=os.path.basename(bname) + "_", suffix=".bat_ana")
+ananame = anafile.name
+
+def get_all_dis(addrs=None):
+
+    addrstr = ""
+    if addrs is not None:
+        addrstr = " ".join([f"--function-at {x}" for x in addrs])
+
+    subprocess.check_output(f"bat-ana {addrstr} --no-post-analysis -o {ananame} {bname} 2>/dev/null", shell=True)
+
+
+    output = subprocess.check_output(f"bat-dis --no-insn-address --no-bb-cfg-arrows --color=off {ananame} 2>/dev/null", shell=True)
+    output = re.sub(b' +', b' ', output)
+
+    func_dis = {}
+    last_func = None
+    current_output = []
+
+    for l in output.splitlines():
+        if l.startswith(b";;; function 0x"):
+            if last_func is not None:
+                func_dis[last_func] = b"\n".join(current_output)
+            last_func = int(l.split()[2], 16)
+            current_output.clear()
+
+        if not b";;" in l:
+            current_output.append(l)
+
+    if last_func is not None:
+        if last_func in func_dis:
+            print("Warning: Ignoring multiple functions at the same address")
+        else:
+            func_dis[last_func] = b"\n".join(current_output)
+
+    return func_dis
+
 def get_funs(f):
-    return "cool"
+    funs = get_all_dis()
+    returns str(funs)
 
 demo = gr.Interface(fn=get_funs, inputs="file", outputs="textarea")
 demo.launch(server_name="0.0.0.0", server_port=7860)