diff options
| author | Malfurious <m@lfurio.us> | 2024-01-13 15:03:30 -0500 |
|---|---|---|
| committer | Malfurious <m@lfurio.us> | 2024-05-19 17:52:10 -0400 |
| commit | debc0a42457619309039579cd1615588d5d221e0 (patch) | |
| tree | 78ae31a1e63a0dabf114942ce14ca0d287b7084c | |
| parent | c71df559f277167501d25057519f627481088c99 (diff) | |
| download | sploit-debc0a42457619309039579cd1615588d5d221e0.tar.gz sploit-debc0a42457619309039579cd1615588d5d221e0.zip | |
main: Automatically provide top-level sploit modules to user scripts
Signed-off-by: Malfurious <m@lfurio.us>
Diffstat (limited to '')
| -rw-r--r-- | sploit/__main__.py | 10 |
1 files changed, 9 insertions, 1 deletions
diff --git a/sploit/__main__.py b/sploit/__main__.py index 5b694a2..5d53ca6 100644 --- a/sploit/__main__.py +++ b/sploit/__main__.py @@ -46,11 +46,19 @@ def target(script, target): print_banner(STATUS, line3='Subprocess Mode') runscript(script, Comm(Process(target))) +def user_scope(comm): + import sploit as lib + scope = { name: getattr(lib, name) for name in dir(lib) } + scope['__version__'] = __version__ + scope['print'] = elog + scope['io'] = comm + return scope + def runscript(script, comm): try: ilog("Running Script...") code = compile(open(script).read(), script, 'exec') - exec(code, {'io': comm, 'print': elog}) + exec(code, user_scope(comm)) ilog("Script Finished!") return except KeyboardInterrupt: |