io_tcp.c File Reference

#include "rz_io.h"
#include "rz_lib.h"
#include <rz_socket.h>
#include <stdio.h>
#include <stdlib.h>
#include "../io_memory.h"

Go to the source code of this file.

Functions
static bool	__check (RzIO *io, const char *pathname, bool many)
static ut8 *	tcpme (const char *pathname, int *code, int *len)
static RzIODesc *	__open (RzIO *io, const char *pathname, int rw, int mode)

Variables
RzIOPlugin	rz_io_plugin_tcp
RZ_API RzLibStruct	rizin_plugin

Function Documentation

__check()

static bool __check ( RzIO *  io, const char *  pathname, bool  many  )

static

Definition at line 11 of file io_tcp.c.

                                                               {
    return (!strncmp(pathname, "tcp://", 6));
}

References pathname.

Referenced by __open().

__open()

static RzIODesc* __open ( RzIO *  io, const char *  pathname, int  rw, int  mode  )

static

Definition at line 67 of file io_tcp.c.

                                                                          {
    if (__check(io, pathname, 0)) {
        int rlen, code;
        RzIOMalloc *mal = RZ_NEW0(RzIOMalloc);
        if (!mal) {
            return NULL;
        }
        mal->offset = 0;
        mal->buf = tcpme(pathname, &code, &rlen);
        if (mal->buf && rlen > 0) {
            mal->size = rlen;
            return rz_io_desc_new(io, &rz_io_plugin_tcp, pathname, rw, mode, mal);
        }
        eprintf("No TCP segment\n");
        free(mal);
    }
    return NULL;
}

References __check(), code, eprintf, free(), mal, NULL, pathname, rz_io_desc_new(), rz_io_plugin_tcp, RZ_NEW0, and tcpme().

tcpme()

static ut8* tcpme ( const char *  pathname, int *  code, int *  len  )

static

Definition at line 15 of file io_tcp.c.

                                                             {
    pathname += 6;
    *code = 404;
#if __UNIX__
    rz_sys_signal(SIGINT, SIG_IGN);
#endif
    if (*pathname == ':') {
        /* listen and wait for connection */
        RzSocket *sl = rz_socket_new(false);
        if (!rz_socket_listen(sl, pathname + 1, NULL)) {
            eprintf("Cannot listen\n");
            rz_socket_free(sl);
            return NULL;
        }
        RzSocket *sc = rz_socket_accept(sl);
        ut8 *res = rz_socket_slurp(sc, len);
        rz_socket_free(sc);
        rz_socket_free(sl);
        if (res) {
            *code = 200;
            return res;
        }
    } else {
        /* connect and slurp the end point */
        char *host = strdup(pathname);
        if (!host) {
            return NULL;
        }
        char *port = strchr(host, ':');
        if (port) {
            *port++ = 0;
            RzSocket *s = rz_socket_new(false);
            if (rz_socket_connect(s, host, port, RZ_SOCKET_PROTO_TCP, 0)) {
                ut8 *res = rz_socket_slurp(s, len);
                if (*len < 1) {
                    RZ_FREE(res);
                } else {
                    *code = 200;
                }
                rz_socket_free(s);
                free(host);
                return res;
            }
            rz_socket_free(s);
        } else {
            eprintf("Missing port.\n");
        }
        free(host);
    }
    return NULL;
}

References eprintf, free(), len, NULL, pathname, RZ_FREE, rz_socket_accept(), rz_socket_connect(), rz_socket_free(), rz_socket_listen(), rz_socket_new(), RZ_SOCKET_PROTO_TCP, rz_socket_slurp(), rz_sys_signal(), s, sc, and strdup().

Referenced by __open().

Variable Documentation

rizin_plugin

RZ_API RzLibStruct rizin_plugin

Initial value:

= {
    .type = RZ_LIB_TYPE_IO,
    .data = &rz_io_plugin_tcp,
    .version = RZ_VERSION
}

Definition at line 100 of file io_tcp.c.

rz_io_plugin_tcp

RzIOPlugin rz_io_plugin_tcp

Initial value:

= {
    .name = "tcp",
    .desc = "Load files via TCP (listen or connect)",
    .uris = "tcp://",
    .license = "LGPL3",
    .open = __open,
    .close = io_memory_close,
    .read = io_memory_read,
    .check = __check,
    .lseek = io_memory_lseek,
    .write = io_memory_write,
}

Definition at line 86 of file io_tcp.c.

Referenced by __open().