Fixes krefs in chan.c
[akaros.git] / kern / src / ns / chan.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 <syscall.h>
15
16 char*
17 channame(struct chan *c)                /* DEBUGGING */
18 {
19         if(c == NULL)
20                 return "<NULL chan>";
21         if(c->name == NULL)
22                 return "<NULL name>";
23         if(c->name->s == NULL)
24                 return "<NULL name.s>";
25         return c->name->s;
26 }
27
28 enum
29 {
30         CNAMESLOP       = 20
31 };
32
33 struct
34 {
35         spinlock_t lock;
36         int     fid;
37         struct chan     *free;
38         struct chan     *list;
39 }chanalloc;
40
41 typedef struct Elemlist Elemlist;
42
43 struct Elemlist
44 {
45         char    *name;  /* copy of name, so '/' can be overwritten */
46         int     ARRAY_SIZEs;
47         char    **elems;
48         int     *off;
49         int     mustbedir;
50 };
51
52 #define SEP(c) ((c) == 0 || (c) == '/')
53 void cleancname(struct cname*);
54
55 int
56 isdotdot(char *p)
57 {
58         return p[0]=='.' && p[1]=='.' && p[2]=='\0';
59 }
60
61 int
62 emptystr(char *s)
63 {
64         if(s == NULL)
65                 return 1;
66         if(s[0] == '\0')
67                 return 1;
68         return 0;
69 }
70
71 /*
72  * Atomically replace *p with copy of s
73  */
74 void
75 kstrdup(char **p, char *s)
76 {
77         int n;
78         char *t, *prev;
79
80         n = strlen(s)+1;
81         /* if it's a user, we can wait for memory; if not, something's very wrong */
82         if (current) {
83                 t = kzmalloc(n, 0);
84         }else{
85                 t = kzmalloc(n, 0);
86                 if(t == NULL)
87                         panic("kstrdup: no memory");
88         }
89         memmove(t, s, n);
90         prev = *p;
91         *p = t;
92         kfree(prev);
93 }
94
95 void
96 chandevreset(void)
97 {
98         int i;
99
100         for(i=0; devtab[i] != NULL; i++)
101                 devtab[i]->reset();
102 }
103
104 void
105 chandevinit(void)
106 {
107         int i;
108
109         for(i=0; devtab[i] != NULL; i++)
110                 devtab[i]->init();
111 }
112
113 void
114 chandevshutdown(void)
115 {
116         int i;
117         
118         /* shutdown in reverse order */
119         for(i=0; devtab[i] != NULL; i++)
120                 ;
121         for(i--; i >= 0; i--)
122                 devtab[i]->shutdown();
123 }
124
125 static void chan_release(struct kref *kref)
126 {
127         struct chan *c = container_of(kref, struct chan, ref);
128         ERRSTACK(1);
129         /* this style discards the error from close().  picture it as
130          * if (waserror()) { } else { close(); } chanfree_no_matter_what();  */
131         if (!waserror()) {
132                 devtab[c->type]->close(c);
133         }
134         /* need to poperror regardless of whether we error'd or not */
135         poperror();
136         /* and chan free no matter what */
137         chanfree(c);
138 }
139
140 struct chan*
141 newchan(void)
142 {
143         struct chan *c;
144
145         spin_lock(&(&chanalloc)->lock);
146         c = chanalloc.free;
147         if(c != 0)
148                 chanalloc.free = c->next;
149         spin_unlock(&(&chanalloc)->lock);
150
151         if(c == NULL) {
152                 c = kzmalloc(sizeof(struct chan), 0);
153                 spin_lock(&(&chanalloc)->lock);
154                 c->fid = ++chanalloc.fid;
155                 c->link = chanalloc.list;
156                 chanalloc.list = c;
157                 spin_unlock(&(&chanalloc)->lock);
158         }
159
160         /* if you get an error before associating with a dev,
161            close calls rootclose, a nop */
162         c->type = 0;
163         c->flag = 0;
164         kref_init(&c->ref, chan_release, 1);
165         c->dev = 0;
166         c->offset = 0;
167         c->iounit = 0;
168         c->umh = 0;
169         c->uri = 0;
170         c->dri = 0;
171         c->aux = 0;
172         c->mchan = 0;
173         c->mcp = 0;
174         c->mux = 0;
175         c->mqid.path = 0;
176         c->mqid.vers = 0;
177         c->mqid.type = 0;
178         c->name = 0;
179         return c;
180 }
181
182 static void __cname_release(struct kref *kref)
183 {
184         struct cname *n = container_of(kref, struct cname, ref);
185         kfree(n->s);
186         kfree(n);
187 }
188
189 struct cname*
190 newcname(char *s)
191 {
192         struct cname *n;
193         int i;
194
195         n = kzmalloc(sizeof(*n), 0);
196         i = strlen(s);
197         n->len = i;
198         n->alen = i+CNAMESLOP;
199         n->s = kzmalloc(n->alen, 0);
200         memmove(n->s, s, i+1);
201         kref_init(&n->ref, __cname_release, 1);
202         return n;
203 }
204
205 void
206 cnameclose(struct cname *n)
207 {
208         if(n == NULL)
209                 return;
210         kref_put(&n->ref);
211 }
212
213 struct cname*
214 addelem(struct cname *n, char *s)
215 {
216         int i, a;
217         char *t;
218         struct cname *new;
219
220         if(s[0]=='.' && s[1]=='\0')
221                 return n;
222
223         if(kref_refcnt(&n->ref) > 1){
224                 /* copy on write */
225                 new = newcname(n->s);
226                 cnameclose(n);
227                 n = new;
228         }
229
230         i = strlen(s);
231         if(n->len+1+i+1 > n->alen){
232                 a = n->len+1+i+1 + CNAMESLOP;
233                 t = kzmalloc(a, 0);
234                 memmove(t, n->s, n->len+1);
235                 kfree(n->s);
236                 n->s = t;
237                 n->alen = a;
238         }
239         if(n->len>0 && n->s[n->len-1]!='/' && s[0]!='/')        /* don't insert extra slash if one is present */
240                 n->s[n->len++] = '/';
241         memmove(n->s+n->len, s, i+1);
242         n->len += i;
243         if(isdotdot(s))
244                 cleancname(n);
245         return n;
246 }
247
248 void
249 chanfree(struct chan *c)
250 {
251         c->flag = CFREE;
252
253         if(c->umh != NULL){
254                 putmhead(c->umh);
255                 c->umh = NULL;
256         }
257         if(c->umc != NULL){
258                 cclose(c->umc);
259                 c->umc = NULL;
260         }
261         if(c->mux != NULL){
262                 //
263                 muxclose(c->mux);
264                 c->mux = NULL;
265         }
266         if(c->mchan != NULL){
267                 cclose(c->mchan);
268                 c->mchan = NULL;
269         }
270
271         cnameclose(c->name);
272
273         spin_lock(&(&chanalloc)->lock);
274         c->next = chanalloc.free;
275         chanalloc.free = c;
276         spin_unlock(&(&chanalloc)->lock);
277 }
278
279 void
280 cclose(struct chan *c)
281 {
282         if(c == 0)
283                 return;
284
285         if(c->flag&CFREE)
286                 panic("cclose %lux", getcallerpc(&c));
287
288         kref_put(&c->ref);
289 }
290
291 /*
292  * Make sure we have the only copy of c.  (Copy on write.)
293  */
294 struct chan*
295 cunique(struct chan *c)
296 {
297         struct chan *nc;
298
299         if(kref_refcnt(&c->ref) != 1) {
300                 nc = cclone(c);
301                 cclose(c);
302                 c = nc;
303         }
304
305         return c;
306 }
307
308 int
309 eqqid(struct qid a, struct qid b)
310 {
311         return a.path==b.path && a.vers==b.vers;
312 }
313
314 int
315 eqchan(struct chan *a, struct chan *b, int pathonly)
316 {
317         if(a->qid.path != b->qid.path)
318                 return 0;
319         if(!pathonly && a->qid.vers!=b->qid.vers)
320                 return 0;
321         if(a->type != b->type)
322                 return 0;
323         if(a->dev != b->dev)
324                 return 0;
325         return 1;
326 }
327
328 int
329 eqchantdqid(struct chan *a, int type, int dev, struct qid qid, int pathonly)
330 {
331         if(a->qid.path != qid.path)
332                 return 0;
333         if(!pathonly && a->qid.vers!=qid.vers)
334                 return 0;
335         if(a->type != type)
336                 return 0;
337         if(a->dev != dev)
338                 return 0;
339         return 1;
340 }
341
342 static void mh_release(struct kref *kref)
343 {
344         struct mhead *mh = container_of(kref, struct mhead, ref);
345         mh->mount = (struct mount*)0xCafeBeef;
346         kfree(mh);
347 }
348
349 struct mhead*
350 newmhead(struct chan *from)
351 {
352         struct mhead *mh;
353
354         mh = kzmalloc(sizeof(struct mhead), 0);
355         kref_init(&mh->ref, mh_release, 1);
356         mh->from = from;
357         kref_get(&from->ref, 1);
358
359 /*
360         n = from->name->len;
361         if(n >= sizeof(mh->fromname))
362                 n = sizeof(mh->fromname)-1;
363         memmove(mh->fromname, from->name->s, n);
364         mh->fromname[n] = 0;
365 */
366         return mh;
367 }
368
369 int
370 cmount(struct chan *new, struct chan *old, int flag, char *spec)
371 {
372         ERRSTACK(2);
373         struct pgrp *pg;
374         int order, flg;
375         struct mhead *m, **l, *mh;
376         struct mount *nm, *f, *um, **h;
377
378         if(QTDIR & (old->qid.type^new->qid.type))
379                 error(Emount);
380
381 if(old->umh)
382         printd("cmount old extra umh\n");
383
384         order = flag&MORDER;
385
386         if((old->qid.type&QTDIR)==0 && order != MREPL)
387                 error(Emount);
388
389         mh = new->umh;
390
391         /*
392          * Not allowed to bind when the old directory
393          * is itself a union.  (Maybe it should be allowed, but I don't see
394          * what the semantics would be.)
395          *
396          * We need to check mh->mount->next to tell unions apart from
397          * simple mount points, so that things like
398          *      mount -c fd /root
399          *      bind -c /root /
400          * work.  The check of mount->mflag catches things like
401          *      mount fd /root
402          *      bind -c /root /
403          * 
404          * This is far more complicated than it should be, but I don't
405          * see an easier way at the moment.             -rsc
406          */
407         if((flag&MCREATE) && mh && mh->mount
408         && (mh->mount->next || !(mh->mount->mflag&MCREATE)))
409                 error(Emount);
410
411         pg = current->pgrp;
412         wlock(&pg->ns);
413
414         l = &MOUNTH(pg, old->qid);
415         for(m = *l; m; m = m->hash) {
416                 if(eqchan(m->from, old, 1))
417                         break;
418                 l = &m->hash;
419         }
420
421         if(m == NULL) {
422                 /*
423                  *  nothing mounted here yet.  create a mount
424                  *  head and add to the hash table.
425                  */
426                 m = newmhead(old);
427                 *l = m;
428
429                 /*
430                  *  if this is a union mount, add the old
431                  *  node to the mount chain.
432                  */
433                 if(order != MREPL)
434                         m->mount = newmount(m, old, 0, 0);
435         }
436         wlock(&m->lock);
437         if(waserror()){
438                 wunlock(&m->lock);
439                 nexterror();
440         }
441         wunlock(&pg->ns);
442
443         nm = newmount(m, new, flag, spec);
444         if(mh != NULL && mh->mount != NULL) {
445                 /*
446                  *  copy a union when binding it onto a directory
447                  */
448                 flg = order;
449                 if(order == MREPL)
450                         flg = MAFTER;
451                 h = &nm->next;
452                 um = mh->mount;
453                 for(um = um->next; um; um = um->next) {
454                         f = newmount(m, um->to, flg, um->spec);
455                         *h = f;
456                         h = &f->next;
457                 }
458         }
459
460         if(m->mount && order == MREPL) {
461                 mountfree(m->mount);
462                 m->mount = 0;
463         }
464
465         if(flag & MCREATE)
466                 nm->mflag |= MCREATE;
467
468         if(m->mount && order == MAFTER) {
469                 for(f = m->mount; f->next; f = f->next)
470                         ;
471                 f->next = nm;
472         }
473         else {
474                 for(f = nm; f->next; f = f->next)
475                         ;
476                 f->next = m->mount;
477                 m->mount = nm;
478         }
479
480         wunlock(&m->lock);
481         poperror();
482         return nm->mountid;
483 }
484
485 void
486 cunmount(struct chan *mnt, struct chan *mounted)
487 {
488         struct pgrp *pg;
489         struct mhead *m, **l;
490         struct mount *f, **p;
491
492         if(mnt->umh)    /* should not happen */
493                 printd("cunmount newp extra umh %p has %p\n", mnt, mnt->umh);
494
495         /*
496          * It _can_ happen that mounted->umh is non-NULL, 
497          * because mounted is the result of namec(Aopen)
498          * (see sysfile.c:/^sysunmount).
499          * If we open a union directory, it will have a umh.
500          * Although surprising, this is okay, since the
501          * cclose will take care of freeing the umh.
502          */
503
504         pg = current->pgrp;
505         wlock(&pg->ns);
506
507         l = &MOUNTH(pg, mnt->qid);
508         for(m = *l; m; m = m->hash) {
509                 if(eqchan(m->from, mnt, 1))
510                         break;
511                 l = &m->hash;
512         }
513
514         if(m == 0) {
515                 wunlock(&pg->ns);
516                 error(Eunmount);
517         }
518
519         wlock(&m->lock);
520         if(mounted == 0) {
521                 *l = m->hash;
522                 wunlock(&pg->ns);
523                 mountfree(m->mount);
524                 m->mount = NULL;
525                 cclose(m->from);
526                 wunlock(&m->lock);
527                 putmhead(m);
528                 return;
529         }
530
531         p = &m->mount;
532         for(f = *p; f; f = f->next) {
533                 /* BUG: Needs to be 2 pass */
534                 if(eqchan(f->to, mounted, 1) ||
535                   (f->to->mchan && eqchan(f->to->mchan, mounted, 1))) {
536                         *p = f->next;
537                         f->next = 0;
538                         mountfree(f);
539                         if(m->mount == NULL) {
540                                 *l = m->hash;
541                                 cclose(m->from);
542                                 wunlock(&m->lock);
543                                 wunlock(&pg->ns);
544                                 putmhead(m);
545                                 return;
546                         }
547                         wunlock(&m->lock);
548                         wunlock(&pg->ns);
549                         return;
550                 }
551                 p = &f->next;
552         }
553         wunlock(&m->lock);
554         wunlock(&pg->ns);
555         error(Eunion);
556 }
557
558 struct chan*
559 cclone(struct chan *c)
560 {
561         struct chan *nc;
562         struct walkqid *wq;
563
564         wq = devtab[c->type]->walk(c, NULL, NULL, 0);
565         if(wq == NULL)
566                 error("clone failed");
567         nc = wq->clone;
568         kfree(wq);
569         nc->name = c->name;
570         if(c->name)
571                 kref_get(&c->name->ref, 1);
572         return nc;
573 }
574
575 int
576 findmount(struct chan **cp,
577           struct mhead **mp, int type, int dev, struct qid qid)
578 {
579         struct pgrp *pg;
580         struct mhead *m;
581
582         pg = current->pgrp;
583         rlock(&pg->ns);
584         for(m = MOUNTH(pg, qid); m; m = m->hash){
585                 rlock(&m->lock);
586 if(m->from == NULL){
587         printd("m %p m->from 0\n", m);
588         runlock(&m->lock);
589         continue;
590 }
591                 if(eqchantdqid(m->from, type, dev, qid, 1)) {
592                         runlock(&pg->ns);
593                         if(mp != NULL){
594                                 kref_get(&m->ref, 1);
595                                 if(*mp != NULL)
596                                         putmhead(*mp);
597                                 *mp = m;
598                         }
599                         if(*cp != NULL)
600                                 cclose(*cp);
601                         kref_get(&m->mount->to->ref, 1);
602                         *cp = m->mount->to;
603                         runlock(&m->lock);
604                         return 1;
605                 }
606                 runlock(&m->lock);
607         }
608
609         runlock(&pg->ns);
610         return 0;
611 }
612
613 int
614 domount(struct chan **cp, struct mhead **mp)
615 {
616         return findmount(cp, mp, (*cp)->type, (*cp)->dev, (*cp)->qid);
617 }
618
619 struct chan*
620 undomount(struct chan *c, struct cname *name)
621 {
622         ERRSTACK(2);
623         struct chan *nc;
624         struct pgrp *pg;
625         struct mount *t;
626         struct mhead **h, **he, *f;
627
628         pg = current->pgrp;
629         rlock(&pg->ns);
630         if(waserror()) {
631                 runlock(&pg->ns);
632                 nexterror();
633         }
634
635         he = &pg->mnthash[MNTHASH];
636         for(h = pg->mnthash; h < he; h++) {
637                 for(f = *h; f; f = f->hash) {
638                         if(strcmp(f->from->name->s, name->s) != 0)
639                                 continue;
640                         for(t = f->mount; t; t = t->next) {
641                                 if(eqchan(c, t->to, 1)) {
642                                         /*
643                                          * We want to come out on the left hand side of the mount
644                                          * point using the element of the union that we entered on.
645                                          * To do this, find the element that has a from name of
646                                          * c->name->s.
647                                          */
648                                         if(strcmp(t->head->from->name->s, name->s) != 0)
649                                                 continue;
650                                         nc = t->head->from;
651                                         kref_get(&nc->ref, 1);
652                                         cclose(c);
653                                         c = nc;
654                                         break;
655                                 }
656                         }
657                 }
658         }
659         poperror();
660         runlock(&pg->ns);
661         return c;
662 }
663
664 /*
665  * Either walks all the way or not at all.  No partial results in *cp.
666  * *nerror is the number of names to display in an error message.
667  */
668 static char Edoesnotexist[] = "does not exist";
669 int
670 walk(struct chan **cp, char **names, int nnames, int nomount, int *nerror)
671 {
672         int dev, dotdot, i, n, nhave, ntry, type;
673         struct chan *c, *nc;
674         struct cname *cname;
675         struct mount *f;
676         struct mhead *mh, *nmh;
677         struct walkqid *wq;
678
679         c = *cp;
680         kref_get(&c->ref, 1);
681         cname = c->name;
682         kref_get(&cname->ref, 1);
683         mh = NULL;
684
685         /*
686          * While we haven't gotten all the way down the path:
687          *    1. step through a mount po int unused_int, if any
688          *    2. send a walk request for initial dotdot or initial prefix without dotdot
689          *    3. move to the first mountpoint along the way.
690          *    4. repeat.
691          *
692          * An invariant is that each time through the loop, c is on the undomount
693          * side of the mount po int unused_int, and c's name is cname.
694          */
695         for(nhave=0; nhave<nnames; nhave+=n){
696                 if((c->qid.type&QTDIR)==0){
697                         if(nerror)
698                                 *nerror = nhave;
699                         cnameclose(cname);
700                         cclose(c);
701                         set_errstr(Enotdir);
702                         if(mh != NULL)
703                                 putmhead(mh);
704                         return -1;
705                 }
706                 ntry = nnames - nhave;
707                 if(ntry > MAXWELEM)
708                         ntry = MAXWELEM;
709                 dotdot = 0;
710                 for(i=0; i<ntry; i++){
711                         if(isdotdot(names[nhave+i])){
712                                 if(i==0) {
713                                         dotdot = 1;
714                                         ntry = 1;
715                                 } else
716                                         ntry = i;
717                                 break;
718                         }
719                 }
720
721                 if(!dotdot && !nomount)
722                         domount(&c, &mh);
723
724                 type = c->type;
725                 dev = c->dev;
726
727                 if((wq = devtab[type]->walk(c, NULL, names+nhave, ntry)) == NULL){
728                         /* try a union mount, if any */
729                         if(mh && !nomount){
730                                 /*
731                                  * mh->mount == c, so start at mh->mount->next
732                                  */
733                                 rlock(&mh->lock);
734                                 for(f = mh->mount->next; f; f = f->next)
735                                         if((wq = devtab[f->to->type]->walk(f->to, NULL, names+nhave, ntry)) != NULL)
736                                                 break;
737                                 runlock(&mh->lock);
738                                 if(f != NULL){
739                                         type = f->to->type;
740                                         dev = f->to->dev;
741                                 }
742                         }
743                         if(wq == NULL){
744                                 cclose(c);
745                                 cnameclose(cname);
746                                 if(nerror)
747                                         *nerror = nhave+1;
748                                 if(mh != NULL)
749                                         putmhead(mh);
750                                 return -1;
751                         }
752                 }
753
754                 nmh = NULL;
755                 if(dotdot) {
756                         assert(wq->nqid == 1);
757                         assert(wq->clone != NULL);
758
759                         cname = addelem(cname, "..");
760                         nc = undomount(wq->clone, cname);
761                         n = 1;
762                 } else {
763                         nc = NULL;
764                         if(!nomount)
765                                 for(i=0; i<wq->nqid && i<ntry-1; i++)
766                                         if(findmount(&nc, &nmh, type, dev, wq->qid[i]))
767                                                 break;
768                         if(nc == NULL){ /* no mount points along path */
769                                 if(wq->clone == NULL){
770                                         cclose(c);
771                                         cnameclose(cname);
772                                         if(wq->nqid==0 || (wq->qid[wq->nqid-1].type&QTDIR)){
773                                                 if(nerror)
774                                                         *nerror = nhave+wq->nqid+1;
775                                                 set_errstr(Edoesnotexist);
776                                         }else{
777                                                 if(nerror)
778                                                         *nerror = nhave+wq->nqid;
779                                                 set_errstr(Enotdir);
780                                         }
781                                         kfree(wq);
782                                         if(mh != NULL)
783                                                 putmhead(mh);
784                                         return -1;
785                                 }
786                                 n = wq->nqid;
787                                 nc = wq->clone;
788                         }else{          /* stopped early, at a mount point */
789                                 if(wq->clone != NULL){
790                                         cclose(wq->clone);
791                                         wq->clone = NULL;
792                                 }
793                                 n = i+1;
794                         }
795                         for(i=0; i<n; i++)
796                                 cname = addelem(cname, names[nhave+i]);
797                 }
798                 cclose(c);
799                 c = nc;
800                 putmhead(mh);
801                 mh = nmh;
802                 kfree(wq);
803         }
804
805         putmhead(mh);
806
807         c = cunique(c);
808
809         if(c->umh != NULL){     //BUG
810                 printd("walk umh\n");
811                 putmhead(c->umh);
812                 c->umh = NULL;
813         }
814
815         cnameclose(c->name);
816         c->name = cname;
817
818         cclose(*cp);
819         *cp = c;
820         if(nerror)
821                 *nerror = 0;
822         return 0;
823 }
824
825 /*
826  * c is a mounted non-creatable directory.  find a creatable one.
827  */
828 struct chan*
829 createdir(struct chan *c, struct mhead *m)
830 {
831         ERRSTACK(2);
832         struct chan *nc;
833         struct mount *f;
834
835         rlock(&m->lock);
836         if(waserror()) {
837                 runlock(&m->lock);
838                 nexterror();
839         }
840         for(f = m->mount; f; f = f->next) {
841                 if(f->mflag&MCREATE) {
842                         nc = cclone(f->to);
843                         runlock(&m->lock);
844                         poperror();
845                         cclose(c);
846                         return nc;
847                 }
848         }
849         error(Enocreate);
850         return 0;
851 }
852
853 /*
854  * In place, rewrite name to compress multiple /, eliminate ., and process ..
855  */
856 void
857 cleancname(struct cname *n)
858 {
859         char *p;
860
861         if(n->s[0] == '#'){
862                 p = strchr(n->s, '/');
863                 if(p == NULL)
864                         return;
865                 cleanname(p);
866
867                 /*
868                  * The correct name is #i rather than #i/,
869                  * but the correct name of #/ is #/.
870                  */
871                 if(strcmp(p, "/")==0 && n->s[1] != '/')
872                         *p = '\0';
873         }else
874                 cleanname(n->s);
875         n->len = strlen(n->s);
876 }
877
878 static void
879 growparse(Elemlist *e)
880 {
881         char **new;
882         int *inew;
883         enum { Delta = 8 };
884
885         if(e->ARRAY_SIZEs % Delta == 0){
886                 new = kzmalloc((e->ARRAY_SIZEs + Delta) * sizeof(char *), 0);
887                 memmove(new, e->elems, e->ARRAY_SIZEs*sizeof( char *));
888                 kfree(e->elems);
889                 e->elems = new;
890                 inew = kzmalloc((e->ARRAY_SIZEs + Delta + 1) * sizeof(int), 0);
891                 memmove(inew, e->off, e->ARRAY_SIZEs*sizeof(int));
892                 kfree(e->off);
893                 e->off = inew;
894         }
895 }
896
897 /*
898  * The name is known to be valid.
899  * Copy the name so slashes can be overwritten.
900  * An empty string will set ARRAY_SIZE=0.
901  * A path ending in / or /. or /.//./ etc. will have
902  * e.mustbedir = 1, so that we correctly
903  * reject, e.g., "/adm/users/." when /adm/users is a file
904  * rather than a directory.
905  */
906 static void
907 parsename(char *name, Elemlist *e)
908 {
909         char *slash;
910
911         kstrdup(&e->name, name);
912         name = e->name;
913         e->ARRAY_SIZEs = 0;
914         e->elems = NULL;
915         e->off = kzmalloc(sizeof(int), 0);
916         e->off[0] = skipslash(name) - name;
917         for(;;){
918                 name = skipslash(name);
919                 if(*name=='\0'){
920                         e->mustbedir = 1;
921                         break;
922                 }
923                 growparse(e);
924                 
925                 e->elems[e->ARRAY_SIZEs++] = name;
926                 /* we may want to do this again some day
927                 slash = utfrune(name, '/');
928                 */
929                 slash = strchr(name, '/');
930                 if(slash == NULL){
931                         e->off[e->ARRAY_SIZEs] = name+strlen(name) - e->name;
932                         e->mustbedir = 0;
933                         break;
934                 }
935                 e->off[e->ARRAY_SIZEs] = slash - e->name;
936                 *slash++ = '\0';
937                 name = slash;
938         }
939 }
940
941 void*
942 memrchr(void *va, int c, long n)
943 {
944         uint8_t *a, *e;
945
946         a = va;
947         for(e=a+n-1; e>a; e--)
948                 if(*e == c)
949                         return e;
950         return NULL;
951 }
952
953 /*
954  * Turn a name into a channel.
955  * &name[0] is known to be a valid address.  It may be a kernel address.
956  *
957  * Opening with amode Aopen, Acreate, or Aremove guarantees
958  * that the result will be the only reference to that particular fid.
959  * This is necessary since we might pass the result to
960  * devtab[]->remove().
961  *
962  * Opening Atodir, Amount, or Aaccess does not guarantee this.
963  *
964  * Opening Aaccess can, under certain conditions, return a
965  * correct Chan* but with an incorrect struct cname attached.
966  * Since the functions that open Aaccess (sysstat, syswstat, sys_stat)
967  * do not use the struct cname*, this avoids an unnecessary clone.
968  */
969 struct chan*
970 namec(char *aname, int amode, int omode, uint32_t perm)
971 {
972         ERRSTACK(4);
973         int n, prefix, len, t, nomount, npath;
974         struct chan *c, *cnew;
975         struct cname *cname;
976         Elemlist e;
977         struct mhead *m;
978         char *createerr, tmperrbuf[ERRMAX];
979         char *name;
980         // Rune r;
981
982         name = aname;
983         if(name[0] == '\0')
984                 error("empty file name");
985         validname(name, 1);
986
987         /*
988          * Find the starting off point (the current slash, the root of
989          * a device tree, or the current dot) as well as the name to
990          * evaluate starting there.
991          */
992         nomount = 0;
993         switch(name[0]){
994         case '/':
995                 c = current->pgrp->slash;
996                 kref_get(&c->ref, 1);
997                 break;
998         
999         case '#':
1000                 nomount = 1;
1001                 get_cur_genbuf()[0] = '\0';
1002                 n = 0;
1003                 while(*name!='\0' && (*name != '/' || n < 2)){
1004                         if(n >= sizeof(get_cur_genbuf())-1)
1005                                 error(Efilename);
1006                         get_cur_genbuf()[n++] = *name++;
1007                 }
1008                 get_cur_genbuf()[n] = '\0';
1009 #if 0
1010                 n = chartorune(&r, get_cur_genbuf()+1)+1;
1011                 if(r == 'M')
1012                         error(Enoattach);
1013 #endif
1014                 if (get_cur_genbuf()[1] == 'M')
1015                         error(Enoattach);
1016                 /*
1017                  *  the nodevs exceptions are
1018                  *      |  it only gives access to pipes you create
1019                  *      e  this process's environment
1020                  *      s  private file2chan creation space
1021                  *      D private secure sockets name space
1022                  *      a private TLS name space
1023                  */
1024                 if(current->pgrp->nodevs &&
1025                    //              (utfrune("|esDa", r) == NULL
1026                    ((strchr("|esDa", get_cur_genbuf()[1]) == NULL)
1027                     || (get_cur_genbuf()[1] == 's' // || r == 's'
1028                         && get_cur_genbuf()[n]!='\0')))
1029                         error(Enoattach);
1030                 t = devno(/*r*/get_cur_genbuf()[1], 1);
1031                 if(t == -1)
1032                         error(Ebadsharp);
1033                 c = devtab[t]->attach(get_cur_genbuf()+n);
1034                 break;
1035
1036         default:
1037                 c = current->pgrp->dot;
1038                 kref_get(&c->ref, 1);
1039                 break;
1040         }
1041         prefix = name - aname;
1042
1043         e.name = NULL;
1044         e.elems = NULL;
1045         e.off = NULL;
1046         e.ARRAY_SIZEs = 0;
1047         if(waserror()){
1048                 cclose(c);
1049                 kfree(e.name);
1050                 kfree(e.elems);
1051                 kfree(e.off);
1052 //dumpmount();
1053                 nexterror();
1054         }
1055
1056         /*
1057          * Build a list of elements in the path.
1058          */
1059         parsename(name, &e);
1060
1061         /*
1062          * On create, ....
1063          */
1064         if(amode == Acreate){
1065                 /* perm must have DMDIR if last element is / or /. */
1066                 if(e.mustbedir && !(perm&DMDIR)){
1067                         npath = e.ARRAY_SIZEs;
1068                         strncpy(tmperrbuf,  "create without DMDIR", sizeof(tmperrbuf));
1069                         goto NameError;
1070                 }
1071
1072                 /* don't try to walk the last path element just yet. */
1073                 if(e.ARRAY_SIZEs == 0)
1074                         error(Eexist);
1075                 e.ARRAY_SIZEs--;
1076         }
1077
1078         if(walk(&c, e.elems, e.ARRAY_SIZEs, nomount, &npath) < 0){
1079                 if(npath < 0 || npath > e.ARRAY_SIZEs){
1080                         printd("namec %s walk error npath=%d\n", aname, npath);
1081                         nexterror();
1082                 }
1083 #warning "fix this mess with errstr and walking"
1084 #if 0
1085                 strncpy(tmperrbuf,  current->errstr, sizeof(tmperrbuf));
1086 #endif
1087         NameError:
1088 #if 0
1089                 len = prefix+e.off[npath];
1090                 if(len < ERRMAX/3 || (name=memrchr(aname, '/', len))==NULL || name==aname)
1091                         snprintf(get_cur_genbuf(), sizeof current->genbuf, "%.*s", len, aname);
1092                 else
1093                         snprintf(get_cur_genbuf(), sizeof current->genbuf, "...%.*s", (int)(len-(name-aname)), name);
1094                 snprintf(current->errstr, ERRMAX, "%#q %s", get_cur_genbuf(), tmperrbuf);
1095 #endif
1096                 error("some kinda name error");
1097                 nexterror();
1098         }
1099
1100         if(e.mustbedir && !(c->qid.type&QTDIR)){
1101                 npath = e.ARRAY_SIZEs;
1102                 strncpy(tmperrbuf,  "not a directory", sizeof(tmperrbuf));
1103                 goto NameError;
1104         }
1105
1106         if(amode == Aopen && (omode&3) == OEXEC && (c->qid.type&QTDIR)){
1107                 npath = e.ARRAY_SIZEs;
1108                 error("cannot exec directory");
1109         }
1110
1111         switch(amode){
1112         case Aaccess:
1113                 if(!nomount)
1114                         domount(&c, NULL);
1115                 break;
1116
1117         case Abind:
1118                 m = NULL;
1119                 if(!nomount)
1120                         domount(&c, &m);
1121                 if(c->umh != NULL)
1122                         putmhead(c->umh);
1123                 c->umh = m;
1124                 break;
1125
1126         case Aremove:
1127         case Aopen:
1128         Open:
1129                 /* save the name; domount might change c */
1130                 cname = c->name;
1131                 kref_get(&cname->ref, 1);
1132                 m = NULL;
1133                 if(!nomount)
1134                         domount(&c, &m);
1135
1136                 /* our own copy to open or remove */
1137                 c = cunique(c);
1138
1139                 /* now it's our copy anyway, we can put the name back */
1140                 cnameclose(c->name);
1141                 c->name = cname;
1142
1143                 switch(amode){
1144                 case Aremove:
1145                         putmhead(m);
1146                         break;
1147
1148                 case Aopen:
1149                 case Acreate:
1150 if(c->umh != NULL){
1151         printd("cunique umh\n");
1152         putmhead(c->umh);
1153         c->umh = NULL;
1154 }
1155
1156                         /* only save the mount head if it's a multiple element union */
1157                         if(m && m->mount && m->mount->next)
1158                                 c->umh = m;
1159                         else
1160                                 putmhead(m);
1161
1162                         if(omode == OEXEC)
1163                                 c->flag &= ~CCACHE;
1164
1165                         c = devtab[c->type]->open(c, omode&~OCEXEC);
1166
1167                         if(omode & OCEXEC)
1168                                 c->flag |= CCEXEC;
1169                         if(omode & ORCLOSE)
1170                                 c->flag |= CRCLOSE;
1171                         break;
1172                 }
1173                 break;
1174
1175         case Atodir:
1176                 /*
1177                  * Directories (e.g. for cd) are left before the mount po int unused_int,
1178                  * so one may mount on / or . and see the effect.
1179                  */
1180                 if(!(c->qid.type & QTDIR))
1181                         error(Enotdir);
1182                 break;
1183
1184         case Amount:
1185                 /*
1186                  * When mounting on an already mounted upon directory,
1187                  * one wants subsequent mounts to be attached to the
1188                  * original directory, not the replacement.  Don't domount.
1189                  */
1190                 break;
1191
1192         case Acreate:
1193                 /*
1194                  * We've already walked all but the last element.
1195                  * If the last exists, try to open it OTRUNC.
1196                  * If omode&OEXCL is set, just give up.
1197                  */
1198                 e.ARRAY_SIZEs++;
1199                 if(walk(&c, e.elems+e.ARRAY_SIZEs-1, 1, nomount, NULL) == 0){
1200                         if(omode&OEXCL)
1201                                 error(Eexist);
1202                         omode |= OTRUNC;
1203                         goto Open;
1204                 }
1205
1206                 /*
1207                  * The semantics of the create(2) system call are that if the
1208                  * file exists and can be written, it is to be opened with truncation.
1209                  * On the other hand, the create(5) message fails if the file exists.
1210                  * If we get two create(2) calls happening simultaneously, 
1211                  * they might both get here and send create(5) messages, but only 
1212                  * one of the messages will succeed.  To provide the expected create(2)
1213                  * semantics, the call with the failed message needs to try the above
1214                  * walk again, opening for truncation.  This correctly solves the 
1215                  * create/create race, in the sense that any observable outcome can
1216                  * be explained as one happening before the other.
1217                  * The create/create race is quite common.  For example, it happens
1218                  * when two rc subshells simultaneously update the same
1219                  * environment variable.
1220                  *
1221                  * The implementation still admits a create/create/remove race:
1222                  * (A) walk to file, fails
1223                  * (B) walk to file, fails
1224                  * (A) create file, succeeds, returns 
1225                  * (B) create file, fails
1226                  * (A) remove file, succeeds, returns
1227                  * (B) walk to file, return failure.
1228                  *
1229                  * This is hardly as common as the create/create race, and is really
1230                  * not too much worse than what might happen if (B) got a hold of a
1231                  * file descriptor and then the file was removed -- either way (B) can't do
1232                  * anything with the result of the create call.  So we don't care about this race.
1233                  *
1234                  * Applications that care about more fine-grained decision of the races
1235                  * can use the OEXCL flag to get at the underlying create(5) semantics;
1236                  * by default we provide the common case.
1237                  *
1238                  * We need to stay behind the mount point in case we
1239                  * need to do the first walk again (should the create fail).
1240                  *
1241                  * We also need to cross the mount point and find the directory
1242                  * in the union in which we should be creating.
1243                  *
1244                  * The channel staying behind is c, the one moving forward is cnew.
1245                  */
1246                 m = NULL;
1247                 cnew = NULL;    /* is this assignment necessary? */
1248                 if(!waserror()){        /* try create */
1249                         if(!nomount && findmount(&cnew, &m, c->type, c->dev, c->qid))
1250                                 cnew = createdir(cnew, m);
1251                         else{
1252                                 cnew = c;
1253                                 kref_get(&cnew->ref, 1);
1254                         }
1255
1256                         /*
1257                          * We need our own copy of the Chan because we're
1258                          * about to send a create, which will move it.  Once we have
1259                          * our own copy, we can fix the name, which might be wrong
1260                          * if findmount gave us a new Chan.
1261                          */
1262                         cnew = cunique(cnew);
1263                         cnameclose(cnew->name);
1264                         cnew->name = c->name;
1265                         kref_get(&cnew->name->ref, 1);
1266
1267                         devtab[cnew->type]->create(cnew, e.elems[e.ARRAY_SIZEs-1], omode&~(OEXCL|OCEXEC), perm);
1268                         poperror();
1269                         if(omode & OCEXEC)
1270                                 cnew->flag |= CCEXEC;
1271                         if(omode & ORCLOSE)
1272                                 cnew->flag |= CRCLOSE;
1273                         if(m)
1274                                 putmhead(m);
1275                         cclose(c);
1276                         c = cnew;
1277                         c->name = addelem(c->name, e.elems[e.ARRAY_SIZEs-1]);
1278                         break;
1279                 }
1280
1281                 /* create failed */
1282                 cclose(cnew);
1283                 if(m)
1284                         putmhead(m);
1285                 if(omode & OEXCL)
1286                         nexterror();
1287                 /* save error */
1288 #warning "more mess with errstr"
1289 #if 0
1290                 createerr = current->errstr;
1291                 current->errstr = tmperrbuf;
1292 #endif
1293                 /* note: we depend that walk does not error */
1294                 if(walk(&c, e.elems+e.ARRAY_SIZEs-1, 1, nomount, NULL) < 0){
1295                         error(createerr);       /* report true error */
1296                 }
1297 #if 0
1298                 set_errstr(createerr);
1299 #endif
1300                 omode |= OTRUNC;
1301                 goto Open;
1302
1303         default:
1304                 panic("unknown namec access %d\n", amode);
1305         }
1306
1307         poperror();
1308
1309         if(e.ARRAY_SIZEs > 0)
1310                 strncpy(get_cur_genbuf(), e.elems[e.ARRAY_SIZEs-1], GENBUF_SZ);
1311         else
1312                 strncpy(get_cur_genbuf(), ".", GENBUF_SZ);
1313
1314         kfree(e.name);
1315         kfree(e.elems);
1316         kfree(e.off);
1317
1318         return c;
1319 }
1320
1321 /*
1322  * name is valid. skip leading / and ./ as much as possible
1323  */
1324 char*
1325 skipslash(char *name)
1326 {
1327         while(name[0]=='/' || (name[0]=='.' && (name[1]==0 || name[1]=='/')))
1328                 name++;
1329         return name;
1330 }
1331
1332 char isfrog[256]={
1333         /*NUL*/ 1, 1, 1, 1, 1, 1, 1, 1,
1334         /*BKS*/ 1, 1, 1, 1, 1, 1, 1, 1,
1335         /*DLE*/ 1, 1, 1, 1, 1, 1, 1, 1,
1336         /*CAN*/ 1, 1, 1, 1, 1, 1, 1, 1,
1337         ['/']   1,
1338         [0x7f]  1,
1339 };
1340
1341 /*
1342  * Check that the name
1343  *  a) is in valid memory.
1344  *  b) is shorter than 2^16 bytes, so it can fit in a 9P string field.
1345  *  c) contains no frogs.
1346  * The first byte is known to be addressible by the requester, so the
1347  * routine works for kernel and user memory both.
1348  * The parameter slashok flags whether a slash character is an error
1349  * or a valid character.
1350  */
1351 void
1352 validname(char *aname, int slashok)
1353 {
1354         char *ename, *name;
1355         int c;
1356
1357         name = aname;
1358         ename = memchr(name, 0, (1<<16));
1359
1360         if(ename==NULL || ename-name>=(1<<16))
1361                 error("name too long");
1362
1363         while(*name){
1364                 /* all characters above '~' are ok */
1365                 c = *( uint8_t *)name;
1366 #if 0       
1367                 if(c >= Runeself)
1368                         name += chartorune(&r, name);
1369 #endif
1370                 if(c >= 0x7f){
1371                         error("Akaros doesn't do UTF-8");
1372                 } else {
1373                         if(isfrog[c])
1374                                 if(!slashok || c!='/'){
1375                                         error("%s: %q", Ebadchar, aname);
1376                         }
1377                         name++;
1378                 }
1379         }
1380 }
1381
1382 void
1383 isdir(struct chan *c)
1384 {
1385         if(c->qid.type & QTDIR)
1386                 return;
1387         error(Enotdir);
1388 }
1389
1390 /*
1391  * This is necessary because there are many
1392  * pointers to the top of a given mount list:
1393  *
1394  *      - the mhead in the namespace hash table
1395  *      - the mhead in chans returned from findmount:
1396  *        used in namec and then by unionread.
1397  *      - the mhead in chans returned from createdir:
1398  *        used in the open/create race protect, which is gone.
1399  *
1400  * The RWlock in the Mhead protects the mount list it contains.
1401  * The mount list is deleted when we cunmount.
1402  * The RWlock ensures that nothing is using the mount list at that time.
1403  *
1404  * It is okay to replace c->mh with whatever you want as 
1405  * long as you are sure you have a unique reference to it.
1406  *
1407  * This comment might belong somewhere else.
1408  */
1409 void
1410 putmhead(struct mhead *m)
1411 {
1412         if (m)
1413                 kref_put(&m->ref);
1414 }