Statistics
| Branch: | Revision:

root / hw / virtio-9p.c @ e95ead32

History | View | Annotate | Download (55.4 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, V9fsString *path, mode_t mode, dev_t dev)
164
{
165
    return s->ops->mknod(&s->ctx, path->data, mode, dev);
166
}
167

    
168
static int v9fs_do_mksock(V9fsState *s, V9fsString *path)
169
{
170
    return s->ops->mksock(&s->ctx, path->data);
171
}
172

    
173
static int v9fs_do_mkdir(V9fsState *s, V9fsString *path, mode_t mode)
174
{
175
    return s->ops->mkdir(&s->ctx, path->data, mode);
176
}
177

    
178
static int v9fs_do_fstat(V9fsState *s, int fd, struct stat *stbuf)
179
{
180
    return s->ops->fstat(&s->ctx, fd, stbuf);
181
}
182

    
183
static int v9fs_do_open2(V9fsState *s, V9fsString *path, int flags, mode_t mode)
184
{
185
    return s->ops->open2(&s->ctx, path->data, flags, mode);
186
}
187

    
188
static int v9fs_do_symlink(V9fsState *s, V9fsString *oldpath,
189
                            V9fsString *newpath)
190
{
191
    return s->ops->symlink(&s->ctx, oldpath->data, newpath->data);
192
}
193

    
194
static int v9fs_do_link(V9fsState *s, V9fsString *oldpath, V9fsString *newpath)
195
{
196
    return s->ops->link(&s->ctx, oldpath->data, newpath->data);
197
}
198

    
199
static int v9fs_do_truncate(V9fsState *s, V9fsString *path, off_t size)
200
{
201
    return s->ops->truncate(&s->ctx, path->data, size);
202
}
203

    
204
static int v9fs_do_rename(V9fsState *s, V9fsString *oldpath,
205
                            V9fsString *newpath)
206
{
207
    return s->ops->rename(&s->ctx, oldpath->data, newpath->data);
208
}
209

    
210
static int v9fs_do_chown(V9fsState *s, V9fsString *path, uid_t uid, gid_t gid)
211
{
212
    return s->ops->chown(&s->ctx, path->data, uid, gid);
213
}
214

    
215
static int v9fs_do_utime(V9fsState *s, V9fsString *path,
216
                            const struct utimbuf *buf)
217
{
218
    return s->ops->utime(&s->ctx, path->data, buf);
219
}
220

    
221
static int v9fs_do_remove(V9fsState *s, V9fsString *path)
222
{
223
    return s->ops->remove(&s->ctx, path->data);
224
}
225

    
226
static int v9fs_do_fsync(V9fsState *s, int fd)
227
{
228
    return s->ops->fsync(&s->ctx, fd);
229
}
230

    
231
static void v9fs_string_init(V9fsString *str)
232
{
233
    str->data = NULL;
234
    str->size = 0;
235
}
236

    
237
static void v9fs_string_free(V9fsString *str)
238
{
239
    qemu_free(str->data);
240
    str->data = NULL;
241
    str->size = 0;
242
}
243

    
244
static void v9fs_string_null(V9fsString *str)
245
{
246
    v9fs_string_free(str);
247
}
248

    
249
static int number_to_string(void *arg, char type)
250
{
251
    unsigned int ret = 0;
252

    
253
    switch (type) {
254
    case 'u': {
255
        unsigned int num = *(unsigned int *)arg;
256

    
257
        do {
258
            ret++;
259
            num = num/10;
260
        } while (num);
261
        break;
262
    }
263
    default:
264
        printf("Number_to_string: Unknown number format\n");
265
        return -1;
266
    }
267

    
268
    return ret;
269
}
270

    
271
static int v9fs_string_alloc_printf(char **strp, const char *fmt, va_list ap)
272
{
273
    va_list ap2;
274
    char *iter = (char *)fmt;
275
    int len = 0;
276
    int nr_args = 0;
277
    char *arg_char_ptr;
278
    unsigned int arg_uint;
279

    
280
    /* Find the number of %'s that denotes an argument */
281
    for (iter = strstr(iter, "%"); iter; iter = strstr(iter, "%")) {
282
        nr_args++;
283
        iter++;
284
    }
285

    
286
    len = strlen(fmt) - 2*nr_args;
287

    
288
    if (!nr_args) {
289
        goto alloc_print;
290
    }
291

    
292
    va_copy(ap2, ap);
293

    
294
    iter = (char *)fmt;
295

    
296
    /* Now parse the format string */
297
    for (iter = strstr(iter, "%"); iter; iter = strstr(iter, "%")) {
298
        iter++;
299
        switch (*iter) {
300
        case 'u':
301
            arg_uint = va_arg(ap2, unsigned int);
302
            len += number_to_string((void *)&arg_uint, 'u');
303
            break;
304
        case 's':
305
            arg_char_ptr = va_arg(ap2, char *);
306
            len += strlen(arg_char_ptr);
307
            break;
308
        case 'c':
309
            len += 1;
310
            break;
311
        default:
312
            fprintf(stderr,
313
                    "v9fs_string_alloc_printf:Incorrect format %c", *iter);
314
            return -1;
315
        }
316
        iter++;
317
    }
318

    
319
alloc_print:
320
    *strp = qemu_malloc((len + 1) * sizeof(**strp));
321

    
322
    return vsprintf(*strp, fmt, ap);
323
}
324

    
325
static void v9fs_string_sprintf(V9fsString *str, const char *fmt, ...)
326
{
327
    va_list ap;
328
    int err;
329

    
330
    v9fs_string_free(str);
331

    
332
    va_start(ap, fmt);
333
    err = v9fs_string_alloc_printf(&str->data, fmt, ap);
334
    BUG_ON(err == -1);
335
    va_end(ap);
336

    
337
    str->size = err;
338
}
339

    
340
static void v9fs_string_copy(V9fsString *lhs, V9fsString *rhs)
341
{
342
    v9fs_string_free(lhs);
343
    v9fs_string_sprintf(lhs, "%s", rhs->data);
344
}
345

    
346
static size_t v9fs_string_size(V9fsString *str)
347
{
348
    return str->size;
349
}
350

    
351
static V9fsFidState *lookup_fid(V9fsState *s, int32_t fid)
352
{
353
    V9fsFidState *f;
354

    
355
    for (f = s->fid_list; f; f = f->next) {
356
        if (f->fid == fid) {
357
            return f;
358
        }
359
    }
360

    
361
    return NULL;
362
}
363

    
364
static V9fsFidState *alloc_fid(V9fsState *s, int32_t fid)
365
{
366
    V9fsFidState *f;
367

    
368
    f = lookup_fid(s, fid);
369
    if (f) {
370
        return NULL;
371
    }
372

    
373
    f = qemu_mallocz(sizeof(V9fsFidState));
374

    
375
    f->fid = fid;
376
    f->fd = -1;
377
    f->dir = NULL;
378

    
379
    f->next = s->fid_list;
380
    s->fid_list = f;
381

    
382
    return f;
383
}
384

    
385
static int free_fid(V9fsState *s, int32_t fid)
386
{
387
    V9fsFidState **fidpp, *fidp;
388

    
389
    for (fidpp = &s->fid_list; *fidpp; fidpp = &(*fidpp)->next) {
390
        if ((*fidpp)->fid == fid) {
391
            break;
392
        }
393
    }
394

    
395
    if (*fidpp == NULL) {
396
        return -ENOENT;
397
    }
398

    
399
    fidp = *fidpp;
400
    *fidpp = fidp->next;
401

    
402
    if (fidp->fd != -1) {
403
        v9fs_do_close(s, fidp->fd);
404
    }
405
    if (fidp->dir) {
406
        v9fs_do_closedir(s, fidp->dir);
407
    }
408
    v9fs_string_free(&fidp->path);
409
    qemu_free(fidp);
410

    
411
    return 0;
412
}
413

    
414
#define P9_QID_TYPE_DIR         0x80
415
#define P9_QID_TYPE_SYMLINK     0x02
416

    
417
#define P9_STAT_MODE_DIR        0x80000000
418
#define P9_STAT_MODE_APPEND     0x40000000
419
#define P9_STAT_MODE_EXCL       0x20000000
420
#define P9_STAT_MODE_MOUNT      0x10000000
421
#define P9_STAT_MODE_AUTH       0x08000000
422
#define P9_STAT_MODE_TMP        0x04000000
423
#define P9_STAT_MODE_SYMLINK    0x02000000
424
#define P9_STAT_MODE_LINK       0x01000000
425
#define P9_STAT_MODE_DEVICE     0x00800000
426
#define P9_STAT_MODE_NAMED_PIPE 0x00200000
427
#define P9_STAT_MODE_SOCKET     0x00100000
428
#define P9_STAT_MODE_SETUID     0x00080000
429
#define P9_STAT_MODE_SETGID     0x00040000
430
#define P9_STAT_MODE_SETVTX     0x00010000
431

    
432
#define P9_STAT_MODE_TYPE_BITS (P9_STAT_MODE_DIR |          \
433
                                P9_STAT_MODE_SYMLINK |      \
434
                                P9_STAT_MODE_LINK |         \
435
                                P9_STAT_MODE_DEVICE |       \
436
                                P9_STAT_MODE_NAMED_PIPE |   \
437
                                P9_STAT_MODE_SOCKET)
438

    
439
/* This is the algorithm from ufs in spfs */
440
static void stat_to_qid(const struct stat *stbuf, V9fsQID *qidp)
441
{
442
    size_t size;
443

    
444
    size = MIN(sizeof(stbuf->st_ino), sizeof(qidp->path));
445
    memcpy(&qidp->path, &stbuf->st_ino, size);
446
    qidp->version = stbuf->st_mtime ^ (stbuf->st_size << 8);
447
    qidp->type = 0;
448
    if (S_ISDIR(stbuf->st_mode)) {
449
        qidp->type |= P9_QID_TYPE_DIR;
450
    }
451
    if (S_ISLNK(stbuf->st_mode)) {
452
        qidp->type |= P9_QID_TYPE_SYMLINK;
453
    }
454
}
455

    
456
static int fid_to_qid(V9fsState *s, V9fsFidState *fidp, V9fsQID *qidp)
457
{
458
    struct stat stbuf;
459
    int err;
460

    
461
    err = v9fs_do_lstat(s, &fidp->path, &stbuf);
462
    if (err) {
463
        return err;
464
    }
465

    
466
    stat_to_qid(&stbuf, qidp);
467
    return 0;
468
}
469

    
470
static V9fsPDU *alloc_pdu(V9fsState *s)
471
{
472
    V9fsPDU *pdu = NULL;
473

    
474
    if (!QLIST_EMPTY(&s->free_list)) {
475
        pdu = QLIST_FIRST(&s->free_list);
476
        QLIST_REMOVE(pdu, next);
477
    }
478
    return pdu;
479
}
480

    
481
static void free_pdu(V9fsState *s, V9fsPDU *pdu)
482
{
483
    if (pdu) {
484
        QLIST_INSERT_HEAD(&s->free_list, pdu, next);
485
    }
486
}
487

    
488
size_t pdu_packunpack(void *addr, struct iovec *sg, int sg_count,
489
                        size_t offset, size_t size, int pack)
490
{
491
    int i = 0;
492
    size_t copied = 0;
493

    
494
    for (i = 0; size && i < sg_count; i++) {
495
        size_t len;
496
        if (offset >= sg[i].iov_len) {
497
            /* skip this sg */
498
            offset -= sg[i].iov_len;
499
            continue;
500
        } else {
501
            len = MIN(sg[i].iov_len - offset, size);
502
            if (pack) {
503
                memcpy(sg[i].iov_base + offset, addr, len);
504
            } else {
505
                memcpy(addr, sg[i].iov_base + offset, len);
506
            }
507
            size -= len;
508
            copied += len;
509
            addr += len;
510
            if (size) {
511
                offset = 0;
512
                continue;
513
            }
514
        }
515
    }
516

    
517
    return copied;
518
}
519

    
520
static size_t pdu_unpack(void *dst, V9fsPDU *pdu, size_t offset, size_t size)
521
{
522
    return pdu_packunpack(dst, pdu->elem.out_sg, pdu->elem.out_num,
523
                         offset, size, 0);
524
}
525

    
526
static size_t pdu_pack(V9fsPDU *pdu, size_t offset, const void *src,
527
                        size_t size)
528
{
529
    return pdu_packunpack((void *)src, pdu->elem.in_sg, pdu->elem.in_num,
530
                             offset, size, 1);
531
}
532

    
533
static int pdu_copy_sg(V9fsPDU *pdu, size_t offset, int rx, struct iovec *sg)
534
{
535
    size_t pos = 0;
536
    int i, j;
537
    struct iovec *src_sg;
538
    unsigned int num;
539

    
540
    if (rx) {
541
        src_sg = pdu->elem.in_sg;
542
        num = pdu->elem.in_num;
543
    } else {
544
        src_sg = pdu->elem.out_sg;
545
        num = pdu->elem.out_num;
546
    }
547

    
548
    j = 0;
549
    for (i = 0; i < num; i++) {
550
        if (offset <= pos) {
551
            sg[j].iov_base = src_sg[i].iov_base;
552
            sg[j].iov_len = src_sg[i].iov_len;
553
            j++;
554
        } else if (offset < (src_sg[i].iov_len + pos)) {
555
            sg[j].iov_base = src_sg[i].iov_base;
556
            sg[j].iov_len = src_sg[i].iov_len;
557
            sg[j].iov_base += (offset - pos);
558
            sg[j].iov_len -= (offset - pos);
559
            j++;
560
        }
561
        pos += src_sg[i].iov_len;
562
    }
563

    
564
    return j;
565
}
566

    
567
static size_t pdu_unmarshal(V9fsPDU *pdu, size_t offset, const char *fmt, ...)
568
{
569
    size_t old_offset = offset;
570
    va_list ap;
571
    int i;
572

    
573
    va_start(ap, fmt);
574
    for (i = 0; fmt[i]; i++) {
575
        switch (fmt[i]) {
576
        case 'b': {
577
            uint8_t *valp = va_arg(ap, uint8_t *);
578
            offset += pdu_unpack(valp, pdu, offset, sizeof(*valp));
579
            break;
580
        }
581
        case 'w': {
582
            uint16_t val, *valp;
583
            valp = va_arg(ap, uint16_t *);
584
            val = le16_to_cpupu(valp);
585
            offset += pdu_unpack(&val, pdu, offset, sizeof(val));
586
            *valp = val;
587
            break;
588
        }
589
        case 'd': {
590
            uint32_t val, *valp;
591
            valp = va_arg(ap, uint32_t *);
592
            val = le32_to_cpupu(valp);
593
            offset += pdu_unpack(&val, pdu, offset, sizeof(val));
594
            *valp = val;
595
            break;
596
        }
597
        case 'q': {
598
            uint64_t val, *valp;
599
            valp = va_arg(ap, uint64_t *);
600
            val = le64_to_cpup(valp);
601
            offset += pdu_unpack(&val, pdu, offset, sizeof(val));
602
            *valp = val;
603
            break;
604
        }
605
        case 'v': {
606
            struct iovec *iov = va_arg(ap, struct iovec *);
607
            int *iovcnt = va_arg(ap, int *);
608
            *iovcnt = pdu_copy_sg(pdu, offset, 0, iov);
609
            break;
610
        }
611
        case 's': {
612
            V9fsString *str = va_arg(ap, V9fsString *);
613
            offset += pdu_unmarshal(pdu, offset, "w", &str->size);
614
            /* FIXME: sanity check str->size */
615
            str->data = qemu_malloc(str->size + 1);
616
            offset += pdu_unpack(str->data, pdu, offset, str->size);
617
            str->data[str->size] = 0;
618
            break;
619
        }
620
        case 'Q': {
621
            V9fsQID *qidp = va_arg(ap, V9fsQID *);
622
            offset += pdu_unmarshal(pdu, offset, "bdq",
623
                        &qidp->type, &qidp->version, &qidp->path);
624
            break;
625
        }
626
        case 'S': {
627
            V9fsStat *statp = va_arg(ap, V9fsStat *);
628
            offset += pdu_unmarshal(pdu, offset, "wwdQdddqsssssddd",
629
                        &statp->size, &statp->type, &statp->dev,
630
                        &statp->qid, &statp->mode, &statp->atime,
631
                        &statp->mtime, &statp->length,
632
                        &statp->name, &statp->uid, &statp->gid,
633
                        &statp->muid, &statp->extension,
634
                        &statp->n_uid, &statp->n_gid,
635
                        &statp->n_muid);
636
            break;
637
        }
638
        default:
639
            break;
640
        }
641
    }
642

    
643
    va_end(ap);
644

    
645
    return offset - old_offset;
646
}
647

    
648
static size_t pdu_marshal(V9fsPDU *pdu, size_t offset, const char *fmt, ...)
649
{
650
    size_t old_offset = offset;
651
    va_list ap;
652
    int i;
653

    
654
    va_start(ap, fmt);
655
    for (i = 0; fmt[i]; i++) {
656
        switch (fmt[i]) {
657
        case 'b': {
658
            uint8_t val = va_arg(ap, int);
659
            offset += pdu_pack(pdu, offset, &val, sizeof(val));
660
            break;
661
        }
662
        case 'w': {
663
            uint16_t val;
664
            cpu_to_le16w(&val, va_arg(ap, int));
665
            offset += pdu_pack(pdu, offset, &val, sizeof(val));
666
            break;
667
        }
668
        case 'd': {
669
            uint32_t val;
670
            cpu_to_le32w(&val, va_arg(ap, uint32_t));
671
            offset += pdu_pack(pdu, offset, &val, sizeof(val));
672
            break;
673
        }
674
        case 'q': {
675
            uint64_t val;
676
            cpu_to_le64w(&val, va_arg(ap, uint64_t));
677
            offset += pdu_pack(pdu, offset, &val, sizeof(val));
678
            break;
679
        }
680
        case 'v': {
681
            struct iovec *iov = va_arg(ap, struct iovec *);
682
            int *iovcnt = va_arg(ap, int *);
683
            *iovcnt = pdu_copy_sg(pdu, offset, 1, iov);
684
            break;
685
        }
686
        case 's': {
687
            V9fsString *str = va_arg(ap, V9fsString *);
688
            offset += pdu_marshal(pdu, offset, "w", str->size);
689
            offset += pdu_pack(pdu, offset, str->data, str->size);
690
            break;
691
        }
692
        case 'Q': {
693
            V9fsQID *qidp = va_arg(ap, V9fsQID *);
694
            offset += pdu_marshal(pdu, offset, "bdq",
695
                        qidp->type, qidp->version, qidp->path);
696
            break;
697
        }
698
        case 'S': {
699
            V9fsStat *statp = va_arg(ap, V9fsStat *);
700
            offset += pdu_marshal(pdu, offset, "wwdQdddqsssssddd",
701
                        statp->size, statp->type, statp->dev,
702
                        &statp->qid, statp->mode, statp->atime,
703
                        statp->mtime, statp->length, &statp->name,
704
                        &statp->uid, &statp->gid, &statp->muid,
705
                        &statp->extension, statp->n_uid,
706
                        statp->n_gid, statp->n_muid);
707
            break;
708
        }
709
        default:
710
            break;
711
        }
712
    }
713
    va_end(ap);
714

    
715
    return offset - old_offset;
716
}
717

    
718
static void complete_pdu(V9fsState *s, V9fsPDU *pdu, ssize_t len)
719
{
720
    int8_t id = pdu->id + 1; /* Response */
721

    
722
    if (len < 0) {
723
        V9fsString str;
724
        int err = -len;
725

    
726
        str.data = strerror(err);
727
        str.size = strlen(str.data);
728

    
729
        len = 7;
730
        len += pdu_marshal(pdu, len, "s", &str);
731
        if (dotu) {
732
            len += pdu_marshal(pdu, len, "d", err);
733
        }
734

    
735
        id = P9_RERROR;
736
    }
737

    
738
    /* fill out the header */
739
    pdu_marshal(pdu, 0, "dbw", (int32_t)len, id, pdu->tag);
740

    
741
    /* keep these in sync */
742
    pdu->size = len;
743
    pdu->id = id;
744

    
745
    /* push onto queue and notify */
746
    virtqueue_push(s->vq, &pdu->elem, len);
747

    
748
    /* FIXME: we should batch these completions */
749
    virtio_notify(&s->vdev, s->vq);
750

    
751
    free_pdu(s, pdu);
752
}
753

    
754
static mode_t v9mode_to_mode(uint32_t mode, V9fsString *extension)
755
{
756
    mode_t ret;
757

    
758
    ret = mode & 0777;
759
    if (mode & P9_STAT_MODE_DIR) {
760
        ret |= S_IFDIR;
761
    }
762

    
763
    if (dotu) {
764
        if (mode & P9_STAT_MODE_SYMLINK) {
765
            ret |= S_IFLNK;
766
        }
767
        if (mode & P9_STAT_MODE_SOCKET) {
768
            ret |= S_IFSOCK;
769
        }
770
        if (mode & P9_STAT_MODE_NAMED_PIPE) {
771
            ret |= S_IFIFO;
772
        }
773
        if (mode & P9_STAT_MODE_DEVICE) {
774
            if (extension && extension->data[0] == 'c') {
775
                ret |= S_IFCHR;
776
            } else {
777
                ret |= S_IFBLK;
778
            }
779
        }
780
    }
781

    
782
    if (!(ret&~0777)) {
783
        ret |= S_IFREG;
784
    }
785

    
786
    if (mode & P9_STAT_MODE_SETUID) {
787
        ret |= S_ISUID;
788
    }
789
    if (mode & P9_STAT_MODE_SETGID) {
790
        ret |= S_ISGID;
791
    }
792
    if (mode & P9_STAT_MODE_SETVTX) {
793
        ret |= S_ISVTX;
794
    }
795

    
796
    return ret;
797
}
798

    
799
static int donttouch_stat(V9fsStat *stat)
800
{
801
    if (stat->type == -1 &&
802
        stat->dev == -1 &&
803
        stat->qid.type == -1 &&
804
        stat->qid.version == -1 &&
805
        stat->qid.path == -1 &&
806
        stat->mode == -1 &&
807
        stat->atime == -1 &&
808
        stat->mtime == -1 &&
809
        stat->length == -1 &&
810
        !stat->name.size &&
811
        !stat->uid.size &&
812
        !stat->gid.size &&
813
        !stat->muid.size &&
814
        stat->n_uid == -1 &&
815
        stat->n_gid == -1 &&
816
        stat->n_muid == -1) {
817
        return 1;
818
    }
819

    
820
    return 0;
821
}
822

    
823
static void v9fs_stat_free(V9fsStat *stat)
824
{
825
    v9fs_string_free(&stat->name);
826
    v9fs_string_free(&stat->uid);
827
    v9fs_string_free(&stat->gid);
828
    v9fs_string_free(&stat->muid);
829
    v9fs_string_free(&stat->extension);
830
}
831

    
832
static uint32_t stat_to_v9mode(const struct stat *stbuf)
833
{
834
    uint32_t mode;
835

    
836
    mode = stbuf->st_mode & 0777;
837
    if (S_ISDIR(stbuf->st_mode)) {
838
        mode |= P9_STAT_MODE_DIR;
839
    }
840

    
841
    if (dotu) {
842
        if (S_ISLNK(stbuf->st_mode)) {
843
            mode |= P9_STAT_MODE_SYMLINK;
844
        }
845

    
846
        if (S_ISSOCK(stbuf->st_mode)) {
847
            mode |= P9_STAT_MODE_SOCKET;
848
        }
849

    
850
        if (S_ISFIFO(stbuf->st_mode)) {
851
            mode |= P9_STAT_MODE_NAMED_PIPE;
852
        }
853

    
854
        if (S_ISBLK(stbuf->st_mode) || S_ISCHR(stbuf->st_mode)) {
855
            mode |= P9_STAT_MODE_DEVICE;
856
        }
857

    
858
        if (stbuf->st_mode & S_ISUID) {
859
            mode |= P9_STAT_MODE_SETUID;
860
        }
861

    
862
        if (stbuf->st_mode & S_ISGID) {
863
            mode |= P9_STAT_MODE_SETGID;
864
        }
865

    
866
        if (stbuf->st_mode & S_ISVTX) {
867
            mode |= P9_STAT_MODE_SETVTX;
868
        }
869
    }
870

    
871
    return mode;
872
}
873

    
874
static int stat_to_v9stat(V9fsState *s, V9fsString *name,
875
                            const struct stat *stbuf,
876
                            V9fsStat *v9stat)
877
{
878
    int err;
879
    const char *str;
880

    
881
    memset(v9stat, 0, sizeof(*v9stat));
882

    
883
    stat_to_qid(stbuf, &v9stat->qid);
884
    v9stat->mode = stat_to_v9mode(stbuf);
885
    v9stat->atime = stbuf->st_atime;
886
    v9stat->mtime = stbuf->st_mtime;
887
    v9stat->length = stbuf->st_size;
888

    
889
    v9fs_string_null(&v9stat->uid);
890
    v9fs_string_null(&v9stat->gid);
891
    v9fs_string_null(&v9stat->muid);
892

    
893
    if (dotu) {
894
        v9stat->n_uid = stbuf->st_uid;
895
        v9stat->n_gid = stbuf->st_gid;
896
        v9stat->n_muid = 0;
897

    
898
        v9fs_string_null(&v9stat->extension);
899

    
900
        if (v9stat->mode & P9_STAT_MODE_SYMLINK) {
901
            err = v9fs_do_readlink(s, name, &v9stat->extension);
902
            if (err == -1) {
903
                err = -errno;
904
                return err;
905
            }
906
            v9stat->extension.data[err] = 0;
907
            v9stat->extension.size = err;
908
        } else if (v9stat->mode & P9_STAT_MODE_DEVICE) {
909
            v9fs_string_sprintf(&v9stat->extension, "%c %u %u",
910
                    S_ISCHR(stbuf->st_mode) ? 'c' : 'b',
911
                    major(stbuf->st_rdev), minor(stbuf->st_rdev));
912
        } else if (S_ISDIR(stbuf->st_mode) || S_ISREG(stbuf->st_mode)) {
913
            v9fs_string_sprintf(&v9stat->extension, "%s %u",
914
                    "HARDLINKCOUNT", stbuf->st_nlink);
915
        }
916
    }
917

    
918
    str = strrchr(name->data, '/');
919
    if (str) {
920
        str += 1;
921
    } else {
922
        str = name->data;
923
    }
924

    
925
    v9fs_string_sprintf(&v9stat->name, "%s", str);
926

    
927
    v9stat->size = 61 +
928
        v9fs_string_size(&v9stat->name) +
929
        v9fs_string_size(&v9stat->uid) +
930
        v9fs_string_size(&v9stat->gid) +
931
        v9fs_string_size(&v9stat->muid) +
932
        v9fs_string_size(&v9stat->extension);
933
    return 0;
934
}
935

    
936
static struct iovec *adjust_sg(struct iovec *sg, int len, int *iovcnt)
937
{
938
    while (len && *iovcnt) {
939
        if (len < sg->iov_len) {
940
            sg->iov_len -= len;
941
            sg->iov_base += len;
942
            len = 0;
943
        } else {
944
            len -= sg->iov_len;
945
            sg++;
946
            *iovcnt -= 1;
947
        }
948
    }
949

    
950
    return sg;
951
}
952

    
953
static struct iovec *cap_sg(struct iovec *sg, int cap, int *cnt)
954
{
955
    int i;
956
    int total = 0;
957

    
958
    for (i = 0; i < *cnt; i++) {
959
        if ((total + sg[i].iov_len) > cap) {
960
            sg[i].iov_len -= ((total + sg[i].iov_len) - cap);
961
            i++;
962
            break;
963
        }
964
        total += sg[i].iov_len;
965
    }
966

    
967
    *cnt = i;
968

    
969
    return sg;
970
}
971

    
972
static void print_sg(struct iovec *sg, int cnt)
973
{
974
    int i;
975

    
976
    printf("sg[%d]: {", cnt);
977
    for (i = 0; i < cnt; i++) {
978
        if (i) {
979
            printf(", ");
980
        }
981
        printf("(%p, %zd)", sg[i].iov_base, sg[i].iov_len);
982
    }
983
    printf("}\n");
984
}
985

    
986
static void v9fs_fix_path(V9fsString *dst, V9fsString *src, int len)
987
{
988
    V9fsString str;
989
    v9fs_string_init(&str);
990
    v9fs_string_copy(&str, dst);
991
    v9fs_string_sprintf(dst, "%s%s", src->data, str.data+len);
992
    v9fs_string_free(&str);
993
}
994

    
995
static void v9fs_version(V9fsState *s, V9fsPDU *pdu)
996
{
997
    int32_t msize;
998
    V9fsString version;
999
    size_t offset = 7;
1000

    
1001
    pdu_unmarshal(pdu, offset, "ds", &msize, &version);
1002

    
1003
    if (strcmp(version.data, "9P2000.u")) {
1004
        v9fs_string_sprintf(&version, "unknown");
1005
    }
1006

    
1007
    offset += pdu_marshal(pdu, offset, "ds", msize, &version);
1008
    complete_pdu(s, pdu, offset);
1009

    
1010
    v9fs_string_free(&version);
1011
}
1012

    
1013
static void v9fs_attach(V9fsState *s, V9fsPDU *pdu)
1014
{
1015
    int32_t fid, afid, n_uname;
1016
    V9fsString uname, aname;
1017
    V9fsFidState *fidp;
1018
    V9fsQID qid;
1019
    size_t offset = 7;
1020
    ssize_t err;
1021

    
1022
    pdu_unmarshal(pdu, offset, "ddssd", &fid, &afid, &uname, &aname, &n_uname);
1023

    
1024
    fidp = alloc_fid(s, fid);
1025
    if (fidp == NULL) {
1026
        err = -EINVAL;
1027
        goto out;
1028
    }
1029

    
1030
    fidp->uid = n_uname;
1031

    
1032
    v9fs_string_sprintf(&fidp->path, "%s", "/");
1033
    err = fid_to_qid(s, fidp, &qid);
1034
    if (err) {
1035
        err = -EINVAL;
1036
        free_fid(s, fid);
1037
        goto out;
1038
    }
1039

    
1040
    offset += pdu_marshal(pdu, offset, "Q", &qid);
1041

    
1042
    err = offset;
1043
out:
1044
    complete_pdu(s, pdu, err);
1045
    v9fs_string_free(&uname);
1046
    v9fs_string_free(&aname);
1047
}
1048

    
1049
static void v9fs_stat_post_lstat(V9fsState *s, V9fsStatState *vs, int err)
1050
{
1051
    if (err == -1) {
1052
        err = -errno;
1053
        goto out;
1054
    }
1055

    
1056
    err = stat_to_v9stat(s, &vs->fidp->path, &vs->stbuf, &vs->v9stat);
1057
    if (err) {
1058
        goto out;
1059
    }
1060
    vs->offset += pdu_marshal(vs->pdu, vs->offset, "wS", 0, &vs->v9stat);
1061
    err = vs->offset;
1062

    
1063
out:
1064
    complete_pdu(s, vs->pdu, err);
1065
    v9fs_stat_free(&vs->v9stat);
1066
    qemu_free(vs);
1067
}
1068

    
1069
static void v9fs_stat(V9fsState *s, V9fsPDU *pdu)
1070
{
1071
    int32_t fid;
1072
    V9fsStatState *vs;
1073
    ssize_t err = 0;
1074

    
1075
    vs = qemu_malloc(sizeof(*vs));
1076
    vs->pdu = pdu;
1077
    vs->offset = 7;
1078

    
1079
    memset(&vs->v9stat, 0, sizeof(vs->v9stat));
1080

    
1081
    pdu_unmarshal(vs->pdu, vs->offset, "d", &fid);
1082

    
1083
    vs->fidp = lookup_fid(s, fid);
1084
    if (vs->fidp == NULL) {
1085
        err = -ENOENT;
1086
        goto out;
1087
    }
1088

    
1089
    err = v9fs_do_lstat(s, &vs->fidp->path, &vs->stbuf);
1090
    v9fs_stat_post_lstat(s, vs, err);
1091
    return;
1092

    
1093
out:
1094
    complete_pdu(s, vs->pdu, err);
1095
    v9fs_stat_free(&vs->v9stat);
1096
    qemu_free(vs);
1097
}
1098

    
1099
static void v9fs_walk_complete(V9fsState *s, V9fsWalkState *vs, int err)
1100
{
1101
    complete_pdu(s, vs->pdu, err);
1102

    
1103
    if (vs->nwnames) {
1104
        for (vs->name_idx = 0; vs->name_idx < vs->nwnames; vs->name_idx++) {
1105
            v9fs_string_free(&vs->wnames[vs->name_idx]);
1106
        }
1107

    
1108
        qemu_free(vs->wnames);
1109
        qemu_free(vs->qids);
1110
    }
1111
}
1112

    
1113
static void v9fs_walk_marshal(V9fsWalkState *vs)
1114
{
1115
    int i;
1116
    vs->offset = 7;
1117
    vs->offset += pdu_marshal(vs->pdu, vs->offset, "w", vs->nwnames);
1118

    
1119
    for (i = 0; i < vs->nwnames; i++) {
1120
        vs->offset += pdu_marshal(vs->pdu, vs->offset, "Q", &vs->qids[i]);
1121
    }
1122
}
1123

    
1124
static void v9fs_walk_post_newfid_lstat(V9fsState *s, V9fsWalkState *vs,
1125
                                                                int err)
1126
{
1127
    if (err == -1) {
1128
        free_fid(s, vs->newfidp->fid);
1129
        v9fs_string_free(&vs->path);
1130
        err = -ENOENT;
1131
        goto out;
1132
    }
1133

    
1134
    stat_to_qid(&vs->stbuf, &vs->qids[vs->name_idx]);
1135

    
1136
    vs->name_idx++;
1137
    if (vs->name_idx < vs->nwnames) {
1138
        v9fs_string_sprintf(&vs->path, "%s/%s", vs->newfidp->path.data,
1139
                                            vs->wnames[vs->name_idx].data);
1140
        v9fs_string_copy(&vs->newfidp->path, &vs->path);
1141

    
1142
        err = v9fs_do_lstat(s, &vs->newfidp->path, &vs->stbuf);
1143
        v9fs_walk_post_newfid_lstat(s, vs, err);
1144
        return;
1145
    }
1146

    
1147
    v9fs_string_free(&vs->path);
1148
    v9fs_walk_marshal(vs);
1149
    err = vs->offset;
1150
out:
1151
    v9fs_walk_complete(s, vs, err);
1152
}
1153

    
1154
static void v9fs_walk_post_oldfid_lstat(V9fsState *s, V9fsWalkState *vs,
1155
        int err)
1156
{
1157
    if (err == -1) {
1158
        v9fs_string_free(&vs->path);
1159
        err = -ENOENT;
1160
        goto out;
1161
    }
1162

    
1163
    stat_to_qid(&vs->stbuf, &vs->qids[vs->name_idx]);
1164
    vs->name_idx++;
1165
    if (vs->name_idx < vs->nwnames) {
1166

    
1167
        v9fs_string_sprintf(&vs->path, "%s/%s",
1168
                vs->fidp->path.data, vs->wnames[vs->name_idx].data);
1169
        v9fs_string_copy(&vs->fidp->path, &vs->path);
1170

    
1171
        err = v9fs_do_lstat(s, &vs->fidp->path, &vs->stbuf);
1172
        v9fs_walk_post_oldfid_lstat(s, vs, err);
1173
        return;
1174
    }
1175

    
1176
    v9fs_string_free(&vs->path);
1177
    v9fs_walk_marshal(vs);
1178
    err = vs->offset;
1179
out:
1180
    v9fs_walk_complete(s, vs, err);
1181
}
1182

    
1183
static void v9fs_walk(V9fsState *s, V9fsPDU *pdu)
1184
{
1185
    int32_t fid, newfid;
1186
    V9fsWalkState *vs;
1187
    int err = 0;
1188
    int i;
1189

    
1190
    vs = qemu_malloc(sizeof(*vs));
1191
    vs->pdu = pdu;
1192
    vs->wnames = NULL;
1193
    vs->qids = NULL;
1194
    vs->offset = 7;
1195

    
1196
    vs->offset += pdu_unmarshal(vs->pdu, vs->offset, "ddw", &fid,
1197
                                            &newfid, &vs->nwnames);
1198

    
1199
    if (vs->nwnames) {
1200
        vs->wnames = qemu_mallocz(sizeof(vs->wnames[0]) * vs->nwnames);
1201

    
1202
        vs->qids = qemu_mallocz(sizeof(vs->qids[0]) * vs->nwnames);
1203

    
1204
        for (i = 0; i < vs->nwnames; i++) {
1205
            vs->offset += pdu_unmarshal(vs->pdu, vs->offset, "s",
1206
                                            &vs->wnames[i]);
1207
        }
1208
    }
1209

    
1210
    vs->fidp = lookup_fid(s, fid);
1211
    if (vs->fidp == NULL) {
1212
        err = -ENOENT;
1213
        goto out;
1214
    }
1215

    
1216
    /* FIXME: is this really valid? */
1217
    if (fid == newfid) {
1218

    
1219
        BUG_ON(vs->fidp->fd != -1);
1220
        BUG_ON(vs->fidp->dir);
1221
        v9fs_string_init(&vs->path);
1222
        vs->name_idx = 0;
1223

    
1224
        if (vs->name_idx < vs->nwnames) {
1225
            v9fs_string_sprintf(&vs->path, "%s/%s",
1226
                vs->fidp->path.data, vs->wnames[vs->name_idx].data);
1227
            v9fs_string_copy(&vs->fidp->path, &vs->path);
1228

    
1229
            err = v9fs_do_lstat(s, &vs->fidp->path, &vs->stbuf);
1230
            v9fs_walk_post_oldfid_lstat(s, vs, err);
1231
            return;
1232
        }
1233
    } else {
1234
        vs->newfidp = alloc_fid(s, newfid);
1235
        if (vs->newfidp == NULL) {
1236
            err = -EINVAL;
1237
            goto out;
1238
        }
1239

    
1240
        vs->newfidp->uid = vs->fidp->uid;
1241
        v9fs_string_init(&vs->path);
1242
        vs->name_idx = 0;
1243
        v9fs_string_copy(&vs->newfidp->path, &vs->fidp->path);
1244

    
1245
        if (vs->name_idx < vs->nwnames) {
1246
            v9fs_string_sprintf(&vs->path, "%s/%s", vs->newfidp->path.data,
1247
                                vs->wnames[vs->name_idx].data);
1248
            v9fs_string_copy(&vs->newfidp->path, &vs->path);
1249

    
1250
            err = v9fs_do_lstat(s, &vs->newfidp->path, &vs->stbuf);
1251
            v9fs_walk_post_newfid_lstat(s, vs, err);
1252
            return;
1253
        }
1254
    }
1255

    
1256
    v9fs_walk_marshal(vs);
1257
    err = vs->offset;
1258
out:
1259
    v9fs_walk_complete(s, vs, err);
1260
}
1261

    
1262
static void v9fs_open_post_opendir(V9fsState *s, V9fsOpenState *vs, int err)
1263
{
1264
    if (vs->fidp->dir == NULL) {
1265
        err = -errno;
1266
        goto out;
1267
    }
1268

    
1269
    vs->offset += pdu_marshal(vs->pdu, vs->offset, "Qd", &vs->qid, 0);
1270
    err = vs->offset;
1271
out:
1272
    complete_pdu(s, vs->pdu, err);
1273
    qemu_free(vs);
1274

    
1275
}
1276

    
1277
static void v9fs_open_post_open(V9fsState *s, V9fsOpenState *vs, int err)
1278
{
1279
    if (vs->fidp->fd == -1) {
1280
        err = -errno;
1281
        goto out;
1282
    }
1283

    
1284
    vs->offset += pdu_marshal(vs->pdu, vs->offset, "Qd", &vs->qid, 0);
1285
    err = vs->offset;
1286
out:
1287
    complete_pdu(s, vs->pdu, err);
1288
    qemu_free(vs);
1289
}
1290

    
1291
static void v9fs_open_post_lstat(V9fsState *s, V9fsOpenState *vs, int err)
1292
{
1293
    if (err) {
1294
        err = -errno;
1295
        goto out;
1296
    }
1297

    
1298
    stat_to_qid(&vs->stbuf, &vs->qid);
1299

    
1300
    if (S_ISDIR(vs->stbuf.st_mode)) {
1301
        vs->fidp->dir = v9fs_do_opendir(s, &vs->fidp->path);
1302
        v9fs_open_post_opendir(s, vs, err);
1303
    } else {
1304
        vs->fidp->fd = v9fs_do_open(s, &vs->fidp->path,
1305
                                    omode_to_uflags(vs->mode));
1306
        v9fs_open_post_open(s, vs, err);
1307
    }
1308
    return;
1309
out:
1310
    complete_pdu(s, vs->pdu, err);
1311
    qemu_free(vs);
1312
}
1313

    
1314
static void v9fs_open(V9fsState *s, V9fsPDU *pdu)
1315
{
1316
    int32_t fid;
1317
    V9fsOpenState *vs;
1318
    ssize_t err = 0;
1319

    
1320

    
1321
    vs = qemu_malloc(sizeof(*vs));
1322
    vs->pdu = pdu;
1323
    vs->offset = 7;
1324

    
1325
    pdu_unmarshal(vs->pdu, vs->offset, "db", &fid, &vs->mode);
1326

    
1327
    vs->fidp = lookup_fid(s, fid);
1328
    if (vs->fidp == NULL) {
1329
        err = -ENOENT;
1330
        goto out;
1331
    }
1332

    
1333
    BUG_ON(vs->fidp->fd != -1);
1334
    BUG_ON(vs->fidp->dir);
1335

    
1336
    err = v9fs_do_lstat(s, &vs->fidp->path, &vs->stbuf);
1337

    
1338
    v9fs_open_post_lstat(s, vs, err);
1339
    return;
1340
out:
1341
    complete_pdu(s, pdu, err);
1342
    qemu_free(vs);
1343
}
1344

    
1345
static void v9fs_clunk(V9fsState *s, V9fsPDU *pdu)
1346
{
1347
    int32_t fid;
1348
    size_t offset = 7;
1349
    int err;
1350

    
1351
    pdu_unmarshal(pdu, offset, "d", &fid);
1352

    
1353
    err = free_fid(s, fid);
1354
    if (err < 0) {
1355
        goto out;
1356
    }
1357

    
1358
    offset = 7;
1359
    err = offset;
1360
out:
1361
    complete_pdu(s, pdu, err);
1362
}
1363

    
1364
static void v9fs_read_post_readdir(V9fsState *, V9fsReadState *, ssize_t);
1365

    
1366
static void v9fs_read_post_seekdir(V9fsState *s, V9fsReadState *vs, ssize_t err)
1367
{
1368
    if (err) {
1369
        goto out;
1370
    }
1371
    v9fs_stat_free(&vs->v9stat);
1372
    v9fs_string_free(&vs->name);
1373
    vs->offset += pdu_marshal(vs->pdu, vs->offset, "d", vs->count);
1374
    vs->offset += vs->count;
1375
    err = vs->offset;
1376
out:
1377
    complete_pdu(s, vs->pdu, err);
1378
    qemu_free(vs);
1379
    return;
1380
}
1381

    
1382
static void v9fs_read_post_dir_lstat(V9fsState *s, V9fsReadState *vs,
1383
                                    ssize_t err)
1384
{
1385
    if (err) {
1386
        err = -errno;
1387
        goto out;
1388
    }
1389
    err = stat_to_v9stat(s, &vs->name, &vs->stbuf, &vs->v9stat);
1390
    if (err) {
1391
        goto out;
1392
    }
1393

    
1394
    vs->len = pdu_marshal(vs->pdu, vs->offset + 4 + vs->count, "S",
1395
                            &vs->v9stat);
1396
    if ((vs->len != (vs->v9stat.size + 2)) ||
1397
            ((vs->count + vs->len) > vs->max_count)) {
1398
        v9fs_do_seekdir(s, vs->fidp->dir, vs->dir_pos);
1399
        v9fs_read_post_seekdir(s, vs, err);
1400
        return;
1401
    }
1402
    vs->count += vs->len;
1403
    v9fs_stat_free(&vs->v9stat);
1404
    v9fs_string_free(&vs->name);
1405
    vs->dir_pos = vs->dent->d_off;
1406
    vs->dent = v9fs_do_readdir(s, vs->fidp->dir);
1407
    v9fs_read_post_readdir(s, vs, err);
1408
    return;
1409
out:
1410
    v9fs_do_seekdir(s, vs->fidp->dir, vs->dir_pos);
1411
    v9fs_read_post_seekdir(s, vs, err);
1412
    return;
1413

    
1414
}
1415

    
1416
static void v9fs_read_post_readdir(V9fsState *s, V9fsReadState *vs, ssize_t err)
1417
{
1418
    if (vs->dent) {
1419
        memset(&vs->v9stat, 0, sizeof(vs->v9stat));
1420
        v9fs_string_init(&vs->name);
1421
        v9fs_string_sprintf(&vs->name, "%s/%s", vs->fidp->path.data,
1422
                            vs->dent->d_name);
1423
        err = v9fs_do_lstat(s, &vs->name, &vs->stbuf);
1424
        v9fs_read_post_dir_lstat(s, vs, err);
1425
        return;
1426
    }
1427

    
1428
    vs->offset += pdu_marshal(vs->pdu, vs->offset, "d", vs->count);
1429
    vs->offset += vs->count;
1430
    err = vs->offset;
1431
    complete_pdu(s, vs->pdu, err);
1432
    qemu_free(vs);
1433
    return;
1434
}
1435

    
1436
static void v9fs_read_post_telldir(V9fsState *s, V9fsReadState *vs, ssize_t err)
1437
{
1438
    vs->dent = v9fs_do_readdir(s, vs->fidp->dir);
1439
    v9fs_read_post_readdir(s, vs, err);
1440
    return;
1441
}
1442

    
1443
static void v9fs_read_post_rewinddir(V9fsState *s, V9fsReadState *vs,
1444
                                       ssize_t err)
1445
{
1446
    vs->dir_pos = v9fs_do_telldir(s, vs->fidp->dir);
1447
    v9fs_read_post_telldir(s, vs, err);
1448
    return;
1449
}
1450

    
1451
static void v9fs_read_post_readv(V9fsState *s, V9fsReadState *vs, ssize_t err)
1452
{
1453
    if (err  < 0) {
1454
        /* IO error return the error */
1455
        err = -errno;
1456
        goto out;
1457
    }
1458
    vs->total += vs->len;
1459
    vs->sg = adjust_sg(vs->sg, vs->len, &vs->cnt);
1460
    if (vs->total < vs->count && vs->len > 0) {
1461
        do {
1462
            if (0) {
1463
                print_sg(vs->sg, vs->cnt);
1464
            }
1465
            vs->len = v9fs_do_readv(s, vs->fidp->fd, vs->sg, vs->cnt);
1466
        } while (vs->len == -1 && errno == EINTR);
1467
        if (vs->len == -1) {
1468
            err  = -errno;
1469
        }
1470
        v9fs_read_post_readv(s, vs, err);
1471
        return;
1472
    }
1473
    vs->offset += pdu_marshal(vs->pdu, vs->offset, "d", vs->total);
1474
    vs->offset += vs->count;
1475
    err = vs->offset;
1476

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

    
1482
static void v9fs_read_post_lseek(V9fsState *s, V9fsReadState *vs, ssize_t err)
1483
{
1484
    if (err == -1) {
1485
        err = -errno;
1486
        goto out;
1487
    }
1488
    vs->sg = cap_sg(vs->sg, vs->count, &vs->cnt);
1489

    
1490
    if (vs->total < vs->count) {
1491
        do {
1492
            if (0) {
1493
                print_sg(vs->sg, vs->cnt);
1494
            }
1495
            vs->len = v9fs_do_readv(s, vs->fidp->fd, vs->sg, vs->cnt);
1496
        } while (vs->len == -1 && errno == EINTR);
1497
        if (vs->len == -1) {
1498
            err  = -errno;
1499
        }
1500
        v9fs_read_post_readv(s, vs, err);
1501
        return;
1502
    }
1503
out:
1504
    complete_pdu(s, vs->pdu, err);
1505
    qemu_free(vs);
1506
}
1507

    
1508
static void v9fs_read(V9fsState *s, V9fsPDU *pdu)
1509
{
1510
    int32_t fid;
1511
    V9fsReadState *vs;
1512
    ssize_t err = 0;
1513

    
1514
    vs = qemu_malloc(sizeof(*vs));
1515
    vs->pdu = pdu;
1516
    vs->offset = 7;
1517
    vs->total = 0;
1518
    vs->len = 0;
1519
    vs->count = 0;
1520

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

    
1523
    vs->fidp = lookup_fid(s, fid);
1524
    if (vs->fidp == NULL) {
1525
        err = -EINVAL;
1526
        goto out;
1527
    }
1528

    
1529
    if (vs->fidp->dir) {
1530
        vs->max_count = vs->count;
1531
        vs->count = 0;
1532
        if (vs->off == 0) {
1533
            v9fs_do_rewinddir(s, vs->fidp->dir);
1534
        }
1535
        v9fs_read_post_rewinddir(s, vs, err);
1536
        return;
1537
    } else if (vs->fidp->fd != -1) {
1538
        vs->sg = vs->iov;
1539
        pdu_marshal(vs->pdu, vs->offset + 4, "v", vs->sg, &vs->cnt);
1540
        err = v9fs_do_lseek(s, vs->fidp->fd, vs->off, SEEK_SET);
1541
        v9fs_read_post_lseek(s, vs, err);
1542
        return;
1543
    } else {
1544
        err = -EINVAL;
1545
    }
1546
out:
1547
    complete_pdu(s, pdu, err);
1548
    qemu_free(vs);
1549
}
1550

    
1551
static void v9fs_write_post_writev(V9fsState *s, V9fsWriteState *vs,
1552
                                   ssize_t err)
1553
{
1554
    if (err  < 0) {
1555
        /* IO error return the error */
1556
        err = -errno;
1557
        goto out;
1558
    }
1559
    vs->total += vs->len;
1560
    vs->sg = adjust_sg(vs->sg, vs->len, &vs->cnt);
1561
    if (vs->total < vs->count && vs->len > 0) {
1562
        do {
1563
            if (0) {
1564
                print_sg(vs->sg, vs->cnt);
1565
            }
1566
            vs->len =  v9fs_do_writev(s, vs->fidp->fd, vs->sg, vs->cnt);
1567
        } while (vs->len == -1 && errno == EINTR);
1568
        if (vs->len == -1) {
1569
            err  = -errno;
1570
        }
1571
        v9fs_write_post_writev(s, vs, err);
1572
        return;
1573
    }
1574
    vs->offset += pdu_marshal(vs->pdu, vs->offset, "d", vs->total);
1575

    
1576
    err = vs->offset;
1577
out:
1578
    complete_pdu(s, vs->pdu, err);
1579
    qemu_free(vs);
1580
}
1581

    
1582
static void v9fs_write_post_lseek(V9fsState *s, V9fsWriteState *vs, ssize_t err)
1583
{
1584
    if (err == -1) {
1585
        err = -errno;
1586
        goto out;
1587
    }
1588
    vs->sg = cap_sg(vs->sg, vs->count, &vs->cnt);
1589

    
1590
    if (vs->total < vs->count) {
1591
        do {
1592
            if (0) {
1593
                print_sg(vs->sg, vs->cnt);
1594
            }
1595
            vs->len = v9fs_do_writev(s, vs->fidp->fd, vs->sg, vs->cnt);
1596
        } while (vs->len == -1 && errno == EINTR);
1597
        if (vs->len == -1) {
1598
            err  = -errno;
1599
        }
1600
        v9fs_write_post_writev(s, vs, err);
1601
        return;
1602
    }
1603

    
1604
out:
1605
    complete_pdu(s, vs->pdu, err);
1606
    qemu_free(vs);
1607
}
1608

    
1609
static void v9fs_write(V9fsState *s, V9fsPDU *pdu)
1610
{
1611
    int32_t fid;
1612
    V9fsWriteState *vs;
1613
    ssize_t err;
1614

    
1615
    vs = qemu_malloc(sizeof(*vs));
1616

    
1617
    vs->pdu = pdu;
1618
    vs->offset = 7;
1619
    vs->sg = vs->iov;
1620
    vs->total = 0;
1621
    vs->len = 0;
1622

    
1623
    pdu_unmarshal(vs->pdu, vs->offset, "dqdv", &fid, &vs->off, &vs->count,
1624
                    vs->sg, &vs->cnt);
1625

    
1626
    vs->fidp = lookup_fid(s, fid);
1627
    if (vs->fidp == NULL) {
1628
        err = -EINVAL;
1629
        goto out;
1630
    }
1631

    
1632
    if (vs->fidp->fd == -1) {
1633
        err = -EINVAL;
1634
        goto out;
1635
    }
1636

    
1637
    err = v9fs_do_lseek(s, vs->fidp->fd, vs->off, SEEK_SET);
1638

    
1639
    v9fs_write_post_lseek(s, vs, err);
1640
    return;
1641

    
1642
out:
1643
    complete_pdu(s, vs->pdu, err);
1644
    qemu_free(vs);
1645
}
1646

    
1647
static void v9fs_post_create(V9fsState *s, V9fsCreateState *vs, int err)
1648
{
1649
    if (err == 0) {
1650
        v9fs_string_copy(&vs->fidp->path, &vs->fullname);
1651
        stat_to_qid(&vs->stbuf, &vs->qid);
1652

    
1653
        vs->offset += pdu_marshal(vs->pdu, vs->offset, "Qd", &vs->qid, 0);
1654

    
1655
        err = vs->offset;
1656
    }
1657

    
1658
    complete_pdu(s, vs->pdu, err);
1659
    v9fs_string_free(&vs->name);
1660
    v9fs_string_free(&vs->extension);
1661
    v9fs_string_free(&vs->fullname);
1662
    qemu_free(vs);
1663
}
1664

    
1665
static void v9fs_create_post_perms(V9fsState *s, V9fsCreateState *vs, int err)
1666
{
1667
    if (err) {
1668
        err = -errno;
1669
    }
1670
    v9fs_post_create(s, vs, err);
1671
}
1672

    
1673
static void v9fs_create_post_opendir(V9fsState *s, V9fsCreateState *vs,
1674
                                                                    int err)
1675
{
1676
    if (!vs->fidp->dir) {
1677
        err = -errno;
1678
    }
1679
    v9fs_post_create(s, vs, err);
1680
}
1681

    
1682
static void v9fs_create_post_dir_lstat(V9fsState *s, V9fsCreateState *vs,
1683
                                                                    int err)
1684
{
1685
    if (err) {
1686
        err = -errno;
1687
        goto out;
1688
    }
1689

    
1690
    vs->fidp->dir = v9fs_do_opendir(s, &vs->fullname);
1691
    v9fs_create_post_opendir(s, vs, err);
1692
    return;
1693

    
1694
out:
1695
    v9fs_post_create(s, vs, err);
1696
}
1697

    
1698
static void v9fs_create_post_mkdir(V9fsState *s, V9fsCreateState *vs, int err)
1699
{
1700
    if (err) {
1701
        err = -errno;
1702
        goto out;
1703
    }
1704

    
1705
    err = v9fs_do_lstat(s, &vs->fullname, &vs->stbuf);
1706
    v9fs_create_post_dir_lstat(s, vs, err);
1707
    return;
1708

    
1709
out:
1710
    v9fs_post_create(s, vs, err);
1711
}
1712

    
1713
static void v9fs_create_post_mksock(V9fsState *s, V9fsCreateState *vs,
1714
                                                                int err)
1715
{
1716
    if (err) {
1717
        err = -errno;
1718
        goto out;
1719
    }
1720

    
1721
    err = v9fs_do_chmod(s, &vs->fullname, vs->perm & 0777);
1722
    v9fs_create_post_perms(s, vs, err);
1723
    return;
1724

    
1725
out:
1726
    v9fs_post_create(s, vs, err);
1727
}
1728

    
1729
static void v9fs_create_post_fstat(V9fsState *s, V9fsCreateState *vs, int err)
1730
{
1731
    if (err) {
1732
        vs->fidp->fd = -1;
1733
        err = -errno;
1734
    }
1735

    
1736
    v9fs_post_create(s, vs, err);
1737
    return;
1738
}
1739

    
1740
static void v9fs_create_post_open2(V9fsState *s, V9fsCreateState *vs, int err)
1741
{
1742
    if (vs->fidp->fd == -1) {
1743
        err = -errno;
1744
        goto out;
1745
    }
1746

    
1747
    err = v9fs_do_fstat(s, vs->fidp->fd, &vs->stbuf);
1748
    v9fs_create_post_fstat(s, vs, err);
1749

    
1750
    return;
1751

    
1752
out:
1753
    v9fs_post_create(s, vs, err);
1754

    
1755
}
1756

    
1757
static void v9fs_create_post_lstat(V9fsState *s, V9fsCreateState *vs, int err)
1758
{
1759

    
1760
    if (err == 0 || errno != ENOENT) {
1761
        err = -errno;
1762
        goto out;
1763
    }
1764

    
1765
    if (vs->perm & P9_STAT_MODE_DIR) {
1766
        err = v9fs_do_mkdir(s, &vs->fullname, vs->perm & 0777);
1767
        v9fs_create_post_mkdir(s, vs, err);
1768
    } else if (vs->perm & P9_STAT_MODE_SYMLINK) {
1769
        err = v9fs_do_symlink(s, &vs->extension, &vs->fullname);
1770
        v9fs_create_post_perms(s, vs, err);
1771
    } else if (vs->perm & P9_STAT_MODE_LINK) {
1772
        int32_t nfid = atoi(vs->extension.data);
1773
        V9fsFidState *nfidp = lookup_fid(s, nfid);
1774
        if (nfidp == NULL) {
1775
            err = -errno;
1776
            v9fs_post_create(s, vs, err);
1777
        }
1778
        err = v9fs_do_link(s, &nfidp->path, &vs->fullname);
1779
        v9fs_create_post_perms(s, vs, err);
1780
    } else if (vs->perm & P9_STAT_MODE_DEVICE) {
1781
        char ctype;
1782
        uint32_t major, minor;
1783
        mode_t nmode = 0;
1784

    
1785
        if (sscanf(vs->extension.data, "%c %u %u", &ctype, &major,
1786
                                        &minor) != 3) {
1787
            err = -errno;
1788
            v9fs_post_create(s, vs, err);
1789
        }
1790

    
1791
        switch (ctype) {
1792
        case 'c':
1793
            nmode = S_IFCHR;
1794
            break;
1795
        case 'b':
1796
            nmode = S_IFBLK;
1797
            break;
1798
        default:
1799
            err = -EIO;
1800
            v9fs_post_create(s, vs, err);
1801
        }
1802

    
1803
        nmode |= vs->perm & 0777;
1804
        err = v9fs_do_mknod(s, &vs->fullname, nmode, makedev(major, minor));
1805
        v9fs_create_post_perms(s, vs, err);
1806
    } else if (vs->perm & P9_STAT_MODE_NAMED_PIPE) {
1807
        err = v9fs_do_mknod(s, &vs->fullname, S_IFIFO | (vs->mode & 0777), 0);
1808
        v9fs_post_create(s, vs, err);
1809
    } else if (vs->perm & P9_STAT_MODE_SOCKET) {
1810
        err = v9fs_do_mksock(s, &vs->fullname);
1811
        v9fs_create_post_mksock(s, vs, err);
1812
    } else {
1813
        vs->fidp->fd = v9fs_do_open2(s, &vs->fullname,
1814
                                omode_to_uflags(vs->mode) | O_CREAT,
1815
                                vs->perm & 0777);
1816
        v9fs_create_post_open2(s, vs, err);
1817
    }
1818

    
1819
    return;
1820

    
1821
out:
1822
    v9fs_post_create(s, vs, err);
1823
}
1824

    
1825
static void v9fs_create(V9fsState *s, V9fsPDU *pdu)
1826
{
1827
    int32_t fid;
1828
    V9fsCreateState *vs;
1829
    int err = 0;
1830

    
1831
    vs = qemu_malloc(sizeof(*vs));
1832
    vs->pdu = pdu;
1833
    vs->offset = 7;
1834

    
1835
    v9fs_string_init(&vs->fullname);
1836

    
1837
    pdu_unmarshal(vs->pdu, vs->offset, "dsdbs", &fid, &vs->name,
1838
                                &vs->perm, &vs->mode, &vs->extension);
1839

    
1840
    vs->fidp = lookup_fid(s, fid);
1841
    if (vs->fidp == NULL) {
1842
        err = -EINVAL;
1843
        goto out;
1844
    }
1845

    
1846
    v9fs_string_sprintf(&vs->fullname, "%s/%s", vs->fidp->path.data,
1847
                                                        vs->name.data);
1848

    
1849
    err = v9fs_do_lstat(s, &vs->fullname, &vs->stbuf);
1850
    v9fs_create_post_lstat(s, vs, err);
1851
    return;
1852

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

    
1860
static void v9fs_flush(V9fsState *s, V9fsPDU *pdu)
1861
{
1862
    /* A nop call with no return */
1863
    complete_pdu(s, pdu, 7);
1864
}
1865

    
1866
static void v9fs_remove_post_remove(V9fsState *s, V9fsRemoveState *vs,
1867
                                                                int err)
1868
{
1869
    /* For TREMOVE we need to clunk the fid even on failed remove */
1870
    err = free_fid(s, vs->fidp->fid);
1871
    if (err < 0) {
1872
        goto out;
1873
    }
1874

    
1875
    err = vs->offset;
1876
out:
1877
    complete_pdu(s, vs->pdu, err);
1878
    qemu_free(vs);
1879
}
1880

    
1881
static void v9fs_remove(V9fsState *s, V9fsPDU *pdu)
1882
{
1883
    int32_t fid;
1884
    V9fsRemoveState *vs;
1885
    int err = 0;
1886

    
1887
    vs = qemu_malloc(sizeof(*vs));
1888
    vs->pdu = pdu;
1889
    vs->offset = 7;
1890

    
1891
    pdu_unmarshal(vs->pdu, vs->offset, "d", &fid);
1892

    
1893
    vs->fidp = lookup_fid(s, fid);
1894
    if (vs->fidp == NULL) {
1895
        err = -EINVAL;
1896
        goto out;
1897
    }
1898

    
1899
    err = v9fs_do_remove(s, &vs->fidp->path);
1900
    v9fs_remove_post_remove(s, vs, err);
1901
    return;
1902

    
1903
out:
1904
    complete_pdu(s, pdu, err);
1905
    qemu_free(vs);
1906
}
1907

    
1908
static void v9fs_wstat_post_truncate(V9fsState *s, V9fsWstatState *vs, int err)
1909
{
1910
    if (err < 0) {
1911
        goto out;
1912
    }
1913

    
1914
    err = vs->offset;
1915

    
1916
out:
1917
    v9fs_stat_free(&vs->v9stat);
1918
    complete_pdu(s, vs->pdu, err);
1919
    qemu_free(vs);
1920
}
1921

    
1922
static void v9fs_wstat_post_rename(V9fsState *s, V9fsWstatState *vs, int err)
1923
{
1924
    if (err < 0) {
1925
        goto out;
1926
    }
1927

    
1928
    if (vs->v9stat.name.size != 0) {
1929
        v9fs_string_free(&vs->nname);
1930
    }
1931

    
1932
    if (vs->v9stat.length != -1) {
1933
        if (v9fs_do_truncate(s, &vs->fidp->path, vs->v9stat.length) < 0) {
1934
            err = -errno;
1935
        }
1936
    }
1937
    v9fs_wstat_post_truncate(s, vs, err);
1938
    return;
1939

    
1940
out:
1941
    v9fs_stat_free(&vs->v9stat);
1942
    complete_pdu(s, vs->pdu, err);
1943
    qemu_free(vs);
1944
}
1945

    
1946
static void v9fs_wstat_post_chown(V9fsState *s, V9fsWstatState *vs, int err)
1947
{
1948
    V9fsFidState *fidp;
1949
    if (err < 0) {
1950
        goto out;
1951
    }
1952

    
1953
    if (vs->v9stat.name.size != 0) {
1954
        char *old_name, *new_name;
1955
        char *end;
1956

    
1957
        old_name = vs->fidp->path.data;
1958
        end = strrchr(old_name, '/');
1959
        if (end) {
1960
            end++;
1961
        } else {
1962
            end = old_name;
1963
        }
1964

    
1965
        new_name = qemu_malloc(end - old_name + vs->v9stat.name.size + 1);
1966

    
1967
        memset(new_name, 0, end - old_name + vs->v9stat.name.size + 1);
1968
        memcpy(new_name, old_name, end - old_name);
1969
        memcpy(new_name + (end - old_name), vs->v9stat.name.data,
1970
                vs->v9stat.name.size);
1971
        vs->nname.data = new_name;
1972
        vs->nname.size = strlen(new_name);
1973

    
1974
        if (strcmp(new_name, vs->fidp->path.data) != 0) {
1975
            if (v9fs_do_rename(s, &vs->fidp->path, &vs->nname)) {
1976
                err = -errno;
1977
            } else {
1978
                /*
1979
                 * Fixup fid's pointing to the old name to
1980
                 * start pointing to the new name
1981
                 */
1982
                for (fidp = s->fid_list; fidp; fidp = fidp->next) {
1983

    
1984
                    if (vs->fidp == fidp) {
1985
                        /*
1986
                         * we replace name of this fid towards the end
1987
                         * so that our below strcmp will work
1988
                         */
1989
                        continue;
1990
                    }
1991
                    if (!strncmp(vs->fidp->path.data, fidp->path.data,
1992
                                 strlen(vs->fidp->path.data))) {
1993
                        /* replace the name */
1994
                        v9fs_fix_path(&fidp->path, &vs->nname,
1995
                                      strlen(vs->fidp->path.data));
1996
                    }
1997
                }
1998
                v9fs_string_copy(&vs->fidp->path, &vs->nname);
1999
            }
2000
        }
2001
    }
2002
    v9fs_wstat_post_rename(s, vs, err);
2003
    return;
2004

    
2005
out:
2006
    v9fs_stat_free(&vs->v9stat);
2007
    complete_pdu(s, vs->pdu, err);
2008
    qemu_free(vs);
2009
}
2010

    
2011
static void v9fs_wstat_post_utime(V9fsState *s, V9fsWstatState *vs, int err)
2012
{
2013
    if (err < 0) {
2014
        goto out;
2015
    }
2016

    
2017
    if (vs->v9stat.n_gid != -1) {
2018
        if (v9fs_do_chown(s, &vs->fidp->path, vs->v9stat.n_uid,
2019
                    vs->v9stat.n_gid)) {
2020
            err = -errno;
2021
        }
2022
    }
2023
    v9fs_wstat_post_chown(s, vs, err);
2024
    return;
2025

    
2026
out:
2027
    v9fs_stat_free(&vs->v9stat);
2028
    complete_pdu(s, vs->pdu, err);
2029
    qemu_free(vs);
2030
}
2031

    
2032
static void v9fs_wstat_post_chmod(V9fsState *s, V9fsWstatState *vs, int err)
2033
{
2034
    if (err < 0) {
2035
        goto out;
2036
    }
2037

    
2038
    if (vs->v9stat.mtime != -1) {
2039
        struct utimbuf tb;
2040
        tb.actime = 0;
2041
        tb.modtime = vs->v9stat.mtime;
2042
        if (v9fs_do_utime(s, &vs->fidp->path, &tb)) {
2043
            err = -errno;
2044
        }
2045
    }
2046

    
2047
    v9fs_wstat_post_utime(s, vs, err);
2048
    return;
2049

    
2050
out:
2051
    v9fs_stat_free(&vs->v9stat);
2052
    complete_pdu(s, vs->pdu, err);
2053
    qemu_free(vs);
2054
}
2055

    
2056
static void v9fs_wstat_post_fsync(V9fsState *s, V9fsWstatState *vs, int err)
2057
{
2058
    if (err == -1) {
2059
        err = -errno;
2060
    }
2061
    v9fs_stat_free(&vs->v9stat);
2062
    complete_pdu(s, vs->pdu, err);
2063
    qemu_free(vs);
2064
}
2065

    
2066
static void v9fs_wstat_post_lstat(V9fsState *s, V9fsWstatState *vs, int err)
2067
{
2068
    uint32_t v9_mode;
2069

    
2070
    if (err == -1) {
2071
        err = -errno;
2072
        goto out;
2073
    }
2074

    
2075
    v9_mode = stat_to_v9mode(&vs->stbuf);
2076

    
2077
    if ((vs->v9stat.mode & P9_STAT_MODE_TYPE_BITS) !=
2078
        (v9_mode & P9_STAT_MODE_TYPE_BITS)) {
2079
            /* Attempting to change the type */
2080
            err = -EIO;
2081
            goto out;
2082
    }
2083

    
2084
    if (v9fs_do_chmod(s, &vs->fidp->path, v9mode_to_mode(vs->v9stat.mode,
2085
                    &vs->v9stat.extension))) {
2086
            err = -errno;
2087
     }
2088
    v9fs_wstat_post_chmod(s, vs, err);
2089
    return;
2090

    
2091
out:
2092
    v9fs_stat_free(&vs->v9stat);
2093
    complete_pdu(s, vs->pdu, err);
2094
    qemu_free(vs);
2095
}
2096

    
2097
static void v9fs_wstat(V9fsState *s, V9fsPDU *pdu)
2098
{
2099
    int32_t fid;
2100
    V9fsWstatState *vs;
2101
    int err = 0;
2102

    
2103
    vs = qemu_malloc(sizeof(*vs));
2104
    vs->pdu = pdu;
2105
    vs->offset = 7;
2106

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

    
2109
    vs->fidp = lookup_fid(s, fid);
2110
    if (vs->fidp == NULL) {
2111
        err = -EINVAL;
2112
        goto out;
2113
    }
2114

    
2115
    /* do we need to sync the file? */
2116
    if (donttouch_stat(&vs->v9stat)) {
2117
        err = v9fs_do_fsync(s, vs->fidp->fd);
2118
        v9fs_wstat_post_fsync(s, vs, err);
2119
        return;
2120
    }
2121

    
2122
    if (vs->v9stat.mode != -1) {
2123
        err = v9fs_do_lstat(s, &vs->fidp->path, &vs->stbuf);
2124
        v9fs_wstat_post_lstat(s, vs, err);
2125
        return;
2126
    }
2127

    
2128
    v9fs_wstat_post_chmod(s, vs, err);
2129
    return;
2130

    
2131
out:
2132
    v9fs_stat_free(&vs->v9stat);
2133
    complete_pdu(s, vs->pdu, err);
2134
    qemu_free(vs);
2135
}
2136

    
2137
typedef void (pdu_handler_t)(V9fsState *s, V9fsPDU *pdu);
2138

    
2139
static pdu_handler_t *pdu_handlers[] = {
2140
    [P9_TVERSION] = v9fs_version,
2141
    [P9_TATTACH] = v9fs_attach,
2142
    [P9_TSTAT] = v9fs_stat,
2143
    [P9_TWALK] = v9fs_walk,
2144
    [P9_TCLUNK] = v9fs_clunk,
2145
    [P9_TOPEN] = v9fs_open,
2146
    [P9_TREAD] = v9fs_read,
2147
#if 0
2148
    [P9_TAUTH] = v9fs_auth,
2149
#endif
2150
    [P9_TFLUSH] = v9fs_flush,
2151
    [P9_TCREATE] = v9fs_create,
2152
    [P9_TWRITE] = v9fs_write,
2153
    [P9_TWSTAT] = v9fs_wstat,
2154
    [P9_TREMOVE] = v9fs_remove,
2155
};
2156

    
2157
static void submit_pdu(V9fsState *s, V9fsPDU *pdu)
2158
{
2159
    pdu_handler_t *handler;
2160

    
2161
    if (debug_9p_pdu) {
2162
        pprint_pdu(pdu);
2163
    }
2164

    
2165
    BUG_ON(pdu->id >= ARRAY_SIZE(pdu_handlers));
2166

    
2167
    handler = pdu_handlers[pdu->id];
2168
    BUG_ON(handler == NULL);
2169

    
2170
    handler(s, pdu);
2171
}
2172

    
2173
static void handle_9p_output(VirtIODevice *vdev, VirtQueue *vq)
2174
{
2175
    V9fsState *s = (V9fsState *)vdev;
2176
    V9fsPDU *pdu;
2177
    ssize_t len;
2178

    
2179
    while ((pdu = alloc_pdu(s)) &&
2180
            (len = virtqueue_pop(vq, &pdu->elem)) != 0) {
2181
        uint8_t *ptr;
2182

    
2183
        BUG_ON(pdu->elem.out_num == 0 || pdu->elem.in_num == 0);
2184
        BUG_ON(pdu->elem.out_sg[0].iov_len < 7);
2185

    
2186
        ptr = pdu->elem.out_sg[0].iov_base;
2187

    
2188
        memcpy(&pdu->size, ptr, 4);
2189
        pdu->id = ptr[4];
2190
        memcpy(&pdu->tag, ptr + 5, 2);
2191

    
2192
        submit_pdu(s, pdu);
2193
    }
2194

    
2195
    free_pdu(s, pdu);
2196
}
2197

    
2198
static uint32_t virtio_9p_get_features(VirtIODevice *vdev, uint32_t features)
2199
{
2200
    features |= 1 << VIRTIO_9P_MOUNT_TAG;
2201
    return features;
2202
}
2203

    
2204
static V9fsState *to_virtio_9p(VirtIODevice *vdev)
2205
{
2206
    return (V9fsState *)vdev;
2207
}
2208

    
2209
static void virtio_9p_get_config(VirtIODevice *vdev, uint8_t *config)
2210
{
2211
    struct virtio_9p_config *cfg;
2212
    V9fsState *s = to_virtio_9p(vdev);
2213

    
2214
    cfg = qemu_mallocz(sizeof(struct virtio_9p_config) +
2215
                        s->tag_len);
2216
    stw_raw(&cfg->tag_len, s->tag_len);
2217
    memcpy(cfg->tag, s->tag, s->tag_len);
2218
    memcpy(config, cfg, s->config_size);
2219
    qemu_free(cfg);
2220
}
2221

    
2222
VirtIODevice *virtio_9p_init(DeviceState *dev, V9fsConf *conf)
2223
 {
2224
    V9fsState *s;
2225
    int i, len;
2226
    struct stat stat;
2227
    FsTypeEntry *fse;
2228

    
2229

    
2230
    s = (V9fsState *)virtio_common_init("virtio-9p",
2231
                                    VIRTIO_ID_9P,
2232
                                    sizeof(struct virtio_9p_config)+
2233
                                    MAX_TAG_LEN,
2234
                                    sizeof(V9fsState));
2235

    
2236
    /* initialize pdu allocator */
2237
    QLIST_INIT(&s->free_list);
2238
    for (i = 0; i < (MAX_REQ - 1); i++) {
2239
        QLIST_INSERT_HEAD(&s->free_list, &s->pdus[i], next);
2240
    }
2241

    
2242
    s->vq = virtio_add_queue(&s->vdev, MAX_REQ, handle_9p_output);
2243

    
2244
    fse = get_fsdev_fsentry(conf->fsdev_id);
2245

    
2246
    if (!fse) {
2247
        /* We don't have a fsdev identified by fsdev_id */
2248
        fprintf(stderr, "Virtio-9p device couldn't find fsdev "
2249
                    "with the id %s\n", conf->fsdev_id);
2250
        exit(1);
2251
    }
2252

    
2253
    if (!fse->path || !conf->tag) {
2254
        /* we haven't specified a mount_tag or the path */
2255
        fprintf(stderr, "fsdev with id %s needs path "
2256
                "and Virtio-9p device needs mount_tag arguments\n",
2257
                conf->fsdev_id);
2258
        exit(1);
2259
    }
2260

    
2261
    if (!strcmp(fse->security_model, "passthrough")) {
2262
        /* Files on the Fileserver set to client user credentials */
2263
        s->ctx.fs_sm = SM_PASSTHROUGH;
2264
    } else if (!strcmp(fse->security_model, "mapped")) {
2265
        /* Files on the fileserver are set to QEMU credentials.
2266
         * Client user credentials are saved in extended attributes.
2267
         */
2268
        s->ctx.fs_sm = SM_MAPPED;
2269
    } else {
2270
        /* user haven't specified a correct security option */
2271
        fprintf(stderr, "one of the following must be specified as the"
2272
                "security option:\n\t security_model=passthrough \n\t "
2273
                "security_model=mapped\n");
2274
        return NULL;
2275
    }
2276

    
2277
    if (lstat(fse->path, &stat)) {
2278
        fprintf(stderr, "share path %s does not exist\n", fse->path);
2279
        exit(1);
2280
    } else if (!S_ISDIR(stat.st_mode)) {
2281
        fprintf(stderr, "share path %s is not a directory \n", fse->path);
2282
        exit(1);
2283
    }
2284

    
2285
    s->ctx.fs_root = qemu_strdup(fse->path);
2286
    len = strlen(conf->tag);
2287
    if (len > MAX_TAG_LEN) {
2288
        len = MAX_TAG_LEN;
2289
    }
2290
    /* s->tag is non-NULL terminated string */
2291
    s->tag = qemu_malloc(len);
2292
    memcpy(s->tag, conf->tag, len);
2293
    s->tag_len = len;
2294
    s->ctx.uid = -1;
2295

    
2296
    s->ops = fse->ops;
2297
    s->vdev.get_features = virtio_9p_get_features;
2298
    s->config_size = sizeof(struct virtio_9p_config) +
2299
                        s->tag_len;
2300
    s->vdev.get_config = virtio_9p_get_config;
2301

    
2302
    return &s->vdev;
2303
}