Merge pull request #7 from JohnRyk/patch-1

update revshellgen.py
t0thkr1s · Aug 14, 2021 · 227137c · 227137c
2 parents cc06f57 + c763711
commit 227137c
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/revshellgen.py b/revshellgen.py
@@ -2,6 +2,7 @@
 # coding=utf-8
 import ipaddress
 import os
+import sys
 import urllib.parse
 from string import Template
 from typing import List
@@ -34,7 +35,7 @@
 
 choices = ['no', 'yes']
 shells = ['/bin/sh', '/bin/bash', '/bin/zsh', '/bin/ksh', '/bin/tcsh', '/bin/dash']
-commands = sorted([command for command in os.listdir('commands')])
+commands = sorted([command for command in os.listdir(sys.path[0] + '/commands')])
 
 
 def print_banner():