Subversion Repositories planix.SVN

/* Copyright © 2003 Russ Cox, MIT; see /sys/src/libsunrpc/COPYING */
#include <u.h>
#include <libc.h>
#include <thread.h>
#include <sunrpc.h>

int chatty;
SunClient *client;

void
usage(void)
{
        fprint(2, "usage: portmap address [cmd]\n"
                "cmd is one of:\n"
                "\tnull\n"
                "\tset prog vers proto port\n"
                "\tunset prog vers proto port\n"
                "\tgetport prog vers proto\n"
                "\tdump (default)\n");
        threadexitsall("usage");
}

void
portCall(SunCall *c, PortCallType type)
{
        c->rpc.prog = PortProgram;
        c->rpc.vers = PortVersion;
        c->rpc.proc = type>>1;
        c->rpc.iscall = !(type&1);
        c->type = type;
}

void
tnull(char **argv)
{
        PortTNull tx;
        PortRNull rx;

        USED(argv);

        memset(&tx, 0, sizeof tx);
        portCall(&tx.call, PortCallTNull);

        memset(&rx, 0, sizeof rx);
        portCall(&rx.call, PortCallRNull);

        if(sunClientRpc(client, 0, &tx.call, &rx.call, nil) < 0)
                sysfatal("rpc: %r");
}

void
tset(char **argv)
{
        PortTSet tx;
        PortRSet rx;

        memset(&tx, 0, sizeof tx);
        portCall(&tx.call, PortCallTSet);
        tx.map.prog = strtol(argv[0], 0, 0);
        tx.map.vers = strtol(argv[1], 0, 0);
        tx.map.prot = strtol(argv[2], 0, 0);
        tx.map.port = strtol(argv[3], 0, 0);

        memset(&rx, 0, sizeof rx);
        portCall(&rx.call, PortCallRSet);

        if(sunClientRpc(client, 0, &tx.call, &rx.call, nil) < 0)
                sysfatal("rpc: %r");

        if(rx.b == 0)
                print("rejected\n");
}

void
tunset(char **argv)
{
        PortTUnset tx;
        PortRUnset rx;

        memset(&tx, 0, sizeof tx);
        portCall(&tx.call, PortCallTUnset);
        tx.map.prog = strtol(argv[0], 0, 0);
        tx.map.vers = strtol(argv[1], 0, 0);
        tx.map.prot = strtol(argv[2], 0, 0);
        tx.map.port = strtol(argv[3], 0, 0);

        memset(&rx, 0, sizeof rx);
        portCall(&rx.call, PortCallRUnset);

        if(sunClientRpc(client, 0, &tx.call, &rx.call, nil) < 0)
                sysfatal("rpc: %r");

        if(rx.b == 0)
                print("rejected\n");
}

void
tgetport(char **argv)
{
        PortTGetport tx;
        PortRGetport rx;

        memset(&tx, 0, sizeof tx);
        portCall(&tx.call, PortCallTGetport);
        tx.map.prog = strtol(argv[0], 0, 0);
        tx.map.vers = strtol(argv[1], 0, 0);
        tx.map.prot = strtol(argv[2], 0, 0);

        memset(&rx, 0, sizeof rx);
        portCall(&rx.call, PortCallRGetport);

        if(sunClientRpc(client, 0, &tx.call, &rx.call, nil) < 0)
                sysfatal("rpc: %r");

        print("%d\n", rx.port);
}

void
tdump(char **argv)
{
        int i;
        uchar *p;
        PortTDump tx;
        PortRDump rx;
        PortMap *m;

        USED(argv);

        memset(&tx, 0, sizeof tx);
        portCall(&tx.call, PortCallTDump);

        memset(&rx, 0, sizeof rx);
        portCall(&rx.call, PortCallRDump);

        if(sunClientRpc(client, 0, &tx.call, &rx.call, &p) < 0)
                sysfatal("rpc: %r");

        for(i=0, m=rx.map; i<rx.nmap; i++, m++)
                print("%ud %ud %ud %ud\n", (uint)m->prog, (uint)m->vers, (uint)m->prot, (uint)m->port);

        free(p);
}

static struct {
        char *cmd;
        int narg;
        void (*fn)(char**);
} tab[] = {
        "null", 0,      tnull,
        "set",  4,      tset,
        "unset",        4,      tunset,
        "getport",      3,      tgetport,
        "dump", 0,      tdump,
};

void
threadmain(int argc, char **argv)
{
        char *dflt[] = { "dump", };
        char *addr, *cmd;
        int i;

        ARGBEGIN{
        case 'R':
                chatty++;
                break;
        }ARGEND

        if(argc < 1)
                usage();

        fmtinstall('B', sunRpcFmt);
        fmtinstall('C', sunCallFmt);
        fmtinstall('H', encodefmt);
        sunFmtInstall(&portProg);

        addr = netmkaddr(argv[0], "udp", "portmap");
        if((client = sunDial(addr)) == nil)
                sysfatal("dial %s: %r", addr);

        client->chatty = chatty;
        sunClientProg(client, &portProg);

        argv++;
        argc--;

        if(argc == 0){
                argc = 1;
                argv = dflt;
        }
        cmd = argv[0];
        argv++;
        argc--;

        for(i=0; i<nelem(tab); i++){
                if(strcmp(tab[i].cmd, cmd) == 0){
                        if(tab[i].narg != argc)
                                usage();
                        (*tab[i].fn)(argv);
                        threadexitsall(nil);
                }
        }
        usage();
}