1 /* flex - tool to generate fast lexical analyzers */
2
3 /* Copyright (c) 1990 The Regents of the University of California. */
4 /* All rights reserved. */
5
6 /* This code is derived from software contributed to Berkeley by */
7 /* Vern Paxson. */
8
9 /* The United States Government has rights in this work pursuant */
10 /* to contract no. DE-AC03-76SF00098 between the United States */
11 /* Department of Energy and the University of California. */
12
13 /* This file is part of flex. */
14
15 /* Redistribution and use in source and binary forms, with or without */
16 /* modification, are permitted provided that the following conditions */
17 /* are met: */
18
19 /* 1. Redistributions of source code must retain the above copyright */
20 /* notice, this list of conditions and the following disclaimer. */
21 /* 2. Redistributions in binary form must reproduce the above copyright */
22 /* notice, this list of conditions and the following disclaimer in the */
23 /* documentation and/or other materials provided with the distribution. */
24
25 /* Neither the name of the University nor the names of its contributors */
26 /* may be used to endorse or promote products derived from this software */
27 /* without specific prior written permission. */
28
29 /* THIS SOFTWARE IS PROVIDED ``AS IS'' AND WITHOUT ANY EXPRESS OR */
30 /* IMPLIED WARRANTIES, INCLUDING, WITHOUT LIMITATION, THE IMPLIED */
31 /* WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR */
32 /* PURPOSE. */
33
34 #include "flexdef.h"
35
36 /* Take note: The buffer object is sometimes used as a String buffer (one
37 * continuous string), and sometimes used as a list of strings, usually line by
38 * line.
39 *
40 * The type is specified in buf_init by the elt_size. If the elt_size is
41 * sizeof(char), then the buffer should be treated as string buffer. If the
42 * elt_size is sizeof(char*), then the buffer should be treated as a list of
43 * strings.
44 *
45 * Certain functions are only appropriate for one type or the other.
46 */
47
48 /* global buffers. */
49 struct Buf userdef_buf; /**< for user #definitions triggered by cmd-line. */
50 struct Buf defs_buf; /**< for #define's autogenerated. List of strings. */
51 struct Buf yydmap_buf; /**< string buffer to hold yydmap elements */
52 struct Buf m4defs_buf; /**< m4 definitions. List of strings. */
53 struct Buf top_buf; /**< contains %top code. String buffer. */
54
buf_print_strings(struct Buf * buf,FILE * out)55 struct Buf *buf_print_strings(struct Buf * buf, FILE* out)
56 {
57 int i;
58
59 if(!buf || !out)
60 return buf;
61
62 for (i=0; i < buf->nelts; i++){
63 const char * s = ((char**)buf->elts)[i];
64 if(s)
65 fprintf(out, "%s", s);
66 }
67 return buf;
68 }
69
70 /* Append a "%s" formatted string to a string buffer */
buf_prints(struct Buf * buf,const char * fmt,const char * s)71 struct Buf *buf_prints (struct Buf *buf, const char *fmt, const char *s)
72 {
73 char *t;
74 size_t tsz;
75
76 t = flex_alloc (tsz = strlen (fmt) + strlen (s) + 1);
77 if (!t)
78 flexfatal (_("Allocation of buffer to print string failed"));
79 snprintf (t, tsz, fmt, s);
80 buf = buf_strappend (buf, t);
81 flex_free (t);
82 return buf;
83 }
84
85 /** Append a line directive to the string buffer.
86 * @param buf A string buffer.
87 * @param filename file name
88 * @param lineno line number
89 * @return buf
90 */
buf_linedir(struct Buf * buf,const char * filename,int lineno)91 struct Buf *buf_linedir (struct Buf *buf, const char* filename, int lineno)
92 {
93 char *dst, *t;
94 const char *src;
95
96 t = flex_alloc (strlen ("#line \"\"\n") + /* constant parts */
97 2 * strlen (filename) + /* filename with possibly all backslashes escaped */
98 NUMCHARLINES + /* line number */
99 1); /* NUL */
100 if (!t)
101 flexfatal (_("Allocation of buffer for line directive failed"));
102 for (dst = t + sprintf (t, "#line %d \"", lineno), src = filename; *src; *dst++ = *src++)
103 if (*src == '\\') /* escape backslashes */
104 *dst++ = '\\';
105 *dst++ = '"';
106 *dst++ = '\n';
107 *dst = '\0';
108 buf = buf_strappend (buf, t);
109 flex_free (t);
110 return buf;
111 }
112
113
114 /** Append the contents of @a src to @a dest.
115 * @param @a dest the destination buffer
116 * @param @a dest the source buffer
117 * @return @a dest
118 */
buf_concat(struct Buf * dest,const struct Buf * src)119 struct Buf *buf_concat(struct Buf* dest, const struct Buf* src)
120 {
121 buf_append(dest, src->elts, src->nelts);
122 return dest;
123 }
124
125
126 /* Appends n characters in str to buf. */
buf_strnappend(buf,str,n)127 struct Buf *buf_strnappend (buf, str, n)
128 struct Buf *buf;
129 const char *str;
130 int n;
131 {
132 buf_append (buf, str, n + 1);
133
134 /* "undo" the '\0' character that buf_append() already copied. */
135 buf->nelts--;
136
137 return buf;
138 }
139
140 /* Appends characters in str to buf. */
buf_strappend(buf,str)141 struct Buf *buf_strappend (buf, str)
142 struct Buf *buf;
143 const char *str;
144 {
145 return buf_strnappend (buf, str, strlen (str));
146 }
147
148 /* appends "#define str def\n" */
buf_strdefine(buf,str,def)149 struct Buf *buf_strdefine (buf, str, def)
150 struct Buf *buf;
151 const char *str;
152 const char *def;
153 {
154 buf_strappend (buf, "#define ");
155 buf_strappend (buf, " ");
156 buf_strappend (buf, str);
157 buf_strappend (buf, " ");
158 buf_strappend (buf, def);
159 buf_strappend (buf, "\n");
160 return buf;
161 }
162
163 /** Pushes "m4_define( [[def]], [[val]])m4_dnl" to end of buffer.
164 * @param buf A buffer as a list of strings.
165 * @param def The m4 symbol to define.
166 * @param val The definition; may be NULL.
167 * @return buf
168 */
buf_m4_define(struct Buf * buf,const char * def,const char * val)169 struct Buf *buf_m4_define (struct Buf *buf, const char* def, const char* val)
170 {
171 const char * fmt = "m4_define( [[%s]], [[%s]])m4_dnl\n";
172 char * str;
173 size_t strsz;
174
175 val = val?val:"";
176 str = (char*)flex_alloc(strsz = strlen(fmt) + strlen(def) + strlen(val) + 2);
177 if (!str)
178 flexfatal (_("Allocation of buffer for m4 def failed"));
179
180 snprintf(str, strsz, fmt, def, val);
181 buf_append(buf, &str, 1);
182 return buf;
183 }
184
185 /** Pushes "m4_undefine([[def]])m4_dnl" to end of buffer.
186 * @param buf A buffer as a list of strings.
187 * @param def The m4 symbol to undefine.
188 * @return buf
189 */
buf_m4_undefine(struct Buf * buf,const char * def)190 struct Buf *buf_m4_undefine (struct Buf *buf, const char* def)
191 {
192 const char * fmt = "m4_undefine( [[%s]])m4_dnl\n";
193 char * str;
194 size_t strsz;
195
196 str = (char*)flex_alloc(strsz = strlen(fmt) + strlen(def) + 2);
197 if (!str)
198 flexfatal (_("Allocation of buffer for m4 undef failed"));
199
200 snprintf(str, strsz, fmt, def);
201 buf_append(buf, &str, 1);
202 return buf;
203 }
204
205 /* create buf with 0 elements, each of size elem_size. */
buf_init(buf,elem_size)206 void buf_init (buf, elem_size)
207 struct Buf *buf;
208 size_t elem_size;
209 {
210 buf->elts = (void *) 0;
211 buf->nelts = 0;
212 buf->elt_size = elem_size;
213 buf->nmax = 0;
214 }
215
216 /* frees memory */
buf_destroy(buf)217 void buf_destroy (buf)
218 struct Buf *buf;
219 {
220 if (buf && buf->elts)
221 flex_free (buf->elts);
222 buf->elts = (void *) 0;
223 }
224
225
226 /* appends ptr[] to buf, grow if necessary.
227 * n_elem is number of elements in ptr[], NOT bytes.
228 * returns buf.
229 * We grow by mod(512) boundaries.
230 */
231
buf_append(buf,ptr,n_elem)232 struct Buf *buf_append (buf, ptr, n_elem)
233 struct Buf *buf;
234 const void *ptr;
235 int n_elem;
236 {
237 int n_alloc = 0;
238
239 if (!ptr || n_elem == 0)
240 return buf;
241
242 /* May need to alloc more. */
243 if (n_elem + buf->nelts > buf->nmax) {
244
245 /* exact amount needed... */
246 n_alloc = (n_elem + buf->nelts) * buf->elt_size;
247
248 /* ...plus some extra */
249 if (((n_alloc * buf->elt_size) % 512) != 0
250 && buf->elt_size < 512)
251 n_alloc +=
252 (512 -
253 ((n_alloc * buf->elt_size) % 512)) /
254 buf->elt_size;
255
256 if (!buf->elts)
257 buf->elts =
258 allocate_array (n_alloc, buf->elt_size);
259 else
260 buf->elts =
261 reallocate_array (buf->elts, n_alloc,
262 buf->elt_size);
263
264 buf->nmax = n_alloc;
265 }
266
267 memcpy ((char *) buf->elts + buf->nelts * buf->elt_size, ptr,
268 n_elem * buf->elt_size);
269 buf->nelts += n_elem;
270
271 return buf;
272 }
273
274 /* vim:set tabstop=8 softtabstop=4 shiftwidth=4: */
275