xref: /netbsd-src/usr.bin/make/str.c (revision b1c86f5f087524e68db12794ee9c3e3da1ab17a0)
1 /*	$NetBSD: str.c,v 1.33 2009/02/25 21:17:21 sno Exp $	*/
2 
3 /*-
4  * Copyright (c) 1988, 1989, 1990, 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  * Adam de Boor.
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 
35 /*-
36  * Copyright (c) 1989 by Berkeley Softworks
37  * All rights reserved.
38  *
39  * This code is derived from software contributed to Berkeley by
40  * Adam de Boor.
41  *
42  * Redistribution and use in source and binary forms, with or without
43  * modification, are permitted provided that the following conditions
44  * are met:
45  * 1. Redistributions of source code must retain the above copyright
46  *    notice, this list of conditions and the following disclaimer.
47  * 2. Redistributions in binary form must reproduce the above copyright
48  *    notice, this list of conditions and the following disclaimer in the
49  *    documentation and/or other materials provided with the distribution.
50  * 3. All advertising materials mentioning features or use of this software
51  *    must display the following acknowledgement:
52  *	This product includes software developed by the University of
53  *	California, Berkeley and its contributors.
54  * 4. Neither the name of the University nor the names of its contributors
55  *    may be used to endorse or promote products derived from this software
56  *    without specific prior written permission.
57  *
58  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
59  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
60  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
61  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
62  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
63  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
64  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
65  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
66  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
67  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
68  * SUCH DAMAGE.
69  */
70 
71 #ifndef MAKE_NATIVE
72 static char rcsid[] = "$NetBSD: str.c,v 1.33 2009/02/25 21:17:21 sno Exp $";
73 #else
74 #include <sys/cdefs.h>
75 #ifndef lint
76 #if 0
77 static char     sccsid[] = "@(#)str.c	5.8 (Berkeley) 6/1/90";
78 #else
79 __RCSID("$NetBSD: str.c,v 1.33 2009/02/25 21:17:21 sno Exp $");
80 #endif
81 #endif				/* not lint */
82 #endif
83 
84 #include "make.h"
85 
86 /*-
87  * str_concat --
88  *	concatenate the two strings, inserting a space or slash between them,
89  *	freeing them if requested.
90  *
91  * returns --
92  *	the resulting string in allocated space.
93  */
94 char *
95 str_concat(const char *s1, const char *s2, int flags)
96 {
97 	int len1, len2;
98 	char *result;
99 
100 	/* get the length of both strings */
101 	len1 = strlen(s1);
102 	len2 = strlen(s2);
103 
104 	/* allocate length plus separator plus EOS */
105 	result = bmake_malloc((u_int)(len1 + len2 + 2));
106 
107 	/* copy first string into place */
108 	memcpy(result, s1, len1);
109 
110 	/* add separator character */
111 	if (flags & STR_ADDSPACE) {
112 		result[len1] = ' ';
113 		++len1;
114 	} else if (flags & STR_ADDSLASH) {
115 		result[len1] = '/';
116 		++len1;
117 	}
118 
119 	/* copy second string plus EOS into place */
120 	memcpy(result + len1, s2, len2 + 1);
121 
122 	return(result);
123 }
124 
125 /*-
126  * brk_string --
127  *	Fracture a string into an array of words (as delineated by tabs or
128  *	spaces) taking quotation marks into account.  Leading tabs/spaces
129  *	are ignored.
130  *
131  * If expand is TRUE, quotes are removed and escape sequences
132  *  such as \r, \t, etc... are expanded.
133  *
134  * returns --
135  *	Pointer to the array of pointers to the words.
136  *      Memory containing the actual words in *buffer.
137  *		Both of these must be free'd by the caller.
138  *      Number of words in *store_argc.
139  */
140 char **
141 brk_string(const char *str, int *store_argc, Boolean expand, char **buffer)
142 {
143 	int argc, ch;
144 	char inquote, *start, *t;
145 	const char *p;
146 	int len;
147 	int argmax = 50, curlen = 0;
148     	char **argv = bmake_malloc((argmax + 1) * sizeof(char *));
149 
150 	/* skip leading space chars. */
151 	for (; *str == ' ' || *str == '\t'; ++str)
152 		continue;
153 
154 	/* allocate room for a copy of the string */
155 	if ((len = strlen(str) + 1) > curlen)
156 		*buffer = bmake_malloc(curlen = len);
157 
158 	/*
159 	 * copy the string; at the same time, parse backslashes,
160 	 * quotes and build the argument list.
161 	 */
162 	argc = 0;
163 	inquote = '\0';
164 	for (p = str, start = t = *buffer;; ++p) {
165 		switch(ch = *p) {
166 		case '"':
167 		case '\'':
168 			if (inquote) {
169 				if (inquote == ch)
170 					inquote = '\0';
171 				else
172 					break;
173 			}
174 			else {
175 				inquote = (char) ch;
176 				/* Don't miss "" or '' */
177 				if (start == NULL && p[1] == inquote) {
178 					if (!expand) {
179 						start = t;
180 						*t++ = ch;
181 					} else
182 						start = t + 1;
183 					p++;
184 					inquote = '\0';
185 					break;
186 				}
187 			}
188 			if (!expand) {
189 				if (!start)
190 					start = t;
191 				*t++ = ch;
192 			}
193 			continue;
194 		case ' ':
195 		case '\t':
196 		case '\n':
197 			if (inquote)
198 				break;
199 			if (!start)
200 				continue;
201 			/* FALLTHROUGH */
202 		case '\0':
203 			/*
204 			 * end of a token -- make sure there's enough argv
205 			 * space and save off a pointer.
206 			 */
207 			if (!start)
208 			    goto done;
209 
210 			*t++ = '\0';
211 			if (argc == argmax) {
212 				argmax *= 2;		/* ramp up fast */
213 				argv = (char **)bmake_realloc(argv,
214 				    (argmax + 1) * sizeof(char *));
215 			}
216 			argv[argc++] = start;
217 			start = NULL;
218 			if (ch == '\n' || ch == '\0') {
219 				if (expand && inquote) {
220 					free(argv);
221 					free(*buffer);
222 					*buffer = NULL;
223 					return NULL;
224 				}
225 				goto done;
226 			}
227 			continue;
228 		case '\\':
229 			if (!expand) {
230 				if (!start)
231 					start = t;
232 				*t++ = '\\';
233 				if (*(p+1) == '\0') /* catch '\' at end of line */
234 					continue;
235 				ch = *++p;
236 				break;
237 			}
238 
239 			switch (ch = *++p) {
240 			case '\0':
241 			case '\n':
242 				/* hmmm; fix it up as best we can */
243 				ch = '\\';
244 				--p;
245 				break;
246 			case 'b':
247 				ch = '\b';
248 				break;
249 			case 'f':
250 				ch = '\f';
251 				break;
252 			case 'n':
253 				ch = '\n';
254 				break;
255 			case 'r':
256 				ch = '\r';
257 				break;
258 			case 't':
259 				ch = '\t';
260 				break;
261 			}
262 			break;
263 		}
264 		if (!start)
265 			start = t;
266 		*t++ = (char) ch;
267 	}
268 done:	argv[argc] = NULL;
269 	*store_argc = argc;
270 	return(argv);
271 }
272 
273 /*
274  * Str_FindSubstring -- See if a string contains a particular substring.
275  *
276  * Input:
277  *	string		String to search.
278  *	substring	Substring to find in string.
279  *
280  * Results: If string contains substring, the return value is the location of
281  * the first matching instance of substring in string.  If string doesn't
282  * contain substring, the return value is NULL.  Matching is done on an exact
283  * character-for-character basis with no wildcards or special characters.
284  *
285  * Side effects: None.
286  */
287 char *
288 Str_FindSubstring(const char *string, const char *substring)
289 {
290 	const char *a, *b;
291 
292 	/*
293 	 * First scan quickly through the two strings looking for a single-
294 	 * character match.  When it's found, then compare the rest of the
295 	 * substring.
296 	 */
297 
298 	for (b = substring; *string != 0; string += 1) {
299 		if (*string != *b)
300 			continue;
301 		a = string;
302 		for (;;) {
303 			if (*b == 0)
304 				return UNCONST(string);
305 			if (*a++ != *b++)
306 				break;
307 		}
308 		b = substring;
309 	}
310 	return NULL;
311 }
312 
313 /*
314  * Str_Match --
315  *
316  * See if a particular string matches a particular pattern.
317  *
318  * Results: Non-zero is returned if string matches pattern, 0 otherwise. The
319  * matching operation permits the following special characters in the
320  * pattern: *?\[] (see the man page for details on what these mean).
321  *
322  * Side effects: None.
323  */
324 int
325 Str_Match(const char *string, const char *pattern)
326 {
327 	char c2;
328 
329 	for (;;) {
330 		/*
331 		 * See if we're at the end of both the pattern and the
332 		 * string. If, we succeeded.  If we're at the end of the
333 		 * pattern but not at the end of the string, we failed.
334 		 */
335 		if (*pattern == 0)
336 			return(!*string);
337 		if (*string == 0 && *pattern != '*')
338 			return(0);
339 		/*
340 		 * Check for a "*" as the next pattern character.  It matches
341 		 * any substring.  We handle this by calling ourselves
342 		 * recursively for each postfix of string, until either we
343 		 * match or we reach the end of the string.
344 		 */
345 		if (*pattern == '*') {
346 			pattern += 1;
347 			if (*pattern == 0)
348 				return(1);
349 			while (*string != 0) {
350 				if (Str_Match(string, pattern))
351 					return(1);
352 				++string;
353 			}
354 			return(0);
355 		}
356 		/*
357 		 * Check for a "?" as the next pattern character.  It matches
358 		 * any single character.
359 		 */
360 		if (*pattern == '?')
361 			goto thisCharOK;
362 		/*
363 		 * Check for a "[" as the next pattern character.  It is
364 		 * followed by a list of characters that are acceptable, or
365 		 * by a range (two characters separated by "-").
366 		 */
367 		if (*pattern == '[') {
368 			++pattern;
369 			for (;;) {
370 				if ((*pattern == ']') || (*pattern == 0))
371 					return(0);
372 				if (*pattern == *string)
373 					break;
374 				if (pattern[1] == '-') {
375 					c2 = pattern[2];
376 					if (c2 == 0)
377 						return(0);
378 					if ((*pattern <= *string) &&
379 					    (c2 >= *string))
380 						break;
381 					if ((*pattern >= *string) &&
382 					    (c2 <= *string))
383 						break;
384 					pattern += 2;
385 				}
386 				++pattern;
387 			}
388 			while ((*pattern != ']') && (*pattern != 0))
389 				++pattern;
390 			goto thisCharOK;
391 		}
392 		/*
393 		 * If the next pattern character is '/', just strip off the
394 		 * '/' so we do exact matching on the character that follows.
395 		 */
396 		if (*pattern == '\\') {
397 			++pattern;
398 			if (*pattern == 0)
399 				return(0);
400 		}
401 		/*
402 		 * There's no special character.  Just make sure that the
403 		 * next characters of each string match.
404 		 */
405 		if (*pattern != *string)
406 			return(0);
407 thisCharOK:	++pattern;
408 		++string;
409 	}
410 }
411 
412 
413 /*-
414  *-----------------------------------------------------------------------
415  * Str_SYSVMatch --
416  *	Check word against pattern for a match (% is wild),
417  *
418  * Input:
419  *	word		Word to examine
420  *	pattern		Pattern to examine against
421  *	len		Number of characters to substitute
422  *
423  * Results:
424  *	Returns the beginning position of a match or null. The number
425  *	of characters matched is returned in len.
426  *
427  * Side Effects:
428  *	None
429  *
430  *-----------------------------------------------------------------------
431  */
432 char *
433 Str_SYSVMatch(const char *word, const char *pattern, int *len)
434 {
435     const char *p = pattern;
436     const char *w = word;
437     const char *m;
438 
439     if (*p == '\0') {
440 	/* Null pattern is the whole string */
441 	*len = strlen(w);
442 	return UNCONST(w);
443     }
444 
445     if ((m = strchr(p, '%')) != NULL) {
446 	/* check that the prefix matches */
447 	for (; p != m && *w && *w == *p; w++, p++)
448 	     continue;
449 
450 	if (p != m)
451 	    return NULL;	/* No match */
452 
453 	if (*++p == '\0') {
454 	    /* No more pattern, return the rest of the string */
455 	    *len = strlen(w);
456 	    return UNCONST(w);
457 	}
458     }
459 
460     m = w;
461 
462     /* Find a matching tail */
463     do
464 	if (strcmp(p, w) == 0) {
465 	    *len = w - m;
466 	    return UNCONST(m);
467 	}
468     while (*w++ != '\0');
469 
470     return NULL;
471 }
472 
473 
474 /*-
475  *-----------------------------------------------------------------------
476  * Str_SYSVSubst --
477  *	Substitute '%' on the pattern with len characters from src.
478  *	If the pattern does not contain a '%' prepend len characters
479  *	from src.
480  *
481  * Results:
482  *	None
483  *
484  * Side Effects:
485  *	Places result on buf
486  *
487  *-----------------------------------------------------------------------
488  */
489 void
490 Str_SYSVSubst(Buffer *buf, char *pat, char *src, int len)
491 {
492     char *m;
493 
494     if ((m = strchr(pat, '%')) != NULL) {
495 	/* Copy the prefix */
496 	Buf_AddBytes(buf, m - pat, pat);
497 	/* skip the % */
498 	pat = m + 1;
499     }
500 
501     /* Copy the pattern */
502     Buf_AddBytes(buf, len, src);
503 
504     /* append the rest */
505     Buf_AddBytes(buf, strlen(pat), pat);
506 }
507