1 056fe1ba 2003-11-23 devnull #include <u.h>
2 056fe1ba 2003-11-23 devnull #include <libc.h>
3 056fe1ba 2003-11-23 devnull #include <venti.h>
5 a09e80f9 2004-05-23 devnull int ventidoublechecksha1 = 1;
7 056fe1ba 2003-11-23 devnull static int
8 056fe1ba 2003-11-23 devnull vtfcallrpc(VtConn *z, VtFcall *ou, VtFcall *in)
10 056fe1ba 2003-11-23 devnull Packet *p;
12 a09e80f9 2004-05-23 devnull if(chattyventi)
13 a09e80f9 2004-05-23 devnull fprint(2, "%s -> %F\n", argv0, ou);
14 056fe1ba 2003-11-23 devnull p = vtfcallpack(ou);
15 056fe1ba 2003-11-23 devnull if(p == nil)
16 056fe1ba 2003-11-23 devnull return -1;
17 056fe1ba 2003-11-23 devnull if((p = vtrpc(z, p)) == nil)
18 056fe1ba 2003-11-23 devnull return -1;
19 056fe1ba 2003-11-23 devnull if(vtfcallunpack(in, p) < 0){
20 056fe1ba 2003-11-23 devnull packetfree(p);
21 056fe1ba 2003-11-23 devnull return -1;
23 a09e80f9 2004-05-23 devnull if(chattyventi)
24 a09e80f9 2004-05-23 devnull fprint(2, "%s <- %F\n", argv0, in);
25 056fe1ba 2003-11-23 devnull if(in->type == VtRerror){
26 056fe1ba 2003-11-23 devnull werrstr(in->error);
27 056fe1ba 2003-11-23 devnull vtfcallclear(in);
28 056fe1ba 2003-11-23 devnull packetfree(p);
29 056fe1ba 2003-11-23 devnull return -1;
31 056fe1ba 2003-11-23 devnull if(in->type != ou->type+1){
32 056fe1ba 2003-11-23 devnull werrstr("type mismatch: sent %c%d got %c%d",
33 056fe1ba 2003-11-23 devnull "TR"[ou->type&1], ou->type>>1,
34 056fe1ba 2003-11-23 devnull "TR"[in->type&1], in->type>>1);
35 056fe1ba 2003-11-23 devnull vtfcallclear(in);
36 056fe1ba 2003-11-23 devnull packetfree(p);
37 056fe1ba 2003-11-23 devnull return -1;
39 056fe1ba 2003-11-23 devnull packetfree(p);
40 056fe1ba 2003-11-23 devnull return 0;
44 056fe1ba 2003-11-23 devnull vthello(VtConn *z)
46 056fe1ba 2003-11-23 devnull VtFcall tx, rx;
48 056fe1ba 2003-11-23 devnull memset(&tx, 0, sizeof tx);
49 056fe1ba 2003-11-23 devnull tx.type = VtThello;
50 056fe1ba 2003-11-23 devnull tx.version = z->version;
51 056fe1ba 2003-11-23 devnull tx.uid = z->uid;
52 056fe1ba 2003-11-23 devnull if(tx.uid == nil)
53 056fe1ba 2003-11-23 devnull tx.uid = "anonymous";
54 056fe1ba 2003-11-23 devnull if(vtfcallrpc(z, &tx, &rx) < 0)
55 056fe1ba 2003-11-23 devnull return -1;
56 056fe1ba 2003-11-23 devnull z->sid = rx.sid;
57 056fe1ba 2003-11-23 devnull rx.sid = 0;
58 056fe1ba 2003-11-23 devnull vtfcallclear(&rx);
59 056fe1ba 2003-11-23 devnull return 0;
63 056fe1ba 2003-11-23 devnull vtreadpacket(VtConn *z, uchar score[VtScoreSize], uint type, int n)
65 056fe1ba 2003-11-23 devnull VtFcall tx, rx;
67 7cb74894 2004-06-17 devnull if(memcmp(score, vtzeroscore, VtScoreSize) == 0)
68 7cb74894 2004-06-17 devnull return packetalloc();
70 056fe1ba 2003-11-23 devnull memset(&tx, 0, sizeof tx);
71 056fe1ba 2003-11-23 devnull tx.type = VtTread;
72 056fe1ba 2003-11-23 devnull tx.dtype = type;
73 056fe1ba 2003-11-23 devnull tx.count = n;
74 056fe1ba 2003-11-23 devnull memmove(tx.score, score, VtScoreSize);
75 056fe1ba 2003-11-23 devnull if(vtfcallrpc(z, &tx, &rx) < 0)
76 056fe1ba 2003-11-23 devnull return nil;
77 056fe1ba 2003-11-23 devnull if(packetsize(rx.data) > n){
78 056fe1ba 2003-11-23 devnull werrstr("read returned too much data");
79 056fe1ba 2003-11-23 devnull packetfree(rx.data);
80 056fe1ba 2003-11-23 devnull return nil;
82 a09e80f9 2004-05-23 devnull if(ventidoublechecksha1){
83 a09e80f9 2004-05-23 devnull packetsha1(rx.data, tx.score);
84 a09e80f9 2004-05-23 devnull if(memcmp(score, tx.score, VtScoreSize) != 0){
85 a09e80f9 2004-05-23 devnull werrstr("read asked for %V got %V", score, tx.score);
86 a09e80f9 2004-05-23 devnull packetfree(rx.data);
87 a09e80f9 2004-05-23 devnull return nil;
90 056fe1ba 2003-11-23 devnull return rx.data;
94 056fe1ba 2003-11-23 devnull vtread(VtConn *z, uchar score[VtScoreSize], uint type, uchar *buf, int n)
97 056fe1ba 2003-11-23 devnull Packet *p;
99 056fe1ba 2003-11-23 devnull if((p = vtreadpacket(z, score, type, n)) == nil)
100 056fe1ba 2003-11-23 devnull return -1;
101 056fe1ba 2003-11-23 devnull nn = packetsize(p);
102 056fe1ba 2003-11-23 devnull if(packetconsume(p, buf, nn) < 0)
103 056fe1ba 2003-11-23 devnull abort();
104 c5eb6860 2004-06-16 devnull packetfree(p);
105 056fe1ba 2003-11-23 devnull return nn;
109 056fe1ba 2003-11-23 devnull vtwritepacket(VtConn *z, uchar score[VtScoreSize], uint type, Packet *p)
111 056fe1ba 2003-11-23 devnull VtFcall tx, rx;
113 7cb74894 2004-06-17 devnull if(packetsize(p) == 0){
114 7cb74894 2004-06-17 devnull memmove(score, vtzeroscore, VtScoreSize);
115 7cb74894 2004-06-17 devnull return 0;
117 056fe1ba 2003-11-23 devnull tx.type = VtTwrite;
118 056fe1ba 2003-11-23 devnull tx.dtype = type;
119 056fe1ba 2003-11-23 devnull tx.data = p;
120 a09e80f9 2004-05-23 devnull if(ventidoublechecksha1)
121 a09e80f9 2004-05-23 devnull packetsha1(p, score);
122 056fe1ba 2003-11-23 devnull if(vtfcallrpc(z, &tx, &rx) < 0)
123 056fe1ba 2003-11-23 devnull return -1;
124 a09e80f9 2004-05-23 devnull if(ventidoublechecksha1){
125 a09e80f9 2004-05-23 devnull if(memcmp(score, rx.score, VtScoreSize) != 0){
126 a09e80f9 2004-05-23 devnull werrstr("sha1 hash mismatch: want %V got %V", score, rx.score);
127 a09e80f9 2004-05-23 devnull return -1;
130 4192ac1d 2004-06-09 devnull memmove(score, rx.score, VtScoreSize);
131 056fe1ba 2003-11-23 devnull return 0;
135 056fe1ba 2003-11-23 devnull vtwrite(VtConn *z, uchar score[VtScoreSize], uint type, uchar *buf, int n)
137 056fe1ba 2003-11-23 devnull Packet *p;
139 056fe1ba 2003-11-23 devnull p = packetforeign(buf, n, nil, nil);
140 056fe1ba 2003-11-23 devnull return vtwritepacket(z, score, type, p);
144 056fe1ba 2003-11-23 devnull vtsync(VtConn *z)
146 056fe1ba 2003-11-23 devnull VtFcall tx, rx;
148 056fe1ba 2003-11-23 devnull tx.type = VtTsync;
149 056fe1ba 2003-11-23 devnull return vtfcallrpc(z, &tx, &rx);
153 056fe1ba 2003-11-23 devnull vtping(VtConn *z)
155 056fe1ba 2003-11-23 devnull VtFcall tx, rx;
157 056fe1ba 2003-11-23 devnull tx.type = VtTping;
158 056fe1ba 2003-11-23 devnull return vtfcallrpc(z, &tx, &rx);
162 056fe1ba 2003-11-23 devnull vtconnect(VtConn *z)
164 056fe1ba 2003-11-23 devnull if(vtversion(z) < 0)
165 056fe1ba 2003-11-23 devnull return -1;
166 056fe1ba 2003-11-23 devnull if(vthello(z) < 0)
167 056fe1ba 2003-11-23 devnull return -1;
168 056fe1ba 2003-11-23 devnull return 0;
