xref: /qemu/tests/qtest/libqos/virtio-9p-client.c (revision a9e0c9c0f14e19d23443ac24c8080b4708d2eab8)
1 /*
2  * 9P network client for VirtIO 9P test cases (based on QTest)
3  *
4  * Copyright (c) 2014 SUSE LINUX Products GmbH
5  *
6  * This work is licensed under the terms of the GNU GPL, version 2 or later.
7  * See the COPYING file in the top-level directory.
8  */
9 
10 /*
11  * Not so fast! You might want to read the 9p developer docs first:
12  * https://wiki.qemu.org/Documentation/9p
13  */
14 
15 #include "qemu/osdep.h"
16 #include "virtio-9p-client.h"
17 
18 #define QVIRTIO_9P_TIMEOUT_US (10 * 1000 * 1000)
19 static QGuestAllocator *alloc;
20 
v9fs_set_allocator(QGuestAllocator * t_alloc)21 void v9fs_set_allocator(QGuestAllocator *t_alloc)
22 {
23     alloc = t_alloc;
24 }
25 
26 /*
27  * Used to auto generate new fids. Start with arbitrary high value to avoid
28  * collision with hard coded fids in basic test code.
29  */
30 static uint32_t fid_generator = 1000;
31 
genfid(void)32 static uint32_t genfid(void)
33 {
34     return fid_generator++;
35 }
36 
37 /**
38  * Splits the @a in string by @a delim into individual (non empty) strings
39  * and outputs them to @a out. The output array @a out is NULL terminated.
40  *
41  * Output array @a out must be freed by calling split_free().
42  *
43  * @returns number of individual elements in output array @a out (without the
44  *          final NULL terminating element)
45  */
split(const char * in,const char * delim,char *** out)46 static int split(const char *in, const char *delim, char ***out)
47 {
48     int n = 0, i = 0;
49     char *tmp, *p;
50 
51     tmp = g_strdup(in);
52     for (p = strtok(tmp, delim); p != NULL; p = strtok(NULL, delim)) {
53         if (strlen(p) > 0) {
54             ++n;
55         }
56     }
57     g_free(tmp);
58 
59     *out = g_new0(char *, n + 1); /* last element NULL delimiter */
60 
61     tmp = g_strdup(in);
62     for (p = strtok(tmp, delim); p != NULL; p = strtok(NULL, delim)) {
63         if (strlen(p) > 0) {
64             (*out)[i++] = g_strdup(p);
65         }
66     }
67     g_free(tmp);
68 
69     return n;
70 }
71 
split_free(char *** out)72 static void split_free(char ***out)
73 {
74     int i;
75     if (!*out) {
76         return;
77     }
78     for (i = 0; (*out)[i]; ++i) {
79         g_free((*out)[i]);
80     }
81     g_free(*out);
82     *out = NULL;
83 }
84 
v9fs_memwrite(P9Req * req,const void * addr,size_t len)85 void v9fs_memwrite(P9Req *req, const void *addr, size_t len)
86 {
87     qtest_memwrite(req->qts, req->t_msg + req->t_off, addr, len);
88     req->t_off += len;
89 }
90 
v9fs_memskip(P9Req * req,size_t len)91 void v9fs_memskip(P9Req *req, size_t len)
92 {
93     req->r_off += len;
94 }
95 
v9fs_memread(P9Req * req,void * addr,size_t len)96 void v9fs_memread(P9Req *req, void *addr, size_t len)
97 {
98     qtest_memread(req->qts, req->r_msg + req->r_off, addr, len);
99     req->r_off += len;
100 }
101 
v9fs_uint8_read(P9Req * req,uint8_t * val)102 void v9fs_uint8_read(P9Req *req, uint8_t *val)
103 {
104     v9fs_memread(req, val, 1);
105 }
106 
v9fs_uint16_write(P9Req * req,uint16_t val)107 void v9fs_uint16_write(P9Req *req, uint16_t val)
108 {
109     uint16_t le_val = cpu_to_le16(val);
110 
111     v9fs_memwrite(req, &le_val, 2);
112 }
113 
v9fs_uint16_read(P9Req * req,uint16_t * val)114 void v9fs_uint16_read(P9Req *req, uint16_t *val)
115 {
116     v9fs_memread(req, val, 2);
117     le16_to_cpus(val);
118 }
119 
v9fs_uint32_write(P9Req * req,uint32_t val)120 void v9fs_uint32_write(P9Req *req, uint32_t val)
121 {
122     uint32_t le_val = cpu_to_le32(val);
123 
124     v9fs_memwrite(req, &le_val, 4);
125 }
126 
v9fs_uint64_write(P9Req * req,uint64_t val)127 void v9fs_uint64_write(P9Req *req, uint64_t val)
128 {
129     uint64_t le_val = cpu_to_le64(val);
130 
131     v9fs_memwrite(req, &le_val, 8);
132 }
133 
v9fs_uint32_read(P9Req * req,uint32_t * val)134 void v9fs_uint32_read(P9Req *req, uint32_t *val)
135 {
136     v9fs_memread(req, val, 4);
137     le32_to_cpus(val);
138 }
139 
v9fs_uint64_read(P9Req * req,uint64_t * val)140 void v9fs_uint64_read(P9Req *req, uint64_t *val)
141 {
142     v9fs_memread(req, val, 8);
143     le64_to_cpus(val);
144 }
145 
146 /* len[2] string[len] */
v9fs_string_size(const char * string)147 uint16_t v9fs_string_size(const char *string)
148 {
149     size_t len = strlen(string);
150 
151     g_assert_cmpint(len, <=, UINT16_MAX - 2);
152 
153     return 2 + len;
154 }
155 
v9fs_string_write(P9Req * req,const char * string)156 void v9fs_string_write(P9Req *req, const char *string)
157 {
158     int len = strlen(string);
159 
160     g_assert_cmpint(len, <=, UINT16_MAX);
161 
162     v9fs_uint16_write(req, (uint16_t) len);
163     v9fs_memwrite(req, string, len);
164 }
165 
v9fs_string_read(P9Req * req,uint16_t * len,char ** string)166 void v9fs_string_read(P9Req *req, uint16_t *len, char **string)
167 {
168     uint16_t local_len;
169 
170     v9fs_uint16_read(req, &local_len);
171     if (len) {
172         *len = local_len;
173     }
174     if (string) {
175         *string = g_malloc(local_len + 1);
176         v9fs_memread(req, *string, local_len);
177         (*string)[local_len] = 0;
178     } else {
179         v9fs_memskip(req, local_len);
180     }
181 }
182 
183 typedef struct {
184     uint32_t size;
185     uint8_t id;
186     uint16_t tag;
187 } QEMU_PACKED P9Hdr;
188 
v9fs_req_init(QVirtio9P * v9p,uint32_t size,uint8_t id,uint16_t tag)189 P9Req *v9fs_req_init(QVirtio9P *v9p, uint32_t size, uint8_t id,
190                      uint16_t tag)
191 {
192     P9Req *req = g_new0(P9Req, 1);
193     uint32_t total_size = 7; /* 9P header has well-known size of 7 bytes */
194     P9Hdr hdr = {
195         .id = id,
196         .tag = cpu_to_le16(tag)
197     };
198 
199     g_assert_cmpint(total_size, <=, UINT32_MAX - size);
200     total_size += size;
201     hdr.size = cpu_to_le32(total_size);
202 
203     g_assert_cmpint(total_size, <=, P9_MAX_SIZE);
204 
205     req->qts = global_qtest;
206     req->v9p = v9p;
207     req->t_size = total_size;
208     req->t_msg = guest_alloc(alloc, req->t_size);
209     v9fs_memwrite(req, &hdr, 7);
210     req->tag = tag;
211     return req;
212 }
213 
v9fs_req_send(P9Req * req)214 void v9fs_req_send(P9Req *req)
215 {
216     QVirtio9P *v9p = req->v9p;
217 
218     req->r_msg = guest_alloc(alloc, P9_MAX_SIZE);
219     req->free_head = qvirtqueue_add(req->qts, v9p->vq, req->t_msg, req->t_size,
220                                     false, true);
221     qvirtqueue_add(req->qts, v9p->vq, req->r_msg, P9_MAX_SIZE, true, false);
222     qvirtqueue_kick(req->qts, v9p->vdev, v9p->vq, req->free_head);
223     req->t_off = 0;
224 }
225 
rmessage_name(uint8_t id)226 static const char *rmessage_name(uint8_t id)
227 {
228     return
229         id == P9_RLERROR ? "RLERROR" :
230         id == P9_RVERSION ? "RVERSION" :
231         id == P9_RATTACH ? "RATTACH" :
232         id == P9_RWALK ? "RWALK" :
233         id == P9_RLOPEN ? "RLOPEN" :
234         id == P9_RWRITE ? "RWRITE" :
235         id == P9_RMKDIR ? "RMKDIR" :
236         id == P9_RLCREATE ? "RLCREATE" :
237         id == P9_RSYMLINK ? "RSYMLINK" :
238         id == P9_RGETATTR ? "RGETATTR" :
239         id == P9_RLINK ? "RLINK" :
240         id == P9_RUNLINKAT ? "RUNLINKAT" :
241         id == P9_RFLUSH ? "RFLUSH" :
242         id == P9_RREADDIR ? "RREADDIR" :
243         "<unknown>";
244 }
245 
v9fs_req_wait_for_reply(P9Req * req,uint32_t * len)246 void v9fs_req_wait_for_reply(P9Req *req, uint32_t *len)
247 {
248     QVirtio9P *v9p = req->v9p;
249 
250     qvirtio_wait_used_elem(req->qts, v9p->vdev, v9p->vq, req->free_head, len,
251                            QVIRTIO_9P_TIMEOUT_US);
252 }
253 
v9fs_req_recv(P9Req * req,uint8_t id)254 void v9fs_req_recv(P9Req *req, uint8_t id)
255 {
256     P9Hdr hdr;
257 
258     v9fs_memread(req, &hdr, 7);
259     hdr.size = ldl_le_p(&hdr.size);
260     hdr.tag = lduw_le_p(&hdr.tag);
261 
262     g_assert_cmpint(hdr.size, >=, 7);
263     g_assert_cmpint(hdr.size, <=, P9_MAX_SIZE);
264     g_assert_cmpint(hdr.tag, ==, req->tag);
265 
266     if (hdr.id != id) {
267         g_printerr("Received response %d (%s) instead of %d (%s)\n",
268                    hdr.id, rmessage_name(hdr.id), id, rmessage_name(id));
269 
270         if (hdr.id == P9_RLERROR) {
271             uint32_t err;
272             v9fs_uint32_read(req, &err);
273             g_printerr("Rlerror has errno %d (%s)\n", err, strerror(err));
274         }
275     }
276     g_assert_cmpint(hdr.id, ==, id);
277 }
278 
v9fs_req_free(P9Req * req)279 void v9fs_req_free(P9Req *req)
280 {
281     guest_free(alloc, req->t_msg);
282     guest_free(alloc, req->r_msg);
283     g_free(req);
284 }
285 
286 /* size[4] Rlerror tag[2] ecode[4] */
v9fs_rlerror(P9Req * req,uint32_t * err)287 void v9fs_rlerror(P9Req *req, uint32_t *err)
288 {
289     v9fs_req_recv(req, P9_RLERROR);
290     v9fs_uint32_read(req, err);
291     v9fs_req_free(req);
292 }
293 
294 /* size[4] Tversion tag[2] msize[4] version[s] */
v9fs_tversion(TVersionOpt opt)295 TVersionRes v9fs_tversion(TVersionOpt opt)
296 {
297     P9Req *req;
298     uint32_t err;
299     uint32_t body_size = 4;
300     uint16_t string_size;
301     uint16_t server_len;
302     g_autofree char *server_version = NULL;
303 
304     g_assert(opt.client);
305 
306     if (!opt.msize) {
307         opt.msize = P9_MAX_SIZE;
308     }
309 
310     if (!opt.tag) {
311         opt.tag = P9_NOTAG;
312     }
313 
314     if (!opt.version) {
315         opt.version = "9P2000.L";
316     }
317 
318     string_size = v9fs_string_size(opt.version);
319     g_assert_cmpint(body_size, <=, UINT32_MAX - string_size);
320     body_size += string_size;
321     req = v9fs_req_init(opt.client, body_size, P9_TVERSION, opt.tag);
322 
323     v9fs_uint32_write(req, opt.msize);
324     v9fs_string_write(req, opt.version);
325     v9fs_req_send(req);
326 
327     if (!opt.requestOnly) {
328         v9fs_req_wait_for_reply(req, NULL);
329         if (opt.expectErr) {
330             v9fs_rlerror(req, &err);
331             g_assert_cmpint(err, ==, opt.expectErr);
332         } else {
333             v9fs_rversion(req, &server_len, &server_version);
334             g_assert_cmpmem(server_version, server_len,
335                             opt.version, strlen(opt.version));
336         }
337         req = NULL; /* request was freed */
338     }
339 
340     return (TVersionRes) {
341         .req = req,
342     };
343 }
344 
345 /* size[4] Rversion tag[2] msize[4] version[s] */
v9fs_rversion(P9Req * req,uint16_t * len,char ** version)346 void v9fs_rversion(P9Req *req, uint16_t *len, char **version)
347 {
348     uint32_t msize;
349 
350     v9fs_req_recv(req, P9_RVERSION);
351     v9fs_uint32_read(req, &msize);
352 
353     g_assert_cmpint(msize, ==, P9_MAX_SIZE);
354 
355     if (len || version) {
356         v9fs_string_read(req, len, version);
357     }
358 
359     v9fs_req_free(req);
360 }
361 
362 /* size[4] Tattach tag[2] fid[4] afid[4] uname[s] aname[s] n_uname[4] */
v9fs_tattach(TAttachOpt opt)363 TAttachRes v9fs_tattach(TAttachOpt opt)
364 {
365     uint32_t err;
366     const char *uname = ""; /* ignored by QEMU */
367     const char *aname = ""; /* ignored by QEMU */
368 
369     g_assert(opt.client);
370     /* expecting either Rattach or Rlerror, but obviously not both */
371     g_assert(!opt.expectErr || !opt.rattach.qid);
372 
373     if (!opt.requestOnly) {
374         v9fs_tversion((TVersionOpt) { .client = opt.client });
375     }
376 
377     if (!opt.n_uname) {
378         opt.n_uname = getuid();
379     }
380 
381     P9Req *req = v9fs_req_init(opt.client, 4 + 4 + 2 + 2 + 4, P9_TATTACH,
382                                opt.tag);
383 
384     v9fs_uint32_write(req, opt.fid);
385     v9fs_uint32_write(req, P9_NOFID);
386     v9fs_string_write(req, uname);
387     v9fs_string_write(req, aname);
388     v9fs_uint32_write(req, opt.n_uname);
389     v9fs_req_send(req);
390 
391     if (!opt.requestOnly) {
392         v9fs_req_wait_for_reply(req, NULL);
393         if (opt.expectErr) {
394             v9fs_rlerror(req, &err);
395             g_assert_cmpint(err, ==, opt.expectErr);
396         } else {
397             v9fs_rattach(req, opt.rattach.qid);
398         }
399         req = NULL; /* request was freed */
400     }
401 
402     return (TAttachRes) {
403         .req = req,
404     };
405 }
406 
407 /* size[4] Rattach tag[2] qid[13] */
v9fs_rattach(P9Req * req,v9fs_qid * qid)408 void v9fs_rattach(P9Req *req, v9fs_qid *qid)
409 {
410     v9fs_req_recv(req, P9_RATTACH);
411     if (qid) {
412         v9fs_memread(req, qid, 13);
413     }
414     v9fs_req_free(req);
415 }
416 
417 /* size[4] Twalk tag[2] fid[4] newfid[4] nwname[2] nwname*(wname[s]) */
v9fs_twalk(TWalkOpt opt)418 TWalkRes v9fs_twalk(TWalkOpt opt)
419 {
420     P9Req *req;
421     int i;
422     uint32_t body_size = 4 + 4 + 2;
423     uint32_t err;
424     char **wnames = NULL;
425 
426     g_assert(opt.client);
427     /* expecting either high- or low-level path, both not both */
428     g_assert(!opt.path || !(opt.nwname || opt.wnames));
429     /* expecting either Rwalk or Rlerror, but obviously not both */
430     g_assert(!opt.expectErr || !(opt.rwalk.nwqid || opt.rwalk.wqid));
431 
432     if (!opt.newfid) {
433         opt.newfid = genfid();
434     }
435 
436     if (opt.path) {
437         opt.nwname = split(opt.path, "/", &wnames);
438         opt.wnames = wnames;
439     }
440 
441     for (i = 0; i < opt.nwname; i++) {
442         uint16_t wname_size = v9fs_string_size(opt.wnames[i]);
443 
444         g_assert_cmpint(body_size, <=, UINT32_MAX - wname_size);
445         body_size += wname_size;
446     }
447     req = v9fs_req_init(opt.client, body_size, P9_TWALK, opt.tag);
448     v9fs_uint32_write(req, opt.fid);
449     v9fs_uint32_write(req, opt.newfid);
450     v9fs_uint16_write(req, opt.nwname);
451     for (i = 0; i < opt.nwname; i++) {
452         v9fs_string_write(req, opt.wnames[i]);
453     }
454     v9fs_req_send(req);
455 
456     if (!opt.requestOnly) {
457         v9fs_req_wait_for_reply(req, NULL);
458         if (opt.expectErr) {
459             v9fs_rlerror(req, &err);
460             g_assert_cmpint(err, ==, opt.expectErr);
461         } else {
462             v9fs_rwalk(req, opt.rwalk.nwqid, opt.rwalk.wqid);
463         }
464         req = NULL; /* request was freed */
465     }
466 
467     split_free(&wnames);
468 
469     return (TWalkRes) {
470         .newfid = opt.newfid,
471         .req = req,
472     };
473 }
474 
475 /* size[4] Rwalk tag[2] nwqid[2] nwqid*(wqid[13]) */
v9fs_rwalk(P9Req * req,uint16_t * nwqid,v9fs_qid ** wqid)476 void v9fs_rwalk(P9Req *req, uint16_t *nwqid, v9fs_qid **wqid)
477 {
478     uint16_t local_nwqid;
479 
480     v9fs_req_recv(req, P9_RWALK);
481     v9fs_uint16_read(req, &local_nwqid);
482     if (nwqid) {
483         *nwqid = local_nwqid;
484     }
485     if (wqid) {
486         *wqid = g_malloc(local_nwqid * 13);
487         v9fs_memread(req, *wqid, local_nwqid * 13);
488     }
489     v9fs_req_free(req);
490 }
491 
492 /* size[4] Tgetattr tag[2] fid[4] request_mask[8] */
v9fs_tgetattr(TGetAttrOpt opt)493 TGetAttrRes v9fs_tgetattr(TGetAttrOpt opt)
494 {
495     P9Req *req;
496     uint32_t err;
497 
498     g_assert(opt.client);
499     /* expecting either Rgetattr or Rlerror, but obviously not both */
500     g_assert(!opt.expectErr || !opt.rgetattr.attr);
501 
502     if (!opt.request_mask) {
503         opt.request_mask = P9_GETATTR_ALL;
504     }
505 
506     req = v9fs_req_init(opt.client, 4 + 8, P9_TGETATTR, opt.tag);
507     v9fs_uint32_write(req, opt.fid);
508     v9fs_uint64_write(req, opt.request_mask);
509     v9fs_req_send(req);
510 
511     if (!opt.requestOnly) {
512         v9fs_req_wait_for_reply(req, NULL);
513         if (opt.expectErr) {
514             v9fs_rlerror(req, &err);
515             g_assert_cmpint(err, ==, opt.expectErr);
516         } else {
517             v9fs_rgetattr(req, opt.rgetattr.attr);
518         }
519         req = NULL; /* request was freed */
520     }
521 
522     return (TGetAttrRes) { .req = req };
523 }
524 
525 /*
526  * size[4] Rgetattr tag[2] valid[8] qid[13] mode[4] uid[4] gid[4] nlink[8]
527  *                  rdev[8] size[8] blksize[8] blocks[8]
528  *                  atime_sec[8] atime_nsec[8] mtime_sec[8] mtime_nsec[8]
529  *                  ctime_sec[8] ctime_nsec[8] btime_sec[8] btime_nsec[8]
530  *                  gen[8] data_version[8]
531  */
v9fs_rgetattr(P9Req * req,v9fs_attr * attr)532 void v9fs_rgetattr(P9Req *req, v9fs_attr *attr)
533 {
534     v9fs_req_recv(req, P9_RGETATTR);
535 
536     v9fs_uint64_read(req, &attr->valid);
537     v9fs_memread(req, &attr->qid, 13);
538     v9fs_uint32_read(req, &attr->mode);
539     v9fs_uint32_read(req, &attr->uid);
540     v9fs_uint32_read(req, &attr->gid);
541     v9fs_uint64_read(req, &attr->nlink);
542     v9fs_uint64_read(req, &attr->rdev);
543     v9fs_uint64_read(req, &attr->size);
544     v9fs_uint64_read(req, &attr->blksize);
545     v9fs_uint64_read(req, &attr->blocks);
546     v9fs_uint64_read(req, &attr->atime_sec);
547     v9fs_uint64_read(req, &attr->atime_nsec);
548     v9fs_uint64_read(req, &attr->mtime_sec);
549     v9fs_uint64_read(req, &attr->mtime_nsec);
550     v9fs_uint64_read(req, &attr->ctime_sec);
551     v9fs_uint64_read(req, &attr->ctime_nsec);
552     v9fs_uint64_read(req, &attr->btime_sec);
553     v9fs_uint64_read(req, &attr->btime_nsec);
554     v9fs_uint64_read(req, &attr->gen);
555     v9fs_uint64_read(req, &attr->data_version);
556 
557     v9fs_req_free(req);
558 }
559 
560 /*
561  * size[4] Tsetattr tag[2] fid[4] valid[4] mode[4] uid[4] gid[4] size[8]
562  *                  atime_sec[8] atime_nsec[8] mtime_sec[8] mtime_nsec[8]
563  */
v9fs_tsetattr(TSetAttrOpt opt)564 TSetAttrRes v9fs_tsetattr(TSetAttrOpt opt)
565 {
566     P9Req *req;
567     uint32_t err;
568 
569     g_assert(opt.client);
570 
571     req = v9fs_req_init(
572         opt.client, 4/*fid*/ + 4/*valid*/ + 4/*mode*/ + 4/*uid*/ + 4/*gid*/ +
573         8/*size*/ + 8/*atime_sec*/ + 8/*atime_nsec*/ + 8/*mtime_sec*/ +
574         8/*mtime_nsec*/, P9_TSETATTR, opt.tag
575     );
576     v9fs_uint32_write(req, opt.fid);
577     v9fs_uint32_write(req, (uint32_t) opt.attr.valid);
578     v9fs_uint32_write(req, opt.attr.mode);
579     v9fs_uint32_write(req, opt.attr.uid);
580     v9fs_uint32_write(req, opt.attr.gid);
581     v9fs_uint64_write(req, opt.attr.size);
582     v9fs_uint64_write(req, opt.attr.atime_sec);
583     v9fs_uint64_write(req, opt.attr.atime_nsec);
584     v9fs_uint64_write(req, opt.attr.mtime_sec);
585     v9fs_uint64_write(req, opt.attr.mtime_nsec);
586     v9fs_req_send(req);
587 
588     if (!opt.requestOnly) {
589         v9fs_req_wait_for_reply(req, NULL);
590         if (opt.expectErr) {
591             v9fs_rlerror(req, &err);
592             g_assert_cmpint(err, ==, opt.expectErr);
593         } else {
594             v9fs_rsetattr(req);
595         }
596         req = NULL; /* request was freed */
597     }
598 
599     return (TSetAttrRes) { .req = req };
600 }
601 
602 /* size[4] Rsetattr tag[2] */
v9fs_rsetattr(P9Req * req)603 void v9fs_rsetattr(P9Req *req)
604 {
605     v9fs_req_recv(req, P9_RSETATTR);
606     v9fs_req_free(req);
607 }
608 
609 /* size[4] Treaddir tag[2] fid[4] offset[8] count[4] */
v9fs_treaddir(TReadDirOpt opt)610 TReadDirRes v9fs_treaddir(TReadDirOpt opt)
611 {
612     P9Req *req;
613     uint32_t err;
614 
615     g_assert(opt.client);
616     /* expecting either Rreaddir or Rlerror, but obviously not both */
617     g_assert(!opt.expectErr || !(opt.rreaddir.count ||
618              opt.rreaddir.nentries || opt.rreaddir.entries));
619 
620     req = v9fs_req_init(opt.client, 4 + 8 + 4, P9_TREADDIR, opt.tag);
621     v9fs_uint32_write(req, opt.fid);
622     v9fs_uint64_write(req, opt.offset);
623     v9fs_uint32_write(req, opt.count);
624     v9fs_req_send(req);
625 
626     if (!opt.requestOnly) {
627         v9fs_req_wait_for_reply(req, NULL);
628         if (opt.expectErr) {
629             v9fs_rlerror(req, &err);
630             g_assert_cmpint(err, ==, opt.expectErr);
631         } else {
632             v9fs_rreaddir(req, opt.rreaddir.count, opt.rreaddir.nentries,
633                           opt.rreaddir.entries);
634         }
635         req = NULL; /* request was freed */
636     }
637 
638     return (TReadDirRes) { .req = req };
639 }
640 
641 /* size[4] Rreaddir tag[2] count[4] data[count] */
v9fs_rreaddir(P9Req * req,uint32_t * count,uint32_t * nentries,struct V9fsDirent ** entries)642 void v9fs_rreaddir(P9Req *req, uint32_t *count, uint32_t *nentries,
643                    struct V9fsDirent **entries)
644 {
645     uint32_t local_count;
646     struct V9fsDirent *e = NULL;
647     /* only used to avoid a leak if entries was NULL */
648     struct V9fsDirent *unused_entries = NULL;
649     uint16_t slen;
650     uint32_t n = 0;
651 
652     v9fs_req_recv(req, P9_RREADDIR);
653     v9fs_uint32_read(req, &local_count);
654 
655     if (count) {
656         *count = local_count;
657     }
658 
659     for (int32_t togo = (int32_t)local_count;
660          togo >= 13 + 8 + 1 + 2;
661          togo -= 13 + 8 + 1 + 2 + slen, ++n)
662     {
663         if (!e) {
664             e = g_new(struct V9fsDirent, 1);
665             if (entries) {
666                 *entries = e;
667             } else {
668                 unused_entries = e;
669             }
670         } else {
671             e = e->next = g_new(struct V9fsDirent, 1);
672         }
673         e->next = NULL;
674         /* qid[13] offset[8] type[1] name[s] */
675         v9fs_memread(req, &e->qid, 13);
676         v9fs_uint64_read(req, &e->offset);
677         v9fs_uint8_read(req, &e->type);
678         v9fs_string_read(req, &slen, &e->name);
679     }
680 
681     if (nentries) {
682         *nentries = n;
683     }
684 
685     v9fs_free_dirents(unused_entries);
686     v9fs_req_free(req);
687 }
688 
v9fs_free_dirents(struct V9fsDirent * e)689 void v9fs_free_dirents(struct V9fsDirent *e)
690 {
691     struct V9fsDirent *next = NULL;
692 
693     for (; e; e = next) {
694         next = e->next;
695         g_free(e->name);
696         g_free(e);
697     }
698 }
699 
700 /* size[4] Tlopen tag[2] fid[4] flags[4] */
v9fs_tlopen(TLOpenOpt opt)701 TLOpenRes v9fs_tlopen(TLOpenOpt opt)
702 {
703     P9Req *req;
704     uint32_t err;
705 
706     g_assert(opt.client);
707     /* expecting either Rlopen or Rlerror, but obviously not both */
708     g_assert(!opt.expectErr || !(opt.rlopen.qid || opt.rlopen.iounit));
709 
710     req = v9fs_req_init(opt.client,  4 + 4, P9_TLOPEN, opt.tag);
711     v9fs_uint32_write(req, opt.fid);
712     v9fs_uint32_write(req, opt.flags);
713     v9fs_req_send(req);
714 
715     if (!opt.requestOnly) {
716         v9fs_req_wait_for_reply(req, NULL);
717         if (opt.expectErr) {
718             v9fs_rlerror(req, &err);
719             g_assert_cmpint(err, ==, opt.expectErr);
720         } else {
721             v9fs_rlopen(req, opt.rlopen.qid, opt.rlopen.iounit);
722         }
723         req = NULL; /* request was freed */
724     }
725 
726     return (TLOpenRes) { .req = req };
727 }
728 
729 /* size[4] Rlopen tag[2] qid[13] iounit[4] */
v9fs_rlopen(P9Req * req,v9fs_qid * qid,uint32_t * iounit)730 void v9fs_rlopen(P9Req *req, v9fs_qid *qid, uint32_t *iounit)
731 {
732     v9fs_req_recv(req, P9_RLOPEN);
733     if (qid) {
734         v9fs_memread(req, qid, 13);
735     } else {
736         v9fs_memskip(req, 13);
737     }
738     if (iounit) {
739         v9fs_uint32_read(req, iounit);
740     }
741     v9fs_req_free(req);
742 }
743 
744 /* size[4] Twrite tag[2] fid[4] offset[8] count[4] data[count] */
v9fs_twrite(TWriteOpt opt)745 TWriteRes v9fs_twrite(TWriteOpt opt)
746 {
747     P9Req *req;
748     uint32_t err;
749     uint32_t body_size = 4 + 8 + 4;
750     uint32_t written = 0;
751 
752     g_assert(opt.client);
753 
754     g_assert_cmpint(body_size, <=, UINT32_MAX - opt.count);
755     body_size += opt.count;
756     req = v9fs_req_init(opt.client, body_size, P9_TWRITE, opt.tag);
757     v9fs_uint32_write(req, opt.fid);
758     v9fs_uint64_write(req, opt.offset);
759     v9fs_uint32_write(req, opt.count);
760     v9fs_memwrite(req, opt.data, opt.count);
761     v9fs_req_send(req);
762 
763     if (!opt.requestOnly) {
764         v9fs_req_wait_for_reply(req, NULL);
765         if (opt.expectErr) {
766             v9fs_rlerror(req, &err);
767             g_assert_cmpint(err, ==, opt.expectErr);
768         } else {
769             v9fs_rwrite(req, &written);
770         }
771         req = NULL; /* request was freed */
772     }
773 
774     return (TWriteRes) {
775         .req = req,
776         .count = written
777     };
778 }
779 
780 /* size[4] Rwrite tag[2] count[4] */
v9fs_rwrite(P9Req * req,uint32_t * count)781 void v9fs_rwrite(P9Req *req, uint32_t *count)
782 {
783     v9fs_req_recv(req, P9_RWRITE);
784     if (count) {
785         v9fs_uint32_read(req, count);
786     }
787     v9fs_req_free(req);
788 }
789 
790 /* size[4] Tflush tag[2] oldtag[2] */
v9fs_tflush(TFlushOpt opt)791 TFlushRes v9fs_tflush(TFlushOpt opt)
792 {
793     P9Req *req;
794     uint32_t err;
795 
796     g_assert(opt.client);
797 
798     req = v9fs_req_init(opt.client, 2, P9_TFLUSH, opt.tag);
799     v9fs_uint32_write(req, opt.oldtag);
800     v9fs_req_send(req);
801 
802     if (!opt.requestOnly) {
803         v9fs_req_wait_for_reply(req, NULL);
804         if (opt.expectErr) {
805             v9fs_rlerror(req, &err);
806             g_assert_cmpint(err, ==, opt.expectErr);
807         } else {
808             v9fs_rflush(req);
809         }
810         req = NULL; /* request was freed */
811     }
812 
813     return (TFlushRes) { .req = req };
814 }
815 
816 /* size[4] Rflush tag[2] */
v9fs_rflush(P9Req * req)817 void v9fs_rflush(P9Req *req)
818 {
819     v9fs_req_recv(req, P9_RFLUSH);
820     v9fs_req_free(req);
821 }
822 
823 /* size[4] Tmkdir tag[2] dfid[4] name[s] mode[4] gid[4] */
v9fs_tmkdir(TMkdirOpt opt)824 TMkdirRes v9fs_tmkdir(TMkdirOpt opt)
825 {
826     P9Req *req;
827     uint32_t err;
828 
829     g_assert(opt.client);
830     /* expecting either hi-level atPath or low-level dfid, but not both */
831     g_assert(!opt.atPath || !opt.dfid);
832     /* expecting either Rmkdir or Rlerror, but obviously not both */
833     g_assert(!opt.expectErr || !opt.rmkdir.qid);
834 
835     if (opt.atPath) {
836         opt.dfid = v9fs_twalk((TWalkOpt) { .client = opt.client,
837                                            .path = opt.atPath }).newfid;
838     }
839 
840     if (!opt.mode) {
841         opt.mode = 0750;
842     }
843 
844     uint32_t body_size = 4 + 4 + 4;
845     uint16_t string_size = v9fs_string_size(opt.name);
846 
847     g_assert_cmpint(body_size, <=, UINT32_MAX - string_size);
848     body_size += string_size;
849 
850     req = v9fs_req_init(opt.client, body_size, P9_TMKDIR, opt.tag);
851     v9fs_uint32_write(req, opt.dfid);
852     v9fs_string_write(req, opt.name);
853     v9fs_uint32_write(req, opt.mode);
854     v9fs_uint32_write(req, opt.gid);
855     v9fs_req_send(req);
856 
857     if (!opt.requestOnly) {
858         v9fs_req_wait_for_reply(req, NULL);
859         if (opt.expectErr) {
860             v9fs_rlerror(req, &err);
861             g_assert_cmpint(err, ==, opt.expectErr);
862         } else {
863             v9fs_rmkdir(req, opt.rmkdir.qid);
864         }
865         req = NULL; /* request was freed */
866     }
867 
868     return (TMkdirRes) { .req = req };
869 }
870 
871 /* size[4] Rmkdir tag[2] qid[13] */
v9fs_rmkdir(P9Req * req,v9fs_qid * qid)872 void v9fs_rmkdir(P9Req *req, v9fs_qid *qid)
873 {
874     v9fs_req_recv(req, P9_RMKDIR);
875     if (qid) {
876         v9fs_memread(req, qid, 13);
877     } else {
878         v9fs_memskip(req, 13);
879     }
880     v9fs_req_free(req);
881 }
882 
883 /* size[4] Tlcreate tag[2] fid[4] name[s] flags[4] mode[4] gid[4] */
v9fs_tlcreate(TlcreateOpt opt)884 TlcreateRes v9fs_tlcreate(TlcreateOpt opt)
885 {
886     P9Req *req;
887     uint32_t err;
888 
889     g_assert(opt.client);
890     /* expecting either hi-level atPath or low-level fid, but not both */
891     g_assert(!opt.atPath || !opt.fid);
892     /* expecting either Rlcreate or Rlerror, but obviously not both */
893     g_assert(!opt.expectErr || !(opt.rlcreate.qid || opt.rlcreate.iounit));
894 
895     if (opt.atPath) {
896         opt.fid = v9fs_twalk((TWalkOpt) { .client = opt.client,
897                                           .path = opt.atPath }).newfid;
898     }
899 
900     if (!opt.mode) {
901         opt.mode = 0750;
902     }
903 
904     uint32_t body_size = 4 + 4 + 4 + 4;
905     uint16_t string_size = v9fs_string_size(opt.name);
906 
907     g_assert_cmpint(body_size, <=, UINT32_MAX - string_size);
908     body_size += string_size;
909 
910     req = v9fs_req_init(opt.client, body_size, P9_TLCREATE, opt.tag);
911     v9fs_uint32_write(req, opt.fid);
912     v9fs_string_write(req, opt.name);
913     v9fs_uint32_write(req, opt.flags);
914     v9fs_uint32_write(req, opt.mode);
915     v9fs_uint32_write(req, opt.gid);
916     v9fs_req_send(req);
917 
918     if (!opt.requestOnly) {
919         v9fs_req_wait_for_reply(req, NULL);
920         if (opt.expectErr) {
921             v9fs_rlerror(req, &err);
922             g_assert_cmpint(err, ==, opt.expectErr);
923         } else {
924             v9fs_rlcreate(req, opt.rlcreate.qid, opt.rlcreate.iounit);
925         }
926         req = NULL; /* request was freed */
927     }
928 
929     return (TlcreateRes) { .req = req };
930 }
931 
932 /* size[4] Rlcreate tag[2] qid[13] iounit[4] */
v9fs_rlcreate(P9Req * req,v9fs_qid * qid,uint32_t * iounit)933 void v9fs_rlcreate(P9Req *req, v9fs_qid *qid, uint32_t *iounit)
934 {
935     v9fs_req_recv(req, P9_RLCREATE);
936     if (qid) {
937         v9fs_memread(req, qid, 13);
938     } else {
939         v9fs_memskip(req, 13);
940     }
941     if (iounit) {
942         v9fs_uint32_read(req, iounit);
943     }
944     v9fs_req_free(req);
945 }
946 
947 /* size[4] Tsymlink tag[2] fid[4] name[s] symtgt[s] gid[4] */
v9fs_tsymlink(TsymlinkOpt opt)948 TsymlinkRes v9fs_tsymlink(TsymlinkOpt opt)
949 {
950     P9Req *req;
951     uint32_t err;
952 
953     g_assert(opt.client);
954     /* expecting either hi-level atPath or low-level fid, but not both */
955     g_assert(!opt.atPath || !opt.fid);
956     /* expecting either Rsymlink or Rlerror, but obviously not both */
957     g_assert(!opt.expectErr || !opt.rsymlink.qid);
958 
959     if (opt.atPath) {
960         opt.fid = v9fs_twalk((TWalkOpt) { .client = opt.client,
961                                           .path = opt.atPath }).newfid;
962     }
963 
964     uint32_t body_size = 4 + 4;
965     uint16_t string_size = v9fs_string_size(opt.name) +
966                            v9fs_string_size(opt.symtgt);
967 
968     g_assert_cmpint(body_size, <=, UINT32_MAX - string_size);
969     body_size += string_size;
970 
971     req = v9fs_req_init(opt.client, body_size, P9_TSYMLINK, opt.tag);
972     v9fs_uint32_write(req, opt.fid);
973     v9fs_string_write(req, opt.name);
974     v9fs_string_write(req, opt.symtgt);
975     v9fs_uint32_write(req, opt.gid);
976     v9fs_req_send(req);
977 
978     if (!opt.requestOnly) {
979         v9fs_req_wait_for_reply(req, NULL);
980         if (opt.expectErr) {
981             v9fs_rlerror(req, &err);
982             g_assert_cmpint(err, ==, opt.expectErr);
983         } else {
984             v9fs_rsymlink(req, opt.rsymlink.qid);
985         }
986         req = NULL; /* request was freed */
987     }
988 
989     return (TsymlinkRes) { .req = req };
990 }
991 
992 /* size[4] Rsymlink tag[2] qid[13] */
v9fs_rsymlink(P9Req * req,v9fs_qid * qid)993 void v9fs_rsymlink(P9Req *req, v9fs_qid *qid)
994 {
995     v9fs_req_recv(req, P9_RSYMLINK);
996     if (qid) {
997         v9fs_memread(req, qid, 13);
998     } else {
999         v9fs_memskip(req, 13);
1000     }
1001     v9fs_req_free(req);
1002 }
1003 
1004 /* size[4] Tlink tag[2] dfid[4] fid[4] name[s] */
v9fs_tlink(TlinkOpt opt)1005 TlinkRes v9fs_tlink(TlinkOpt opt)
1006 {
1007     P9Req *req;
1008     uint32_t err;
1009 
1010     g_assert(opt.client);
1011     /* expecting either hi-level atPath or low-level dfid, but not both */
1012     g_assert(!opt.atPath || !opt.dfid);
1013     /* expecting either hi-level toPath or low-level fid, but not both */
1014     g_assert(!opt.toPath || !opt.fid);
1015 
1016     if (opt.atPath) {
1017         opt.dfid = v9fs_twalk((TWalkOpt) { .client = opt.client,
1018                                            .path = opt.atPath }).newfid;
1019     }
1020     if (opt.toPath) {
1021         opt.fid = v9fs_twalk((TWalkOpt) { .client = opt.client,
1022                                           .path = opt.toPath }).newfid;
1023     }
1024 
1025     uint32_t body_size = 4 + 4;
1026     uint16_t string_size = v9fs_string_size(opt.name);
1027 
1028     g_assert_cmpint(body_size, <=, UINT32_MAX - string_size);
1029     body_size += string_size;
1030 
1031     req = v9fs_req_init(opt.client, body_size, P9_TLINK, opt.tag);
1032     v9fs_uint32_write(req, opt.dfid);
1033     v9fs_uint32_write(req, opt.fid);
1034     v9fs_string_write(req, opt.name);
1035     v9fs_req_send(req);
1036 
1037     if (!opt.requestOnly) {
1038         v9fs_req_wait_for_reply(req, NULL);
1039         if (opt.expectErr) {
1040             v9fs_rlerror(req, &err);
1041             g_assert_cmpint(err, ==, opt.expectErr);
1042         } else {
1043             v9fs_rlink(req);
1044         }
1045         req = NULL; /* request was freed */
1046     }
1047 
1048     return (TlinkRes) { .req = req };
1049 }
1050 
1051 /* size[4] Rlink tag[2] */
v9fs_rlink(P9Req * req)1052 void v9fs_rlink(P9Req *req)
1053 {
1054     v9fs_req_recv(req, P9_RLINK);
1055     v9fs_req_free(req);
1056 }
1057 
1058 /* size[4] Tunlinkat tag[2] dirfd[4] name[s] flags[4] */
v9fs_tunlinkat(TunlinkatOpt opt)1059 TunlinkatRes v9fs_tunlinkat(TunlinkatOpt opt)
1060 {
1061     P9Req *req;
1062     uint32_t err;
1063 
1064     g_assert(opt.client);
1065     /* expecting either hi-level atPath or low-level dirfd, but not both */
1066     g_assert(!opt.atPath || !opt.dirfd);
1067 
1068     if (opt.atPath) {
1069         opt.dirfd = v9fs_twalk((TWalkOpt) { .client = opt.client,
1070                                             .path = opt.atPath }).newfid;
1071     }
1072 
1073     uint32_t body_size = 4 + 4;
1074     uint16_t string_size = v9fs_string_size(opt.name);
1075 
1076     g_assert_cmpint(body_size, <=, UINT32_MAX - string_size);
1077     body_size += string_size;
1078 
1079     req = v9fs_req_init(opt.client, body_size, P9_TUNLINKAT, opt.tag);
1080     v9fs_uint32_write(req, opt.dirfd);
1081     v9fs_string_write(req, opt.name);
1082     v9fs_uint32_write(req, opt.flags);
1083     v9fs_req_send(req);
1084 
1085     if (!opt.requestOnly) {
1086         v9fs_req_wait_for_reply(req, NULL);
1087         if (opt.expectErr) {
1088             v9fs_rlerror(req, &err);
1089             g_assert_cmpint(err, ==, opt.expectErr);
1090         } else {
1091             v9fs_runlinkat(req);
1092         }
1093         req = NULL; /* request was freed */
1094     }
1095 
1096     return (TunlinkatRes) { .req = req };
1097 }
1098 
1099 /* size[4] Runlinkat tag[2] */
v9fs_runlinkat(P9Req * req)1100 void v9fs_runlinkat(P9Req *req)
1101 {
1102     v9fs_req_recv(req, P9_RUNLINKAT);
1103     v9fs_req_free(req);
1104 }
1105