summaryrefslogtreecommitdiffstats
path: root/sploit/main.py
blob: ee0495f63d9f139c9f68b3e097952de3d6ba7457 (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
49
50
51
52
53
54
55
56
57
58
59
60
import argparse
import tempfile
import traceback

from sploit.comm import *

def main():
    parser = argparse.ArgumentParser(description='Execute Sploit Script Against Target')
    parser.add_argument('-d', '--daemon', action='store_true',
                        help='run in "daemon" mode with pipes instead of a designated target')
    parser.add_argument('script',
                        help='exploit script to run')
    parser.add_argument('target', nargs=argparse.REMAINDER,
                        help='target program to exploit')
    args = parser.parse_args()

    try:
        if(len(args.target)>0):
            if(args.daemon):
                print("Target Given. Ignoring Daemon Flag...")
            target(args.script, args.target)
        else:
            if(args.daemon):
                daemon(args.script)
            else:
                pipe(args.script)
    except KeyboardInterrupt:
        pass

def daemon(script):
    print("Running in Pipe Daemon Mode...")
    with tempfile.TemporaryDirectory() as tmpdir:
        while(True):
            try:
                p = Pipes(tmpdir)
            except KeyboardInterrupt:
                break
            try:
                runscript(script, Comm(p));
            except KeyboardInterrupt:
                pass
            except:
                traceback.print_exc()
            del p

def pipe(script):
    print("Running in Pipe Mode...");
    runscript(script, Comm(Pipes()));

def target(script, target):
    print("Running in Target Mode...")
    runscript(script, Comm(Process(target)));

def runscript(script, comm):
    print("Running Script...")
    exec(open(script).read())
    print("Script Finished!")
    for line in comm.back.stdin:
        log(line)