From 83661f21aa0e4c231f227fe4bce5e1b9cd176699 Mon Sep 17 00:00:00 2001
From: Jur van den Berg <Jurl.berg@gmail.com>
Date: Wed, 17 Apr 2019 23:40:29 +0200
Subject: [PATCH] Fix disassembly wrong name for main

---
 src/disassembler.rs | 12 ++++++++----
 1 file changed, 8 insertions(+), 4 deletions(-)

diff --git a/src/disassembler.rs b/src/disassembler.rs
index 4a62730..f5cf1ce 100644
--- a/src/disassembler.rs
+++ b/src/disassembler.rs
@@ -353,10 +353,14 @@ impl Disassembler {
     }
 
     fn disasm_method(&mut self, i: i32, pc: usize, args: usize, vars: usize) -> Result<Method> {
-        let name = self
-            .symbols
-            .lookup_method(pc)
-            .unwrap_or_else(|| format!("func_{}", i));
+
+        let name = if pc == 0 {
+            "main".to_string()
+        } else {
+            self.symbols
+                .lookup_method(pc)
+                .unwrap_or_else(|| format!("func_{}", i))
+        };
         let mut method = Method::new(name, pc, args, vars);
 
         while self.text.has_i8() && !self.is_method(self.text.cur()) {