8a0d15b860acc8cbec911f7fa13506b092fd87d4
[akaros.git] / kern / src / ns / sysfile.c
1 // INFERNO
2 #include <vfs.h>
3 #include <kfs.h>
4 #include <slab.h>
5 #include <kmalloc.h>
6 #include <kref.h>
7 #include <string.h>
8 #include <stdio.h>
9 #include <assert.h>
10 #include <error.h>
11 #include <cpio.h>
12 #include <pmap.h>
13 #include <smp.h>
14 #include <ip.h>
15
16 enum {
17         DIRSIZE = STATFIXLEN + 32 * 4,
18         DIRREADLIM = 2048,      /* should handle the largest reasonable directory entry */
19         DIRREADSIZE=8192,       /* Just read a lot. Memory is cheap, lots of bandwidth,
20                                  * and RPCs are very expensive. At the same time,
21                                  * let's not yet exceed a common MSIZE. */
22 };
23
24 int newfd(struct chan *c, int oflags)
25 {
26         int ret = insert_obj_fdt(&current->open_files, c, 0,
27                                  oflags & O_CLOEXEC ? FD_CLOEXEC : 0,
28                                  FALSE, FALSE);
29         if (ret >= 0)
30                 cclose(c);
31         return ret;
32 }
33
34 struct chan *fdtochan(struct fd_table *fdt, int fd, int mode, int chkmnt,
35                       int iref)
36 {
37         struct chan *c;
38
39         c = lookup_fd(fdt, fd, iref, FALSE);
40         if (!c) {
41                 /* We lost the info about why there was a problem (we used to track file
42                  * group closed too, can add that in later). */
43                 set_errno(EBADF);
44                 error("Bad FD %d\n", fd);
45         }
46         if (chkmnt && (c->flag & CMSG)) {
47                 if (iref)
48                         cclose(c);
49                 set_errno(EBADF);
50                 error(Ebadusefd);
51         }
52         if (mode < 0)
53                 return c;
54         if ((mode & c->mode) != mode) {
55                 if (iref)
56                         cclose(c);
57                 set_errno(EBADF);
58                 error("FD access mode failure: chan mode 0x%x, wanted 0x%x",
59                       c->mode, mode);
60         }
61         return c;
62 }
63
64 long kchanio(void *vc, void *buf, int n, int mode)
65 {
66         ERRSTACK(1);
67         int r;
68         struct chan *c;
69
70         c = vc;
71         if (waserror()) {
72                 poperror();
73                 return -1;
74         }
75
76         if (mode == O_READ)
77                 r = devtab[c->type].read(c, buf, n, c->offset);
78         else if (mode == O_WRITE)
79                 r = devtab[c->type].write(c, buf, n, c->offset);
80         else
81                 error("kchanio: use only O_READ xor O_WRITE");
82
83         spin_lock(&c->lock);
84         c->offset += r;
85         spin_unlock(&c->lock);
86         poperror();
87         return r;
88 }
89
90 int openmode(uint32_t omode)
91 {
92 /* GIANT WARNING: if this ever throws, ipopen (and probably many others) will
93  * screw up refcnts of Qctl, err, data, etc */
94 #if 0
95         /* this is the old plan9 style.  i think they want to turn exec into read,
96          * and strip off anything higher, and just return the RD/WR style bits.  not
97          * stuff like ORCLOSE.  the lack of OEXCL might be a bug on their part (it's
98          * the only one of their non-RW-related flags that isn't masked out).
99          *
100          * Note that we no longer convert OEXEC/O_EXEC to O_READ, and instead return
101          * just the O_ACCMODE bits. */
102         if (o >= (OTRUNC | OCEXEC | ORCLOSE | OEXEC))
103                 error(Ebadarg);
104         o &= ~(OTRUNC | OCEXEC | ORCLOSE);
105         if (o > OEXEC)
106                 error(Ebadarg);
107         if (o == OEXEC)
108                 return OREAD;
109         return o;
110 #endif
111         /* no error checking (we have a shitload of flags anyway), and we return the
112          * basic access modes (RD/WR/ETC) */
113         return omode & O_ACCMODE;
114 }
115
116 void fdclose(struct fd_table *fdt, int fd)
117 {
118         close_fd(fdt, fd);
119 }
120
121 int syschdir(char *path)
122 {
123         ERRSTACK(1);
124         struct chan *c;
125         struct pgrp *pg;
126
127         if (waserror()) {
128                 poperror();
129                 return -1;
130         }
131
132         c = namec(path, Atodir, 0, 0);
133         pg = current->pgrp;
134         cclose(pg->dot);
135         pg->dot = c;
136         poperror();
137         return 0;
138 }
139
140 int sysclose(int fd)
141 {
142         ERRSTACK(1);
143         struct fd_table *fdt = &current->open_files;
144
145         if (waserror()) {
146                 poperror();
147                 return -1;
148         }
149         /*
150          * Take no reference on the chan because we don't really need the
151          * data structure, and are calling fdtochan only for error checks.
152          * fdclose takes care of processes racing through here.
153          */
154         fdtochan(fdt, fd, -1, 0, 0);
155         fdclose(fdt, fd);
156         poperror();
157         return 0;
158 }
159
160 int syscreate(char *path, int mode, uint32_t perm)
161 {
162         ERRSTACK(2);
163         int fd;
164         struct chan *c;
165
166         if (waserror()) {
167                 poperror();
168                 return -1;
169         }
170
171         openmode(mode & ~O_EXCL);       /* error check only; OEXCL okay here */
172         c = namec(path, Acreate, mode, perm);
173         if (waserror()) {
174                 cclose(c);
175                 nexterror();
176         }
177         fd = newfd(c, mode);    /* 9ns mode is the O_FLAGS and perm is glibc mode */
178         if (fd < 0)
179                 error(Enofd);
180         poperror();
181
182         poperror();
183         return fd;
184 }
185
186 int sysdup(int old)
187 {
188         ERRSTACK(1);
189         int fd;
190         struct chan *c;
191
192         if (waserror()) {
193                 poperror();
194                 return -1;
195         }
196         c = fdtochan(&current->open_files, old, -1, 0, 1);
197         if (c->qid.type & QTAUTH) {
198                 cclose(c);
199                 error(Eperm);
200         }
201         fd = newfd(c, 0);
202         if (fd < 0) {
203                 cclose(c);
204                 error(Enofd);
205         }
206         poperror();
207         return fd;
208 }
209
210 /* Could pass in the fdt instead of the proc, but we used to need the to_proc
211  * for now so we can claim a VFS FD.  Careful, we don't close the old chan. */
212 int sys_dup_to(struct proc *from_proc, unsigned int from_fd,
213                struct proc *to_proc, unsigned int to_fd)
214 {
215         ERRSTACK(1);
216         int ret;
217         struct chan *c;
218
219         if (waserror()) {
220                 poperror();
221                 return -1;
222         }
223         c = fdtochan(&from_proc->open_files, from_fd, -1, 0, 1);
224         if (c->qid.type & QTAUTH) {
225                 cclose(c);
226                 error(Eperm);
227         }
228         ret = insert_obj_fdt(&to_proc->open_files, c, to_fd, 0, TRUE, FALSE);
229         /* drop the ref from fdtochan.  if insert succeeded, there is one other ref
230          * stored in the FDT */
231         cclose(c);
232         if (ret < 0)
233                 error("Can't insert FD %d into FDG", to_fd);
234         poperror();
235         return 0;
236 }
237
238 char *sysfd2path(int fd)
239 {
240         ERRSTACK(1);
241         struct chan *c;
242         char *s;
243
244         if (waserror()) {
245                 poperror();
246                 return NULL;
247         }
248         c = fdtochan(&current->open_files, fd, -1, 0, 1);
249         s = NULL;
250         if (c->name != NULL) {
251                 s = kzmalloc(c->name->len + 1, 0);
252                 if (s == NULL) {
253                         cclose(c);
254                         error(Enomem);
255                 }
256                 memmove(s, c->name->s, c->name->len + 1);
257         }
258         cclose(c);
259         poperror();
260         return s;
261 }
262
263 int sysfauth(int fd, char *aname)
264 {
265         ERRSTACK(2);
266         struct chan *c, *ac;
267
268         if (waserror()) {
269                 poperror();
270                 return -1;
271         }
272
273         validname(aname, 0);
274         c = fdtochan(&current->open_files, fd, O_RDWR, 0, 1);
275         if (waserror()) {
276                 cclose(c);
277                 nexterror();
278         }
279
280         ac = mntauth(c, aname);
281
282         /* at this point ac is responsible for keeping c alive */
283         poperror();     /* c */
284         cclose(c);
285
286         if (waserror()) {
287                 cclose(ac);
288                 nexterror();
289         }
290
291         fd = newfd(ac, 0);
292         if (fd < 0)
293                 error(Enofd);
294         poperror();     /* ac */
295
296         poperror();
297
298         return fd;
299 }
300
301 int sysfversion(int fd, unsigned int msize, char *vers, unsigned int arglen)
302 {
303         ERRSTACK(2);
304         int m;
305         struct chan *c;
306
307         if (waserror()) {
308                 poperror();
309                 return -1;
310         }
311
312         /* check there's a NUL in the version string */
313         if (arglen == 0 || memchr(vers, 0, arglen) == 0)
314                 error(Ebadarg);
315
316         c = fdtochan(&current->open_files, fd, O_RDWR, 0, 1);
317         if (waserror()) {
318                 cclose(c);
319                 nexterror();
320         }
321
322         m = mntversion(c, vers, msize, arglen);
323
324         poperror();
325         cclose(c);
326
327         poperror();
328         return m;
329 }
330
331 int syspipe(int fd[2])
332 {
333         ERRSTACK(1);
334         struct dev *d;
335         struct chan *c[2];
336         static char *names[] = { "data", "data1" };
337
338         d = &devtab[devno("pipe", 0)];
339         c[0] = 0;
340         c[1] = 0;
341         fd[0] = -1;
342         fd[1] = -1;
343         if (waserror()) {
344                 /* need to remove from the fd table and make sure the chan is closed
345                  * exactly once.  if fd[i] >= 0, then the fd is valid (or it was!) and
346                  * the fd table has the only ref (newfd() currently decrefs/consumes the
347                  * reference).  cclose() doesn't care if you pass it 0 (like kfree()). */
348                 if (fd[0] >= 0)
349                         close_fd(&current->open_files, fd[0]);
350                 else
351                         cclose(c[0]);
352                 if (fd[1] >= 0)
353                         close_fd(&current->open_files, fd[1]);
354                 else
355                         cclose(c[1]);
356                 poperror();
357                 return -1;
358         }
359         c[0] = namec("#pipe", Atodir, 0, 0);
360         c[1] = cclone(c[0]);
361         if (walk(&c[0], &names[0], 1, FALSE, NULL) < 0)
362                 error(Egreg);
363         if (walk(&c[1], &names[1], 1, FALSE, NULL) < 0)
364                 error(Egreg);
365         c[0] = d->open(c[0], O_RDWR);
366         c[1] = d->open(c[1], O_RDWR);
367         fd[0] = newfd(c[0], 0);
368         if (fd[0] < 0)
369                 error(Enofd);
370         fd[1] = newfd(c[1], 0);
371         if (fd[1] < 0)
372                 error(Enofd);
373         poperror();
374         return 0;
375 }
376
377 int sysfwstat(int fd, uint8_t * buf, int n)
378 {
379         ERRSTACK(2);
380         struct chan *c;
381
382         if (waserror()) {
383                 poperror();
384                 return -1;
385         }
386
387         validstat(buf, n, 0);
388         c = fdtochan(&current->open_files, fd, -1, 1, 1);
389         if (waserror()) {
390                 cclose(c);
391                 nexterror();
392         }
393         n = devtab[c->type].wstat(c, buf, n);
394         poperror();
395         cclose(c);
396
397         poperror();
398         return n;
399 }
400
401 long bindmount(struct chan *c, char *old, int flag, char *spec)
402 {
403         ERRSTACK(1);
404         int ret;
405         struct chan *c1;
406
407         if (flag > MMASK || (flag & MORDER) == (MBEFORE | MAFTER))
408                 error(Ebadarg);
409
410         c1 = namec(old, Amount, 0, 0);
411         if (waserror()) {
412                 cclose(c1);
413                 nexterror();
414         }
415         ret = cmount(c, c1, flag, spec);
416
417         poperror();
418         cclose(c1);
419         return ret;
420 }
421
422 int sysbind(char *new, char *old, int flags)
423 {
424         ERRSTACK(2);
425         long r;
426         struct chan *c0;
427
428         if (waserror()) {
429                 poperror();
430                 return -1;
431         }
432
433         c0 = namec(new, Abind, 0, 0);
434         if (waserror()) {
435                 cclose(c0);
436                 nexterror();
437         }
438         r = bindmount(c0, old, flags, "");
439         poperror();
440         cclose(c0);
441
442         poperror();
443         return r;
444 }
445
446 int sysmount(int fd, int afd, char *old, int flags, char *spec)
447 {
448         ERRSTACK(1);
449         long r;
450         volatile struct {
451                 struct chan *c;
452         } c0;
453         volatile struct {
454                 struct chan *c;
455         } bc;
456         volatile struct {
457                 struct chan *c;
458         } ac;
459         struct mntparam mntparam;
460
461         ac.c = NULL;
462         bc.c = NULL;
463         c0.c = NULL;
464         if (waserror()) {
465                 cclose(ac.c);
466                 cclose(bc.c);
467                 cclose(c0.c);
468                 poperror();
469                 return -1;
470         }
471         bc.c = fdtochan(&current->open_files, fd, O_RDWR, 0, 1);
472         if (afd >= 0)
473                 ac.c = fdtochan(&current->open_files, afd, O_RDWR, 0, 1);
474         mntparam.chan = bc.c;
475         mntparam.authchan = ac.c;
476         mntparam.spec = spec;
477         mntparam.flags = flags;
478         c0.c = devtab[devno("mnt", 0)].attach((char *)&mntparam);
479
480         r = bindmount(c0.c, old, flags, spec);
481         poperror();
482         cclose(ac.c);
483         cclose(bc.c);
484         cclose(c0.c);
485
486         return r;
487 }
488
489 int sysunmount(char *src_path, char *onto_path)
490 {
491         ERRSTACK(1);
492         volatile struct {
493                 struct chan *c;
494         } cmount;
495         volatile struct {
496                 struct chan *c;
497         } cmounted;
498
499         cmount.c = NULL;
500         cmounted.c = NULL;
501         if (waserror()) {
502                 cclose(cmount.c);
503                 cclose(cmounted.c);
504                 poperror();
505                 return -1;
506         }
507
508         cmount.c = namec(onto_path, Amount, 0, 0);
509         if (src_path != NULL && src_path[0] != '\0') {
510                 /*
511                  * This has to be namec(..., Aopen, ...) because
512                  * if arg[0] is something like /srv/cs or /fd/0,
513                  * opening it is the only way to get at the real
514                  * Chan underneath.
515                  */
516                 cmounted.c = namec(src_path, Aopen, O_READ, 0);
517         }
518
519         cunmount(cmount.c, cmounted.c);
520         poperror();
521         cclose(cmount.c);
522         cclose(cmounted.c);
523         return 0;
524 }
525
526 int sysopen(char *path, int vfs_flags)
527 {
528         ERRSTACK(2);
529         int fd;
530         struct chan *c;
531
532         if (waserror()) {
533                 poperror();
534                 return -1;
535         }
536
537         openmode(vfs_flags);    /* error check only */
538         c = namec(path, Aopen, vfs_flags, 0);
539         if (waserror()) {
540                 cclose(c);
541                 nexterror();
542         }
543         fd = newfd(c, vfs_flags);
544         if (fd < 0)
545                 error(Enofd);
546         poperror();
547
548         poperror();
549         return fd;
550 }
551
552 long unionread(struct chan *c, void *va, long n)
553 {
554         ERRSTACK(1);
555         int i;
556         long nr;
557         struct mhead *m;
558         struct mount *mount;
559
560         qlock(&c->umqlock);
561         m = c->umh;
562         rlock(&m->lock);
563         mount = m->mount;
564         /* bring mount in sync with c->uri and c->umc */
565         for (i = 0; mount != NULL && i < c->uri; i++)
566                 mount = mount->next;
567
568         nr = 0;
569         while (mount != NULL) {
570                 /* Error causes component of union to be skipped */
571                 if (mount->to) {
572                         /* normally we want to discard the error, but for our ghetto kdirent
573                          * hack, we need to repeat unionread if we saw a Eshort */
574                         if (waserror()) {
575                                 if (!strcmp(current_errstr(), Eshort)) {
576                                         runlock(&m->lock);
577                                         qunlock(&c->umqlock);
578                                         nexterror();
579                                 }
580                                 /* poperror done below for either branch */
581                         } else {
582                                 if (c->umc == NULL) {
583                                         c->umc = cclone(mount->to);
584                                         c->umc = devtab[c->umc->type].open(c->umc,
585                                                                            O_READ);
586                                 }
587
588                                 nr = devtab[c->umc->type].read(c->umc, va, n, c->umc->offset);
589                                 if (nr < 0)
590                                         nr = 0; /* dev.c can return -1 */
591                                 c->umc->offset += nr;
592                         }
593                         poperror();     /* pop regardless */
594                 }
595                 if (nr > 0)
596                         break;
597
598                 /* Advance to next element */
599                 c->uri++;
600                 if (c->umc) {
601                         cclose(c->umc);
602                         c->umc = NULL;
603                 }
604                 mount = mount->next;
605         }
606         runlock(&m->lock);
607         qunlock(&c->umqlock);
608         return nr;
609 }
610
611 static void unionrewind(struct chan *c)
612 {
613         qlock(&c->umqlock);
614         c->uri = 0;
615         if (c->umc) {
616                 cclose(c->umc);
617                 c->umc = NULL;
618         }
619         qunlock(&c->umqlock);
620 }
621
622 static long rread(int fd, void *va, long n, int64_t * offp)
623 {
624         ERRSTACK(3);
625         int dir;
626         struct chan *c;
627         int64_t off;
628
629         /* dirty dirent hack */
630         void *real_va = va;
631
632         if (waserror()) {
633                 poperror();
634                 return -1;
635         }
636
637         c = fdtochan(&current->open_files, fd, O_READ, 1, 1);
638         if (waserror()) {
639                 cclose(c);
640                 nexterror();
641         }
642
643         if (n < 0)
644                 error(Etoosmall);
645
646         dir = c->qid.type & QTDIR;
647
648         /* kdirent hack: userspace is expecting kdirents, but all of 9ns
649          * produces Ms.  Just save up what we don't use and append the
650          * new stuff later. Allocate DIRREADSIZE bytes for that purpose.
651          */
652         if (dir) {
653                 int amt;
654                 /* expecting only one dirent at a time, o/w we're busted */
655                 assert(n >= sizeof(struct kdirent));
656                 if (!c->buf) {
657                         c->buf=kmalloc(DIRREADSIZE, KMALLOC_WAIT);
658                         c->bufused = 0;
659                 }
660                 /* Attempt to extract an M, in case there was some already */
661                 amt = convM2kdirent(c->buf, c->bufused, real_va, 0);
662                 if (amt) {
663                         c->bufused -= amt;
664                         memmove(c->buf, c->buf + amt, c->bufused);
665                         n = sizeof(struct kdirent);
666                         goto out;
667                 }
668                 /* debugging */
669                 if (waserror()) {
670                         printk("Well, sysread of a dir sucks.%s \n", current_errstr());
671                         nexterror();
672                 }
673                 va = c->buf + c->bufused;
674                 n = DIRREADSIZE - c->bufused;
675         }
676
677         /* this is the normal plan9 read */
678         if (dir && c->umh)
679                 n = unionread(c, va, n);
680         else {
681                 if (offp == NULL) {
682                         spin_lock(&c->lock);    /* lock for int64_t assignment */
683                         off = c->offset;
684                         spin_unlock(&c->lock);
685                 } else
686                         off = *offp;
687                 if (off < 0)
688                         error(Enegoff);
689                 if (off == 0) {
690                         if (offp == NULL) {
691                                 spin_lock(&c->lock);
692                                 c->offset = 0;
693                                 c->dri = 0;
694                                 spin_unlock(&c->lock);
695                         }
696                         unionrewind(c);
697                 }
698                 if (! c->ateof) {
699                         n = devtab[c->type].read(c, va, n, off);
700                         if (n == 0 && dir)
701                                 c->ateof = 1;
702                 } else {
703                         n = 0;
704                 }
705                 spin_lock(&c->lock);
706                 c->offset += n;
707                 spin_unlock(&c->lock);
708         }
709
710         /* dirty kdirent hack */
711         if (dir) {
712                 int amt;
713                 c->bufused = c->bufused + n;
714                 /* extract an M from the front, then shift the remainder back */
715                 amt = convM2kdirent(c->buf, c->bufused, real_va, 0);
716                 c->bufused -= amt;
717                 memmove(c->buf, c->buf + amt, c->bufused);
718                 n = amt ? sizeof(struct kdirent) : 0;
719                 poperror();     /* matching our debugging waserror */
720         }
721
722 out:
723         poperror();
724         cclose(c);
725
726         poperror();
727         return n;
728 }
729
730 /* Reads exactly n bytes from chan c, starting at its offset.  Can block, but if
731  * we get 0 back too soon (EOF or error), then we'll error out with Eshort.
732  * That might need a little work - if there was a previous error, then we
733  * clobbered it and only know Eshort but not why we completed early. */
734 void read_exactly_n(struct chan *c, void *vp, long n)
735 {
736         char *p;
737         long nn;
738         int total = 0, want = n;
739
740         p = vp;
741         while (n > 0) {
742                 nn = devtab[c->type].read(c, p, n, c->offset);
743                 printd("readn: Got %d@%lld\n", nn, c->offset);
744                 if (nn == 0)
745                         error("%s: wanted %d, got %d", Eshort, want, total);
746                 spin_lock(&c->lock);
747                 c->offset += nn;
748                 spin_unlock(&c->lock);
749                 p += nn;
750                 n -= nn;
751                 total += nn;
752         }
753 }
754
755 long sysread(int fd, void *va, long n)
756 {
757         return rread(fd, va, n, NULL);
758 }
759
760 long syspread(int fd, void *va, long n, int64_t off)
761 {
762         return rread(fd, va, n, &off);
763 }
764
765 int sysremove(char *path)
766 {
767         ERRSTACK(2);
768         struct chan *c;
769
770         if (waserror()) {
771                 poperror();
772                 return -1;
773         }
774
775         c = namec(path, Aremove, 0, 0);
776         if (waserror()) {
777                 c->type = -1;   /* see below */
778                 cclose(c);
779                 nexterror();
780         }
781         devtab[c->type].remove(c);
782         /*
783          * Remove clunks the fid, but we need to recover the Chan
784          * so fake it up.  -1 aborts the dev's close.
785          */
786         c->type = -1;
787         poperror();
788         cclose(c);
789
790         poperror();
791         return 0;
792 }
793
794 int64_t sysseek(int fd, int64_t off, int whence)
795 {
796         ERRSTACK(2);
797         struct dir *dir;
798         struct chan *c;
799
800         if (waserror()) {
801                 poperror();
802                 return -1;
803         }
804
805         c = fdtochan(&current->open_files, fd, -1, 1, 1);
806         if (waserror()) {
807                 cclose(c);
808                 nexterror();
809         }
810
811         /* TODO: WTF is this?  Is pipe magically the only device that isn't
812          * seekable? */
813         if (!strcmp(devtab[c->type].name, "pipe"))
814                 error(Eisstream);
815
816         switch (whence) {
817                 case 0:
818                         if (c->qid.type & QTDIR) {
819                                 if (off != 0)
820                                         error(Eisdir);
821                                 unionrewind(c);
822                         } else if (off < 0)
823                                 error(Enegoff);
824                         spin_lock(&c->lock);    /* lock for int64_t assignment */
825                         c->offset = off;
826                         spin_unlock(&c->lock);
827                         break;
828
829                 case 1:
830                         if (c->qid.type & QTDIR)
831                                 error(Eisdir);
832                         spin_lock(&c->lock);    /* lock for read/write update */
833                         off += c->offset;
834                         if (off < 0) {
835                                 spin_unlock(&c->lock);
836                                 error(Enegoff);
837                         }
838                         c->offset = off;
839                         spin_unlock(&c->lock);
840                         break;
841
842                 case 2:
843                         if (c->qid.type & QTDIR)
844                                 error(Eisdir);
845                         dir = chandirstat(c);
846                         if (dir == NULL)
847                                 error("internal error: stat error in seek");
848                         off += dir->length;
849                         kfree(dir);
850                         if (off < 0)
851                                 error(Enegoff);
852                         spin_lock(&c->lock);    /* lock for read/write update */
853                         c->offset = off;
854                         spin_unlock(&c->lock);
855                         break;
856
857                 default:
858                         error(Ebadarg);
859                         break;
860         }
861         poperror();
862         c->dri = 0;
863         cclose(c);
864         poperror();
865         return off;
866 }
867
868 void validstat(uint8_t * s, int n, int slashok)
869 {
870         
871         int m;
872         char buf[64];
873
874         if (statcheck(s, n) < 0)
875                 error(Ebadstat);
876         /* verify that name entry is acceptable */
877         s += STATFIXLEN - 4 * BIT16SZ;  /* location of first string */
878         /*
879          * s now points at count for first string.
880          * if it's too long, let the server decide; this is
881          * only for his protection anyway. otherwise
882          * we'd have to allocate and waserror.
883          */
884         m = GBIT16(s);
885         s += BIT16SZ;
886         if (m + 1 > sizeof buf) {
887                 return;
888         }
889         memmove(buf, s, m);
890         buf[m] = '\0';
891         /* name could be '/' */
892         if (strcmp(buf, "/") != 0)
893                 validname(buf, slashok);
894 }
895
896 int sysfstat(int fd, uint8_t *buf, int n)
897 {
898         ERRSTACK(2);
899         struct chan *c;
900
901         if (waserror()) {
902                 poperror();
903                 return -1;
904         }
905
906         c = fdtochan(&current->open_files, fd, -1, 0, 1);
907         if (waserror()) {
908                 cclose(c);
909                 nexterror();
910         }
911         devtab[c->type].stat(c, buf, n);
912
913         poperror();
914         cclose(c);
915
916         poperror();
917         return n;
918 }
919
920 int sysfstatakaros(int fd, struct kstat *ks)
921 {
922         
923         int n = 4096;
924         uint8_t *buf;
925         buf = kmalloc(n, KMALLOC_WAIT);
926         n = sysfstat(fd, buf, n);
927         if (n > 0) {
928                 convM2kstat(buf, n, ks);
929                 n = 0;
930         }
931         kfree(buf);
932         return n;
933 }
934
935 int sysstat(char *path, uint8_t *buf, int n)
936 {
937         ERRSTACK(2);
938         struct chan *c;
939
940         if (waserror()) {
941                 poperror();
942                 return -1;
943         }
944
945         c = namec(path, Aaccess, 0, 0);
946         if (waserror()) {
947                 cclose(c);
948                 nexterror();
949         }
950         devtab[c->type].stat(c, buf, n);
951         poperror();
952         cclose(c);
953
954         poperror();
955
956         return n;
957 }
958
959 int sysstatakaros(char *path, struct kstat *ks)
960 {
961         
962         int n = 4096;
963         uint8_t *buf;
964         buf = kmalloc(n, KMALLOC_WAIT);
965         n = sysstat(path, buf, n);
966         if (n > 0) {
967                 convM2kstat(buf, n, ks);
968                 n = 0;
969         }
970         kfree(buf);
971         return n;
972 }
973
974 static long rwrite(int fd, void *va, long n, int64_t * offp)
975 {
976         ERRSTACK(3);
977         struct chan *c;
978         struct dir *dir;
979         int64_t off;
980         long m;
981
982         if (waserror()) {
983                 poperror();
984                 return -1;
985         }
986         c = fdtochan(&current->open_files, fd, O_WRITE, 1, 1);
987         if (waserror()) {
988                 cclose(c);
989                 nexterror();
990         }
991         if (c->qid.type & QTDIR)
992                 error(Eisdir);
993
994         if (n < 0)
995                 error(Etoosmall);
996
997         if (offp == NULL) {
998                 /* append changes the offset to the end, and even if we fail later, this
999                  * change will persist */
1000                 if (c->flag & O_APPEND) {
1001                         dir = chandirstat(c);
1002                         if (!dir)
1003                                 error("internal error: stat error in append write");
1004                         spin_lock(&c->lock);    /* legacy lock for int64 assignment */
1005                         c->offset = dir->length;
1006                         spin_unlock(&c->lock);
1007                         kfree(dir);
1008                 }
1009                 spin_lock(&c->lock);
1010                 off = c->offset;
1011                 c->offset += n;
1012                 spin_unlock(&c->lock);
1013         } else
1014                 off = *offp;
1015
1016         if (waserror()) {
1017                 if (offp == NULL) {
1018                         spin_lock(&c->lock);
1019                         c->offset -= n;
1020                         spin_unlock(&c->lock);
1021                 }
1022                 nexterror();
1023         }
1024         if (off < 0)
1025                 error(Enegoff);
1026         m = devtab[c->type].write(c, va, n, off);
1027         poperror();
1028
1029         if (offp == NULL && m < n) {
1030                 spin_lock(&c->lock);
1031                 c->offset -= n - m;
1032                 spin_unlock(&c->lock);
1033         }
1034
1035         poperror();
1036         cclose(c);
1037
1038         poperror();
1039         return n;
1040 }
1041
1042 long syswrite(int fd, void *va, long n)
1043 {
1044         return rwrite(fd, va, n, NULL);
1045 }
1046
1047 long syspwrite(int fd, void *va, long n, int64_t off)
1048 {
1049         return rwrite(fd, va, n, &off);
1050 }
1051
1052 int syswstat(char *path, uint8_t * buf, int n)
1053 {
1054         ERRSTACK(2);
1055         struct chan *c;
1056
1057         if (waserror()) {
1058                 poperror();
1059                 return -1;
1060         }
1061
1062         validstat(buf, n, 0);
1063         c = namec(path, Aaccess, 0, 0);
1064         if (waserror()) {
1065                 cclose(c);
1066                 nexterror();
1067         }
1068         n = devtab[c->type].wstat(c, buf, n);
1069         poperror();
1070         cclose(c);
1071
1072         poperror();
1073         return n;
1074 }
1075
1076 struct dir *chandirstat(struct chan *c)
1077 {
1078         ERRSTACK(1);
1079         struct dir *d;
1080         uint8_t *buf;
1081         int n, nd, i;
1082
1083         nd = DIRSIZE;
1084         for (i = 0; i < 2; i++) {       /* should work by the second try */
1085                 d = kzmalloc(sizeof(struct dir) + nd, 0);
1086                 buf = (uint8_t *) & d[1];
1087                 if (waserror()) {
1088                         kfree(d);
1089                         poperror();
1090                         return NULL;
1091                 }
1092                 n = devtab[c->type].stat(c, buf, nd);
1093                 poperror();
1094                 if (n < BIT16SZ) {
1095                         kfree(d);
1096                         return NULL;
1097                 }
1098                 nd = GBIT16((uint8_t *) buf) + BIT16SZ; /* size needed to store whole stat buffer including count */
1099                 if (nd <= n) {
1100                         convM2D(buf, n, d, (char *)&d[1]);
1101                         return d;
1102                 }
1103                 /* else sizeof(Dir)+nd is plenty */
1104                 kfree(d);
1105         }
1106         return NULL;
1107
1108 }
1109
1110 struct dir *sysdirstat(char *name)
1111 {
1112         ERRSTACK(2);
1113         struct chan *c;
1114         struct dir *d;
1115
1116         if (waserror()) {
1117                 poperror();
1118                 return NULL;
1119         }
1120
1121         c = namec(name, Aaccess, 0, 0);
1122         if (waserror()) {
1123                 cclose(c);
1124                 nexterror();
1125         }
1126         d = chandirstat(c);
1127         poperror();
1128         cclose(c);
1129
1130         poperror();
1131         return d;
1132 }
1133
1134 struct dir *sysdirfstat(int fd)
1135 {
1136         ERRSTACK(2);
1137         struct chan *c;
1138         struct dir *d;
1139
1140         if (waserror()) {
1141                 poperror();
1142                 return NULL;
1143         }
1144
1145         c = fdtochan(&current->open_files, fd, -1, 0, 1);
1146         if (waserror()) {
1147                 cclose(c);
1148                 nexterror();
1149         }
1150         d = chandirstat(c);
1151         poperror();
1152         cclose(c);
1153
1154         poperror();
1155         return d;
1156 }
1157
1158 int sysdirwstat(char *name, struct dir *dir)
1159 {
1160         
1161         uint8_t *buf;
1162         int r;
1163
1164         r = sizeD2M(dir);
1165         buf = kzmalloc(r, 0);
1166         convD2M(dir, buf, r);
1167         r = syswstat(name, buf, r);
1168         kfree(buf);
1169         return r < 0 ? r : 0;
1170 }
1171
1172 int sysdirfwstat(int fd, struct dir *dir)
1173 {
1174         
1175         uint8_t *buf;
1176         int r;
1177
1178         r = sizeD2M(dir);
1179         buf = kzmalloc(r, 0);
1180         convD2M(dir, buf, r);
1181         r = sysfwstat(fd, buf, r);
1182         kfree(buf);
1183         return r < 0 ? r : 0;
1184 }
1185
1186 static long dirpackage(uint8_t * buf, long ts, struct kdirent **d)
1187 {
1188         
1189         char *s;
1190         long ss, i, n, nn, m = 0;
1191
1192         *d = NULL;
1193         if (ts <= 0) {
1194                 return ts;
1195         }
1196
1197         /*
1198          * first find number of all stats, check they look like stats, & size all associated strings
1199          */
1200         ss = 0;
1201         n = 0;
1202         for (i = 0; i < ts; i += m) {
1203                 m = BIT16SZ + GBIT16(&buf[i]);
1204                 if (statcheck(&buf[i], m) < 0)
1205                         break;
1206                 ss += m;
1207                 n++;
1208         }
1209
1210         if (i != ts)
1211                 error("bad directory format");
1212
1213         *d = kzmalloc(n * sizeof(**d) + ss, 0);
1214         if (*d == NULL)
1215                 error(Enomem);
1216
1217         /*
1218          * then convert all buffers
1219          */
1220         s = (char *)*d + n * sizeof(**d);
1221         nn = 0;
1222         for (i = 0; i < ts; i += m) {
1223                 m = BIT16SZ + GBIT16((uint8_t *) & buf[i]);
1224                 if (nn >= n || /*convM2D */ convM2kdirent(&buf[i], m, *d + nn, s) != m) {
1225                         kfree(*d);
1226                         *d = NULL;
1227                         error("bad directory entry");
1228                 }
1229                 nn++;
1230                 s += m;
1231         }
1232
1233         return nn;
1234 }
1235
1236 long sysdirread(int fd, struct kdirent **d)
1237 {
1238         ERRSTACK(2);
1239         uint8_t *buf;
1240         long ts;
1241
1242         *d = NULL;
1243         if (waserror()) {
1244                 poperror();
1245                 return -1;
1246         }
1247         buf = kzmalloc(DIRREADLIM, 0);
1248         if (buf == NULL)
1249                 error(Enomem);
1250         if (waserror()) {
1251                 kfree(buf);
1252                 nexterror();
1253         }
1254         ts = sysread(fd, buf, DIRREADLIM);
1255         if (ts >= 0)
1256                 ts = dirpackage(buf, ts, d);
1257         poperror();
1258         kfree(buf);
1259         poperror();
1260         return ts;
1261 }
1262
1263 int sysiounit(int fd)
1264 {
1265         ERRSTACK(1);
1266         struct chan *c;
1267         int n;
1268
1269         c = fdtochan(&current->open_files, fd, -1, 0, 1);
1270         if (waserror()) {
1271                 cclose(c);
1272                 poperror();
1273                 return 0;       /* n.b. */
1274         }
1275         n = c->iounit;
1276         poperror();
1277         cclose(c);
1278         return n;
1279 }
1280
1281 void print_chaninfo(struct chan *c)
1282 {
1283         
1284         char buf[64] = { 0 };
1285         bool has_dev = c->type != -1;
1286         if (has_dev && !devtab[c->type].chaninfo) {
1287                 printk("Chan type %d has no chaninfo!\n", c->type);
1288                 has_dev = FALSE;
1289         }
1290         printk("Chan pathname: %s ref %d, Dev: %s, Devinfo: %s",
1291                    c->name ? c->name->s : "no cname",
1292                    kref_refcnt(&c->ref),
1293                    has_dev ? devtab[c->type].name : "no dev",
1294                    has_dev ? devtab[c->type].chaninfo(c, buf, sizeof(buf)) : "");
1295         if (!has_dev)
1296                 printk("qid.path: %p\n", c->qid.path);
1297         printk("\n");
1298 }
1299
1300 /* TODO: 9ns ns inheritance flags: Shared, copied, or empty.  The old fgrp is
1301  * managed by the fd_table, which is handled outside this function.  We share
1302  * the pgrp. */
1303 int plan9setup(struct proc *new_proc, struct proc *parent, int flags)
1304 {
1305         
1306         struct proc *old_current;
1307         struct kref *new_dot_ref;
1308         ERRSTACK(1);
1309         if (waserror()) {
1310                 printk("plan9setup failed, %s\n", current_errstr());
1311                 poperror();
1312                 return -1;
1313         }
1314         if (!parent) {
1315                 /* We are probably spawned by the kernel directly, and have no parent to
1316                  * inherit from.
1317                  *
1318                  * TODO: One problem is namec wants a current set for things like
1319                  * genbuf.  So we'll use new_proc for this bootstrapping.  Note
1320                  * switch_to() also loads the cr3. */
1321                 new_proc->pgrp = newpgrp();
1322                 old_current = switch_to(new_proc);
1323                 new_proc->slash = namec("#root", Atodir, 0, 0);
1324                 if (!new_proc->slash)
1325                         panic("no root device");
1326                 switch_back(new_proc, old_current);
1327                 /* Want the name to be "/" instead of "#root" */
1328                 cnameclose(new_proc->slash->name);
1329                 new_proc->slash->name = newcname("/");
1330                 new_proc->dot = cclone(new_proc->slash);
1331                 poperror();
1332                 return 0;
1333         }
1334         /* Shared semantics */
1335         kref_get(&parent->pgrp->ref, 1);
1336         new_proc->pgrp = parent->pgrp;
1337         /* copy semantics on / and . (doesn't make a lot of sense in akaros o/w) */
1338         /* / should never disappear while we hold a ref to parent */
1339         chan_incref(parent->slash);
1340         new_proc->slash = parent->slash;
1341         /* dot could change concurrently, and we could fail to gain a ref if whoever
1342          * decref'd dot triggered the release.  if that did happen, new_proc->dot
1343          * should update and we can try again. */
1344         while (!(new_dot_ref = kref_get_not_zero(&parent->dot->ref, 1)))
1345                 cpu_relax();
1346         /* And now, we can't trust parent->dot, and need to determine our dot from
1347          * the ref we obtained. */
1348         new_proc->dot = container_of(new_dot_ref, struct chan, ref);
1349         poperror();
1350         return 0;
1351 }
1352
1353 /* Open flags, create modes, access types, file flags, and all that...
1354  *
1355  * there are a bunch of things here:
1356  *              1) file creation flags (e.g. O_TRUNC)
1357  *              2) file status flags (e.g. O_APPEND)
1358  *              3) file open modes (e.g. O_RDWR)
1359  *              4) file descriptor flags (e.g. CLOEXEC)
1360  *              5) file creation mode (e.g. S_IRWXU)
1361  * the 1-4 are passed in via open's vfs_flags, and the 5 via mode only when
1362  * O_CREATE is set.
1363  *
1364  * file creation flags (1) only matter when creating, but aren't permanent.
1365  * O_EXCL, O_DIRECTORY, O_TRUNC, etc.
1366  *
1367  * file status flags (2) are per struct file/chan.  stuff like O_APPEND,
1368  * O_ASYNC, etc.  we convert those to an internal flag bit and store in c->flags
1369  *
1370  * the open mode (3) matters for a given FD/chan (chan->mode), and should be
1371  * stored in the chan. (c->mode) stuff like O_RDONLY.
1372  *
1373  * the file descriptor flags (4) clearly are in the FD.  note that the same
1374  * file/chan can be opened by two different FDs, with different flags.  the only
1375  * one anyone uses is CLOEXEC.  while exec may not last long in akaros, i can
1376  * imagine similar "never pass to children" flags/meanings.
1377  *
1378  * the file creation mode (5) matters for the device's permissions; given this,
1379  * it should be stored in the device/inode.  ACLs fall under this category.
1380  *
1381  * finally, only certain categories can be edited afterwards: file status flags
1382  * (2), FD flags (4), and file permissions (5). */
1383 int fd_getfl(int fd)
1384 {
1385         ERRSTACK(1);
1386         struct chan *c;
1387         int ret;
1388
1389         if (waserror()) {
1390                 poperror();
1391                 return -1;
1392         }
1393         c = fdtochan(&current->open_files, fd, -1, 0, 1);
1394
1395         ret = c->mode;
1396         ret |= c->flag & CEXTERNAL_FLAGS;
1397
1398         cclose(c);
1399         poperror();
1400         return ret;
1401 }
1402
1403 static bool cexternal_flags_differ(int set1, int set2, int flags)
1404 {
1405         flags &= CEXTERNAL_FLAGS;
1406         return (set1 & flags) ^ (set2 & flags);
1407 }
1408
1409 int fd_setfl(int fd, int flags)
1410 {
1411         ERRSTACK(1);
1412         struct chan *c;
1413
1414         if (waserror()) {
1415                 poperror();
1416                 return -1;
1417         }
1418         c = fdtochan(&current->open_files, fd, -1, 0, 1);
1419         if (cexternal_flags_differ(flags, c->flag, O_CLOEXEC)) {
1420                 /* TODO: The whole CCEXEC / O_CLOEXEC on 9ns needs work */
1421                 set_errno(EINVAL);
1422                 error("can't toggle O_CLOEXEC with setfl");
1423         }
1424         if (cexternal_flags_differ(flags, c->flag, O_PATH)) {
1425                 set_errno(EINVAL);
1426                 error("can't toggle O_PATH with setfl");
1427         }
1428         if (cexternal_flags_differ(flags, c->flag, O_NONBLOCK)) {
1429                 /* If we want to let them toggle NONBLOCK, it'd require a device op */
1430                 set_errno(EINVAL);
1431                 error("can't set O_NONBLOCK, use a device-specific ctl command");
1432         }
1433         c->flag = (c->flag & ~CEXTERNAL_FLAGS) | (flags & CEXTERNAL_FLAGS);
1434         cclose(c);
1435         poperror();
1436         return 0;
1437 }