30 Years of Decompilation and the Unsolved Structuring Problem: Part 1
mahaloz.re
mahaloz.re
(Why? Just for fun, https://github.com/albertz/PyCParser, even more just-for-fun goal was this: https://github.com/albertz/PyCPython).
It literally would translate the C code in equivalent Python code, using ctypes heavily. It was mostly straight-forward, except for mapping goto (thus related to this control flow structuring problem).
Of course, there are some hacks to introduce goto in Python, which in many cases would operate on the Python bytecode, which actually has the JUMP_ABSOLUTE op, but there are also other ways (https://stackoverflow.com/questions/6959360/goto-in-python).
I could also have translated C directly to equivalent Python bytecode and not Python source code, but I really wanted to have Python source code.
My ugly solution worked basically like this: Whenever there was some goto in a function, it would translate it as follows:
First, we flatten any Python AST into a series of statements, with even more goto's added for while/for loops, if's, etc. (We can avoid doing this for all sub-ASTs where there are no no goto-labels. The goal is to have all goto-labels at the top-level, not inside a sub-AST.)
Only conditional goto's can stay. All Gotos and Goto-labels are marked somehow as special elements. So, we end up with sth like:
x()
y()
<goto-label "a">
z()
<goto-stmnt "a">
w()
if v(): <goto-stmnt "a">
q()
Now, we can implement the goto-handling based on this flattened code:- Add a big endless loop around it. After the final statement, a break would leave the loop.
- Before the loop, we add the statement `goto = None`.
- The goto-labels will split the code into multiple part, where we add some `if goto is None:` before each part (excluding the goto-labels).
- For the goto-labels itself, we add this code:
if goto == <goto-label>: goto = None
- For every goto-statement, we add this code: goto = <goto-stmnt>; continue
See here: https://github.com/albertz/PyCParser/blob/master/goto.pyWe even do this in the angr decompiler, found here: https://github.com/angr/angr/blob/8e48d001e18a913ecd4ed2e995...
You may also be interested in the Relooper and Stackifier algorithms, which produce more efficient programs than a simple loop/switch.
There are tools like Remill (https://www.trailofbits.com/opensource/#binary) that disassemble to LLVM IR that can be fed back to the compiler to produce a new binary. You can actually target a different architecture, so it works for binary translation too.
jmp rax
...some binary data...
Where the value of "rax" depends on some input, so the disassembler can never be sure that "some binary data" is actually "data" or "code".
As in, could you abstract away the CFG and feed it into an LLM trained on a bunch of CFGs and corresponding ASTs? Especially given that the LLM could be trained on a lot of code that may very well have been reused in the code being decompiled. Even if not the same code, the same algorithms may be similar enough to improve the structuring output.
There's lots to be researched here, but as the post alludes to, (public) development is limited. My hope is that there will be more work in this space to enable porting of closed-source applications on older architectures to newer architectures; I have a few ideas on how to go about that, but not enough time to look into it.
[0]: https://arxiv.org/abs/2310.06530
[EDIT: Removed a comment about an apparent OCR issue, which has been fixed. Thanks for this great write-up!]