C3-Client: refactored collectChannels
This commit is contained in:
parent
fd3618a765
commit
ded4b69aa3
|
@ -589,7 +589,7 @@ def onMattermostPurge(args):
|
||||||
'name' : 'ChannelCommandGroup'
|
'name' : 'ChannelCommandGroup'
|
||||||
}
|
}
|
||||||
|
|
||||||
channels = collectChannelsToSendCommand(args, 'mattermost')
|
channels = collectChannels(args, 'mattermost')
|
||||||
|
|
||||||
if len(channels) == 0:
|
if len(channels) == 0:
|
||||||
print('[-] No channels could be found to receive Mattermost purge command.')
|
print('[-] No channels could be found to receive Mattermost purge command.')
|
||||||
|
@ -675,9 +675,6 @@ def onJitter(args):
|
||||||
|
|
||||||
if ret.status_code == 201:
|
if ret.status_code == 201:
|
||||||
print(f'[+] Channel {channel["name"]} (id: {channel["iid"]}) running on {channel["kind"]} {channel["agent"]["name"]} (id: {channel["agent"]["agentId"]}) got its Jitter updated to {args.min_jitter}...{args.max_jitter}\n')
|
print(f'[+] Channel {channel["name"]} (id: {channel["iid"]}) running on {channel["kind"]} {channel["agent"]["name"]} (id: {channel["agent"]["agentId"]}) got its Jitter updated to {args.min_jitter}...{args.max_jitter}\n')
|
||||||
#else:
|
|
||||||
# print(f'[-] Could not update channel\'s {channel["name"]} (id: {channel["iid"]}) running on {channel["kind"]} {channel["agent"]["name"]} (id: {channel["agent"]["agentId"]}) Jitter! Result: {ret.status_code} ({ret.text})\n')
|
|
||||||
|
|
||||||
|
|
||||||
def onLDAPClear(args):
|
def onLDAPClear(args):
|
||||||
data = {
|
data = {
|
||||||
|
@ -690,7 +687,7 @@ def onLDAPClear(args):
|
||||||
'name' : 'ChannelCommandGroup'
|
'name' : 'ChannelCommandGroup'
|
||||||
}
|
}
|
||||||
|
|
||||||
channels = collectChannelsToSendCommand(args, 'ldap')
|
channels = collectChannels(args, 'ldap')
|
||||||
|
|
||||||
if len(channels) == 0:
|
if len(channels) == 0:
|
||||||
print('[-] No channels could be found to receive LDAP clear attribute command.')
|
print('[-] No channels could be found to receive LDAP clear attribute command.')
|
||||||
|
@ -716,7 +713,7 @@ def onMSSQLClearTable(args):
|
||||||
'name' : 'ChannelCommandGroup'
|
'name' : 'ChannelCommandGroup'
|
||||||
}
|
}
|
||||||
|
|
||||||
channels = collectChannelsToSendCommand(args, 'table name')
|
channels = collectChannels(args, 'mssql')
|
||||||
|
|
||||||
if len(channels) == 0:
|
if len(channels) == 0:
|
||||||
print('[-] No channels could be found to receive MSSQL clear DB table command.')
|
print('[-] No channels could be found to receive MSSQL clear DB table command.')
|
||||||
|
@ -742,7 +739,7 @@ def onUncShareFileClear(args):
|
||||||
'name' : 'ChannelCommandGroup'
|
'name' : 'ChannelCommandGroup'
|
||||||
}
|
}
|
||||||
|
|
||||||
channels = collectChannelsToSendCommand(args, 'filesystem path')
|
channels = collectChannels(args, 'uncsharefile')
|
||||||
|
|
||||||
if len(channels) == 0:
|
if len(channels) == 0:
|
||||||
print('[-] No channels could be found to receive UncShareFile remove all message files command.')
|
print('[-] No channels could be found to receive UncShareFile remove all message files command.')
|
||||||
|
@ -768,7 +765,7 @@ def onDropboxClear(args):
|
||||||
'name' : 'ChannelCommandGroup'
|
'name' : 'ChannelCommandGroup'
|
||||||
}
|
}
|
||||||
|
|
||||||
channels = collectChannelsToSendCommand(args, 'dropbox token')
|
channels = collectChannels(args, 'dropbox')
|
||||||
|
|
||||||
if len(channels) == 0:
|
if len(channels) == 0:
|
||||||
print('[-] No channels could be found to receive Dropbox remove all message files command.')
|
print('[-] No channels could be found to receive Dropbox remove all message files command.')
|
||||||
|
@ -794,7 +791,7 @@ def onGithubClear(args):
|
||||||
'name' : 'ChannelCommandGroup'
|
'name' : 'ChannelCommandGroup'
|
||||||
}
|
}
|
||||||
|
|
||||||
channels = collectChannelsToSendCommand(args, 'github token')
|
channels = collectChannels(args, 'github')
|
||||||
|
|
||||||
if len(channels) == 0:
|
if len(channels) == 0:
|
||||||
print('[-] No channels could be found to receive Github remove all message files command.')
|
print('[-] No channels could be found to receive Github remove all message files command.')
|
||||||
|
@ -820,7 +817,7 @@ def onGoogleDriveClear(args):
|
||||||
'name' : 'ChannelCommandGroup'
|
'name' : 'ChannelCommandGroup'
|
||||||
}
|
}
|
||||||
|
|
||||||
channels = collectChannelsToSendCommand(args, 'github token')
|
channels = collectChannels(args, 'googledrive')
|
||||||
|
|
||||||
if len(channels) == 0:
|
if len(channels) == 0:
|
||||||
print('[-] No channels could be found to receive GoogleDrive remove all message files command.')
|
print('[-] No channels could be found to receive GoogleDrive remove all message files command.')
|
||||||
|
@ -835,88 +832,77 @@ def onGoogleDriveClear(args):
|
||||||
else:
|
else:
|
||||||
print(f'[+] Cleared GoogleDrive message files on C3 channel {channel["channelId"]} on gateway {channel["gateway"]["name"]}')
|
print(f'[+] Cleared GoogleDrive message files on C3 channel {channel["channelId"]} on gateway {channel["gateway"]["name"]}')
|
||||||
|
|
||||||
def collectChannelsToSendCommand(args, channelKeyword):
|
def getDeviceName(gateway, devicesType, deviceType):
|
||||||
relays = collectRelays(args)
|
capability = processCapability(gateway)
|
||||||
gateways = getRequest('/api/gateway')
|
name = list(capability[devicesType].keys())[list(capability[devicesType].values()).index(deviceType)]
|
||||||
|
|
||||||
channel_id = None
|
return name
|
||||||
if hasattr(args, 'channel_id') and args.channel_id != None:
|
|
||||||
|
def collectChannels(args, channelName):
|
||||||
|
channels = []
|
||||||
|
gateways = getRequest('/api/gateway')
|
||||||
|
gateway_id = ''
|
||||||
|
relay_id = ''
|
||||||
|
channel_id = ''
|
||||||
|
|
||||||
|
if hasattr(args, 'gateway_id'):
|
||||||
|
gateway_id = args.gateway_id
|
||||||
|
|
||||||
|
if hasattr(args, 'relay_id'):
|
||||||
|
relay_id = args.relay_id
|
||||||
|
|
||||||
|
if hasattr(args, 'channel_id'):
|
||||||
channel_id = args.channel_id
|
channel_id = args.channel_id
|
||||||
|
|
||||||
channels = []
|
|
||||||
|
|
||||||
for gateway, relay in relays:
|
|
||||||
if 'channels' in relay.keys():
|
|
||||||
channel_num = 0
|
|
||||||
|
|
||||||
for c in relay['channels']:
|
|
||||||
channel_num += 1
|
|
||||||
Logger.dbg(f'Iterating over channel {c["iid"]} on Relay ...')
|
|
||||||
if channel_id != None:
|
|
||||||
if c['iid'] == channel_id:
|
|
||||||
Logger.dbg(f'Adding channel {c["iid"]} in Relay {relay["name"]}.')
|
|
||||||
channels.append({
|
|
||||||
'url' : f'/api/gateway/{gateway["agentId"]}/relay/{relay["agentId"]}/channel/{c["iid"]}/command',
|
|
||||||
'gateway' : gateway,
|
|
||||||
'relay' : relay,
|
|
||||||
'channelId' : c['iid'],
|
|
||||||
})
|
|
||||||
continue
|
|
||||||
else:
|
|
||||||
for arg in c['propertiesText']['arguments']:
|
|
||||||
if type(arg) == dict:
|
|
||||||
if channelKeyword in arg['name'].lower() or ("description" in arg.keys() and channelKeyword in arg['description'].lower()):
|
|
||||||
Logger.dbg(f'Adding channel {c["iid"]} in Relay {relay["name"]}.')
|
|
||||||
channels.append({
|
|
||||||
'url' : f'/api/gateway/{gateway["agentId"]}/relay/{relay["agentId"]}/channel/{c["iid"]}/command',
|
|
||||||
'gateway' : gateway,
|
|
||||||
'relay' : relay,
|
|
||||||
'channelId' : c['iid'],
|
|
||||||
})
|
|
||||||
break
|
|
||||||
|
|
||||||
for _gateway in gateways:
|
for _gateway in gateways:
|
||||||
|
if len(gateway_id) > 0:
|
||||||
|
if _gateway["agentId"].lower() != gateway_id.lower() and _gateway["name"].lower() != gateway_id.lower():
|
||||||
|
continue
|
||||||
|
|
||||||
gateway = getRequest(f'/api/gateway/{_gateway["agentId"]}')
|
gateway = getRequest(f'/api/gateway/{_gateway["agentId"]}')
|
||||||
|
|
||||||
if type(gateway) != dict:
|
for channel in gateway['channels']:
|
||||||
|
if len(channel_id) > 0:
|
||||||
|
if channel["iid"].lower() != channel_id.lower():
|
||||||
continue
|
continue
|
||||||
|
|
||||||
if 'channels' in gateway.keys():
|
name = getDeviceName(gateway, 'channels', channel['type'])
|
||||||
channel_num = 0
|
|
||||||
hadGatewayId = False
|
|
||||||
|
|
||||||
if hasattr(args, 'gateway_id') and args.gateway_id != None:
|
if name.lower() != channelName.lower():
|
||||||
hadGatewayId = True
|
|
||||||
if (args.gateway_id == gateway['agentId'].lower()) or (args.gateway_id == gateway['name'].lower()):
|
|
||||||
pass
|
|
||||||
else:
|
|
||||||
continue
|
continue
|
||||||
|
|
||||||
Logger.dbg(f'Checking channels bound to Gateway {gateway["name"]} / {gateway["agentId"]}')
|
Logger.dbg(f'Adding channel {c["iid"]} in Relay {relay["name"]}.')
|
||||||
|
channels.append({
|
||||||
|
'url' : f'/api/gateway/{gateway["agentId"]}/relay/{relay["agentId"]}/channel/{channel["iid"]}/command',
|
||||||
|
'gateway' : gateway,
|
||||||
|
'relay' : relay,
|
||||||
|
'channelId' : channel['iid'],
|
||||||
|
})
|
||||||
|
|
||||||
for c in gateway['channels']:
|
for relay in gateway['relays']:
|
||||||
channel_num += 1
|
if len(relay_id) > 0:
|
||||||
Logger.dbg(f'Iterating over channel {c["iid"]} in Gateway...')
|
if relay["agentId"].lower() != relay_id.lower() and relay["name"].lower() != relay_id.lower():
|
||||||
if channel_id != None:
|
continue
|
||||||
if c['iid'] == channel_id:
|
|
||||||
Logger.dbg(f'Adding channel {c["iid"]} in gateway {gateway["name"]}.')
|
if 'channels' in relay.keys():
|
||||||
|
for channel in relay['channels']:
|
||||||
|
if len(channel_id) > 0:
|
||||||
|
if channel["iid"].lower() != channel_id.lower():
|
||||||
|
continue
|
||||||
|
|
||||||
|
name = getDeviceName(gateway, 'channels', channel['type'])
|
||||||
|
|
||||||
|
if name.lower() != channelName.lower():
|
||||||
|
continue
|
||||||
|
|
||||||
|
Logger.dbg(f'Adding channel {channel["iid"]} in Relay {relay["name"]}.')
|
||||||
channels.append({
|
channels.append({
|
||||||
'url' : f'/api/gateway/{gateway["agentId"]}/channel/{c["iid"]}/command',
|
'url' : f'/api/gateway/{gateway["agentId"]}/relay/{relay["agentId"]}/channel/{channel["iid"]}/command',
|
||||||
'gateway' : gateway,
|
'gateway' : gateway,
|
||||||
'channelId' : c['iid'],
|
'relay' : relay,
|
||||||
|
'channelId' : channel['iid'],
|
||||||
})
|
})
|
||||||
break
|
|
||||||
else:
|
|
||||||
for arg in c['propertiesText']['arguments']:
|
|
||||||
if type(arg) == dict:
|
|
||||||
if channelKeyword in arg['name'].lower() or ("description" in arg.keys() and channelKeyword in arg['description'].lower()):
|
|
||||||
Logger.dbg(f'Adding channel {c["iid"]} in gateway {gateway["name"]}.')
|
|
||||||
channels.append({
|
|
||||||
'url' : f'/api/gateway/{gateway["agentId"]}/channel/{c["iid"]}/command',
|
|
||||||
'gateway' : gateway,
|
|
||||||
'channelId' : c['iid'],
|
|
||||||
})
|
|
||||||
break
|
|
||||||
return channels
|
return channels
|
||||||
|
|
||||||
def shell(cmd, alternative = False, stdErrToStdout = False, surpressStderr = False):
|
def shell(cmd, alternative = False, stdErrToStdout = False, surpressStderr = False):
|
||||||
|
@ -1090,11 +1076,10 @@ def getValueOrRandom(val, N = 6):
|
||||||
def closeRelay(gateway, relay):
|
def closeRelay(gateway, relay):
|
||||||
gateway = getRequest(f'/api/gateway/{gateway["agentId"]}')
|
gateway = getRequest(f'/api/gateway/{gateway["agentId"]}')
|
||||||
relayMeta = getRequest(f'/api/gateway/{gateway["agentId"]}/relay/{relay["agentId"]}')
|
relayMeta = getRequest(f'/api/gateway/{gateway["agentId"]}/relay/{relay["agentId"]}')
|
||||||
capability = processCapability(gateway)
|
|
||||||
|
|
||||||
print('\n[.] step 1: Closing bound Peripherals')
|
print('\n[.] step 1: Closing bound Peripherals')
|
||||||
for peri in relayMeta['peripherals']:
|
for peri in relayMeta['peripherals']:
|
||||||
name = list(capability['peripherals'].keys())[list(capability['peripherals'].values()).index(peri['type'])]
|
name = getDeviceName(gateway, 'peripherals', peri['type'])
|
||||||
Logger.info(f'Closing relay\'s peripheral {name} id:{peri["iid"]}')
|
Logger.info(f'Closing relay\'s peripheral {name} id:{peri["iid"]}')
|
||||||
closePeripheral(gateway, relay, name, peri['iid'])
|
closePeripheral(gateway, relay, name, peri['iid'])
|
||||||
|
|
||||||
|
@ -1107,7 +1092,7 @@ def closeRelay(gateway, relay):
|
||||||
grcChannel = chan
|
grcChannel = chan
|
||||||
continue
|
continue
|
||||||
|
|
||||||
chanName = list(capability['channels'].keys())[list(capability['channels'].values()).index(chan['type'])]
|
chanName = getDeviceName(gateway, 'channels', chan['type'])
|
||||||
Logger.info(f'Closing relay\'s channel {chanName} id:{chan["iid"]}')
|
Logger.info(f'Closing relay\'s channel {chanName} id:{chan["iid"]}')
|
||||||
|
|
||||||
chan['url'] = f'/api/gateway/{gateway["agentId"]}/relay/{relay["agentId"]}/channel/{chan["iid"]}/command'
|
chan['url'] = f'/api/gateway/{gateway["agentId"]}/relay/{relay["agentId"]}/channel/{chan["iid"]}/command'
|
||||||
|
@ -1116,7 +1101,7 @@ def closeRelay(gateway, relay):
|
||||||
if not grcChannel:
|
if not grcChannel:
|
||||||
Logger.fatal(f'Could not determine Gateway-Return Channel of the specified Relay {relay["name"]} / {relay["agentId"]}. \n Probably its unreachable or already closed.')
|
Logger.fatal(f'Could not determine Gateway-Return Channel of the specified Relay {relay["name"]} / {relay["agentId"]}. \n Probably its unreachable or already closed.')
|
||||||
|
|
||||||
closeChannel(grcChannel, list(capability['channels'].keys())[list(capability['channels'].values()).index(grcChannel['type'])])
|
closeChannel(grcChannel, getDeviceName(gateway, 'channels', grcChannel['type']))
|
||||||
|
|
||||||
print('\n[.] step 3: closing Relay itself')
|
print('\n[.] step 3: closing Relay itself')
|
||||||
data = {
|
data = {
|
||||||
|
@ -1149,7 +1134,7 @@ def closeRelay(gateway, relay):
|
||||||
else:
|
else:
|
||||||
chan['url'] = f'/api/gateway/{gateway["agentId"]}/relay/{relayNode["agentId"]}/channel/{chan["iid"]}/command'
|
chan['url'] = f'/api/gateway/{gateway["agentId"]}/relay/{relayNode["agentId"]}/channel/{chan["iid"]}/command'
|
||||||
|
|
||||||
closeChannel(chan, list(capability['channels'].keys())[list(capability['channels'].values()).index(chan['type'])])
|
closeChannel(chan, getDeviceName(gateway, 'channels', chan['type']))
|
||||||
closed = True
|
closed = True
|
||||||
break
|
break
|
||||||
if closed: break
|
if closed: break
|
||||||
|
@ -1259,7 +1244,7 @@ def onCloseChannel(args):
|
||||||
if len(args.gateway_id) > 0:
|
if len(args.gateway_id) > 0:
|
||||||
if gateway["agentId"].lower() == args.agent_id.lower() or gateway["name"].lower() == args.agent_id.lower():
|
if gateway["agentId"].lower() == args.agent_id.lower() or gateway["name"].lower() == args.agent_id.lower():
|
||||||
for channel in gateway['channels']:
|
for channel in gateway['channels']:
|
||||||
name = list(capability['channels'].keys())[list(capability['channels'].values()).index(channel['type'])]
|
name = getDeviceName(gateway, 'channels', channel['type'])
|
||||||
if len(args.channel_id) == 0 or (name.lower() == args.channel_id.lower() or channel['iid'] == args.channel_id):
|
if len(args.channel_id) == 0 or (name.lower() == args.channel_id.lower() or channel['iid'] == args.channel_id):
|
||||||
_type = 'non-negotiation'
|
_type = 'non-negotiation'
|
||||||
if 'isReturnChannel' in channel.keys() and channel['isReturnChannel']: _type = 'grc'
|
if 'isReturnChannel' in channel.keys() and channel['isReturnChannel']: _type = 'grc'
|
||||||
|
@ -1279,7 +1264,7 @@ def onCloseChannel(args):
|
||||||
continue
|
continue
|
||||||
|
|
||||||
for channel in relay['channels']:
|
for channel in relay['channels']:
|
||||||
name = list(capability['channels'].keys())[list(capability['channels'].values()).index(channel['type'])]
|
name = getDeviceName(gateway, 'channels', channel['type'])
|
||||||
if len(args.channel_id) == 0 or (name.lower() == args.channel_id.lower() or channel['iid'] == args.channel_id):
|
if len(args.channel_id) == 0 or (name.lower() == args.channel_id.lower() or channel['iid'] == args.channel_id):
|
||||||
|
|
||||||
_type = 'non-negotiation'
|
_type = 'non-negotiation'
|
||||||
|
|
Loading…
Reference in New Issue