diff options
Diffstat (limited to 'tools')
| -rw-r--r-- | tools/sploit/sploit/main.py | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/tools/sploit/sploit/main.py b/tools/sploit/sploit/main.py index ebcbd41..6d181b8 100644 --- a/tools/sploit/sploit/main.py +++ b/tools/sploit/sploit/main.py @@ -25,10 +25,10 @@ def main(): def daemon(script): print("Running in Pipe Daemon Mode...") - tmpdir = tempfile.TemporaryDirectory() - tmp = os.path.split(tmpdir.name)[1] - while(True): - runscript(script,Comm(Pipes(tmp))); + with tempfile.TemporaryDirectory() as tmpdir: + tmp = os.path.split(tmpdir)[1] + while(True): + runscript(script,Comm(Pipes(tmp))); def pipe(script): print("Running in Pipe Mode..."); |