Statistics
| Branch: | Revision:

root / hw / 9pfs / virtio-9p.c @ 9694b5d1

History | View | Annotate | Download (94.1 kB)

1
/*
2
 * Virtio 9p backend
3
 *
4
 * Copyright IBM, Corp. 2010
5
 *
6
 * Authors:
7
 *  Anthony Liguori   <aliguori@us.ibm.com>
8
 *
9
 * This work is licensed under the terms of the GNU GPL, version 2.  See
10
 * the COPYING file in the top-level directory.
11
 *
12
 */
13

    
14
#include "virtio.h"
15
#include "pc.h"
16
#include "qemu_socket.h"
17
#include "virtio-9p.h"
18
#include "fsdev/qemu-fsdev.h"
19
#include "virtio-9p-debug.h"
20
#include "virtio-9p-xattr.h"
21

    
22
int debug_9p_pdu;
23

    
24
enum {
25
    Oread   = 0x00,
26
    Owrite  = 0x01,
27
    Ordwr   = 0x02,
28
    Oexec   = 0x03,
29
    Oexcl   = 0x04,
30
    Otrunc  = 0x10,
31
    Orexec  = 0x20,
32
    Orclose = 0x40,
33
    Oappend = 0x80,
34
};
35

    
36
static int omode_to_uflags(int8_t mode)
37
{
38
    int ret = 0;
39

    
40
    switch (mode & 3) {
41
    case Oread:
42
        ret = O_RDONLY;
43
        break;
44
    case Ordwr:
45
        ret = O_RDWR;
46
        break;
47
    case Owrite:
48
        ret = O_WRONLY;
49
        break;
50
    case Oexec:
51
        ret = O_RDONLY;
52
        break;
53
    }
54

    
55
    if (mode & Otrunc) {
56
        ret |= O_TRUNC;
57
    }
58

    
59
    if (mode & Oappend) {
60
        ret |= O_APPEND;
61
    }
62

    
63
    if (mode & Oexcl) {
64
        ret |= O_EXCL;
65
    }
66

    
67
    return ret;
68
}
69

    
70
void cred_init(FsCred *credp)
71
{
72
    credp->fc_uid = -1;
73
    credp->fc_gid = -1;
74
    credp->fc_mode = -1;
75
    credp->fc_rdev = -1;
76
}
77

    
78
static int v9fs_do_lstat(V9fsState *s, V9fsString *path, struct stat *stbuf)
79
{
80
    return s->ops->lstat(&s->ctx, path->data, stbuf);
81
}
82

    
83
static ssize_t v9fs_do_readlink(V9fsState *s, V9fsString *path, V9fsString *buf)
84
{
85
    ssize_t len;
86

    
87
    buf->data = qemu_malloc(1024);
88

    
89
    len = s->ops->readlink(&s->ctx, path->data, buf->data, 1024 - 1);
90
    if (len > -1) {
91
        buf->size = len;
92
        buf->data[len] = 0;
93
    }
94

    
95
    return len;
96
}
97

    
98
static int v9fs_do_close(V9fsState *s, int fd)
99
{
100
    return s->ops->close(&s->ctx, fd);
101
}
102

    
103
static int v9fs_do_closedir(V9fsState *s, DIR *dir)
104
{
105
    return s->ops->closedir(&s->ctx, dir);
106
}
107

    
108
static int v9fs_do_open(V9fsState *s, V9fsString *path, int flags)
109
{
110
    return s->ops->open(&s->ctx, path->data, flags);
111
}
112

    
113
static DIR *v9fs_do_opendir(V9fsState *s, V9fsString *path)
114
{
115
    return s->ops->opendir(&s->ctx, path->data);
116
}
117

    
118
static void v9fs_do_rewinddir(V9fsState *s, DIR *dir)
119
{
120
    return s->ops->rewinddir(&s->ctx, dir);
121
}
122

    
123
static off_t v9fs_do_telldir(V9fsState *s, DIR *dir)
124
{
125
    return s->ops->telldir(&s->ctx, dir);
126
}
127

    
128
static struct dirent *v9fs_do_readdir(V9fsState *s, DIR *dir)
129
{
130
    return s->ops->readdir(&s->ctx, dir);
131
}
132

    
133
static void v9fs_do_seekdir(V9fsState *s, DIR *dir, off_t off)
134
{
135
    return s->ops->seekdir(&s->ctx, dir, off);
136
}
137

    
138
static int v9fs_do_preadv(V9fsState *s, int fd, const struct iovec *iov,
139
                            int iovcnt, int64_t offset)
140
{
141
    return s->ops->preadv(&s->ctx, fd, iov, iovcnt, offset);
142
}
143

    
144
static int v9fs_do_pwritev(V9fsState *s, int fd, const struct iovec *iov,
145
                       int iovcnt, int64_t offset)
146
{
147
    return s->ops->pwritev(&s->ctx, fd, iov, iovcnt, offset);
148
}
149

    
150
static int v9fs_do_chmod(V9fsState *s, V9fsString *path, mode_t mode)
151
{
152
    FsCred cred;
153
    cred_init(&cred);
154
    cred.fc_mode = mode;
155
    return s->ops->chmod(&s->ctx, path->data, &cred);
156
}
157

    
158
static int v9fs_do_mknod(V9fsState *s, char *name,
159
        mode_t mode, dev_t dev, uid_t uid, gid_t gid)
160
{
161
    FsCred cred;
162
    cred_init(&cred);
163
    cred.fc_uid = uid;
164
    cred.fc_gid = gid;
165
    cred.fc_mode = mode;
166
    cred.fc_rdev = dev;
167
    return s->ops->mknod(&s->ctx, name, &cred);
168
}
169

    
170
static int v9fs_do_mkdir(V9fsState *s, char *name, mode_t mode,
171
                uid_t uid, gid_t gid)
172
{
173
    FsCred cred;
174

    
175
    cred_init(&cred);
176
    cred.fc_uid = uid;
177
    cred.fc_gid = gid;
178
    cred.fc_mode = mode;
179

    
180
    return s->ops->mkdir(&s->ctx, name, &cred);
181
}
182

    
183
static int v9fs_do_fstat(V9fsState *s, int fd, struct stat *stbuf)
184
{
185
    return s->ops->fstat(&s->ctx, fd, stbuf);
186
}
187

    
188
static int v9fs_do_open2(V9fsState *s, char *fullname, uid_t uid, gid_t gid,
189
        int flags, int mode)
190
{
191
    FsCred cred;
192

    
193
    cred_init(&cred);
194
    cred.fc_uid = uid;
195
    cred.fc_gid = gid;
196
    cred.fc_mode = mode & 07777;
197

    
198
    return s->ops->open2(&s->ctx, fullname, flags, &cred);
199
}
200

    
201
static int v9fs_do_symlink(V9fsState *s, V9fsFidState *fidp,
202
        const char *oldpath, const char *newpath, gid_t gid)
203
{
204
    FsCred cred;
205
    cred_init(&cred);
206
    cred.fc_uid = fidp->uid;
207
    cred.fc_gid = gid;
208
    cred.fc_mode = 0777;
209

    
210
    return s->ops->symlink(&s->ctx, oldpath, newpath, &cred);
211
}
212

    
213
static int v9fs_do_link(V9fsState *s, V9fsString *oldpath, V9fsString *newpath)
214
{
215
    return s->ops->link(&s->ctx, oldpath->data, newpath->data);
216
}
217

    
218
static int v9fs_do_truncate(V9fsState *s, V9fsString *path, off_t size)
219
{
220
    return s->ops->truncate(&s->ctx, path->data, size);
221
}
222

    
223
static int v9fs_do_rename(V9fsState *s, V9fsString *oldpath,
224
                            V9fsString *newpath)
225
{
226
    return s->ops->rename(&s->ctx, oldpath->data, newpath->data);
227
}
228

    
229
static int v9fs_do_chown(V9fsState *s, V9fsString *path, uid_t uid, gid_t gid)
230
{
231
    FsCred cred;
232
    cred_init(&cred);
233
    cred.fc_uid = uid;
234
    cred.fc_gid = gid;
235

    
236
    return s->ops->chown(&s->ctx, path->data, &cred);
237
}
238

    
239
static int v9fs_do_utimensat(V9fsState *s, V9fsString *path,
240
                                           const struct timespec times[2])
241
{
242
    return s->ops->utimensat(&s->ctx, path->data, times);
243
}
244

    
245
static int v9fs_do_remove(V9fsState *s, V9fsString *path)
246
{
247
    return s->ops->remove(&s->ctx, path->data);
248
}
249

    
250
static int v9fs_do_fsync(V9fsState *s, int fd, int datasync)
251
{
252
    return s->ops->fsync(&s->ctx, fd, datasync);
253
}
254

    
255
static int v9fs_do_statfs(V9fsState *s, V9fsString *path, struct statfs *stbuf)
256
{
257
    return s->ops->statfs(&s->ctx, path->data, stbuf);
258
}
259

    
260
static ssize_t v9fs_do_lgetxattr(V9fsState *s, V9fsString *path,
261
                             V9fsString *xattr_name,
262
                             void *value, size_t size)
263
{
264
    return s->ops->lgetxattr(&s->ctx, path->data,
265
                             xattr_name->data, value, size);
266
}
267

    
268
static ssize_t v9fs_do_llistxattr(V9fsState *s, V9fsString *path,
269
                              void *value, size_t size)
270
{
271
    return s->ops->llistxattr(&s->ctx, path->data,
272
                              value, size);
273
}
274

    
275
static int v9fs_do_lsetxattr(V9fsState *s, V9fsString *path,
276
                             V9fsString *xattr_name,
277
                             void *value, size_t size, int flags)
278
{
279
    return s->ops->lsetxattr(&s->ctx, path->data,
280
                             xattr_name->data, value, size, flags);
281
}
282

    
283
static int v9fs_do_lremovexattr(V9fsState *s, V9fsString *path,
284
                                V9fsString *xattr_name)
285
{
286
    return s->ops->lremovexattr(&s->ctx, path->data,
287
                                xattr_name->data);
288
}
289

    
290

    
291
static void v9fs_string_init(V9fsString *str)
292
{
293
    str->data = NULL;
294
    str->size = 0;
295
}
296

    
297
static void v9fs_string_free(V9fsString *str)
298
{
299
    qemu_free(str->data);
300
    str->data = NULL;
301
    str->size = 0;
302
}
303

    
304
static void v9fs_string_null(V9fsString *str)
305
{
306
    v9fs_string_free(str);
307
}
308

    
309
static int number_to_string(void *arg, char type)
310
{
311
    unsigned int ret = 0;
312

    
313
    switch (type) {
314
    case 'u': {
315
        unsigned int num = *(unsigned int *)arg;
316

    
317
        do {
318
            ret++;
319
            num = num/10;
320
        } while (num);
321
        break;
322
    }
323
    case 'U': {
324
        unsigned long num = *(unsigned long *)arg;
325
        do {
326
            ret++;
327
            num = num/10;
328
        } while (num);
329
        break;
330
    }
331
    default:
332
        printf("Number_to_string: Unknown number format\n");
333
        return -1;
334
    }
335

    
336
    return ret;
337
}
338

    
339
static int GCC_FMT_ATTR(2, 0)
340
v9fs_string_alloc_printf(char **strp, const char *fmt, va_list ap)
341
{
342
    va_list ap2;
343
    char *iter = (char *)fmt;
344
    int len = 0;
345
    int nr_args = 0;
346
    char *arg_char_ptr;
347
    unsigned int arg_uint;
348
    unsigned long arg_ulong;
349

    
350
    /* Find the number of %'s that denotes an argument */
351
    for (iter = strstr(iter, "%"); iter; iter = strstr(iter, "%")) {
352
        nr_args++;
353
        iter++;
354
    }
355

    
356
    len = strlen(fmt) - 2*nr_args;
357

    
358
    if (!nr_args) {
359
        goto alloc_print;
360
    }
361

    
362
    va_copy(ap2, ap);
363

    
364
    iter = (char *)fmt;
365

    
366
    /* Now parse the format string */
367
    for (iter = strstr(iter, "%"); iter; iter = strstr(iter, "%")) {
368
        iter++;
369
        switch (*iter) {
370
        case 'u':
371
            arg_uint = va_arg(ap2, unsigned int);
372
            len += number_to_string((void *)&arg_uint, 'u');
373
            break;
374
        case 'l':
375
            if (*++iter == 'u') {
376
                arg_ulong = va_arg(ap2, unsigned long);
377
                len += number_to_string((void *)&arg_ulong, 'U');
378
            } else {
379
                return -1;
380
            }
381
            break;
382
        case 's':
383
            arg_char_ptr = va_arg(ap2, char *);
384
            len += strlen(arg_char_ptr);
385
            break;
386
        case 'c':
387
            len += 1;
388
            break;
389
        default:
390
            fprintf(stderr,
391
                    "v9fs_string_alloc_printf:Incorrect format %c", *iter);
392
            return -1;
393
        }
394
        iter++;
395
    }
396

    
397
alloc_print:
398
    *strp = qemu_malloc((len + 1) * sizeof(**strp));
399

    
400
    return vsprintf(*strp, fmt, ap);
401
}
402

    
403
static void GCC_FMT_ATTR(2, 3)
404
v9fs_string_sprintf(V9fsString *str, const char *fmt, ...)
405
{
406
    va_list ap;
407
    int err;
408

    
409
    v9fs_string_free(str);
410

    
411
    va_start(ap, fmt);
412
    err = v9fs_string_alloc_printf(&str->data, fmt, ap);
413
    BUG_ON(err == -1);
414
    va_end(ap);
415

    
416
    str->size = err;
417
}
418

    
419
static void v9fs_string_copy(V9fsString *lhs, V9fsString *rhs)
420
{
421
    v9fs_string_free(lhs);
422
    v9fs_string_sprintf(lhs, "%s", rhs->data);
423
}
424

    
425
static size_t v9fs_string_size(V9fsString *str)
426
{
427
    return str->size;
428
}
429

    
430
static V9fsFidState *lookup_fid(V9fsState *s, int32_t fid)
431
{
432
    V9fsFidState *f;
433

    
434
    for (f = s->fid_list; f; f = f->next) {
435
        if (f->fid == fid) {
436
            return f;
437
        }
438
    }
439

    
440
    return NULL;
441
}
442

    
443
static V9fsFidState *alloc_fid(V9fsState *s, int32_t fid)
444
{
445
    V9fsFidState *f;
446

    
447
    f = lookup_fid(s, fid);
448
    if (f) {
449
        return NULL;
450
    }
451

    
452
    f = qemu_mallocz(sizeof(V9fsFidState));
453

    
454
    f->fid = fid;
455
    f->fid_type = P9_FID_NONE;
456

    
457
    f->next = s->fid_list;
458
    s->fid_list = f;
459

    
460
    return f;
461
}
462

    
463
static int v9fs_xattr_fid_clunk(V9fsState *s, V9fsFidState *fidp)
464
{
465
    int retval = 0;
466

    
467
    if (fidp->fs.xattr.copied_len == -1) {
468
        /* getxattr/listxattr fid */
469
        goto free_value;
470
    }
471
    /*
472
     * if this is fid for setxattr. clunk should
473
     * result in setxattr localcall
474
     */
475
    if (fidp->fs.xattr.len != fidp->fs.xattr.copied_len) {
476
        /* clunk after partial write */
477
        retval = -EINVAL;
478
        goto free_out;
479
    }
480
    if (fidp->fs.xattr.len) {
481
        retval = v9fs_do_lsetxattr(s, &fidp->path, &fidp->fs.xattr.name,
482
                                   fidp->fs.xattr.value,
483
                                   fidp->fs.xattr.len,
484
                                   fidp->fs.xattr.flags);
485
    } else {
486
        retval = v9fs_do_lremovexattr(s, &fidp->path, &fidp->fs.xattr.name);
487
    }
488
free_out:
489
    v9fs_string_free(&fidp->fs.xattr.name);
490
free_value:
491
    if (fidp->fs.xattr.value) {
492
        qemu_free(fidp->fs.xattr.value);
493
    }
494
    return retval;
495
}
496

    
497
static int free_fid(V9fsState *s, int32_t fid)
498
{
499
    int retval = 0;
500
    V9fsFidState **fidpp, *fidp;
501

    
502
    for (fidpp = &s->fid_list; *fidpp; fidpp = &(*fidpp)->next) {
503
        if ((*fidpp)->fid == fid) {
504
            break;
505
        }
506
    }
507

    
508
    if (*fidpp == NULL) {
509
        return -ENOENT;
510
    }
511

    
512
    fidp = *fidpp;
513
    *fidpp = fidp->next;
514

    
515
    if (fidp->fid_type == P9_FID_FILE) {
516
        v9fs_do_close(s, fidp->fs.fd);
517
    } else if (fidp->fid_type == P9_FID_DIR) {
518
        v9fs_do_closedir(s, fidp->fs.dir);
519
    } else if (fidp->fid_type == P9_FID_XATTR) {
520
        retval = v9fs_xattr_fid_clunk(s, fidp);
521
    }
522
    v9fs_string_free(&fidp->path);
523
    qemu_free(fidp);
524

    
525
    return retval;
526
}
527

    
528
#define P9_QID_TYPE_DIR         0x80
529
#define P9_QID_TYPE_SYMLINK     0x02
530

    
531
#define P9_STAT_MODE_DIR        0x80000000
532
#define P9_STAT_MODE_APPEND     0x40000000
533
#define P9_STAT_MODE_EXCL       0x20000000
534
#define P9_STAT_MODE_MOUNT      0x10000000
535
#define P9_STAT_MODE_AUTH       0x08000000
536
#define P9_STAT_MODE_TMP        0x04000000
537
#define P9_STAT_MODE_SYMLINK    0x02000000
538
#define P9_STAT_MODE_LINK       0x01000000
539
#define P9_STAT_MODE_DEVICE     0x00800000
540
#define P9_STAT_MODE_NAMED_PIPE 0x00200000
541
#define P9_STAT_MODE_SOCKET     0x00100000
542
#define P9_STAT_MODE_SETUID     0x00080000
543
#define P9_STAT_MODE_SETGID     0x00040000
544
#define P9_STAT_MODE_SETVTX     0x00010000
545

    
546
#define P9_STAT_MODE_TYPE_BITS (P9_STAT_MODE_DIR |          \
547
                                P9_STAT_MODE_SYMLINK |      \
548
                                P9_STAT_MODE_LINK |         \
549
                                P9_STAT_MODE_DEVICE |       \
550
                                P9_STAT_MODE_NAMED_PIPE |   \
551
                                P9_STAT_MODE_SOCKET)
552

    
553
/* This is the algorithm from ufs in spfs */
554
static void stat_to_qid(const struct stat *stbuf, V9fsQID *qidp)
555
{
556
    size_t size;
557

    
558
    size = MIN(sizeof(stbuf->st_ino), sizeof(qidp->path));
559
    memcpy(&qidp->path, &stbuf->st_ino, size);
560
    qidp->version = stbuf->st_mtime ^ (stbuf->st_size << 8);
561
    qidp->type = 0;
562
    if (S_ISDIR(stbuf->st_mode)) {
563
        qidp->type |= P9_QID_TYPE_DIR;
564
    }
565
    if (S_ISLNK(stbuf->st_mode)) {
566
        qidp->type |= P9_QID_TYPE_SYMLINK;
567
    }
568
}
569

    
570
static int fid_to_qid(V9fsState *s, V9fsFidState *fidp, V9fsQID *qidp)
571
{
572
    struct stat stbuf;
573
    int err;
574

    
575
    err = v9fs_do_lstat(s, &fidp->path, &stbuf);
576
    if (err) {
577
        return err;
578
    }
579

    
580
    stat_to_qid(&stbuf, qidp);
581
    return 0;
582
}
583

    
584
static V9fsPDU *alloc_pdu(V9fsState *s)
585
{
586
    V9fsPDU *pdu = NULL;
587

    
588
    if (!QLIST_EMPTY(&s->free_list)) {
589
        pdu = QLIST_FIRST(&s->free_list);
590
        QLIST_REMOVE(pdu, next);
591
    }
592
    return pdu;
593
}
594

    
595
static void free_pdu(V9fsState *s, V9fsPDU *pdu)
596
{
597
    if (pdu) {
598
        if (debug_9p_pdu) {
599
            pprint_pdu(pdu);
600
        }
601
        QLIST_INSERT_HEAD(&s->free_list, pdu, next);
602
    }
603
}
604

    
605
size_t pdu_packunpack(void *addr, struct iovec *sg, int sg_count,
606
                        size_t offset, size_t size, int pack)
607
{
608
    int i = 0;
609
    size_t copied = 0;
610

    
611
    for (i = 0; size && i < sg_count; i++) {
612
        size_t len;
613
        if (offset >= sg[i].iov_len) {
614
            /* skip this sg */
615
            offset -= sg[i].iov_len;
616
            continue;
617
        } else {
618
            len = MIN(sg[i].iov_len - offset, size);
619
            if (pack) {
620
                memcpy(sg[i].iov_base + offset, addr, len);
621
            } else {
622
                memcpy(addr, sg[i].iov_base + offset, len);
623
            }
624
            size -= len;
625
            copied += len;
626
            addr += len;
627
            if (size) {
628
                offset = 0;
629
                continue;
630
            }
631
        }
632
    }
633

    
634
    return copied;
635
}
636

    
637
static size_t pdu_unpack(void *dst, V9fsPDU *pdu, size_t offset, size_t size)
638
{
639
    return pdu_packunpack(dst, pdu->elem.out_sg, pdu->elem.out_num,
640
                         offset, size, 0);
641
}
642

    
643
static size_t pdu_pack(V9fsPDU *pdu, size_t offset, const void *src,
644
                        size_t size)
645
{
646
    return pdu_packunpack((void *)src, pdu->elem.in_sg, pdu->elem.in_num,
647
                             offset, size, 1);
648
}
649

    
650
static int pdu_copy_sg(V9fsPDU *pdu, size_t offset, int rx, struct iovec *sg)
651
{
652
    size_t pos = 0;
653
    int i, j;
654
    struct iovec *src_sg;
655
    unsigned int num;
656

    
657
    if (rx) {
658
        src_sg = pdu->elem.in_sg;
659
        num = pdu->elem.in_num;
660
    } else {
661
        src_sg = pdu->elem.out_sg;
662
        num = pdu->elem.out_num;
663
    }
664

    
665
    j = 0;
666
    for (i = 0; i < num; i++) {
667
        if (offset <= pos) {
668
            sg[j].iov_base = src_sg[i].iov_base;
669
            sg[j].iov_len = src_sg[i].iov_len;
670
            j++;
671
        } else if (offset < (src_sg[i].iov_len + pos)) {
672
            sg[j].iov_base = src_sg[i].iov_base;
673
            sg[j].iov_len = src_sg[i].iov_len;
674
            sg[j].iov_base += (offset - pos);
675
            sg[j].iov_len -= (offset - pos);
676
            j++;
677
        }
678
        pos += src_sg[i].iov_len;
679
    }
680

    
681
    return j;
682
}
683

    
684
static size_t pdu_unmarshal(V9fsPDU *pdu, size_t offset, const char *fmt, ...)
685
{
686
    size_t old_offset = offset;
687
    va_list ap;
688
    int i;
689

    
690
    va_start(ap, fmt);
691
    for (i = 0; fmt[i]; i++) {
692
        switch (fmt[i]) {
693
        case 'b': {
694
            uint8_t *valp = va_arg(ap, uint8_t *);
695
            offset += pdu_unpack(valp, pdu, offset, sizeof(*valp));
696
            break;
697
        }
698
        case 'w': {
699
            uint16_t val, *valp;
700
            valp = va_arg(ap, uint16_t *);
701
            offset += pdu_unpack(&val, pdu, offset, sizeof(val));
702
            *valp = le16_to_cpu(val);
703
            break;
704
        }
705
        case 'd': {
706
            uint32_t val, *valp;
707
            valp = va_arg(ap, uint32_t *);
708
            offset += pdu_unpack(&val, pdu, offset, sizeof(val));
709
            *valp = le32_to_cpu(val);
710
            break;
711
        }
712
        case 'q': {
713
            uint64_t val, *valp;
714
            valp = va_arg(ap, uint64_t *);
715
            offset += pdu_unpack(&val, pdu, offset, sizeof(val));
716
            *valp = le64_to_cpu(val);
717
            break;
718
        }
719
        case 'v': {
720
            struct iovec *iov = va_arg(ap, struct iovec *);
721
            int *iovcnt = va_arg(ap, int *);
722
            *iovcnt = pdu_copy_sg(pdu, offset, 0, iov);
723
            break;
724
        }
725
        case 's': {
726
            V9fsString *str = va_arg(ap, V9fsString *);
727
            offset += pdu_unmarshal(pdu, offset, "w", &str->size);
728
            /* FIXME: sanity check str->size */
729
            str->data = qemu_malloc(str->size + 1);
730
            offset += pdu_unpack(str->data, pdu, offset, str->size);
731
            str->data[str->size] = 0;
732
            break;
733
        }
734
        case 'Q': {
735
            V9fsQID *qidp = va_arg(ap, V9fsQID *);
736
            offset += pdu_unmarshal(pdu, offset, "bdq",
737
                        &qidp->type, &qidp->version, &qidp->path);
738
            break;
739
        }
740
        case 'S': {
741
            V9fsStat *statp = va_arg(ap, V9fsStat *);
742
            offset += pdu_unmarshal(pdu, offset, "wwdQdddqsssssddd",
743
                        &statp->size, &statp->type, &statp->dev,
744
                        &statp->qid, &statp->mode, &statp->atime,
745
                        &statp->mtime, &statp->length,
746
                        &statp->name, &statp->uid, &statp->gid,
747
                        &statp->muid, &statp->extension,
748
                        &statp->n_uid, &statp->n_gid,
749
                        &statp->n_muid);
750
            break;
751
        }
752
        case 'I': {
753
            V9fsIattr *iattr = va_arg(ap, V9fsIattr *);
754
            offset += pdu_unmarshal(pdu, offset, "ddddqqqqq",
755
                        &iattr->valid, &iattr->mode,
756
                        &iattr->uid, &iattr->gid, &iattr->size,
757
                        &iattr->atime_sec, &iattr->atime_nsec,
758
                        &iattr->mtime_sec, &iattr->mtime_nsec);
759
            break;
760
        }
761
        default:
762
            break;
763
        }
764
    }
765

    
766
    va_end(ap);
767

    
768
    return offset - old_offset;
769
}
770

    
771
static size_t pdu_marshal(V9fsPDU *pdu, size_t offset, const char *fmt, ...)
772
{
773
    size_t old_offset = offset;
774
    va_list ap;
775
    int i;
776

    
777
    va_start(ap, fmt);
778
    for (i = 0; fmt[i]; i++) {
779
        switch (fmt[i]) {
780
        case 'b': {
781
            uint8_t val = va_arg(ap, int);
782
            offset += pdu_pack(pdu, offset, &val, sizeof(val));
783
            break;
784
        }
785
        case 'w': {
786
            uint16_t val;
787
            cpu_to_le16w(&val, va_arg(ap, int));
788
            offset += pdu_pack(pdu, offset, &val, sizeof(val));
789
            break;
790
        }
791
        case 'd': {
792
            uint32_t val;
793
            cpu_to_le32w(&val, va_arg(ap, uint32_t));
794
            offset += pdu_pack(pdu, offset, &val, sizeof(val));
795
            break;
796
        }
797
        case 'q': {
798
            uint64_t val;
799
            cpu_to_le64w(&val, va_arg(ap, uint64_t));
800
            offset += pdu_pack(pdu, offset, &val, sizeof(val));
801
            break;
802
        }
803
        case 'v': {
804
            struct iovec *iov = va_arg(ap, struct iovec *);
805
            int *iovcnt = va_arg(ap, int *);
806
            *iovcnt = pdu_copy_sg(pdu, offset, 1, iov);
807
            break;
808
        }
809
        case 's': {
810
            V9fsString *str = va_arg(ap, V9fsString *);
811
            offset += pdu_marshal(pdu, offset, "w", str->size);
812
            offset += pdu_pack(pdu, offset, str->data, str->size);
813
            break;
814
        }
815
        case 'Q': {
816
            V9fsQID *qidp = va_arg(ap, V9fsQID *);
817
            offset += pdu_marshal(pdu, offset, "bdq",
818
                        qidp->type, qidp->version, qidp->path);
819
            break;
820
        }
821
        case 'S': {
822
            V9fsStat *statp = va_arg(ap, V9fsStat *);
823
            offset += pdu_marshal(pdu, offset, "wwdQdddqsssssddd",
824
                        statp->size, statp->type, statp->dev,
825
                        &statp->qid, statp->mode, statp->atime,
826
                        statp->mtime, statp->length, &statp->name,
827
                        &statp->uid, &statp->gid, &statp->muid,
828
                        &statp->extension, statp->n_uid,
829
                        statp->n_gid, statp->n_muid);
830
            break;
831
        }
832
        case 'A': {
833
            V9fsStatDotl *statp = va_arg(ap, V9fsStatDotl *);
834
            offset += pdu_marshal(pdu, offset, "qQdddqqqqqqqqqqqqqqq",
835
                        statp->st_result_mask,
836
                        &statp->qid, statp->st_mode,
837
                        statp->st_uid, statp->st_gid,
838
                        statp->st_nlink, statp->st_rdev,
839
                        statp->st_size, statp->st_blksize, statp->st_blocks,
840
                        statp->st_atime_sec, statp->st_atime_nsec,
841
                        statp->st_mtime_sec, statp->st_mtime_nsec,
842
                        statp->st_ctime_sec, statp->st_ctime_nsec,
843
                        statp->st_btime_sec, statp->st_btime_nsec,
844
                        statp->st_gen, statp->st_data_version);
845
            break;
846
        }
847
        default:
848
            break;
849
        }
850
    }
851
    va_end(ap);
852

    
853
    return offset - old_offset;
854
}
855

    
856
static void complete_pdu(V9fsState *s, V9fsPDU *pdu, ssize_t len)
857
{
858
    int8_t id = pdu->id + 1; /* Response */
859

    
860
    if (len < 0) {
861
        int err = -len;
862
        len = 7;
863

    
864
        if (s->proto_version != V9FS_PROTO_2000L) {
865
            V9fsString str;
866

    
867
            str.data = strerror(err);
868
            str.size = strlen(str.data);
869

    
870
            len += pdu_marshal(pdu, len, "s", &str);
871
            id = P9_RERROR;
872
        }
873

    
874
        len += pdu_marshal(pdu, len, "d", err);
875

    
876
        if (s->proto_version == V9FS_PROTO_2000L) {
877
            id = P9_RLERROR;
878
        }
879
    }
880

    
881
    /* fill out the header */
882
    pdu_marshal(pdu, 0, "dbw", (int32_t)len, id, pdu->tag);
883

    
884
    /* keep these in sync */
885
    pdu->size = len;
886
    pdu->id = id;
887

    
888
    /* push onto queue and notify */
889
    virtqueue_push(s->vq, &pdu->elem, len);
890

    
891
    /* FIXME: we should batch these completions */
892
    virtio_notify(&s->vdev, s->vq);
893

    
894
    free_pdu(s, pdu);
895
}
896

    
897
static mode_t v9mode_to_mode(uint32_t mode, V9fsString *extension)
898
{
899
    mode_t ret;
900

    
901
    ret = mode & 0777;
902
    if (mode & P9_STAT_MODE_DIR) {
903
        ret |= S_IFDIR;
904
    }
905

    
906
    if (mode & P9_STAT_MODE_SYMLINK) {
907
        ret |= S_IFLNK;
908
    }
909
    if (mode & P9_STAT_MODE_SOCKET) {
910
        ret |= S_IFSOCK;
911
    }
912
    if (mode & P9_STAT_MODE_NAMED_PIPE) {
913
        ret |= S_IFIFO;
914
    }
915
    if (mode & P9_STAT_MODE_DEVICE) {
916
        if (extension && extension->data[0] == 'c') {
917
            ret |= S_IFCHR;
918
        } else {
919
            ret |= S_IFBLK;
920
        }
921
    }
922

    
923
    if (!(ret&~0777)) {
924
        ret |= S_IFREG;
925
    }
926

    
927
    if (mode & P9_STAT_MODE_SETUID) {
928
        ret |= S_ISUID;
929
    }
930
    if (mode & P9_STAT_MODE_SETGID) {
931
        ret |= S_ISGID;
932
    }
933
    if (mode & P9_STAT_MODE_SETVTX) {
934
        ret |= S_ISVTX;
935
    }
936

    
937
    return ret;
938
}
939

    
940
static int donttouch_stat(V9fsStat *stat)
941
{
942
    if (stat->type == -1 &&
943
        stat->dev == -1 &&
944
        stat->qid.type == -1 &&
945
        stat->qid.version == -1 &&
946
        stat->qid.path == -1 &&
947
        stat->mode == -1 &&
948
        stat->atime == -1 &&
949
        stat->mtime == -1 &&
950
        stat->length == -1 &&
951
        !stat->name.size &&
952
        !stat->uid.size &&
953
        !stat->gid.size &&
954
        !stat->muid.size &&
955
        stat->n_uid == -1 &&
956
        stat->n_gid == -1 &&
957
        stat->n_muid == -1) {
958
        return 1;
959
    }
960

    
961
    return 0;
962
}
963

    
964
static void v9fs_stat_free(V9fsStat *stat)
965
{
966
    v9fs_string_free(&stat->name);
967
    v9fs_string_free(&stat->uid);
968
    v9fs_string_free(&stat->gid);
969
    v9fs_string_free(&stat->muid);
970
    v9fs_string_free(&stat->extension);
971
}
972

    
973
static uint32_t stat_to_v9mode(const struct stat *stbuf)
974
{
975
    uint32_t mode;
976

    
977
    mode = stbuf->st_mode & 0777;
978
    if (S_ISDIR(stbuf->st_mode)) {
979
        mode |= P9_STAT_MODE_DIR;
980
    }
981

    
982
    if (S_ISLNK(stbuf->st_mode)) {
983
        mode |= P9_STAT_MODE_SYMLINK;
984
    }
985

    
986
    if (S_ISSOCK(stbuf->st_mode)) {
987
        mode |= P9_STAT_MODE_SOCKET;
988
    }
989

    
990
    if (S_ISFIFO(stbuf->st_mode)) {
991
        mode |= P9_STAT_MODE_NAMED_PIPE;
992
    }
993

    
994
    if (S_ISBLK(stbuf->st_mode) || S_ISCHR(stbuf->st_mode)) {
995
        mode |= P9_STAT_MODE_DEVICE;
996
    }
997

    
998
    if (stbuf->st_mode & S_ISUID) {
999
        mode |= P9_STAT_MODE_SETUID;
1000
    }
1001

    
1002
    if (stbuf->st_mode & S_ISGID) {
1003
        mode |= P9_STAT_MODE_SETGID;
1004
    }
1005

    
1006
    if (stbuf->st_mode & S_ISVTX) {
1007
        mode |= P9_STAT_MODE_SETVTX;
1008
    }
1009

    
1010
    return mode;
1011
}
1012

    
1013
static int stat_to_v9stat(V9fsState *s, V9fsString *name,
1014
                            const struct stat *stbuf,
1015
                            V9fsStat *v9stat)
1016
{
1017
    int err;
1018
    const char *str;
1019

    
1020
    memset(v9stat, 0, sizeof(*v9stat));
1021

    
1022
    stat_to_qid(stbuf, &v9stat->qid);
1023
    v9stat->mode = stat_to_v9mode(stbuf);
1024
    v9stat->atime = stbuf->st_atime;
1025
    v9stat->mtime = stbuf->st_mtime;
1026
    v9stat->length = stbuf->st_size;
1027

    
1028
    v9fs_string_null(&v9stat->uid);
1029
    v9fs_string_null(&v9stat->gid);
1030
    v9fs_string_null(&v9stat->muid);
1031

    
1032
    v9stat->n_uid = stbuf->st_uid;
1033
    v9stat->n_gid = stbuf->st_gid;
1034
    v9stat->n_muid = 0;
1035

    
1036
    v9fs_string_null(&v9stat->extension);
1037

    
1038
    if (v9stat->mode & P9_STAT_MODE_SYMLINK) {
1039
        err = v9fs_do_readlink(s, name, &v9stat->extension);
1040
        if (err == -1) {
1041
            err = -errno;
1042
            return err;
1043
        }
1044
        v9stat->extension.data[err] = 0;
1045
        v9stat->extension.size = err;
1046
    } else if (v9stat->mode & P9_STAT_MODE_DEVICE) {
1047
        v9fs_string_sprintf(&v9stat->extension, "%c %u %u",
1048
                S_ISCHR(stbuf->st_mode) ? 'c' : 'b',
1049
                major(stbuf->st_rdev), minor(stbuf->st_rdev));
1050
    } else if (S_ISDIR(stbuf->st_mode) || S_ISREG(stbuf->st_mode)) {
1051
        v9fs_string_sprintf(&v9stat->extension, "%s %lu",
1052
                "HARDLINKCOUNT", (unsigned long)stbuf->st_nlink);
1053
    }
1054

    
1055
    str = strrchr(name->data, '/');
1056
    if (str) {
1057
        str += 1;
1058
    } else {
1059
        str = name->data;
1060
    }
1061

    
1062
    v9fs_string_sprintf(&v9stat->name, "%s", str);
1063

    
1064
    v9stat->size = 61 +
1065
        v9fs_string_size(&v9stat->name) +
1066
        v9fs_string_size(&v9stat->uid) +
1067
        v9fs_string_size(&v9stat->gid) +
1068
        v9fs_string_size(&v9stat->muid) +
1069
        v9fs_string_size(&v9stat->extension);
1070
    return 0;
1071
}
1072

    
1073
#define P9_STATS_MODE          0x00000001ULL
1074
#define P9_STATS_NLINK         0x00000002ULL
1075
#define P9_STATS_UID           0x00000004ULL
1076
#define P9_STATS_GID           0x00000008ULL
1077
#define P9_STATS_RDEV          0x00000010ULL
1078
#define P9_STATS_ATIME         0x00000020ULL
1079
#define P9_STATS_MTIME         0x00000040ULL
1080
#define P9_STATS_CTIME         0x00000080ULL
1081
#define P9_STATS_INO           0x00000100ULL
1082
#define P9_STATS_SIZE          0x00000200ULL
1083
#define P9_STATS_BLOCKS        0x00000400ULL
1084

    
1085
#define P9_STATS_BTIME         0x00000800ULL
1086
#define P9_STATS_GEN           0x00001000ULL
1087
#define P9_STATS_DATA_VERSION  0x00002000ULL
1088

    
1089
#define P9_STATS_BASIC         0x000007ffULL /* Mask for fields up to BLOCKS */
1090
#define P9_STATS_ALL           0x00003fffULL /* Mask for All fields above */
1091

    
1092

    
1093
static void stat_to_v9stat_dotl(V9fsState *s, const struct stat *stbuf,
1094
                            V9fsStatDotl *v9lstat)
1095
{
1096
    memset(v9lstat, 0, sizeof(*v9lstat));
1097

    
1098
    v9lstat->st_mode = stbuf->st_mode;
1099
    v9lstat->st_nlink = stbuf->st_nlink;
1100
    v9lstat->st_uid = stbuf->st_uid;
1101
    v9lstat->st_gid = stbuf->st_gid;
1102
    v9lstat->st_rdev = stbuf->st_rdev;
1103
    v9lstat->st_size = stbuf->st_size;
1104
    v9lstat->st_blksize = stbuf->st_blksize;
1105
    v9lstat->st_blocks = stbuf->st_blocks;
1106
    v9lstat->st_atime_sec = stbuf->st_atime;
1107
    v9lstat->st_atime_nsec = stbuf->st_atim.tv_nsec;
1108
    v9lstat->st_mtime_sec = stbuf->st_mtime;
1109
    v9lstat->st_mtime_nsec = stbuf->st_mtim.tv_nsec;
1110
    v9lstat->st_ctime_sec = stbuf->st_ctime;
1111
    v9lstat->st_ctime_nsec = stbuf->st_ctim.tv_nsec;
1112
    /* Currently we only support BASIC fields in stat */
1113
    v9lstat->st_result_mask = P9_STATS_BASIC;
1114

    
1115
    stat_to_qid(stbuf, &v9lstat->qid);
1116
}
1117

    
1118
static struct iovec *adjust_sg(struct iovec *sg, int len, int *iovcnt)
1119
{
1120
    while (len && *iovcnt) {
1121
        if (len < sg->iov_len) {
1122
            sg->iov_len -= len;
1123
            sg->iov_base += len;
1124
            len = 0;
1125
        } else {
1126
            len -= sg->iov_len;
1127
            sg++;
1128
            *iovcnt -= 1;
1129
        }
1130
    }
1131

    
1132
    return sg;
1133
}
1134

    
1135
static struct iovec *cap_sg(struct iovec *sg, int cap, int *cnt)
1136
{
1137
    int i;
1138
    int total = 0;
1139

    
1140
    for (i = 0; i < *cnt; i++) {
1141
        if ((total + sg[i].iov_len) > cap) {
1142
            sg[i].iov_len -= ((total + sg[i].iov_len) - cap);
1143
            i++;
1144
            break;
1145
        }
1146
        total += sg[i].iov_len;
1147
    }
1148

    
1149
    *cnt = i;
1150

    
1151
    return sg;
1152
}
1153

    
1154
static void print_sg(struct iovec *sg, int cnt)
1155
{
1156
    int i;
1157

    
1158
    printf("sg[%d]: {", cnt);
1159
    for (i = 0; i < cnt; i++) {
1160
        if (i) {
1161
            printf(", ");
1162
        }
1163
        printf("(%p, %zd)", sg[i].iov_base, sg[i].iov_len);
1164
    }
1165
    printf("}\n");
1166
}
1167

    
1168
static void v9fs_fix_path(V9fsString *dst, V9fsString *src, int len)
1169
{
1170
    V9fsString str;
1171
    v9fs_string_init(&str);
1172
    v9fs_string_copy(&str, dst);
1173
    v9fs_string_sprintf(dst, "%s%s", src->data, str.data+len);
1174
    v9fs_string_free(&str);
1175
}
1176

    
1177
static void v9fs_version(V9fsState *s, V9fsPDU *pdu)
1178
{
1179
    V9fsString version;
1180
    size_t offset = 7;
1181

    
1182
    pdu_unmarshal(pdu, offset, "ds", &s->msize, &version);
1183

    
1184
    if (!strcmp(version.data, "9P2000.u")) {
1185
        s->proto_version = V9FS_PROTO_2000U;
1186
    } else if (!strcmp(version.data, "9P2000.L")) {
1187
        s->proto_version = V9FS_PROTO_2000L;
1188
    } else {
1189
        v9fs_string_sprintf(&version, "unknown");
1190
    }
1191

    
1192
    offset += pdu_marshal(pdu, offset, "ds", s->msize, &version);
1193
    complete_pdu(s, pdu, offset);
1194

    
1195
    v9fs_string_free(&version);
1196
}
1197

    
1198
static void v9fs_attach(V9fsState *s, V9fsPDU *pdu)
1199
{
1200
    int32_t fid, afid, n_uname;
1201
    V9fsString uname, aname;
1202
    V9fsFidState *fidp;
1203
    V9fsQID qid;
1204
    size_t offset = 7;
1205
    ssize_t err;
1206

    
1207
    pdu_unmarshal(pdu, offset, "ddssd", &fid, &afid, &uname, &aname, &n_uname);
1208

    
1209
    fidp = alloc_fid(s, fid);
1210
    if (fidp == NULL) {
1211
        err = -EINVAL;
1212
        goto out;
1213
    }
1214

    
1215
    fidp->uid = n_uname;
1216

    
1217
    v9fs_string_sprintf(&fidp->path, "%s", "/");
1218
    err = fid_to_qid(s, fidp, &qid);
1219
    if (err) {
1220
        err = -EINVAL;
1221
        free_fid(s, fid);
1222
        goto out;
1223
    }
1224

    
1225
    offset += pdu_marshal(pdu, offset, "Q", &qid);
1226

    
1227
    err = offset;
1228
out:
1229
    complete_pdu(s, pdu, err);
1230
    v9fs_string_free(&uname);
1231
    v9fs_string_free(&aname);
1232
}
1233

    
1234
static void v9fs_stat_post_lstat(V9fsState *s, V9fsStatState *vs, int err)
1235
{
1236
    if (err == -1) {
1237
        err = -errno;
1238
        goto out;
1239
    }
1240

    
1241
    err = stat_to_v9stat(s, &vs->fidp->path, &vs->stbuf, &vs->v9stat);
1242
    if (err) {
1243
        goto out;
1244
    }
1245
    vs->offset += pdu_marshal(vs->pdu, vs->offset, "wS", 0, &vs->v9stat);
1246
    err = vs->offset;
1247

    
1248
out:
1249
    complete_pdu(s, vs->pdu, err);
1250
    v9fs_stat_free(&vs->v9stat);
1251
    qemu_free(vs);
1252
}
1253

    
1254
static void v9fs_stat(V9fsState *s, V9fsPDU *pdu)
1255
{
1256
    int32_t fid;
1257
    V9fsStatState *vs;
1258
    ssize_t err = 0;
1259

    
1260
    vs = qemu_malloc(sizeof(*vs));
1261
    vs->pdu = pdu;
1262
    vs->offset = 7;
1263

    
1264
    memset(&vs->v9stat, 0, sizeof(vs->v9stat));
1265

    
1266
    pdu_unmarshal(vs->pdu, vs->offset, "d", &fid);
1267

    
1268
    vs->fidp = lookup_fid(s, fid);
1269
    if (vs->fidp == NULL) {
1270
        err = -ENOENT;
1271
        goto out;
1272
    }
1273

    
1274
    err = v9fs_do_lstat(s, &vs->fidp->path, &vs->stbuf);
1275
    v9fs_stat_post_lstat(s, vs, err);
1276
    return;
1277

    
1278
out:
1279
    complete_pdu(s, vs->pdu, err);
1280
    v9fs_stat_free(&vs->v9stat);
1281
    qemu_free(vs);
1282
}
1283

    
1284
static void v9fs_getattr_post_lstat(V9fsState *s, V9fsStatStateDotl *vs,
1285
                                                                int err)
1286
{
1287
    if (err == -1) {
1288
        err = -errno;
1289
        goto out;
1290
    }
1291

    
1292
    stat_to_v9stat_dotl(s, &vs->stbuf, &vs->v9stat_dotl);
1293
    vs->offset += pdu_marshal(vs->pdu, vs->offset, "A", &vs->v9stat_dotl);
1294
    err = vs->offset;
1295

    
1296
out:
1297
    complete_pdu(s, vs->pdu, err);
1298
    qemu_free(vs);
1299
}
1300

    
1301
static void v9fs_getattr(V9fsState *s, V9fsPDU *pdu)
1302
{
1303
    int32_t fid;
1304
    V9fsStatStateDotl *vs;
1305
    ssize_t err = 0;
1306
    V9fsFidState *fidp;
1307
    uint64_t request_mask;
1308

    
1309
    vs = qemu_malloc(sizeof(*vs));
1310
    vs->pdu = pdu;
1311
    vs->offset = 7;
1312

    
1313
    memset(&vs->v9stat_dotl, 0, sizeof(vs->v9stat_dotl));
1314

    
1315
    pdu_unmarshal(vs->pdu, vs->offset, "dq", &fid, &request_mask);
1316

    
1317
    fidp = lookup_fid(s, fid);
1318
    if (fidp == NULL) {
1319
        err = -ENOENT;
1320
        goto out;
1321
    }
1322

    
1323
    /* Currently we only support BASIC fields in stat, so there is no
1324
     * need to look at request_mask.
1325
     */
1326
    err = v9fs_do_lstat(s, &fidp->path, &vs->stbuf);
1327
    v9fs_getattr_post_lstat(s, vs, err);
1328
    return;
1329

    
1330
out:
1331
    complete_pdu(s, vs->pdu, err);
1332
    qemu_free(vs);
1333
}
1334

    
1335
/* From Linux kernel code */
1336
#define ATTR_MODE    (1 << 0)
1337
#define ATTR_UID     (1 << 1)
1338
#define ATTR_GID     (1 << 2)
1339
#define ATTR_SIZE    (1 << 3)
1340
#define ATTR_ATIME   (1 << 4)
1341
#define ATTR_MTIME   (1 << 5)
1342
#define ATTR_CTIME   (1 << 6)
1343
#define ATTR_MASK    127
1344
#define ATTR_ATIME_SET  (1 << 7)
1345
#define ATTR_MTIME_SET  (1 << 8)
1346

    
1347
static void v9fs_setattr_post_truncate(V9fsState *s, V9fsSetattrState *vs,
1348
                                                                  int err)
1349
{
1350
    if (err == -1) {
1351
        err = -errno;
1352
        goto out;
1353
    }
1354
    err = vs->offset;
1355

    
1356
out:
1357
    complete_pdu(s, vs->pdu, err);
1358
    qemu_free(vs);
1359
}
1360

    
1361
static void v9fs_setattr_post_chown(V9fsState *s, V9fsSetattrState *vs, int err)
1362
{
1363
    if (err == -1) {
1364
        err = -errno;
1365
        goto out;
1366
    }
1367

    
1368
    if (vs->v9iattr.valid & (ATTR_SIZE)) {
1369
        err = v9fs_do_truncate(s, &vs->fidp->path, vs->v9iattr.size);
1370
    }
1371
    v9fs_setattr_post_truncate(s, vs, err);
1372
    return;
1373

    
1374
out:
1375
    complete_pdu(s, vs->pdu, err);
1376
    qemu_free(vs);
1377
}
1378

    
1379
static void v9fs_setattr_post_utimensat(V9fsState *s, V9fsSetattrState *vs,
1380
                                                                   int err)
1381
{
1382
    if (err == -1) {
1383
        err = -errno;
1384
        goto out;
1385
    }
1386

    
1387
    /* If the only valid entry in iattr is ctime we can call
1388
     * chown(-1,-1) to update the ctime of the file
1389
     */
1390
    if ((vs->v9iattr.valid & (ATTR_UID | ATTR_GID)) ||
1391
            ((vs->v9iattr.valid & ATTR_CTIME)
1392
            && !((vs->v9iattr.valid & ATTR_MASK) & ~ATTR_CTIME))) {
1393
        if (!(vs->v9iattr.valid & ATTR_UID)) {
1394
            vs->v9iattr.uid = -1;
1395
        }
1396
        if (!(vs->v9iattr.valid & ATTR_GID)) {
1397
            vs->v9iattr.gid = -1;
1398
        }
1399
        err = v9fs_do_chown(s, &vs->fidp->path, vs->v9iattr.uid,
1400
                                                vs->v9iattr.gid);
1401
    }
1402
    v9fs_setattr_post_chown(s, vs, err);
1403
    return;
1404

    
1405
out:
1406
    complete_pdu(s, vs->pdu, err);
1407
    qemu_free(vs);
1408
}
1409

    
1410
static void v9fs_setattr_post_chmod(V9fsState *s, V9fsSetattrState *vs, int err)
1411
{
1412
    if (err == -1) {
1413
        err = -errno;
1414
        goto out;
1415
    }
1416

    
1417
    if (vs->v9iattr.valid & (ATTR_ATIME | ATTR_MTIME)) {
1418
        struct timespec times[2];
1419
        if (vs->v9iattr.valid & ATTR_ATIME) {
1420
            if (vs->v9iattr.valid & ATTR_ATIME_SET) {
1421
                times[0].tv_sec = vs->v9iattr.atime_sec;
1422
                times[0].tv_nsec = vs->v9iattr.atime_nsec;
1423
            } else {
1424
                times[0].tv_nsec = UTIME_NOW;
1425
            }
1426
        } else {
1427
            times[0].tv_nsec = UTIME_OMIT;
1428
        }
1429

    
1430
        if (vs->v9iattr.valid & ATTR_MTIME) {
1431
            if (vs->v9iattr.valid & ATTR_MTIME_SET) {
1432
                times[1].tv_sec = vs->v9iattr.mtime_sec;
1433
                times[1].tv_nsec = vs->v9iattr.mtime_nsec;
1434
            } else {
1435
                times[1].tv_nsec = UTIME_NOW;
1436
            }
1437
        } else {
1438
            times[1].tv_nsec = UTIME_OMIT;
1439
        }
1440
        err = v9fs_do_utimensat(s, &vs->fidp->path, times);
1441
    }
1442
    v9fs_setattr_post_utimensat(s, vs, err);
1443
    return;
1444

    
1445
out:
1446
    complete_pdu(s, vs->pdu, err);
1447
    qemu_free(vs);
1448
}
1449

    
1450
static void v9fs_setattr(V9fsState *s, V9fsPDU *pdu)
1451
{
1452
    int32_t fid;
1453
    V9fsSetattrState *vs;
1454
    int err = 0;
1455

    
1456
    vs = qemu_malloc(sizeof(*vs));
1457
    vs->pdu = pdu;
1458
    vs->offset = 7;
1459

    
1460
    pdu_unmarshal(pdu, vs->offset, "dI", &fid, &vs->v9iattr);
1461

    
1462
    vs->fidp = lookup_fid(s, fid);
1463
    if (vs->fidp == NULL) {
1464
        err = -EINVAL;
1465
        goto out;
1466
    }
1467

    
1468
    if (vs->v9iattr.valid & ATTR_MODE) {
1469
        err = v9fs_do_chmod(s, &vs->fidp->path, vs->v9iattr.mode);
1470
    }
1471

    
1472
    v9fs_setattr_post_chmod(s, vs, err);
1473
    return;
1474

    
1475
out:
1476
    complete_pdu(s, vs->pdu, err);
1477
    qemu_free(vs);
1478
}
1479

    
1480
static void v9fs_walk_complete(V9fsState *s, V9fsWalkState *vs, int err)
1481
{
1482
    complete_pdu(s, vs->pdu, err);
1483

    
1484
    if (vs->nwnames && vs->nwnames <= P9_MAXWELEM) {
1485
        for (vs->name_idx = 0; vs->name_idx < vs->nwnames; vs->name_idx++) {
1486
            v9fs_string_free(&vs->wnames[vs->name_idx]);
1487
        }
1488

    
1489
        qemu_free(vs->wnames);
1490
        qemu_free(vs->qids);
1491
    }
1492
}
1493

    
1494
static void v9fs_walk_marshal(V9fsWalkState *vs)
1495
{
1496
    int i;
1497
    vs->offset = 7;
1498
    vs->offset += pdu_marshal(vs->pdu, vs->offset, "w", vs->nwnames);
1499

    
1500
    for (i = 0; i < vs->nwnames; i++) {
1501
        vs->offset += pdu_marshal(vs->pdu, vs->offset, "Q", &vs->qids[i]);
1502
    }
1503
}
1504

    
1505
static void v9fs_walk_post_newfid_lstat(V9fsState *s, V9fsWalkState *vs,
1506
                                                                int err)
1507
{
1508
    if (err == -1) {
1509
        free_fid(s, vs->newfidp->fid);
1510
        v9fs_string_free(&vs->path);
1511
        err = -ENOENT;
1512
        goto out;
1513
    }
1514

    
1515
    stat_to_qid(&vs->stbuf, &vs->qids[vs->name_idx]);
1516

    
1517
    vs->name_idx++;
1518
    if (vs->name_idx < vs->nwnames) {
1519
        v9fs_string_sprintf(&vs->path, "%s/%s", vs->newfidp->path.data,
1520
                                            vs->wnames[vs->name_idx].data);
1521
        v9fs_string_copy(&vs->newfidp->path, &vs->path);
1522

    
1523
        err = v9fs_do_lstat(s, &vs->newfidp->path, &vs->stbuf);
1524
        v9fs_walk_post_newfid_lstat(s, vs, err);
1525
        return;
1526
    }
1527

    
1528
    v9fs_string_free(&vs->path);
1529
    v9fs_walk_marshal(vs);
1530
    err = vs->offset;
1531
out:
1532
    v9fs_walk_complete(s, vs, err);
1533
}
1534

    
1535
static void v9fs_walk_post_oldfid_lstat(V9fsState *s, V9fsWalkState *vs,
1536
        int err)
1537
{
1538
    if (err == -1) {
1539
        v9fs_string_free(&vs->path);
1540
        err = -ENOENT;
1541
        goto out;
1542
    }
1543

    
1544
    stat_to_qid(&vs->stbuf, &vs->qids[vs->name_idx]);
1545
    vs->name_idx++;
1546
    if (vs->name_idx < vs->nwnames) {
1547

    
1548
        v9fs_string_sprintf(&vs->path, "%s/%s",
1549
                vs->fidp->path.data, vs->wnames[vs->name_idx].data);
1550
        v9fs_string_copy(&vs->fidp->path, &vs->path);
1551

    
1552
        err = v9fs_do_lstat(s, &vs->fidp->path, &vs->stbuf);
1553
        v9fs_walk_post_oldfid_lstat(s, vs, err);
1554
        return;
1555
    }
1556

    
1557
    v9fs_string_free(&vs->path);
1558
    v9fs_walk_marshal(vs);
1559
    err = vs->offset;
1560
out:
1561
    v9fs_walk_complete(s, vs, err);
1562
}
1563

    
1564
static void v9fs_walk(V9fsState *s, V9fsPDU *pdu)
1565
{
1566
    int32_t fid, newfid;
1567
    V9fsWalkState *vs;
1568
    int err = 0;
1569
    int i;
1570

    
1571
    vs = qemu_malloc(sizeof(*vs));
1572
    vs->pdu = pdu;
1573
    vs->wnames = NULL;
1574
    vs->qids = NULL;
1575
    vs->offset = 7;
1576

    
1577
    vs->offset += pdu_unmarshal(vs->pdu, vs->offset, "ddw", &fid,
1578
                                            &newfid, &vs->nwnames);
1579

    
1580
    if (vs->nwnames && vs->nwnames <= P9_MAXWELEM) {
1581
        vs->wnames = qemu_mallocz(sizeof(vs->wnames[0]) * vs->nwnames);
1582

    
1583
        vs->qids = qemu_mallocz(sizeof(vs->qids[0]) * vs->nwnames);
1584

    
1585
        for (i = 0; i < vs->nwnames; i++) {
1586
            vs->offset += pdu_unmarshal(vs->pdu, vs->offset, "s",
1587
                                            &vs->wnames[i]);
1588
        }
1589
    } else if (vs->nwnames > P9_MAXWELEM) {
1590
        err = -EINVAL;
1591
        goto out;
1592
    }
1593

    
1594
    vs->fidp = lookup_fid(s, fid);
1595
    if (vs->fidp == NULL) {
1596
        err = -ENOENT;
1597
        goto out;
1598
    }
1599

    
1600
    /* FIXME: is this really valid? */
1601
    if (fid == newfid) {
1602

    
1603
        BUG_ON(vs->fidp->fid_type != P9_FID_NONE);
1604
        v9fs_string_init(&vs->path);
1605
        vs->name_idx = 0;
1606

    
1607
        if (vs->name_idx < vs->nwnames) {
1608
            v9fs_string_sprintf(&vs->path, "%s/%s",
1609
                vs->fidp->path.data, vs->wnames[vs->name_idx].data);
1610
            v9fs_string_copy(&vs->fidp->path, &vs->path);
1611

    
1612
            err = v9fs_do_lstat(s, &vs->fidp->path, &vs->stbuf);
1613
            v9fs_walk_post_oldfid_lstat(s, vs, err);
1614
            return;
1615
        }
1616
    } else {
1617
        vs->newfidp = alloc_fid(s, newfid);
1618
        if (vs->newfidp == NULL) {
1619
            err = -EINVAL;
1620
            goto out;
1621
        }
1622

    
1623
        vs->newfidp->uid = vs->fidp->uid;
1624
        v9fs_string_init(&vs->path);
1625
        vs->name_idx = 0;
1626
        v9fs_string_copy(&vs->newfidp->path, &vs->fidp->path);
1627

    
1628
        if (vs->name_idx < vs->nwnames) {
1629
            v9fs_string_sprintf(&vs->path, "%s/%s", vs->newfidp->path.data,
1630
                                vs->wnames[vs->name_idx].data);
1631
            v9fs_string_copy(&vs->newfidp->path, &vs->path);
1632

    
1633
            err = v9fs_do_lstat(s, &vs->newfidp->path, &vs->stbuf);
1634
            v9fs_walk_post_newfid_lstat(s, vs, err);
1635
            return;
1636
        }
1637
    }
1638

    
1639
    v9fs_walk_marshal(vs);
1640
    err = vs->offset;
1641
out:
1642
    v9fs_walk_complete(s, vs, err);
1643
}
1644

    
1645
static int32_t get_iounit(V9fsState *s, V9fsString *name)
1646
{
1647
    struct statfs stbuf;
1648
    int32_t iounit = 0;
1649

    
1650
    /*
1651
     * iounit should be multiples of f_bsize (host filesystem block size
1652
     * and as well as less than (client msize - P9_IOHDRSZ))
1653
     */
1654
    if (!v9fs_do_statfs(s, name, &stbuf)) {
1655
        iounit = stbuf.f_bsize;
1656
        iounit *= (s->msize - P9_IOHDRSZ)/stbuf.f_bsize;
1657
    }
1658

    
1659
    if (!iounit) {
1660
        iounit = s->msize - P9_IOHDRSZ;
1661
    }
1662
    return iounit;
1663
}
1664

    
1665
static void v9fs_open_post_opendir(V9fsState *s, V9fsOpenState *vs, int err)
1666
{
1667
    if (vs->fidp->fs.dir == NULL) {
1668
        err = -errno;
1669
        goto out;
1670
    }
1671
    vs->fidp->fid_type = P9_FID_DIR;
1672
    vs->offset += pdu_marshal(vs->pdu, vs->offset, "Qd", &vs->qid, 0);
1673
    err = vs->offset;
1674
out:
1675
    complete_pdu(s, vs->pdu, err);
1676
    qemu_free(vs);
1677

    
1678
}
1679

    
1680
static void v9fs_open_post_getiounit(V9fsState *s, V9fsOpenState *vs)
1681
{
1682
    int err;
1683
    vs->offset += pdu_marshal(vs->pdu, vs->offset, "Qd", &vs->qid, vs->iounit);
1684
    err = vs->offset;
1685
    complete_pdu(s, vs->pdu, err);
1686
    qemu_free(vs);
1687
}
1688

    
1689
static void v9fs_open_post_open(V9fsState *s, V9fsOpenState *vs, int err)
1690
{
1691
    if (vs->fidp->fs.fd == -1) {
1692
        err = -errno;
1693
        goto out;
1694
    }
1695
    vs->fidp->fid_type = P9_FID_FILE;
1696
    vs->iounit = get_iounit(s, &vs->fidp->path);
1697
    v9fs_open_post_getiounit(s, vs);
1698
    return;
1699
out:
1700
    complete_pdu(s, vs->pdu, err);
1701
    qemu_free(vs);
1702
}
1703

    
1704
static void v9fs_open_post_lstat(V9fsState *s, V9fsOpenState *vs, int err)
1705
{
1706
    int flags;
1707

    
1708
    if (err) {
1709
        err = -errno;
1710
        goto out;
1711
    }
1712

    
1713
    stat_to_qid(&vs->stbuf, &vs->qid);
1714

    
1715
    if (S_ISDIR(vs->stbuf.st_mode)) {
1716
        vs->fidp->fs.dir = v9fs_do_opendir(s, &vs->fidp->path);
1717
        v9fs_open_post_opendir(s, vs, err);
1718
    } else {
1719
        if (s->proto_version == V9FS_PROTO_2000L) {
1720
            flags = vs->mode;
1721
            flags &= ~(O_NOCTTY | O_ASYNC | O_CREAT);
1722
            /* Ignore direct disk access hint until the server supports it. */
1723
            flags &= ~O_DIRECT;
1724
        } else {
1725
            flags = omode_to_uflags(vs->mode);
1726
        }
1727
        vs->fidp->fs.fd = v9fs_do_open(s, &vs->fidp->path, flags);
1728
        v9fs_open_post_open(s, vs, err);
1729
    }
1730
    return;
1731
out:
1732
    complete_pdu(s, vs->pdu, err);
1733
    qemu_free(vs);
1734
}
1735

    
1736
static void v9fs_open(V9fsState *s, V9fsPDU *pdu)
1737
{
1738
    int32_t fid;
1739
    V9fsOpenState *vs;
1740
    ssize_t err = 0;
1741

    
1742
    vs = qemu_malloc(sizeof(*vs));
1743
    vs->pdu = pdu;
1744
    vs->offset = 7;
1745
    vs->mode = 0;
1746

    
1747
    if (s->proto_version == V9FS_PROTO_2000L) {
1748
        pdu_unmarshal(vs->pdu, vs->offset, "dd", &fid, &vs->mode);
1749
    } else {
1750
        pdu_unmarshal(vs->pdu, vs->offset, "db", &fid, &vs->mode);
1751
    }
1752

    
1753
    vs->fidp = lookup_fid(s, fid);
1754
    if (vs->fidp == NULL) {
1755
        err = -ENOENT;
1756
        goto out;
1757
    }
1758

    
1759
    BUG_ON(vs->fidp->fid_type != P9_FID_NONE);
1760

    
1761
    err = v9fs_do_lstat(s, &vs->fidp->path, &vs->stbuf);
1762

    
1763
    v9fs_open_post_lstat(s, vs, err);
1764
    return;
1765
out:
1766
    complete_pdu(s, pdu, err);
1767
    qemu_free(vs);
1768
}
1769

    
1770
static void v9fs_post_lcreate(V9fsState *s, V9fsLcreateState *vs, int err)
1771
{
1772
    if (err == 0) {
1773
        v9fs_string_copy(&vs->fidp->path, &vs->fullname);
1774
        stat_to_qid(&vs->stbuf, &vs->qid);
1775
        vs->offset += pdu_marshal(vs->pdu, vs->offset, "Qd", &vs->qid,
1776
                vs->iounit);
1777
        err = vs->offset;
1778
    } else {
1779
        vs->fidp->fid_type = P9_FID_NONE;
1780
        err = -errno;
1781
        if (vs->fidp->fs.fd > 0) {
1782
            close(vs->fidp->fs.fd);
1783
        }
1784
    }
1785

    
1786
    complete_pdu(s, vs->pdu, err);
1787
    v9fs_string_free(&vs->name);
1788
    v9fs_string_free(&vs->fullname);
1789
    qemu_free(vs);
1790
}
1791

    
1792
static void v9fs_lcreate_post_get_iounit(V9fsState *s, V9fsLcreateState *vs,
1793
        int err)
1794
{
1795
    if (err) {
1796
        err = -errno;
1797
        goto out;
1798
    }
1799
    err = v9fs_do_lstat(s, &vs->fullname, &vs->stbuf);
1800

    
1801
out:
1802
    v9fs_post_lcreate(s, vs, err);
1803
}
1804

    
1805
static void v9fs_lcreate_post_do_open2(V9fsState *s, V9fsLcreateState *vs,
1806
        int err)
1807
{
1808
    if (vs->fidp->fs.fd == -1) {
1809
        err = -errno;
1810
        goto out;
1811
    }
1812
    vs->fidp->fid_type = P9_FID_FILE;
1813
    vs->iounit =  get_iounit(s, &vs->fullname);
1814
    v9fs_lcreate_post_get_iounit(s, vs, err);
1815
    return;
1816

    
1817
out:
1818
    v9fs_post_lcreate(s, vs, err);
1819
}
1820

    
1821
static void v9fs_lcreate(V9fsState *s, V9fsPDU *pdu)
1822
{
1823
    int32_t dfid, flags, mode;
1824
    gid_t gid;
1825
    V9fsLcreateState *vs;
1826
    ssize_t err = 0;
1827

    
1828
    vs = qemu_malloc(sizeof(*vs));
1829
    vs->pdu = pdu;
1830
    vs->offset = 7;
1831

    
1832
    v9fs_string_init(&vs->fullname);
1833

    
1834
    pdu_unmarshal(vs->pdu, vs->offset, "dsddd", &dfid, &vs->name, &flags,
1835
            &mode, &gid);
1836

    
1837
    vs->fidp = lookup_fid(s, dfid);
1838
    if (vs->fidp == NULL) {
1839
        err = -ENOENT;
1840
        goto out;
1841
    }
1842

    
1843
    v9fs_string_sprintf(&vs->fullname, "%s/%s", vs->fidp->path.data,
1844
             vs->name.data);
1845

    
1846
    /* Ignore direct disk access hint until the server supports it. */
1847
    flags &= ~O_DIRECT;
1848

    
1849
    vs->fidp->fs.fd = v9fs_do_open2(s, vs->fullname.data, vs->fidp->uid,
1850
            gid, flags, mode);
1851
    v9fs_lcreate_post_do_open2(s, vs, err);
1852
    return;
1853

    
1854
out:
1855
    complete_pdu(s, vs->pdu, err);
1856
    v9fs_string_free(&vs->name);
1857
    qemu_free(vs);
1858
}
1859

    
1860
static void v9fs_post_do_fsync(V9fsState *s, V9fsPDU *pdu, int err)
1861
{
1862
    if (err == -1) {
1863
        err = -errno;
1864
    }
1865
    complete_pdu(s, pdu, err);
1866
}
1867

    
1868
static void v9fs_fsync(V9fsState *s, V9fsPDU *pdu)
1869
{
1870
    int32_t fid;
1871
    size_t offset = 7;
1872
    V9fsFidState *fidp;
1873
    int datasync;
1874
    int err;
1875

    
1876
    pdu_unmarshal(pdu, offset, "dd", &fid, &datasync);
1877
    fidp = lookup_fid(s, fid);
1878
    if (fidp == NULL) {
1879
        err = -ENOENT;
1880
        v9fs_post_do_fsync(s, pdu, err);
1881
        return;
1882
    }
1883
    err = v9fs_do_fsync(s, fidp->fs.fd, datasync);
1884
    v9fs_post_do_fsync(s, pdu, err);
1885
}
1886

    
1887
static void v9fs_clunk(V9fsState *s, V9fsPDU *pdu)
1888
{
1889
    int32_t fid;
1890
    size_t offset = 7;
1891
    int err;
1892

    
1893
    pdu_unmarshal(pdu, offset, "d", &fid);
1894

    
1895
    err = free_fid(s, fid);
1896
    if (err < 0) {
1897
        goto out;
1898
    }
1899

    
1900
    offset = 7;
1901
    err = offset;
1902
out:
1903
    complete_pdu(s, pdu, err);
1904
}
1905

    
1906
static void v9fs_read_post_readdir(V9fsState *, V9fsReadState *, ssize_t);
1907

    
1908
static void v9fs_read_post_seekdir(V9fsState *s, V9fsReadState *vs, ssize_t err)
1909
{
1910
    if (err) {
1911
        goto out;
1912
    }
1913
    v9fs_stat_free(&vs->v9stat);
1914
    v9fs_string_free(&vs->name);
1915
    vs->offset += pdu_marshal(vs->pdu, vs->offset, "d", vs->count);
1916
    vs->offset += vs->count;
1917
    err = vs->offset;
1918
out:
1919
    complete_pdu(s, vs->pdu, err);
1920
    qemu_free(vs);
1921
    return;
1922
}
1923

    
1924
static void v9fs_read_post_dir_lstat(V9fsState *s, V9fsReadState *vs,
1925
                                    ssize_t err)
1926
{
1927
    if (err) {
1928
        err = -errno;
1929
        goto out;
1930
    }
1931
    err = stat_to_v9stat(s, &vs->name, &vs->stbuf, &vs->v9stat);
1932
    if (err) {
1933
        goto out;
1934
    }
1935

    
1936
    vs->len = pdu_marshal(vs->pdu, vs->offset + 4 + vs->count, "S",
1937
                            &vs->v9stat);
1938
    if ((vs->len != (vs->v9stat.size + 2)) ||
1939
            ((vs->count + vs->len) > vs->max_count)) {
1940
        v9fs_do_seekdir(s, vs->fidp->fs.dir, vs->dir_pos);
1941
        v9fs_read_post_seekdir(s, vs, err);
1942
        return;
1943
    }
1944
    vs->count += vs->len;
1945
    v9fs_stat_free(&vs->v9stat);
1946
    v9fs_string_free(&vs->name);
1947
    vs->dir_pos = vs->dent->d_off;
1948
    vs->dent = v9fs_do_readdir(s, vs->fidp->fs.dir);
1949
    v9fs_read_post_readdir(s, vs, err);
1950
    return;
1951
out:
1952
    v9fs_do_seekdir(s, vs->fidp->fs.dir, vs->dir_pos);
1953
    v9fs_read_post_seekdir(s, vs, err);
1954
    return;
1955

    
1956
}
1957

    
1958
static void v9fs_read_post_readdir(V9fsState *s, V9fsReadState *vs, ssize_t err)
1959
{
1960
    if (vs->dent) {
1961
        memset(&vs->v9stat, 0, sizeof(vs->v9stat));
1962
        v9fs_string_init(&vs->name);
1963
        v9fs_string_sprintf(&vs->name, "%s/%s", vs->fidp->path.data,
1964
                            vs->dent->d_name);
1965
        err = v9fs_do_lstat(s, &vs->name, &vs->stbuf);
1966
        v9fs_read_post_dir_lstat(s, vs, err);
1967
        return;
1968
    }
1969

    
1970
    vs->offset += pdu_marshal(vs->pdu, vs->offset, "d", vs->count);
1971
    vs->offset += vs->count;
1972
    err = vs->offset;
1973
    complete_pdu(s, vs->pdu, err);
1974
    qemu_free(vs);
1975
    return;
1976
}
1977

    
1978
static void v9fs_read_post_telldir(V9fsState *s, V9fsReadState *vs, ssize_t err)
1979
{
1980
    vs->dent = v9fs_do_readdir(s, vs->fidp->fs.dir);
1981
    v9fs_read_post_readdir(s, vs, err);
1982
    return;
1983
}
1984

    
1985
static void v9fs_read_post_rewinddir(V9fsState *s, V9fsReadState *vs,
1986
                                       ssize_t err)
1987
{
1988
    vs->dir_pos = v9fs_do_telldir(s, vs->fidp->fs.dir);
1989
    v9fs_read_post_telldir(s, vs, err);
1990
    return;
1991
}
1992

    
1993
static void v9fs_read_post_preadv(V9fsState *s, V9fsReadState *vs, ssize_t err)
1994
{
1995
    if (err  < 0) {
1996
        /* IO error return the error */
1997
        err = -errno;
1998
        goto out;
1999
    }
2000
    vs->total += vs->len;
2001
    vs->sg = adjust_sg(vs->sg, vs->len, &vs->cnt);
2002
    if (vs->total < vs->count && vs->len > 0) {
2003
        do {
2004
            if (0) {
2005
                print_sg(vs->sg, vs->cnt);
2006
            }
2007
            vs->len = v9fs_do_preadv(s, vs->fidp->fs.fd, vs->sg, vs->cnt,
2008
                      vs->off);
2009
            if (vs->len > 0) {
2010
                vs->off += vs->len;
2011
            }
2012
        } while (vs->len == -1 && errno == EINTR);
2013
        if (vs->len == -1) {
2014
            err  = -errno;
2015
        }
2016
        v9fs_read_post_preadv(s, vs, err);
2017
        return;
2018
    }
2019
    vs->offset += pdu_marshal(vs->pdu, vs->offset, "d", vs->total);
2020
    vs->offset += vs->count;
2021
    err = vs->offset;
2022

    
2023
out:
2024
    complete_pdu(s, vs->pdu, err);
2025
    qemu_free(vs);
2026
}
2027

    
2028
static void v9fs_xattr_read(V9fsState *s, V9fsReadState *vs)
2029
{
2030
    ssize_t err = 0;
2031
    int read_count;
2032
    int64_t xattr_len;
2033

    
2034
    xattr_len = vs->fidp->fs.xattr.len;
2035
    read_count = xattr_len - vs->off;
2036
    if (read_count > vs->count) {
2037
        read_count = vs->count;
2038
    } else if (read_count < 0) {
2039
        /*
2040
         * read beyond XATTR value
2041
         */
2042
        read_count = 0;
2043
    }
2044
    vs->offset += pdu_marshal(vs->pdu, vs->offset, "d", read_count);
2045
    vs->offset += pdu_pack(vs->pdu, vs->offset,
2046
                           ((char *)vs->fidp->fs.xattr.value) + vs->off,
2047
                           read_count);
2048
    err = vs->offset;
2049
    complete_pdu(s, vs->pdu, err);
2050
    qemu_free(vs);
2051
}
2052

    
2053
static void v9fs_read(V9fsState *s, V9fsPDU *pdu)
2054
{
2055
    int32_t fid;
2056
    V9fsReadState *vs;
2057
    ssize_t err = 0;
2058

    
2059
    vs = qemu_malloc(sizeof(*vs));
2060
    vs->pdu = pdu;
2061
    vs->offset = 7;
2062
    vs->total = 0;
2063
    vs->len = 0;
2064
    vs->count = 0;
2065

    
2066
    pdu_unmarshal(vs->pdu, vs->offset, "dqd", &fid, &vs->off, &vs->count);
2067

    
2068
    vs->fidp = lookup_fid(s, fid);
2069
    if (vs->fidp == NULL) {
2070
        err = -EINVAL;
2071
        goto out;
2072
    }
2073

    
2074
    if (vs->fidp->fid_type == P9_FID_DIR) {
2075
        vs->max_count = vs->count;
2076
        vs->count = 0;
2077
        if (vs->off == 0) {
2078
            v9fs_do_rewinddir(s, vs->fidp->fs.dir);
2079
        }
2080
        v9fs_read_post_rewinddir(s, vs, err);
2081
        return;
2082
    } else if (vs->fidp->fid_type == P9_FID_FILE) {
2083
        vs->sg = vs->iov;
2084
        pdu_marshal(vs->pdu, vs->offset + 4, "v", vs->sg, &vs->cnt);
2085
        vs->sg = cap_sg(vs->sg, vs->count, &vs->cnt);
2086
        if (vs->total <= vs->count) {
2087
            vs->len = v9fs_do_preadv(s, vs->fidp->fs.fd, vs->sg, vs->cnt,
2088
                                    vs->off);
2089
            if (vs->len > 0) {
2090
                vs->off += vs->len;
2091
            }
2092
            err = vs->len;
2093
            v9fs_read_post_preadv(s, vs, err);
2094
        }
2095
        return;
2096
    } else if (vs->fidp->fid_type == P9_FID_XATTR) {
2097
        v9fs_xattr_read(s, vs);
2098
        return;
2099
    } else {
2100
        err = -EINVAL;
2101
    }
2102
out:
2103
    complete_pdu(s, pdu, err);
2104
    qemu_free(vs);
2105
}
2106

    
2107
typedef struct V9fsReadDirState {
2108
    V9fsPDU *pdu;
2109
    V9fsFidState *fidp;
2110
    V9fsQID qid;
2111
    off_t saved_dir_pos;
2112
    struct dirent *dent;
2113
    int32_t count;
2114
    int32_t max_count;
2115
    size_t offset;
2116
    int64_t initial_offset;
2117
    V9fsString name;
2118
} V9fsReadDirState;
2119

    
2120
static void v9fs_readdir_post_seekdir(V9fsState *s, V9fsReadDirState *vs)
2121
{
2122
    vs->offset += pdu_marshal(vs->pdu, vs->offset, "d", vs->count);
2123
    vs->offset += vs->count;
2124
    complete_pdu(s, vs->pdu, vs->offset);
2125
    qemu_free(vs);
2126
    return;
2127
}
2128

    
2129
/* Size of each dirent on the wire: size of qid (13) + size of offset (8)
2130
 * size of type (1) + size of name.size (2) + strlen(name.data)
2131
 */
2132
#define V9_READDIR_DATA_SZ (24 + strlen(vs->name.data))
2133

    
2134
static void v9fs_readdir_post_readdir(V9fsState *s, V9fsReadDirState *vs)
2135
{
2136
    int len;
2137
    size_t size;
2138

    
2139
    if (vs->dent) {
2140
        v9fs_string_init(&vs->name);
2141
        v9fs_string_sprintf(&vs->name, "%s", vs->dent->d_name);
2142

    
2143
        if ((vs->count + V9_READDIR_DATA_SZ) > vs->max_count) {
2144
            /* Ran out of buffer. Set dir back to old position and return */
2145
            v9fs_do_seekdir(s, vs->fidp->fs.dir, vs->saved_dir_pos);
2146
            v9fs_readdir_post_seekdir(s, vs);
2147
            return;
2148
        }
2149

    
2150
        /* Fill up just the path field of qid because the client uses
2151
         * only that. To fill the entire qid structure we will have
2152
         * to stat each dirent found, which is expensive
2153
         */
2154
        size = MIN(sizeof(vs->dent->d_ino), sizeof(vs->qid.path));
2155
        memcpy(&vs->qid.path, &vs->dent->d_ino, size);
2156
        /* Fill the other fields with dummy values */
2157
        vs->qid.type = 0;
2158
        vs->qid.version = 0;
2159

    
2160
        len = pdu_marshal(vs->pdu, vs->offset+4+vs->count, "Qqbs",
2161
                              &vs->qid, vs->dent->d_off,
2162
                              vs->dent->d_type, &vs->name);
2163
        vs->count += len;
2164
        v9fs_string_free(&vs->name);
2165
        vs->saved_dir_pos = vs->dent->d_off;
2166
        vs->dent = v9fs_do_readdir(s, vs->fidp->fs.dir);
2167
        v9fs_readdir_post_readdir(s, vs);
2168
        return;
2169
    }
2170

    
2171
    vs->offset += pdu_marshal(vs->pdu, vs->offset, "d", vs->count);
2172
    vs->offset += vs->count;
2173
    complete_pdu(s, vs->pdu, vs->offset);
2174
    qemu_free(vs);
2175
    return;
2176
}
2177

    
2178
static void v9fs_readdir_post_telldir(V9fsState *s, V9fsReadDirState *vs)
2179
{
2180
    vs->dent = v9fs_do_readdir(s, vs->fidp->fs.dir);
2181
    v9fs_readdir_post_readdir(s, vs);
2182
    return;
2183
}
2184

    
2185
static void v9fs_readdir_post_setdir(V9fsState *s, V9fsReadDirState *vs)
2186
{
2187
    vs->saved_dir_pos = v9fs_do_telldir(s, vs->fidp->fs.dir);
2188
    v9fs_readdir_post_telldir(s, vs);
2189
    return;
2190
}
2191

    
2192
static void v9fs_readdir(V9fsState *s, V9fsPDU *pdu)
2193
{
2194
    int32_t fid;
2195
    V9fsReadDirState *vs;
2196
    ssize_t err = 0;
2197
    size_t offset = 7;
2198

    
2199
    vs = qemu_malloc(sizeof(*vs));
2200
    vs->pdu = pdu;
2201
    vs->offset = 7;
2202
    vs->count = 0;
2203

    
2204
    pdu_unmarshal(vs->pdu, offset, "dqd", &fid, &vs->initial_offset,
2205
                                                        &vs->max_count);
2206

    
2207
    vs->fidp = lookup_fid(s, fid);
2208
    if (vs->fidp == NULL || !(vs->fidp->fs.dir)) {
2209
        err = -EINVAL;
2210
        goto out;
2211
    }
2212

    
2213
    if (vs->initial_offset == 0) {
2214
        v9fs_do_rewinddir(s, vs->fidp->fs.dir);
2215
    } else {
2216
        v9fs_do_seekdir(s, vs->fidp->fs.dir, vs->initial_offset);
2217
    }
2218

    
2219
    v9fs_readdir_post_setdir(s, vs);
2220
    return;
2221

    
2222
out:
2223
    complete_pdu(s, pdu, err);
2224
    qemu_free(vs);
2225
    return;
2226
}
2227

    
2228
static void v9fs_write_post_pwritev(V9fsState *s, V9fsWriteState *vs,
2229
                                   ssize_t err)
2230
{
2231
    if (err  < 0) {
2232
        /* IO error return the error */
2233
        err = -errno;
2234
        goto out;
2235
    }
2236
    vs->total += vs->len;
2237
    vs->sg = adjust_sg(vs->sg, vs->len, &vs->cnt);
2238
    if (vs->total < vs->count && vs->len > 0) {
2239
        do {
2240
            if (0) {
2241
                print_sg(vs->sg, vs->cnt);
2242
            }
2243
            vs->len = v9fs_do_pwritev(s, vs->fidp->fs.fd, vs->sg, vs->cnt,
2244
                      vs->off);
2245
            if (vs->len > 0) {
2246
                vs->off += vs->len;
2247
            }
2248
        } while (vs->len == -1 && errno == EINTR);
2249
        if (vs->len == -1) {
2250
            err  = -errno;
2251
        }
2252
        v9fs_write_post_pwritev(s, vs, err);
2253
        return;
2254
    }
2255
    vs->offset += pdu_marshal(vs->pdu, vs->offset, "d", vs->total);
2256
    err = vs->offset;
2257
out:
2258
    complete_pdu(s, vs->pdu, err);
2259
    qemu_free(vs);
2260
}
2261

    
2262
static void v9fs_xattr_write(V9fsState *s, V9fsWriteState *vs)
2263
{
2264
    int i, to_copy;
2265
    ssize_t err = 0;
2266
    int write_count;
2267
    int64_t xattr_len;
2268

    
2269
    xattr_len = vs->fidp->fs.xattr.len;
2270
    write_count = xattr_len - vs->off;
2271
    if (write_count > vs->count) {
2272
        write_count = vs->count;
2273
    } else if (write_count < 0) {
2274
        /*
2275
         * write beyond XATTR value len specified in
2276
         * xattrcreate
2277
         */
2278
        err = -ENOSPC;
2279
        goto out;
2280
    }
2281
    vs->offset += pdu_marshal(vs->pdu, vs->offset, "d", write_count);
2282
    err = vs->offset;
2283
    vs->fidp->fs.xattr.copied_len += write_count;
2284
    /*
2285
     * Now copy the content from sg list
2286
     */
2287
    for (i = 0; i < vs->cnt; i++) {
2288
        if (write_count > vs->sg[i].iov_len) {
2289
            to_copy = vs->sg[i].iov_len;
2290
        } else {
2291
            to_copy = write_count;
2292
        }
2293
        memcpy((char *)vs->fidp->fs.xattr.value + vs->off,
2294
               vs->sg[i].iov_base, to_copy);
2295
        /* updating vs->off since we are not using below */
2296
        vs->off += to_copy;
2297
        write_count -= to_copy;
2298
    }
2299
out:
2300
    complete_pdu(s, vs->pdu, err);
2301
    qemu_free(vs);
2302
}
2303

    
2304
static void v9fs_write(V9fsState *s, V9fsPDU *pdu)
2305
{
2306
    int32_t fid;
2307
    V9fsWriteState *vs;
2308
    ssize_t err;
2309

    
2310
    vs = qemu_malloc(sizeof(*vs));
2311

    
2312
    vs->pdu = pdu;
2313
    vs->offset = 7;
2314
    vs->sg = vs->iov;
2315
    vs->total = 0;
2316
    vs->len = 0;
2317

    
2318
    pdu_unmarshal(vs->pdu, vs->offset, "dqdv", &fid, &vs->off, &vs->count,
2319
                  vs->sg, &vs->cnt);
2320

    
2321
    vs->fidp = lookup_fid(s, fid);
2322
    if (vs->fidp == NULL) {
2323
        err = -EINVAL;
2324
        goto out;
2325
    }
2326

    
2327
    if (vs->fidp->fid_type == P9_FID_FILE) {
2328
        if (vs->fidp->fs.fd == -1) {
2329
            err = -EINVAL;
2330
            goto out;
2331
        }
2332
    } else if (vs->fidp->fid_type == P9_FID_XATTR) {
2333
        /*
2334
         * setxattr operation
2335
         */
2336
        v9fs_xattr_write(s, vs);
2337
        return;
2338
    } else {
2339
        err = -EINVAL;
2340
        goto out;
2341
    }
2342
    vs->sg = cap_sg(vs->sg, vs->count, &vs->cnt);
2343
    if (vs->total <= vs->count) {
2344
        vs->len = v9fs_do_pwritev(s, vs->fidp->fs.fd, vs->sg, vs->cnt, vs->off);
2345
        if (vs->len > 0) {
2346
            vs->off += vs->len;
2347
        }
2348
        err = vs->len;
2349
        v9fs_write_post_pwritev(s, vs, err);
2350
    }
2351
    return;
2352
out:
2353
    complete_pdu(s, vs->pdu, err);
2354
    qemu_free(vs);
2355
}
2356

    
2357
static void v9fs_create_post_getiounit(V9fsState *s, V9fsCreateState *vs)
2358
{
2359
    int err;
2360
    v9fs_string_copy(&vs->fidp->path, &vs->fullname);
2361
    stat_to_qid(&vs->stbuf, &vs->qid);
2362

    
2363
    vs->offset += pdu_marshal(vs->pdu, vs->offset, "Qd", &vs->qid, vs->iounit);
2364
    err = vs->offset;
2365

    
2366
    complete_pdu(s, vs->pdu, err);
2367
    v9fs_string_free(&vs->name);
2368
    v9fs_string_free(&vs->extension);
2369
    v9fs_string_free(&vs->fullname);
2370
    qemu_free(vs);
2371
}
2372

    
2373
static void v9fs_post_create(V9fsState *s, V9fsCreateState *vs, int err)
2374
{
2375
    if (err == 0) {
2376
        vs->iounit = get_iounit(s, &vs->fidp->path);
2377
        v9fs_create_post_getiounit(s, vs);
2378
        return;
2379
    }
2380

    
2381
    complete_pdu(s, vs->pdu, err);
2382
    v9fs_string_free(&vs->name);
2383
    v9fs_string_free(&vs->extension);
2384
    v9fs_string_free(&vs->fullname);
2385
    qemu_free(vs);
2386
}
2387

    
2388
static void v9fs_create_post_perms(V9fsState *s, V9fsCreateState *vs, int err)
2389
{
2390
    if (err) {
2391
        err = -errno;
2392
    }
2393
    v9fs_post_create(s, vs, err);
2394
}
2395

    
2396
static void v9fs_create_post_opendir(V9fsState *s, V9fsCreateState *vs,
2397
                                                                    int err)
2398
{
2399
    if (!vs->fidp->fs.dir) {
2400
        err = -errno;
2401
    }
2402
    vs->fidp->fid_type = P9_FID_DIR;
2403
    v9fs_post_create(s, vs, err);
2404
}
2405

    
2406
static void v9fs_create_post_dir_lstat(V9fsState *s, V9fsCreateState *vs,
2407
                                                                    int err)
2408
{
2409
    if (err) {
2410
        err = -errno;
2411
        goto out;
2412
    }
2413

    
2414
    vs->fidp->fs.dir = v9fs_do_opendir(s, &vs->fullname);
2415
    v9fs_create_post_opendir(s, vs, err);
2416
    return;
2417

    
2418
out:
2419
    v9fs_post_create(s, vs, err);
2420
}
2421

    
2422
static void v9fs_create_post_mkdir(V9fsState *s, V9fsCreateState *vs, int err)
2423
{
2424
    if (err) {
2425
        err = -errno;
2426
        goto out;
2427
    }
2428

    
2429
    err = v9fs_do_lstat(s, &vs->fullname, &vs->stbuf);
2430
    v9fs_create_post_dir_lstat(s, vs, err);
2431
    return;
2432

    
2433
out:
2434
    v9fs_post_create(s, vs, err);
2435
}
2436

    
2437
static void v9fs_create_post_fstat(V9fsState *s, V9fsCreateState *vs, int err)
2438
{
2439
    if (err) {
2440
        vs->fidp->fid_type = P9_FID_NONE;
2441
        close(vs->fidp->fs.fd);
2442
        err = -errno;
2443
    }
2444
    v9fs_post_create(s, vs, err);
2445
    return;
2446
}
2447

    
2448
static void v9fs_create_post_open2(V9fsState *s, V9fsCreateState *vs, int err)
2449
{
2450
    if (vs->fidp->fs.fd == -1) {
2451
        err = -errno;
2452
        goto out;
2453
    }
2454
    vs->fidp->fid_type = P9_FID_FILE;
2455
    err = v9fs_do_fstat(s, vs->fidp->fs.fd, &vs->stbuf);
2456
    v9fs_create_post_fstat(s, vs, err);
2457

    
2458
    return;
2459

    
2460
out:
2461
    v9fs_post_create(s, vs, err);
2462

    
2463
}
2464

    
2465
static void v9fs_create_post_lstat(V9fsState *s, V9fsCreateState *vs, int err)
2466
{
2467

    
2468
    if (err == 0 || errno != ENOENT) {
2469
        err = -errno;
2470
        goto out;
2471
    }
2472

    
2473
    if (vs->perm & P9_STAT_MODE_DIR) {
2474
        err = v9fs_do_mkdir(s, vs->fullname.data, vs->perm & 0777,
2475
                vs->fidp->uid, -1);
2476
        v9fs_create_post_mkdir(s, vs, err);
2477
    } else if (vs->perm & P9_STAT_MODE_SYMLINK) {
2478
        err = v9fs_do_symlink(s, vs->fidp, vs->extension.data,
2479
                vs->fullname.data, -1);
2480
        v9fs_create_post_perms(s, vs, err);
2481
    } else if (vs->perm & P9_STAT_MODE_LINK) {
2482
        int32_t nfid = atoi(vs->extension.data);
2483
        V9fsFidState *nfidp = lookup_fid(s, nfid);
2484
        if (nfidp == NULL) {
2485
            err = -errno;
2486
            v9fs_post_create(s, vs, err);
2487
        }
2488
        err = v9fs_do_link(s, &nfidp->path, &vs->fullname);
2489
        v9fs_create_post_perms(s, vs, err);
2490
    } else if (vs->perm & P9_STAT_MODE_DEVICE) {
2491
        char ctype;
2492
        uint32_t major, minor;
2493
        mode_t nmode = 0;
2494

    
2495
        if (sscanf(vs->extension.data, "%c %u %u", &ctype, &major,
2496
                                        &minor) != 3) {
2497
            err = -errno;
2498
            v9fs_post_create(s, vs, err);
2499
        }
2500

    
2501
        switch (ctype) {
2502
        case 'c':
2503
            nmode = S_IFCHR;
2504
            break;
2505
        case 'b':
2506
            nmode = S_IFBLK;
2507
            break;
2508
        default:
2509
            err = -EIO;
2510
            v9fs_post_create(s, vs, err);
2511
        }
2512

    
2513
        nmode |= vs->perm & 0777;
2514
        err = v9fs_do_mknod(s, vs->fullname.data, nmode,
2515
                makedev(major, minor), vs->fidp->uid, -1);
2516
        v9fs_create_post_perms(s, vs, err);
2517
    } else if (vs->perm & P9_STAT_MODE_NAMED_PIPE) {
2518
        err = v9fs_do_mknod(s, vs->fullname.data, S_IFIFO | (vs->perm & 0777),
2519
                0, vs->fidp->uid, -1);
2520
        v9fs_post_create(s, vs, err);
2521
    } else if (vs->perm & P9_STAT_MODE_SOCKET) {
2522
        err = v9fs_do_mknod(s, vs->fullname.data, S_IFSOCK | (vs->perm & 0777),
2523
                0, vs->fidp->uid, -1);
2524
        v9fs_post_create(s, vs, err);
2525
    } else {
2526
        vs->fidp->fs.fd = v9fs_do_open2(s, vs->fullname.data, vs->fidp->uid,
2527
                -1, omode_to_uflags(vs->mode)|O_CREAT, vs->perm);
2528

    
2529
        v9fs_create_post_open2(s, vs, err);
2530
    }
2531

    
2532
    return;
2533

    
2534
out:
2535
    v9fs_post_create(s, vs, err);
2536
}
2537

    
2538
static void v9fs_create(V9fsState *s, V9fsPDU *pdu)
2539
{
2540
    int32_t fid;
2541
    V9fsCreateState *vs;
2542
    int err = 0;
2543

    
2544
    vs = qemu_malloc(sizeof(*vs));
2545
    vs->pdu = pdu;
2546
    vs->offset = 7;
2547

    
2548
    v9fs_string_init(&vs->fullname);
2549

    
2550
    pdu_unmarshal(vs->pdu, vs->offset, "dsdbs", &fid, &vs->name,
2551
                                &vs->perm, &vs->mode, &vs->extension);
2552

    
2553
    vs->fidp = lookup_fid(s, fid);
2554
    if (vs->fidp == NULL) {
2555
        err = -EINVAL;
2556
        goto out;
2557
    }
2558

    
2559
    v9fs_string_sprintf(&vs->fullname, "%s/%s", vs->fidp->path.data,
2560
                                                        vs->name.data);
2561

    
2562
    err = v9fs_do_lstat(s, &vs->fullname, &vs->stbuf);
2563
    v9fs_create_post_lstat(s, vs, err);
2564
    return;
2565

    
2566
out:
2567
    complete_pdu(s, vs->pdu, err);
2568
    v9fs_string_free(&vs->name);
2569
    v9fs_string_free(&vs->extension);
2570
    qemu_free(vs);
2571
}
2572

    
2573
static void v9fs_post_symlink(V9fsState *s, V9fsSymlinkState *vs, int err)
2574
{
2575
    if (err == 0) {
2576
        stat_to_qid(&vs->stbuf, &vs->qid);
2577
        vs->offset += pdu_marshal(vs->pdu, vs->offset, "Q", &vs->qid);
2578
        err = vs->offset;
2579
    } else {
2580
        err = -errno;
2581
    }
2582
    complete_pdu(s, vs->pdu, err);
2583
    v9fs_string_free(&vs->name);
2584
    v9fs_string_free(&vs->symname);
2585
    v9fs_string_free(&vs->fullname);
2586
    qemu_free(vs);
2587
}
2588

    
2589
static void v9fs_symlink_post_do_symlink(V9fsState *s, V9fsSymlinkState *vs,
2590
        int err)
2591
{
2592
    if (err) {
2593
        goto out;
2594
    }
2595
    err = v9fs_do_lstat(s, &vs->fullname, &vs->stbuf);
2596
out:
2597
    v9fs_post_symlink(s, vs, err);
2598
}
2599

    
2600
static void v9fs_symlink(V9fsState *s, V9fsPDU *pdu)
2601
{
2602
    int32_t dfid;
2603
    V9fsSymlinkState *vs;
2604
    int err = 0;
2605
    gid_t gid;
2606

    
2607
    vs = qemu_malloc(sizeof(*vs));
2608
    vs->pdu = pdu;
2609
    vs->offset = 7;
2610

    
2611
    v9fs_string_init(&vs->fullname);
2612

    
2613
    pdu_unmarshal(vs->pdu, vs->offset, "dssd", &dfid, &vs->name,
2614
            &vs->symname, &gid);
2615

    
2616
    vs->dfidp = lookup_fid(s, dfid);
2617
    if (vs->dfidp == NULL) {
2618
        err = -EINVAL;
2619
        goto out;
2620
    }
2621

    
2622
    v9fs_string_sprintf(&vs->fullname, "%s/%s", vs->dfidp->path.data,
2623
            vs->name.data);
2624
    err = v9fs_do_symlink(s, vs->dfidp, vs->symname.data,
2625
            vs->fullname.data, gid);
2626
    v9fs_symlink_post_do_symlink(s, vs, err);
2627
    return;
2628

    
2629
out:
2630
    complete_pdu(s, vs->pdu, err);
2631
    v9fs_string_free(&vs->name);
2632
    v9fs_string_free(&vs->symname);
2633
    qemu_free(vs);
2634
}
2635

    
2636
static void v9fs_flush(V9fsState *s, V9fsPDU *pdu)
2637
{
2638
    /* A nop call with no return */
2639
    complete_pdu(s, pdu, 7);
2640
}
2641

    
2642
static void v9fs_link(V9fsState *s, V9fsPDU *pdu)
2643
{
2644
    int32_t dfid, oldfid;
2645
    V9fsFidState *dfidp, *oldfidp;
2646
    V9fsString name, fullname;
2647
    size_t offset = 7;
2648
    int err = 0;
2649

    
2650
    v9fs_string_init(&fullname);
2651

    
2652
    pdu_unmarshal(pdu, offset, "dds", &dfid, &oldfid, &name);
2653

    
2654
    dfidp = lookup_fid(s, dfid);
2655
    if (dfidp == NULL) {
2656
        err = -errno;
2657
        goto out;
2658
    }
2659

    
2660
    oldfidp = lookup_fid(s, oldfid);
2661
    if (oldfidp == NULL) {
2662
        err = -errno;
2663
        goto out;
2664
    }
2665

    
2666
    v9fs_string_sprintf(&fullname, "%s/%s", dfidp->path.data, name.data);
2667
    err = offset;
2668
    err = v9fs_do_link(s, &oldfidp->path, &fullname);
2669
    if (err) {
2670
        err = -errno;
2671
    }
2672
    v9fs_string_free(&fullname);
2673

    
2674
out:
2675
    v9fs_string_free(&name);
2676
    complete_pdu(s, pdu, err);
2677
}
2678

    
2679
static void v9fs_remove_post_remove(V9fsState *s, V9fsRemoveState *vs,
2680
                                                                int err)
2681
{
2682
    if (err < 0) {
2683
        err = -errno;
2684
    } else {
2685
        err = vs->offset;
2686
    }
2687

    
2688
    /* For TREMOVE we need to clunk the fid even on failed remove */
2689
    free_fid(s, vs->fidp->fid);
2690

    
2691
    complete_pdu(s, vs->pdu, err);
2692
    qemu_free(vs);
2693
}
2694

    
2695
static void v9fs_remove(V9fsState *s, V9fsPDU *pdu)
2696
{
2697
    int32_t fid;
2698
    V9fsRemoveState *vs;
2699
    int err = 0;
2700

    
2701
    vs = qemu_malloc(sizeof(*vs));
2702
    vs->pdu = pdu;
2703
    vs->offset = 7;
2704

    
2705
    pdu_unmarshal(vs->pdu, vs->offset, "d", &fid);
2706

    
2707
    vs->fidp = lookup_fid(s, fid);
2708
    if (vs->fidp == NULL) {
2709
        err = -EINVAL;
2710
        goto out;
2711
    }
2712

    
2713
    err = v9fs_do_remove(s, &vs->fidp->path);
2714
    v9fs_remove_post_remove(s, vs, err);
2715
    return;
2716

    
2717
out:
2718
    complete_pdu(s, pdu, err);
2719
    qemu_free(vs);
2720
}
2721

    
2722
static void v9fs_wstat_post_truncate(V9fsState *s, V9fsWstatState *vs, int err)
2723
{
2724
    if (err < 0) {
2725
        goto out;
2726
    }
2727

    
2728
    err = vs->offset;
2729

    
2730
out:
2731
    v9fs_stat_free(&vs->v9stat);
2732
    complete_pdu(s, vs->pdu, err);
2733
    qemu_free(vs);
2734
}
2735

    
2736
static void v9fs_wstat_post_rename(V9fsState *s, V9fsWstatState *vs, int err)
2737
{
2738
    if (err < 0) {
2739
        goto out;
2740
    }
2741
    if (vs->v9stat.length != -1) {
2742
        if (v9fs_do_truncate(s, &vs->fidp->path, vs->v9stat.length) < 0) {
2743
            err = -errno;
2744
        }
2745
    }
2746
    v9fs_wstat_post_truncate(s, vs, err);
2747
    return;
2748

    
2749
out:
2750
    v9fs_stat_free(&vs->v9stat);
2751
    complete_pdu(s, vs->pdu, err);
2752
    qemu_free(vs);
2753
}
2754

    
2755
static int v9fs_complete_rename(V9fsState *s, V9fsRenameState *vs)
2756
{
2757
    int err = 0;
2758
    char *old_name, *new_name;
2759
    char *end;
2760

    
2761
    if (vs->newdirfid != -1) {
2762
        V9fsFidState *dirfidp;
2763
        dirfidp = lookup_fid(s, vs->newdirfid);
2764

    
2765
        if (dirfidp == NULL) {
2766
            err = -ENOENT;
2767
            goto out;
2768
        }
2769

    
2770
        BUG_ON(dirfidp->fid_type != P9_FID_NONE);
2771

    
2772
        new_name = qemu_mallocz(dirfidp->path.size + vs->name.size + 2);
2773

    
2774
        strcpy(new_name, dirfidp->path.data);
2775
        strcat(new_name, "/");
2776
        strcat(new_name + dirfidp->path.size, vs->name.data);
2777
    } else {
2778
        old_name = vs->fidp->path.data;
2779
        end = strrchr(old_name, '/');
2780
        if (end) {
2781
            end++;
2782
        } else {
2783
            end = old_name;
2784
        }
2785
        new_name = qemu_mallocz(end - old_name + vs->name.size + 1);
2786

    
2787
        strncat(new_name, old_name, end - old_name);
2788
        strncat(new_name + (end - old_name), vs->name.data, vs->name.size);
2789
    }
2790

    
2791
    v9fs_string_free(&vs->name);
2792
    vs->name.data = qemu_strdup(new_name);
2793
    vs->name.size = strlen(new_name);
2794

    
2795
    if (strcmp(new_name, vs->fidp->path.data) != 0) {
2796
        if (v9fs_do_rename(s, &vs->fidp->path, &vs->name)) {
2797
            err = -errno;
2798
        } else {
2799
            V9fsFidState *fidp;
2800
            /*
2801
            * Fixup fid's pointing to the old name to
2802
            * start pointing to the new name
2803
            */
2804
            for (fidp = s->fid_list; fidp; fidp = fidp->next) {
2805
                if (vs->fidp == fidp) {
2806
                    /*
2807
                    * we replace name of this fid towards the end
2808
                    * so that our below strcmp will work
2809
                    */
2810
                    continue;
2811
                }
2812
                if (!strncmp(vs->fidp->path.data, fidp->path.data,
2813
                    strlen(vs->fidp->path.data))) {
2814
                    /* replace the name */
2815
                    v9fs_fix_path(&fidp->path, &vs->name,
2816
                                  strlen(vs->fidp->path.data));
2817
                }
2818
            }
2819
            v9fs_string_copy(&vs->fidp->path, &vs->name);
2820
        }
2821
    }
2822
out:
2823
    v9fs_string_free(&vs->name);
2824
    return err;
2825
}
2826

    
2827
static void v9fs_rename_post_rename(V9fsState *s, V9fsRenameState *vs, int err)
2828
{
2829
    complete_pdu(s, vs->pdu, err);
2830
    qemu_free(vs);
2831
}
2832

    
2833
static void v9fs_wstat_post_chown(V9fsState *s, V9fsWstatState *vs, int err)
2834
{
2835
    if (err < 0) {
2836
        goto out;
2837
    }
2838

    
2839
    if (vs->v9stat.name.size != 0) {
2840
        V9fsRenameState *vr;
2841

    
2842
        vr = qemu_mallocz(sizeof(V9fsRenameState));
2843
        vr->newdirfid = -1;
2844
        vr->pdu = vs->pdu;
2845
        vr->fidp = vs->fidp;
2846
        vr->offset = vs->offset;
2847
        vr->name.size = vs->v9stat.name.size;
2848
        vr->name.data = qemu_strdup(vs->v9stat.name.data);
2849

    
2850
        err = v9fs_complete_rename(s, vr);
2851
        qemu_free(vr);
2852
    }
2853
    v9fs_wstat_post_rename(s, vs, err);
2854
    return;
2855

    
2856
out:
2857
    v9fs_stat_free(&vs->v9stat);
2858
    complete_pdu(s, vs->pdu, err);
2859
    qemu_free(vs);
2860
}
2861

    
2862
static void v9fs_rename(V9fsState *s, V9fsPDU *pdu)
2863
{
2864
    int32_t fid;
2865
    V9fsRenameState *vs;
2866
    ssize_t err = 0;
2867

    
2868
    vs = qemu_malloc(sizeof(*vs));
2869
    vs->pdu = pdu;
2870
    vs->offset = 7;
2871

    
2872
    pdu_unmarshal(vs->pdu, vs->offset, "dds", &fid, &vs->newdirfid, &vs->name);
2873

    
2874
    vs->fidp = lookup_fid(s, fid);
2875
    if (vs->fidp == NULL) {
2876
        err = -ENOENT;
2877
        goto out;
2878
    }
2879

    
2880
    BUG_ON(vs->fidp->fid_type != P9_FID_NONE);
2881

    
2882
    err = v9fs_complete_rename(s, vs);
2883
    v9fs_rename_post_rename(s, vs, err);
2884
    return;
2885
out:
2886
    complete_pdu(s, vs->pdu, err);
2887
    qemu_free(vs);
2888
}
2889

    
2890
static void v9fs_wstat_post_utime(V9fsState *s, V9fsWstatState *vs, int err)
2891
{
2892
    if (err < 0) {
2893
        goto out;
2894
    }
2895

    
2896
    if (vs->v9stat.n_gid != -1 || vs->v9stat.n_uid != -1) {
2897
        if (v9fs_do_chown(s, &vs->fidp->path, vs->v9stat.n_uid,
2898
                    vs->v9stat.n_gid)) {
2899
            err = -errno;
2900
        }
2901
    }
2902
    v9fs_wstat_post_chown(s, vs, err);
2903
    return;
2904

    
2905
out:
2906
    v9fs_stat_free(&vs->v9stat);
2907
    complete_pdu(s, vs->pdu, err);
2908
    qemu_free(vs);
2909
}
2910

    
2911
static void v9fs_wstat_post_chmod(V9fsState *s, V9fsWstatState *vs, int err)
2912
{
2913
    if (err < 0) {
2914
        goto out;
2915
    }
2916

    
2917
    if (vs->v9stat.mtime != -1 || vs->v9stat.atime != -1) {
2918
        struct timespec times[2];
2919
        if (vs->v9stat.atime != -1) {
2920
            times[0].tv_sec = vs->v9stat.atime;
2921
            times[0].tv_nsec = 0;
2922
        } else {
2923
            times[0].tv_nsec = UTIME_OMIT;
2924
        }
2925
        if (vs->v9stat.mtime != -1) {
2926
            times[1].tv_sec = vs->v9stat.mtime;
2927
            times[1].tv_nsec = 0;
2928
        } else {
2929
            times[1].tv_nsec = UTIME_OMIT;
2930
        }
2931

    
2932
        if (v9fs_do_utimensat(s, &vs->fidp->path, times)) {
2933
            err = -errno;
2934
        }
2935
    }
2936

    
2937
    v9fs_wstat_post_utime(s, vs, err);
2938
    return;
2939

    
2940
out:
2941
    v9fs_stat_free(&vs->v9stat);
2942
    complete_pdu(s, vs->pdu, err);
2943
    qemu_free(vs);
2944
}
2945

    
2946
static void v9fs_wstat_post_fsync(V9fsState *s, V9fsWstatState *vs, int err)
2947
{
2948
    if (err == -1) {
2949
        err = -errno;
2950
    }
2951
    v9fs_stat_free(&vs->v9stat);
2952
    complete_pdu(s, vs->pdu, err);
2953
    qemu_free(vs);
2954
}
2955

    
2956
static void v9fs_wstat_post_lstat(V9fsState *s, V9fsWstatState *vs, int err)
2957
{
2958
    uint32_t v9_mode;
2959

    
2960
    if (err == -1) {
2961
        err = -errno;
2962
        goto out;
2963
    }
2964

    
2965
    v9_mode = stat_to_v9mode(&vs->stbuf);
2966

    
2967
    if ((vs->v9stat.mode & P9_STAT_MODE_TYPE_BITS) !=
2968
        (v9_mode & P9_STAT_MODE_TYPE_BITS)) {
2969
            /* Attempting to change the type */
2970
            err = -EIO;
2971
            goto out;
2972
    }
2973

    
2974
    if (v9fs_do_chmod(s, &vs->fidp->path, v9mode_to_mode(vs->v9stat.mode,
2975
                    &vs->v9stat.extension))) {
2976
            err = -errno;
2977
     }
2978
    v9fs_wstat_post_chmod(s, vs, err);
2979
    return;
2980

    
2981
out:
2982
    v9fs_stat_free(&vs->v9stat);
2983
    complete_pdu(s, vs->pdu, err);
2984
    qemu_free(vs);
2985
}
2986

    
2987
static void v9fs_wstat(V9fsState *s, V9fsPDU *pdu)
2988
{
2989
    int32_t fid;
2990
    V9fsWstatState *vs;
2991
    int err = 0;
2992

    
2993
    vs = qemu_malloc(sizeof(*vs));
2994
    vs->pdu = pdu;
2995
    vs->offset = 7;
2996

    
2997
    pdu_unmarshal(pdu, vs->offset, "dwS", &fid, &vs->unused, &vs->v9stat);
2998

    
2999
    vs->fidp = lookup_fid(s, fid);
3000
    if (vs->fidp == NULL) {
3001
        err = -EINVAL;
3002
        goto out;
3003
    }
3004

    
3005
    /* do we need to sync the file? */
3006
    if (donttouch_stat(&vs->v9stat)) {
3007
        err = v9fs_do_fsync(s, vs->fidp->fs.fd, 0);
3008
        v9fs_wstat_post_fsync(s, vs, err);
3009
        return;
3010
    }
3011

    
3012
    if (vs->v9stat.mode != -1) {
3013
        err = v9fs_do_lstat(s, &vs->fidp->path, &vs->stbuf);
3014
        v9fs_wstat_post_lstat(s, vs, err);
3015
        return;
3016
    }
3017

    
3018
    v9fs_wstat_post_chmod(s, vs, err);
3019
    return;
3020

    
3021
out:
3022
    v9fs_stat_free(&vs->v9stat);
3023
    complete_pdu(s, vs->pdu, err);
3024
    qemu_free(vs);
3025
}
3026

    
3027
static void v9fs_statfs_post_statfs(V9fsState *s, V9fsStatfsState *vs, int err)
3028
{
3029
    int32_t bsize_factor;
3030

    
3031
    if (err) {
3032
        err = -errno;
3033
        goto out;
3034
    }
3035

    
3036
    /*
3037
     * compute bsize factor based on host file system block size
3038
     * and client msize
3039
     */
3040
    bsize_factor = (s->msize - P9_IOHDRSZ)/vs->stbuf.f_bsize;
3041
    if (!bsize_factor) {
3042
        bsize_factor = 1;
3043
    }
3044
    vs->v9statfs.f_type = vs->stbuf.f_type;
3045
    vs->v9statfs.f_bsize = vs->stbuf.f_bsize;
3046
    vs->v9statfs.f_bsize *= bsize_factor;
3047
    /*
3048
     * f_bsize is adjusted(multiplied) by bsize factor, so we need to
3049
     * adjust(divide) the number of blocks, free blocks and available
3050
     * blocks by bsize factor
3051
     */
3052
    vs->v9statfs.f_blocks = vs->stbuf.f_blocks/bsize_factor;
3053
    vs->v9statfs.f_bfree = vs->stbuf.f_bfree/bsize_factor;
3054
    vs->v9statfs.f_bavail = vs->stbuf.f_bavail/bsize_factor;
3055
    vs->v9statfs.f_files = vs->stbuf.f_files;
3056
    vs->v9statfs.f_ffree = vs->stbuf.f_ffree;
3057
    vs->v9statfs.fsid_val = (unsigned int) vs->stbuf.f_fsid.__val[0] |
3058
                        (unsigned long long)vs->stbuf.f_fsid.__val[1] << 32;
3059
    vs->v9statfs.f_namelen = vs->stbuf.f_namelen;
3060

    
3061
    vs->offset += pdu_marshal(vs->pdu, vs->offset, "ddqqqqqqd",
3062
         vs->v9statfs.f_type, vs->v9statfs.f_bsize, vs->v9statfs.f_blocks,
3063
         vs->v9statfs.f_bfree, vs->v9statfs.f_bavail, vs->v9statfs.f_files,
3064
         vs->v9statfs.f_ffree, vs->v9statfs.fsid_val,
3065
         vs->v9statfs.f_namelen);
3066

    
3067
out:
3068
    complete_pdu(s, vs->pdu, vs->offset);
3069
    qemu_free(vs);
3070
}
3071

    
3072
static void v9fs_statfs(V9fsState *s, V9fsPDU *pdu)
3073
{
3074
    V9fsStatfsState *vs;
3075
    ssize_t err = 0;
3076

    
3077
    vs = qemu_malloc(sizeof(*vs));
3078
    vs->pdu = pdu;
3079
    vs->offset = 7;
3080

    
3081
    memset(&vs->v9statfs, 0, sizeof(vs->v9statfs));
3082

    
3083
    pdu_unmarshal(vs->pdu, vs->offset, "d", &vs->fid);
3084

    
3085
    vs->fidp = lookup_fid(s, vs->fid);
3086
    if (vs->fidp == NULL) {
3087
        err = -ENOENT;
3088
        goto out;
3089
    }
3090

    
3091
    err = v9fs_do_statfs(s, &vs->fidp->path, &vs->stbuf);
3092
    v9fs_statfs_post_statfs(s, vs, err);
3093
    return;
3094

    
3095
out:
3096
    complete_pdu(s, vs->pdu, err);
3097
    qemu_free(vs);
3098
}
3099

    
3100
static void v9fs_mknod_post_lstat(V9fsState *s, V9fsMkState *vs, int err)
3101
{
3102
    if (err == -1) {
3103
        err = -errno;
3104
        goto out;
3105
    }
3106

    
3107
    stat_to_qid(&vs->stbuf, &vs->qid);
3108
    vs->offset += pdu_marshal(vs->pdu, vs->offset, "Q", &vs->qid);
3109
    err = vs->offset;
3110
out:
3111
    complete_pdu(s, vs->pdu, err);
3112
    v9fs_string_free(&vs->fullname);
3113
    v9fs_string_free(&vs->name);
3114
    qemu_free(vs);
3115
}
3116

    
3117
static void v9fs_mknod_post_mknod(V9fsState *s, V9fsMkState *vs, int err)
3118
{
3119
    if (err == -1) {
3120
        err = -errno;
3121
        goto out;
3122
    }
3123

    
3124
    err = v9fs_do_lstat(s, &vs->fullname, &vs->stbuf);
3125
    v9fs_mknod_post_lstat(s, vs, err);
3126
    return;
3127
out:
3128
    complete_pdu(s, vs->pdu, err);
3129
    v9fs_string_free(&vs->fullname);
3130
    v9fs_string_free(&vs->name);
3131
    qemu_free(vs);
3132
}
3133

    
3134
static void v9fs_mknod(V9fsState *s, V9fsPDU *pdu)
3135
{
3136
    int32_t fid;
3137
    V9fsMkState *vs;
3138
    int err = 0;
3139
    V9fsFidState *fidp;
3140
    gid_t gid;
3141
    int mode;
3142
    int major, minor;
3143

    
3144
    vs = qemu_malloc(sizeof(*vs));
3145
    vs->pdu = pdu;
3146
    vs->offset = 7;
3147

    
3148
    v9fs_string_init(&vs->fullname);
3149
    pdu_unmarshal(vs->pdu, vs->offset, "dsdddd", &fid, &vs->name, &mode,
3150
        &major, &minor, &gid);
3151

    
3152
    fidp = lookup_fid(s, fid);
3153
    if (fidp == NULL) {
3154
        err = -ENOENT;
3155
        goto out;
3156
    }
3157

    
3158
    v9fs_string_sprintf(&vs->fullname, "%s/%s", fidp->path.data, vs->name.data);
3159
    err = v9fs_do_mknod(s, vs->fullname.data, mode, makedev(major, minor),
3160
        fidp->uid, gid);
3161
    v9fs_mknod_post_mknod(s, vs, err);
3162
    return;
3163

    
3164
out:
3165
    complete_pdu(s, vs->pdu, err);
3166
    v9fs_string_free(&vs->fullname);
3167
    v9fs_string_free(&vs->name);
3168
    qemu_free(vs);
3169
}
3170

    
3171
/*
3172
 * Implement posix byte range locking code
3173
 * Server side handling of locking code is very simple, because 9p server in
3174
 * QEMU can handle only one client. And most of the lock handling
3175
 * (like conflict, merging) etc is done by the VFS layer itself, so no need to
3176
 * do any thing in * qemu 9p server side lock code path.
3177
 * So when a TLOCK request comes, always return success
3178
 */
3179

    
3180
static void v9fs_lock(V9fsState *s, V9fsPDU *pdu)
3181
{
3182
    int32_t fid, err = 0;
3183
    V9fsLockState *vs;
3184

    
3185
    vs = qemu_mallocz(sizeof(*vs));
3186
    vs->pdu = pdu;
3187
    vs->offset = 7;
3188

    
3189
    vs->flock = qemu_malloc(sizeof(*vs->flock));
3190
    pdu_unmarshal(vs->pdu, vs->offset, "dbdqqds", &fid, &vs->flock->type,
3191
                &vs->flock->flags, &vs->flock->start, &vs->flock->length,
3192
                            &vs->flock->proc_id, &vs->flock->client_id);
3193

    
3194
    vs->status = P9_LOCK_ERROR;
3195

    
3196
    /* We support only block flag now (that too ignored currently) */
3197
    if (vs->flock->flags & ~P9_LOCK_FLAGS_BLOCK) {
3198
        err = -EINVAL;
3199
        goto out;
3200
    }
3201
    vs->fidp = lookup_fid(s, fid);
3202
    if (vs->fidp == NULL) {
3203
        err = -ENOENT;
3204
        goto out;
3205
    }
3206

    
3207
    err = v9fs_do_fstat(s, vs->fidp->fs.fd, &vs->stbuf);
3208
    if (err < 0) {
3209
        err = -errno;
3210
        goto out;
3211
    }
3212
    vs->status = P9_LOCK_SUCCESS;
3213
out:
3214
    vs->offset += pdu_marshal(vs->pdu, vs->offset, "b", vs->status);
3215
    complete_pdu(s, vs->pdu, err);
3216
    qemu_free(vs->flock);
3217
    qemu_free(vs);
3218
}
3219

    
3220
/*
3221
 * When a TGETLOCK request comes, always return success because all lock
3222
 * handling is done by client's VFS layer.
3223
 */
3224

    
3225
static void v9fs_getlock(V9fsState *s, V9fsPDU *pdu)
3226
{
3227
    int32_t fid, err = 0;
3228
    V9fsGetlockState *vs;
3229

    
3230
    vs = qemu_mallocz(sizeof(*vs));
3231
    vs->pdu = pdu;
3232
    vs->offset = 7;
3233

    
3234
    vs->glock = qemu_malloc(sizeof(*vs->glock));
3235
    pdu_unmarshal(vs->pdu, vs->offset, "dbqqds", &fid, &vs->glock->type,
3236
                &vs->glock->start, &vs->glock->length, &vs->glock->proc_id,
3237
                &vs->glock->client_id);
3238

    
3239
    vs->fidp = lookup_fid(s, fid);
3240
    if (vs->fidp == NULL) {
3241
        err = -ENOENT;
3242
        goto out;
3243
    }
3244

    
3245
    err = v9fs_do_fstat(s, vs->fidp->fs.fd, &vs->stbuf);
3246
    if (err < 0) {
3247
        err = -errno;
3248
        goto out;
3249
    }
3250
    vs->glock->type = F_UNLCK;
3251
    vs->offset += pdu_marshal(vs->pdu, vs->offset, "bqqds", vs->glock->type,
3252
                vs->glock->start, vs->glock->length, vs->glock->proc_id,
3253
                &vs->glock->client_id);
3254
out:
3255
    complete_pdu(s, vs->pdu, err);
3256
    qemu_free(vs->glock);
3257
    qemu_free(vs);
3258
}
3259

    
3260
static void v9fs_mkdir_post_lstat(V9fsState *s, V9fsMkState *vs, int err)
3261
{
3262
    if (err == -1) {
3263
        err = -errno;
3264
        goto out;
3265
    }
3266

    
3267
    stat_to_qid(&vs->stbuf, &vs->qid);
3268
    vs->offset += pdu_marshal(vs->pdu, vs->offset, "Q", &vs->qid);
3269
    err = vs->offset;
3270
out:
3271
    complete_pdu(s, vs->pdu, err);
3272
    v9fs_string_free(&vs->fullname);
3273
    v9fs_string_free(&vs->name);
3274
    qemu_free(vs);
3275
}
3276

    
3277
static void v9fs_mkdir_post_mkdir(V9fsState *s, V9fsMkState *vs, int err)
3278
{
3279
    if (err == -1) {
3280
        err = -errno;
3281
        goto out;
3282
    }
3283

    
3284
    err = v9fs_do_lstat(s, &vs->fullname, &vs->stbuf);
3285
    v9fs_mkdir_post_lstat(s, vs, err);
3286
    return;
3287
out:
3288
    complete_pdu(s, vs->pdu, err);
3289
    v9fs_string_free(&vs->fullname);
3290
    v9fs_string_free(&vs->name);
3291
    qemu_free(vs);
3292
}
3293

    
3294
static void v9fs_mkdir(V9fsState *s, V9fsPDU *pdu)
3295
{
3296
    int32_t fid;
3297
    V9fsMkState *vs;
3298
    int err = 0;
3299
    V9fsFidState *fidp;
3300
    gid_t gid;
3301
    int mode;
3302

    
3303
    vs = qemu_malloc(sizeof(*vs));
3304
    vs->pdu = pdu;
3305
    vs->offset = 7;
3306

    
3307
    v9fs_string_init(&vs->fullname);
3308
    pdu_unmarshal(vs->pdu, vs->offset, "dsdd", &fid, &vs->name, &mode,
3309
        &gid);
3310

    
3311
    fidp = lookup_fid(s, fid);
3312
    if (fidp == NULL) {
3313
        err = -ENOENT;
3314
        goto out;
3315
    }
3316

    
3317
    v9fs_string_sprintf(&vs->fullname, "%s/%s", fidp->path.data, vs->name.data);
3318
    err = v9fs_do_mkdir(s, vs->fullname.data, mode, fidp->uid, gid);
3319
    v9fs_mkdir_post_mkdir(s, vs, err);
3320
    return;
3321

    
3322
out:
3323
    complete_pdu(s, vs->pdu, err);
3324
    v9fs_string_free(&vs->fullname);
3325
    v9fs_string_free(&vs->name);
3326
    qemu_free(vs);
3327
}
3328

    
3329
static void v9fs_post_xattr_getvalue(V9fsState *s, V9fsXattrState *vs, int err)
3330
{
3331

    
3332
    if (err < 0) {
3333
        err = -errno;
3334
        free_fid(s, vs->xattr_fidp->fid);
3335
        goto out;
3336
    }
3337
    vs->offset += pdu_marshal(vs->pdu, vs->offset, "q", vs->size);
3338
    err = vs->offset;
3339
out:
3340
    complete_pdu(s, vs->pdu, err);
3341
    v9fs_string_free(&vs->name);
3342
    qemu_free(vs);
3343
    return;
3344
}
3345

    
3346
static void v9fs_post_xattr_check(V9fsState *s, V9fsXattrState *vs, ssize_t err)
3347
{
3348
    if (err < 0) {
3349
        err = -errno;
3350
        free_fid(s, vs->xattr_fidp->fid);
3351
        goto out;
3352
    }
3353
    /*
3354
     * Read the xattr value
3355
     */
3356
    vs->xattr_fidp->fs.xattr.len = vs->size;
3357
    vs->xattr_fidp->fid_type = P9_FID_XATTR;
3358
    vs->xattr_fidp->fs.xattr.copied_len = -1;
3359
    if (vs->size) {
3360
        vs->xattr_fidp->fs.xattr.value = qemu_malloc(vs->size);
3361
        err = v9fs_do_lgetxattr(s, &vs->xattr_fidp->path,
3362
                                &vs->name, vs->xattr_fidp->fs.xattr.value,
3363
                                vs->xattr_fidp->fs.xattr.len);
3364
    }
3365
    v9fs_post_xattr_getvalue(s, vs, err);
3366
    return;
3367
out:
3368
    complete_pdu(s, vs->pdu, err);
3369
    v9fs_string_free(&vs->name);
3370
    qemu_free(vs);
3371
}
3372

    
3373
static void v9fs_post_lxattr_getvalue(V9fsState *s,
3374
                                      V9fsXattrState *vs, int err)
3375
{
3376
    if (err < 0) {
3377
        err = -errno;
3378
        free_fid(s, vs->xattr_fidp->fid);
3379
        goto out;
3380
    }
3381
    vs->offset += pdu_marshal(vs->pdu, vs->offset, "q", vs->size);
3382
    err = vs->offset;
3383
out:
3384
    complete_pdu(s, vs->pdu, err);
3385
    v9fs_string_free(&vs->name);
3386
    qemu_free(vs);
3387
    return;
3388
}
3389

    
3390
static void v9fs_post_lxattr_check(V9fsState *s,
3391
                                   V9fsXattrState *vs, ssize_t err)
3392
{
3393
    if (err < 0) {
3394
        err = -errno;
3395
        free_fid(s, vs->xattr_fidp->fid);
3396
        goto out;
3397
    }
3398
    /*
3399
     * Read the xattr value
3400
     */
3401
    vs->xattr_fidp->fs.xattr.len = vs->size;
3402
    vs->xattr_fidp->fid_type = P9_FID_XATTR;
3403
    vs->xattr_fidp->fs.xattr.copied_len = -1;
3404
    if (vs->size) {
3405
        vs->xattr_fidp->fs.xattr.value = qemu_malloc(vs->size);
3406
        err = v9fs_do_llistxattr(s, &vs->xattr_fidp->path,
3407
                                 vs->xattr_fidp->fs.xattr.value,
3408
                                 vs->xattr_fidp->fs.xattr.len);
3409
    }
3410
    v9fs_post_lxattr_getvalue(s, vs, err);
3411
    return;
3412
out:
3413
    complete_pdu(s, vs->pdu, err);
3414
    v9fs_string_free(&vs->name);
3415
    qemu_free(vs);
3416
}
3417

    
3418
static void v9fs_xattrwalk(V9fsState *s, V9fsPDU *pdu)
3419
{
3420
    ssize_t err = 0;
3421
    V9fsXattrState *vs;
3422
    int32_t fid, newfid;
3423

    
3424
    vs = qemu_malloc(sizeof(*vs));
3425
    vs->pdu = pdu;
3426
    vs->offset = 7;
3427

    
3428
    pdu_unmarshal(vs->pdu, vs->offset, "dds", &fid, &newfid, &vs->name);
3429
    vs->file_fidp = lookup_fid(s, fid);
3430
    if (vs->file_fidp == NULL) {
3431
        err = -ENOENT;
3432
        goto out;
3433
    }
3434

    
3435
    vs->xattr_fidp = alloc_fid(s, newfid);
3436
    if (vs->xattr_fidp == NULL) {
3437
        err = -EINVAL;
3438
        goto out;
3439
    }
3440

    
3441
    v9fs_string_copy(&vs->xattr_fidp->path, &vs->file_fidp->path);
3442
    if (vs->name.data[0] == 0) {
3443
        /*
3444
         * listxattr request. Get the size first
3445
         */
3446
        vs->size = v9fs_do_llistxattr(s, &vs->xattr_fidp->path,
3447
                                      NULL, 0);
3448
        if (vs->size < 0) {
3449
            err = vs->size;
3450
        }
3451
        v9fs_post_lxattr_check(s, vs, err);
3452
        return;
3453
    } else {
3454
        /*
3455
         * specific xattr fid. We check for xattr
3456
         * presence also collect the xattr size
3457
         */
3458
        vs->size = v9fs_do_lgetxattr(s, &vs->xattr_fidp->path,
3459
                                     &vs->name, NULL, 0);
3460
        if (vs->size < 0) {
3461
            err = vs->size;
3462
        }
3463
        v9fs_post_xattr_check(s, vs, err);
3464
        return;
3465
    }
3466
out:
3467
    complete_pdu(s, vs->pdu, err);
3468
    v9fs_string_free(&vs->name);
3469
    qemu_free(vs);
3470
}
3471

    
3472
static void v9fs_xattrcreate(V9fsState *s, V9fsPDU *pdu)
3473
{
3474
    int flags;
3475
    int32_t fid;
3476
    ssize_t err = 0;
3477
    V9fsXattrState *vs;
3478

    
3479
    vs = qemu_malloc(sizeof(*vs));
3480
    vs->pdu = pdu;
3481
    vs->offset = 7;
3482

    
3483
    pdu_unmarshal(vs->pdu, vs->offset, "dsqd",
3484
                  &fid, &vs->name, &vs->size, &flags);
3485

    
3486
    vs->file_fidp = lookup_fid(s, fid);
3487
    if (vs->file_fidp == NULL) {
3488
        err = -EINVAL;
3489
        goto out;
3490
    }
3491

    
3492
    /* Make the file fid point to xattr */
3493
    vs->xattr_fidp = vs->file_fidp;
3494
    vs->xattr_fidp->fid_type = P9_FID_XATTR;
3495
    vs->xattr_fidp->fs.xattr.copied_len = 0;
3496
    vs->xattr_fidp->fs.xattr.len = vs->size;
3497
    vs->xattr_fidp->fs.xattr.flags = flags;
3498
    v9fs_string_init(&vs->xattr_fidp->fs.xattr.name);
3499
    v9fs_string_copy(&vs->xattr_fidp->fs.xattr.name, &vs->name);
3500
    if (vs->size)
3501
        vs->xattr_fidp->fs.xattr.value = qemu_malloc(vs->size);
3502
    else
3503
        vs->xattr_fidp->fs.xattr.value = NULL;
3504

    
3505
out:
3506
    complete_pdu(s, vs->pdu, err);
3507
    v9fs_string_free(&vs->name);
3508
    qemu_free(vs);
3509
}
3510

    
3511
static void v9fs_readlink_post_readlink(V9fsState *s, V9fsReadLinkState *vs,
3512
                                                    int err)
3513
{
3514
    if (err < 0) {
3515
        err = -errno;
3516
        goto out;
3517
    }
3518
    vs->offset += pdu_marshal(vs->pdu, vs->offset, "s", &vs->target);
3519
    err = vs->offset;
3520
out:
3521
    complete_pdu(s, vs->pdu, err);
3522
    v9fs_string_free(&vs->target);
3523
    qemu_free(vs);
3524
}
3525

    
3526
static void v9fs_readlink(V9fsState *s, V9fsPDU *pdu)
3527
{
3528
    int32_t fid;
3529
    V9fsReadLinkState *vs;
3530
    int err = 0;
3531
    V9fsFidState *fidp;
3532

    
3533
    vs = qemu_malloc(sizeof(*vs));
3534
    vs->pdu = pdu;
3535
    vs->offset = 7;
3536

    
3537
    pdu_unmarshal(vs->pdu, vs->offset, "d", &fid);
3538

    
3539
    fidp = lookup_fid(s, fid);
3540
    if (fidp == NULL) {
3541
        err = -ENOENT;
3542
        goto out;
3543
    }
3544

    
3545
    v9fs_string_init(&vs->target);
3546
    err = v9fs_do_readlink(s, &fidp->path, &vs->target);
3547
    v9fs_readlink_post_readlink(s, vs, err);
3548
    return;
3549
out:
3550
    complete_pdu(s, vs->pdu, err);
3551
    qemu_free(vs);
3552
}
3553

    
3554
typedef void (pdu_handler_t)(V9fsState *s, V9fsPDU *pdu);
3555

    
3556
static pdu_handler_t *pdu_handlers[] = {
3557
    [P9_TREADDIR] = v9fs_readdir,
3558
    [P9_TSTATFS] = v9fs_statfs,
3559
    [P9_TGETATTR] = v9fs_getattr,
3560
    [P9_TSETATTR] = v9fs_setattr,
3561
    [P9_TXATTRWALK] = v9fs_xattrwalk,
3562
    [P9_TXATTRCREATE] = v9fs_xattrcreate,
3563
    [P9_TMKNOD] = v9fs_mknod,
3564
    [P9_TRENAME] = v9fs_rename,
3565
    [P9_TLOCK] = v9fs_lock,
3566
    [P9_TGETLOCK] = v9fs_getlock,
3567
    [P9_TREADLINK] = v9fs_readlink,
3568
    [P9_TMKDIR] = v9fs_mkdir,
3569
    [P9_TVERSION] = v9fs_version,
3570
    [P9_TLOPEN] = v9fs_open,
3571
    [P9_TATTACH] = v9fs_attach,
3572
    [P9_TSTAT] = v9fs_stat,
3573
    [P9_TWALK] = v9fs_walk,
3574
    [P9_TCLUNK] = v9fs_clunk,
3575
    [P9_TFSYNC] = v9fs_fsync,
3576
    [P9_TOPEN] = v9fs_open,
3577
    [P9_TREAD] = v9fs_read,
3578
#if 0
3579
    [P9_TAUTH] = v9fs_auth,
3580
#endif
3581
    [P9_TFLUSH] = v9fs_flush,
3582
    [P9_TLINK] = v9fs_link,
3583
    [P9_TSYMLINK] = v9fs_symlink,
3584
    [P9_TCREATE] = v9fs_create,
3585
    [P9_TLCREATE] = v9fs_lcreate,
3586
    [P9_TWRITE] = v9fs_write,
3587
    [P9_TWSTAT] = v9fs_wstat,
3588
    [P9_TREMOVE] = v9fs_remove,
3589
};
3590

    
3591
static void submit_pdu(V9fsState *s, V9fsPDU *pdu)
3592
{
3593
    pdu_handler_t *handler;
3594

    
3595
    if (debug_9p_pdu) {
3596
        pprint_pdu(pdu);
3597
    }
3598

    
3599
    BUG_ON(pdu->id >= ARRAY_SIZE(pdu_handlers));
3600

    
3601
    handler = pdu_handlers[pdu->id];
3602
    BUG_ON(handler == NULL);
3603

    
3604
    handler(s, pdu);
3605
}
3606

    
3607
static void handle_9p_output(VirtIODevice *vdev, VirtQueue *vq)
3608
{
3609
    V9fsState *s = (V9fsState *)vdev;
3610
    V9fsPDU *pdu;
3611
    ssize_t len;
3612

    
3613
    while ((pdu = alloc_pdu(s)) &&
3614
            (len = virtqueue_pop(vq, &pdu->elem)) != 0) {
3615
        uint8_t *ptr;
3616

    
3617
        BUG_ON(pdu->elem.out_num == 0 || pdu->elem.in_num == 0);
3618
        BUG_ON(pdu->elem.out_sg[0].iov_len < 7);
3619

    
3620
        ptr = pdu->elem.out_sg[0].iov_base;
3621

    
3622
        memcpy(&pdu->size, ptr, 4);
3623
        pdu->id = ptr[4];
3624
        memcpy(&pdu->tag, ptr + 5, 2);
3625

    
3626
        submit_pdu(s, pdu);
3627
    }
3628

    
3629
    free_pdu(s, pdu);
3630
}
3631

    
3632
static uint32_t virtio_9p_get_features(VirtIODevice *vdev, uint32_t features)
3633
{
3634
    features |= 1 << VIRTIO_9P_MOUNT_TAG;
3635
    return features;
3636
}
3637

    
3638
static V9fsState *to_virtio_9p(VirtIODevice *vdev)
3639
{
3640
    return (V9fsState *)vdev;
3641
}
3642

    
3643
static void virtio_9p_get_config(VirtIODevice *vdev, uint8_t *config)
3644
{
3645
    struct virtio_9p_config *cfg;
3646
    V9fsState *s = to_virtio_9p(vdev);
3647

    
3648
    cfg = qemu_mallocz(sizeof(struct virtio_9p_config) +
3649
                        s->tag_len);
3650
    stw_raw(&cfg->tag_len, s->tag_len);
3651
    memcpy(cfg->tag, s->tag, s->tag_len);
3652
    memcpy(config, cfg, s->config_size);
3653
    qemu_free(cfg);
3654
}
3655

    
3656
VirtIODevice *virtio_9p_init(DeviceState *dev, V9fsConf *conf)
3657
 {
3658
    V9fsState *s;
3659
    int i, len;
3660
    struct stat stat;
3661
    FsTypeEntry *fse;
3662

    
3663

    
3664
    s = (V9fsState *)virtio_common_init("virtio-9p",
3665
                                    VIRTIO_ID_9P,
3666
                                    sizeof(struct virtio_9p_config)+
3667
                                    MAX_TAG_LEN,
3668
                                    sizeof(V9fsState));
3669

    
3670
    /* initialize pdu allocator */
3671
    QLIST_INIT(&s->free_list);
3672
    for (i = 0; i < (MAX_REQ - 1); i++) {
3673
        QLIST_INSERT_HEAD(&s->free_list, &s->pdus[i], next);
3674
    }
3675

    
3676
    s->vq = virtio_add_queue(&s->vdev, MAX_REQ, handle_9p_output);
3677

    
3678
    fse = get_fsdev_fsentry(conf->fsdev_id);
3679

    
3680
    if (!fse) {
3681
        /* We don't have a fsdev identified by fsdev_id */
3682
        fprintf(stderr, "Virtio-9p device couldn't find fsdev with the "
3683
                "id = %s\n", conf->fsdev_id ? conf->fsdev_id : "NULL");
3684
        exit(1);
3685
    }
3686

    
3687
    if (!fse->path || !conf->tag) {
3688
        /* we haven't specified a mount_tag or the path */
3689
        fprintf(stderr, "fsdev with id %s needs path "
3690
                "and Virtio-9p device needs mount_tag arguments\n",
3691
                conf->fsdev_id);
3692
        exit(1);
3693
    }
3694

    
3695
    if (!strcmp(fse->security_model, "passthrough")) {
3696
        /* Files on the Fileserver set to client user credentials */
3697
        s->ctx.fs_sm = SM_PASSTHROUGH;
3698
        s->ctx.xops = passthrough_xattr_ops;
3699
    } else if (!strcmp(fse->security_model, "mapped")) {
3700
        /* Files on the fileserver are set to QEMU credentials.
3701
         * Client user credentials are saved in extended attributes.
3702
         */
3703
        s->ctx.fs_sm = SM_MAPPED;
3704
        s->ctx.xops = mapped_xattr_ops;
3705
    } else if (!strcmp(fse->security_model, "none")) {
3706
        /*
3707
         * Files on the fileserver are set to QEMU credentials.
3708
         */
3709
        s->ctx.fs_sm = SM_NONE;
3710
        s->ctx.xops = none_xattr_ops;
3711
    } else {
3712
        fprintf(stderr, "Default to security_model=none. You may want"
3713
                " enable advanced security model using "
3714
                "security option:\n\t security_model=passthrough \n\t "
3715
                "security_model=mapped\n");
3716
        s->ctx.fs_sm = SM_NONE;
3717
        s->ctx.xops = none_xattr_ops;
3718
    }
3719

    
3720
    if (lstat(fse->path, &stat)) {
3721
        fprintf(stderr, "share path %s does not exist\n", fse->path);
3722
        exit(1);
3723
    } else if (!S_ISDIR(stat.st_mode)) {
3724
        fprintf(stderr, "share path %s is not a directory \n", fse->path);
3725
        exit(1);
3726
    }
3727

    
3728
    s->ctx.fs_root = qemu_strdup(fse->path);
3729
    len = strlen(conf->tag);
3730
    if (len > MAX_TAG_LEN) {
3731
        len = MAX_TAG_LEN;
3732
    }
3733
    /* s->tag is non-NULL terminated string */
3734
    s->tag = qemu_malloc(len);
3735
    memcpy(s->tag, conf->tag, len);
3736
    s->tag_len = len;
3737
    s->ctx.uid = -1;
3738

    
3739
    s->ops = fse->ops;
3740
    s->vdev.get_features = virtio_9p_get_features;
3741
    s->config_size = sizeof(struct virtio_9p_config) +
3742
                        s->tag_len;
3743
    s->vdev.get_config = virtio_9p_get_config;
3744

    
3745
    return &s->vdev;
3746
}