sys/src/cmd/venti/ro.c

   1 /* Copyright (c) 2004 Russ Cox */
   2 #include <u.h>
   3 #include <libc.h>
   4 #include <venti.h>
   5 #include <thread.h>
   6 #include <libsec.h>
   7
   8 #ifndef _UNISTD_H_
   9 #pragma varargck type "F" VtFcall*
  10 #pragma varargck type "T" void
  11 #endif
  12
  13 VtConn *z;
  14 int verbose;
  15
  16 enum
  17 {
  18         STACK = 8192
  19 };
  20
  21 void
  22 usage(void)
  23 {
  24         fprint(2, "usage: venti/ro [-v] [-a address] [-h address]\n");
  25         threadexitsall("usage");
  26 }
  27
  28 void
  29 readthread(void *v)
  30 {
  31         char err[ERRMAX];
  32         VtReq *r;
  33         uchar *buf;
  34         int n;
  35
  36         r = v;
  37         buf = vtmalloc(r->tx.count);
  38         if((n=vtread(z, r->tx.score, r->tx.blocktype, buf, r->tx.count)) < 0){
  39                 r->rx.msgtype = VtRerror;
  40                 rerrstr(err, sizeof err);
  41                 r->rx.error = vtstrdup(err);
  42                 free(buf);
  43         }else{
  44                 r->rx.data = packetforeign(buf, n, free, buf);
  45         }
  46         if(verbose)
  47                 fprint(2, "-> %F\n", &r->rx);
  48         vtrespond(r);
  49 }
  50
  51 void
  52 threadmain(int argc, char **argv)
  53 {
  54         VtReq *r;
  55         VtSrv *srv;
  56         char *address, *ventiaddress;
  57
  58         fmtinstall('F', vtfcallfmt);
  59         fmtinstall('V', vtscorefmt);
  60
  61         address = "tcp!*!venti";
  62         ventiaddress = nil;
  63
  64         ARGBEGIN{
  65         case 'v':
  66                 verbose++;
  67                 break;
  68         case 'a':
  69                 address = EARGF(usage());
  70                 break;
  71         case 'h':
  72                 ventiaddress = EARGF(usage());
  73                 break;
  74         default:
  75                 usage();
  76         }ARGEND
  77
  78         if((z = vtdial(ventiaddress)) == nil)
  79                 sysfatal("vtdial %s: %r", ventiaddress);
  80         if(vtconnect(z) < 0)
  81                 sysfatal("vtconnect: %r");
  82
  83         srv = vtlisten(address);
  84         if(srv == nil)
  85                 sysfatal("vtlisten %s: %r", address);
  86
  87         while((r = vtgetreq(srv)) != nil){
  88                 r->rx.msgtype = r->tx.msgtype+1;
  89                 if(verbose)
  90                         fprint(2, "<- %F\n", &r->tx);
  91                 switch(r->tx.msgtype){
  92                 case VtTping:
  93                         break;
  94                 case VtTgoodbye:
  95                         break;
  96                 case VtTread:
  97                         threadcreate(readthread, r, 16384);
  98                         continue;
  99                 case VtTwrite:
 100                         r->rx.error = vtstrdup("read-only server");
 101                         r->rx.msgtype = VtRerror;
 102                         break;
 103                 case VtTsync:
 104                         break;
 105                 }
 106                 if(verbose)
 107                         fprint(2, "-> %F\n", &r->rx);
 108                 vtrespond(r);
 109         }
 110         threadexitsall(nil);
 111 }
 112