Statistics
| Branch: | Revision:

root / hw / virtio-9p.c @ 771e9d4c

History | View | Annotate | Download (80.2 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

    
21
int dotu = 1;
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_readv(V9fsState *s, int fd, const struct iovec *iov,
139
                            int iovcnt)
140
{
141
    return s->ops->readv(&s->ctx, fd, iov, iovcnt);
142
}
143

    
144
static off_t v9fs_do_lseek(V9fsState *s, int fd, off_t offset, int whence)
145
{
146
    return s->ops->lseek(&s->ctx, fd, offset, whence);
147
}
148

    
149
static int v9fs_do_writev(V9fsState *s, int fd, const struct iovec *iov,
150
                       int iovcnt)
151
{
152
    return s->ops->writev(&s->ctx, fd, iov, iovcnt);
153
}
154

    
155
static int v9fs_do_chmod(V9fsState *s, V9fsString *path, mode_t mode)
156
{
157
    FsCred cred;
158
    cred_init(&cred);
159
    cred.fc_mode = mode;
160
    return s->ops->chmod(&s->ctx, path->data, &cred);
161
}
162

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

    
175
static int v9fs_do_mkdir(V9fsState *s, char *name, mode_t mode,
176
                uid_t uid, gid_t gid)
177
{
178
    FsCred cred;
179

    
180
    cred_init(&cred);
181
    cred.fc_uid = uid;
182
    cred.fc_gid = gid;
183
    cred.fc_mode = mode;
184

    
185
    return s->ops->mkdir(&s->ctx, name, &cred);
186
}
187

    
188
static int v9fs_do_fstat(V9fsState *s, int fd, struct stat *stbuf)
189
{
190
    return s->ops->fstat(&s->ctx, fd, stbuf);
191
}
192

    
193
static int v9fs_do_open2(V9fsState *s, char *fullname, uid_t uid, gid_t gid,
194
        int flags, int mode)
195
{
196
    FsCred cred;
197

    
198
    cred_init(&cred);
199
    cred.fc_uid = uid;
200
    cred.fc_gid = gid;
201
    cred.fc_mode = mode & 07777;
202
    flags = flags;
203

    
204
    return s->ops->open2(&s->ctx, fullname, flags, &cred);
205
}
206

    
207
static int v9fs_do_symlink(V9fsState *s, V9fsFidState *fidp,
208
        const char *oldpath, const char *newpath, gid_t gid)
209
{
210
    FsCred cred;
211
    cred_init(&cred);
212
    cred.fc_uid = fidp->uid;
213
    cred.fc_gid = gid;
214
    cred.fc_mode = 0777;
215

    
216
    return s->ops->symlink(&s->ctx, oldpath, newpath, &cred);
217
}
218

    
219
static int v9fs_do_link(V9fsState *s, V9fsString *oldpath, V9fsString *newpath)
220
{
221
    return s->ops->link(&s->ctx, oldpath->data, newpath->data);
222
}
223

    
224
static int v9fs_do_truncate(V9fsState *s, V9fsString *path, off_t size)
225
{
226
    return s->ops->truncate(&s->ctx, path->data, size);
227
}
228

    
229
static int v9fs_do_rename(V9fsState *s, V9fsString *oldpath,
230
                            V9fsString *newpath)
231
{
232
    return s->ops->rename(&s->ctx, oldpath->data, newpath->data);
233
}
234

    
235
static int v9fs_do_chown(V9fsState *s, V9fsString *path, uid_t uid, gid_t gid)
236
{
237
    FsCred cred;
238
    cred_init(&cred);
239
    cred.fc_uid = uid;
240
    cred.fc_gid = gid;
241

    
242
    return s->ops->chown(&s->ctx, path->data, &cred);
243
}
244

    
245
static int v9fs_do_utimensat(V9fsState *s, V9fsString *path,
246
                                           const struct timespec times[2])
247
{
248
    return s->ops->utimensat(&s->ctx, path->data, times);
249
}
250

    
251
static int v9fs_do_remove(V9fsState *s, V9fsString *path)
252
{
253
    return s->ops->remove(&s->ctx, path->data);
254
}
255

    
256
static int v9fs_do_fsync(V9fsState *s, int fd)
257
{
258
    return s->ops->fsync(&s->ctx, fd);
259
}
260

    
261
static int v9fs_do_statfs(V9fsState *s, V9fsString *path, struct statfs *stbuf)
262
{
263
    return s->ops->statfs(&s->ctx, path->data, stbuf);
264
}
265

    
266
static void v9fs_string_init(V9fsString *str)
267
{
268
    str->data = NULL;
269
    str->size = 0;
270
}
271

    
272
static void v9fs_string_free(V9fsString *str)
273
{
274
    qemu_free(str->data);
275
    str->data = NULL;
276
    str->size = 0;
277
}
278

    
279
static void v9fs_string_null(V9fsString *str)
280
{
281
    v9fs_string_free(str);
282
}
283

    
284
static int number_to_string(void *arg, char type)
285
{
286
    unsigned int ret = 0;
287

    
288
    switch (type) {
289
    case 'u': {
290
        unsigned int num = *(unsigned int *)arg;
291

    
292
        do {
293
            ret++;
294
            num = num/10;
295
        } while (num);
296
        break;
297
    }
298
    default:
299
        printf("Number_to_string: Unknown number format\n");
300
        return -1;
301
    }
302

    
303
    return ret;
304
}
305

    
306
static int v9fs_string_alloc_printf(char **strp, const char *fmt, va_list ap)
307
{
308
    va_list ap2;
309
    char *iter = (char *)fmt;
310
    int len = 0;
311
    int nr_args = 0;
312
    char *arg_char_ptr;
313
    unsigned int arg_uint;
314

    
315
    /* Find the number of %'s that denotes an argument */
316
    for (iter = strstr(iter, "%"); iter; iter = strstr(iter, "%")) {
317
        nr_args++;
318
        iter++;
319
    }
320

    
321
    len = strlen(fmt) - 2*nr_args;
322

    
323
    if (!nr_args) {
324
        goto alloc_print;
325
    }
326

    
327
    va_copy(ap2, ap);
328

    
329
    iter = (char *)fmt;
330

    
331
    /* Now parse the format string */
332
    for (iter = strstr(iter, "%"); iter; iter = strstr(iter, "%")) {
333
        iter++;
334
        switch (*iter) {
335
        case 'u':
336
            arg_uint = va_arg(ap2, unsigned int);
337
            len += number_to_string((void *)&arg_uint, 'u');
338
            break;
339
        case 's':
340
            arg_char_ptr = va_arg(ap2, char *);
341
            len += strlen(arg_char_ptr);
342
            break;
343
        case 'c':
344
            len += 1;
345
            break;
346
        default:
347
            fprintf(stderr,
348
                    "v9fs_string_alloc_printf:Incorrect format %c", *iter);
349
            return -1;
350
        }
351
        iter++;
352
    }
353

    
354
alloc_print:
355
    *strp = qemu_malloc((len + 1) * sizeof(**strp));
356

    
357
    return vsprintf(*strp, fmt, ap);
358
}
359

    
360
static void v9fs_string_sprintf(V9fsString *str, const char *fmt, ...)
361
{
362
    va_list ap;
363
    int err;
364

    
365
    v9fs_string_free(str);
366

    
367
    va_start(ap, fmt);
368
    err = v9fs_string_alloc_printf(&str->data, fmt, ap);
369
    BUG_ON(err == -1);
370
    va_end(ap);
371

    
372
    str->size = err;
373
}
374

    
375
static void v9fs_string_copy(V9fsString *lhs, V9fsString *rhs)
376
{
377
    v9fs_string_free(lhs);
378
    v9fs_string_sprintf(lhs, "%s", rhs->data);
379
}
380

    
381
static size_t v9fs_string_size(V9fsString *str)
382
{
383
    return str->size;
384
}
385

    
386
static V9fsFidState *lookup_fid(V9fsState *s, int32_t fid)
387
{
388
    V9fsFidState *f;
389

    
390
    for (f = s->fid_list; f; f = f->next) {
391
        if (f->fid == fid) {
392
            return f;
393
        }
394
    }
395

    
396
    return NULL;
397
}
398

    
399
static V9fsFidState *alloc_fid(V9fsState *s, int32_t fid)
400
{
401
    V9fsFidState *f;
402

    
403
    f = lookup_fid(s, fid);
404
    if (f) {
405
        return NULL;
406
    }
407

    
408
    f = qemu_mallocz(sizeof(V9fsFidState));
409

    
410
    f->fid = fid;
411
    f->fd = -1;
412
    f->dir = NULL;
413

    
414
    f->next = s->fid_list;
415
    s->fid_list = f;
416

    
417
    return f;
418
}
419

    
420
static int free_fid(V9fsState *s, int32_t fid)
421
{
422
    V9fsFidState **fidpp, *fidp;
423

    
424
    for (fidpp = &s->fid_list; *fidpp; fidpp = &(*fidpp)->next) {
425
        if ((*fidpp)->fid == fid) {
426
            break;
427
        }
428
    }
429

    
430
    if (*fidpp == NULL) {
431
        return -ENOENT;
432
    }
433

    
434
    fidp = *fidpp;
435
    *fidpp = fidp->next;
436

    
437
    if (fidp->fd != -1) {
438
        v9fs_do_close(s, fidp->fd);
439
    }
440
    if (fidp->dir) {
441
        v9fs_do_closedir(s, fidp->dir);
442
    }
443
    v9fs_string_free(&fidp->path);
444
    qemu_free(fidp);
445

    
446
    return 0;
447
}
448

    
449
#define P9_QID_TYPE_DIR         0x80
450
#define P9_QID_TYPE_SYMLINK     0x02
451

    
452
#define P9_STAT_MODE_DIR        0x80000000
453
#define P9_STAT_MODE_APPEND     0x40000000
454
#define P9_STAT_MODE_EXCL       0x20000000
455
#define P9_STAT_MODE_MOUNT      0x10000000
456
#define P9_STAT_MODE_AUTH       0x08000000
457
#define P9_STAT_MODE_TMP        0x04000000
458
#define P9_STAT_MODE_SYMLINK    0x02000000
459
#define P9_STAT_MODE_LINK       0x01000000
460
#define P9_STAT_MODE_DEVICE     0x00800000
461
#define P9_STAT_MODE_NAMED_PIPE 0x00200000
462
#define P9_STAT_MODE_SOCKET     0x00100000
463
#define P9_STAT_MODE_SETUID     0x00080000
464
#define P9_STAT_MODE_SETGID     0x00040000
465
#define P9_STAT_MODE_SETVTX     0x00010000
466

    
467
#define P9_STAT_MODE_TYPE_BITS (P9_STAT_MODE_DIR |          \
468
                                P9_STAT_MODE_SYMLINK |      \
469
                                P9_STAT_MODE_LINK |         \
470
                                P9_STAT_MODE_DEVICE |       \
471
                                P9_STAT_MODE_NAMED_PIPE |   \
472
                                P9_STAT_MODE_SOCKET)
473

    
474
/* This is the algorithm from ufs in spfs */
475
static void stat_to_qid(const struct stat *stbuf, V9fsQID *qidp)
476
{
477
    size_t size;
478

    
479
    size = MIN(sizeof(stbuf->st_ino), sizeof(qidp->path));
480
    memcpy(&qidp->path, &stbuf->st_ino, size);
481
    qidp->version = stbuf->st_mtime ^ (stbuf->st_size << 8);
482
    qidp->type = 0;
483
    if (S_ISDIR(stbuf->st_mode)) {
484
        qidp->type |= P9_QID_TYPE_DIR;
485
    }
486
    if (S_ISLNK(stbuf->st_mode)) {
487
        qidp->type |= P9_QID_TYPE_SYMLINK;
488
    }
489
}
490

    
491
static int fid_to_qid(V9fsState *s, V9fsFidState *fidp, V9fsQID *qidp)
492
{
493
    struct stat stbuf;
494
    int err;
495

    
496
    err = v9fs_do_lstat(s, &fidp->path, &stbuf);
497
    if (err) {
498
        return err;
499
    }
500

    
501
    stat_to_qid(&stbuf, qidp);
502
    return 0;
503
}
504

    
505
static V9fsPDU *alloc_pdu(V9fsState *s)
506
{
507
    V9fsPDU *pdu = NULL;
508

    
509
    if (!QLIST_EMPTY(&s->free_list)) {
510
        pdu = QLIST_FIRST(&s->free_list);
511
        QLIST_REMOVE(pdu, next);
512
    }
513
    return pdu;
514
}
515

    
516
static void free_pdu(V9fsState *s, V9fsPDU *pdu)
517
{
518
    if (pdu) {
519
        QLIST_INSERT_HEAD(&s->free_list, pdu, next);
520
    }
521
}
522

    
523
size_t pdu_packunpack(void *addr, struct iovec *sg, int sg_count,
524
                        size_t offset, size_t size, int pack)
525
{
526
    int i = 0;
527
    size_t copied = 0;
528

    
529
    for (i = 0; size && i < sg_count; i++) {
530
        size_t len;
531
        if (offset >= sg[i].iov_len) {
532
            /* skip this sg */
533
            offset -= sg[i].iov_len;
534
            continue;
535
        } else {
536
            len = MIN(sg[i].iov_len - offset, size);
537
            if (pack) {
538
                memcpy(sg[i].iov_base + offset, addr, len);
539
            } else {
540
                memcpy(addr, sg[i].iov_base + offset, len);
541
            }
542
            size -= len;
543
            copied += len;
544
            addr += len;
545
            if (size) {
546
                offset = 0;
547
                continue;
548
            }
549
        }
550
    }
551

    
552
    return copied;
553
}
554

    
555
static size_t pdu_unpack(void *dst, V9fsPDU *pdu, size_t offset, size_t size)
556
{
557
    return pdu_packunpack(dst, pdu->elem.out_sg, pdu->elem.out_num,
558
                         offset, size, 0);
559
}
560

    
561
static size_t pdu_pack(V9fsPDU *pdu, size_t offset, const void *src,
562
                        size_t size)
563
{
564
    return pdu_packunpack((void *)src, pdu->elem.in_sg, pdu->elem.in_num,
565
                             offset, size, 1);
566
}
567

    
568
static int pdu_copy_sg(V9fsPDU *pdu, size_t offset, int rx, struct iovec *sg)
569
{
570
    size_t pos = 0;
571
    int i, j;
572
    struct iovec *src_sg;
573
    unsigned int num;
574

    
575
    if (rx) {
576
        src_sg = pdu->elem.in_sg;
577
        num = pdu->elem.in_num;
578
    } else {
579
        src_sg = pdu->elem.out_sg;
580
        num = pdu->elem.out_num;
581
    }
582

    
583
    j = 0;
584
    for (i = 0; i < num; i++) {
585
        if (offset <= pos) {
586
            sg[j].iov_base = src_sg[i].iov_base;
587
            sg[j].iov_len = src_sg[i].iov_len;
588
            j++;
589
        } else if (offset < (src_sg[i].iov_len + pos)) {
590
            sg[j].iov_base = src_sg[i].iov_base;
591
            sg[j].iov_len = src_sg[i].iov_len;
592
            sg[j].iov_base += (offset - pos);
593
            sg[j].iov_len -= (offset - pos);
594
            j++;
595
        }
596
        pos += src_sg[i].iov_len;
597
    }
598

    
599
    return j;
600
}
601

    
602
static size_t pdu_unmarshal(V9fsPDU *pdu, size_t offset, const char *fmt, ...)
603
{
604
    size_t old_offset = offset;
605
    va_list ap;
606
    int i;
607

    
608
    va_start(ap, fmt);
609
    for (i = 0; fmt[i]; i++) {
610
        switch (fmt[i]) {
611
        case 'b': {
612
            uint8_t *valp = va_arg(ap, uint8_t *);
613
            offset += pdu_unpack(valp, pdu, offset, sizeof(*valp));
614
            break;
615
        }
616
        case 'w': {
617
            uint16_t val, *valp;
618
            valp = va_arg(ap, uint16_t *);
619
            val = le16_to_cpupu(valp);
620
            offset += pdu_unpack(&val, pdu, offset, sizeof(val));
621
            *valp = val;
622
            break;
623
        }
624
        case 'd': {
625
            uint32_t val, *valp;
626
            valp = va_arg(ap, uint32_t *);
627
            val = le32_to_cpupu(valp);
628
            offset += pdu_unpack(&val, pdu, offset, sizeof(val));
629
            *valp = val;
630
            break;
631
        }
632
        case 'q': {
633
            uint64_t val, *valp;
634
            valp = va_arg(ap, uint64_t *);
635
            val = le64_to_cpup(valp);
636
            offset += pdu_unpack(&val, pdu, offset, sizeof(val));
637
            *valp = val;
638
            break;
639
        }
640
        case 'v': {
641
            struct iovec *iov = va_arg(ap, struct iovec *);
642
            int *iovcnt = va_arg(ap, int *);
643
            *iovcnt = pdu_copy_sg(pdu, offset, 0, iov);
644
            break;
645
        }
646
        case 's': {
647
            V9fsString *str = va_arg(ap, V9fsString *);
648
            offset += pdu_unmarshal(pdu, offset, "w", &str->size);
649
            /* FIXME: sanity check str->size */
650
            str->data = qemu_malloc(str->size + 1);
651
            offset += pdu_unpack(str->data, pdu, offset, str->size);
652
            str->data[str->size] = 0;
653
            break;
654
        }
655
        case 'Q': {
656
            V9fsQID *qidp = va_arg(ap, V9fsQID *);
657
            offset += pdu_unmarshal(pdu, offset, "bdq",
658
                        &qidp->type, &qidp->version, &qidp->path);
659
            break;
660
        }
661
        case 'S': {
662
            V9fsStat *statp = va_arg(ap, V9fsStat *);
663
            offset += pdu_unmarshal(pdu, offset, "wwdQdddqsssssddd",
664
                        &statp->size, &statp->type, &statp->dev,
665
                        &statp->qid, &statp->mode, &statp->atime,
666
                        &statp->mtime, &statp->length,
667
                        &statp->name, &statp->uid, &statp->gid,
668
                        &statp->muid, &statp->extension,
669
                        &statp->n_uid, &statp->n_gid,
670
                        &statp->n_muid);
671
            break;
672
        }
673
        case 'I': {
674
            V9fsIattr *iattr = va_arg(ap, V9fsIattr *);
675
            offset += pdu_unmarshal(pdu, offset, "ddddqqqqq",
676
                        &iattr->valid, &iattr->mode,
677
                        &iattr->uid, &iattr->gid, &iattr->size,
678
                        &iattr->atime_sec, &iattr->atime_nsec,
679
                        &iattr->mtime_sec, &iattr->mtime_nsec);
680
            break;
681
        }
682
        default:
683
            break;
684
        }
685
    }
686

    
687
    va_end(ap);
688

    
689
    return offset - old_offset;
690
}
691

    
692
static size_t pdu_marshal(V9fsPDU *pdu, size_t offset, const char *fmt, ...)
693
{
694
    size_t old_offset = offset;
695
    va_list ap;
696
    int i;
697

    
698
    va_start(ap, fmt);
699
    for (i = 0; fmt[i]; i++) {
700
        switch (fmt[i]) {
701
        case 'b': {
702
            uint8_t val = va_arg(ap, int);
703
            offset += pdu_pack(pdu, offset, &val, sizeof(val));
704
            break;
705
        }
706
        case 'w': {
707
            uint16_t val;
708
            cpu_to_le16w(&val, va_arg(ap, int));
709
            offset += pdu_pack(pdu, offset, &val, sizeof(val));
710
            break;
711
        }
712
        case 'd': {
713
            uint32_t val;
714
            cpu_to_le32w(&val, va_arg(ap, uint32_t));
715
            offset += pdu_pack(pdu, offset, &val, sizeof(val));
716
            break;
717
        }
718
        case 'q': {
719
            uint64_t val;
720
            cpu_to_le64w(&val, va_arg(ap, uint64_t));
721
            offset += pdu_pack(pdu, offset, &val, sizeof(val));
722
            break;
723
        }
724
        case 'v': {
725
            struct iovec *iov = va_arg(ap, struct iovec *);
726
            int *iovcnt = va_arg(ap, int *);
727
            *iovcnt = pdu_copy_sg(pdu, offset, 1, iov);
728
            break;
729
        }
730
        case 's': {
731
            V9fsString *str = va_arg(ap, V9fsString *);
732
            offset += pdu_marshal(pdu, offset, "w", str->size);
733
            offset += pdu_pack(pdu, offset, str->data, str->size);
734
            break;
735
        }
736
        case 'Q': {
737
            V9fsQID *qidp = va_arg(ap, V9fsQID *);
738
            offset += pdu_marshal(pdu, offset, "bdq",
739
                        qidp->type, qidp->version, qidp->path);
740
            break;
741
        }
742
        case 'S': {
743
            V9fsStat *statp = va_arg(ap, V9fsStat *);
744
            offset += pdu_marshal(pdu, offset, "wwdQdddqsssssddd",
745
                        statp->size, statp->type, statp->dev,
746
                        &statp->qid, statp->mode, statp->atime,
747
                        statp->mtime, statp->length, &statp->name,
748
                        &statp->uid, &statp->gid, &statp->muid,
749
                        &statp->extension, statp->n_uid,
750
                        statp->n_gid, statp->n_muid);
751
            break;
752
        }
753
        case 'A': {
754
            V9fsStatDotl *statp = va_arg(ap, V9fsStatDotl *);
755
            offset += pdu_marshal(pdu, offset, "qQdddqqqqqqqqqqqqqqq",
756
                        statp->st_result_mask,
757
                        &statp->qid, statp->st_mode,
758
                        statp->st_uid, statp->st_gid,
759
                        statp->st_nlink, statp->st_rdev,
760
                        statp->st_size, statp->st_blksize, statp->st_blocks,
761
                        statp->st_atime_sec, statp->st_atime_nsec,
762
                        statp->st_mtime_sec, statp->st_mtime_nsec,
763
                        statp->st_ctime_sec, statp->st_ctime_nsec,
764
                        statp->st_btime_sec, statp->st_btime_nsec,
765
                        statp->st_gen, statp->st_data_version);
766
            break;
767
        }
768
        default:
769
            break;
770
        }
771
    }
772
    va_end(ap);
773

    
774
    return offset - old_offset;
775
}
776

    
777
static void complete_pdu(V9fsState *s, V9fsPDU *pdu, ssize_t len)
778
{
779
    int8_t id = pdu->id + 1; /* Response */
780

    
781
    if (len < 0) {
782
        V9fsString str;
783
        int err = -len;
784

    
785
        str.data = strerror(err);
786
        str.size = strlen(str.data);
787

    
788
        len = 7;
789
        len += pdu_marshal(pdu, len, "s", &str);
790
        if (dotu) {
791
            len += pdu_marshal(pdu, len, "d", err);
792
        }
793

    
794
        id = P9_RERROR;
795
    }
796

    
797
    /* fill out the header */
798
    pdu_marshal(pdu, 0, "dbw", (int32_t)len, id, pdu->tag);
799

    
800
    /* keep these in sync */
801
    pdu->size = len;
802
    pdu->id = id;
803

    
804
    /* push onto queue and notify */
805
    virtqueue_push(s->vq, &pdu->elem, len);
806

    
807
    /* FIXME: we should batch these completions */
808
    virtio_notify(&s->vdev, s->vq);
809

    
810
    free_pdu(s, pdu);
811
}
812

    
813
static mode_t v9mode_to_mode(uint32_t mode, V9fsString *extension)
814
{
815
    mode_t ret;
816

    
817
    ret = mode & 0777;
818
    if (mode & P9_STAT_MODE_DIR) {
819
        ret |= S_IFDIR;
820
    }
821

    
822
    if (dotu) {
823
        if (mode & P9_STAT_MODE_SYMLINK) {
824
            ret |= S_IFLNK;
825
        }
826
        if (mode & P9_STAT_MODE_SOCKET) {
827
            ret |= S_IFSOCK;
828
        }
829
        if (mode & P9_STAT_MODE_NAMED_PIPE) {
830
            ret |= S_IFIFO;
831
        }
832
        if (mode & P9_STAT_MODE_DEVICE) {
833
            if (extension && extension->data[0] == 'c') {
834
                ret |= S_IFCHR;
835
            } else {
836
                ret |= S_IFBLK;
837
            }
838
        }
839
    }
840

    
841
    if (!(ret&~0777)) {
842
        ret |= S_IFREG;
843
    }
844

    
845
    if (mode & P9_STAT_MODE_SETUID) {
846
        ret |= S_ISUID;
847
    }
848
    if (mode & P9_STAT_MODE_SETGID) {
849
        ret |= S_ISGID;
850
    }
851
    if (mode & P9_STAT_MODE_SETVTX) {
852
        ret |= S_ISVTX;
853
    }
854

    
855
    return ret;
856
}
857

    
858
static int donttouch_stat(V9fsStat *stat)
859
{
860
    if (stat->type == -1 &&
861
        stat->dev == -1 &&
862
        stat->qid.type == -1 &&
863
        stat->qid.version == -1 &&
864
        stat->qid.path == -1 &&
865
        stat->mode == -1 &&
866
        stat->atime == -1 &&
867
        stat->mtime == -1 &&
868
        stat->length == -1 &&
869
        !stat->name.size &&
870
        !stat->uid.size &&
871
        !stat->gid.size &&
872
        !stat->muid.size &&
873
        stat->n_uid == -1 &&
874
        stat->n_gid == -1 &&
875
        stat->n_muid == -1) {
876
        return 1;
877
    }
878

    
879
    return 0;
880
}
881

    
882
static void v9fs_stat_free(V9fsStat *stat)
883
{
884
    v9fs_string_free(&stat->name);
885
    v9fs_string_free(&stat->uid);
886
    v9fs_string_free(&stat->gid);
887
    v9fs_string_free(&stat->muid);
888
    v9fs_string_free(&stat->extension);
889
}
890

    
891
static uint32_t stat_to_v9mode(const struct stat *stbuf)
892
{
893
    uint32_t mode;
894

    
895
    mode = stbuf->st_mode & 0777;
896
    if (S_ISDIR(stbuf->st_mode)) {
897
        mode |= P9_STAT_MODE_DIR;
898
    }
899

    
900
    if (dotu) {
901
        if (S_ISLNK(stbuf->st_mode)) {
902
            mode |= P9_STAT_MODE_SYMLINK;
903
        }
904

    
905
        if (S_ISSOCK(stbuf->st_mode)) {
906
            mode |= P9_STAT_MODE_SOCKET;
907
        }
908

    
909
        if (S_ISFIFO(stbuf->st_mode)) {
910
            mode |= P9_STAT_MODE_NAMED_PIPE;
911
        }
912

    
913
        if (S_ISBLK(stbuf->st_mode) || S_ISCHR(stbuf->st_mode)) {
914
            mode |= P9_STAT_MODE_DEVICE;
915
        }
916

    
917
        if (stbuf->st_mode & S_ISUID) {
918
            mode |= P9_STAT_MODE_SETUID;
919
        }
920

    
921
        if (stbuf->st_mode & S_ISGID) {
922
            mode |= P9_STAT_MODE_SETGID;
923
        }
924

    
925
        if (stbuf->st_mode & S_ISVTX) {
926
            mode |= P9_STAT_MODE_SETVTX;
927
        }
928
    }
929

    
930
    return mode;
931
}
932

    
933
static int stat_to_v9stat(V9fsState *s, V9fsString *name,
934
                            const struct stat *stbuf,
935
                            V9fsStat *v9stat)
936
{
937
    int err;
938
    const char *str;
939

    
940
    memset(v9stat, 0, sizeof(*v9stat));
941

    
942
    stat_to_qid(stbuf, &v9stat->qid);
943
    v9stat->mode = stat_to_v9mode(stbuf);
944
    v9stat->atime = stbuf->st_atime;
945
    v9stat->mtime = stbuf->st_mtime;
946
    v9stat->length = stbuf->st_size;
947

    
948
    v9fs_string_null(&v9stat->uid);
949
    v9fs_string_null(&v9stat->gid);
950
    v9fs_string_null(&v9stat->muid);
951

    
952
    if (dotu) {
953
        v9stat->n_uid = stbuf->st_uid;
954
        v9stat->n_gid = stbuf->st_gid;
955
        v9stat->n_muid = 0;
956

    
957
        v9fs_string_null(&v9stat->extension);
958

    
959
        if (v9stat->mode & P9_STAT_MODE_SYMLINK) {
960
            err = v9fs_do_readlink(s, name, &v9stat->extension);
961
            if (err == -1) {
962
                err = -errno;
963
                return err;
964
            }
965
            v9stat->extension.data[err] = 0;
966
            v9stat->extension.size = err;
967
        } else if (v9stat->mode & P9_STAT_MODE_DEVICE) {
968
            v9fs_string_sprintf(&v9stat->extension, "%c %u %u",
969
                    S_ISCHR(stbuf->st_mode) ? 'c' : 'b',
970
                    major(stbuf->st_rdev), minor(stbuf->st_rdev));
971
        } else if (S_ISDIR(stbuf->st_mode) || S_ISREG(stbuf->st_mode)) {
972
            v9fs_string_sprintf(&v9stat->extension, "%s %u",
973
                    "HARDLINKCOUNT", stbuf->st_nlink);
974
        }
975
    }
976

    
977
    str = strrchr(name->data, '/');
978
    if (str) {
979
        str += 1;
980
    } else {
981
        str = name->data;
982
    }
983

    
984
    v9fs_string_sprintf(&v9stat->name, "%s", str);
985

    
986
    v9stat->size = 61 +
987
        v9fs_string_size(&v9stat->name) +
988
        v9fs_string_size(&v9stat->uid) +
989
        v9fs_string_size(&v9stat->gid) +
990
        v9fs_string_size(&v9stat->muid) +
991
        v9fs_string_size(&v9stat->extension);
992
    return 0;
993
}
994

    
995
#define P9_STATS_MODE          0x00000001ULL
996
#define P9_STATS_NLINK         0x00000002ULL
997
#define P9_STATS_UID           0x00000004ULL
998
#define P9_STATS_GID           0x00000008ULL
999
#define P9_STATS_RDEV          0x00000010ULL
1000
#define P9_STATS_ATIME         0x00000020ULL
1001
#define P9_STATS_MTIME         0x00000040ULL
1002
#define P9_STATS_CTIME         0x00000080ULL
1003
#define P9_STATS_INO           0x00000100ULL
1004
#define P9_STATS_SIZE          0x00000200ULL
1005
#define P9_STATS_BLOCKS        0x00000400ULL
1006

    
1007
#define P9_STATS_BTIME         0x00000800ULL
1008
#define P9_STATS_GEN           0x00001000ULL
1009
#define P9_STATS_DATA_VERSION  0x00002000ULL
1010

    
1011
#define P9_STATS_BASIC         0x000007ffULL /* Mask for fields up to BLOCKS */
1012
#define P9_STATS_ALL           0x00003fffULL /* Mask for All fields above */
1013

    
1014

    
1015
static void stat_to_v9stat_dotl(V9fsState *s, const struct stat *stbuf,
1016
                            V9fsStatDotl *v9lstat)
1017
{
1018
    memset(v9lstat, 0, sizeof(*v9lstat));
1019

    
1020
    v9lstat->st_mode = stbuf->st_mode;
1021
    v9lstat->st_nlink = stbuf->st_nlink;
1022
    v9lstat->st_uid = stbuf->st_uid;
1023
    v9lstat->st_gid = stbuf->st_gid;
1024
    v9lstat->st_rdev = stbuf->st_rdev;
1025
    v9lstat->st_size = stbuf->st_size;
1026
    v9lstat->st_blksize = stbuf->st_blksize;
1027
    v9lstat->st_blocks = stbuf->st_blocks;
1028
    v9lstat->st_atime_sec = stbuf->st_atime;
1029
    v9lstat->st_atime_nsec = stbuf->st_atim.tv_nsec;
1030
    v9lstat->st_mtime_sec = stbuf->st_mtime;
1031
    v9lstat->st_mtime_nsec = stbuf->st_mtim.tv_nsec;
1032
    v9lstat->st_ctime_sec = stbuf->st_ctime;
1033
    v9lstat->st_ctime_nsec = stbuf->st_ctim.tv_nsec;
1034
    /* Currently we only support BASIC fields in stat */
1035
    v9lstat->st_result_mask = P9_STATS_BASIC;
1036

    
1037
    stat_to_qid(stbuf, &v9lstat->qid);
1038
}
1039

    
1040
static struct iovec *adjust_sg(struct iovec *sg, int len, int *iovcnt)
1041
{
1042
    while (len && *iovcnt) {
1043
        if (len < sg->iov_len) {
1044
            sg->iov_len -= len;
1045
            sg->iov_base += len;
1046
            len = 0;
1047
        } else {
1048
            len -= sg->iov_len;
1049
            sg++;
1050
            *iovcnt -= 1;
1051
        }
1052
    }
1053

    
1054
    return sg;
1055
}
1056

    
1057
static struct iovec *cap_sg(struct iovec *sg, int cap, int *cnt)
1058
{
1059
    int i;
1060
    int total = 0;
1061

    
1062
    for (i = 0; i < *cnt; i++) {
1063
        if ((total + sg[i].iov_len) > cap) {
1064
            sg[i].iov_len -= ((total + sg[i].iov_len) - cap);
1065
            i++;
1066
            break;
1067
        }
1068
        total += sg[i].iov_len;
1069
    }
1070

    
1071
    *cnt = i;
1072

    
1073
    return sg;
1074
}
1075

    
1076
static void print_sg(struct iovec *sg, int cnt)
1077
{
1078
    int i;
1079

    
1080
    printf("sg[%d]: {", cnt);
1081
    for (i = 0; i < cnt; i++) {
1082
        if (i) {
1083
            printf(", ");
1084
        }
1085
        printf("(%p, %zd)", sg[i].iov_base, sg[i].iov_len);
1086
    }
1087
    printf("}\n");
1088
}
1089

    
1090
static void v9fs_fix_path(V9fsString *dst, V9fsString *src, int len)
1091
{
1092
    V9fsString str;
1093
    v9fs_string_init(&str);
1094
    v9fs_string_copy(&str, dst);
1095
    v9fs_string_sprintf(dst, "%s%s", src->data, str.data+len);
1096
    v9fs_string_free(&str);
1097
}
1098

    
1099
static void v9fs_version(V9fsState *s, V9fsPDU *pdu)
1100
{
1101
    V9fsString version;
1102
    size_t offset = 7;
1103

    
1104
    pdu_unmarshal(pdu, offset, "ds", &s->msize, &version);
1105

    
1106
    if (!strcmp(version.data, "9P2000.u")) {
1107
        s->proto_version = V9FS_PROTO_2000U;
1108
    } else if (!strcmp(version.data, "9P2000.L")) {
1109
        s->proto_version = V9FS_PROTO_2000L;
1110
    } else {
1111
        v9fs_string_sprintf(&version, "unknown");
1112
    }
1113

    
1114
    offset += pdu_marshal(pdu, offset, "ds", s->msize, &version);
1115
    complete_pdu(s, pdu, offset);
1116

    
1117
    v9fs_string_free(&version);
1118
}
1119

    
1120
static void v9fs_attach(V9fsState *s, V9fsPDU *pdu)
1121
{
1122
    int32_t fid, afid, n_uname;
1123
    V9fsString uname, aname;
1124
    V9fsFidState *fidp;
1125
    V9fsQID qid;
1126
    size_t offset = 7;
1127
    ssize_t err;
1128

    
1129
    pdu_unmarshal(pdu, offset, "ddssd", &fid, &afid, &uname, &aname, &n_uname);
1130

    
1131
    fidp = alloc_fid(s, fid);
1132
    if (fidp == NULL) {
1133
        err = -EINVAL;
1134
        goto out;
1135
    }
1136

    
1137
    fidp->uid = n_uname;
1138

    
1139
    v9fs_string_sprintf(&fidp->path, "%s", "/");
1140
    err = fid_to_qid(s, fidp, &qid);
1141
    if (err) {
1142
        err = -EINVAL;
1143
        free_fid(s, fid);
1144
        goto out;
1145
    }
1146

    
1147
    offset += pdu_marshal(pdu, offset, "Q", &qid);
1148

    
1149
    err = offset;
1150
out:
1151
    complete_pdu(s, pdu, err);
1152
    v9fs_string_free(&uname);
1153
    v9fs_string_free(&aname);
1154
}
1155

    
1156
static void v9fs_stat_post_lstat(V9fsState *s, V9fsStatState *vs, int err)
1157
{
1158
    if (err == -1) {
1159
        err = -errno;
1160
        goto out;
1161
    }
1162

    
1163
    err = stat_to_v9stat(s, &vs->fidp->path, &vs->stbuf, &vs->v9stat);
1164
    if (err) {
1165
        goto out;
1166
    }
1167
    vs->offset += pdu_marshal(vs->pdu, vs->offset, "wS", 0, &vs->v9stat);
1168
    err = vs->offset;
1169

    
1170
out:
1171
    complete_pdu(s, vs->pdu, err);
1172
    v9fs_stat_free(&vs->v9stat);
1173
    qemu_free(vs);
1174
}
1175

    
1176
static void v9fs_stat(V9fsState *s, V9fsPDU *pdu)
1177
{
1178
    int32_t fid;
1179
    V9fsStatState *vs;
1180
    ssize_t err = 0;
1181

    
1182
    vs = qemu_malloc(sizeof(*vs));
1183
    vs->pdu = pdu;
1184
    vs->offset = 7;
1185

    
1186
    memset(&vs->v9stat, 0, sizeof(vs->v9stat));
1187

    
1188
    pdu_unmarshal(vs->pdu, vs->offset, "d", &fid);
1189

    
1190
    vs->fidp = lookup_fid(s, fid);
1191
    if (vs->fidp == NULL) {
1192
        err = -ENOENT;
1193
        goto out;
1194
    }
1195

    
1196
    err = v9fs_do_lstat(s, &vs->fidp->path, &vs->stbuf);
1197
    v9fs_stat_post_lstat(s, vs, err);
1198
    return;
1199

    
1200
out:
1201
    complete_pdu(s, vs->pdu, err);
1202
    v9fs_stat_free(&vs->v9stat);
1203
    qemu_free(vs);
1204
}
1205

    
1206
static void v9fs_getattr_post_lstat(V9fsState *s, V9fsStatStateDotl *vs,
1207
                                                                int err)
1208
{
1209
    if (err == -1) {
1210
        err = -errno;
1211
        goto out;
1212
    }
1213

    
1214
    stat_to_v9stat_dotl(s, &vs->stbuf, &vs->v9stat_dotl);
1215
    vs->offset += pdu_marshal(vs->pdu, vs->offset, "A", &vs->v9stat_dotl);
1216
    err = vs->offset;
1217

    
1218
out:
1219
    complete_pdu(s, vs->pdu, err);
1220
    qemu_free(vs);
1221
}
1222

    
1223
static void v9fs_getattr(V9fsState *s, V9fsPDU *pdu)
1224
{
1225
    int32_t fid;
1226
    V9fsStatStateDotl *vs;
1227
    ssize_t err = 0;
1228
    V9fsFidState *fidp;
1229
    uint64_t request_mask;
1230

    
1231
    vs = qemu_malloc(sizeof(*vs));
1232
    vs->pdu = pdu;
1233
    vs->offset = 7;
1234

    
1235
    memset(&vs->v9stat_dotl, 0, sizeof(vs->v9stat_dotl));
1236

    
1237
    pdu_unmarshal(vs->pdu, vs->offset, "dq", &fid, &request_mask);
1238

    
1239
    fidp = lookup_fid(s, fid);
1240
    if (fidp == NULL) {
1241
        err = -ENOENT;
1242
        goto out;
1243
    }
1244

    
1245
    /* Currently we only support BASIC fields in stat, so there is no
1246
     * need to look at request_mask.
1247
     */
1248
    err = v9fs_do_lstat(s, &fidp->path, &vs->stbuf);
1249
    v9fs_getattr_post_lstat(s, vs, err);
1250
    return;
1251

    
1252
out:
1253
    complete_pdu(s, vs->pdu, err);
1254
    qemu_free(vs);
1255
}
1256

    
1257
/* From Linux kernel code */
1258
#define ATTR_MODE    (1 << 0)
1259
#define ATTR_UID     (1 << 1)
1260
#define ATTR_GID     (1 << 2)
1261
#define ATTR_SIZE    (1 << 3)
1262
#define ATTR_ATIME   (1 << 4)
1263
#define ATTR_MTIME   (1 << 5)
1264
#define ATTR_CTIME   (1 << 6)
1265
#define ATTR_MASK    127
1266
#define ATTR_ATIME_SET  (1 << 7)
1267
#define ATTR_MTIME_SET  (1 << 8)
1268

    
1269
static void v9fs_setattr_post_truncate(V9fsState *s, V9fsSetattrState *vs,
1270
                                                                  int err)
1271
{
1272
    if (err == -1) {
1273
        err = -errno;
1274
        goto out;
1275
    }
1276
    err = vs->offset;
1277

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

    
1283
static void v9fs_setattr_post_chown(V9fsState *s, V9fsSetattrState *vs, int err)
1284
{
1285
    if (err == -1) {
1286
        err = -errno;
1287
        goto out;
1288
    }
1289

    
1290
    if (vs->v9iattr.valid & (ATTR_SIZE)) {
1291
        err = v9fs_do_truncate(s, &vs->fidp->path, vs->v9iattr.size);
1292
    }
1293
    v9fs_setattr_post_truncate(s, vs, err);
1294
    return;
1295

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

    
1301
static void v9fs_setattr_post_utimensat(V9fsState *s, V9fsSetattrState *vs,
1302
                                                                   int err)
1303
{
1304
    if (err == -1) {
1305
        err = -errno;
1306
        goto out;
1307
    }
1308

    
1309
    /* If the only valid entry in iattr is ctime we can call
1310
     * chown(-1,-1) to update the ctime of the file
1311
     */
1312
    if ((vs->v9iattr.valid & (ATTR_UID | ATTR_GID)) ||
1313
            ((vs->v9iattr.valid & ATTR_CTIME)
1314
            && !((vs->v9iattr.valid & ATTR_MASK) & ~ATTR_CTIME))) {
1315
        if (!(vs->v9iattr.valid & ATTR_UID)) {
1316
            vs->v9iattr.uid = -1;
1317
        }
1318
        if (!(vs->v9iattr.valid & ATTR_GID)) {
1319
            vs->v9iattr.gid = -1;
1320
        }
1321
        err = v9fs_do_chown(s, &vs->fidp->path, vs->v9iattr.uid,
1322
                                                vs->v9iattr.gid);
1323
    }
1324
    v9fs_setattr_post_chown(s, vs, err);
1325
    return;
1326

    
1327
out:
1328
    complete_pdu(s, vs->pdu, err);
1329
    qemu_free(vs);
1330
}
1331

    
1332
static void v9fs_setattr_post_chmod(V9fsState *s, V9fsSetattrState *vs, int err)
1333
{
1334
    if (err == -1) {
1335
        err = -errno;
1336
        goto out;
1337
    }
1338

    
1339
    if (vs->v9iattr.valid & (ATTR_ATIME | ATTR_MTIME)) {
1340
        struct timespec times[2];
1341
        if (vs->v9iattr.valid & ATTR_ATIME) {
1342
            if (vs->v9iattr.valid & ATTR_ATIME_SET) {
1343
                times[0].tv_sec = vs->v9iattr.atime_sec;
1344
                times[0].tv_nsec = vs->v9iattr.atime_nsec;
1345
            } else {
1346
                times[0].tv_nsec = UTIME_NOW;
1347
            }
1348
        } else {
1349
            times[0].tv_nsec = UTIME_OMIT;
1350
        }
1351

    
1352
        if (vs->v9iattr.valid & ATTR_MTIME) {
1353
            if (vs->v9iattr.valid & ATTR_MTIME_SET) {
1354
                times[1].tv_sec = vs->v9iattr.mtime_sec;
1355
                times[1].tv_nsec = vs->v9iattr.mtime_nsec;
1356
            } else {
1357
                times[1].tv_nsec = UTIME_NOW;
1358
            }
1359
        } else {
1360
            times[1].tv_nsec = UTIME_OMIT;
1361
        }
1362
        err = v9fs_do_utimensat(s, &vs->fidp->path, times);
1363
    }
1364
    v9fs_setattr_post_utimensat(s, vs, err);
1365
    return;
1366

    
1367
out:
1368
    complete_pdu(s, vs->pdu, err);
1369
    qemu_free(vs);
1370
}
1371

    
1372
static void v9fs_setattr(V9fsState *s, V9fsPDU *pdu)
1373
{
1374
    int32_t fid;
1375
    V9fsSetattrState *vs;
1376
    int err = 0;
1377

    
1378
    vs = qemu_malloc(sizeof(*vs));
1379
    vs->pdu = pdu;
1380
    vs->offset = 7;
1381

    
1382
    pdu_unmarshal(pdu, vs->offset, "dI", &fid, &vs->v9iattr);
1383

    
1384
    vs->fidp = lookup_fid(s, fid);
1385
    if (vs->fidp == NULL) {
1386
        err = -EINVAL;
1387
        goto out;
1388
    }
1389

    
1390
    if (vs->v9iattr.valid & ATTR_MODE) {
1391
        err = v9fs_do_chmod(s, &vs->fidp->path, vs->v9iattr.mode);
1392
    }
1393

    
1394
    v9fs_setattr_post_chmod(s, vs, err);
1395
    return;
1396

    
1397
out:
1398
    complete_pdu(s, vs->pdu, err);
1399
    qemu_free(vs);
1400
}
1401

    
1402
static void v9fs_walk_complete(V9fsState *s, V9fsWalkState *vs, int err)
1403
{
1404
    complete_pdu(s, vs->pdu, err);
1405

    
1406
    if (vs->nwnames) {
1407
        for (vs->name_idx = 0; vs->name_idx < vs->nwnames; vs->name_idx++) {
1408
            v9fs_string_free(&vs->wnames[vs->name_idx]);
1409
        }
1410

    
1411
        qemu_free(vs->wnames);
1412
        qemu_free(vs->qids);
1413
    }
1414
}
1415

    
1416
static void v9fs_walk_marshal(V9fsWalkState *vs)
1417
{
1418
    int i;
1419
    vs->offset = 7;
1420
    vs->offset += pdu_marshal(vs->pdu, vs->offset, "w", vs->nwnames);
1421

    
1422
    for (i = 0; i < vs->nwnames; i++) {
1423
        vs->offset += pdu_marshal(vs->pdu, vs->offset, "Q", &vs->qids[i]);
1424
    }
1425
}
1426

    
1427
static void v9fs_walk_post_newfid_lstat(V9fsState *s, V9fsWalkState *vs,
1428
                                                                int err)
1429
{
1430
    if (err == -1) {
1431
        free_fid(s, vs->newfidp->fid);
1432
        v9fs_string_free(&vs->path);
1433
        err = -ENOENT;
1434
        goto out;
1435
    }
1436

    
1437
    stat_to_qid(&vs->stbuf, &vs->qids[vs->name_idx]);
1438

    
1439
    vs->name_idx++;
1440
    if (vs->name_idx < vs->nwnames) {
1441
        v9fs_string_sprintf(&vs->path, "%s/%s", vs->newfidp->path.data,
1442
                                            vs->wnames[vs->name_idx].data);
1443
        v9fs_string_copy(&vs->newfidp->path, &vs->path);
1444

    
1445
        err = v9fs_do_lstat(s, &vs->newfidp->path, &vs->stbuf);
1446
        v9fs_walk_post_newfid_lstat(s, vs, err);
1447
        return;
1448
    }
1449

    
1450
    v9fs_string_free(&vs->path);
1451
    v9fs_walk_marshal(vs);
1452
    err = vs->offset;
1453
out:
1454
    v9fs_walk_complete(s, vs, err);
1455
}
1456

    
1457
static void v9fs_walk_post_oldfid_lstat(V9fsState *s, V9fsWalkState *vs,
1458
        int err)
1459
{
1460
    if (err == -1) {
1461
        v9fs_string_free(&vs->path);
1462
        err = -ENOENT;
1463
        goto out;
1464
    }
1465

    
1466
    stat_to_qid(&vs->stbuf, &vs->qids[vs->name_idx]);
1467
    vs->name_idx++;
1468
    if (vs->name_idx < vs->nwnames) {
1469

    
1470
        v9fs_string_sprintf(&vs->path, "%s/%s",
1471
                vs->fidp->path.data, vs->wnames[vs->name_idx].data);
1472
        v9fs_string_copy(&vs->fidp->path, &vs->path);
1473

    
1474
        err = v9fs_do_lstat(s, &vs->fidp->path, &vs->stbuf);
1475
        v9fs_walk_post_oldfid_lstat(s, vs, err);
1476
        return;
1477
    }
1478

    
1479
    v9fs_string_free(&vs->path);
1480
    v9fs_walk_marshal(vs);
1481
    err = vs->offset;
1482
out:
1483
    v9fs_walk_complete(s, vs, err);
1484
}
1485

    
1486
static void v9fs_walk(V9fsState *s, V9fsPDU *pdu)
1487
{
1488
    int32_t fid, newfid;
1489
    V9fsWalkState *vs;
1490
    int err = 0;
1491
    int i;
1492

    
1493
    vs = qemu_malloc(sizeof(*vs));
1494
    vs->pdu = pdu;
1495
    vs->wnames = NULL;
1496
    vs->qids = NULL;
1497
    vs->offset = 7;
1498

    
1499
    vs->offset += pdu_unmarshal(vs->pdu, vs->offset, "ddw", &fid,
1500
                                            &newfid, &vs->nwnames);
1501

    
1502
    if (vs->nwnames) {
1503
        vs->wnames = qemu_mallocz(sizeof(vs->wnames[0]) * vs->nwnames);
1504

    
1505
        vs->qids = qemu_mallocz(sizeof(vs->qids[0]) * vs->nwnames);
1506

    
1507
        for (i = 0; i < vs->nwnames; i++) {
1508
            vs->offset += pdu_unmarshal(vs->pdu, vs->offset, "s",
1509
                                            &vs->wnames[i]);
1510
        }
1511
    }
1512

    
1513
    vs->fidp = lookup_fid(s, fid);
1514
    if (vs->fidp == NULL) {
1515
        err = -ENOENT;
1516
        goto out;
1517
    }
1518

    
1519
    /* FIXME: is this really valid? */
1520
    if (fid == newfid) {
1521

    
1522
        BUG_ON(vs->fidp->fd != -1);
1523
        BUG_ON(vs->fidp->dir);
1524
        v9fs_string_init(&vs->path);
1525
        vs->name_idx = 0;
1526

    
1527
        if (vs->name_idx < vs->nwnames) {
1528
            v9fs_string_sprintf(&vs->path, "%s/%s",
1529
                vs->fidp->path.data, vs->wnames[vs->name_idx].data);
1530
            v9fs_string_copy(&vs->fidp->path, &vs->path);
1531

    
1532
            err = v9fs_do_lstat(s, &vs->fidp->path, &vs->stbuf);
1533
            v9fs_walk_post_oldfid_lstat(s, vs, err);
1534
            return;
1535
        }
1536
    } else {
1537
        vs->newfidp = alloc_fid(s, newfid);
1538
        if (vs->newfidp == NULL) {
1539
            err = -EINVAL;
1540
            goto out;
1541
        }
1542

    
1543
        vs->newfidp->uid = vs->fidp->uid;
1544
        v9fs_string_init(&vs->path);
1545
        vs->name_idx = 0;
1546
        v9fs_string_copy(&vs->newfidp->path, &vs->fidp->path);
1547

    
1548
        if (vs->name_idx < vs->nwnames) {
1549
            v9fs_string_sprintf(&vs->path, "%s/%s", vs->newfidp->path.data,
1550
                                vs->wnames[vs->name_idx].data);
1551
            v9fs_string_copy(&vs->newfidp->path, &vs->path);
1552

    
1553
            err = v9fs_do_lstat(s, &vs->newfidp->path, &vs->stbuf);
1554
            v9fs_walk_post_newfid_lstat(s, vs, err);
1555
            return;
1556
        }
1557
    }
1558

    
1559
    v9fs_walk_marshal(vs);
1560
    err = vs->offset;
1561
out:
1562
    v9fs_walk_complete(s, vs, err);
1563
}
1564

    
1565
static int32_t get_iounit(V9fsState *s, V9fsString *name)
1566
{
1567
    struct statfs stbuf;
1568
    int32_t iounit = 0;
1569

    
1570
    /*
1571
     * iounit should be multiples of f_bsize (host filesystem block size
1572
     * and as well as less than (client msize - P9_IOHDRSZ))
1573
     */
1574
    if (!v9fs_do_statfs(s, name, &stbuf)) {
1575
        iounit = stbuf.f_bsize;
1576
        iounit *= (s->msize - P9_IOHDRSZ)/stbuf.f_bsize;
1577
    }
1578

    
1579
    if (!iounit) {
1580
        iounit = s->msize - P9_IOHDRSZ;
1581
    }
1582
    return iounit;
1583
}
1584

    
1585
static void v9fs_open_post_opendir(V9fsState *s, V9fsOpenState *vs, int err)
1586
{
1587
    if (vs->fidp->dir == NULL) {
1588
        err = -errno;
1589
        goto out;
1590
    }
1591

    
1592
    vs->offset += pdu_marshal(vs->pdu, vs->offset, "Qd", &vs->qid, 0);
1593
    err = vs->offset;
1594
out:
1595
    complete_pdu(s, vs->pdu, err);
1596
    qemu_free(vs);
1597

    
1598
}
1599

    
1600
static void v9fs_open_post_getiounit(V9fsState *s, V9fsOpenState *vs)
1601
{
1602
    int err;
1603
    vs->offset += pdu_marshal(vs->pdu, vs->offset, "Qd", &vs->qid, vs->iounit);
1604
    err = vs->offset;
1605
    complete_pdu(s, vs->pdu, err);
1606
    qemu_free(vs);
1607
}
1608

    
1609
static void v9fs_open_post_open(V9fsState *s, V9fsOpenState *vs, int err)
1610
{
1611
    if (vs->fidp->fd == -1) {
1612
        err = -errno;
1613
        goto out;
1614
    }
1615

    
1616
    vs->iounit = get_iounit(s, &vs->fidp->path);
1617
    v9fs_open_post_getiounit(s, vs);
1618
    return;
1619
out:
1620
    complete_pdu(s, vs->pdu, err);
1621
    qemu_free(vs);
1622
}
1623

    
1624
static inline int valid_flags(int flag)
1625
{
1626
    if (flag & O_NOCTTY || flag & O_NONBLOCK || flag & O_ASYNC ||
1627
            flag & O_CLOEXEC)
1628
        return 0;
1629
    else
1630
        return 1;
1631
}
1632

    
1633
static void v9fs_open_post_lstat(V9fsState *s, V9fsOpenState *vs, int err)
1634
{
1635
    int flags;
1636

    
1637
    if (err) {
1638
        err = -errno;
1639
        goto out;
1640
    }
1641

    
1642
    stat_to_qid(&vs->stbuf, &vs->qid);
1643

    
1644
    if (S_ISDIR(vs->stbuf.st_mode)) {
1645
        vs->fidp->dir = v9fs_do_opendir(s, &vs->fidp->path);
1646
        v9fs_open_post_opendir(s, vs, err);
1647
    } else {
1648
        if (s->proto_version == V9FS_PROTO_2000L) {
1649
            if (!valid_flags(vs->mode)) {
1650
                err = -EINVAL;
1651
                goto out;
1652
            }
1653
            flags = vs->mode;
1654
        } else {
1655
            flags = omode_to_uflags(vs->mode);
1656
        }
1657
        vs->fidp->fd = v9fs_do_open(s, &vs->fidp->path, flags);
1658
        v9fs_open_post_open(s, vs, err);
1659
    }
1660
    return;
1661
out:
1662
    complete_pdu(s, vs->pdu, err);
1663
    qemu_free(vs);
1664
}
1665

    
1666
static void v9fs_open(V9fsState *s, V9fsPDU *pdu)
1667
{
1668
    int32_t fid;
1669
    V9fsOpenState *vs;
1670
    ssize_t err = 0;
1671

    
1672
    vs = qemu_malloc(sizeof(*vs));
1673
    vs->pdu = pdu;
1674
    vs->offset = 7;
1675
    vs->mode = 0;
1676

    
1677
    if (s->proto_version == V9FS_PROTO_2000L) {
1678
        pdu_unmarshal(vs->pdu, vs->offset, "dd", &fid, &vs->mode);
1679
    } else {
1680
        pdu_unmarshal(vs->pdu, vs->offset, "db", &fid, &vs->mode);
1681
    }
1682

    
1683
    vs->fidp = lookup_fid(s, fid);
1684
    if (vs->fidp == NULL) {
1685
        err = -ENOENT;
1686
        goto out;
1687
    }
1688

    
1689
    BUG_ON(vs->fidp->fd != -1);
1690
    BUG_ON(vs->fidp->dir);
1691

    
1692
    err = v9fs_do_lstat(s, &vs->fidp->path, &vs->stbuf);
1693

    
1694
    v9fs_open_post_lstat(s, vs, err);
1695
    return;
1696
out:
1697
    complete_pdu(s, pdu, err);
1698
    qemu_free(vs);
1699
}
1700

    
1701
static void v9fs_post_lcreate(V9fsState *s, V9fsLcreateState *vs, int err)
1702
{
1703
    if (err == 0) {
1704
        v9fs_string_copy(&vs->fidp->path, &vs->fullname);
1705
        stat_to_qid(&vs->stbuf, &vs->qid);
1706
        vs->offset += pdu_marshal(vs->pdu, vs->offset, "Qd", &vs->qid,
1707
                &vs->iounit);
1708
        err = vs->offset;
1709
    } else {
1710
        err = -errno;
1711
    }
1712

    
1713
    complete_pdu(s, vs->pdu, err);
1714
    v9fs_string_free(&vs->name);
1715
    v9fs_string_free(&vs->fullname);
1716
    qemu_free(vs);
1717
}
1718

    
1719
static void v9fs_lcreate_post_get_iounit(V9fsState *s, V9fsLcreateState *vs,
1720
        int err)
1721
{
1722
    if (err) {
1723
        err = -errno;
1724
        goto out;
1725
    }
1726
    err = v9fs_do_lstat(s, &vs->fullname, &vs->stbuf);
1727

    
1728
out:
1729
    v9fs_post_lcreate(s, vs, err);
1730
}
1731

    
1732
static void v9fs_lcreate_post_do_open2(V9fsState *s, V9fsLcreateState *vs,
1733
        int err)
1734
{
1735
    if (vs->fidp->fd == -1) {
1736
        err = -errno;
1737
        goto out;
1738
    }
1739
    vs->iounit =  get_iounit(s, &vs->fullname);
1740
    v9fs_lcreate_post_get_iounit(s, vs, err);
1741
    return;
1742

    
1743
out:
1744
    v9fs_post_lcreate(s, vs, err);
1745
}
1746

    
1747
static void v9fs_lcreate(V9fsState *s, V9fsPDU *pdu)
1748
{
1749
    int32_t dfid, flags, mode;
1750
    gid_t gid;
1751
    V9fsLcreateState *vs;
1752
    ssize_t err = 0;
1753

    
1754
    vs = qemu_malloc(sizeof(*vs));
1755
    vs->pdu = pdu;
1756
    vs->offset = 7;
1757

    
1758
    v9fs_string_init(&vs->fullname);
1759

    
1760
    pdu_unmarshal(vs->pdu, vs->offset, "dsddd", &dfid, &vs->name, &flags,
1761
            &mode, &gid);
1762

    
1763
    vs->fidp = lookup_fid(s, dfid);
1764
    if (vs->fidp == NULL) {
1765
        err = -ENOENT;
1766
        goto out;
1767
    }
1768

    
1769
    v9fs_string_sprintf(&vs->fullname, "%s/%s", vs->fidp->path.data,
1770
             vs->name.data);
1771

    
1772
    vs->fidp->fd = v9fs_do_open2(s, vs->fullname.data, vs->fidp->uid,
1773
            gid, flags, mode);
1774
    v9fs_lcreate_post_do_open2(s, vs, err);
1775
    return;
1776

    
1777
out:
1778
    complete_pdu(s, vs->pdu, err);
1779
    v9fs_string_free(&vs->name);
1780
    qemu_free(vs);
1781
}
1782

    
1783
static void v9fs_clunk(V9fsState *s, V9fsPDU *pdu)
1784
{
1785
    int32_t fid;
1786
    size_t offset = 7;
1787
    int err;
1788

    
1789
    pdu_unmarshal(pdu, offset, "d", &fid);
1790

    
1791
    err = free_fid(s, fid);
1792
    if (err < 0) {
1793
        goto out;
1794
    }
1795

    
1796
    offset = 7;
1797
    err = offset;
1798
out:
1799
    complete_pdu(s, pdu, err);
1800
}
1801

    
1802
static void v9fs_read_post_readdir(V9fsState *, V9fsReadState *, ssize_t);
1803

    
1804
static void v9fs_read_post_seekdir(V9fsState *s, V9fsReadState *vs, ssize_t err)
1805
{
1806
    if (err) {
1807
        goto out;
1808
    }
1809
    v9fs_stat_free(&vs->v9stat);
1810
    v9fs_string_free(&vs->name);
1811
    vs->offset += pdu_marshal(vs->pdu, vs->offset, "d", vs->count);
1812
    vs->offset += vs->count;
1813
    err = vs->offset;
1814
out:
1815
    complete_pdu(s, vs->pdu, err);
1816
    qemu_free(vs);
1817
    return;
1818
}
1819

    
1820
static void v9fs_read_post_dir_lstat(V9fsState *s, V9fsReadState *vs,
1821
                                    ssize_t err)
1822
{
1823
    if (err) {
1824
        err = -errno;
1825
        goto out;
1826
    }
1827
    err = stat_to_v9stat(s, &vs->name, &vs->stbuf, &vs->v9stat);
1828
    if (err) {
1829
        goto out;
1830
    }
1831

    
1832
    vs->len = pdu_marshal(vs->pdu, vs->offset + 4 + vs->count, "S",
1833
                            &vs->v9stat);
1834
    if ((vs->len != (vs->v9stat.size + 2)) ||
1835
            ((vs->count + vs->len) > vs->max_count)) {
1836
        v9fs_do_seekdir(s, vs->fidp->dir, vs->dir_pos);
1837
        v9fs_read_post_seekdir(s, vs, err);
1838
        return;
1839
    }
1840
    vs->count += vs->len;
1841
    v9fs_stat_free(&vs->v9stat);
1842
    v9fs_string_free(&vs->name);
1843
    vs->dir_pos = vs->dent->d_off;
1844
    vs->dent = v9fs_do_readdir(s, vs->fidp->dir);
1845
    v9fs_read_post_readdir(s, vs, err);
1846
    return;
1847
out:
1848
    v9fs_do_seekdir(s, vs->fidp->dir, vs->dir_pos);
1849
    v9fs_read_post_seekdir(s, vs, err);
1850
    return;
1851

    
1852
}
1853

    
1854
static void v9fs_read_post_readdir(V9fsState *s, V9fsReadState *vs, ssize_t err)
1855
{
1856
    if (vs->dent) {
1857
        memset(&vs->v9stat, 0, sizeof(vs->v9stat));
1858
        v9fs_string_init(&vs->name);
1859
        v9fs_string_sprintf(&vs->name, "%s/%s", vs->fidp->path.data,
1860
                            vs->dent->d_name);
1861
        err = v9fs_do_lstat(s, &vs->name, &vs->stbuf);
1862
        v9fs_read_post_dir_lstat(s, vs, err);
1863
        return;
1864
    }
1865

    
1866
    vs->offset += pdu_marshal(vs->pdu, vs->offset, "d", vs->count);
1867
    vs->offset += vs->count;
1868
    err = vs->offset;
1869
    complete_pdu(s, vs->pdu, err);
1870
    qemu_free(vs);
1871
    return;
1872
}
1873

    
1874
static void v9fs_read_post_telldir(V9fsState *s, V9fsReadState *vs, ssize_t err)
1875
{
1876
    vs->dent = v9fs_do_readdir(s, vs->fidp->dir);
1877
    v9fs_read_post_readdir(s, vs, err);
1878
    return;
1879
}
1880

    
1881
static void v9fs_read_post_rewinddir(V9fsState *s, V9fsReadState *vs,
1882
                                       ssize_t err)
1883
{
1884
    vs->dir_pos = v9fs_do_telldir(s, vs->fidp->dir);
1885
    v9fs_read_post_telldir(s, vs, err);
1886
    return;
1887
}
1888

    
1889
static void v9fs_read_post_readv(V9fsState *s, V9fsReadState *vs, ssize_t err)
1890
{
1891
    if (err  < 0) {
1892
        /* IO error return the error */
1893
        err = -errno;
1894
        goto out;
1895
    }
1896
    vs->total += vs->len;
1897
    vs->sg = adjust_sg(vs->sg, vs->len, &vs->cnt);
1898
    if (vs->total < vs->count && vs->len > 0) {
1899
        do {
1900
            if (0) {
1901
                print_sg(vs->sg, vs->cnt);
1902
            }
1903
            vs->len = v9fs_do_readv(s, vs->fidp->fd, vs->sg, vs->cnt);
1904
        } while (vs->len == -1 && errno == EINTR);
1905
        if (vs->len == -1) {
1906
            err  = -errno;
1907
        }
1908
        v9fs_read_post_readv(s, vs, err);
1909
        return;
1910
    }
1911
    vs->offset += pdu_marshal(vs->pdu, vs->offset, "d", vs->total);
1912
    vs->offset += vs->count;
1913
    err = vs->offset;
1914

    
1915
out:
1916
    complete_pdu(s, vs->pdu, err);
1917
    qemu_free(vs);
1918
}
1919

    
1920
static void v9fs_read_post_lseek(V9fsState *s, V9fsReadState *vs, ssize_t err)
1921
{
1922
    if (err == -1) {
1923
        err = -errno;
1924
        goto out;
1925
    }
1926
    vs->sg = cap_sg(vs->sg, vs->count, &vs->cnt);
1927

    
1928
    if (vs->total < vs->count) {
1929
        do {
1930
            if (0) {
1931
                print_sg(vs->sg, vs->cnt);
1932
            }
1933
            vs->len = v9fs_do_readv(s, vs->fidp->fd, vs->sg, vs->cnt);
1934
        } while (vs->len == -1 && errno == EINTR);
1935
        if (vs->len == -1) {
1936
            err  = -errno;
1937
        }
1938
        v9fs_read_post_readv(s, vs, err);
1939
        return;
1940
    }
1941
out:
1942
    complete_pdu(s, vs->pdu, err);
1943
    qemu_free(vs);
1944
}
1945

    
1946
static void v9fs_read(V9fsState *s, V9fsPDU *pdu)
1947
{
1948
    int32_t fid;
1949
    V9fsReadState *vs;
1950
    ssize_t err = 0;
1951

    
1952
    vs = qemu_malloc(sizeof(*vs));
1953
    vs->pdu = pdu;
1954
    vs->offset = 7;
1955
    vs->total = 0;
1956
    vs->len = 0;
1957
    vs->count = 0;
1958

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

    
1961
    vs->fidp = lookup_fid(s, fid);
1962
    if (vs->fidp == NULL) {
1963
        err = -EINVAL;
1964
        goto out;
1965
    }
1966

    
1967
    if (vs->fidp->dir) {
1968
        vs->max_count = vs->count;
1969
        vs->count = 0;
1970
        if (vs->off == 0) {
1971
            v9fs_do_rewinddir(s, vs->fidp->dir);
1972
        }
1973
        v9fs_read_post_rewinddir(s, vs, err);
1974
        return;
1975
    } else if (vs->fidp->fd != -1) {
1976
        vs->sg = vs->iov;
1977
        pdu_marshal(vs->pdu, vs->offset + 4, "v", vs->sg, &vs->cnt);
1978
        err = v9fs_do_lseek(s, vs->fidp->fd, vs->off, SEEK_SET);
1979
        v9fs_read_post_lseek(s, vs, err);
1980
        return;
1981
    } else {
1982
        err = -EINVAL;
1983
    }
1984
out:
1985
    complete_pdu(s, pdu, err);
1986
    qemu_free(vs);
1987
}
1988

    
1989
typedef struct V9fsReadDirState {
1990
    V9fsPDU *pdu;
1991
    V9fsFidState *fidp;
1992
    V9fsQID qid;
1993
    off_t saved_dir_pos;
1994
    struct dirent *dent;
1995
    int32_t count;
1996
    int32_t max_count;
1997
    size_t offset;
1998
    int64_t initial_offset;
1999
    V9fsString name;
2000
} V9fsReadDirState;
2001

    
2002
static void v9fs_readdir_post_seekdir(V9fsState *s, V9fsReadDirState *vs)
2003
{
2004
    vs->offset += pdu_marshal(vs->pdu, vs->offset, "d", vs->count);
2005
    vs->offset += vs->count;
2006
    complete_pdu(s, vs->pdu, vs->offset);
2007
    qemu_free(vs);
2008
    return;
2009
}
2010

    
2011
/* Size of each dirent on the wire: size of qid (13) + size of offset (8)
2012
 * size of type (1) + size of name.size (2) + strlen(name.data)
2013
 */
2014
#define V9_READDIR_DATA_SZ (24 + strlen(vs->name.data))
2015

    
2016
static void v9fs_readdir_post_readdir(V9fsState *s, V9fsReadDirState *vs)
2017
{
2018
    int len;
2019
    size_t size;
2020

    
2021
    if (vs->dent) {
2022
        v9fs_string_init(&vs->name);
2023
        v9fs_string_sprintf(&vs->name, "%s", vs->dent->d_name);
2024

    
2025
        if ((vs->count + V9_READDIR_DATA_SZ) > vs->max_count) {
2026
            /* Ran out of buffer. Set dir back to old position and return */
2027
            v9fs_do_seekdir(s, vs->fidp->dir, vs->saved_dir_pos);
2028
            v9fs_readdir_post_seekdir(s, vs);
2029
            return;
2030
        }
2031

    
2032
        /* Fill up just the path field of qid because the client uses
2033
         * only that. To fill the entire qid structure we will have
2034
         * to stat each dirent found, which is expensive
2035
         */
2036
        size = MIN(sizeof(vs->dent->d_ino), sizeof(vs->qid.path));
2037
        memcpy(&vs->qid.path, &vs->dent->d_ino, size);
2038
        /* Fill the other fields with dummy values */
2039
        vs->qid.type = 0;
2040
        vs->qid.version = 0;
2041

    
2042
        len = pdu_marshal(vs->pdu, vs->offset+4+vs->count, "Qqbs",
2043
                              &vs->qid, vs->dent->d_off,
2044
                              vs->dent->d_type, &vs->name);
2045
        vs->count += len;
2046
        v9fs_string_free(&vs->name);
2047
        vs->saved_dir_pos = vs->dent->d_off;
2048
        vs->dent = v9fs_do_readdir(s, vs->fidp->dir);
2049
        v9fs_readdir_post_readdir(s, vs);
2050
        return;
2051
    }
2052

    
2053
    vs->offset += pdu_marshal(vs->pdu, vs->offset, "d", vs->count);
2054
    vs->offset += vs->count;
2055
    complete_pdu(s, vs->pdu, vs->offset);
2056
    qemu_free(vs);
2057
    return;
2058
}
2059

    
2060
static void v9fs_readdir_post_telldir(V9fsState *s, V9fsReadDirState *vs)
2061
{
2062
    vs->dent = v9fs_do_readdir(s, vs->fidp->dir);
2063
    v9fs_readdir_post_readdir(s, vs);
2064
    return;
2065
}
2066

    
2067
static void v9fs_readdir_post_setdir(V9fsState *s, V9fsReadDirState *vs)
2068
{
2069
    vs->saved_dir_pos = v9fs_do_telldir(s, vs->fidp->dir);
2070
    v9fs_readdir_post_telldir(s, vs);
2071
    return;
2072
}
2073

    
2074
static void v9fs_readdir(V9fsState *s, V9fsPDU *pdu)
2075
{
2076
    int32_t fid;
2077
    V9fsReadDirState *vs;
2078
    ssize_t err = 0;
2079
    size_t offset = 7;
2080

    
2081
    vs = qemu_malloc(sizeof(*vs));
2082
    vs->pdu = pdu;
2083
    vs->offset = 7;
2084
    vs->count = 0;
2085

    
2086
    pdu_unmarshal(vs->pdu, offset, "dqd", &fid, &vs->initial_offset,
2087
                                                        &vs->max_count);
2088

    
2089
    vs->fidp = lookup_fid(s, fid);
2090
    if (vs->fidp == NULL || !(vs->fidp->dir)) {
2091
        err = -EINVAL;
2092
        goto out;
2093
    }
2094

    
2095
    if (vs->initial_offset == 0) {
2096
        v9fs_do_rewinddir(s, vs->fidp->dir);
2097
    } else {
2098
        v9fs_do_seekdir(s, vs->fidp->dir, vs->initial_offset);
2099
    }
2100

    
2101
    v9fs_readdir_post_setdir(s, vs);
2102
    return;
2103

    
2104
out:
2105
    complete_pdu(s, pdu, err);
2106
    qemu_free(vs);
2107
    return;
2108
}
2109

    
2110
static void v9fs_write_post_writev(V9fsState *s, V9fsWriteState *vs,
2111
                                   ssize_t err)
2112
{
2113
    if (err  < 0) {
2114
        /* IO error return the error */
2115
        err = -errno;
2116
        goto out;
2117
    }
2118
    vs->total += vs->len;
2119
    vs->sg = adjust_sg(vs->sg, vs->len, &vs->cnt);
2120
    if (vs->total < vs->count && vs->len > 0) {
2121
        do {
2122
            if (0) {
2123
                print_sg(vs->sg, vs->cnt);
2124
            }
2125
            vs->len =  v9fs_do_writev(s, vs->fidp->fd, vs->sg, vs->cnt);
2126
        } while (vs->len == -1 && errno == EINTR);
2127
        if (vs->len == -1) {
2128
            err  = -errno;
2129
        }
2130
        v9fs_write_post_writev(s, vs, err);
2131
        return;
2132
    }
2133
    vs->offset += pdu_marshal(vs->pdu, vs->offset, "d", vs->total);
2134

    
2135
    err = vs->offset;
2136
out:
2137
    complete_pdu(s, vs->pdu, err);
2138
    qemu_free(vs);
2139
}
2140

    
2141
static void v9fs_write_post_lseek(V9fsState *s, V9fsWriteState *vs, ssize_t err)
2142
{
2143
    if (err == -1) {
2144
        err = -errno;
2145
        goto out;
2146
    }
2147
    vs->sg = cap_sg(vs->sg, vs->count, &vs->cnt);
2148

    
2149
    if (vs->total < vs->count) {
2150
        do {
2151
            if (0) {
2152
                print_sg(vs->sg, vs->cnt);
2153
            }
2154
            vs->len = v9fs_do_writev(s, vs->fidp->fd, vs->sg, vs->cnt);
2155
        } while (vs->len == -1 && errno == EINTR);
2156
        if (vs->len == -1) {
2157
            err  = -errno;
2158
        }
2159
        v9fs_write_post_writev(s, vs, err);
2160
        return;
2161
    }
2162

    
2163
out:
2164
    complete_pdu(s, vs->pdu, err);
2165
    qemu_free(vs);
2166
}
2167

    
2168
static void v9fs_write(V9fsState *s, V9fsPDU *pdu)
2169
{
2170
    int32_t fid;
2171
    V9fsWriteState *vs;
2172
    ssize_t err;
2173

    
2174
    vs = qemu_malloc(sizeof(*vs));
2175

    
2176
    vs->pdu = pdu;
2177
    vs->offset = 7;
2178
    vs->sg = vs->iov;
2179
    vs->total = 0;
2180
    vs->len = 0;
2181

    
2182
    pdu_unmarshal(vs->pdu, vs->offset, "dqdv", &fid, &vs->off, &vs->count,
2183
                    vs->sg, &vs->cnt);
2184

    
2185
    vs->fidp = lookup_fid(s, fid);
2186
    if (vs->fidp == NULL) {
2187
        err = -EINVAL;
2188
        goto out;
2189
    }
2190

    
2191
    if (vs->fidp->fd == -1) {
2192
        err = -EINVAL;
2193
        goto out;
2194
    }
2195

    
2196
    err = v9fs_do_lseek(s, vs->fidp->fd, vs->off, SEEK_SET);
2197

    
2198
    v9fs_write_post_lseek(s, vs, err);
2199
    return;
2200

    
2201
out:
2202
    complete_pdu(s, vs->pdu, err);
2203
    qemu_free(vs);
2204
}
2205

    
2206
static void v9fs_create_post_getiounit(V9fsState *s, V9fsCreateState *vs)
2207
{
2208
    int err;
2209
    v9fs_string_copy(&vs->fidp->path, &vs->fullname);
2210
    stat_to_qid(&vs->stbuf, &vs->qid);
2211

    
2212
    vs->offset += pdu_marshal(vs->pdu, vs->offset, "Qd", &vs->qid, vs->iounit);
2213
    err = vs->offset;
2214

    
2215
    complete_pdu(s, vs->pdu, err);
2216
    v9fs_string_free(&vs->name);
2217
    v9fs_string_free(&vs->extension);
2218
    v9fs_string_free(&vs->fullname);
2219
    qemu_free(vs);
2220
}
2221

    
2222
static void v9fs_post_create(V9fsState *s, V9fsCreateState *vs, int err)
2223
{
2224
    if (err == 0) {
2225
        vs->iounit = get_iounit(s, &vs->fidp->path);
2226
        v9fs_create_post_getiounit(s, vs);
2227
        return;
2228
    }
2229

    
2230
    complete_pdu(s, vs->pdu, err);
2231
    v9fs_string_free(&vs->name);
2232
    v9fs_string_free(&vs->extension);
2233
    v9fs_string_free(&vs->fullname);
2234
    qemu_free(vs);
2235
}
2236

    
2237
static void v9fs_create_post_perms(V9fsState *s, V9fsCreateState *vs, int err)
2238
{
2239
    if (err) {
2240
        err = -errno;
2241
    }
2242
    v9fs_post_create(s, vs, err);
2243
}
2244

    
2245
static void v9fs_create_post_opendir(V9fsState *s, V9fsCreateState *vs,
2246
                                                                    int err)
2247
{
2248
    if (!vs->fidp->dir) {
2249
        err = -errno;
2250
    }
2251
    v9fs_post_create(s, vs, err);
2252
}
2253

    
2254
static void v9fs_create_post_dir_lstat(V9fsState *s, V9fsCreateState *vs,
2255
                                                                    int err)
2256
{
2257
    if (err) {
2258
        err = -errno;
2259
        goto out;
2260
    }
2261

    
2262
    vs->fidp->dir = v9fs_do_opendir(s, &vs->fullname);
2263
    v9fs_create_post_opendir(s, vs, err);
2264
    return;
2265

    
2266
out:
2267
    v9fs_post_create(s, vs, err);
2268
}
2269

    
2270
static void v9fs_create_post_mkdir(V9fsState *s, V9fsCreateState *vs, int err)
2271
{
2272
    if (err) {
2273
        err = -errno;
2274
        goto out;
2275
    }
2276

    
2277
    err = v9fs_do_lstat(s, &vs->fullname, &vs->stbuf);
2278
    v9fs_create_post_dir_lstat(s, vs, err);
2279
    return;
2280

    
2281
out:
2282
    v9fs_post_create(s, vs, err);
2283
}
2284

    
2285
static void v9fs_create_post_fstat(V9fsState *s, V9fsCreateState *vs, int err)
2286
{
2287
    if (err) {
2288
        vs->fidp->fd = -1;
2289
        err = -errno;
2290
    }
2291

    
2292
    v9fs_post_create(s, vs, err);
2293
    return;
2294
}
2295

    
2296
static void v9fs_create_post_open2(V9fsState *s, V9fsCreateState *vs, int err)
2297
{
2298
    if (vs->fidp->fd == -1) {
2299
        err = -errno;
2300
        goto out;
2301
    }
2302

    
2303
    err = v9fs_do_fstat(s, vs->fidp->fd, &vs->stbuf);
2304
    v9fs_create_post_fstat(s, vs, err);
2305

    
2306
    return;
2307

    
2308
out:
2309
    v9fs_post_create(s, vs, err);
2310

    
2311
}
2312

    
2313
static void v9fs_create_post_lstat(V9fsState *s, V9fsCreateState *vs, int err)
2314
{
2315

    
2316
    if (err == 0 || errno != ENOENT) {
2317
        err = -errno;
2318
        goto out;
2319
    }
2320

    
2321
    if (vs->perm & P9_STAT_MODE_DIR) {
2322
        err = v9fs_do_mkdir(s, vs->fullname.data, vs->perm & 0777,
2323
                vs->fidp->uid, -1);
2324
        v9fs_create_post_mkdir(s, vs, err);
2325
    } else if (vs->perm & P9_STAT_MODE_SYMLINK) {
2326
        err = v9fs_do_symlink(s, vs->fidp, vs->extension.data,
2327
                vs->fullname.data, -1);
2328
        v9fs_create_post_perms(s, vs, err);
2329
    } else if (vs->perm & P9_STAT_MODE_LINK) {
2330
        int32_t nfid = atoi(vs->extension.data);
2331
        V9fsFidState *nfidp = lookup_fid(s, nfid);
2332
        if (nfidp == NULL) {
2333
            err = -errno;
2334
            v9fs_post_create(s, vs, err);
2335
        }
2336
        err = v9fs_do_link(s, &nfidp->path, &vs->fullname);
2337
        v9fs_create_post_perms(s, vs, err);
2338
    } else if (vs->perm & P9_STAT_MODE_DEVICE) {
2339
        char ctype;
2340
        uint32_t major, minor;
2341
        mode_t nmode = 0;
2342

    
2343
        if (sscanf(vs->extension.data, "%c %u %u", &ctype, &major,
2344
                                        &minor) != 3) {
2345
            err = -errno;
2346
            v9fs_post_create(s, vs, err);
2347
        }
2348

    
2349
        switch (ctype) {
2350
        case 'c':
2351
            nmode = S_IFCHR;
2352
            break;
2353
        case 'b':
2354
            nmode = S_IFBLK;
2355
            break;
2356
        default:
2357
            err = -EIO;
2358
            v9fs_post_create(s, vs, err);
2359
        }
2360

    
2361
        nmode |= vs->perm & 0777;
2362
        err = v9fs_do_mknod(s, vs->fullname.data, nmode,
2363
                makedev(major, minor), vs->fidp->uid, -1);
2364
        v9fs_create_post_perms(s, vs, err);
2365
    } else if (vs->perm & P9_STAT_MODE_NAMED_PIPE) {
2366
        err = v9fs_do_mknod(s, vs->fullname.data, S_IFIFO | (vs->perm & 0777),
2367
                0, vs->fidp->uid, -1);
2368
        v9fs_post_create(s, vs, err);
2369
    } else if (vs->perm & P9_STAT_MODE_SOCKET) {
2370
        err = v9fs_do_mknod(s, vs->fullname.data, S_IFSOCK | (vs->perm & 0777),
2371
                0, vs->fidp->uid, -1);
2372
        v9fs_post_create(s, vs, err);
2373
    } else {
2374
        vs->fidp->fd = v9fs_do_open2(s, vs->fullname.data, vs->fidp->uid,
2375
                -1, omode_to_uflags(vs->mode)|O_CREAT, vs->perm);
2376

    
2377
        v9fs_create_post_open2(s, vs, err);
2378
    }
2379

    
2380
    return;
2381

    
2382
out:
2383
    v9fs_post_create(s, vs, err);
2384
}
2385

    
2386
static void v9fs_create(V9fsState *s, V9fsPDU *pdu)
2387
{
2388
    int32_t fid;
2389
    V9fsCreateState *vs;
2390
    int err = 0;
2391

    
2392
    vs = qemu_malloc(sizeof(*vs));
2393
    vs->pdu = pdu;
2394
    vs->offset = 7;
2395

    
2396
    v9fs_string_init(&vs->fullname);
2397

    
2398
    pdu_unmarshal(vs->pdu, vs->offset, "dsdbs", &fid, &vs->name,
2399
                                &vs->perm, &vs->mode, &vs->extension);
2400

    
2401
    vs->fidp = lookup_fid(s, fid);
2402
    if (vs->fidp == NULL) {
2403
        err = -EINVAL;
2404
        goto out;
2405
    }
2406

    
2407
    v9fs_string_sprintf(&vs->fullname, "%s/%s", vs->fidp->path.data,
2408
                                                        vs->name.data);
2409

    
2410
    err = v9fs_do_lstat(s, &vs->fullname, &vs->stbuf);
2411
    v9fs_create_post_lstat(s, vs, err);
2412
    return;
2413

    
2414
out:
2415
    complete_pdu(s, vs->pdu, err);
2416
    v9fs_string_free(&vs->name);
2417
    v9fs_string_free(&vs->extension);
2418
    qemu_free(vs);
2419
}
2420

    
2421
static void v9fs_post_symlink(V9fsState *s, V9fsSymlinkState *vs, int err)
2422
{
2423
    if (err == 0) {
2424
        stat_to_qid(&vs->stbuf, &vs->qid);
2425
        vs->offset += pdu_marshal(vs->pdu, vs->offset, "Q", &vs->qid);
2426
        err = vs->offset;
2427
    } else {
2428
        err = -errno;
2429
    }
2430
    complete_pdu(s, vs->pdu, err);
2431
    v9fs_string_free(&vs->name);
2432
    v9fs_string_free(&vs->symname);
2433
    v9fs_string_free(&vs->fullname);
2434
    qemu_free(vs);
2435
}
2436

    
2437
static void v9fs_symlink_post_do_symlink(V9fsState *s, V9fsSymlinkState *vs,
2438
        int err)
2439
{
2440
    if (err) {
2441
        goto out;
2442
    }
2443
    err = v9fs_do_lstat(s, &vs->fullname, &vs->stbuf);
2444
out:
2445
    v9fs_post_symlink(s, vs, err);
2446
}
2447

    
2448
static void v9fs_symlink(V9fsState *s, V9fsPDU *pdu)
2449
{
2450
    int32_t dfid;
2451
    V9fsSymlinkState *vs;
2452
    int err = 0;
2453
    gid_t gid;
2454

    
2455
    vs = qemu_malloc(sizeof(*vs));
2456
    vs->pdu = pdu;
2457
    vs->offset = 7;
2458

    
2459
    v9fs_string_init(&vs->fullname);
2460

    
2461
    pdu_unmarshal(vs->pdu, vs->offset, "dssd", &dfid, &vs->name,
2462
            &vs->symname, &gid);
2463

    
2464
    vs->dfidp = lookup_fid(s, dfid);
2465
    if (vs->dfidp == NULL) {
2466
        err = -EINVAL;
2467
        goto out;
2468
    }
2469

    
2470
    v9fs_string_sprintf(&vs->fullname, "%s/%s", vs->dfidp->path.data,
2471
            vs->name.data);
2472
    err = v9fs_do_symlink(s, vs->dfidp, vs->symname.data,
2473
            vs->fullname.data, gid);
2474
    v9fs_symlink_post_do_symlink(s, vs, err);
2475
    return;
2476

    
2477
out:
2478
    complete_pdu(s, vs->pdu, err);
2479
    v9fs_string_free(&vs->name);
2480
    v9fs_string_free(&vs->symname);
2481
    qemu_free(vs);
2482
}
2483

    
2484
static void v9fs_flush(V9fsState *s, V9fsPDU *pdu)
2485
{
2486
    /* A nop call with no return */
2487
    complete_pdu(s, pdu, 7);
2488
}
2489

    
2490
static void v9fs_link(V9fsState *s, V9fsPDU *pdu)
2491
{
2492
    int32_t dfid, oldfid;
2493
    V9fsFidState *dfidp, *oldfidp;
2494
    V9fsString name, fullname;
2495
    size_t offset = 7;
2496
    int err = 0;
2497

    
2498
    v9fs_string_init(&fullname);
2499

    
2500
    pdu_unmarshal(pdu, offset, "dds", &dfid, &oldfid, &name);
2501

    
2502
    dfidp = lookup_fid(s, dfid);
2503
    if (dfidp == NULL) {
2504
        err = -errno;
2505
        goto out;
2506
    }
2507

    
2508
    oldfidp = lookup_fid(s, oldfid);
2509
    if (oldfidp == NULL) {
2510
        err = -errno;
2511
        goto out;
2512
    }
2513

    
2514
    v9fs_string_sprintf(&fullname, "%s/%s", dfidp->path.data, name.data);
2515
    err = offset;
2516
    err = v9fs_do_link(s, &oldfidp->path, &fullname);
2517
    if (err) {
2518
        err = -errno;
2519
    }
2520
    v9fs_string_free(&fullname);
2521

    
2522
out:
2523
    v9fs_string_free(&name);
2524
    complete_pdu(s, pdu, err);
2525
}
2526

    
2527
static void v9fs_remove_post_remove(V9fsState *s, V9fsRemoveState *vs,
2528
                                                                int err)
2529
{
2530
    if (err < 0) {
2531
        err = -errno;
2532
    } else {
2533
        err = vs->offset;
2534
    }
2535

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

    
2539
    complete_pdu(s, vs->pdu, err);
2540
    qemu_free(vs);
2541
}
2542

    
2543
static void v9fs_remove(V9fsState *s, V9fsPDU *pdu)
2544
{
2545
    int32_t fid;
2546
    V9fsRemoveState *vs;
2547
    int err = 0;
2548

    
2549
    vs = qemu_malloc(sizeof(*vs));
2550
    vs->pdu = pdu;
2551
    vs->offset = 7;
2552

    
2553
    pdu_unmarshal(vs->pdu, vs->offset, "d", &fid);
2554

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

    
2561
    err = v9fs_do_remove(s, &vs->fidp->path);
2562
    v9fs_remove_post_remove(s, vs, err);
2563
    return;
2564

    
2565
out:
2566
    complete_pdu(s, pdu, err);
2567
    qemu_free(vs);
2568
}
2569

    
2570
static void v9fs_wstat_post_truncate(V9fsState *s, V9fsWstatState *vs, int err)
2571
{
2572
    if (err < 0) {
2573
        goto out;
2574
    }
2575

    
2576
    err = vs->offset;
2577

    
2578
out:
2579
    v9fs_stat_free(&vs->v9stat);
2580
    complete_pdu(s, vs->pdu, err);
2581
    qemu_free(vs);
2582
}
2583

    
2584
static void v9fs_wstat_post_rename(V9fsState *s, V9fsWstatState *vs, int err)
2585
{
2586
    if (err < 0) {
2587
        goto out;
2588
    }
2589
    if (vs->v9stat.length != -1) {
2590
        if (v9fs_do_truncate(s, &vs->fidp->path, vs->v9stat.length) < 0) {
2591
            err = -errno;
2592
        }
2593
    }
2594
    v9fs_wstat_post_truncate(s, vs, err);
2595
    return;
2596

    
2597
out:
2598
    v9fs_stat_free(&vs->v9stat);
2599
    complete_pdu(s, vs->pdu, err);
2600
    qemu_free(vs);
2601
}
2602

    
2603
static int v9fs_complete_rename(V9fsState *s, V9fsRenameState *vs)
2604
{
2605
    int err = 0;
2606
    char *old_name, *new_name;
2607
    char *end;
2608

    
2609
    if (vs->newdirfid != -1) {
2610
        V9fsFidState *dirfidp;
2611
        dirfidp = lookup_fid(s, vs->newdirfid);
2612

    
2613
        if (dirfidp == NULL) {
2614
            err = -ENOENT;
2615
            goto out;
2616
        }
2617

    
2618
        BUG_ON(dirfidp->fd != -1);
2619
        BUG_ON(dirfidp->dir);
2620

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

    
2623
        strcpy(new_name, dirfidp->path.data);
2624
        strcat(new_name, "/");
2625
        strcat(new_name + dirfidp->path.size, vs->name.data);
2626
    } else {
2627
        old_name = vs->fidp->path.data;
2628
        end = strrchr(old_name, '/');
2629
        if (end) {
2630
            end++;
2631
        } else {
2632
            end = old_name;
2633
        }
2634
        new_name = qemu_mallocz(end - old_name + vs->name.size + 1);
2635

    
2636
        strncat(new_name, old_name, end - old_name);
2637
        strncat(new_name + (end - old_name), vs->name.data, vs->name.size);
2638
    }
2639

    
2640
    v9fs_string_free(&vs->name);
2641
    vs->name.data = qemu_strdup(new_name);
2642
    vs->name.size = strlen(new_name);
2643

    
2644
    if (strcmp(new_name, vs->fidp->path.data) != 0) {
2645
        if (v9fs_do_rename(s, &vs->fidp->path, &vs->name)) {
2646
            err = -errno;
2647
        } else {
2648
            V9fsFidState *fidp;
2649
            /*
2650
            * Fixup fid's pointing to the old name to
2651
            * start pointing to the new name
2652
            */
2653
            for (fidp = s->fid_list; fidp; fidp = fidp->next) {
2654
                if (vs->fidp == fidp) {
2655
                    /*
2656
                    * we replace name of this fid towards the end
2657
                    * so that our below strcmp will work
2658
                    */
2659
                    continue;
2660
                }
2661
                if (!strncmp(vs->fidp->path.data, fidp->path.data,
2662
                    strlen(vs->fidp->path.data))) {
2663
                    /* replace the name */
2664
                    v9fs_fix_path(&fidp->path, &vs->name,
2665
                                  strlen(vs->fidp->path.data));
2666
                }
2667
            }
2668
            v9fs_string_copy(&vs->fidp->path, &vs->name);
2669
        }
2670
    }
2671
out:
2672
    v9fs_string_free(&vs->name);
2673
    return err;
2674
}
2675

    
2676
static void v9fs_rename_post_rename(V9fsState *s, V9fsRenameState *vs, int err)
2677
{
2678
    complete_pdu(s, vs->pdu, err);
2679
    qemu_free(vs);
2680
}
2681

    
2682
static void v9fs_wstat_post_chown(V9fsState *s, V9fsWstatState *vs, int err)
2683
{
2684
    if (err < 0) {
2685
        goto out;
2686
    }
2687

    
2688
    if (vs->v9stat.name.size != 0) {
2689
        V9fsRenameState *vr;
2690

    
2691
        vr = qemu_malloc(sizeof(V9fsRenameState));
2692
        memset(vr, sizeof(*vr), 0);
2693
        vr->newdirfid = -1;
2694
        vr->pdu = vs->pdu;
2695
        vr->fidp = vs->fidp;
2696
        vr->offset = vs->offset;
2697
        vr->name.size = vs->v9stat.name.size;
2698
        vr->name.data = qemu_strdup(vs->v9stat.name.data);
2699

    
2700
        err = v9fs_complete_rename(s, vr);
2701
        qemu_free(vr);
2702
    }
2703
    v9fs_wstat_post_rename(s, vs, err);
2704
    return;
2705

    
2706
out:
2707
    v9fs_stat_free(&vs->v9stat);
2708
    complete_pdu(s, vs->pdu, err);
2709
    qemu_free(vs);
2710
}
2711

    
2712
static void v9fs_rename(V9fsState *s, V9fsPDU *pdu)
2713
{
2714
    int32_t fid;
2715
    V9fsRenameState *vs;
2716
    ssize_t err = 0;
2717

    
2718
    vs = qemu_malloc(sizeof(*vs));
2719
    vs->pdu = pdu;
2720
    vs->offset = 7;
2721

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

    
2724
    vs->fidp = lookup_fid(s, fid);
2725
    if (vs->fidp == NULL) {
2726
        err = -ENOENT;
2727
        goto out;
2728
    }
2729

    
2730
    BUG_ON(vs->fidp->fd != -1);
2731
    BUG_ON(vs->fidp->dir);
2732

    
2733
    err = v9fs_complete_rename(s, vs);
2734
    v9fs_rename_post_rename(s, vs, err);
2735
    return;
2736
out:
2737
    complete_pdu(s, vs->pdu, err);
2738
    qemu_free(vs);
2739
}
2740

    
2741
static void v9fs_wstat_post_utime(V9fsState *s, V9fsWstatState *vs, int err)
2742
{
2743
    if (err < 0) {
2744
        goto out;
2745
    }
2746

    
2747
    if (vs->v9stat.n_gid != -1 || vs->v9stat.n_uid != -1) {
2748
        if (v9fs_do_chown(s, &vs->fidp->path, vs->v9stat.n_uid,
2749
                    vs->v9stat.n_gid)) {
2750
            err = -errno;
2751
        }
2752
    }
2753
    v9fs_wstat_post_chown(s, vs, err);
2754
    return;
2755

    
2756
out:
2757
    v9fs_stat_free(&vs->v9stat);
2758
    complete_pdu(s, vs->pdu, err);
2759
    qemu_free(vs);
2760
}
2761

    
2762
static void v9fs_wstat_post_chmod(V9fsState *s, V9fsWstatState *vs, int err)
2763
{
2764
    if (err < 0) {
2765
        goto out;
2766
    }
2767

    
2768
    if (vs->v9stat.mtime != -1 || vs->v9stat.atime != -1) {
2769
        struct timespec times[2];
2770
        if (vs->v9stat.atime != -1) {
2771
            times[0].tv_sec = vs->v9stat.atime;
2772
            times[0].tv_nsec = 0;
2773
        } else {
2774
            times[0].tv_nsec = UTIME_OMIT;
2775
        }
2776
        if (vs->v9stat.mtime != -1) {
2777
            times[1].tv_sec = vs->v9stat.mtime;
2778
            times[1].tv_nsec = 0;
2779
        } else {
2780
            times[1].tv_nsec = UTIME_OMIT;
2781
        }
2782

    
2783
        if (v9fs_do_utimensat(s, &vs->fidp->path, times)) {
2784
            err = -errno;
2785
        }
2786
    }
2787

    
2788
    v9fs_wstat_post_utime(s, vs, err);
2789
    return;
2790

    
2791
out:
2792
    v9fs_stat_free(&vs->v9stat);
2793
    complete_pdu(s, vs->pdu, err);
2794
    qemu_free(vs);
2795
}
2796

    
2797
static void v9fs_wstat_post_fsync(V9fsState *s, V9fsWstatState *vs, int err)
2798
{
2799
    if (err == -1) {
2800
        err = -errno;
2801
    }
2802
    v9fs_stat_free(&vs->v9stat);
2803
    complete_pdu(s, vs->pdu, err);
2804
    qemu_free(vs);
2805
}
2806

    
2807
static void v9fs_wstat_post_lstat(V9fsState *s, V9fsWstatState *vs, int err)
2808
{
2809
    uint32_t v9_mode;
2810

    
2811
    if (err == -1) {
2812
        err = -errno;
2813
        goto out;
2814
    }
2815

    
2816
    v9_mode = stat_to_v9mode(&vs->stbuf);
2817

    
2818
    if ((vs->v9stat.mode & P9_STAT_MODE_TYPE_BITS) !=
2819
        (v9_mode & P9_STAT_MODE_TYPE_BITS)) {
2820
            /* Attempting to change the type */
2821
            err = -EIO;
2822
            goto out;
2823
    }
2824

    
2825
    if (v9fs_do_chmod(s, &vs->fidp->path, v9mode_to_mode(vs->v9stat.mode,
2826
                    &vs->v9stat.extension))) {
2827
            err = -errno;
2828
     }
2829
    v9fs_wstat_post_chmod(s, vs, err);
2830
    return;
2831

    
2832
out:
2833
    v9fs_stat_free(&vs->v9stat);
2834
    complete_pdu(s, vs->pdu, err);
2835
    qemu_free(vs);
2836
}
2837

    
2838
static void v9fs_wstat(V9fsState *s, V9fsPDU *pdu)
2839
{
2840
    int32_t fid;
2841
    V9fsWstatState *vs;
2842
    int err = 0;
2843

    
2844
    vs = qemu_malloc(sizeof(*vs));
2845
    vs->pdu = pdu;
2846
    vs->offset = 7;
2847

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

    
2850
    vs->fidp = lookup_fid(s, fid);
2851
    if (vs->fidp == NULL) {
2852
        err = -EINVAL;
2853
        goto out;
2854
    }
2855

    
2856
    /* do we need to sync the file? */
2857
    if (donttouch_stat(&vs->v9stat)) {
2858
        err = v9fs_do_fsync(s, vs->fidp->fd);
2859
        v9fs_wstat_post_fsync(s, vs, err);
2860
        return;
2861
    }
2862

    
2863
    if (vs->v9stat.mode != -1) {
2864
        err = v9fs_do_lstat(s, &vs->fidp->path, &vs->stbuf);
2865
        v9fs_wstat_post_lstat(s, vs, err);
2866
        return;
2867
    }
2868

    
2869
    v9fs_wstat_post_chmod(s, vs, err);
2870
    return;
2871

    
2872
out:
2873
    v9fs_stat_free(&vs->v9stat);
2874
    complete_pdu(s, vs->pdu, err);
2875
    qemu_free(vs);
2876
}
2877

    
2878
static void v9fs_statfs_post_statfs(V9fsState *s, V9fsStatfsState *vs, int err)
2879
{
2880
    int32_t bsize_factor;
2881

    
2882
    if (err) {
2883
        err = -errno;
2884
        goto out;
2885
    }
2886

    
2887
    /*
2888
     * compute bsize factor based on host file system block size
2889
     * and client msize
2890
     */
2891
    bsize_factor = (s->msize - P9_IOHDRSZ)/vs->stbuf.f_bsize;
2892
    if (!bsize_factor) {
2893
        bsize_factor = 1;
2894
    }
2895
    vs->v9statfs.f_type = vs->stbuf.f_type;
2896
    vs->v9statfs.f_bsize = vs->stbuf.f_bsize;
2897
    vs->v9statfs.f_bsize *= bsize_factor;
2898
    /*
2899
     * f_bsize is adjusted(multiplied) by bsize factor, so we need to
2900
     * adjust(divide) the number of blocks, free blocks and available
2901
     * blocks by bsize factor
2902
     */
2903
    vs->v9statfs.f_blocks = vs->stbuf.f_blocks/bsize_factor;
2904
    vs->v9statfs.f_bfree = vs->stbuf.f_bfree/bsize_factor;
2905
    vs->v9statfs.f_bavail = vs->stbuf.f_bavail/bsize_factor;
2906
    vs->v9statfs.f_files = vs->stbuf.f_files;
2907
    vs->v9statfs.f_ffree = vs->stbuf.f_ffree;
2908
    vs->v9statfs.fsid_val = (unsigned int) vs->stbuf.f_fsid.__val[0] |
2909
                        (unsigned long long)vs->stbuf.f_fsid.__val[1] << 32;
2910
    vs->v9statfs.f_namelen = vs->stbuf.f_namelen;
2911

    
2912
    vs->offset += pdu_marshal(vs->pdu, vs->offset, "ddqqqqqqd",
2913
         vs->v9statfs.f_type, vs->v9statfs.f_bsize, vs->v9statfs.f_blocks,
2914
         vs->v9statfs.f_bfree, vs->v9statfs.f_bavail, vs->v9statfs.f_files,
2915
         vs->v9statfs.f_ffree, vs->v9statfs.fsid_val,
2916
         vs->v9statfs.f_namelen);
2917

    
2918
out:
2919
    complete_pdu(s, vs->pdu, vs->offset);
2920
    qemu_free(vs);
2921
}
2922

    
2923
static void v9fs_statfs(V9fsState *s, V9fsPDU *pdu)
2924
{
2925
    V9fsStatfsState *vs;
2926
    ssize_t err = 0;
2927

    
2928
    vs = qemu_malloc(sizeof(*vs));
2929
    vs->pdu = pdu;
2930
    vs->offset = 7;
2931

    
2932
    memset(&vs->v9statfs, 0, sizeof(vs->v9statfs));
2933

    
2934
    pdu_unmarshal(vs->pdu, vs->offset, "d", &vs->fid);
2935

    
2936
    vs->fidp = lookup_fid(s, vs->fid);
2937
    if (vs->fidp == NULL) {
2938
        err = -ENOENT;
2939
        goto out;
2940
    }
2941

    
2942
    err = v9fs_do_statfs(s, &vs->fidp->path, &vs->stbuf);
2943
    v9fs_statfs_post_statfs(s, vs, err);
2944
    return;
2945

    
2946
out:
2947
    complete_pdu(s, vs->pdu, err);
2948
    qemu_free(vs);
2949
}
2950

    
2951
static void v9fs_mknod_post_lstat(V9fsState *s, V9fsMkState *vs, int err)
2952
{
2953
    if (err == -1) {
2954
        err = -errno;
2955
        goto out;
2956
    }
2957

    
2958
    stat_to_qid(&vs->stbuf, &vs->qid);
2959
    vs->offset += pdu_marshal(vs->pdu, vs->offset, "Q", &vs->qid);
2960
    err = vs->offset;
2961
out:
2962
    complete_pdu(s, vs->pdu, err);
2963
    v9fs_string_free(&vs->fullname);
2964
    v9fs_string_free(&vs->name);
2965
    qemu_free(vs);
2966
}
2967

    
2968
static void v9fs_mknod_post_mknod(V9fsState *s, V9fsMkState *vs, int err)
2969
{
2970
    if (err == -1) {
2971
        err = -errno;
2972
        goto out;
2973
    }
2974

    
2975
    err = v9fs_do_lstat(s, &vs->fullname, &vs->stbuf);
2976
    v9fs_mknod_post_lstat(s, vs, err);
2977
    return;
2978
out:
2979
    complete_pdu(s, vs->pdu, err);
2980
    v9fs_string_free(&vs->fullname);
2981
    v9fs_string_free(&vs->name);
2982
    qemu_free(vs);
2983
}
2984

    
2985
static void v9fs_mknod(V9fsState *s, V9fsPDU *pdu)
2986
{
2987
    int32_t fid;
2988
    V9fsMkState *vs;
2989
    int err = 0;
2990
    V9fsFidState *fidp;
2991
    gid_t gid;
2992
    int mode;
2993
    int major, minor;
2994

    
2995
    vs = qemu_malloc(sizeof(*vs));
2996
    vs->pdu = pdu;
2997
    vs->offset = 7;
2998

    
2999
    v9fs_string_init(&vs->fullname);
3000
    pdu_unmarshal(vs->pdu, vs->offset, "dsdddd", &fid, &vs->name, &mode,
3001
        &major, &minor, &gid);
3002

    
3003
    fidp = lookup_fid(s, fid);
3004
    if (fidp == NULL) {
3005
        err = -ENOENT;
3006
        goto out;
3007
    }
3008

    
3009
    v9fs_string_sprintf(&vs->fullname, "%s/%s", fidp->path.data, vs->name.data);
3010
    err = v9fs_do_mknod(s, vs->fullname.data, mode, makedev(major, minor),
3011
        fidp->uid, gid);
3012
    v9fs_mknod_post_mknod(s, vs, err);
3013
    return;
3014

    
3015
out:
3016
    complete_pdu(s, vs->pdu, err);
3017
    v9fs_string_free(&vs->fullname);
3018
    v9fs_string_free(&vs->name);
3019
    qemu_free(vs);
3020
}
3021

    
3022
static void v9fs_mkdir_post_lstat(V9fsState *s, V9fsMkState *vs, int err)
3023
{
3024
    if (err == -1) {
3025
        err = -errno;
3026
        goto out;
3027
    }
3028

    
3029
    stat_to_qid(&vs->stbuf, &vs->qid);
3030
    vs->offset += pdu_marshal(vs->pdu, vs->offset, "Q", &vs->qid);
3031
    err = vs->offset;
3032
out:
3033
    complete_pdu(s, vs->pdu, err);
3034
    v9fs_string_free(&vs->fullname);
3035
    v9fs_string_free(&vs->name);
3036
    qemu_free(vs);
3037
}
3038

    
3039
static void v9fs_mkdir_post_mkdir(V9fsState *s, V9fsMkState *vs, int err)
3040
{
3041
    if (err == -1) {
3042
        err = -errno;
3043
        goto out;
3044
    }
3045

    
3046
    err = v9fs_do_lstat(s, &vs->fullname, &vs->stbuf);
3047
    v9fs_mkdir_post_lstat(s, vs, err);
3048
    return;
3049
out:
3050
    complete_pdu(s, vs->pdu, err);
3051
    v9fs_string_free(&vs->fullname);
3052
    v9fs_string_free(&vs->name);
3053
    qemu_free(vs);
3054
}
3055

    
3056
static void v9fs_mkdir(V9fsState *s, V9fsPDU *pdu)
3057
{
3058
    int32_t fid;
3059
    V9fsMkState *vs;
3060
    int err = 0;
3061
    V9fsFidState *fidp;
3062
    gid_t gid;
3063
    int mode;
3064

    
3065
    vs = qemu_malloc(sizeof(*vs));
3066
    vs->pdu = pdu;
3067
    vs->offset = 7;
3068

    
3069
    v9fs_string_init(&vs->fullname);
3070
    pdu_unmarshal(vs->pdu, vs->offset, "dsdd", &fid, &vs->name, &mode,
3071
        &gid);
3072

    
3073
    fidp = lookup_fid(s, fid);
3074
    if (fidp == NULL) {
3075
        err = -ENOENT;
3076
        goto out;
3077
    }
3078

    
3079
    v9fs_string_sprintf(&vs->fullname, "%s/%s", fidp->path.data, vs->name.data);
3080
    err = v9fs_do_mkdir(s, vs->fullname.data, mode, fidp->uid, gid);
3081
    v9fs_mkdir_post_mkdir(s, vs, err);
3082
    return;
3083

    
3084
out:
3085
    complete_pdu(s, vs->pdu, err);
3086
    v9fs_string_free(&vs->fullname);
3087
    v9fs_string_free(&vs->name);
3088
    qemu_free(vs);
3089
}
3090

    
3091
typedef void (pdu_handler_t)(V9fsState *s, V9fsPDU *pdu);
3092

    
3093
static pdu_handler_t *pdu_handlers[] = {
3094
    [P9_TREADDIR] = v9fs_readdir,
3095
    [P9_TSTATFS] = v9fs_statfs,
3096
    [P9_TGETATTR] = v9fs_getattr,
3097
    [P9_TSETATTR] = v9fs_setattr,
3098
    [P9_TMKNOD] = v9fs_mknod,
3099
    [P9_TRENAME] = v9fs_rename,
3100
    [P9_TMKDIR] = v9fs_mkdir,
3101
    [P9_TVERSION] = v9fs_version,
3102
    [P9_TLOPEN] = v9fs_open,
3103
    [P9_TATTACH] = v9fs_attach,
3104
    [P9_TSTAT] = v9fs_stat,
3105
    [P9_TWALK] = v9fs_walk,
3106
    [P9_TCLUNK] = v9fs_clunk,
3107
    [P9_TOPEN] = v9fs_open,
3108
    [P9_TREAD] = v9fs_read,
3109
#if 0
3110
    [P9_TAUTH] = v9fs_auth,
3111
#endif
3112
    [P9_TFLUSH] = v9fs_flush,
3113
    [P9_TLINK] = v9fs_link,
3114
    [P9_TSYMLINK] = v9fs_symlink,
3115
    [P9_TCREATE] = v9fs_create,
3116
    [P9_TLCREATE] = v9fs_lcreate,
3117
    [P9_TWRITE] = v9fs_write,
3118
    [P9_TWSTAT] = v9fs_wstat,
3119
    [P9_TREMOVE] = v9fs_remove,
3120
};
3121

    
3122
static void submit_pdu(V9fsState *s, V9fsPDU *pdu)
3123
{
3124
    pdu_handler_t *handler;
3125

    
3126
    if (debug_9p_pdu) {
3127
        pprint_pdu(pdu);
3128
    }
3129

    
3130
    BUG_ON(pdu->id >= ARRAY_SIZE(pdu_handlers));
3131

    
3132
    handler = pdu_handlers[pdu->id];
3133
    BUG_ON(handler == NULL);
3134

    
3135
    handler(s, pdu);
3136
}
3137

    
3138
static void handle_9p_output(VirtIODevice *vdev, VirtQueue *vq)
3139
{
3140
    V9fsState *s = (V9fsState *)vdev;
3141
    V9fsPDU *pdu;
3142
    ssize_t len;
3143

    
3144
    while ((pdu = alloc_pdu(s)) &&
3145
            (len = virtqueue_pop(vq, &pdu->elem)) != 0) {
3146
        uint8_t *ptr;
3147

    
3148
        BUG_ON(pdu->elem.out_num == 0 || pdu->elem.in_num == 0);
3149
        BUG_ON(pdu->elem.out_sg[0].iov_len < 7);
3150

    
3151
        ptr = pdu->elem.out_sg[0].iov_base;
3152

    
3153
        memcpy(&pdu->size, ptr, 4);
3154
        pdu->id = ptr[4];
3155
        memcpy(&pdu->tag, ptr + 5, 2);
3156

    
3157
        submit_pdu(s, pdu);
3158
    }
3159

    
3160
    free_pdu(s, pdu);
3161
}
3162

    
3163
static uint32_t virtio_9p_get_features(VirtIODevice *vdev, uint32_t features)
3164
{
3165
    features |= 1 << VIRTIO_9P_MOUNT_TAG;
3166
    return features;
3167
}
3168

    
3169
static V9fsState *to_virtio_9p(VirtIODevice *vdev)
3170
{
3171
    return (V9fsState *)vdev;
3172
}
3173

    
3174
static void virtio_9p_get_config(VirtIODevice *vdev, uint8_t *config)
3175
{
3176
    struct virtio_9p_config *cfg;
3177
    V9fsState *s = to_virtio_9p(vdev);
3178

    
3179
    cfg = qemu_mallocz(sizeof(struct virtio_9p_config) +
3180
                        s->tag_len);
3181
    stw_raw(&cfg->tag_len, s->tag_len);
3182
    memcpy(cfg->tag, s->tag, s->tag_len);
3183
    memcpy(config, cfg, s->config_size);
3184
    qemu_free(cfg);
3185
}
3186

    
3187
VirtIODevice *virtio_9p_init(DeviceState *dev, V9fsConf *conf)
3188
 {
3189
    V9fsState *s;
3190
    int i, len;
3191
    struct stat stat;
3192
    FsTypeEntry *fse;
3193

    
3194

    
3195
    s = (V9fsState *)virtio_common_init("virtio-9p",
3196
                                    VIRTIO_ID_9P,
3197
                                    sizeof(struct virtio_9p_config)+
3198
                                    MAX_TAG_LEN,
3199
                                    sizeof(V9fsState));
3200

    
3201
    /* initialize pdu allocator */
3202
    QLIST_INIT(&s->free_list);
3203
    for (i = 0; i < (MAX_REQ - 1); i++) {
3204
        QLIST_INSERT_HEAD(&s->free_list, &s->pdus[i], next);
3205
    }
3206

    
3207
    s->vq = virtio_add_queue(&s->vdev, MAX_REQ, handle_9p_output);
3208

    
3209
    fse = get_fsdev_fsentry(conf->fsdev_id);
3210

    
3211
    if (!fse) {
3212
        /* We don't have a fsdev identified by fsdev_id */
3213
        fprintf(stderr, "Virtio-9p device couldn't find fsdev "
3214
                    "with the id %s\n", conf->fsdev_id);
3215
        exit(1);
3216
    }
3217

    
3218
    if (!fse->path || !conf->tag) {
3219
        /* we haven't specified a mount_tag or the path */
3220
        fprintf(stderr, "fsdev with id %s needs path "
3221
                "and Virtio-9p device needs mount_tag arguments\n",
3222
                conf->fsdev_id);
3223
        exit(1);
3224
    }
3225

    
3226
    if (!strcmp(fse->security_model, "passthrough")) {
3227
        /* Files on the Fileserver set to client user credentials */
3228
        s->ctx.fs_sm = SM_PASSTHROUGH;
3229
    } else if (!strcmp(fse->security_model, "mapped")) {
3230
        /* Files on the fileserver are set to QEMU credentials.
3231
         * Client user credentials are saved in extended attributes.
3232
         */
3233
        s->ctx.fs_sm = SM_MAPPED;
3234
    } else {
3235
        /* user haven't specified a correct security option */
3236
        fprintf(stderr, "one of the following must be specified as the"
3237
                "security option:\n\t security_model=passthrough \n\t "
3238
                "security_model=mapped\n");
3239
        return NULL;
3240
    }
3241

    
3242
    if (lstat(fse->path, &stat)) {
3243
        fprintf(stderr, "share path %s does not exist\n", fse->path);
3244
        exit(1);
3245
    } else if (!S_ISDIR(stat.st_mode)) {
3246
        fprintf(stderr, "share path %s is not a directory \n", fse->path);
3247
        exit(1);
3248
    }
3249

    
3250
    s->ctx.fs_root = qemu_strdup(fse->path);
3251
    len = strlen(conf->tag);
3252
    if (len > MAX_TAG_LEN) {
3253
        len = MAX_TAG_LEN;
3254
    }
3255
    /* s->tag is non-NULL terminated string */
3256
    s->tag = qemu_malloc(len);
3257
    memcpy(s->tag, conf->tag, len);
3258
    s->tag_len = len;
3259
    s->ctx.uid = -1;
3260

    
3261
    s->ops = fse->ops;
3262
    s->vdev.get_features = virtio_9p_get_features;
3263
    s->config_size = sizeof(struct virtio_9p_config) +
3264
                        s->tag_len;
3265
    s->vdev.get_config = virtio_9p_get_config;
3266

    
3267
    return &s->vdev;
3268
}