def initialize_runners(social_channels):
# start a runner for each user in a channel
for channel, users in social_channels.items():
if isinstance(users, list):
for user in users:
runner = Runner(channel, user)
runner.stalk()
def destroy_all(dto):
host = dto['host']
uids = Logger.get_uid_list()
for uid in uids:
args = Command.get_command('destroy', 'destroy_injection' ) \
+ uid
Runner.run_adhoc(host, args)
Logger.clear_uid_file()
return "Destroyed all injections successfully"
def inject_random(dto):
host = dto['host']
target = K8sObserver.get_random_name(dto)
if dto.get('cpu-percent') == None:
# Inject Pod
namespace = dto['namespace']
args = Command.get_command('pod_injection', 'pod_delete') \
+ "--names " + target \
+ " --namespace " + namespace \
+ Command.get_config()
else:
# Inject Node
cpu_percent = dto['cpu-percent']
args = Command.get_command('node_injection', 'cpu_load') \
+ "--cpu-percent " + cpu_percent \
+ " --names " + target \
+ Command.get_config()
r_success_dict = Runner.run_adhoc(host, args)
# Handle result
return Handler.get_stdout_info(r_success_dict)
def destroy_injection(dto):
host = dto['host']
args = Command.get_command('destroy', 'destroy_injection') \
+ dto['uid']
r_success_dict = Runner.run_adhoc(host, args)
return Handler.get_stdout_info(r_success_dict)
def inject_pod_network_dns(dto):
host = dto['host']
args = Command.get_command('pod_injection', 'pod_network_dns') \
+ Command.parser(dto) \
+ Command.get_config()
r_success_dict = Runner.run_adhoc(host, args)
return Handler.get_stdout_info(r_success_dict)
def inject_node_cpu(dto):
host = dto['host']
args = Command.get_command('node_injection', 'cpu_load') \
+ Command.parser(dto) \
+ Command.get_config()
#print(args)
r_success_dict = Runner.run_adhoc(host, args)
return Handler.get_stdout_info(r_success_dict)
def inject_node_network_loss(dto):
(host, names) = (dto['host'], dto['names'])
args = Command.get_command('node_injection', 'network_loss') \
+ Command.parser(dto) \
+ Command.get_command('network_interface', 'node_' + names) \
+ Command.get_config()
#print(args)
r_success_dict = Runner.run_adhoc(host, args)
return Handler.get_stdout_info(r_success_dict)
def inject_pod_delete_by_name(dto):
host = dto['host']
args = Command.get_command('pod_injection', 'pod_delete') \
+ Command.parser(dto) \
+ Command.get_config()
#print(args)
r_success_dict = Runner.run_adhoc(host, args)
return Handler.get_stdout_info(r_success_dict)
def get_info(dto):
# Decodes dto and get commands
(host, namespace) = (dto.get('host'), dto.get('namespace'))
if namespace == None:
args = Command.get_command('kubernetes_info', 'node_info')
elif namespace == 'all':
args = Command.get_command('kubernetes_info', 'pod_info') + " --all-namespaces"
else:
args = Command.get_command('kubernetes_info', 'pod_info') + " --namespace " + namespace
# Run ansible
r_success_dict = Runner.run_adhoc(host,args)
# Handle return result
ip = list(r_success_dict.keys())[0]
return r_success_dict[ip]["stdout_lines"]