usr.bin/m4/look.c

   1 /*      $OpenBSD: look.c,v 1.23 2014/05/12 19:11:19 espie 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_calloc(size_t, size_t, void *);
  55 static void hash_free(void *, 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_calloc, hash_free, element_alloc };
  63
  64 struct ohash macros;
  65
  66 /* Support routines for hash tables.  */
  67 void *
  68 hash_calloc(size_t n, size_t s, void *u __unused)
  69 {
  70         void *storage = xcalloc(n, s, "hash alloc");
  71         return storage;
  72 }
  73
  74 void
  75 hash_free(void *p, void *u __unused)
  76 {
  77         free(p);
  78 }
  79
  80 void *
  81 element_alloc(size_t s, void *u __unused)
  82 {
  83         return xalloc(s, "element alloc");
  84 }
  85
  86 void
  87 init_macros(void)
  88 {
  89         ohash_init(&macros, 10, &macro_info);
  90 }
  91
  92 /*
  93  * find name in the hash table
  94  */
  95 ndptr
  96 lookup(const char *name)
  97 {
  98         return ohash_find(&macros, ohash_qlookup(&macros, name));
  99 }
 100
 101 struct macro_definition *
 102 lookup_macro_definition(const char *name)
 103 {
 104         ndptr p;
 105
 106         p = ohash_find(&macros, ohash_qlookup(&macros, name));
 107         if (p)
 108                 return p->d;
 109         else
 110                 return NULL;
 111 }
 112
 113 static void
 114 setup_definition(struct macro_definition *d, const char *defn, const char *name)
 115 {
 116         ndptr p;
 117
 118         if (strncmp(defn, BUILTIN_MARKER, sizeof(BUILTIN_MARKER)-1) == 0 &&
 119             (p = macro_getbuiltin(defn+sizeof(BUILTIN_MARKER)-1)) != NULL) {
 120                 d->type = macro_builtin_type(p);
 121                 d->defn = xstrdup(defn+sizeof(BUILTIN_MARKER)-1);
 122         } else {
 123                 if (!*defn)
 124                         d->defn = __DECONST(char *, null);
 125                 else
 126                         d->defn = xstrdup(defn);
 127                 d->type = MACRTYPE;
 128         }
 129         if (STREQ(name, defn))
 130                 d->type |= RECDEF;
 131 }
 132
 133 static ndptr
 134 create_entry(const char *name)
 135 {
 136         const char *end = NULL;
 137         unsigned int i;
 138         ndptr n;
 139
 140         i = ohash_qlookupi(&macros, name, &end);
 141         n = ohash_find(&macros, i);
 142         if (n == NULL) {
 143                 n = ohash_create_entry(&macro_info, name, &end);
 144                 ohash_insert(&macros, i, n);
 145                 n->trace_flags = FLAG_NO_TRACE;
 146                 n->builtin_type = MACRTYPE;
 147                 n->d = NULL;
 148         }
 149         return n;
 150 }
 151
 152 void
 153 macro_define(const char *name, const char *defn)
 154 {
 155         ndptr n = create_entry(name);
 156         if (n->d != NULL) {
 157                 if (n->d->defn != null)
 158                         free(n->d->defn);
 159         } else {
 160                 n->d = xalloc(sizeof(struct macro_definition), NULL);
 161                 n->d->next = NULL;
 162         }
 163         setup_definition(n->d, defn, name);
 164 }
 165
 166 void
 167 macro_pushdef(const char *name, const char *defn)
 168 {
 169         ndptr n;
 170         struct macro_definition *d;
 171
 172         n = create_entry(name);
 173         d = xalloc(sizeof(struct macro_definition), NULL);
 174         d->next = n->d;
 175         n->d = d;
 176         setup_definition(n->d, defn, name);
 177 }
 178
 179 void
 180 macro_undefine(const char *name)
 181 {
 182         ndptr n = lookup(name);
 183         if (n != NULL) {
 184                 struct macro_definition *r, *r2;
 185
 186                 for (r = n->d; r != NULL; r = r2) {
 187                         r2 = r->next;
 188                         if (r->defn != null)
 189                                 free(r->defn);
 190                         free(r);
 191                 }
 192                 n->d = NULL;
 193         }
 194 }
 195
 196 void
 197 macro_popdef(const char *name)
 198 {
 199         ndptr n = lookup(name);
 200
 201         if (n != NULL) {
 202                 struct macro_definition *r = n->d;
 203                 if (r != NULL) {
 204                         n->d = r->next;
 205                         if (r->defn != null)
 206                                 free(r->defn);
 207                         free(r);
 208                 }
 209         }
 210 }
 211
 212 void
 213 macro_for_all(void (*f)(const char *, struct macro_definition *))
 214 {
 215         ndptr n;
 216         unsigned int i;
 217
 218         for (n = ohash_first(&macros, &i); n != NULL;
 219             n = ohash_next(&macros, &i))
 220                 if (n->d != NULL)
 221                         f(n->name, n->d);
 222 }
 223
 224 void
 225 setup_builtin(const char *name, unsigned int type)
 226 {
 227         ndptr n;
 228         char *name2;
 229
 230         if (prefix_builtins) {
 231                 name2 = xalloc(strlen(name)+3+1, NULL);
 232                 memcpy(name2, "m4_", 3);
 233                 memcpy(name2 + 3, name, strlen(name)+1);
 234         } else
 235                 name2 = xstrdup(name);
 236
 237         n = create_entry(name2);
 238         n->builtin_type = type;
 239         n->d = xalloc(sizeof(struct macro_definition), NULL);
 240         n->d->defn = name2;
 241         n->d->type = type;
 242         n->d->next = NULL;
 243 }
 244
 245 void
 246 mark_traced(const char *name, int on)
 247 {
 248         ndptr p;
 249         unsigned int i;
 250
 251         if (name == NULL) {
 252                 if (on)
 253                         trace_flags |= TRACE_ALL;
 254                 else
 255                         trace_flags &= ~TRACE_ALL;
 256                 for (p = ohash_first(&macros, &i); p != NULL;
 257                     p = ohash_next(&macros, &i))
 258                         p->trace_flags = FLAG_NO_TRACE;
 259         } else {
 260                 p = create_entry(name);
 261                 p->trace_flags = on;
 262         }
 263 }
 264
 265 ndptr
 266 macro_getbuiltin(const char *name)
 267 {
 268         ndptr p;
 269
 270         p = lookup(name);
 271         if (p == NULL || p->builtin_type == MACRTYPE)
 272                 return NULL;
 273         else
 274                 return p;
 275 }