nineproto.c revision 1.3 1 /* $NetBSD: nineproto.c,v 1.3 2007/05/05 15:49:51 pooka Exp $ */
2
3 /*
4 * Copyright (c) 2007 Antti Kantee. All Rights Reserved.
5 *
6 * Redistribution and use in source and binary forms, with or without
7 * modification, are permitted provided that the following conditions
8 * are met:
9 * 1. Redistributions of source code must retain the above copyright
10 * notice, this list of conditions and the following disclaimer.
11 * 2. Redistributions in binary form must reproduce the above copyright
12 * notice, this list of conditions and the following disclaimer in the
13 * documentation and/or other materials provided with the distribution.
14 *
15 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS
16 * OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
17 * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
18 * DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
19 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
20 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
21 * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
22 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
23 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
24 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
25 * SUCH DAMAGE.
26 */
27
28 #include <sys/cdefs.h>
29 #ifndef lint
30 __RCSID("$NetBSD: nineproto.c,v 1.3 2007/05/05 15:49:51 pooka Exp $");
31 #endif /* !lint */
32
33 #include <sys/types.h>
34
35 #include <errno.h>
36 #include <grp.h>
37 #include <pwd.h>
38 #include <puffs.h>
39 #include <stdio.h>
40 #include <stdlib.h>
41
42 #include "ninepuffs.h"
43 #include "nineproto.h"
44
45 int
46 proto_getqid(struct puffs_framebuf *pb, struct qid9p *qid)
47 {
48
49 if (puffs_framebuf_tellsize(pb) - puffs_framebuf_telloff(pb) < 13)
50 return ENOBUFS;
51
52 p9pbuf_get_1(pb, &qid->qidtype);
53 p9pbuf_get_4(pb, &qid->qidvers);
54 p9pbuf_get_8(pb, &qid->qidpath);
55
56 return 0;
57 }
58
59 static uid_t
60 ustr2uid(char *uid)
61 {
62 struct passwd *pw;
63
64 pw = getpwnam(uid);
65 if (pw == NULL)
66 return 0; /* XXXXX */
67
68 return pw->pw_uid;
69 }
70
71 static gid_t
72 gstr2gid(char *gid)
73 {
74 struct group *grr;
75
76 grr = getgrnam(gid);
77 if (grr == NULL)
78 return 0; /* more XXXX */
79
80 return grr->gr_gid;
81 }
82
83 static const char *
84 uid2ustr(uid_t uid)
85 {
86 struct passwd *pw;
87
88 pw = getpwuid(uid);
89 if (pw == NULL)
90 return "root"; /* XXXXX */
91
92 return pw->pw_name;
93 }
94
95 static const char *
96 gid2gstr(gid_t gid)
97 {
98 struct group *grr;
99
100 grr = getgrgid(gid);
101 if (grr == NULL)
102 return "wheel"; /* XXXXXX */
103
104 return grr->gr_name;
105 }
106
107 #define GETFIELD(a,b,unitsize) \
108 do { \
109 if (size < unitsize) return EPROTO; \
110 if ((rv = (a(pb, b)))) return rv; \
111 size -= unitsize; \
112 } while (/*CONSTCOND*/0)
113 #define GETSTR(val,strsize) \
114 do { \
115 if ((rv = p9pbuf_get_str(pb, val, strsize))) return rv; \
116 if (*strsize > size) return EPROTO; \
117 size -= *strsize; \
118 } while (/*CONSTCOND*/0)
119 int
120 proto_getstat(struct puffs_framebuf *pb, struct vattr *vap,
121 char **name, uint16_t *rs)
122 {
123 char *uid, *gid;
124 struct qid9p qid;
125 uint64_t flen;
126 uint32_t rdev, mode, atime, mtime;
127 uint16_t size, v16;
128 int rv;
129
130 /* check size */
131 if ((rv = p9pbuf_get_2(pb, &size)))
132 return rv;
133 if (puffs_framebuf_tellsize(pb) - puffs_framebuf_telloff(pb) < size)
134 return ENOBUFS;
135
136 if (rs)
137 *rs = size+2; /* compensate for size field itself */
138
139 GETFIELD(p9pbuf_get_2, &v16, 2);
140 if (v16)
141 printf("%d\n", v16);
142 GETFIELD(p9pbuf_get_4, &rdev, 4);
143 GETFIELD(proto_getqid, &qid, 13);
144 GETFIELD(p9pbuf_get_4, &mode, 4);
145 GETFIELD(p9pbuf_get_4, &atime, 4);
146 GETFIELD(p9pbuf_get_4, &mtime, 4);
147 GETFIELD(p9pbuf_get_8, &flen, 8);
148 GETSTR(name, &v16);
149 GETSTR(&uid, &v16);
150 GETSTR(&gid, &v16);
151
152 if (rdev)
153 printf("%d\n", rdev);
154 vap->va_rdev = rdev;
155 vap->va_mode = mode & 0777; /* may contain other uninteresting bits */
156 vap->va_atime.tv_sec = atime;
157 vap->va_mtime.tv_sec = mtime;
158 vap->va_ctime.tv_sec = mtime;
159 vap->va_atime.tv_nsec=vap->va_mtime.tv_nsec=vap->va_ctime.tv_nsec = 0;
160 vap->va_birthtime.tv_sec = vap->va_birthtime.tv_nsec = 0;
161 vap->va_size = vap->va_bytes = flen;
162 vap->va_uid = ustr2uid(uid);
163 vap->va_gid = gstr2gid(gid);
164 free(uid);
165 free(gid);
166 qid2vattr(vap, &qid);
167
168 /* some defaults */
169 if (vap->va_type == VDIR)
170 vap->va_nlink = 1906;
171 else
172 vap->va_nlink = 1;
173 vap->va_blocksize = 512;
174 vap->va_flags = vap->va_vaflags = 0;
175 vap->va_filerev = PUFFS_VNOVAL;
176
177 /* muid, not used */
178 GETSTR(NULL, &v16);
179
180 return 0;
181 }
182
183 int
184 proto_cc_dupfid(struct puffs_cc *pcc, p9pfid_t oldfid, p9pfid_t newfid)
185 {
186 struct puffs9p *p9p = puffs_cc_getspecific(pcc);
187 struct puffs_framebuf *pb;
188 p9ptag_t tag = NEXTTAG(p9p);
189 uint16_t qids;
190 int rv, error = 0;
191
192 pb = p9pbuf_makeout();
193 p9pbuf_put_1(pb, P9PROTO_T_WALK);
194 p9pbuf_put_2(pb, tag);
195 p9pbuf_put_4(pb, oldfid);
196 p9pbuf_put_4(pb, newfid);
197 p9pbuf_put_2(pb, 0);
198 puffs_framebuf_enqueue_cc(pcc, pb);
199
200 rv = proto_expect_walk_nqids(pb, &qids);
201 if (rv)
202 error = rv;
203 if (qids != 0)
204 error = EPROTO;
205
206 puffs_framebuf_destroy(pb);
207 return error;
208 }
209
210 int
211 proto_cc_clunkfid(struct puffs_cc *pcc, p9pfid_t fid, int waitforit)
212 {
213 struct puffs9p *p9p = puffs_cc_getspecific(pcc);
214 struct puffs_framebuf *pb;
215 p9ptag_t tag = NEXTTAG(p9p);
216 int error = 0;
217
218 pb = p9pbuf_makeout();
219 p9pbuf_put_1(pb, P9PROTO_T_CLUNK);
220 p9pbuf_put_2(pb, tag);
221 p9pbuf_put_4(pb, fid);
222
223 if (waitforit) {
224 puffs_framebuf_enqueue_cc(pcc, pb);
225 if (p9pbuf_get_type(pb) != P9PROTO_R_CLUNK)
226 error = EPROTO;
227 puffs_framebuf_destroy(pb);
228 } else {
229 puffs_framebuf_enqueue_justsend(puffs_cc_getusermount(pcc),
230 pb, 1);
231 }
232
233 return error;
234 }
235
236 /*
237 * walk a new fid, then open it
238 */
239 int
240 proto_cc_open(struct puffs_cc *pcc, p9pfid_t fid, p9pfid_t newfid, int mode)
241 {
242 struct puffs9p *p9p = puffs_cc_getspecific(pcc);
243 struct puffs_framebuf *pb;
244 p9ptag_t tag = NEXTTAG(p9p);
245 int error;
246
247 error = proto_cc_dupfid(pcc, fid, newfid);
248 if (error)
249 return error;
250
251 pb = p9pbuf_makeout();
252 p9pbuf_put_1(pb, P9PROTO_T_OPEN);
253 p9pbuf_put_2(pb, tag);
254 p9pbuf_put_4(pb, newfid);
255 p9pbuf_put_1(pb, mode);
256 puffs_framebuf_enqueue_cc(pcc, pb);
257 if (p9pbuf_get_type(pb) != P9PROTO_R_OPEN)
258 error = EPROTO;
259
260 puffs_framebuf_destroy(pb);
261 return error;
262 }
263
264 void
265 proto_make_stat(struct puffs_framebuf *pb, const struct vattr *vap,
266 const char *filename)
267 {
268 struct vattr fakeva;
269 uint32_t mode, atime, mtime;
270 uint64_t flen;
271 const char *owner, *group;
272 int startoff, curoff;
273
274 if (vap == NULL) {
275 puffs_vattr_null(&fakeva);
276 vap = &fakeva;
277 }
278
279 startoff = puffs_framebuf_telloff(pb);
280 puffs_framebuf_seekset(pb, startoff + 2+2); /* stat[n] incl. stat[2] */
281
282 if (vap->va_mode != (mode_t)PUFFS_VNOVAL)
283 mode = vap->va_mode;
284 else
285 mode = P9PROTO_STAT_NOVAL4;
286 if (vap->va_atime.tv_sec != (time_t)PUFFS_VNOVAL)
287 atime = vap->va_atime.tv_sec;
288 else
289 atime = P9PROTO_STAT_NOVAL4;
290 if (vap->va_mtime.tv_sec != (time_t)PUFFS_VNOVAL)
291 mtime = vap->va_mtime.tv_sec;
292 else
293 mtime = P9PROTO_STAT_NOVAL4;
294 if (vap->va_size != (u_quad_t)PUFFS_VNOVAL)
295 flen = vap->va_size;
296 else
297 flen = P9PROTO_STAT_NOVAL8;
298 if (vap->va_uid != (uid_t)PUFFS_VNOVAL)
299 owner = uid2ustr(vap->va_uid);
300 else
301 owner = "";
302 if (vap->va_gid != (gid_t)PUFFS_VNOVAL)
303 group = gid2gstr(vap->va_gid);
304 else
305 group = "";
306
307 p9pbuf_put_2(pb, P9PROTO_STAT_NOVAL2); /* kernel type */
308 p9pbuf_put_4(pb, P9PROTO_STAT_NOVAL4); /* dev */
309 p9pbuf_put_1(pb, P9PROTO_STAT_NOVAL1); /* type */
310 p9pbuf_put_4(pb, P9PROTO_STAT_NOVAL4); /* version */
311 p9pbuf_put_8(pb, P9PROTO_STAT_NOVAL8); /* path */
312 p9pbuf_put_4(pb, mode);
313 p9pbuf_put_4(pb, atime);
314 p9pbuf_put_4(pb, mtime);
315 p9pbuf_put_8(pb, flen);
316 p9pbuf_put_str(pb, filename ? filename : "");
317 p9pbuf_put_str(pb, owner);
318 p9pbuf_put_str(pb, group);
319 p9pbuf_put_str(pb, ""); /* muid */
320
321 curoff = puffs_framebuf_telloff(pb);
322 puffs_framebuf_seekset(pb, startoff);
323 p9pbuf_put_2(pb, curoff-(startoff+2)); /* stat[n] size */
324 p9pbuf_put_2(pb, curoff-(startoff+4)); /* size[2] stat */
325
326 puffs_framebuf_seekset(pb, curoff);
327 }
328
329 int
330 proto_expect_walk_nqids(struct puffs_framebuf *pb, uint16_t *nqids)
331 {
332
333 if (p9pbuf_get_type(pb) != P9PROTO_R_WALK)
334 return EPROTO;
335 return p9pbuf_get_2(pb, nqids);
336 }
337
338 int
339 proto_expect_qid(struct puffs_framebuf *pb, uint8_t op, struct qid9p *qid)
340 {
341
342 if (p9pbuf_get_type(pb) != op)
343 return EPROTO;
344 return proto_getqid(pb, qid);
345 }
346
347 int
348 proto_expect_stat(struct puffs_framebuf *pb, struct vattr *va)
349 {
350 uint16_t dummy;
351 int rv;
352
353 if (p9pbuf_get_type(pb) != P9PROTO_R_STAT)
354 return EPROTO;
355 if ((rv = p9pbuf_get_2(pb, &dummy)))
356 return rv;
357 return proto_getstat(pb, va, NULL, NULL);
358 }
359