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

Fix for issue #178

1.6
Harmj0y 2016-04-24 10:29:11 -04:00
parent f3142287fe
commit f699ec510d
1 changed files with 10 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
lib/common/agents.py
View File

@ -54,9 +54,12 @@ class Agents:
# get the current and previous URIs for tasking
currentURIs,oldURIs = self.get_agent_uris(agentID)
if not oldURIs: oldURIs = ''
self.agents[agentID]['currentURIs'] = currentURIs
self.agents[agentID]['oldURIs'] = oldURIs
self.agents[agentID]['currentURIs'] = currentURIs.split(',')
if not oldURIs:
self.agents[agentID]['oldURIs'] = []
else:
self.agents[agentID]['oldURIs'] = oldURIs.split(',')
# pull out common configs from the main menu object in empire.py
self.ipWhiteList = self.mainMenu.ipWhiteList
@ -125,8 +128,8 @@ class Agents:
# initialize the tasking/result buffers along with the client session key
sessionKey = self.get_agent_session_key(sessionID)
# TODO: should oldURIs be a string or list?
self.agents[sessionID] = {'sessionKey':sessionKey, 'functions':[], 'currentURIs':requestUris, 'oldURIs': ''}
self.agents[sessionID] = {'sessionKey':sessionKey, 'functions':[], 'currentURIs':requestUris.split(','), 'oldURIs': []}
# report the initial checkin in the reporting database
cur = self.conn.cursor()
@ -642,8 +645,8 @@ class Agents:
print helpers.color("[!] Agent " + agentName + " not active.")
else:
# update the URIs in the cache
self.agents[sessionID]['oldURIs'] = oldURIs
self.agents[sessionID]['currentURIs'] = parts[0]
self.agents[sessionID]['oldURIs'] = oldURIs.split(',')
self.agents[sessionID]['currentURIs'] = parts[0].split(',')
# if no additional headers
if len(parts) == 2: