swisskyrepo
/
NetExec
mirror of https://github.com/swisskyrepo/NetExec.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Merge remote-tracking branch 'upstream/master'

main
Zak 2022-10-19 14:41:17 +02:00
parent 46e596a233 7f1b2a3ab5
commit 51b3958175
1 changed files with 14 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
cme/cmedb.py
View File

@ -328,7 +328,9 @@ class CMEDBMenu(cmd.Cmd):
pass pass
def do_workspace(self, line): def do_workspace(self, line):
helpString = "[-] wordkspace create <targetName> | workspace list | workspace <targetName>"
if not line: if not line:
print(helpString)
return return
line = line.strip() line = line.strip()
@ -364,12 +366,24 @@ class CMEDBMenu(cmd.Cmd):
self.do_workspace(new_workspace) self.do_workspace(new_workspace)
elif line.split()[0] == 'list':
print("[*] Enumerating Workspaces")
for workspace in os.listdir(os.path.join(self.workspace_dir)):
if(workspace==self.workspace):
print("==> "+workspace)
else:
print(workspace)
elif os.path.exists(os.path.join(self.workspace_dir, line)): elif os.path.exists(os.path.join(self.workspace_dir, line)):
self.config.set('CME', 'workspace', line) self.config.set('CME', 'workspace', line)
self.write_configfile() self.write_configfile()
self.workspace = line self.workspace = line
self.prompt = 'cmedb ({}) > '.format(line) self.prompt = 'cmedb ({}) > '.format(line)
else:
print(helpString)
def do_exit(self, line): def do_exit(self, line):
sys.exit(0) sys.exit(0)