2b978659b3465c6720763e5727c8695f24e26ea9
[dragonfly.git] / usr.bin / make / var.c
1 /*-
2  * Copyright (c) 1988, 1989, 1990, 1993
3  *      The Regents of the University of California.  All rights reserved.
4  * Copyright (c) 1989 by Berkeley Softworks
5  * 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. All advertising materials mentioning features or use of this software
19  *    must display the following acknowledgement:
20  *      This product includes software developed by the University of
21  *      California, Berkeley and its contributors.
22  * 4. Neither the name of the University nor the names of its contributors
23  *    may be used to endorse or promote products derived from this software
24  *    without specific prior written permission.
25  *
26  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
27  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
28  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
29  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
30  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
31  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
32  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
33  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
34  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
35  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
36  * SUCH DAMAGE.
37  *
38  * @(#)var.c    8.3 (Berkeley) 3/19/94
39  * $FreeBSD: src/usr.bin/make/var.c,v 1.16.2.3 2002/02/27 14:18:57 cjc Exp $
40  * $DragonFly: src/usr.bin/make/var.c,v 1.60 2005/02/06 23:17:16 okumoto Exp $
41  */
42
43 /*-
44  * var.c --
45  *      Variable-handling functions
46  *
47  * Interface:
48  *      Var_Set             Set the value of a variable in the given
49  *                          context. The variable is created if it doesn't
50  *                          yet exist. The value and variable name need not
51  *                          be preserved.
52  *
53  *      Var_Append          Append more characters to an existing variable
54  *                          in the given context. The variable needn't
55  *                          exist already -- it will be created if it doesn't.
56  *                          A space is placed between the old value and the
57  *                          new one.
58  *
59  *      Var_Exists          See if a variable exists.
60  *
61  *      Var_Value           Return the value of a variable in a context or
62  *                          NULL if the variable is undefined.
63  *
64  *      Var_Subst           Substitute named variable, or all variables if
65  *                          NULL in a string using
66  *                          the given context as the top-most one. If the
67  *                          third argument is non-zero, Parse_Error is
68  *                          called if any variables are undefined.
69  *
70  *      Var_Parse           Parse a variable expansion from a string and
71  *                          return the result and the number of characters
72  *                          consumed.
73  *
74  *      Var_Delete          Delete a variable in a context.
75  *
76  *      Var_Init            Initialize this module.
77  *
78  * Debugging:
79  *      Var_Dump            Print out all variables defined in the given
80  *                          context.
81  *
82  * XXX: There's a lot of duplication in these functions.
83  */
84
85 #include <ctype.h>
86 #include <stdlib.h>
87 #include <string.h>
88
89 #include "buf.h"
90 #include "config.h"
91 #include "globals.h"
92 #include "GNode.h"
93 #include "make.h"
94 #include "nonints.h"
95 #include "parse.h"
96 #include "str.h"
97 #include "targ.h"
98 #include "util.h"
99 #include "var.h"
100
101 /*
102  * This is a harmless return value for Var_Parse that can be used by Var_Subst
103  * to determine if there was an error in parsing -- easier than returning
104  * a flag, as things outside this module don't give a hoot.
105  */
106 char    var_Error[] = "";
107
108 /*
109  * Similar to var_Error, but returned when the 'err' flag for Var_Parse is
110  * set false. Why not just use a constant? Well, gcc likes to condense
111  * identical string instances...
112  */
113 static char     varNoError[] = "";
114
115 /*
116  * Internally, variables are contained in four different contexts.
117  *      1) the environment. They may not be changed. If an environment
118  *          variable is appended-to, the result is placed in the global
119  *          context.
120  *      2) the global context. Variables set in the Makefile are located in
121  *          the global context. It is the penultimate context searched when
122  *          substituting.
123  *      3) the command-line context. All variables set on the command line
124  *         are placed in this context. They are UNALTERABLE once placed here.
125  *      4) the local context. Each target has associated with it a context
126  *         list. On this list are located the structures describing such
127  *         local variables as $(@) and $(*)
128  * The four contexts are searched in the reverse order from which they are
129  * listed.
130  */
131 GNode          *VAR_GLOBAL;   /* variables from the makefile */
132 GNode          *VAR_CMD;      /* variables defined on the command-line */
133
134 #define FIND_CMD        0x1   /* look in VAR_CMD when searching */
135 #define FIND_GLOBAL     0x2   /* look in VAR_GLOBAL as well */
136 #define FIND_ENV        0x4   /* look in the environment also */
137
138 #define OPEN_PAREN              '('
139 #define CLOSE_PAREN             ')'
140 #define OPEN_BRACKET            '{'
141 #define CLOSE_BRACKET           '}'
142
143 static Var *VarCreate(const char [], const char [], int);
144 static void VarDestroy(Var *, Boolean);
145 static char *VarGetPattern(GNode *, int, char **, int, int *, size_t *,
146                            VarPattern *);
147 static char *VarModify(char *, VarModifyProc *, void *);
148 static int VarPrintVar(void *, void *);
149
150 /*-
151  *-----------------------------------------------------------------------
152  * VarCmp  --
153  *      See if the given variable matches the named one. Called from
154  *      Lst_Find when searching for a variable of a given name.
155  *
156  * Results:
157  *      0 if they match. non-zero otherwise.
158  *
159  * Side Effects:
160  *      none
161  *-----------------------------------------------------------------------
162  */
163 static int
164 VarCmp(const void *v, const void *name)
165 {
166
167     return (strcmp(name, ((const Var *)v)->name));
168 }
169
170 /*-
171  *-----------------------------------------------------------------------
172  * VarPossiblyExpand --
173  *      Expand a variable name's embedded variables in the given context.
174  *
175  * Results:
176  *      The contents of name, possibly expanded.
177  *-----------------------------------------------------------------------
178  */
179 static char *
180 VarPossiblyExpand(const char *name, GNode *ctxt)
181 {
182         char   *tmp;
183
184         /*
185          * XXX make a temporary copy of the name because Var_Subst insists
186          * on writing into the string.
187          */
188         tmp = estrdup(name);
189         if (strchr(name, '$') != NULL) {
190                 Buffer  *buf;
191                 char    *str;
192
193                 buf = Var_Subst(NULL, tmp, ctxt, 0);
194                 str = Buf_GetAll(buf, NULL);
195                 Buf_Destroy(buf, FALSE);
196
197                 return(str);
198         } else {
199                 return(tmp);
200         }
201 }
202
203 /*-
204  *-----------------------------------------------------------------------
205  * VarFind --
206  *      Find the given variable in the given context and any other contexts
207  *      indicated.
208  *
209  *      Flags:
210  *              FIND_GLOBAL     set means look in the VAR_GLOBAL context too
211  *              FIND_CMD        set means to look in the VAR_CMD context too
212  *              FIND_ENV        set means to look in the environment
213  *
214  * Results:
215  *      A pointer to the structure describing the desired variable or
216  *      NULL if the variable does not exist.
217  *
218  * Side Effects:
219  *      None
220  *-----------------------------------------------------------------------
221  */
222 static Var *
223 VarFind(const char *name, GNode *ctxt, int flags)
224 {
225     Boolean             localCheckEnvFirst;
226     LstNode             *var;
227     Var                 *v;
228
229     /*
230      * If the variable name begins with a '.', it could very well be one of
231      * the local ones.  We check the name against all the local variables
232      * and substitute the short version in for 'name' if it matches one of
233      * them.
234      */
235     if (name[0] == '.' && isupper((unsigned char)name[1]))
236         switch (name[1]) {
237         case 'A':
238                 if (!strcmp(name, ".ALLSRC"))
239                         name = ALLSRC;
240                 if (!strcmp(name, ".ARCHIVE"))
241                         name = ARCHIVE;
242                 break;
243         case 'I':
244                 if (!strcmp(name, ".IMPSRC"))
245                         name = IMPSRC;
246                 break;
247         case 'M':
248                 if (!strcmp(name, ".MEMBER"))
249                         name = MEMBER;
250                 break;
251         case 'O':
252                 if (!strcmp(name, ".OODATE"))
253                         name = OODATE;
254                 break;
255         case 'P':
256                 if (!strcmp(name, ".PREFIX"))
257                         name = PREFIX;
258                 break;
259         case 'T':
260                 if (!strcmp(name, ".TARGET"))
261                         name = TARGET;
262                 break;
263         }
264
265     /*
266      * Note whether this is one of the specific variables we were told through
267      * the -E flag to use environment-variable-override for.
268      */
269     if (Lst_Find(&envFirstVars, name, (CompareProc *)strcmp) != NULL) {
270         localCheckEnvFirst = TRUE;
271     } else {
272         localCheckEnvFirst = FALSE;
273     }
274
275     /*
276      * First look for the variable in the given context. If it's not there,
277      * look for it in VAR_CMD, VAR_GLOBAL and the environment, in that order,
278      * depending on the FIND_* flags in 'flags'
279      */
280     var = Lst_Find(&ctxt->context, name, VarCmp);
281
282     if ((var == NULL) && (flags & FIND_CMD) && (ctxt != VAR_CMD)) {
283         var = Lst_Find(&VAR_CMD->context, name, VarCmp);
284     }
285     if ((var == NULL) && (flags & FIND_GLOBAL) && (ctxt != VAR_GLOBAL) &&
286         !checkEnvFirst && !localCheckEnvFirst)
287     {
288         var = Lst_Find(&VAR_GLOBAL->context, name, VarCmp);
289     }
290     if ((var == NULL) && (flags & FIND_ENV)) {
291         char *env;
292
293         if ((env = getenv(name)) != NULL) {
294             v = VarCreate(name, env, VAR_FROM_ENV);
295
296             return (v);
297         } else if ((checkEnvFirst || localCheckEnvFirst) &&
298                    (flags & FIND_GLOBAL) && (ctxt != VAR_GLOBAL))
299         {
300             var = Lst_Find(&VAR_GLOBAL->context, name, VarCmp);
301             if (var == NULL) {
302                 return (NULL);
303             } else {
304                 return (Lst_Datum(var));
305             }
306         } else {
307             return (NULL);
308         }
309     } else if (var == NULL) {
310         return (NULL);
311     } else {
312         return (Lst_Datum(var));
313     }
314 }
315
316 /*-
317  *-----------------------------------------------------------------------
318  * VarAdd  --
319  *      Add a new variable of name name and value val to the given context.
320  *
321  * Results:
322  *      None
323  *
324  * Side Effects:
325  *      The new variable is placed at the front of the given context
326  *      The name and val arguments are duplicated so they may
327  *      safely be freed.
328  *-----------------------------------------------------------------------
329  */
330 static void
331 VarAdd(const char *name, const char *val, GNode *ctxt)
332 {
333     Lst_AtFront(&ctxt->context, VarCreate(name, val, 0));
334
335     DEBUGF(VAR, ("%s:%s = %s\n", ctxt->name, name, val));
336 }
337
338 /*
339  * Create a Var object.
340  *
341  * @param name          Name of variable.
342  * @param value         Value of variable.
343  * @param flags         Flags set on variable.
344  */
345 static Var *
346 VarCreate(const char name[], const char value[], int flags)
347 {
348     Var *v;
349
350     v = emalloc(sizeof(Var));
351     v->name     = estrdup(name);
352     v->val      = Buf_Init(0);
353     v->flags    = flags;
354
355     if (value != NULL) {
356         Buf_Append(v->val, value);
357     }
358     return (v);
359 }
360
361 /*
362  * Destroy a Var object.
363  *
364  * @param v     Object to destroy.
365  * @param f     true if internal buffer in Buffer object is to be
366  *              removed.
367  */
368 static void
369 VarDestroy(Var *v, Boolean f)
370 {
371     Buf_Destroy(v->val, f);
372     free(v->name);
373     free(v);
374 }
375
376 /*-
377  *-----------------------------------------------------------------------
378  * Var_Delete --
379  *      Remove a variable from a context.
380  *
381  * Results:
382  *      None.
383  *
384  * Side Effects:
385  *      The Var structure is removed and freed.
386  *
387  *-----------------------------------------------------------------------
388  */
389 void
390 Var_Delete(const char *name, GNode *ctxt)
391 {
392     LstNode *ln;
393
394     DEBUGF(VAR, ("%s:delete %s\n", ctxt->name, name));
395     ln = Lst_Find(&ctxt->context, name, VarCmp);
396     if (ln != NULL) {
397         VarDestroy(Lst_Datum(ln), TRUE);
398         Lst_Remove(&ctxt->context, ln);
399     }
400 }
401
402 /*-
403  *-----------------------------------------------------------------------
404  * Var_Set --
405  *      Set the variable name to the value val in the given context.
406  *
407  * Results:
408  *      None.
409  *
410  * Side Effects:
411  *      If the variable doesn't yet exist, a new record is created for it.
412  *      Else the old value is freed and the new one stuck in its place
413  *
414  * Notes:
415  *      The variable is searched for only in its context before being
416  *      created in that context. I.e. if the context is VAR_GLOBAL,
417  *      only VAR_GLOBAL->context is searched. Likewise if it is VAR_CMD, only
418  *      VAR_CMD->context is searched. This is done to avoid the literally
419  *      thousands of unnecessary strcmp's that used to be done to
420  *      set, say, $(@) or $(<).
421  *-----------------------------------------------------------------------
422  */
423 void
424 Var_Set(const char *name, const char *val, GNode *ctxt)
425 {
426     Var         *v;
427     char        *n;
428
429     /*
430      * We only look for a variable in the given context since anything set
431      * here will override anything in a lower context, so there's not much
432      * point in searching them all just to save a bit of memory...
433      */
434     n = VarPossiblyExpand(name, ctxt);
435     v = VarFind(n, ctxt, 0);
436     if (v == NULL) {
437         VarAdd(n, val, ctxt);
438     } else {
439         Buf_Clear(v->val);
440         Buf_Append(v->val, val);
441
442         DEBUGF(VAR, ("%s:%s = %s\n", ctxt->name, n, val));
443     }
444     /*
445      * Any variables given on the command line are automatically exported
446      * to the environment (as per POSIX standard)
447      */
448     if (ctxt == VAR_CMD || (v != (Var *)NULL && (v->flags & VAR_TO_ENV))) {
449         setenv(n, val, 1);
450     }
451     free(n);
452 }
453
454 /*
455  * Var_SetEnv --
456  *      Set the VAR_TO_ENV flag on a variable
457  */
458 void
459 Var_SetEnv(const char *name, GNode *ctxt)
460 {
461     Var *v;
462
463     v = VarFind(name, ctxt, FIND_CMD|FIND_GLOBAL|FIND_ENV);
464     if (v) {
465         if ((v->flags & VAR_TO_ENV) == 0) {
466             v->flags |= VAR_TO_ENV;
467             setenv(v->name, Buf_GetAll(v->val, NULL), 1);
468         }
469     } else {
470         Error("Cannot set environment flag on non-existant variable %s", name);
471     }
472 }
473
474 /*-
475  *-----------------------------------------------------------------------
476  * Var_Append --
477  *      The variable of the given name has the given value appended to it in
478  *      the given context.
479  *
480  * Results:
481  *      None
482  *
483  * Side Effects:
484  *      If the variable doesn't exist, it is created. Else the strings
485  *      are concatenated (with a space in between).
486  *
487  * Notes:
488  *      Only if the variable is being sought in the global context is the
489  *      environment searched.
490  *      XXX: Knows its calling circumstances in that if called with ctxt
491  *      an actual target, it will only search that context since only
492  *      a local variable could be being appended to. This is actually
493  *      a big win and must be tolerated.
494  *-----------------------------------------------------------------------
495  */
496 void
497 Var_Append(const char *name, const char *val, GNode *ctxt)
498 {
499     Var         *v;
500     char        *n;
501
502     n = VarPossiblyExpand(name, ctxt);
503     v = VarFind(n, ctxt, (ctxt == VAR_GLOBAL) ? FIND_ENV : 0);
504
505     if (v == NULL) {
506         VarAdd(n, val, ctxt);
507     } else {
508         Buf_AddByte(v->val, (Byte)' ');
509         Buf_Append(v->val, val);
510
511         DEBUGF(VAR, ("%s:%s = %s\n", ctxt->name, n,
512                (char *)Buf_GetAll(v->val, (size_t *)NULL)));
513
514         if (v->flags & VAR_FROM_ENV) {
515             /*
516              * If the original variable came from the environment, we
517              * have to install it in the global context (we could place
518              * it in the environment, but then we should provide a way to
519              * export other variables...)
520              */
521             v->flags &= ~VAR_FROM_ENV;
522             Lst_AtFront(&ctxt->context, v);
523         }
524     }
525     free(n);
526 }
527
528 /*-
529  *-----------------------------------------------------------------------
530  * Var_Exists --
531  *      See if the given variable exists.
532  *
533  * Results:
534  *      TRUE if it does, FALSE if it doesn't
535  *
536  * Side Effects:
537  *      None.
538  *
539  *-----------------------------------------------------------------------
540  */
541 Boolean
542 Var_Exists(const char *name, GNode *ctxt)
543 {
544     Var         *v;
545     char        *n;
546
547     n = VarPossiblyExpand(name, ctxt);
548     v = VarFind(n, ctxt, FIND_CMD|FIND_GLOBAL|FIND_ENV);
549     free(n);
550
551     if (v == NULL) {
552         return (FALSE);
553     } else if (v->flags & VAR_FROM_ENV) {
554         VarDestroy(v, TRUE);
555     }
556     return (TRUE);
557 }
558
559 /*-
560  *-----------------------------------------------------------------------
561  * Var_Value --
562  *      Return the value of the named variable in the given context
563  *
564  * Results:
565  *      The value if the variable exists, NULL if it doesn't
566  *
567  * Side Effects:
568  *      None
569  *-----------------------------------------------------------------------
570  */
571 char *
572 Var_Value(const char *name, GNode *ctxt, char **frp)
573 {
574     Var         *v;
575     char        *n;
576
577     n = VarPossiblyExpand(name, ctxt);
578     v = VarFind(n, ctxt, FIND_ENV | FIND_GLOBAL | FIND_CMD);
579     free(n);
580     *frp = NULL;
581     if (v != NULL) {
582         char *p = (char *)Buf_GetAll(v->val, (size_t *)NULL);
583
584         if (v->flags & VAR_FROM_ENV) {
585             VarDestroy(v, FALSE);
586             *frp = p;
587         }
588         return (p);
589     } else {
590         return (NULL);
591     }
592 }
593
594 /*-
595  *-----------------------------------------------------------------------
596  * VarModify --
597  *      Modify each of the words of the passed string using the given
598  *      function. Used to implement all modifiers.
599  *
600  * Results:
601  *      A string of all the words modified appropriately.
602  *
603  * Side Effects:
604  *      None.
605  *
606  *-----------------------------------------------------------------------
607  */
608 static char *
609 VarModify(char *str, VarModifyProc *modProc, void *datum)
610 {
611     Buffer        *buf;             /* Buffer for the new string */
612     Boolean       addSpace;         /* TRUE if need to add a space to the
613                                      * buffer before adding the trimmed
614                                      * word */
615     char **av;                      /* word list [first word does not count] */
616     int ac, i;
617
618     buf = Buf_Init(0);
619     addSpace = FALSE;
620
621     av = brk_string(str, &ac, FALSE);
622
623     for (i = 1; i < ac; i++)
624         addSpace = (*modProc)(av[i], addSpace, buf, datum);
625
626     Buf_AddByte(buf, '\0');
627     str = (char *)Buf_GetAll(buf, (size_t *)NULL);
628     Buf_Destroy(buf, FALSE);
629     return (str);
630 }
631
632 /*-
633  *-----------------------------------------------------------------------
634  * VarSortWords --
635  *      Sort the words in the string.
636  *
637  * Input:
638  *      str             String whose words should be sorted
639  *      cmp             A comparison function to control the ordering
640  *
641  * Results:
642  *      A string containing the words sorted
643  *
644  * Side Effects:
645  *      None.
646  *
647  *-----------------------------------------------------------------------
648  */
649 static char *
650 VarSortWords(char *str, int (*cmp)(const void *, const void *))
651 {
652         Buffer *buf;
653         char **av;
654         int ac, i;
655
656         buf = Buf_Init(0);
657         av = brk_string(str, &ac, FALSE);
658         qsort(av + 1, ac - 1, sizeof(char *), cmp);
659         for (i = 1; i < ac; i++) {
660                 Buf_Append(buf, av[i]);
661                 Buf_AddByte(buf, (Byte)((i < ac - 1) ? ' ' : '\0'));
662         }
663         str = (char *)Buf_GetAll(buf, (size_t *)NULL);
664         Buf_Destroy(buf, FALSE);
665         return (str);
666 }
667
668 static int
669 SortIncreasing(const void *l, const void *r)
670 {
671
672         return (strcmp(*(const char* const*)l, *(const char* const*)r));
673 }
674
675 /*-
676  *-----------------------------------------------------------------------
677  * VarGetPattern --
678  *      Pass through the tstr looking for 1) escaped delimiters,
679  *      '$'s and backslashes (place the escaped character in
680  *      uninterpreted) and 2) unescaped $'s that aren't before
681  *      the delimiter (expand the variable substitution unless flags
682  *      has VAR_NOSUBST set).
683  *      Return the expanded string or NULL if the delimiter was missing
684  *      If pattern is specified, handle escaped ampersands, and replace
685  *      unescaped ampersands with the lhs of the pattern.
686  *
687  * Results:
688  *      A string of all the words modified appropriately.
689  *      If length is specified, return the string length of the buffer
690  *      If flags is specified and the last character of the pattern is a
691  *      $ set the VAR_MATCH_END bit of flags.
692  *
693  * Side Effects:
694  *      None.
695  *-----------------------------------------------------------------------
696  */
697 static char *
698 VarGetPattern(GNode *ctxt, int err, char **tstr, int delim, int *flags,
699     size_t *length, VarPattern *pattern)
700 {
701     char *cp;
702     Buffer *buf = Buf_Init(0);
703     size_t junk;
704
705     if (length == NULL)
706         length = &junk;
707
708 #define IS_A_MATCH(cp, delim) \
709     ((cp[0] == '\\') && ((cp[1] == delim) ||  \
710      (cp[1] == '\\') || (cp[1] == '$') || (pattern && (cp[1] == '&'))))
711
712     /*
713      * Skim through until the matching delimiter is found;
714      * pick up variable substitutions on the way. Also allow
715      * backslashes to quote the delimiter, $, and \, but don't
716      * touch other backslashes.
717      */
718     for (cp = *tstr; *cp && (*cp != delim); cp++) {
719         if (IS_A_MATCH(cp, delim)) {
720             Buf_AddByte(buf, (Byte)cp[1]);
721             cp++;
722         } else if (*cp == '$') {
723             if (cp[1] == delim) {
724                 if (flags == NULL)
725                     Buf_AddByte(buf, (Byte)*cp);
726                 else
727                     /*
728                      * Unescaped $ at end of pattern => anchor
729                      * pattern at end.
730                      */
731                     *flags |= VAR_MATCH_END;
732             } else {
733                 if (flags == NULL || (*flags & VAR_NOSUBST) == 0) {
734                     char   *cp2;
735                     size_t len;
736                     Boolean freeIt;
737
738                     /*
739                      * If unescaped dollar sign not before the
740                      * delimiter, assume it's a variable
741                      * substitution and recurse.
742                      */
743                     cp2 = Var_Parse(cp, ctxt, err, &len, &freeIt);
744                     Buf_Append(buf, cp2);
745                     if (freeIt)
746                         free(cp2);
747                     cp += len - 1;
748                 } else {
749                     char *cp2 = &cp[1];
750
751                     if (*cp2 == OPEN_PAREN || *cp2 == OPEN_BRACKET) {
752                         /*
753                          * Find the end of this variable reference
754                          * and suck it in without further ado.
755                          * It will be interperated later.
756                          */
757                         int have = *cp2;
758                         int want = (*cp2 == OPEN_PAREN) ? CLOSE_PAREN : CLOSE_BRACKET;
759                         int depth = 1;
760
761                         for (++cp2; *cp2 != '\0' && depth > 0; ++cp2) {
762                             if (cp2[-1] != '\\') {
763                                 if (*cp2 == have)
764                                     ++depth;
765                                 if (*cp2 == want)
766                                     --depth;
767                             }
768                         }
769                         Buf_AppendRange(buf, cp, cp2);
770                         cp = --cp2;
771                     } else
772                         Buf_AddByte(buf, (Byte)*cp);
773                 }
774             }
775         }
776         else if (pattern && *cp == '&')
777             Buf_AddBytes(buf, pattern->leftLen, (Byte *)pattern->lhs);
778         else
779             Buf_AddByte(buf, (Byte)*cp);
780     }
781
782     Buf_AddByte(buf, (Byte)'\0');
783
784     if (*cp != delim) {
785         *tstr = cp;
786         *length = 0;
787         return (NULL);
788     } else {
789         *tstr = ++cp;
790         cp = (char *)Buf_GetAll(buf, length);
791         *length -= 1;   /* Don't count the NULL */
792         Buf_Destroy(buf, FALSE);
793         return (cp);
794     }
795 }
796
797 /*-
798  *-----------------------------------------------------------------------
799  * Var_Quote --
800  *      Quote shell meta-characters in the string
801  *
802  * Results:
803  *      The quoted string
804  *
805  * Side Effects:
806  *      None.
807  *
808  *-----------------------------------------------------------------------
809  */
810 char *
811 Var_Quote(const char *str)
812 {
813     Buffer       *buf;
814     /* This should cover most shells :-( */
815     static char meta[] = "\n \t'`\";&<>()|*?{}[]\\$!#^~";
816     char          *ret;
817
818     buf = Buf_Init(MAKE_BSIZE);
819     for (; *str; str++) {
820         if (strchr(meta, *str) != NULL)
821             Buf_AddByte(buf, (Byte)'\\');
822         Buf_AddByte(buf, (Byte)*str);
823     }
824     Buf_AddByte(buf, (Byte)'\0');
825     ret = Buf_GetAll(buf, NULL);
826     Buf_Destroy(buf, FALSE);
827     return (ret);
828 }
829
830 /*-
831  *-----------------------------------------------------------------------
832  * VarREError --
833  *      Print the error caused by a regcomp or regexec call.
834  *
835  * Results:
836  *      None.
837  *
838  * Side Effects:
839  *      An error gets printed.
840  *
841  *-----------------------------------------------------------------------
842  */
843 void
844 VarREError(int err, regex_t *pat, const char *str)
845 {
846     char *errbuf;
847     int errlen;
848
849     errlen = regerror(err, pat, 0, 0);
850     errbuf = emalloc(errlen);
851     regerror(err, pat, errbuf, errlen);
852     Error("%s: %s", str, errbuf);
853     free(errbuf);
854 }
855
856 /*-
857  *-----------------------------------------------------------------------
858  * Var_Parse --
859  *      Given the start of a variable invocation, extract the variable
860  *      name and find its value, then modify it according to the
861  *      specification.
862  *
863  * Results:
864  *      The (possibly-modified) value of the variable or var_Error if the
865  *      specification is invalid. The length of the specification is
866  *      placed in *lengthPtr (for invalid specifications, this is just
867  *      2 to skip the '$' and the following letter, or 1 if '$' was the
868  *      last character in the string).
869  *      A Boolean in *freePtr telling whether the returned string should
870  *      be freed by the caller.
871  *
872  * Side Effects:
873  *      None.
874  *
875  *-----------------------------------------------------------------------
876  */
877 char *
878 Var_Parse(char *str, GNode *ctxt, Boolean err, size_t *lengthPtr,
879     Boolean *freePtr)
880 {
881     char            *tstr;      /* Pointer into str */
882     Var             *v;         /* Variable in invocation */
883     char            *cp;        /* Secondary pointer into str (place marker
884                                  * for tstr) */
885     Boolean         haveModifier;/* TRUE if have modifiers for the variable */
886     char            endc;       /* Ending character when variable in parens
887                                  * or braces */
888     char            startc=0;   /* Starting character when variable in parens
889                                  * or braces */
890     int             cnt;        /* Used to count brace pairs when variable in
891                                  * in parens or braces */
892     char            *start;
893     char             delim;
894     Boolean         dynamic;    /* TRUE if the variable is local and we're
895                                  * expanding it in a non-local context. This
896                                  * is done to support dynamic sources. The
897                                  * result is just the invocation, unaltered */
898     int         vlen;           /* length of variable name, after embedded variable
899                                  * expansion */
900
901     *freePtr = FALSE;
902     dynamic = FALSE;
903     start = str;
904
905     /*
906      * It is assumed that Var_Parse() is called with str[0] == '$'
907      */
908
909     if (str[1] != OPEN_PAREN && str[1] != OPEN_BRACKET) {
910         /*
911          * If it's not bounded by braces of some sort, life is much simpler.
912          * We just need to check for the first character and return the
913          * value if it exists.
914          */
915         char      name[2];
916
917         name[0] = str[1];
918         name[1] = '\0';
919
920         v = VarFind(name, ctxt, FIND_ENV | FIND_GLOBAL | FIND_CMD);
921         if (v == (Var *)NULL) {
922             if (str[1] != '\0')
923                 *lengthPtr = 2;
924             else
925                 *lengthPtr = 1;
926
927             if ((ctxt == VAR_CMD) || (ctxt == VAR_GLOBAL)) {
928                 /*
929                  * If substituting a local variable in a non-local context,
930                  * assume it's for dynamic source stuff. We have to handle
931                  * this specially and return the longhand for the variable
932                  * with the dollar sign escaped so it makes it back to the
933                  * caller. Only four of the local variables are treated
934                  * specially as they are the only four that will be set
935                  * when dynamic sources are expanded.
936                  */
937                 /* XXX: It looks like $% and $! are reversed here */
938                 switch (str[1]) {
939                     case '@':
940                         return ("$(.TARGET)");
941                     case '%':
942                         return ("$(.ARCHIVE)");
943                     case '*':
944                         return ("$(.PREFIX)");
945                     case '!':
946                         return ("$(.MEMBER)");
947                     default:
948                         break;
949                 }
950             }
951             /*
952              * Error
953              */
954             return (err ? var_Error : varNoError);
955         } else {
956             haveModifier = FALSE;
957             tstr = &str[1];
958             endc = str[1];
959         }
960     } else {
961         /* build up expanded variable name in this buffer */
962         Buffer  *buf = Buf_Init(MAKE_BSIZE);
963
964         /*
965          * Skip to the end character or a colon, whichever comes first,
966          * replacing embedded variables as we go.
967          */
968         startc = str[1];
969         endc = (startc == OPEN_PAREN) ? CLOSE_PAREN : CLOSE_BRACKET;
970
971         tstr = str + 2;;
972         while (*tstr != '\0' && *tstr != endc && *tstr != ':') {
973             if (*tstr == '$') {
974                 size_t  rlen;
975                 Boolean rfree;
976                 char    *rval;
977
978                 rval = Var_Parse(tstr, ctxt, err, &rlen, &rfree);
979                 if (rval == var_Error) {
980                         Fatal("Error expanding embedded variable.");
981                 } else if (rval != NULL) {
982                         Buf_Append(buf, rval);
983                         if (rfree)
984                                 free(rval);
985                 }
986                 tstr += rlen - 1;
987             } else {
988                 Buf_AddByte(buf, (Byte)*tstr);
989             }
990             tstr++;
991         }
992
993         if (*tstr == '\0') {
994             /*
995              * If we never did find the end character, return NULL
996              * right now, setting the length to be the distance to
997              * the end of the string, since that's what make does.
998              */
999             *lengthPtr = tstr - str;
1000             return (var_Error);
1001         }
1002
1003         haveModifier = (*tstr == ':');
1004         *tstr = '\0';                   /* modify input string */
1005
1006         Buf_AddByte(buf, (Byte)'\0');
1007         str = Buf_GetAll(buf, (size_t *)NULL);
1008         vlen = strlen(str);
1009
1010         v = VarFind(str, ctxt, FIND_ENV | FIND_GLOBAL | FIND_CMD);
1011         if ((v == (Var *)NULL) && (ctxt != VAR_CMD) && (ctxt != VAR_GLOBAL) &&
1012             (vlen == 2) && (str[1] == 'F' || str[1] == 'D'))
1013         {
1014             /*
1015              * Check for bogus D and F forms of local variables since we're
1016              * in a local context and the name is the right length.
1017              */
1018             switch (str[0]) {
1019                 case '@':
1020                 case '%':
1021                 case '*':
1022                 case '!':
1023                 case '>':
1024                 case '<':
1025                 {
1026                     char    vname[2];
1027                     char    *val;
1028
1029                     /*
1030                      * Well, it's local -- go look for it.
1031                      */
1032                     vname[0] = str[0];
1033                     vname[1] = '\0';
1034                     v = VarFind(vname, ctxt, 0);
1035
1036                     if (v != NULL && !haveModifier) {
1037                         /*
1038                          * No need for nested expansion or anything, as we're
1039                          * the only one who sets these things and we sure don't
1040                          * put nested invocations in them...
1041                          */
1042                         val = (char *)Buf_GetAll(v->val, (size_t *)NULL);
1043
1044                         if (str[1] == 'D') {
1045                             val = VarModify(val, VarHead, (void *)NULL);
1046                         } else {
1047                             val = VarModify(val, VarTail, (void *)NULL);
1048                         }
1049                         /*
1050                          * Resulting string is dynamically allocated, so
1051                          * tell caller to free it.
1052                          */
1053                         *freePtr = TRUE;
1054                         *lengthPtr = tstr-start+1;
1055                         *tstr = endc;
1056                         Buf_Destroy(buf, TRUE);
1057                         return (val);
1058                     }
1059                     break;
1060                 default:
1061                     break;
1062                 }
1063             }
1064         }
1065
1066         if (v == (Var *)NULL) {
1067             if (((vlen == 1) ||
1068                  (((vlen == 2) && (str[1] == 'F' || str[1] == 'D')))) &&
1069                 ((ctxt == VAR_CMD) || (ctxt == VAR_GLOBAL)))
1070             {
1071                 /*
1072                  * If substituting a local variable in a non-local context,
1073                  * assume it's for dynamic source stuff. We have to handle
1074                  * this specially and return the longhand for the variable
1075                  * with the dollar sign escaped so it makes it back to the
1076                  * caller. Only four of the local variables are treated
1077                  * specially as they are the only four that will be set
1078                  * when dynamic sources are expanded.
1079                  */
1080                 switch (str[0]) {
1081                     case '@':
1082                     case '%':
1083                     case '*':
1084                     case '!':
1085                         dynamic = TRUE;
1086                         break;
1087                     default:
1088                         break;
1089                 }
1090             } else if ((vlen > 2) && (str[0] == '.') &&
1091                        isupper((unsigned char)str[1]) &&
1092                        ((ctxt == VAR_CMD) || (ctxt == VAR_GLOBAL)))
1093             {
1094                 int     len;
1095
1096                 len = vlen - 1;
1097                 if ((strncmp(str, ".TARGET", len) == 0) ||
1098                     (strncmp(str, ".ARCHIVE", len) == 0) ||
1099                     (strncmp(str, ".PREFIX", len) == 0) ||
1100                     (strncmp(str, ".MEMBER", len) == 0))
1101                 {
1102                     dynamic = TRUE;
1103                 }
1104             }
1105
1106             if (!haveModifier) {
1107                 /*
1108                  * No modifiers -- have specification length so we can return
1109                  * now.
1110                  */
1111                 *lengthPtr = tstr - start + 1;
1112                 *tstr = endc;
1113                 if (dynamic) {
1114                     str = emalloc(*lengthPtr + 1);
1115                     strncpy(str, start, *lengthPtr);
1116                     str[*lengthPtr] = '\0';
1117                     *freePtr = TRUE;
1118                     Buf_Destroy(buf, TRUE);
1119                     return (str);
1120                 } else {
1121                     Buf_Destroy(buf, TRUE);
1122                     return (err ? var_Error : varNoError);
1123                 }
1124             } else {
1125                 /*
1126                  * Still need to get to the end of the variable specification,
1127                  * so kludge up a Var structure for the modifications
1128                  */
1129                 v = VarCreate(str, NULL, VAR_JUNK);
1130             }
1131         }
1132         Buf_Destroy(buf, TRUE);
1133     }
1134
1135     if (v->flags & VAR_IN_USE) {
1136         Fatal("Variable %s is recursive.", v->name);
1137         /*NOTREACHED*/
1138     } else {
1139         v->flags |= VAR_IN_USE;
1140     }
1141
1142     /*
1143      * Before doing any modification, we have to make sure the value
1144      * has been fully expanded. If it looks like recursion might be
1145      * necessary (there's a dollar sign somewhere in the variable's value)
1146      * we just call Var_Subst to do any other substitutions that are
1147      * necessary. Note that the value returned by Var_Subst will have
1148      * been dynamically-allocated, so it will need freeing when we
1149      * return.
1150      */
1151     str = (char *)Buf_GetAll(v->val, (size_t *)NULL);
1152     if (strchr(str, '$') != NULL) {
1153         Buffer  *buf;
1154
1155         buf = Var_Subst(NULL, str, ctxt, err);
1156         str = Buf_GetAll(buf, NULL);
1157         Buf_Destroy(buf, FALSE);
1158
1159         *freePtr = TRUE;
1160     }
1161
1162     v->flags &= ~VAR_IN_USE;
1163
1164     /*
1165      * Now we need to apply any modifiers the user wants applied.
1166      * These are:
1167      *            :M<pattern>   words which match the given <pattern>.
1168      *                          <pattern> is of the standard file
1169      *                          wildcarding form.
1170      *            :S<d><pat1><d><pat2><d>[g]
1171      *                          Substitute <pat2> for <pat1> in the value
1172      *            :C<d><pat1><d><pat2><d>[g]
1173      *                          Substitute <pat2> for regex <pat1> in the value
1174      *            :H            Substitute the head of each word
1175      *            :T            Substitute the tail of each word
1176      *            :E            Substitute the extension (minus '.') of
1177      *                          each word
1178      *            :R            Substitute the root of each word
1179      *                          (pathname minus the suffix).
1180      *            :lhs=rhs      Like :S, but the rhs goes to the end of
1181      *                          the invocation.
1182      *            :U            Converts variable to upper-case.
1183      *            :L            Converts variable to lower-case.
1184      */
1185     if ((str != NULL) && haveModifier) {
1186         /*
1187          * Skip initial colon while putting it back.
1188          */
1189         *tstr++ = ':';
1190         while (*tstr != endc) {
1191             char        *newStr;    /* New value to return */
1192             char        termc;      /* Character which terminated scan */
1193
1194             DEBUGF(VAR, ("Applying :%c to \"%s\"\n", *tstr, str));
1195             switch (*tstr) {
1196                 case 'N':
1197                 case 'M':
1198                 {
1199                     char    *pattern;
1200                     char    *cp2;
1201                     Boolean copy;
1202
1203                     copy = FALSE;
1204                     for (cp = tstr + 1;
1205                          *cp != '\0' && *cp != ':' && *cp != endc;
1206                          cp++)
1207                     {
1208                         if (*cp == '\\' && (cp[1] == ':' || cp[1] == endc)) {
1209                             copy = TRUE;
1210                             cp++;
1211                         }
1212                     }
1213                     termc = *cp;
1214                     *cp = '\0';
1215                     if (copy) {
1216                         /*
1217                          * Need to compress the \:'s out of the pattern, so
1218                          * allocate enough room to hold the uncompressed
1219                          * pattern (note that cp started at tstr+1, so
1220                          * cp - tstr takes the null byte into account) and
1221                          * compress the pattern into the space.
1222                          */
1223                         pattern = emalloc(cp - tstr);
1224                         for (cp2 = pattern, cp = tstr + 1;
1225                              *cp != '\0';
1226                              cp++, cp2++)
1227                         {
1228                             if ((*cp == '\\') &&
1229                                 (cp[1] == ':' || cp[1] == endc)) {
1230                                     cp++;
1231                             }
1232                             *cp2 = *cp;
1233                         }
1234                         *cp2 = '\0';
1235                     } else {
1236                         pattern = &tstr[1];
1237                     }
1238                     if (*tstr == 'M' || *tstr == 'm') {
1239                         newStr = VarModify(str, VarMatch, pattern);
1240                     } else {
1241                         newStr = VarModify(str, VarNoMatch, pattern);
1242                     }
1243                     if (copy) {
1244                         free(pattern);
1245                     }
1246                     break;
1247                 }
1248                 case 'S':
1249                 {
1250                     VarPattern      pattern;
1251                     char            del;
1252                     Buffer          *buf;       /* Buffer for patterns */
1253
1254                     pattern.flags = 0;
1255                     del = tstr[1];
1256                     tstr += 2;
1257
1258                     /*
1259                      * If pattern begins with '^', it is anchored to the
1260                      * start of the word -- skip over it and flag pattern.
1261                      */
1262                     if (*tstr == '^') {
1263                         pattern.flags |= VAR_MATCH_START;
1264                         tstr += 1;
1265                     }
1266
1267                     buf = Buf_Init(0);
1268
1269                     /*
1270                      * Pass through the lhs looking for 1) escaped delimiters,
1271                      * '$'s and backslashes (place the escaped character in
1272                      * uninterpreted) and 2) unescaped $'s that aren't before
1273                      * the delimiter (expand the variable substitution).
1274                      * The result is left in the Buffer buf.
1275                      */
1276                     for (cp = tstr; *cp != '\0' && *cp != del; cp++) {
1277                         if ((*cp == '\\') &&
1278                             ((cp[1] == del) ||
1279                              (cp[1] == '$') ||
1280                              (cp[1] == '\\')))
1281                         {
1282                             Buf_AddByte(buf, (Byte)cp[1]);
1283                             cp++;
1284                         } else if (*cp == '$') {
1285                             if (cp[1] != del) {
1286                                 /*
1287                                  * If unescaped dollar sign not before the
1288                                  * delimiter, assume it's a variable
1289                                  * substitution and recurse.
1290                                  */
1291                                 char        *cp2;
1292                                 size_t len;
1293                                 Boolean     freeIt;
1294
1295                                 cp2 = Var_Parse(cp, ctxt, err, &len, &freeIt);
1296                                 Buf_Append(buf, cp2);
1297                                 if (freeIt) {
1298                                     free(cp2);
1299                                 }
1300                                 cp += len - 1;
1301                             } else {
1302                                 /*
1303                                  * Unescaped $ at end of pattern => anchor
1304                                  * pattern at end.
1305                                  */
1306                                 pattern.flags |= VAR_MATCH_END;
1307                             }
1308                         } else {
1309                             Buf_AddByte(buf, (Byte)*cp);
1310                         }
1311                     }
1312
1313                     Buf_AddByte(buf, (Byte)'\0');
1314
1315                     /*
1316                      * If lhs didn't end with the delimiter, complain and
1317                      * exit.
1318                      */
1319                     if (*cp != del) {
1320                         Fatal("Unclosed substitution for %s (%c missing)",
1321                               v->name, del);
1322                     }
1323
1324                     /*
1325                      * Fetch pattern and destroy buffer, but preserve the data
1326                      * in it, since that's our lhs. Note that Buf_GetAll
1327                      * will return the actual number of bytes, which includes
1328                      * the null byte, so we have to decrement the length by
1329                      * one.
1330                      */
1331                     pattern.lhs = (char *)Buf_GetAll(buf, &pattern.leftLen);
1332                     pattern.leftLen--;
1333                     Buf_Destroy(buf, FALSE);
1334
1335                     /*
1336                      * Now comes the replacement string. Three things need to
1337                      * be done here: 1) need to compress escaped delimiters and
1338                      * ampersands and 2) need to replace unescaped ampersands
1339                      * with the l.h.s. (since this isn't regexp, we can do
1340                      * it right here) and 3) expand any variable substitutions.
1341                      */
1342                     buf = Buf_Init(0);
1343
1344                     tstr = cp + 1;
1345                     for (cp = tstr; *cp != '\0' && *cp != del; cp++) {
1346                         if ((*cp == '\\') &&
1347                             ((cp[1] == del) ||
1348                              (cp[1] == '&') ||
1349                              (cp[1] == '\\') ||
1350                              (cp[1] == '$')))
1351                         {
1352                             Buf_AddByte(buf, (Byte)cp[1]);
1353                             cp++;
1354                         } else if ((*cp == '$') && (cp[1] != del)) {
1355                             char    *cp2;
1356                             size_t len;
1357                             Boolean freeIt;
1358
1359                             cp2 = Var_Parse(cp, ctxt, err, &len, &freeIt);
1360                             Buf_Append(buf, cp2);
1361                             cp += len - 1;
1362                             if (freeIt) {
1363                                 free(cp2);
1364                             }
1365                         } else if (*cp == '&') {
1366                             Buf_AddBytes(buf, pattern.leftLen,
1367                                          (Byte *)pattern.lhs);
1368                         } else {
1369                             Buf_AddByte(buf, (Byte)*cp);
1370                         }
1371                     }
1372
1373                     Buf_AddByte(buf, (Byte)'\0');
1374
1375                     /*
1376                      * If didn't end in delimiter character, complain
1377                      */
1378                     if (*cp != del) {
1379                         Fatal("Unclosed substitution for %s (%c missing)",
1380                               v->name, del);
1381                     }
1382
1383                     pattern.rhs = (char *)Buf_GetAll(buf, &pattern.rightLen);
1384                     pattern.rightLen--;
1385                     Buf_Destroy(buf, FALSE);
1386
1387                     /*
1388                      * Check for global substitution. If 'g' after the final
1389                      * delimiter, substitution is global and is marked that
1390                      * way.
1391                      */
1392                     cp++;
1393                     if (*cp == 'g') {
1394                         pattern.flags |= VAR_SUB_GLOBAL;
1395                         cp++;
1396                     }
1397
1398                     /*
1399                      * Global substitution of the empty string causes an
1400                      * infinite number of matches, unless anchored by '^'
1401                      * (start of string) or '$' (end of string). Catch the
1402                      * infinite substitution here.
1403                      * Note that flags can only contain the 3 bits we're
1404                      * interested in so we don't have to mask unrelated
1405                      * bits. We can test for equality.
1406                      */
1407                     if (!pattern.leftLen && pattern.flags == VAR_SUB_GLOBAL)
1408                         Fatal("Global substitution of the empty string");
1409
1410                     termc = *cp;
1411                     newStr = VarModify(str, VarSubstitute, &pattern);
1412                     /*
1413                      * Free the two strings.
1414                      */
1415                     free(pattern.lhs);
1416                     free(pattern.rhs);
1417                     break;
1418                 }
1419                 case 'C':
1420                 {
1421                     VarREPattern    pattern;
1422                     char           *re;
1423                     int             error;
1424
1425                     pattern.flags = 0;
1426                     delim = tstr[1];
1427                     tstr += 2;
1428
1429                     cp = tstr;
1430
1431                     if ((re = VarGetPattern(ctxt, err, &cp, delim, NULL,
1432                         NULL, NULL)) == NULL) {
1433                         /* was: goto cleanup */
1434                         *lengthPtr = cp - start + 1;
1435                         if (*freePtr)
1436                             free(str);
1437                         if (delim != '\0')
1438                             Fatal("Unclosed substitution for %s (%c missing)",
1439                                   v->name, delim);
1440                         return (var_Error);
1441                     }
1442
1443                     if ((pattern.replace = VarGetPattern(ctxt, err, &cp,
1444                         delim, NULL, NULL, NULL)) == NULL){
1445                         free(re);
1446
1447                         /* was: goto cleanup */
1448                         *lengthPtr = cp - start + 1;
1449                         if (*freePtr)
1450                             free(str);
1451                         if (delim != '\0')
1452                             Fatal("Unclosed substitution for %s (%c missing)",
1453                                   v->name, delim);
1454                         return (var_Error);
1455                     }
1456
1457                     for (;; cp++) {
1458                         switch (*cp) {
1459                         case 'g':
1460                             pattern.flags |= VAR_SUB_GLOBAL;
1461                             continue;
1462                         case '1':
1463                             pattern.flags |= VAR_SUB_ONE;
1464                             continue;
1465                         default:
1466                             break;
1467                         }
1468                         break;
1469                     }
1470
1471                     termc = *cp;
1472
1473                     error = regcomp(&pattern.re, re, REG_EXTENDED);
1474                     free(re);
1475                     if (error)  {
1476                         *lengthPtr = cp - start + 1;
1477                         VarREError(error, &pattern.re, "RE substitution error");
1478                         free(pattern.replace);
1479                         return (var_Error);
1480                     }
1481
1482                     pattern.nsub = pattern.re.re_nsub + 1;
1483                     if (pattern.nsub < 1)
1484                         pattern.nsub = 1;
1485                     if (pattern.nsub > 10)
1486                         pattern.nsub = 10;
1487                     pattern.matches = emalloc(pattern.nsub *
1488                                               sizeof(regmatch_t));
1489                     newStr = VarModify(str, VarRESubstitute, &pattern);
1490                     regfree(&pattern.re);
1491                     free(pattern.replace);
1492                     free(pattern.matches);
1493                     break;
1494                 }
1495                 case 'L':
1496                     if (tstr[1] == endc || tstr[1] == ':') {
1497                         Buffer *buf;
1498                         buf = Buf_Init(MAKE_BSIZE);
1499                         for (cp = str; *cp ; cp++)
1500                             Buf_AddByte(buf, (Byte)tolower(*cp));
1501
1502                         Buf_AddByte(buf, (Byte)'\0');
1503                         newStr = (char *)Buf_GetAll(buf, (size_t *)NULL);
1504                         Buf_Destroy(buf, FALSE);
1505
1506                         cp = tstr + 1;
1507                         termc = *cp;
1508                         break;
1509                     }
1510                     /* FALLTHROUGH */
1511                 case 'O':
1512                     if (tstr[1] == endc || tstr[1] == ':') {
1513                         newStr = VarSortWords(str, SortIncreasing);
1514                         cp = tstr + 1;
1515                         termc = *cp;
1516                         break;
1517                     }
1518                     /* FALLTHROUGH */
1519                 case 'Q':
1520                     if (tstr[1] == endc || tstr[1] == ':') {
1521                         newStr = Var_Quote(str);
1522                         cp = tstr + 1;
1523                         termc = *cp;
1524                         break;
1525                     }
1526                     /*FALLTHRU*/
1527                 case 'T':
1528                     if (tstr[1] == endc || tstr[1] == ':') {
1529                         newStr = VarModify(str, VarTail, (void *)NULL);
1530                         cp = tstr + 1;
1531                         termc = *cp;
1532                         break;
1533                     }
1534                     /*FALLTHRU*/
1535                 case 'U':
1536                     if (tstr[1] == endc || tstr[1] == ':') {
1537                         Buffer *buf;
1538                         buf = Buf_Init(MAKE_BSIZE);
1539                         for (cp = str; *cp ; cp++)
1540                             Buf_AddByte(buf, (Byte)toupper(*cp));
1541
1542                         Buf_AddByte(buf, (Byte)'\0');
1543                         newStr = (char *)Buf_GetAll(buf, (size_t *)NULL);
1544                         Buf_Destroy(buf, FALSE);
1545
1546                         cp = tstr + 1;
1547                         termc = *cp;
1548                         break;
1549                     }
1550                     /* FALLTHROUGH */
1551                 case 'H':
1552                     if (tstr[1] == endc || tstr[1] == ':') {
1553                         newStr = VarModify(str, VarHead, (void *)NULL);
1554                         cp = tstr + 1;
1555                         termc = *cp;
1556                         break;
1557                     }
1558                     /*FALLTHRU*/
1559                 case 'E':
1560                     if (tstr[1] == endc || tstr[1] == ':') {
1561                         newStr = VarModify(str, VarSuffix, (void *)NULL);
1562                         cp = tstr + 1;
1563                         termc = *cp;
1564                         break;
1565                     }
1566                     /*FALLTHRU*/
1567                 case 'R':
1568                     if (tstr[1] == endc || tstr[1] == ':') {
1569                         newStr = VarModify(str, VarRoot, (void *)NULL);
1570                         cp = tstr + 1;
1571                         termc = *cp;
1572                         break;
1573                     }
1574                     /*FALLTHRU*/
1575 #ifdef SUNSHCMD
1576                 case 's':
1577                     if (tstr[1] == 'h' && (tstr[2] == endc || tstr[2] == ':')) {
1578                         const char *error;
1579                         {
1580                             Buffer *buf = Cmd_Exec(str, &error);
1581                             newStr = Buf_GetAll(buf, NULL);
1582                             Buf_Destroy(buf, FALSE);
1583                         }
1584                         if (error)
1585                             Error(error, str);
1586                         cp = tstr + 2;
1587                         termc = *cp;
1588                         break;
1589                     }
1590                     /*FALLTHRU*/
1591 #endif
1592                 default:
1593                 {
1594 #ifdef SYSVVARSUB
1595                     /*
1596                      * This can either be a bogus modifier or a System-V
1597                      * substitution command.
1598                      */
1599                     VarPattern      pattern;
1600                     Boolean         eqFound;
1601
1602                     pattern.flags = 0;
1603                     eqFound = FALSE;
1604                     /*
1605                      * First we make a pass through the string trying
1606                      * to verify it is a SYSV-make-style translation:
1607                      * it must be: <string1>=<string2>)
1608                      */
1609                     cp = tstr;
1610                     cnt = 1;
1611                     while (*cp != '\0' && cnt) {
1612                         if (*cp == '=') {
1613                             eqFound = TRUE;
1614                             /* continue looking for endc */
1615                         }
1616                         else if (*cp == endc)
1617                             cnt--;
1618                         else if (*cp == startc)
1619                             cnt++;
1620                         if (cnt)
1621                             cp++;
1622                     }
1623                     if (*cp == endc && eqFound) {
1624
1625                         /*
1626                          * Now we break this sucker into the lhs and
1627                          * rhs. We must null terminate them of course.
1628                          */
1629                         cp = tstr;
1630
1631                         delim = '=';
1632                         if ((pattern.lhs = VarGetPattern(ctxt,
1633                             err, &cp, delim, &pattern.flags, &pattern.leftLen,
1634                             NULL)) == NULL) {
1635                                 /* was: goto cleanup */
1636                                 *lengthPtr = cp - start + 1;
1637                                 if (*freePtr)
1638                                     free(str);
1639                                 if (delim != '\0')
1640                                     Fatal("Unclosed substitution for %s (%c missing)",
1641                                           v->name, delim);
1642                                 return (var_Error);
1643                         }
1644
1645                         delim = endc;
1646                         if ((pattern.rhs = VarGetPattern(ctxt,
1647                             err, &cp, delim, NULL, &pattern.rightLen,
1648                             &pattern)) == NULL) {
1649                                 /* was: goto cleanup */
1650                                 *lengthPtr = cp - start + 1;
1651                                 if (*freePtr)
1652                                     free(str);
1653                                 if (delim != '\0')
1654                                     Fatal("Unclosed substitution for %s (%c missing)",
1655                                           v->name, delim);
1656                                 return (var_Error);
1657                         }
1658
1659                         /*
1660                          * SYSV modifications happen through the whole
1661                          * string. Note the pattern is anchored at the end.
1662                          */
1663                         termc = *--cp;
1664                         delim = '\0';
1665                         newStr = VarModify(str, VarSYSVMatch, &pattern);
1666
1667                         free(pattern.lhs);
1668                         free(pattern.rhs);
1669
1670                         termc = endc;
1671                     } else
1672 #endif
1673                     {
1674                         Error("Unknown modifier '%c'\n", *tstr);
1675                         for (cp = tstr+1;
1676                              *cp != ':' && *cp != endc && *cp != '\0';
1677                              cp++)
1678                                  continue;
1679                         termc = *cp;
1680                         newStr = var_Error;
1681                     }
1682                 }
1683             }
1684             DEBUGF(VAR, ("Result is \"%s\"\n", newStr));
1685
1686             if (*freePtr) {
1687                 free(str);
1688             }
1689             str = newStr;
1690             if (str != var_Error) {
1691                 *freePtr = TRUE;
1692             } else {
1693                 *freePtr = FALSE;
1694             }
1695             if (termc == '\0') {
1696                 Error("Unclosed variable specification for %s", v->name);
1697             } else if (termc == ':') {
1698                 *cp++ = termc;
1699             } else {
1700                 *cp = termc;
1701             }
1702             tstr = cp;
1703         }
1704         *lengthPtr = tstr - start + 1;
1705     } else {
1706         *lengthPtr = tstr - start + 1;
1707         *tstr = endc;
1708     }
1709
1710     if (v->flags & VAR_FROM_ENV) {
1711         Boolean   destroy = FALSE;
1712
1713         if (str != (char *)Buf_GetAll(v->val, (size_t *)NULL)) {
1714             destroy = TRUE;
1715         } else {
1716             /*
1717              * Returning the value unmodified, so tell the caller to free
1718              * the thing.
1719              */
1720             *freePtr = TRUE;
1721         }
1722         VarDestroy(v, destroy);
1723     } else if (v->flags & VAR_JUNK) {
1724         /*
1725          * Perform any free'ing needed and set *freePtr to FALSE so the caller
1726          * doesn't try to free a static pointer.
1727          */
1728         if (*freePtr) {
1729             free(str);
1730         }
1731         *freePtr = FALSE;
1732         VarDestroy(v, TRUE);
1733         if (dynamic) {
1734             str = emalloc(*lengthPtr + 1);
1735             strncpy(str, start, *lengthPtr);
1736             str[*lengthPtr] = '\0';
1737             *freePtr = TRUE;
1738         } else {
1739             str = err ? var_Error : varNoError;
1740         }
1741     }
1742     return (str);
1743 }
1744
1745 /*-
1746  *-----------------------------------------------------------------------
1747  * Var_Subst  --
1748  *      Substitute for all variables in the given string in the given context
1749  *      If undefErr is TRUE, Parse_Error will be called when an undefined
1750  *      variable is encountered.
1751  *
1752  * Results:
1753  *      The resulting string.
1754  *
1755  * Side Effects:
1756  *      None. The old string must be freed by the caller
1757  *-----------------------------------------------------------------------
1758  */
1759 Buffer *
1760 Var_Subst(const char *var, const char *str, GNode *ctxt, Boolean undefErr)
1761 {
1762     Boolean     errorReported;
1763     Buffer      *buf;           /* Buffer for forming things */
1764
1765     /*
1766      * Set TRUE if an error has already been reported to prevent a
1767      * plethora of messages when recursing.
1768      */
1769     errorReported = FALSE;
1770
1771     buf = Buf_Init(0);
1772     while (*str) {
1773         if (var == NULL && (str[0] == '$') && (str[1] == '$')) {
1774             /*
1775              * A dollar sign may be escaped either with another dollar sign.
1776              * In such a case, we skip over the escape character and store the
1777              * dollar sign into the buffer directly.
1778              */
1779             Buf_AddByte(buf, (Byte)str[0]);
1780             str += 2;
1781
1782         } else if (str[0] == '$') {
1783             char        *val;   /* Value to substitute for a variable */
1784             size_t      length; /* Length of the variable invocation */
1785             Boolean     doFree; /* Set true if val should be freed */
1786             /*
1787              * Variable invocation.
1788              */
1789             if (var != NULL) {
1790                 int expand;
1791                 for (;;) {
1792                     if (str[1] == OPEN_PAREN || str[1] == OPEN_BRACKET) {
1793                         const char *p = str + 2;
1794
1795                         /*
1796                          * Scan up to the end of the variable name.
1797                          */
1798                         while (*p != '\0' &&
1799                                *p != ':' &&
1800                                *p != CLOSE_PAREN &&
1801                                *p != CLOSE_BRACKET &&
1802                                *p != '$') {
1803                             ++p;
1804                         }
1805
1806                         /*
1807                          * A variable inside the variable. We cannot expand
1808                          * the external variable yet, so we try again with
1809                          * the nested one
1810                          */
1811                         if (*p == '$') {
1812                             Buf_AppendRange(buf, str, p);
1813                             str = p;
1814                             continue;
1815                         }
1816
1817                         if (var[p - (str + 2)] == '\0' && strncmp(var, str + 2, p - (str + 2)) == 0) {
1818                             expand = TRUE;
1819                         } else {
1820                             /*
1821                              * Not the variable we want to expand, scan
1822                              * until the next variable
1823                              */
1824                             while (*p != '$' && *p != '\0')
1825                                 ++p;
1826
1827                             Buf_AppendRange(buf, str, p);
1828                             str = p;
1829                             expand = FALSE;
1830                         }
1831
1832                     } else {
1833                         /*
1834                          * Single letter variable name.
1835                          */
1836                         if (var[1] == '\0' && str[1] == var[0]) {
1837                             expand = TRUE;
1838                         } else {
1839                             Buf_AddBytes(buf, 2, (const Byte *)str);
1840                             str += 2;
1841                             expand = FALSE;
1842                         }
1843                     }
1844                     break;
1845                 }
1846                 if (!expand)
1847                     continue;
1848             }
1849
1850             val = Var_Parse(str, ctxt, undefErr, &length, &doFree);
1851
1852             /*
1853              * When we come down here, val should either point to the
1854              * value of this variable, suitably modified, or be NULL.
1855              * Length should be the total length of the potential
1856              * variable invocation (from $ to end character...)
1857              */
1858             if (val == var_Error || val == varNoError) {
1859                 /*
1860                  * If performing old-time variable substitution, skip over
1861                  * the variable and continue with the substitution. Otherwise,
1862                  * store the dollar sign and advance str so we continue with
1863                  * the string...
1864                  */
1865                 if (oldVars) {
1866                     str += length;
1867                 } else if (undefErr) {
1868                     /*
1869                      * If variable is undefined, complain and skip the
1870                      * variable. The complaint will stop us from doing anything
1871                      * when the file is parsed.
1872                      */
1873                     if (!errorReported) {
1874                         Parse_Error(PARSE_FATAL,
1875                                      "Undefined variable \"%.*s\"",length,str);
1876                     }
1877                     str += length;
1878                     errorReported = TRUE;
1879                 } else {
1880                     Buf_AddByte(buf, (Byte)*str);
1881                     str += 1;
1882                 }
1883             } else {
1884                 /*
1885                  * We've now got a variable structure to store in. But first,
1886                  * advance the string pointer.
1887                  */
1888                 str += length;
1889
1890                 /*
1891                  * Copy all the characters from the variable value straight
1892                  * into the new string.
1893                  */
1894                 Buf_Append(buf, val);
1895                 if (doFree) {
1896                     free(val);
1897                 }
1898             }
1899
1900         } else {
1901             /*
1902              * Skip as many characters as possible -- either to the end of
1903              * the string or to the next dollar sign (variable invocation).
1904              */
1905             const char  *cp = str;
1906
1907             do {
1908                 str++;
1909             } while (str[0] != '$' && str[0] != '\0');
1910
1911             Buf_AppendRange(buf, cp, str);
1912         }
1913     }
1914
1915     return (buf);
1916 }
1917
1918 /*-
1919  *-----------------------------------------------------------------------
1920  * Var_GetTail --
1921  *      Return the tail from each of a list of words. Used to set the
1922  *      System V local variables.
1923  *
1924  * Results:
1925  *      The resulting string.
1926  *
1927  * Side Effects:
1928  *      None.
1929  *
1930  *-----------------------------------------------------------------------
1931  */
1932 char *
1933 Var_GetTail(char *file)
1934 {
1935
1936     return (VarModify(file, VarTail, (void *)NULL));
1937 }
1938
1939 /*-
1940  *-----------------------------------------------------------------------
1941  * Var_GetHead --
1942  *      Find the leading components of a (list of) filename(s).
1943  *      XXX: VarHead does not replace foo by ., as (sun) System V make
1944  *      does.
1945  *
1946  * Results:
1947  *      The leading components.
1948  *
1949  * Side Effects:
1950  *      None.
1951  *
1952  *-----------------------------------------------------------------------
1953  */
1954 char *
1955 Var_GetHead(char *file)
1956 {
1957
1958     return (VarModify(file, VarHead, (void *)NULL));
1959 }
1960
1961 /*-
1962  *-----------------------------------------------------------------------
1963  * Var_Init --
1964  *      Initialize the module
1965  *
1966  * Results:
1967  *      None
1968  *
1969  * Side Effects:
1970  *      The VAR_CMD and VAR_GLOBAL contexts are created
1971  *-----------------------------------------------------------------------
1972  */
1973 void
1974 Var_Init(void)
1975 {
1976
1977     VAR_GLOBAL = Targ_NewGN("Global");
1978     VAR_CMD = Targ_NewGN("Command");
1979 }
1980
1981 /****************** PRINT DEBUGGING INFO *****************/
1982 static int
1983 VarPrintVar(void *vp, void *dummy __unused)
1984 {
1985     Var    *v = (Var *) vp;
1986
1987     printf("%-16s = %s\n", v->name, (char *)Buf_GetAll(v->val, (size_t *)NULL));
1988     return (0);
1989 }
1990
1991 /*-
1992  *-----------------------------------------------------------------------
1993  * Var_Dump --
1994  *      print all variables in a context
1995  *-----------------------------------------------------------------------
1996  */
1997 void
1998 Var_Dump(GNode *ctxt)
1999 {
2000
2001     Lst_ForEach(&ctxt->context, VarPrintVar, (void *)NULL);
2002 }