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

Truncate shellcode when logging it to the DB

chunking
m0rv4i 2019-02-11 22:25:03 +00:00
parent ca9d6e7f0a
commit df908b3f06
1 changed files with 12 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
Tasks.py
View File

@ -19,24 +19,25 @@ def newTask(path):
user_command = command user_command = command
hostinfo = DB.get_hostinfo(RandomURI) hostinfo = DB.get_hostinfo(RandomURI)
now = datetime.datetime.now() now = datetime.datetime.now()
if (command.lower().startswith("$shellcode64")) or (command.lower().startswith("$shellcode64")) :
user_command = command[0:150]+"......TRUNCATED......"+command[-80:]
elif (command.lower().startswith("run-exe core.program core inject-shellcode")) :
user_command = command[0:150]+"......TRUNCATED......"+command[-80:]
elif (command.lower().startswith("$shellcode86")) or (command.lower().startswith("$shellcode86")) :
user_command = command[0:150]+"......TRUNCATED......"+command[-80:]
taskId = DB.insert_task(RandomURI, user_command, user) taskId = DB.insert_task(RandomURI, user_command, user)
taskIdStr = "0" * (5 - len(str(taskId))) + str(taskId) taskIdStr = "0" * (5 - len(str(taskId))) + str(taskId)
print Colours.YELLOW,"" if len(str(taskId)) > 5:
print "Task %s issued against implant %s on host %s %s (%s)" % (taskIdStr, hostinfo[0],hostinfo[3],hostinfo[11],now.strftime("%m/%d/%Y %H:%M:%S")) raise ValueError('Task ID is greater than 5 characters which is not supported.')
if (command.lower().startswith("$shellcode64")) or (command.lower().startswith("$shellcode64")) : print Colours.YELLOW
print "Loading Shellcode",Colours.END print "Task %s issued against implant %s on host %s\\%s @ %s (%s)" % (taskIdStr, hostinfo[0],hostinfo[11],hostinfo[2],hostinfo[3],now.strftime("%m/%d/%Y %H:%M:%S"))
elif (command.lower().startswith("run-exe core.program core inject-shellcode")) : if "upload-file" in command.lower():
print command[0:150]+"......TRUNCATED......"+command[-80:],Colours.END
elif (command.lower().startswith("$shellcode86")) or (command.lower().startswith("$shellcode86")) :
print "Loading Shellcode",Colours.END
elif "upload-file" in command.lower():
print "Uploading File",Colours.END print "Uploading File",Colours.END
else: else:
try: try:
print command,Colours.END print user_command,Colours.END
except Exception as e: except Exception as e:
print "Cannot print output: %s" % e print "Cannot print output: %s" % e
if a[2].startswith("loadmodule"): if a[2].startswith("loadmodule"):
try: try:
module_name = (a[2]).replace("loadmodule ","") module_name = (a[2]).replace("loadmodule ","")
@ -50,8 +51,6 @@ def newTask(path):
except Exception as e: except Exception as e:
print "Cannot find module, loadmodule is case sensitive!" print "Cannot find module, loadmodule is case sensitive!"
print e print e
if len(str(taskId)) > 5:
raise ValueError('Task ID is greater than 5 characters which is not supported.')
command = taskIdStr + command command = taskIdStr + command
if commands: if commands:
commands += "!d-3dion@LD!-d" + command commands += "!d-3dion@LD!-d" + command