usr.bin/m4/look.c

   1 /*      $OpenBSD: look.c,v 1.22 2010/09/07 19:58:09 marco Exp $ */
   2
   3 /*
   4  * Copyright (c) 1989, 1993
   5  *      The Regents of the University of California.  All rights reserved.
   6  *
   7  * This code is derived from software contributed to Berkeley by
   8  * Ozan Yigit at York University.
   9  *
  10  * Redistribution and use in source and binary forms, with or without
  11  * modification, are permitted provided that the following conditions
  12  * are met:
  13  * 1. Redistributions of source code must retain the above copyright
  14  *    notice, this list of conditions and the following disclaimer.
  15  * 2. Redistributions in binary form must reproduce the above copyright
  16  *    notice, this list of conditions and the following disclaimer in the
  17  *    documentation and/or other materials provided with the distribution.
  18  * 3. Neither the name of the University nor the names of its contributors
  19  *    may be used to endorse or promote products derived from this software
  20  *    without specific prior written permission.
  21  *
  22  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
  23  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  24  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  25  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
  26  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  27  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  28  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  29  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  30  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  31  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  32  * SUCH DAMAGE.
  33  */
  34 #include <sys/cdefs.h>
  35 __FBSDID("$FreeBSD$");
  36
  37 /*
  38  * look.c
  39  * Facility: m4 macro processor
  40  * by: oz
  41  */
  42
  43 #include <sys/types.h>
  44 #include <stdio.h>
  45 #include <stdlib.h>
  46 #include <stdint.h>
  47 #include <stddef.h>
  48 #include <string.h>
  49 #include <ohash.h>
  50 #include "mdef.h"
  51 #include "stdd.h"
  52 #include "extern.h"
  53
  54 static void *hash_alloc(size_t, void *);
  55 static void hash_free(void *, size_t, void *);
  56 static void *element_alloc(size_t, void *);
  57 static void setup_definition(struct macro_definition *, const char *,
  58     const char *);
  59
  60 static struct ohash_info macro_info = {
  61         offsetof(struct ndblock, name),
  62         NULL, hash_alloc, hash_free, element_alloc };
  63
  64 struct ohash macros;
  65
  66 /* Support routines for hash tables.  */
  67 void *
  68 hash_alloc(size_t s, __unused void *u)
  69 {
  70         void *storage = xalloc(s, "hash alloc");
  71         if (storage)
  72                 memset(storage, 0, s);
  73         return storage;
  74 }
  75
  76 void
  77 hash_free(void *p, __unused size_t s, __unused void *u)
  78 {
  79         free(p);
  80 }
  81
  82 void *
  83 element_alloc(size_t s, __unused void *u)
  84 {
  85         return xalloc(s, "element alloc");
  86 }
  87
  88 void
  89 init_macros(void)
  90 {
  91         ohash_init(&macros, 10, &macro_info);
  92 }
  93
  94 /*
  95  * find name in the hash table
  96  */
  97 ndptr
  98 lookup(const char *name)
  99 {
 100         return ohash_find(&macros, ohash_qlookup(&macros, name));
 101 }
 102
 103 struct macro_definition *
 104 lookup_macro_definition(const char *name)
 105 {
 106         ndptr p;
 107
 108         p = ohash_find(&macros, ohash_qlookup(&macros, name));
 109         if (p)
 110                 return p->d;
 111         else
 112                 return NULL;
 113 }
 114
 115 static void
 116 setup_definition(struct macro_definition *d, const char *defn, const char *name)
 117 {
 118         ndptr p;
 119
 120         if (strncmp(defn, BUILTIN_MARKER, sizeof(BUILTIN_MARKER)-1) == 0 &&
 121             (p = macro_getbuiltin(defn+sizeof(BUILTIN_MARKER)-1)) != NULL) {
 122                 d->type = macro_builtin_type(p);
 123                 d->defn = xstrdup(defn+sizeof(BUILTIN_MARKER)-1);
 124         } else {
 125                 if (!*defn)
 126                         d->defn = __DECONST(char *, null);
 127                 else
 128                         d->defn = xstrdup(defn);
 129                 d->type = MACRTYPE;
 130         }
 131         if (STREQ(name, defn))
 132                 d->type |= RECDEF;
 133 }
 134
 135 static ndptr
 136 create_entry(const char *name)
 137 {
 138         const char *end = NULL;
 139         unsigned int i;
 140         ndptr n;
 141
 142         i = ohash_qlookupi(&macros, name, &end);
 143         n = ohash_find(&macros, i);
 144         if (n == NULL) {
 145                 n = ohash_create_entry(&macro_info, name, &end);
 146                 ohash_insert(&macros, i, n);
 147                 n->trace_flags = FLAG_NO_TRACE;
 148                 n->builtin_type = MACRTYPE;
 149                 n->d = NULL;
 150         }
 151         return n;
 152 }
 153
 154 void
 155 macro_define(const char *name, const char *defn)
 156 {
 157         ndptr n = create_entry(name);
 158         if (n->d != NULL) {
 159                 if (n->d->defn != null)
 160                         free(n->d->defn);
 161         } else {
 162                 n->d = xalloc(sizeof(struct macro_definition), NULL);
 163                 n->d->next = NULL;
 164         }
 165         setup_definition(n->d, defn, name);
 166 }
 167
 168 void
 169 macro_pushdef(const char *name, const char *defn)
 170 {
 171         ndptr n;
 172         struct macro_definition *d;
 173
 174         n = create_entry(name);
 175         d = xalloc(sizeof(struct macro_definition), NULL);
 176         d->next = n->d;
 177         n->d = d;
 178         setup_definition(n->d, defn, name);
 179 }
 180
 181 void
 182 macro_undefine(const char *name)
 183 {
 184         ndptr n = lookup(name);
 185         if (n != NULL) {
 186                 struct macro_definition *r, *r2;
 187
 188                 for (r = n->d; r != NULL; r = r2) {
 189                         r2 = r->next;
 190                         if (r->defn != null)
 191                                 free(r->defn);
 192                         free(r);
 193                 }
 194                 n->d = NULL;
 195         }
 196 }
 197
 198 void
 199 macro_popdef(const char *name)
 200 {
 201         ndptr n = lookup(name);
 202
 203         if (n != NULL) {
 204                 struct macro_definition *r = n->d;
 205                 if (r != NULL) {
 206                         n->d = r->next;
 207                         if (r->defn != null)
 208                                 free(r->defn);
 209                         free(r);
 210                 }
 211         }
 212 }
 213
 214 void
 215 macro_for_all(void (*f)(const char *, struct macro_definition *))
 216 {
 217         ndptr n;
 218         unsigned int i;
 219
 220         for (n = ohash_first(&macros, &i); n != NULL;
 221             n = ohash_next(&macros, &i))
 222                 if (n->d != NULL)
 223                         f(n->name, n->d);
 224 }
 225
 226 void
 227 setup_builtin(const char *name, unsigned int type)
 228 {
 229         ndptr n;
 230         char *name2;
 231
 232         if (prefix_builtins) {
 233                 name2 = xalloc(strlen(name)+3+1, NULL);
 234                 memcpy(name2, "m4_", 3);
 235                 memcpy(name2 + 3, name, strlen(name)+1);
 236         } else
 237                 name2 = xstrdup(name);
 238
 239         n = create_entry(name2);
 240         n->builtin_type = type;
 241         n->d = xalloc(sizeof(struct macro_definition), NULL);
 242         n->d->defn = name2;
 243         n->d->type = type;
 244         n->d->next = NULL;
 245 }
 246
 247 void
 248 mark_traced(const char *name, int on)
 249 {
 250         ndptr p;
 251         unsigned int i;
 252
 253         if (name == NULL) {
 254                 if (on)
 255                         trace_flags |= TRACE_ALL;
 256                 else
 257                         trace_flags &= ~TRACE_ALL;
 258                 for (p = ohash_first(&macros, &i); p != NULL;
 259                     p = ohash_next(&macros, &i))
 260                         p->trace_flags = FLAG_NO_TRACE;
 261         } else {
 262                 p = create_entry(name);
 263                 p->trace_flags = on;
 264         }
 265 }
 266
 267 ndptr
 268 macro_getbuiltin(const char *name)
 269 {
 270         ndptr p;
 271
 272         p = lookup(name);
 273         if (p == NULL || p->builtin_type == MACRTYPE)
 274                 return NULL;
 275         else
 276                 return p;
 277 }