Fix issue #325

lifting-bits · May 19, 2023 · eaf23b4 · eaf23b4
1 parent c40d377
commit eaf23b4
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 0 deletions.
diff --git a/lib/AST/IRToASTVisitor.cpp b/lib/AST/IRToASTVisitor.cpp
@@ -1186,6 +1186,7 @@ void IRToASTVisitor::VisitFunctionDecl(llvm::Function &func) {
       fdecl->addDecl(var);
     } else if (inst.hasNUsesOrMore(2) ||
                (inst.hasNUsesOrMore(1) && llvm::isa<llvm::CallInst>(inst)) ||
+               (inst.hasNUsesOrMore(1) && llvm::isa<llvm::LoadInst>(inst)) ||
                llvm::isa<llvm::PHINode>(inst)) {
       if (!inst.getType()->isVoidTy()) {
         auto GetPrefix{[&](llvm::Instruction *inst) {

diff --git a/tests/tools/decomp/issue_325_goto.c b/tests/tools/decomp/issue_325_goto.c
@@ -0,0 +1,17 @@
+#include <stdio.h>
+int main(int argc, const char** argv) {
+  int i = argc;
+  if (i == 1) {  // no args
+    putchar(4 + '0');
+    i += 1;
+    goto n5;
+  }
+  if (i > 1) {  // args
+    putchar(5 + '0');
+    goto n7;
+  }
+n5:
+  putchar(6 + '0');
+n7:
+  putchar(7 + '0');
+}