Merge pull request #652 from shoxxdj/master

Improve CMEDB HELP after loosing too much time with workspace ..
main
mpgn 2022-10-19 09:30:48 +02:00 committed by GitHub
commit 7f1b2a3ab5
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 14 additions and 0 deletions

View File

@ -328,7 +328,9 @@ class CMEDBMenu(cmd.Cmd):
pass
def do_workspace(self, line):
helpString = "[-] wordkspace create <targetName> | workspace list | workspace <targetName>"
if not line:
print(helpString)
return
line = line.strip()
@ -364,12 +366,24 @@ class CMEDBMenu(cmd.Cmd):
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)):
self.config.set('CME', 'workspace', line)
self.write_configfile()
self.workspace = line
self.prompt = 'cmedb ({}) > '.format(line)
else:
print(helpString)
def do_exit(self, line):
sys.exit(0)