usr.bin/rdist/lookup.c

   1 /*
   2  * Copyright (c) 1983, 1993
   3  *      The Regents of the University of California.  All rights reserved.
   4  *
   5  * Redistribution and use in source and binary forms, with or without
   6  * modification, are permitted provided that the following conditions
   7  * are met:
   8  * 1. Redistributions of source code must retain the above copyright
   9  *    notice, this list of conditions and the following disclaimer.
  10  * 2. Redistributions in binary form must reproduce the above copyright
  11  *    notice, this list of conditions and the following disclaimer in the
  12  *    documentation and/or other materials provided with the distribution.
  13  * 3. All advertising materials mentioning features or use of this software
  14  *    must display the following acknowledgement:
  15  *      This product includes software developed by the University of
  16  *      California, Berkeley and its contributors.
  17  * 4. Neither the name of the University nor the names of its contributors
  18  *    may be used to endorse or promote products derived from this software
  19  *    without specific prior written permission.
  20  *
  21  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
  22  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  23  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  24  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
  25  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  26  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  27  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  28  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  29  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  30  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  31  * SUCH DAMAGE.
  32  */
  33
  34 #ifndef lint
  35 #if 0
  36 static char sccsid[] = "@(#)lookup.c    8.1 (Berkeley) 6/9/93";
  37 #endif
  38 static const char rcsid[] =
  39   "$FreeBSD: src/usr.bin/rdist/lookup.c,v 1.6 1999/08/28 01:05:07 peter Exp $";
  40 #endif /* not lint */
  41
  42 #include "defs.h"
  43
  44         /* symbol types */
  45 #define VAR     1
  46 #define CONST   2
  47
  48 struct syment {
  49         int     s_type;
  50         char    *s_name;
  51         struct  namelist *s_value;
  52         struct  syment *s_next;
  53 };
  54
  55 static struct syment *hashtab[HASHSIZE];
  56
  57 /*
  58  * Define a variable from a command line argument.
  59  */
  60 void
  61 define(name)
  62         char *name;
  63 {
  64         register char *cp, *s;
  65         register struct namelist *nl;
  66         struct namelist *value = NULL;
  67
  68         if (debug)
  69                 printf("define(%s)\n", name);
  70
  71         cp = index(name, '=');
  72         if (cp == NULL)
  73                 value = NULL;
  74         else if (cp[1] == '\0') {
  75                 *cp = '\0';
  76                 value = NULL;
  77         } else if (cp[1] != '(') {
  78                 *cp++ = '\0';
  79                 value = makenl(cp);
  80         } else {
  81                 nl = NULL;
  82                 *cp++ = '\0';
  83                 do
  84                         cp++;
  85                 while (*cp == ' ' || *cp == '\t');
  86                 for (s = cp; ; s++) {
  87                         switch (*s) {
  88                         case ')':
  89                                 *s = '\0';
  90                         case '\0':
  91                                 break;
  92                         case ' ':
  93                         case '\t':
  94                                 *s++ = '\0';
  95                                 while (*s == ' ' || *s == '\t')
  96                                         s++;
  97                                 if (*s == ')')
  98                                         *s = '\0';
  99                                 break;
 100                         default:
 101                                 continue;
 102                         }
 103                         if (nl == NULL)
 104                                 value = nl = makenl(cp);
 105                         else {
 106                                 nl->n_next = makenl(cp);
 107                                 nl = nl->n_next;
 108                         }
 109                         if (*s == '\0')
 110                                 break;
 111                         cp = s;
 112                 }
 113         }
 114         (void) lookup(name, REPLACE, value);
 115 }
 116
 117 /*
 118  * Lookup name in the table and return a pointer to it.
 119  * LOOKUP - just do lookup, return NULL if not found.
 120  * INSERT - insert name with value, error if already defined.
 121  * REPLACE - insert or replace name with value.
 122  */
 123
 124 struct namelist *
 125 lookup(name, action, value)
 126         char *name;
 127         int action;
 128         struct namelist *value;
 129 {
 130         register unsigned n;
 131         register char *cp;
 132         register struct syment *s;
 133         char buf[BUFSIZ];
 134
 135         if (debug)
 136                 printf("lookup(%s, %d, %p)\n", name, action, value);
 137
 138         n = 0;
 139         for (cp = name; *cp; )
 140                 n += *cp++;
 141         n %= HASHSIZE;
 142
 143         for (s = hashtab[n]; s != NULL; s = s->s_next) {
 144                 if (strcmp(name, s->s_name))
 145                         continue;
 146                 if (action != LOOKUP) {
 147                         if (action != INSERT || s->s_type != CONST) {
 148                                 (void)snprintf(buf, sizeof(buf),
 149                                     "%s redefined", name);
 150                                 yyerror(buf);
 151                         }
 152                 }
 153                 return(s->s_value);
 154         }
 155
 156         if (action == LOOKUP) {
 157                 (void)snprintf(buf, sizeof(buf), "%s undefined", name);
 158                 yyerror(buf);
 159                 return(NULL);
 160         }
 161
 162         s = ALLOC(syment);
 163         if (s == NULL)
 164                 fatal("ran out of memory\n");
 165         s->s_next = hashtab[n];
 166         hashtab[n] = s;
 167         s->s_type = action == INSERT ? VAR : CONST;
 168         s->s_name = name;
 169         s->s_value = value;
 170         return(value);
 171 }