Google Git
Sign in
plan9port / plan9 / 58acc3d4427314035b00cb29e240570e87634f64 / . / src / libventi / fcall.c
blob: 9f721b426367cb35acb5ccd703361c9dbfd1a338 [file] [log] [blame]
rsc056fe1b2003-11-23 18:19:58 +0000[diff] [blame]1#include <u.h>
2#include <libc.h>
3#include <venti.h>
4
5Packet*
6vtfcallpack(VtFcall *f)
7{
8 uchar buf[4];
9 Packet *p;
10
11 p = packetalloc();
12
rsc7643b262005-07-13 10:52:39 +0000[diff] [blame]13 buf[0] = f->msgtype;
rsc056fe1b2003-11-23 18:19:58 +0000[diff] [blame]14 buf[1] = f->tag;
15 packetappend(p, buf, 2);
16
rsc7643b262005-07-13 10:52:39 +0000[diff] [blame]17 switch(f->msgtype){
rsc056fe1b2003-11-23 18:19:58 +0000[diff] [blame]18 default:
rsc7643b262005-07-13 10:52:39 +0000[diff] [blame]19 werrstr("vtfcallpack: unknown packet type %d", f->msgtype);
rsc056fe1b2003-11-23 18:19:58 +0000[diff] [blame]20 goto Err;
21
22 case VtRerror:
23 if(vtputstring(p, f->error) < 0)
24 goto Err;
25 break;
26
27 case VtTping:
28 break;
29
30 case VtRping:
31 break;
32
33 case VtThello:
34 if(vtputstring(p, f->version) < 0
35 || vtputstring(p, f->uid) < 0)
36 goto Err;
37 buf[0] = f->strength;
38 buf[1] = f->ncrypto;
39 packetappend(p, buf, 2);
40 packetappend(p, f->crypto, f->ncrypto);
41 buf[0] = f->ncodec;
42 packetappend(p, buf, 1);
43 packetappend(p, f->codec, f->ncodec);
44 break;
45
46 case VtRhello:
47 if(vtputstring(p, f->sid) < 0)
48 goto Err;
49 buf[0] = f->rcrypto;
50 buf[1] = f->rcodec;
51 packetappend(p, buf, 2);
52 break;
53
54 case VtTgoodbye:
55 break;
56
57 case VtTread:
58 packetappend(p, f->score, VtScoreSize);
rsc7643b262005-07-13 10:52:39 +0000[diff] [blame]59 buf[0] = vttodisktype(f->blocktype);
rsc056fe1b2003-11-23 18:19:58 +0000[diff] [blame]60 if(~buf[0] == 0)
61 goto Err;
62 buf[1] = 0;
63 buf[2] = f->count >> 8;
64 buf[3] = f->count;
65 packetappend(p, buf, 4);
66 break;
67
68 case VtRread:
69 packetconcat(p, f->data);
70 break;
71
72 case VtTwrite:
rsc7643b262005-07-13 10:52:39 +0000[diff] [blame]73 buf[0] = vttodisktype(f->blocktype);
rsc056fe1b2003-11-23 18:19:58 +0000[diff] [blame]74 if(~buf[0] == 0)
75 goto Err;
76 buf[1] = 0;
77 buf[2] = 0;
78 buf[3] = 0;
79 packetappend(p, buf, 4);
80 packetconcat(p, f->data);
81 break;
82
83 case VtRwrite:
84 packetappend(p, f->score, VtScoreSize);
85 break;
86
87 case VtTsync:
88 break;
89
90 case VtRsync:
91 break;
92 }
93
94 return p;
95
96Err:
97 packetfree(p);
98 return nil;
99}
100
101int
102vtfcallunpack(VtFcall *f, Packet *p)
103{
104 uchar buf[4];
105
106 memset(f, 0, sizeof *f);
107
108 if(packetconsume(p, buf, 2) < 0)
109 return -1;
110
rsc7643b262005-07-13 10:52:39 +0000[diff] [blame]111 f->msgtype = buf[0];
rsc056fe1b2003-11-23 18:19:58 +0000[diff] [blame]112 f->tag = buf[1];
113
rsc7643b262005-07-13 10:52:39 +0000[diff] [blame]114 switch(f->msgtype){
rsc056fe1b2003-11-23 18:19:58 +0000[diff] [blame]115 default:
rsc7643b262005-07-13 10:52:39 +0000[diff] [blame]116 werrstr("vtfcallunpack: unknown bad packet type %d", f->msgtype);
rsc056fe1b2003-11-23 18:19:58 +0000[diff] [blame]117 vtfcallclear(f);
118 return -1;
119
120 case VtRerror:
121 if(vtgetstring(p, &f->error) < 0)
122 goto Err;
123 break;
124
125 case VtTping:
126 break;
127
128 case VtRping:
129 break;
130
131 case VtThello:
132 if(vtgetstring(p, &f->version) < 0
133 || vtgetstring(p, &f->uid) < 0
134 || packetconsume(p, buf, 2) < 0)
135 goto Err;
136 f->strength = buf[0];
137 f->ncrypto = buf[1];
138 if(f->ncrypto){
139 f->crypto = vtmalloc(f->ncrypto);
140 if(packetconsume(p, buf, f->ncrypto) < 0)
141 goto Err;
142 }
143 if(packetconsume(p, buf, 1) < 0)
144 goto Err;
145 f->ncodec = buf[0];
146 if(f->ncodec){
147 f->codec = vtmalloc(f->ncodec);
148 if(packetconsume(p, buf, f->ncodec) < 0)
149 goto Err;
150 }
151 break;
152
153 case VtRhello:
154 if(vtgetstring(p, &f->sid) < 0
155 || packetconsume(p, buf, 2) < 0)
156 goto Err;
157 f->rcrypto = buf[0];
158 f->rcodec = buf[1];
159 break;
160
161 case VtTgoodbye:
162 break;
163
164 case VtTread:
165 if(packetconsume(p, f->score, VtScoreSize) < 0
166 || packetconsume(p, buf, 4) < 0)
167 goto Err;
rsc7643b262005-07-13 10:52:39 +0000[diff] [blame]168 f->blocktype = vtfromdisktype(buf[0]);
169 if(~f->blocktype == 0)
rsc056fe1b2003-11-23 18:19:58 +0000[diff] [blame]170 goto Err;
171 f->count = (buf[2] << 8) | buf[3];
172 break;
173
174 case VtRread:
175 f->data = packetalloc();
176 packetconcat(f->data, p);
177 break;
178
179 case VtTwrite:
180 if(packetconsume(p, buf, 4) < 0)
181 goto Err;
rsc7643b262005-07-13 10:52:39 +0000[diff] [blame]182 f->blocktype = vtfromdisktype(buf[0]);
183 if(~f->blocktype == 0)
rsc056fe1b2003-11-23 18:19:58 +0000[diff] [blame]184 goto Err;
185 f->data = packetalloc();
186 packetconcat(f->data, p);
187 break;
188
189 case VtRwrite:
190 if(packetconsume(p, f->score, VtScoreSize) < 0)
191 goto Err;
192 break;
193
194 case VtTsync:
195 break;
196
197 case VtRsync:
198 break;
199 }
200
201 if(packetsize(p) != 0)
202 goto Err;
203
204 return 0;
205
206Err:
207 werrstr("bad packet");
208 return -1;
209}
210
211void
212vtfcallclear(VtFcall *f)
213{
214 vtfree(f->error);
215 f->error = nil;
216 vtfree(f->uid);
217 f->uid = nil;
218 vtfree(f->sid);
219 f->sid = nil;
220 vtfree(f->version);
221 f->version = nil;
222 vtfree(f->crypto);
223 f->crypto = nil;
224 vtfree(f->codec);
225 f->codec = nil;
226 vtfree(f->auth);
227 f->auth = nil;
228 packetfree(f->data);
rsc7cb74892004-06-17 18:51:50 +0000[diff] [blame]229 f->data = nil;
rsc056fe1b2003-11-23 18:19:58 +0000[diff] [blame]230}