simplify code

Author: tesuji

pwnlib/elf/elf.py

@@ -1164,14 +1164,11 @@ def libc_start_main_return(self):
         to list all calls inside __libc_start_main, find the call to exit
         after the call to main and select the previous call.
         """
-        if '__libc_start_main' not in self.functions:
+        func = self.functions.get('__libc_start_main')
+        exit_addr = self.symbols.get('exit')
+        if not (func and exit_addr):
             return 0
 
-        if 'exit' not in self.symbols:
-            return 0
-
-        func = self.functions['__libc_start_main']
-        exit_addr = self.symbols['exit']
         # `__libc_start_call_main` is usually smaller than `__libc_start_main`,
         # (except for powerpc which uses a bigger `generic_start_main`), so
         # we might disassemble a bit too much, but it's a good dynamic estimate.
@@ -1184,9 +1181,8 @@ def libc_start_main_return(self):
         call_return_offset = 1
         call_instructions = set([cs.CS_GRP_CALL])
         if self.arch in ['arm', 'thumb']:
-            if b'armhf' in self.linker:
-                # FIXME: I have no idea why setting self.arch = 'armhf' does not work
-                eabi = 'hf'
+            # FIXME: I have no idea why setting self.arch = 'armhf' does not work
+            if b'armhf' in self.linker: eabi = 'hf'
             if exit_addr & 1: exit_addr -= 1
         elif self.arch == 'aarch64':
             pass
@@ -1197,9 +1193,8 @@ def libc_start_main_return(self):
             pass
         elif self.arch in ['ppc', 'powerpc', 'powerpc64']:
             callee_size *= 2
-            if exit_addr & 1 == 0:
-                # powepc often jumps to the local entry point after TOC setup
-                exit_addr += 8
+            # powepc often jumps to the local entry point after TOC setup
+            if exit_addr & 1 == 0: exit_addr += 8
             pass
         elif self.arch in ['em_s390', 's390']:
             imm_index = 1