rz-agent.c File Reference

#include <rz_main.h>
#include <rz_core.h>
#include <rz_socket.h>

Go to the source code of this file.

Macros
#define	USE_IOS_JETSAM   0

Functions
static int	usage (int v)
static int	showversion (void)
RZ_API int	rz_main_rz_agent (int argc, const char **argv)

Variables
static const char *	page_index

Macro Definition Documentation

USE_IOS_JETSAM

#define USE_IOS_JETSAM   0

Definition at line 43 of file rz-agent.c.

Function Documentation

rz_main_rz_agent()

RZ_API int rz_main_rz_agent	(	int	argc,
		const char **	argv
	)

Definition at line 61 of file rz-agent.c.

                                                         {
    RzSocket *s;
    RzSocketHTTPOptions so;
    RzSocketHTTPRequest *rs;
    int c;
    int dodaemon = 0;
    bool listenlocal = true;
    const char *port = "8080";
    const char *httpauthfile = NULL;
    char *pfile = NULL;
    memset(&so, 0, sizeof(so));
 
    RzGetopt opt;
    rz_getopt_init(&opt, argc, argv, "adhup:t:sv");
    while ((c = rz_getopt_next(&opt)) != -1) {
        switch (c) {
        case 'a':
            listenlocal = false;
            break;
        case 'd':
            dodaemon = 1;
            break;
        case 'h':
            return usage(1);
        case 'v':
            return showversion();
        case 'u':
            so.httpauth = true;
            break;
        case 't':
            httpauthfile = opt.arg;
            break;
        case 'p':
            port = opt.arg;
            break;
        default:
            return usage(0);
        }
    }
    if (opt.ind != argc) {
        return usage(0);
    }
 
    so.accept_timeout = 0;
    so.timeout = 3;
 
    if (so.httpauth) {
        if (!httpauthfile) {
            eprintf("No authentication user list set\n");
            return usage(0);
        }
 
        size_t sz;
        pfile = rz_file_slurp(httpauthfile, &sz);
        if (pfile) {
            so.authtokens = rz_str_split_list(pfile, "\n", 0);
        } else {
            eprintf("Empty list of HTTP users\\n");
            return usage(0);
        }
    }
#if USE_IOS_JETSAM
    memorystatus_control(MEMORYSTATUS_CMD_SET_JETSAM_TASK_LIMIT, getpid(), 256, NULL, 0);
#endif
    if (dodaemon) {
#if HAVE_FORK
        int pid = rz_sys_fork();
        if (pid > 0) {
            printf("%d\n", pid);
            return 0;
        }
#endif
    }
    s = rz_socket_new(false);
    s->local = listenlocal;
    if (!rz_socket_listen(s, port, NULL)) {
        eprintf("Cannot listen on %d\n", s->port);
        rz_socket_free(s);
        return 1;
    }
 
    eprintf("http://localhost:%d/\n", s->port);
 
    (void)rz_cons_new();
 
    while (!rz_cons_singleton()->context->breaked) {
        char *result_heap = NULL;
        const char *result = page_index;
 
        rs = rz_socket_http_accept(s, &so);
        if (!rs) {
            continue;
        }
        if (!rs->auth) {
            rz_socket_http_response(rs, 401, "", 0, NULL);
        }
        if (!strcmp(rs->method, "GET")) {
            if (!strncmp(rs->path, "/proc/kill/", 11)) {
                // TODO: show page here?
                int pid = atoi(rs->path + 11);
                if (pid > 0) {
#if __WINDOWS__
                    rz_sys_kill(pid, 0);
#else
                    rz_sys_kill(pid, SIGKILL);
#endif
                }
            } else if (!strncmp(rs->path, "/file/open/", 11)) {
                int pid;
                int session_port = 3000 + rz_num_rand(1024);
                char *filename = rs->path + 11;
                char *escaped_filename = rz_str_escape(filename);
                size_t escaped_len = strlen(escaped_filename);
                size_t cmd_len = escaped_len + 40;
                char *cmd;
 
                if (!(cmd = malloc(cmd_len))) {
                    perror("malloc");
                    return 1;
                }
                snprintf(cmd, cmd_len, "rizin -q %s-e http.port=%d -cRh \"%s\"",
                    listenlocal ? "" : "-e http.bind=public ",
                    session_port, escaped_filename);
 
                // TODO: use rz_sys api to get pid when running in bg
                pid = rz_sys_cmdbg(cmd);
                free(cmd);
                free(escaped_filename);
                result = result_heap = malloc(1024 + escaped_len);
                if (!result) {
                    perror("malloc");
                    free(pfile);
                    rz_list_free(so.authtokens);
                    return 1;
                }
                sprintf(result_heap,
                    "<html><body>"
                    "<a href='/'>back</a><hr size=1/>"
                    " - <a target='_blank' href='http://localhost:%d/'>open</a><br />"
                    " - <a href='/proc/kill/%d'>kill</a><br />"
                    "</body></html>",
                    session_port, pid);
                eprintf("\nchild pid %d\n\n", pid);
            }
        }
        rz_socket_http_response(rs, 200, result, 0, NULL);
        rz_socket_http_close(rs);
        free(result_heap);
        result_heap = NULL;
    }
    rz_cons_free();
    free(pfile);
    rz_list_free(so.authtokens);
    rz_socket_free(s);
    return 0;
}

References rz_socket_http_options::accept_timeout, rz_getopt_t::arg, argv, rz_socket_http_options::authtokens, rz_cons_context_t::breaked, c, cmd, rz_cons_t::context, eprintf, free(), rz_socket_http_options::httpauth, rz_getopt_t::ind, rz_socket_t::local, malloc(), memset(), NULL, page_index, pid, rz_socket_t::port, printf(), rs, rz_cons_free(), rz_cons_new(), rz_cons_singleton(), rz_file_slurp(), rz_getopt_init(), rz_getopt_next(), rz_list_free(), rz_num_rand(), rz_socket_free(), rz_socket_http_accept(), rz_socket_http_close(), rz_socket_http_response(), rz_socket_listen(), rz_socket_new(), rz_str_escape(), rz_str_split_list(), rz_sys_cmdbg(), rz_sys_fork(), rz_sys_kill(), s, showversion(), SIGKILL, snprintf, sprintf, rz_socket_http_options::timeout, and usage().

Referenced by MAIN_NAME().

showversion()

static int showversion ( void  )

static

Definition at line 57 of file rz-agent.c.

                             {
    return rz_main_version_print("rz-agent");
}

References rz_main_version_print().

Referenced by rz_main_rz_agent().

usage()

static int usage ( int  v )

static

Definition at line 46 of file rz-agent.c.

                        {
    printf("Usage: rz-agent [-adhs] [-p port]\n"
           "  -a        listen for everyone (localhost by default)\n"
           "  -d        run in daemon mode (background)\n"
           "  -h        show this help message\n"
           "  -u        enable http Authorization access\n"
           "  -t        user:password authentication file\n"
           "  -p [port] specify listening port (defaults to 8080)\n");
    return !v;
}

References printf(), and v.

Referenced by rz_main_rz_agent().

Variable Documentation

page_index

const char* page_index

static

Initial value:

=
    "<html>"
    "<head>"
    "<title>rz-agent</title>"
    "<script>"
    "function file_open() {"
    "  var file = window.prompt ('path to file?');"
    "  window.open ('/file/open/'+file);"
    "}"
    "</script>"
    "</head>"
    "<body>"
    "  <h2>rz-agent</h2>"
    "  <hr size=1 />"
    "File<br />"
    " - <a href='javascript:file_open()'>open</a><br />"
    " - <a href='/file/list'>list</a><br />"
    " - push<br />"
    " - pull<br />"
    " - delete<br />"
    " - chmod<br />"
    "Processes<br />"
    " - list<br />"
    "Sessions<br />"
    " - list<br />"
    "</body>"
    "</html>"

Definition at line 8 of file rz-agent.c.

Referenced by rebase_bytes_v1(), rebase_bytes_v2(), rebase_bytes_v3(), reconstruct_chained_fixup(), and rz_main_rz_agent().