xref: /dragonfly/usr.bin/m4/look.c (revision 9b9cf381) 
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  * $FreeBSD: src/usr.bin/m4/look.c,v 1.10 2012/11/17 01:54:24 svnexp Exp $
35  */
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 }
278