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
49
|
from argparse import ArgumentParser, REMAINDER
import gc
import tempfile
import traceback
from sploit.comm import *
from sploit.log 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):
ilog("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):
ilog("Running in Target Mode...")
runscript(script, Comm(Process(target)))
def runscript(script, comm):
try:
ilog("Running Script...")
code = compile(open(script).read(), script, 'exec')
exec(code, {'io': comm, 'print': elog})
ilog("Script Finished!")
comm.readall()
return
except KeyboardInterrupt:
pass
except:
ilog(traceback.format_exc(), end='', color=ERROR)
finally:
gc.collect()
ilog("Script Ended Early!", color=WARNING)
|
