def do_check(self,args):
""" test all agents login ssh """
agentsCount = 0
self.settings['check'] = []
for agent in self.db.execute(C.selectAllBots):
agent = list(agent)
agent.insert(len(agent),self.sshConnect.ssh(agent[1],agent[2],agent[3],agent[4],checkconnect=True).status)
self.settings['check'].append(agent)
color.display_messages('Available Bots:', info=True, sublime=True)
print tabulate(self.settings['check'], headers=C.headersCheck)
for items in self.settings['check']:
if search('ON',items[6]):agentsCount +=1
color.linefeed()
color.display_messages('Online Agents: {}'.format(color.setcolor(str(agentsCount), color='blue')), info=True)
def do_register(self,args):
""" add bot on database """
arg_parser = argparse.ArgumentParser(prog='register',description='add bot on database clients')
arg_parser.add_argument('--host', dest='host',metavar='<Host>', help='ipaddress/host/dns connect ssh')
arg_parser.add_argument('--pass', dest='password',metavar='<Password> <filepath>', help='password/ssh key ssh client')
arg_parser.add_argument('-u', '--user',dest='user', metavar='<user>', help='delete all bot registered')
arg_parser.add_argument('-p', '--port', dest='port',metavar='<Port>',default='22', help='port connect ssh')
arg_parser.add_argument('-f', '--file', dest='file',metavar='<filepath>', help='imports clients from a file')
try:
args=arg_parser.parse_args(shlex.split(args))
except: return
if args.host and args.password and args.user:
color.display_messages('Insert Data: SQL statement will insert a new row', info=True)
C.DB_insert(self.con, self.db, args.host, args.port, args.user, args.password)
color.display_messages('credentials ssh added with success', sucess=True)
elif args.file:
self.all_bot_checked = []
color.display_messages('searching for: {} ...'.format(path.realpath(args.file)),info=True)
if path.exists(args.file):
self.lines_all_read = [line.rstrip('\n') for line in open(path.realpath(args.file),'r')]
for items in self.lines_all_read:
if len(items.split()) == 4:
self.all_bot_checked.append(items)
if len(self.all_bot_checked) == 0:
color.display_messages('Instruction for -f argumments:',info=True,sublime=True)
print('You need to use the separator character [space] in this format below\n')
print('-----cut here -------\n')
print('root:~# cat example.txt\n')
print('192.168.0.100 22 DemoBOT P@ssW0rd or ssh_key file')
print('192.168.0.120 22 b1tifi botP@ssW0rd ssh_key file')
print('\n-----cut here -------\n')
print('')
return None
self.ListBot = []
color.display_messages('All agents imported from file:', info=True,sublime=True)
for agent in self.all_bot_checked: self.ListBot += list([agent.split()])
print tabulate(self.ListBot, headers=C.headersimport)
color.linefeed()
choise = raw_input('{}[*]{} Do you want to import?(S/N):'.format(color.colors.GREEN,color.colors.ENDC,))
if choise.lower() == 's':
color.display_messages('Importing agents...',info=True)
for agent in self.all_bot_checked:
C.DB_insert(self.con, self.db,
agent.split()[0],agent.split()[1],agent.split()[2],agent.split()[3])
return color.display_messages('all agents ssh added with success', sucess=True)
color.display_messages('import was been canceled.', error=True)
else:
color.display_messages('file: could not be found', error=True)
else:
arg_parser.print_help()
def do_list(self,args):
""" list/check/filter list agents on database """
arg_parser = argparse.ArgumentParser(prog='list',description='interact with one/all agents')
arg_parser.add_argument('-i', '--id', dest='id',type=int,metavar='<id>', help='list agent by id')
arg_parser.add_argument('-c', '--check',dest='check', action='store_true', help='check credentials by agent Available')
arg_parser.add_argument('-d', '--db',dest='database', action='store_true', help='list all agents on database')
try:
args=arg_parser.parse_args(shlex.split(args))
except: return
self.settings['all'] = {}
self.listbotsprint = []
for agent in self.db.execute(C.selectAllBots):self.settings['all'][agent[0]] = agent
if self.settings['all'] == {}:
return color.display_messages('No Agents registered',info=True)
if args.database:
if args.id and not args.check:
if not args.id in self.settings['all'].keys():
return color.display_messages('ID not registered',info=True)
color.display_messages('Agents:', info=True, sublime=True)
agent = list(self.settings['all'][args.id])
self.listbotsprint += list([agent])
print tabulate(self.listbotsprint, headers=C.headersCheck)
elif args.check and args.id:
if not args.id in self.settings['all'].keys():
return color.display_messages('ID not registered', info=True)
color.display_messages('Agents:', info=True, sublime=True)
agent = list(self.settings['all'][args.id])
agent.insert(len(agent),self.sshConnect.ssh(agent[1],agent[2],agent[3],agent[4],checkconnect=True).status)
self.listbotsprint += list([agent])
print tabulate(self.listbotsprint, headers=C.headersCheck)
elif args.database and not args.check:
color.display_messages('Agents:', info=True, sublime=True)
for bots in self.settings['all'].items():self.listbotsprint += list([bots[1]])
print tabulate(self.listbotsprint, headers=C.headers)
return color.linefeed()
elif args.database and not args.id:
color.display_messages('Agents:', info=True, sublime=True)
for agent in self.settings['all'].items():
agent = list(agent[1])
agent.insert(len(agent),self.sshConnect.ssh(agent[1],agent[2],agent[3],agent[4],checkconnect=True).status)
self.listbotsprint += list([agent])
print tabulate(self.listbotsprint, headers=C.headersCheck)
color.linefeed()
else:
arg_parser.print_help()
def do_agents(self,args):
""" list all agents in interacting """
list_agents = []
# get all session activated
if self.settings['agents'].keys() != {}:
for agent in self.settings['agents'].keys():
if self.settings['agents'][agent]['tunel'] != None:
if self.settings['agents'][agent]['tunel'].activated:
list_agents.append([agent,self.settings['agents'][agent]['creds']['Host'],
self.settings['agents'][agent]['creds']['Port'],
self.settings['agents'][agent]['tunel'].session.name,
self.settings['agents'][agent]['tunel'].session.pid])
if list_agents != []:
color.display_messages('Session Agents:', info=True, sublime=True)
print tabulate(list_agents,headers=C.headersAgents)
color.linefeed()
return color.display_messages('Online Agents: {}'.format(color.setcolor(str(len(list_agents)),
color='blue')), info=True)
color.display_messages('No agents in interacting', info=True)
def do_help(self,args):
""" show this help """
names = self.get_names()
cmds_doc = []
names.sort()
pname = ''
color.display_messages('Available Commands:', info=True, sublime=True)
for name in names:
if name[:3] == 'do_':
pname = name
cmd = name[3:]
if getattr(self, name).__doc__:
cmds_doc.append((cmd, getattr(self, name).__doc__))
else:
cmds_doc.append((cmd, ""))
#self.stdout.write('%s\n'%str(self.doc_header))
self.stdout.write(' {} {}\n'.format('Commands', 'Description'))
self.stdout.write(' {} {}\n'.format('--------', '-----------'))
for command,doc in cmds_doc:
self.stdout.write(' {:<10} {}\n'.format(command, doc))
color.linefeed()