look.c revision 269162
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: head/usr.bin/m4/look.c 269162 2014-07-27 22:54:13Z bapt $"); 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 54static void *hash_calloc(size_t, size_t, void *); 55static void hash_free(void *, void *); 56static void *element_alloc(size_t, void *); 57static void setup_definition(struct macro_definition *, const char *, 58 const char *); 59 60static struct ohash_info macro_info = { 61 offsetof(struct ndblock, name), 62 NULL, hash_calloc, hash_free, element_alloc }; 63 64struct ohash macros; 65 66/* Support routines for hash tables. */ 67void * 68hash_calloc(size_t n, size_t s, void *u __unused) 69{ 70 void *storage = xcalloc(n, s, "hash alloc"); 71 return storage; 72} 73 74void 75hash_free(void *p, void *u __unused) 76{ 77 free(p); 78} 79 80void * 81element_alloc(size_t s, void *u __unused) 82{ 83 return xalloc(s, "element alloc"); 84} 85 86void 87init_macros(void) 88{ 89 ohash_init(¯os, 10, ¯o_info); 90} 91 92/* 93 * find name in the hash table 94 */ 95ndptr 96lookup(const char *name) 97{ 98 return ohash_find(¯os, ohash_qlookup(¯os, name)); 99} 100 101struct macro_definition * 102lookup_macro_definition(const char *name) 103{ 104 ndptr p; 105 106 p = ohash_find(¯os, ohash_qlookup(¯os, name)); 107 if (p) 108 return p->d; 109 else 110 return NULL; 111} 112 113static void 114setup_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 133static ndptr 134create_entry(const char *name) 135{ 136 const char *end = NULL; 137 unsigned int i; 138 ndptr n; 139 140 i = ohash_qlookupi(¯os, name, &end); 141 n = ohash_find(¯os, i); 142 if (n == NULL) { 143 n = ohash_create_entry(¯o_info, name, &end); 144 ohash_insert(¯os, i, n); 145 n->trace_flags = FLAG_NO_TRACE; 146 n->builtin_type = MACRTYPE; 147 n->d = NULL; 148 } 149 return n; 150} 151 152void 153macro_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 166void 167macro_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 179void 180macro_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 196void 197macro_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 212void 213macro_for_all(void (*f)(const char *, struct macro_definition *)) 214{ 215 ndptr n; 216 unsigned int i; 217 218 for (n = ohash_first(¯os, &i); n != NULL; 219 n = ohash_next(¯os, &i)) 220 if (n->d != NULL) 221 f(n->name, n->d); 222} 223 224void 225setup_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 245void 246mark_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(¯os, &i); p != NULL; 257 p = ohash_next(¯os, &i)) 258 p->trace_flags = FLAG_NO_TRACE; 259 } else { 260 p = create_entry(name); 261 p->trace_flags = on; 262 } 263} 264 265ndptr 266macro_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} 276