Exemple #1
0
 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)
Exemple #2
0
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,
                }
            )