Merge pull request #4 from giladreich/autopep8

Apply autopep8 rules.

.pep8

@@ -0,0 +1,6 @@
+[pycodestyle]
+max_line_length = 120
+aggressive = 0
+recursive = true
+indent-size = 2
+

projects/00_angr_find/generate.py

@@ -1,6 +1,10 @@
-import sys, random, os, tempfile
+import sys
+import random
+import os
+import tempfile
 from templite import Templite
 
+
 def generate(argv):
   if len(argv) != 3:
     print('Usage: ./generate.py [seed] [output_file]')
@@ -23,5 +27,6 @@ def generate(argv):
     temp.seek(0)
     os.system('gcc -m32 -o ' + output_file + ' ' + temp.name)
 
+
 if __name__ == '__main__':
   generate(sys.argv)

projects/00_angr_find/solver.py

@@ -25,6 +25,7 @@
 import angr
 import sys
 
+
 def main(argv):
   # Create an Angr project.
   # If you want to be able to point to the binary from the command line, you can
@@ -45,8 +46,8 @@ def main(argv):
   simulation = project.factory.simgr(initial_state)
 
   # Explore the binary to attempt to find the address that prints "Good Job."
-  # You will have to find the address you want to find and insert it here. 
-  # This function will keep executing until it either finds a solution or it 
+  # You will have to find the address you want to find and insert it here.
+  # This function will keep executing until it either finds a solution or it
   # has explored every possible path through the executable.
   # (!)
   print_good_address = ???  # :integer (probably in hexadecimal)
@@ -61,13 +62,14 @@ def main(argv):
     # target address.
     solution_state = simulation.found[0]
 
-    # Print the string that Angr wrote to stdin to follow solution_state. This 
+    # Print the string that Angr wrote to stdin to follow solution_state. This
     # is our solution.
     print(solution_state.posix.dumps(sys.stdin.fileno()).decode())
   else:
     # If Angr could not find a path that reaches print_good_address, throw an
     # error. Perhaps you mistyped the print_good_address?
     raise Exception('Could not find the solution')
 
+
 if __name__ == '__main__':
   main(sys.argv)

projects/01_angr_avoid/generate.py

@@ -1,6 +1,10 @@
-import sys, random, os, tempfile
+import sys
+import random
+import os
+import tempfile
 from templite import Templite
 
+
 def generate(argv):
   if len(argv) != 3:
     print('Usage: ./generate.py [seed] [output_file]')
@@ -25,5 +29,6 @@ def generate(argv):
     temp.seek(0)
     os.system('gcc -m32 -o ' + output_file + ' ' + temp.name)
 
+
 if __name__ == '__main__':
   generate(sys.argv)

projects/01_angr_avoid/solver.py

@@ -1,16 +1,17 @@
 import angr
 import sys
 
+
 def main(argv):
   path_to_binary = ???
   project = angr.Project(path_to_binary)
   initial_state = project.factory.entry_state()
   simulation = project.factory.simgr(initial_state)
 
   # Explore the binary, but this time, instead of only looking for a state that
-  # reaches the print_good_address, also find a state that does not reach 
+  # reaches the print_good_address, also find a state that does not reach
   # will_not_succeed_address. The binary is pretty large, to save you some time,
-  # everything you will need to look at is near the beginning of the address 
+  # everything you will need to look at is near the beginning of the address
   # space.
   # (!)
   print_good_address = ???
@@ -23,5 +24,6 @@ def main(argv):
   else:
     raise Exception('Could not find the solution')
 
+
 if __name__ == '__main__':
   main(sys.argv)

projects/02_angr_find_condition/generate.py

@@ -1,6 +1,10 @@
-import sys, random, os, tempfile
+import sys
+import random
+import os
+import tempfile
 from templite import Templite
 
+
 def generate(argv):
   if len(argv) != 3:
     print('Usage: ./generate.py [seed] [output_file]')
@@ -23,5 +27,6 @@ def generate(argv):
     temp.seek(0)
     os.system('gcc -m32 -o ' + output_file + ' ' + temp.name)
 
+
 if __name__ == '__main__':
   generate(sys.argv)

projects/02_angr_find_condition/solver.py

@@ -13,6 +13,7 @@
 import angr
 import sys
 
+
 def main(argv):
   path_to_binary = argv[1]
   project = angr.Project(path_to_binary)
@@ -47,5 +48,6 @@ def should_abort(state):
   else:
     raise Exception('Could not find the solution')
 
+
 if __name__ == '__main__':
   main(sys.argv)

projects/03_angr_symbolic_registers/generate.py

@@ -1,6 +1,10 @@
-import sys, random, os, tempfile
+import sys
+import random
+import os
+import tempfile
 from templite import Templite
 
+
 def generate(argv):
   if len(argv) != 3:
     print('Usage: ./generate.py [seed] [output_file]')
@@ -23,5 +27,6 @@ def generate(argv):
     temp.seek(0)
     os.system('gcc -m32 -o ' + output_file + ' ' + temp.name + ' 2>/dev/null')
 
+
 if __name__ == '__main__':
   generate(sys.argv)

projects/03_angr_symbolic_registers/solver.py

@@ -1,11 +1,12 @@
-# Angr doesn't currently support reading multiple things with scanf (Ex: 
+# Angr doesn't currently support reading multiple things with scanf (Ex:
 # scanf("%u %u).) You will have to tell the simulation engine to begin the
 # program after scanf is called and manually inject the symbols into registers.
 
 import angr
 import claripy
 import sys
 
+
 def main(argv):
   path_to_binary = argv[1]
   project = angr.Project(path_to_binary)
@@ -19,7 +20,7 @@ def main(argv):
 
   # Create a symbolic bitvector (the datatype Angr uses to inject symbolic
   # values into the binary.) The first parameter is just a name Angr uses
-  # to reference it. 
+  # to reference it.
   # You will have to construct multiple bitvectors. Copy the two lines below
   # and change the variable names. To figure out how many (and of what size)
   # you need, dissassemble the binary and determine the format parameter passed
@@ -74,5 +75,6 @@ def should_abort(state):
   else:
     raise Exception('Could not find the solution')
 
+
 if __name__ == '__main__':
   main(sys.argv)

projects/04_angr_symbolic_stack/generate.py

@@ -1,6 +1,10 @@
-import sys, random, os, tempfile
+import sys
+import random
+import os
+import tempfile
 from templite import Templite
 
+
 def generate(argv):
   if len(argv) != 3:
     print('Usage: ./generate.py [seed] [output_file]')
@@ -23,5 +27,6 @@ def generate(argv):
     temp.seek(0)
     os.system('gcc -fno-stack-protector -m32 -o ' + output_file + ' ' + temp.name)
 
+
 if __name__ == '__main__':
   generate(sys.argv)

projects/04_angr_symbolic_stack/solver.py

@@ -11,6 +11,7 @@
 import claripy
 import sys
 
+
 def main(argv):
   path_to_binary = argv[1]
   project = angr.Project(path_to_binary)
@@ -86,7 +87,7 @@ def main(argv):
   #   push   %eax
   #   push   $0x80489c3
   #   call   8048370 <__isoc99_scanf@plt>
-  #   add    $0x10,%esp 
+  #   add    $0x10,%esp
   # As you can see, the call to scanf looks like this:
   # scanf(  0x80489c3,   ebp - 0xc,   ebp - 0x10  )
   #      format_string    password0    password1
@@ -155,5 +156,6 @@ def should_abort(state):
   else:
     raise Exception('Could not find the solution')
 
+
 if __name__ == '__main__':
   main(sys.argv)

projects/05_angr_symbolic_memory/generate.py

@@ -1,6 +1,11 @@
-import sys, random, os, tempfile, string
+import sys
+import random
+import os
+import tempfile
+import string
 from templite import Templite
 
+
 def generate(argv):
   if len(argv) != 3:
     print('Usage: ./generate.py [seed] [output_file]')
@@ -26,5 +31,6 @@ def generate(argv):
     temp.seek(0)
     os.system('gcc -m32 -o ' + output_file + ' ' + temp.name)
 
+
 if __name__ == '__main__':
   generate(sys.argv)

projects/05_angr_symbolic_memory/solver.py

@@ -2,6 +2,7 @@
 import claripy
 import sys
 
+
 def main(argv):
   path_to_binary = argv[1]
   project = angr.Project(path_to_binary)
@@ -42,13 +43,14 @@ def should_abort(state):
     # Therefore, we will use eval, with named parameter cast_to=str
     # which returns a string instead of an integer.
     # (!)
-    solution0 = solution_state.solver.eval(password0,cast_to=bytes).decode()
+    solution0 = solution_state.solver.eval(password0, cast_to=bytes).decode()
     ...
     solution = ???
 
     print(solution)
   else:
     raise Exception('Could not find the solution')
 
+
 if __name__ == '__main__':
   main(sys.argv)

projects/06_angr_symbolic_dynamic_memory/generate.py

@@ -1,6 +1,11 @@
-import sys, random, os, tempfile, string
+import sys
+import random
+import os
+import tempfile
+import string
 from templite import Templite
 
+
 def generate(argv):
   if len(argv) != 3:
     print('Usage: ./generate.py [seed] [output_file]')
@@ -25,5 +30,6 @@ def generate(argv):
     temp.seek(0)
     os.system('gcc -m32 -o ' + output_file + ' ' + temp.name)
 
+
 if __name__ == '__main__':
   generate(sys.argv)

projects/06_angr_symbolic_dynamic_memory/solver.py

@@ -2,6 +2,7 @@
 import claripy
 import sys
 
+
 def main(argv):
   path_to_binary = argv[1]
   project = angr.Project(path_to_binary)
@@ -50,13 +51,14 @@ def should_abort(state):
   if simulation.found:
     solution_state = simulation.found[0]
 
-    solution0 = solution_state.se.eval(password0,cast_to=str)
+    solution0 = solution_state.se.eval(password0, cast_to=str)
     ...
     solution = ???
 
     print(solution)
   else:
     raise Exception('Could not find the solution')
 
+
 if __name__ == '__main__':
   main(sys.argv)

projects/07_angr_symbolic_file/generate.py

@@ -1,6 +1,10 @@
-import sys, random, os, tempfile
+import sys
+import random
+import os
+import tempfile
 from templite import Templite
 
+
 def generate(argv):
   if len(argv) != 3:
     print('Usage: ./generate.py [seed] [output_file]')
@@ -25,5 +29,6 @@ def generate(argv):
     temp.seek(0)
     os.system('gcc -m32 -o ' + output_file + ' ' + temp.name)
 
+
 if __name__ == '__main__':
   generate(sys.argv)