kern/src/ns/parse.c

   1 /* Copyright © 1994-1999 Lucent Technologies Inc.  All rights reserved.
   2  * Portions Copyright © 1997-1999 Vita Nuova Limited
   3  * Portions Copyright © 2000-2007 Vita Nuova Holdings Limited
   4  *                                (www.vitanuova.com)
   5  * Revisions Copyright © 2000-2007 Lucent Technologies Inc. and others
   6  *
   7  * Modified for the Akaros operating system:
   8  * Copyright (c) 2013-2014 The Regents of the University of California
   9  * Copyright (c) 2013-2015 Google Inc.
  10  *
  11  * Permission is hereby granted, free of charge, to any person obtaining a copy
  12  * of this software and associated documentation files (the "Software"), to deal
  13  * in the Software without restriction, including without limitation the rights
  14  * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
  15  * copies of the Software, and to permit persons to whom the Software is
  16  * furnished to do so, subject to the following conditions:
  17  *
  18  * The above copyright notice and this permission notice shall be included in
  19  * all copies or substantial portions of the Software.
  20  *
  21  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  22  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  23  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL THE
  24  * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  25  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
  26  * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
  27  * SOFTWARE. */
  28
  29 #include <slab.h>
  30 #include <kmalloc.h>
  31 #include <kref.h>
  32 #include <string.h>
  33 #include <stdio.h>
  34 #include <assert.h>
  35 #include <error.h>
  36 #include <cpio.h>
  37 #include <pmap.h>
  38 #include <smp.h>
  39 #include <net/ip.h>
  40
  41 /*
  42  * Generous estimate of number of fields, including terminal NULL pointer
  43  */
  44 static int ncmdfield(char *p, int n)
  45 {
  46         int white, nwhite;
  47         char *ep;
  48         int nf;
  49
  50         if (p == NULL)
  51                 return 1;
  52
  53         nf = 0;
  54         ep = p + n;
  55         white = 1;      /* first text will start field */
  56         while (p < ep) {
  57                 nwhite = (strchr(" \t\r\n", *p++ & 0xFF) != 0); /* UTF is irrelevant */
  58                 if (white && !nwhite)   /* beginning of field */
  59                         nf++;
  60                 white = nwhite;
  61         }
  62         return nf + 1;  /* +1 for NULL */
  63 }
  64
  65 /*
  66  *  parse a command written to a device
  67  */
  68 struct cmdbuf *parsecmd(char *p, int n)
  69 {
  70         ERRSTACK(1);
  71         struct cmdbuf *volatile cb;
  72         int nf;
  73         char *sp;
  74
  75         nf = ncmdfield(p, n);
  76
  77         /* allocate Cmdbuf plus string pointers plus copy of string including \0 */
  78         sp = kzmalloc(sizeof(*cb) + nf * sizeof(char *) + n + 1, 0);
  79         cb = (struct cmdbuf *)sp;
  80         cb->f = (char **)(&cb[1]);
  81         cb->buf = (char *)(&cb->f[nf]);
  82
  83         if (current != NULL && waserror()) {
  84                 kfree(cb);
  85                 nexterror();
  86         }
  87         memmove(cb->buf, p, n);
  88         if (current != NULL)
  89                 poperror();
  90
  91         /* dump new line and null terminate */
  92         if (n > 0 && cb->buf[n - 1] == '\n')
  93                 n--;
  94         cb->buf[n] = '\0';
  95
  96         cb->nf = tokenize(cb->buf, cb->f, nf - 1);
  97         cb->f[cb->nf] = NULL;
  98
  99         return cb;
 100 }
 101
 102 /*
 103  * Reconstruct original message, for error diagnostic
 104  */
 105 void cmderror(struct cmdbuf *cb, char *s)
 106 {
 107         int i;
 108         char *p, *e;
 109
 110         p = get_cur_genbuf();
 111         e = p + GENBUF_SZ - 10;
 112         p = seprintf(p, e, "%s \"", s);
 113         for (i = 0; i < cb->nf; i++) {
 114                 if (i > 0)
 115                         p = seprintf(p, e, " ");
 116                 p = seprintf(p, e, "%s", cb->f[i]);
 117         }
 118         seprintf(p, e, "\"");
 119         error(EFAIL, get_cur_genbuf());
 120 }
 121
 122 void debugcmd(struct cmdbuf *cb)
 123 {
 124         printk("cb %p, nr %d\n", cb, cb->nf);
 125         for (int i = 0; i < cb->nf; i++) {
 126                 printk("%d: %s\n", i, cb->f[i]);
 127         }
 128 }
 129
 130 /*
 131  * Look up entry in table
 132  */
 133 struct cmdtab *lookupcmd(struct cmdbuf *cb, struct cmdtab *ctab, int nctab)
 134 {
 135         int i;
 136         struct cmdtab *ct;
 137
 138         if (cb->nf == 0)
 139                 error(EFAIL, "empty control message");
 140
 141         for (ct = ctab, i = 0; i < nctab; i++, ct++) {
 142                 if (strcmp(ct->cmd, "*") != 0)  /* wildcard always matches */
 143                         if (strcmp(ct->cmd, cb->f[0]) != 0)
 144                                 continue;
 145                 if (ct->narg != 0 && ct->narg != cb->nf)
 146                         cmderror(cb, "wrong number of args");
 147                 return ct;
 148         }
 149
 150         cmderror(cb, "unknown control message");
 151         return NULL;
 152 }