def request(self, request_body, file=None):
username, hostname = self.host.split('@', 1)
self.client.connect(hostname, username=username)
stdin, stdout, stderr = self.client.exec_command(request_body)
if file:
stdin.write(file.read())
stdin.flush()
stdin.channel.shutdown(2)
results = stderr.read()
else:
results = stderr.read()
return process_response(results)
def main(argv=sys.argv):
if len(argv) < 2:
print "thundercat - client for interfacing with nappingcat servers"
print "usage:"
print "\tthundercat addremote <nickname> <*****@*****.**>"
print "\tthundercat help <nickname> <command>"
print "\tthundercat <nickname> -- list commands"
print "\tthundercat <nickname> <command> <args> -- send a command to the host"
return
subcommand = argv[1]
if subcommand == "addremote":
addhost.add_host(argv[1:4])
discover.discover_command(["discover", argv[2]])
elif subcommand == "help":
host_info = config.get_host_info(argv[2])
print host_info["endpoints"][argv[3]]["help_text"].replace("{EXECUTABLE}", os.path.basename(argv[0]))
else:
nickname = argv[1]
host_info = config.get_host_info(nickname)
if host_info:
endpoint = host_info["endpoints"].get(argv[2], None) if len(argv) > 2 else None
if endpoint:
simple_regex = endpoint["regex"]
for possibility, arg_names in simple_regex:
response_dict = {}
available_args = argv[3:]
available_args.reverse()
for arg_name in arg_names:
parameter_type = endpoint["params"].pop(arg_name, None)
if isinstance(parameter_type, list):
available_args.reverse()
response_dict[arg_name] = parameter_type[1].join(available_args)
available_args = []
else:
try:
response_dict[arg_name] = available_args.pop()
except IndexError:
pass
file = None
for key, type in endpoint["params"].iteritems():
if type == "stdin":
with open(available_args.pop(), "r") as input:
file = StringIO.StringIO(input.read())
break
try:
result = possibility % response_dict
transport = Transport(host_info["url"])
response = transport.request(result, file)
try:
module = import_module("thundercat.plugins.%s" % argv[2])
target = getattr(module, "main", lambda x: x)
print target(nickname, host_info, response)
except ImportError:
print response
return
except (TypeError, KeyError) as e:
pass
else:
print process_response(
{
"content": {
"message": "Bad endpoint %s\nAvailable endpoints:\n\t%s\n"
% (
argv[2] if len(argv) > 2 else "<none provided>",
"\n\t".join(host_info["endpoints"].keys()),
)
},
"status_code": 404,
}
)
else:
print process_response(
{
"content": {
"message": "Bad host %s\nAvailable hosts:\n\t%s\n"
% (argv[1], "\n\t".join(config.get_available_host_list()))
},
"status_code": 404,
}
)