sys/src/cmd/venti/read.c

   1 #include <u.h>
   2 #include <libc.h>
   3 #include <venti.h>
   4 #include <libsec.h>
   5 #include <thread.h>
   6
   7 void
   8 usage(void)
   9 {
  10         fprint(2, "usage: read [-h host] [-t type] score\n");
  11         threadexitsall("usage");
  12 }
  13
  14 void
  15 threadmain(int argc, char *argv[])
  16 {
  17         int type, n;
  18         uchar score[VtScoreSize];
  19         uchar *buf;
  20         VtConn *z;
  21         char *host;
  22
  23         fmtinstall('F', vtfcallfmt);
  24         fmtinstall('V', vtscorefmt);
  25
  26         host = nil;
  27         type = -1;
  28         ARGBEGIN{
  29         case 'h':
  30                 host = EARGF(usage());
  31                 break;
  32         case 't':
  33                 type = atoi(EARGF(usage()));
  34                 break;
  35         default:
  36                 usage();
  37                 break;
  38         }ARGEND
  39
  40         if(argc != 1)
  41                 usage();
  42
  43         if(vtparsescore(argv[0], nil, score) < 0)
  44                 sysfatal("could not parse score '%s': %r", argv[0]);
  45
  46         buf = vtmallocz(VtMaxLumpSize);
  47
  48         z = vtdial(host);
  49         if(z == nil)
  50                 sysfatal("could not connect to server: %r");
  51
  52         if(vtconnect(z) < 0)
  53                 sysfatal("vtconnect: %r");
  54
  55         if(type == -1){
  56                 n = -1;
  57                 for(type=0; type<VtMaxType; type++){
  58                         n = vtread(z, score, type, buf, VtMaxLumpSize);
  59                         if(n >= 0){
  60                                 fprint(2, "venti/read%s%s %V %d\n", host ? " -h" : "", host ? host : "",
  61                                         score, type);
  62                                 break;
  63                         }
  64                 }
  65         }else
  66                 n = vtread(z, score, type, buf, VtMaxLumpSize);
  67
  68         vthangup(z);
  69         if(n < 0)
  70                 sysfatal("could not read block: %r");
  71         if(write(1, buf, n) != n)
  72                 sysfatal("write: %r");
  73         threadexitsall(0);
  74 }