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
usage(void)22 usage(void)
23 {
24 fprint(2, "usage: venti/ro [-v] [-a address] [-h address]\n");
25 threadexitsall("usage");
26 }
27
28 void
readthread(void * v)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
threadmain(int argc,char ** argv)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