Preemptively fixes walk malloc errors
[akaros.git] / kern / drivers / dev / dev.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 extern uint32_t kerndate;
17 extern char *eve;
18
19 void
20 mkqid(struct qid *q, int64_t path, uint32_t vers, int type)
21 {
22         q->type = type;
23         q->vers = vers;
24         q->path = path;
25 }
26
27 int
28 devno(int c, int user)
29 {
30         int i;
31
32         for(i = 0; devtab[i] != NULL; i++) {
33                 if(devtab[i]->dc == c)
34                         return i;
35         }
36         if(user == 0)
37                 panic("devno %C 0x%ux", c, c);
38
39         return -1;
40 }
41
42 void
43 devdir(struct chan *c, struct qid qid, char *n,
44        int64_t length, char *user, long perm, struct dir *db)
45 {
46         db->name = n;
47         if(c->flag&CMSG)
48                 qid.type |= QTMOUNT;
49         db->qid = qid;
50         db->type = devtab[c->type]->dc;
51         db->dev = c->dev;
52         db->mode = perm;
53         db->mode |= qid.type << 24;
54         db->atime = seconds();
55         db->mtime = kerndate;
56         db->length = length;
57         db->uid = user;
58         db->gid = eve;
59         db->muid = user;
60 }
61
62 /*
63  * the zeroth element of the table MUST be the directory itself for ..
64 */
65 int
66 devgen(struct chan *c, char *unused_char_p_t, struct dirtab *tab, int ntab, int i,
67        struct dir *dp)
68 {
69         if(tab == 0)
70                 return -1;
71         if(i != DEVDOTDOT){
72                 /* skip over the first element, that for . itself */
73                 i++;
74                 if(i >= ntab)
75                         return -1;
76                 tab += i;
77         }
78         devdir(c, tab->qid, tab->name, tab->length, eve, tab->perm, dp);
79         return 1;
80 }
81
82 void
83 devreset(void)
84 {
85 }
86
87 void
88 devinit(void)
89 {
90 }
91
92 void
93 devshutdown(void)
94 {
95 }
96
97 struct chan*
98 devattach(int tc, char *spec)
99 {
100         struct chan *c;
101         char *buf;
102
103         c = newchan();
104         mkqid(&c->qid, 0, 0, QTDIR);
105         c->type = devno(tc, 0);
106         if(spec == NULL)
107                 spec = "";
108         buf = kzmalloc(4 + strlen(spec) + 1, 0);
109         snprintf(buf, sizeof(buf), "#%C%s", tc, spec);
110         c->name = newcname(buf);
111         kfree(buf);
112         return c;
113 }
114
115
116 struct chan*
117 devclone(struct chan *c)
118 {
119         struct chan *nc;
120
121         if(c->flag & COPEN)
122                 panic("clone of open file type %C\n", devtab[c->type]->dc);
123
124         nc = newchan();
125
126         nc->type = c->type;
127         nc->dev = c->dev;
128         nc->mode = c->mode;
129         nc->qid = c->qid;
130         nc->offset = c->offset;
131         nc->umh = NULL;
132         nc->mountid = c->mountid;
133         nc->aux = c->aux;
134         nc->mqid = c->mqid;
135         nc->mcp = c->mcp;
136         return nc;
137 }
138
139 struct walkqid*
140 devwalk(struct chan *c,
141         struct chan *nc, char **name, int nname,
142         struct dirtab *tab, int ntab, Devgen *gen)
143 {
144         ERRSTACK(2);
145         int i, j;
146         volatile int alloc; /* to keep waserror from optimizing this out */
147         struct walkqid *wq;
148         char *n;
149         struct dir dir;
150
151         if(nname > 0)
152                 isdir(c);
153
154         alloc = 0;
155         wq = kzmalloc(sizeof(struct walkqid) + nname * sizeof(struct qid),
156                       KMALLOC_WAIT);
157         if(waserror()){
158                 if(alloc && wq->clone!=NULL)
159                         cclose(wq->clone);
160                 kfree(wq);
161                 return NULL;
162         }
163         if(nc == NULL){
164                 nc = devclone(c);
165                 nc->type = 0;   /* device doesn't know about this channel yet */
166                 alloc = 1;
167         }
168         wq->clone = nc;
169
170         for(j=0; j<nname; j++){
171                 if(!(nc->qid.type&QTDIR)){
172                         if(j==0)
173                                 error(Enotdir);
174                         goto Done;
175                 }
176                 n = name[j];
177                 if(strcmp(n, ".") == 0){
178     Accept:
179                         wq->qid[wq->nqid++] = nc->qid;
180                         continue;
181                 }
182                 if(strcmp(n, "..") == 0){
183                         (*gen)(nc, NULL, tab, ntab, DEVDOTDOT, &dir);
184                         nc->qid = dir.qid;
185                         goto Accept;
186                 }
187                 /*
188                  * Ugly problem: If we're using devgen, make sure we're
189                  * walking the directory itself, represented by the first
190                  * entry in the table, and not trying to step into a sub-
191                  * directory of the table, e.g. /net/net. Devgen itself
192                  * should take care of the problem, but it doesn't have
193                  * the necessary information (that we're doing a walk).
194                  */
195                 if(gen==devgen && nc->qid.path!=tab[0].qid.path)
196                         goto Notfound;
197                 for(i=0;; i++) {
198                         switch((*gen)(nc, n, tab, ntab, i, &dir)){
199                         case -1:
200                         Notfound:
201                                 if(j == 0)
202                                         error(Enonexist);
203                                 set_errstr(Enonexist);
204                                 goto Done;
205                         case 0:
206                                 continue;
207                         case 1:
208                                 if(strcmp(n, dir.name) == 0){
209                                         nc->qid = dir.qid;
210                                         goto Accept;
211                                 }
212                                 continue;
213                         }
214                 }
215         }
216         /*
217          * We processed at least one name, so will return some data.
218          * If we didn't process all nname entries succesfully, we drop
219          * the cloned channel and return just the Qids of the walks.
220          */
221 Done:
222         poperror();
223         if(wq->nqid < nname){
224                 if(alloc)
225                         cclose(wq->clone);
226                 wq->clone = NULL;
227         }else if(wq->clone){
228                 /* attach cloned channel to same device */
229                 wq->clone->type = c->type;
230         }
231         return wq;
232 }
233
234 int
235 devstat(struct chan *c, uint8_t *db, int n,
236         struct dirtab *tab, int ntab, Devgen *gen)
237 {
238         int i;
239         struct dir dir;
240         char *p, *elem;
241
242         for(i=0;; i++)
243                 switch((*gen)(c, NULL, tab, ntab, i, &dir)){
244                 case -1:
245                         if(c->qid.type & QTDIR){
246                                 if(c->name == NULL)
247                                         elem = "???";
248                                 else if(strcmp(c->name->s, "/") == 0)
249                                         elem = "/";
250                                 else
251                                         for(elem=p=c->name->s; *p; p++)
252                                                 if(*p == '/')
253                                                         elem = p+1;
254                                 devdir(c, c->qid, elem, 0, eve, DMDIR|0555, &dir);
255                                 n = convD2M(&dir, db, n);
256                                 if(n == 0)
257                                         error(Ebadarg);
258                                 return n;
259                         }
260                         printd("%s %s: devstat %C %llux\n",
261                                 up->text, up->env->user,
262                                 devtab[c->type]->dc, c->qid.path);
263
264                         error(Enonexist);
265                 case 0:
266                         break;
267                 case 1:
268                         if(c->qid.path == dir.qid.path) {
269                                 if(c->flag&CMSG)
270                                         dir.mode |= DMMOUNT;
271                                 n = convD2M(&dir, db, n);
272                                 if(n == 0)
273                                         error(Ebadarg);
274                                 return n;
275                         }
276                         break;
277                 }
278 }
279
280 long
281 devdirread(struct chan *c, char *d, long n,
282            struct dirtab *tab, int ntab, Devgen *gen)
283 {
284         long m, dsz;
285         /* this is gross. Make it 2 so we have room at the end for
286          * bad things.
287          */
288         struct dir dir[4];
289
290         for(m=0; m<n; c->dri++) {
291                 switch((*gen)(c, NULL, tab, ntab, c->dri, &dir[0])){
292                 case -1:
293                         return m;
294
295                 case 0:
296                         break;
297
298                 case 1:
299                         dsz = convD2M(&dir[0], ( uint8_t *)d, n-m);
300                         if(dsz <= BIT16SZ){     /* <= not < because this isn't stat; read is stuck */
301                                 if(m == 0)
302                                         error(Eshort);
303                                 return m;
304                         }
305                         m += dsz;
306                         d += dsz;
307                         break;
308                 }
309         }
310
311         return m;
312 }
313
314 /*
315  * error(Eperm) if open permission not granted for up->env->user.
316  */
317 void
318 devpermcheck(char *fileuid, uint32_t perm, int omode)
319 {
320         uint32_t t;
321         static int access[] = { 0400, 0200, 0600, 0100 };
322
323         if(strcmp(current->user, fileuid) == 0)
324                 perm <<= 0;
325         else
326         if(strcmp(current->user, eve) == 0)
327                 perm <<= 3;
328         else
329                 perm <<= 6;
330
331         t = access[omode&3];
332         if((t&perm) != t)
333                 error(Eperm);
334 }
335
336 struct chan*
337 devopen(struct chan *c, int omode, struct dirtab *tab, int ntab, Devgen *gen)
338 {
339         int i;
340         struct dir dir;
341
342         for(i=0;; i++) {
343                 switch((*gen)(c, NULL, tab, ntab, i, &dir)){
344                 case -1:
345                         goto Return;
346                 case 0:
347                         break;
348                 case 1:
349                         if(c->qid.path == dir.qid.path) {
350                                 devpermcheck(dir.uid, dir.mode, omode);
351                                 goto Return;
352                         }
353                         break;
354                 }
355         }
356 Return:
357         c->offset = 0;
358         if((c->qid.type&QTDIR) && omode!=OREAD)
359                 error(Eperm);
360         c->mode = openmode(omode);
361         c->flag |= COPEN;
362         return c;
363 }
364
365 void
366 devcreate(struct chan*c, char *unused_char_p_t, int unused_int, uint32_t u)
367 {
368         error(Eperm);
369 }
370
371 struct block*
372 devbread(struct chan *c, long n, uint32_t offset)
373 {
374         ERRSTACK(2);
375         struct block *bp;
376
377         bp = allocb(n);
378         if(bp == 0)
379                 error(Enomem);
380         if(waserror()) {
381                 freeb(bp);
382                 nexterror();
383         }
384         bp->wp += devtab[c->type]->read(c, bp->wp, n, offset);
385         poperror();
386         return bp;
387 }
388
389 long
390 devbwrite(struct chan *c, struct block *bp, uint32_t offset)
391 {
392         ERRSTACK(2);
393         long n;
394
395         if(waserror()) {
396                 freeb(bp);
397                 nexterror();
398         }
399         n = devtab[c->type]->write(c, bp->rp, BLEN(bp), offset);
400         poperror();
401         freeb(bp);
402
403         return n;
404 }
405
406 void
407 devremove(struct chan*c)
408 {
409         error(Eperm);
410 }
411
412 int
413 devwstat(struct chan*c, uint8_t *unused_uint8_p_t, int i)
414 {
415         error(Eperm);
416         return 0;
417 }
418
419 void
420 devpower(int i)
421 {
422         error(Eperm);
423 }
424
425 #if 0
426 int
427 devconfig( int unused_int, char *c, DevConf *)
428 {
429         error(Eperm);
430         return 0;
431 }
432 #endif
433 /*
434  * check that the name in a wstat is plausible
435  */
436 void
437 validwstatname(char *name)
438 {
439         validname(name, 0);
440         if(strcmp(name, ".") == 0 || strcmp(name, "..") == 0)
441                 error(Efilename);
442 }
443
444 struct dev*
445 devbyname(char *name)
446 {
447         int i;
448
449         for(i = 0; devtab[i] != NULL; i++)
450                 if(strcmp(devtab[i]->name, name) == 0)
451                         return devtab[i];
452         return NULL;
453 }
454
455