Subversion Repositories planix.SVN

/* Copyright (c) 2004 Russ Cox */
#include <u.h>
#include <libc.h>
#include <venti.h>
#include <thread.h>
#include <libsec.h>

#ifndef _UNISTD_H_
#pragma varargck type "F" VtFcall*
#pragma varargck type "T" void
#endif

VtConn *z;
int verbose;

enum
{
        STACK = 8192
};

void
usage(void)
{
        fprint(2, "usage: venti/ro [-v] [-a address] [-h address]\n");
        threadexitsall("usage");
}

void
readthread(void *v)
{
        char err[ERRMAX];
        VtReq *r;
        uchar *buf;
        int n;
        
        r = v;
        buf = vtmalloc(r->tx.count);
        if((n=vtread(z, r->tx.score, r->tx.blocktype, buf, r->tx.count)) < 0){
                r->rx.msgtype = VtRerror;
                rerrstr(err, sizeof err);
                r->rx.error = vtstrdup(err);
                free(buf);
        }else{
                r->rx.data = packetforeign(buf, n, free, buf);
        }
        if(verbose)
                fprint(2, "-> %F\n", &r->rx);
        vtrespond(r);
}

void
threadmain(int argc, char **argv)
{
        VtReq *r;
        VtSrv *srv;
        char *address, *ventiaddress;

        fmtinstall('F', vtfcallfmt);
        fmtinstall('V', vtscorefmt);
        
        address = "tcp!*!venti";
        ventiaddress = nil;
        
        ARGBEGIN{
        case 'v':
                verbose++;
                break;
        case 'a':
                address = EARGF(usage());
                break;
        case 'h':
                ventiaddress = EARGF(usage());
                break;
        default:
                usage();
        }ARGEND

        if((z = vtdial(ventiaddress)) == nil)
                sysfatal("vtdial %s: %r", ventiaddress);
        if(vtconnect(z) < 0)
                sysfatal("vtconnect: %r");

        srv = vtlisten(address);
        if(srv == nil)
                sysfatal("vtlisten %s: %r", address);

        while((r = vtgetreq(srv)) != nil){
                r->rx.msgtype = r->tx.msgtype+1;
                if(verbose)
                        fprint(2, "<- %F\n", &r->tx);
                switch(r->tx.msgtype){
                case VtTping:
                        break;
                case VtTgoodbye:
                        break;
                case VtTread:
                        threadcreate(readthread, r, 16384);
                        continue;
                case VtTwrite:
                        r->rx.error = vtstrdup("read-only server");
                        r->rx.msgtype = VtRerror;
                        break;
                case VtTsync:
                        break;
                }
                if(verbose)
                        fprint(2, "-> %F\n", &r->rx);
                vtrespond(r);
        }
        threadexitsall(nil);
}