Update printSections.py + update de executáveis

T2/Parte2/.gitignore

+.idea
\ No newline at end of file

T2/Parte2/printSections.py

+import pefile
+import sys
+import os
+
+files = []
+
+
+# Trata o input do usuário
+def handle_input():
+    try:
+        if os.path.isdir(sys.argv[1]):
+            for file in os.listdir(sys.argv[1]):
+                files.append(os.path.join(sys.argv[1], file))
+        elif os.path.isfile(sys.argv[1]):
+            files.append(sys.argv[1])
+    except IndexError:
+        print("ERROR")
+    except FileNotFoundError:
+        print("ERROR")
+
+
+# Encontra as seções executáveis e imprime na saida padrão
+def print_executable_sections():
+    objeto = {}
+    for file in files:
+        pe = pefile.PE(file)
+        executable = []
+        for section in pe.sections:
+            if section.IMAGE_SCN_MEM_EXECUTE:
+                executable.append(section.Name.decode('utf-8').rstrip('\x00'))
+        objeto[os.path.basename(file)] = executable
+    print(objeto)
+
+
+if __name__ == '__main__':
+    handle_input()
+    print_executable_sections()