blob: 845602970e17e91a9fbd68bcd27bb04d503c07c2 (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
|
from argparse import ArgumentParser, REMAINDER
import gc
import tempfile
import traceback
from sploit.comm import *
def main():
parser = ArgumentParser(description='Execute Sploit script against target')
parser.add_argument('script', help='Exploit script to run')
parser.add_argument('target', nargs=REMAINDER, help='Target program to exploit')
args = parser.parse_args()
if(len(args.target)>0):
target(args.script, args.target)
else:
pipe(args.script)
def pipe(script):
print("Running in Pipe Mode...")
with tempfile.TemporaryDirectory() as tmpdir:
while(True):
try:
p = Pipes(tmpdir)
except KeyboardInterrupt:
break
runscript(script, Comm(p))
del p
def target(script, target):
print("Running in Target Mode...")
runscript(script, Comm(Process(target)))
def runscript(script, comm):
try:
print("Running Script...")
code = compile(open(script).read(), script, 'exec')
exec(code, {'io': comm})
print("Script Finished!")
comm.readall()
return
except KeyboardInterrupt:
pass
except:
traceback.print_exc()
finally:
gc.collect()
print("Script Ended Early!")
|
