Update xlat.c
[freeradius.git] / src / main / xlat.c
1 /*
2  *   This program is free software; you can redistribute it and/or modify
3  *   it under the terms of the GNU General Public License as published by
4  *   the Free Software Foundation; either version 2 of the License, or
5  *   (at your option) any later version.
6  *
7  *   This program is distributed in the hope that it will be useful,
8  *   but WITHOUT ANY WARRANTY; without even the implied warranty of
9  *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
10  *   GNU General Public License for more details.
11  *
12  *   You should have received a copy of the GNU General Public License
13  *   along with this program; if not, write to the Free Software
14  *   Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
15  */
16
17 /**
18  * $Id$
19  *
20  * @file xlat.c
21  * @brief String expansion ("translation"). Implements %Attribute -> value
22  *
23  * @copyright 2000,2006  The FreeRADIUS server project
24  * @copyright 2000  Alan DeKok <aland@ox.org>
25  */
26
27 RCSID("$Id$")
28
29 #include <freeradius-devel/radiusd.h>
30 #include <freeradius-devel/parser.h>
31 #include <freeradius-devel/rad_assert.h>
32 #include <freeradius-devel/base64.h>
33
34 #include <ctype.h>
35
36 typedef struct xlat_t {
37         char                    name[MAX_STRING_LEN];   //!< Name of the xlat expansion.
38         int                     length;                 //!< Length of name.
39         void                    *instance;              //!< Module instance passed to xlat and escape functions.
40         xlat_func_t             func;                   //!< xlat function.
41         xlat_escape_t   escape;                 //!< Escape function to apply to dynamic input to func.
42         bool                    internal;               //!< If true, cannot be redefined.
43 } xlat_t;
44
45 typedef enum {
46         XLAT_LITERAL,           //!< Literal string
47         XLAT_PERCENT,           //!< Literal string with %v
48         XLAT_MODULE,            //!< xlat module
49         XLAT_VIRTUAL,           //!< virtual attribute
50         XLAT_ATTRIBUTE,         //!< xlat attribute
51 #ifdef HAVE_REGEX
52         XLAT_REGEX,             //!< regex reference
53 #endif
54         XLAT_ALTERNATE          //!< xlat conditional syntax :-
55 } xlat_state_t;
56
57 struct xlat_exp {
58         char const *fmt;        //!< The format string.
59         size_t len;             //!< Length of the format string.
60
61         xlat_state_t type;      //!< type of this expansion.
62         xlat_exp_t *next;       //!< Next in the list.
63
64         xlat_exp_t *child;      //!< Nested expansion.
65         xlat_exp_t *alternate;  //!< Alternative expansion if this one expanded to a zero length string.
66
67         vp_tmpl_t attr; //!< An attribute template.
68         xlat_t const *xlat;     //!< The xlat expansion to expand format with.
69 };
70
71 typedef struct xlat_out {
72         char const *out;        //!< Output data.
73         size_t len;             //!< Length of the output string.
74 } xlat_out_t;
75
76 static rbtree_t *xlat_root = NULL;
77
78 #ifdef WITH_UNLANG
79 static char const * const xlat_foreach_names[] = {"Foreach-Variable-0",
80                                                   "Foreach-Variable-1",
81                                                   "Foreach-Variable-2",
82                                                   "Foreach-Variable-3",
83                                                   "Foreach-Variable-4",
84                                                   "Foreach-Variable-5",
85                                                   "Foreach-Variable-6",
86                                                   "Foreach-Variable-7",
87                                                   "Foreach-Variable-8",
88                                                   "Foreach-Variable-9",
89                                                   NULL};
90 #endif
91
92
93 static int xlat_inst[] = { 0, 1, 2, 3, 4, 5, 6, 7, 8, 9 };      /* up to 10 for foreach */
94
95 /** Print length of its RHS.
96  *
97  */
98 static ssize_t xlat_strlen(UNUSED void *instance, UNUSED REQUEST *request,
99                            char const *fmt, char *out, size_t outlen)
100 {
101         snprintf(out, outlen, "%u", (unsigned int) strlen(fmt));
102         return strlen(out);
103 }
104
105 /** Print the size of the attribute in bytes.
106  *
107  */
108 static ssize_t xlat_length(UNUSED void *instance, REQUEST *request,
109                            char const *fmt, char *out, size_t outlen)
110 {
111         VALUE_PAIR *vp;
112         while (isspace((int) *fmt)) fmt++;
113
114         if ((radius_get_vp(&vp, request, fmt) < 0) || !vp) {
115                 *out = '\0';
116                 return 0;
117         }
118
119         snprintf(out, outlen, "%zu", vp->vp_length);
120         return strlen(out);
121 }
122
123 /** Print data as integer, not as VALUE.
124  *
125  */
126 static ssize_t xlat_integer(UNUSED void *instance, REQUEST *request,
127                             char const *fmt, char *out, size_t outlen)
128 {
129         VALUE_PAIR      *vp;
130
131         uint64_t        int64 = 0;      /* Needs to be initialised to zero */
132         uint32_t        int32 = 0;      /* Needs to be initialised to zero */
133
134         while (isspace((int) *fmt)) fmt++;
135
136         if ((radius_get_vp(&vp, request, fmt) < 0) || !vp) {
137                 *out = '\0';
138                 return 0;
139         }
140
141         switch (vp->da->type) {
142         case PW_TYPE_OCTETS:
143         case PW_TYPE_STRING:
144                 if (vp->vp_length > 8) {
145                         break;
146                 }
147
148                 if (vp->vp_length > 4) {
149                         memcpy(&int64, vp->vp_octets, vp->vp_length);
150                         return snprintf(out, outlen, "%" PRIu64, htonll(int64));
151                 }
152
153                 memcpy(&int32, vp->vp_octets, vp->vp_length);
154                 return snprintf(out, outlen, "%i", htonl(int32));
155
156         case PW_TYPE_INTEGER64:
157                 return snprintf(out, outlen, "%" PRIu64, vp->vp_integer64);
158
159         /*
160          *      IP addresses are treated specially, as parsing functions assume the value
161          *      is bigendian and will convert it for us.
162          */
163         case PW_TYPE_IPV4_ADDR:
164                 return snprintf(out, outlen, "%u", htonl(vp->vp_ipaddr));
165
166         case PW_TYPE_IPV4_PREFIX:
167                 return snprintf(out, outlen, "%u", htonl((*(uint32_t *)(vp->vp_ipv4prefix + 2))));
168
169         case PW_TYPE_INTEGER:
170                 return snprintf(out, outlen, "%u", vp->vp_integer);
171
172         case PW_TYPE_DATE:
173                 return snprintf(out, outlen, "%u", vp->vp_date);
174
175         case PW_TYPE_BYTE:
176                 return snprintf(out, outlen, "%u", (unsigned int) vp->vp_byte);
177
178         case PW_TYPE_SHORT:
179                 return snprintf(out, outlen, "%u", (unsigned int) vp->vp_short);
180
181         /*
182          *      Ethernet is weird... It's network related, so we assume to it should be
183          *      bigendian.
184          */
185         case PW_TYPE_ETHERNET:
186                 memcpy(&int64, vp->vp_ether, vp->vp_length);
187                 return snprintf(out, outlen, "%" PRIu64, htonll(int64));
188
189         case PW_TYPE_SIGNED:
190                 return snprintf(out, outlen, "%i", vp->vp_signed);
191
192         case PW_TYPE_IPV6_ADDR:
193                 return fr_prints_uint128(out, outlen, ntohlll(*(uint128_t const *) &vp->vp_ipv6addr));
194
195         case PW_TYPE_IPV6_PREFIX:
196                 return fr_prints_uint128(out, outlen, ntohlll(*(uint128_t const *) &vp->vp_ipv6prefix[2]));
197
198         default:
199                 break;
200         }
201
202         REDEBUG("Type '%s' of length %zu cannot be converted to integer",
203                 fr_int2str(dict_attr_types, vp->da->type, "???"), vp->vp_length);
204         *out = '\0';
205
206         return -1;
207 }
208
209 /** Print data as hex, not as VALUE.
210  *
211  */
212 static ssize_t xlat_hex(UNUSED void *instance, REQUEST *request,
213                         char const *fmt, char *out, size_t outlen)
214 {
215         size_t i;
216         VALUE_PAIR *vp;
217         uint8_t const *p;
218         ssize_t ret;
219         size_t  len;
220         value_data_t dst;
221         uint8_t const *buff = NULL;
222
223         while (isspace((int) *fmt)) fmt++;
224
225         if ((radius_get_vp(&vp, request, fmt) < 0) || !vp) {
226         error:
227                 *out = '\0';
228                 return -1;
229         }
230
231         /*
232          *      The easy case.
233          */
234         if (vp->da->type == PW_TYPE_OCTETS) {
235                 p = vp->vp_octets;
236                 len = vp->vp_length;
237         /*
238          *      Cast the value_data_t of the VP to an octets string and
239          *      print that as hex.
240          */
241         } else {
242                 ret = value_data_cast(request, &dst, PW_TYPE_OCTETS, NULL, vp->da->type,
243                                       NULL, &vp->data, vp->vp_length);
244                 if (ret < 0) {
245                         REDEBUG("%s", fr_strerror());
246                         goto error;
247                 }
248                 len = (size_t) ret;
249                 p = buff = dst.octets;
250         }
251
252         rad_assert(p);
253
254         /*
255          *      Don't truncate the data.
256          */
257         if (outlen < (len * 2)) {
258                 rad_const_free(buff);
259                 goto error;
260         }
261
262         for (i = 0; i < len; i++) {
263                 snprintf(out + 2*i, 3, "%02x", p[i]);
264         }
265         rad_const_free(buff);
266
267         return len * 2;
268 }
269
270 /** Return the tag of an attribute reference
271  *
272  */
273 static ssize_t xlat_tag(UNUSED void *instance, REQUEST *request,
274                         char const *fmt, char *out, size_t outlen)
275 {
276         VALUE_PAIR *vp;
277
278         while (isspace((int) *fmt)) fmt++;
279
280         if ((radius_get_vp(&vp, request, fmt) < 0) || !vp) {
281                 *out = '\0';
282                 return 0;
283         }
284
285         if (!vp->da->flags.has_tag || !TAG_VALID(vp->tag)) {
286                 *out = '\0';
287                 return 0;
288         }
289
290         return snprintf(out, outlen, "%u", vp->tag);
291 }
292
293 /** Return the vendor of an attribute reference
294  *
295  */
296 static ssize_t xlat_vendor(UNUSED void *instance, REQUEST *request,
297                            char const *fmt, char *out, size_t outlen)
298 {
299         VALUE_PAIR *vp;
300         DICT_VENDOR *vendor;
301
302         while (isspace((int) *fmt)) fmt++;
303
304         if ((radius_get_vp(&vp, request, fmt) < 0) || !vp) {
305                 *out = '\0';
306                 return 0;
307         }
308
309         vendor = dict_vendorbyvalue(vp->da->vendor);
310         if (!vendor) {
311                 *out = '\0';
312                 return 0;
313         }
314         strlcpy(out, vendor->name, outlen);
315
316         return vendor->length;
317 }
318
319 /** Return the vendor number of an attribute reference
320  *
321  */
322 static ssize_t xlat_vendor_num(UNUSED void *instance, REQUEST *request,
323                                char const *fmt, char *out, size_t outlen)
324 {
325         VALUE_PAIR *vp;
326
327         while (isspace((int) *fmt)) fmt++;
328
329         if ((radius_get_vp(&vp, request, fmt) < 0) || !vp) {
330                 *out = '\0';
331                 return 0;
332         }
333
334         return snprintf(out, outlen, "%u", vp->da->vendor);
335 }
336
337 /** Return the attribute name of an attribute reference
338  *
339  */
340 static ssize_t xlat_attr(UNUSED void *instance, REQUEST *request,
341                          char const *fmt, char *out, size_t outlen)
342 {
343         VALUE_PAIR *vp;
344
345         while (isspace((int) *fmt)) fmt++;
346
347         if ((radius_get_vp(&vp, request, fmt) < 0) || !vp) {
348                 *out = '\0';
349                 return 0;
350         }
351         strlcpy(out, vp->da->name, outlen);
352
353         return strlen(vp->da->name);
354 }
355
356 /** Return the attribute number of an attribute reference
357  *
358  */
359 static ssize_t xlat_attr_num(UNUSED void *instance, REQUEST *request,
360                              char const *fmt, char *out, size_t outlen)
361 {
362         VALUE_PAIR *vp;
363
364         while (isspace((int) *fmt)) fmt++;
365
366         if ((radius_get_vp(&vp, request, fmt) < 0) || !vp) {
367                 *out = '\0';
368                 return 0;
369         }
370
371         return snprintf(out, outlen, "%u", vp->da->attr);
372 }
373
374 /** Print out attribute info
375  *
376  * Prints out all instances of a current attribute, or all attributes in a list.
377  *
378  * At higher debugging levels, also prints out alternative decodings of the same
379  * value. This is helpful to determine types for unknown attributes of long
380  * passed vendors, or just crazy/broken NAS.
381  *
382  * This expands to a zero length string.
383  */
384 static ssize_t xlat_debug_attr(UNUSED void *instance, REQUEST *request, char const *fmt,
385                                char *out, UNUSED size_t outlen)
386 {
387         VALUE_PAIR *vp;
388         vp_cursor_t cursor;
389
390         vp_tmpl_t vpt;
391
392         if (!RDEBUG_ENABLED2) {
393                 *out = '\0';
394                 return -1;
395         }
396
397         while (isspace((int) *fmt)) fmt++;
398
399         if (tmpl_from_attr_str(&vpt, fmt, REQUEST_CURRENT, PAIR_LIST_REQUEST, false, false) <= 0) {
400                 RDEBUG("%s", fr_strerror());
401                 return -1;
402         }
403
404         RIDEBUG("Attributes matching \"%s\"", fmt);
405
406         RINDENT();
407         for (vp = tmpl_cursor_init(NULL, &cursor, request, &vpt);
408              vp;
409              vp = tmpl_cursor_next(&cursor, &vpt)) {
410                 FR_NAME_NUMBER const *type;
411                 char *value;
412
413                 value = vp_aprints_value(vp, vp, '\'');
414                 if (vp->da->flags.has_tag) {
415                         RIDEBUG2("&%s:%s:%i %s %s",
416                                 fr_int2str(pair_lists, vpt.tmpl_list, "<INVALID>"),
417                                 vp->da->name,
418                                 vp->tag,
419                                 fr_int2str(fr_tokens, vp->op, "<INVALID>"),
420                                 value);
421                 } else {
422                         RIDEBUG2("&%s:%s %s %s",
423                                 fr_int2str(pair_lists, vpt.tmpl_list, "<INVALID>"),
424                                 vp->da->name,
425                                 fr_int2str(fr_tokens, vp->op, "<INVALID>"),
426                                 value);
427                 }
428                 talloc_free(value);
429
430                 if (!RDEBUG_ENABLED3) continue;
431
432                 if (vp->da->vendor) {
433                         DICT_VENDOR *dv;
434
435                         dv = dict_vendorbyvalue(vp->da->vendor);
436                         RIDEBUG2("Vendor : %i (%s)", vp->da->vendor, dv ? dv->name : "unknown");
437                 }
438                 RIDEBUG2("Type   : %s", fr_int2str(dict_attr_types, vp->da->type, "<INVALID>"));
439                 RIDEBUG2("Length : %zu", vp->vp_length);
440
441                 if (!RDEBUG_ENABLED4) continue;
442
443                 type = dict_attr_types;
444                 while (type->name) {
445                         int pad;
446
447                         value_data_t *dst = NULL;
448
449                         ssize_t ret;
450
451                         if ((PW_TYPE) type->number == vp->da->type) {
452                                 goto next_type;
453                         }
454
455                         switch (type->number) {
456                         case PW_TYPE_INVALID:           /* Not real type */
457                         case PW_TYPE_MAX:               /* Not real type */
458                         case PW_TYPE_EXTENDED:          /* Not safe/appropriate */
459                         case PW_TYPE_LONG_EXTENDED:     /* Not safe/appropriate */
460                         case PW_TYPE_TLV:               /* Not safe/appropriate */
461                         case PW_TYPE_EVS:               /* Not safe/appropriate */
462                         case PW_TYPE_VSA:               /* @fixme We need special behaviour for these */
463                         case PW_TYPE_COMBO_IP_ADDR:     /* Covered by IPv4 address IPv6 address */
464                         case PW_TYPE_COMBO_IP_PREFIX:   /* Covered by IPv4 address IPv6 address */
465                         case PW_TYPE_TIMEVAL:           /* Not a VALUE_PAIR type */
466                                 goto next_type;
467
468                         default:
469                                 break;
470                         }
471
472                         dst = talloc_zero(vp, value_data_t);
473                         ret = value_data_cast(dst, dst, type->number, NULL, vp->da->type, vp->da,
474                                               &vp->data, vp->vp_length);
475                         if (ret < 0) goto next_type;    /* We expect some to fail */
476
477                         value = value_data_aprints(dst, type->number, NULL, dst, (size_t)ret, '\'');
478                         if (!value) goto next_type;
479
480                         if ((pad = (11 - strlen(type->name))) < 0) {
481                                 pad = 0;
482                         }
483
484                         RINDENT();
485                         RDEBUG2("as %s%*s: %s", type->name, pad, " ", value);
486                         REXDENT();
487
488                 next_type:
489                         talloc_free(dst);
490                         type++;
491                 }
492         }
493
494         *out = '\0';
495         return 0;
496 }
497
498 /** Processes fmt as a map string and applies it to the current request
499  *
500  * e.g. "%{map:&User-Name := 'foo'}"
501  *
502  * Allows sets of modifications to be cached and then applied.
503  * Useful for processing generic attributes from LDAP.
504  */
505 static ssize_t xlat_map(UNUSED void *instance, REQUEST *request,
506                         char const *fmt, char *out, size_t outlen)
507 {
508         vp_map_t *map = NULL;
509         int ret;
510
511         if (map_afrom_attr_str(request, &map, fmt,
512                                REQUEST_CURRENT, PAIR_LIST_REQUEST,
513                                REQUEST_CURRENT, PAIR_LIST_REQUEST) < 0) {
514                 REDEBUG("Failed parsing \"%s\" as map: %s", fmt, fr_strerror());
515                 return -1;
516         }
517
518         RINDENT();
519         ret = map_to_request(request, map, map_to_vp, NULL);
520         REXDENT();
521         talloc_free(map);
522         if (ret < 0) return strlcpy(out, "0", outlen);
523
524         return strlcpy(out, "1", outlen);
525 }
526
527 /** Prints the current module processing the request
528  *
529  */
530 static ssize_t xlat_module(UNUSED void *instance, REQUEST *request,
531                            UNUSED char const *fmt, char *out, size_t outlen)
532 {
533         strlcpy(out, request->module, outlen);
534
535         return strlen(out);
536 }
537
538 #if defined(HAVE_REGEX) && defined(HAVE_PCRE)
539 static ssize_t xlat_regex(UNUSED void *instance, REQUEST *request,
540                           char const *fmt, char *out, size_t outlen)
541 {
542         char *p;
543         size_t len;
544
545         if (regex_request_to_sub_named(request, &p, request, fmt) < 0) {
546                 *out = '\0';
547                 return 0;
548         }
549
550         len = talloc_array_length(p);
551         if (len > outlen) {
552                 RDEBUG("Insufficient buffer space to write subcapture value, needed %zu bytes, have %zu bytes",
553                        len, outlen);
554                 return -1;
555         }
556         strlcpy(out, p, outlen);
557
558         return len - 1; /* - \0 */
559 }
560 #endif
561
562 #ifdef WITH_UNLANG
563 /** Implements the Foreach-Variable-X
564  *
565  * @see modcall()
566  */
567 static ssize_t xlat_foreach(void *instance, REQUEST *request,
568                             UNUSED char const *fmt, char *out, size_t outlen)
569 {
570         VALUE_PAIR      **pvp;
571         size_t          len;
572
573         /*
574          *      See modcall, "FOREACH" for how this works.
575          */
576         pvp = (VALUE_PAIR **) request_data_reference(request, (void *)radius_get_vp, *(int*) instance);
577         if (!pvp || !*pvp) {
578                 *out = '\0';
579                 return 0;
580         }
581
582         len = vp_prints_value(out, outlen, *pvp, 0);
583         if (is_truncated(len, outlen)) {
584                 RDEBUG("Insufficient buffer space to write foreach value");
585                 return -1;
586         }
587
588         return len;
589 }
590 #endif
591
592 /** Print data as string, if possible.
593  *
594  * If attribute "Foo" is defined as "octets" it will normally
595  * be printed as 0x0a0a0a. The xlat "%{string:Foo}" will instead
596  * expand to "\n\n\n"
597  */
598 static ssize_t xlat_string(UNUSED void *instance, REQUEST *request,
599                            char const *fmt, char *out, size_t outlen)
600 {
601         size_t len;
602         ssize_t ret;
603         VALUE_PAIR *vp;
604         uint8_t const *p;
605
606         while (isspace((int) *fmt)) fmt++;
607
608         if (outlen < 3) {
609         nothing:
610                 *out = '\0';
611                 return 0;
612         }
613
614         if ((radius_get_vp(&vp, request, fmt) < 0) || !vp) goto nothing;
615
616         ret = rad_vp2data(&p, vp);
617         if (ret < 0) {
618                 return ret;
619         }
620
621         switch (vp->da->type) {
622         case PW_TYPE_OCTETS:
623                 len = fr_prints(out, outlen, (char const *) p, vp->vp_length, '"');
624                 break;
625
626         case PW_TYPE_STRING:
627                 len = strlcpy(out, vp->vp_strvalue, outlen);
628                 break;
629
630         default:
631                 len = fr_prints(out, outlen, (char const *) p, ret, '\0');
632                 break;
633         }
634
635         return len;
636 }
637
638 /** xlat expand string attribute value
639  *
640  */
641 static ssize_t xlat_xlat(UNUSED void *instance, REQUEST *request,
642                         char const *fmt, char *out, size_t outlen)
643 {
644         VALUE_PAIR *vp;
645
646         while (isspace((int) *fmt)) fmt++;
647
648         if (outlen < 3) {
649         nothing:
650                 *out = '\0';
651                 return 0;
652         }
653
654         if ((radius_get_vp(&vp, request, fmt) < 0) || !vp) goto nothing;
655
656         return radius_xlat(out, outlen, request, vp->vp_strvalue, NULL, NULL);
657 }
658
659 /** Dynamically change the debugging level for the current request
660  *
661  * Example %{debug:3}
662  */
663 static ssize_t xlat_debug(UNUSED void *instance, REQUEST *request,
664                           char const *fmt, char *out, size_t outlen)
665 {
666         int level = 0;
667
668         /*
669          *  Expand to previous (or current) level
670          */
671         snprintf(out, outlen, "%d", request->log.lvl);
672
673         /*
674          *  Assume we just want to get the current value and NOT set it to 0
675          */
676         if (!*fmt)
677                 goto done;
678
679         level = atoi(fmt);
680         if (level == 0) {
681                 request->log.lvl = RAD_REQUEST_LVL_NONE;
682                 request->log.func = NULL;
683         } else {
684                 if (level > 4) level = 4;
685
686                 request->log.lvl = level;
687                 request->log.func = vradlog_request;
688         }
689
690         done:
691         return strlen(out);
692 }
693
694 /*
695  *      Compare two xlat_t structs, based ONLY on the module name.
696  */
697 static int xlat_cmp(void const *one, void const *two)
698 {
699         xlat_t const *a = one;
700         xlat_t const *b = two;
701
702         if (a->length != b->length) {
703                 return a->length - b->length;
704         }
705
706         return memcmp(a->name, b->name, a->length);
707 }
708
709
710 /*
711  *      find the appropriate registered xlat function.
712  */
713 static xlat_t *xlat_find(char const *name)
714 {
715         xlat_t my_xlat;
716
717         strlcpy(my_xlat.name, name, sizeof(my_xlat.name));
718         my_xlat.length = strlen(my_xlat.name);
719
720         return rbtree_finddata(xlat_root, &my_xlat);
721 }
722
723
724 /** Register an xlat function.
725  *
726  * @param[in] name xlat name.
727  * @param[in] func xlat function to be called.
728  * @param[in] escape function to sanitize any sub expansions passed to the xlat function.
729  * @param[in] instance of module that's registering the xlat function.
730  * @return 0 on success, -1 on failure
731  */
732 int xlat_register(char const *name, xlat_func_t func, xlat_escape_t escape, void *instance)
733 {
734         xlat_t  *c;
735         xlat_t  my_xlat;
736         rbnode_t *node;
737
738         if (!name || !*name) {
739                 DEBUG("xlat_register: Invalid xlat name");
740                 return -1;
741         }
742
743         /*
744          *      First time around, build up the tree...
745          *
746          *      FIXME: This code should be hoisted out of this function,
747          *      and into a global "initialization".  But it isn't critical...
748          */
749         if (!xlat_root) {
750 #ifdef WITH_UNLANG
751                 int i;
752 #endif
753
754                 xlat_root = rbtree_create(NULL, xlat_cmp, NULL, RBTREE_FLAG_REPLACE);
755                 if (!xlat_root) {
756                         DEBUG("xlat_register: Failed to create tree");
757                         return -1;
758                 }
759
760 #ifdef WITH_UNLANG
761                 for (i = 0; xlat_foreach_names[i] != NULL; i++) {
762                         xlat_register(xlat_foreach_names[i],
763                                       xlat_foreach, NULL, &xlat_inst[i]);
764                         c = xlat_find(xlat_foreach_names[i]);
765                         rad_assert(c != NULL);
766                         c->internal = true;
767                 }
768 #endif
769
770 #define XLAT_REGISTER(_x) xlat_register(STRINGIFY(_x), xlat_ ## _x, NULL, NULL); \
771                 c = xlat_find(STRINGIFY(_x)); \
772                 rad_assert(c != NULL); \
773                 c->internal = true
774
775                 XLAT_REGISTER(integer);
776                 XLAT_REGISTER(strlen);
777                 XLAT_REGISTER(length);
778                 XLAT_REGISTER(hex);
779                 XLAT_REGISTER(tag);
780                 XLAT_REGISTER(vendor);
781                 XLAT_REGISTER(vendor_num);
782                 XLAT_REGISTER(attr);
783                 XLAT_REGISTER(attr_num);
784                 XLAT_REGISTER(string);
785                 XLAT_REGISTER(xlat);
786                 XLAT_REGISTER(map);
787                 XLAT_REGISTER(module);
788                 XLAT_REGISTER(debug_attr);
789 #if defined(HAVE_REGEX) && defined(HAVE_PCRE)
790                 XLAT_REGISTER(regex);
791 #endif
792
793                 xlat_register("debug", xlat_debug, NULL, &xlat_inst[0]);
794                 c = xlat_find("debug");
795                 rad_assert(c != NULL);
796                 c->internal = true;
797         }
798
799         /*
800          *      If it already exists, replace the instance.
801          */
802         strlcpy(my_xlat.name, name, sizeof(my_xlat.name));
803         my_xlat.length = strlen(my_xlat.name);
804         c = rbtree_finddata(xlat_root, &my_xlat);
805         if (c) {
806                 if (c->internal) {
807                         DEBUG("xlat_register: Cannot re-define internal xlat");
808                         return -1;
809                 }
810
811                 c->func = func;
812                 c->escape = escape;
813                 c->instance = instance;
814                 return 0;
815         }
816
817         /*
818          *      Doesn't exist.  Create it.
819          */
820         c = talloc_zero(xlat_root, xlat_t);
821
822         c->func = func;
823         c->escape = escape;
824         strlcpy(c->name, name, sizeof(c->name));
825         c->length = strlen(c->name);
826         c->instance = instance;
827
828         node = rbtree_insert_node(xlat_root, c);
829         if (!node) {
830                 talloc_free(c);
831                 return -1;
832         }
833
834         /*
835          *      Ensure that the data is deleted when the node is
836          *      deleted.
837          *
838          *      @todo: Maybe this should be the other way around...
839          *      when a thing IN the tree is deleted, it's automatically
840          *      removed from the tree.  But for now, this works.
841          */
842         (void) talloc_steal(node, c);
843         return 0;
844 }
845
846 /** Unregister an xlat function
847  *
848  * We can only have one function to call per name, so the passing of "func"
849  * here is extraneous.
850  *
851  * @param[in] name xlat to unregister.
852  * @param[in] func unused.
853  * @param[in] instance data.
854  */
855 void xlat_unregister(char const *name, UNUSED xlat_func_t func, void *instance)
856 {
857         xlat_t  *c;
858         xlat_t          my_xlat;
859
860         if (!name || !xlat_root) return;
861
862         strlcpy(my_xlat.name, name, sizeof(my_xlat.name));
863         my_xlat.length = strlen(my_xlat.name);
864
865         c = rbtree_finddata(xlat_root, &my_xlat);
866         if (!c) return;
867
868         if (c->instance != instance) return;
869
870         rbtree_deletebydata(xlat_root, c);
871 }
872
873 static int xlat_unregister_callback(void *instance, void *data)
874 {
875         xlat_t *c = (xlat_t *) data;
876
877         if (c->instance != instance) return 0; /* keep walking */
878
879         return 2;               /* delete it */
880 }
881
882 void xlat_unregister_module(void *instance)
883 {
884         rbtree_walk(xlat_root, RBTREE_DELETE_ORDER, xlat_unregister_callback, instance);
885 }
886
887 /*
888  *      Internal redundant handler for xlats
889  */
890 typedef enum xlat_redundant_type_t {
891         XLAT_INVALID = 0,
892         XLAT_REDUNDANT,
893         XLAT_LOAD_BALANCE,
894         XLAT_REDUNDANT_LOAD_BALANCE,
895 } xlat_redundant_type_t;
896
897 typedef struct xlat_redundant_t {
898         xlat_redundant_type_t type;
899         uint32_t        count;
900         CONF_SECTION *cs;
901 } xlat_redundant_t;
902
903
904 static ssize_t xlat_redundant(void *instance, REQUEST *request,
905                               char const *fmt, char *out, size_t outlen)
906 {
907         xlat_redundant_t *xr = instance;
908         CONF_ITEM *ci;
909         char const *name;
910         xlat_t *xlat;
911
912         rad_assert(xr->type == XLAT_REDUNDANT);
913
914         /*
915          *      Pick the first xlat which succeeds
916          */
917         for (ci = cf_item_find_next(xr->cs, NULL);
918              ci != NULL;
919              ci = cf_item_find_next(xr->cs, ci)) {
920                 ssize_t rcode;
921
922                 if (!cf_item_is_pair(ci)) continue;
923
924                 name = cf_pair_attr(cf_item_to_pair(ci));
925                 rad_assert(name != NULL);
926
927                 xlat = xlat_find(name);
928                 if (!xlat) continue;
929
930                 rcode = xlat->func(xlat->instance, request, fmt, out, outlen);
931                 if (rcode <= 0) continue;
932                 return rcode;
933         }
934
935         /*
936          *      Everything failed.  Oh well.
937          */
938         *out  = 0;
939         return 0;
940 }
941
942
943 static ssize_t xlat_load_balance(void *instance, REQUEST *request,
944                               char const *fmt, char *out, size_t outlen)
945 {
946         uint32_t count = 0;
947         xlat_redundant_t *xr = instance;
948         CONF_ITEM *ci;
949         CONF_ITEM *found = NULL;
950         char const *name;
951         xlat_t *xlat;
952
953         /*
954          *      Choose a child at random.
955          */
956         for (ci = cf_item_find_next(xr->cs, NULL);
957              ci != NULL;
958              ci = cf_item_find_next(xr->cs, ci)) {
959                 if (!cf_item_is_pair(ci)) continue;
960                 count++;
961
962                 /*
963                  *      Replace the previously found one with a random
964                  *      new one.
965                  */
966                 if ((count * (fr_rand() & 0xffff)) < (uint32_t) 0x10000) {
967                         found = ci;
968                 }
969         }
970
971         /*
972          *      Plain load balancing: do one child, and only one child.
973          */
974         if (xr->type == XLAT_LOAD_BALANCE) {
975                 name = cf_pair_attr(cf_item_to_pair(found));
976                 rad_assert(name != NULL);
977
978                 xlat = xlat_find(name);
979                 if (!xlat) return -1;
980
981                 return xlat->func(xlat->instance, request, fmt, out, outlen);
982         }
983
984         rad_assert(xr->type == XLAT_REDUNDANT_LOAD_BALANCE);
985
986         /*
987          *      Try the random one we found.  If it fails, keep going
988          *      through the rest of the children.
989          */
990         ci = found;
991         do {
992                 name = cf_pair_attr(cf_item_to_pair(ci));
993                 rad_assert(name != NULL);
994
995                 xlat = xlat_find(name);
996                 if (xlat) {
997                         ssize_t rcode;
998
999                         rcode = xlat->func(xlat->instance, request, fmt, out, outlen);
1000                         if (rcode > 0) return rcode;
1001                 }
1002
1003                 /*
1004                  *      Go to the next one, wrapping around at the end.
1005                  */
1006                 ci = cf_item_find_next(xr->cs, ci);
1007                 if (!ci) ci = cf_item_find_next(xr->cs, NULL);
1008         } while (ci != found);
1009
1010         return -1;
1011 }
1012
1013
1014 bool xlat_register_redundant(CONF_SECTION *cs)
1015 {
1016         char const *name1, *name2;
1017         xlat_redundant_t *xr;
1018
1019         name1 = cf_section_name1(cs);
1020         name2 = cf_section_name2(cs);
1021
1022         if (!name2) return false;
1023
1024         if (xlat_find(name2)) {
1025                 cf_log_err_cs(cs, "An expansion is already registered for this name");
1026                 return false;
1027         }
1028
1029         xr = talloc_zero(cs, xlat_redundant_t);
1030         if (!xr) return false;
1031
1032         if (strcmp(name1, "redundant") == 0) {
1033                 xr->type = XLAT_REDUNDANT;
1034
1035         } else if (strcmp(name1, "redundant-load-balance") == 0) {
1036                 xr->type = XLAT_REDUNDANT_LOAD_BALANCE;
1037
1038         } else if (strcmp(name1, "load-balance") == 0) {
1039                 xr->type = XLAT_LOAD_BALANCE;
1040
1041         } else {
1042                 return false;
1043         }
1044
1045         xr->cs = cs;
1046
1047         /*
1048          *      Get the number of children for load balancing.
1049          */
1050         if (xr->type == XLAT_REDUNDANT) {
1051                 if (xlat_register(name2, xlat_redundant, NULL, xr) < 0) {
1052                         talloc_free(xr);
1053                         return false;
1054                 }
1055
1056         } else {
1057                 CONF_ITEM *ci;
1058
1059                 for (ci = cf_item_find_next(cs, NULL);
1060                      ci != NULL;
1061                      ci = cf_item_find_next(cs, ci)) {
1062                         if (!cf_item_is_pair(ci)) continue;
1063
1064                         if (!xlat_find(cf_pair_attr(cf_item_to_pair(ci)))) {
1065                                 talloc_free(xr);
1066                                 return false;
1067                         }
1068
1069                         xr->count++;
1070                 }
1071
1072                 if (xlat_register(name2, xlat_load_balance, NULL, xr) < 0) {
1073                         talloc_free(xr);
1074                         return false;
1075                 }
1076         }
1077
1078         return true;
1079 }
1080
1081
1082 /** Crappy temporary function to add attribute ref support to xlats
1083  *
1084  * This needs to die, and hopefully will die, when xlat functions accept
1085  * xlat node structures.
1086  *
1087  * Provides either a pointer to a buffer which contains the value of the reference VALUE_PAIR
1088  * in an architecture independent format. Or a pointer to the start of the fmt string.
1089  *
1090  * The pointer is only guaranteed to be valid between calls to xlat_fmt_to_ref,
1091  * and so long as the source VALUE_PAIR is not freed.
1092  *
1093  * @param out where to write a pointer to the buffer to the data the xlat function needs to work on.
1094  * @param request current request.
1095  * @param fmt string.
1096  * @returns the length of the data or -1 on error.
1097  */
1098 ssize_t xlat_fmt_to_ref(uint8_t const **out, REQUEST *request, char const *fmt)
1099 {
1100         VALUE_PAIR *vp;
1101
1102         while (isspace((int) *fmt)) fmt++;
1103
1104         if (fmt[0] == '&') {
1105                 if ((radius_get_vp(&vp, request, fmt) < 0) || !vp) {
1106                         *out = NULL;
1107                         return -1;
1108                 }
1109
1110                 return rad_vp2data(out, vp);
1111         }
1112
1113         *out = (uint8_t const *)fmt;
1114         return strlen(fmt);
1115 }
1116
1117 /** De-register all xlat functions, used mainly for debugging.
1118  *
1119  */
1120 void xlat_free(void)
1121 {
1122         rbtree_free(xlat_root);
1123 }
1124
1125 #ifdef DEBUG_XLAT
1126 #  define XLAT_DEBUG DEBUG3
1127 #else
1128 #  define XLAT_DEBUG(...)
1129 #endif
1130
1131 static ssize_t xlat_tokenize_expansion(TALLOC_CTX *ctx, char *fmt, xlat_exp_t **head,
1132                                        char const **error);
1133 static ssize_t xlat_tokenize_literal(TALLOC_CTX *ctx, char *fmt, xlat_exp_t **head,
1134                                      bool brace, char const **error);
1135 static size_t xlat_process(char **out, REQUEST *request, xlat_exp_t const * const head,
1136                            xlat_escape_t escape, void *escape_ctx);
1137
1138 static ssize_t xlat_tokenize_alternation(TALLOC_CTX *ctx, char *fmt, xlat_exp_t **head,
1139                                          char const **error)
1140 {
1141         ssize_t slen;
1142         char *p;
1143         xlat_exp_t *node;
1144
1145         rad_assert(fmt[0] == '%');
1146         rad_assert(fmt[1] == '{');
1147         rad_assert(fmt[2] == '%');
1148         rad_assert(fmt[3] == '{');
1149
1150         XLAT_DEBUG("ALTERNATE <-- %s", fmt);
1151
1152         node = talloc_zero(ctx, xlat_exp_t);
1153         node->type = XLAT_ALTERNATE;
1154
1155         p = fmt + 2;
1156         slen = xlat_tokenize_expansion(node, p, &node->child, error);
1157         if (slen <= 0) {
1158                 talloc_free(node);
1159                 return slen - (p - fmt);
1160         }
1161         p += slen;
1162
1163         if (p[0] != ':') {
1164                 talloc_free(node);
1165                 *error = "Expected ':' after first expansion";
1166                 return -(p - fmt);
1167         }
1168         p++;
1169
1170         if (p[0] != '-') {
1171                 talloc_free(node);
1172                 *error = "Expected '-' after ':'";
1173                 return -(p - fmt);
1174         }
1175         p++;
1176
1177         /*
1178          *      Allow the RHS to be empty as a special case.
1179          */
1180         if (*p == '}') {
1181                 /*
1182                  *      Hack up an empty string.
1183                  */
1184                 node->alternate = talloc_zero(node, xlat_exp_t);
1185                 node->alternate->type = XLAT_LITERAL;
1186                 node->alternate->fmt = talloc_typed_strdup(node->alternate, "");
1187                 *(p++) = '\0';
1188
1189         } else {
1190                 slen = xlat_tokenize_literal(node, p,  &node->alternate, true, error);
1191                 if (slen <= 0) {
1192                         talloc_free(node);
1193                         return slen - (p - fmt);
1194                 }
1195
1196                 if (!node->alternate) {
1197                         talloc_free(node);
1198                         *error = "Empty expansion is invalid";
1199                         return -(p - fmt);
1200                 }
1201                 p += slen;
1202         }
1203
1204         *head = node;
1205         return p - fmt;
1206 }
1207
1208 static ssize_t xlat_tokenize_expansion(TALLOC_CTX *ctx, char *fmt, xlat_exp_t **head,
1209                                        char const **error)
1210 {
1211         ssize_t slen;
1212         char *p, *q;
1213         xlat_exp_t *node;
1214         long num;
1215
1216         rad_assert(fmt[0] == '%');
1217         rad_assert(fmt[1] == '{');
1218
1219         /*
1220          *      %{%{...}:-bar}
1221          */
1222         if ((fmt[2] == '%') && (fmt[3] == '{')) return xlat_tokenize_alternation(ctx, fmt, head, error);
1223
1224         XLAT_DEBUG("EXPANSION <-- %s", fmt);
1225         node = talloc_zero(ctx, xlat_exp_t);
1226         node->fmt = fmt + 2;
1227         node->len = 0;
1228
1229 #ifdef HAVE_REGEX
1230         /*
1231          *      Handle regex's specially.
1232          */
1233         p = fmt + 2;
1234         num = strtol(p, &q, 10);
1235         if (p != q && (*q == '}')) {
1236                 XLAT_DEBUG("REGEX <-- %s", fmt);
1237                 *q = '\0';
1238
1239                 if ((num > REQUEST_MAX_REGEX) || (num < 0)) {
1240                         talloc_free(node);
1241                         *error = "Invalid regex reference.  Must be in range 0-" STRINGIFY(REQUEST_MAX_REGEX);
1242                         return -2;
1243                 }
1244                 node->attr.tmpl_num = num;
1245
1246                 node->type = XLAT_REGEX;
1247                 *head = node;
1248
1249                 return (q - fmt) + 1;
1250         }
1251 #endif /* HAVE_REGEX */
1252
1253         /*
1254          *      %{Attr-Name}
1255          *      %{Attr-Name[#]}
1256          *      %{Tunnel-Password:1}
1257          *      %{Tunnel-Password:1[#]}
1258          *      %{request:Attr-Name}
1259          *      %{request:Tunnel-Password:1}
1260          *      %{request:Tunnel-Password:1[#]}
1261          *      %{mod:foo}
1262          */
1263
1264         /*
1265          *      This is for efficiency, so we don't search for an xlat,
1266          *      when what's being referenced is obviously an attribute.
1267          */
1268         p = fmt + 2;
1269         for (q = p; *q != '\0'; q++) {
1270                 if (*q == ':') break;
1271
1272                 if (isspace((int) *q)) break;
1273
1274                 if (*q == '[') continue;
1275
1276                 if (*q == '}') break;
1277         }
1278
1279         /*
1280          *      Check for empty expressions %{}
1281          */
1282         if ((*q == '}') && (q == p)) {
1283                 talloc_free(node);
1284                 *error = "Empty expression is invalid";
1285                 return -(p - fmt);
1286         }
1287
1288         /*
1289          *      Might be a module name reference.
1290          *
1291          *      If it's not, it's an attribute or parse error.
1292          */
1293         if (*q == ':') {
1294                 *q = '\0';
1295                 node->xlat = xlat_find(node->fmt);
1296                 if (node->xlat) {
1297                         /*
1298                          *      %{mod:foo}
1299                          */
1300                         node->type = XLAT_MODULE;
1301
1302                         p = q + 1;
1303                         XLAT_DEBUG("MOD <-- %s ... %s", node->fmt, p);
1304
1305                         slen = xlat_tokenize_literal(node, p, &node->child, true, error);
1306                         if (slen < 0) {
1307                                 talloc_free(node);
1308                                 return slen - (p - fmt);
1309                         }
1310                         p += slen;
1311
1312                         *head = node;
1313                         rad_assert(node->next == NULL);
1314
1315                         return p - fmt;
1316                 }
1317                 *q = ':';       /* Avoids a strdup */
1318         }
1319
1320         /*
1321          *      The first token ends with:
1322          *      - '[' - Which is an attribute index, so it must be an attribute.
1323          *      - '}' - The end of the expansion, which means it was a bareword.
1324          */
1325         slen = tmpl_from_attr_substr(&node->attr, p, REQUEST_CURRENT, PAIR_LIST_REQUEST, true, true);
1326         if (slen <= 0) {
1327                 /*
1328                  *      If the parse error occurred before the ':'
1329                  *      then the error is changed to 'Unknown module',
1330                  *      as it was more likely to be a bad module name,
1331                  *      than a request qualifier.
1332                  */
1333                 if ((*q == ':') && ((p + (slen * -1)) < q)) {
1334                         *error = "Unknown module";
1335                 } else {
1336                         *error = fr_strerror();
1337                 }
1338
1339                 talloc_free(node);
1340                 return slen - (p - fmt);
1341         }
1342
1343         /*
1344          *      Might be a virtual XLAT attribute
1345          */
1346         if (node->attr.type == TMPL_TYPE_ATTR_UNDEFINED) {
1347                 node->xlat = xlat_find(node->attr.tmpl_unknown_name);
1348                 if (node->xlat && node->xlat->instance && !node->xlat->internal) {
1349                         talloc_free(node);
1350                         *error = "Missing content in expansion";
1351                         return -(p - fmt) - slen;
1352                 }
1353
1354                 if (node->xlat) {
1355                         node->type = XLAT_VIRTUAL;
1356                         node->fmt = node->attr.tmpl_unknown_name;
1357
1358                         XLAT_DEBUG("VIRTUAL <-- %s", node->fmt);
1359                         *head = node;
1360                         rad_assert(node->next == NULL);
1361                         q++;
1362                         return q - fmt;
1363                 }
1364
1365                 talloc_free(node);
1366                 *error = "Unknown attribute";
1367                 return -(p - fmt);
1368         }
1369
1370         node->type = XLAT_ATTRIBUTE;
1371         p += slen;
1372
1373         if (*p != '}') {
1374                 talloc_free(node);
1375                 *error = "No matching closing brace";
1376                 return -1;      /* second character of format string */
1377         }
1378         *p++ = '\0';
1379         *head = node;
1380         rad_assert(node->next == NULL);
1381
1382         return p - fmt;
1383 }
1384
1385
1386 static ssize_t xlat_tokenize_literal(TALLOC_CTX *ctx, char *fmt, xlat_exp_t **head,
1387                                      bool brace, char const **error)
1388 {
1389         char *p;
1390         xlat_exp_t *node;
1391
1392         if (!*fmt) return 0;
1393
1394         XLAT_DEBUG("LITERAL <-- %s", fmt);
1395
1396         node = talloc_zero(ctx, xlat_exp_t);
1397         node->fmt = fmt;
1398         node->len = 0;
1399         node->type = XLAT_LITERAL;
1400
1401         p = fmt;
1402
1403         while (*p) {
1404                 if (*p == '\\') {
1405                         if (!p[1]) {
1406                                 talloc_free(node);
1407                                 *error = "Invalid escape at end of string";
1408                                 return -(p - fmt);
1409                         }
1410
1411                         p += 2;
1412                         node->len += 2;
1413                         continue;
1414                 }
1415
1416                 /*
1417                  *      Process the expansion.
1418                  */
1419                 if ((p[0] == '%') && (p[1] == '{')) {
1420                         ssize_t slen;
1421
1422                         XLAT_DEBUG("EXPANSION-2 <-- %s", node->fmt);
1423
1424                         slen = xlat_tokenize_expansion(node, p, &node->next, error);
1425                         if (slen <= 0) {
1426                                 talloc_free(node);
1427                                 return slen - (p - fmt);
1428                         }
1429                         *p = '\0'; /* end the literal */
1430                         p += slen;
1431
1432                         rad_assert(node->next != NULL);
1433
1434                         /*
1435                          *      Short-circuit the recursive call.
1436                          *      This saves another function call and
1437                          *      memory allocation.
1438                          */
1439                         if (!*p) break;
1440
1441                         /*
1442                          *      "foo %{User-Name} bar"
1443                          *      LITERAL         "foo "
1444                          *      EXPANSION       User-Name
1445                          *      LITERAL         " bar"
1446                          */
1447                         slen = xlat_tokenize_literal(node->next, p, &(node->next->next), brace, error);
1448                         rad_assert(slen != 0);
1449                         if (slen < 0) {
1450                                 talloc_free(node);
1451                                 return slen - (p - fmt);
1452                         }
1453
1454                         brace = false; /* it was found above, or else the above code errored out */
1455                         p += slen;
1456                         break;  /* stop processing the string */
1457                 }
1458
1459                 /*
1460                  *      Check for valid single-character expansions.
1461                  */
1462                 if (p[0] == '%') {
1463                         ssize_t slen;
1464                         xlat_exp_t *next;
1465
1466                         if (!p[1] || !strchr("%}dlmntDGHISTYv", p[1])) {
1467                                 talloc_free(node);
1468                                 *error = "Invalid variable expansion";
1469                                 p++;
1470                                 return - (p - fmt);
1471                         }
1472
1473                         next = talloc_zero(node, xlat_exp_t);
1474                         next->len = 1;
1475
1476                         switch (p[1]) {
1477                         case '%':
1478                         case '}':
1479                                 next->fmt = talloc_strndup(next, p + 1, 1);
1480
1481                                 XLAT_DEBUG("LITERAL-ESCAPED <-- %s", next->fmt);
1482                                 next->type = XLAT_LITERAL;
1483                                 break;
1484
1485                         default:
1486                                 next->fmt = p + 1;
1487
1488                                 XLAT_DEBUG("PERCENT <-- %c", *next->fmt);
1489                                 next->type = XLAT_PERCENT;
1490                                 break;
1491                         }
1492
1493                         node->next = next;
1494                         *p = '\0';
1495                         p += 2;
1496
1497                         if (!*p) break;
1498
1499                         /*
1500                          *      And recurse.
1501                          */
1502                         slen = xlat_tokenize_literal(node->next, p, &(node->next->next), brace, error);
1503                         rad_assert(slen != 0);
1504                         if (slen < 0) {
1505                                 talloc_free(node);
1506                                 return slen - (p - fmt);
1507                         }
1508
1509                         brace = false; /* it was found above, or else the above code errored out */
1510                         p += slen;
1511                         break;  /* stop processing the string */
1512                 }
1513
1514                 /*
1515                  *      If required, eat the brace.
1516                  */
1517                 if (brace && (*p == '}')) {
1518                         brace = false;
1519                         *p = '\0';
1520                         p++;
1521                         break;
1522                 }
1523
1524                 p++;
1525                 node->len++;
1526         }
1527
1528         /*
1529          *      We were told to look for a brace, but we ran off of
1530          *      the end of the string before we found one.
1531          */
1532         if (brace) {
1533                 *error = "Missing closing brace at end of string";
1534                 return -(p - fmt);
1535         }
1536
1537         /*
1538          *      Squash zero-width literals
1539          */
1540         if (node->len > 0) {
1541                 *head = node;
1542
1543         } else {
1544                 (void) talloc_steal(ctx, node->next);
1545                 *head = node->next;
1546                 talloc_free(node);
1547         }
1548
1549         return p - fmt;
1550 }
1551
1552
1553 static char const xlat_tabs[] = "                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                               ";
1554
1555 static void xlat_tokenize_debug(xlat_exp_t const *node, int lvl)
1556 {
1557         rad_assert(node != NULL);
1558
1559         if (lvl >= (int) sizeof(xlat_tabs)) lvl = sizeof(xlat_tabs);
1560
1561         while (node) {
1562                 switch (node->type) {
1563                 case XLAT_LITERAL:
1564                         DEBUG("%.*sliteral --> %s", lvl, xlat_tabs, node->fmt);
1565                         break;
1566
1567                 case XLAT_PERCENT:
1568                         DEBUG("%.*spercent --> %c", lvl, xlat_tabs, node->fmt[0]);
1569                         break;
1570
1571                 case XLAT_ATTRIBUTE:
1572                         rad_assert(node->attr.tmpl_da != NULL);
1573                         DEBUG("%.*sattribute --> %s", lvl, xlat_tabs, node->attr.tmpl_da->name);
1574                         rad_assert(node->child == NULL);
1575                         if ((node->attr.tmpl_tag != TAG_ANY) || (node->attr.tmpl_num != NUM_ANY)) {
1576                                 DEBUG("%.*s{", lvl, xlat_tabs);
1577
1578                                 DEBUG("%.*sref  %d", lvl + 1, xlat_tabs, node->attr.tmpl_request);
1579                                 DEBUG("%.*slist %d", lvl + 1, xlat_tabs, node->attr.tmpl_list);
1580
1581                                 if (node->attr.tmpl_tag != TAG_ANY) {
1582                                         DEBUG("%.*stag %d", lvl + 1, xlat_tabs, node->attr.tmpl_tag);
1583                                 }
1584                                 if (node->attr.tmpl_num != NUM_ANY) {
1585                                         if (node->attr.tmpl_num == NUM_COUNT) {
1586                                                 DEBUG("%.*s[#]", lvl + 1, xlat_tabs);
1587                                         } else if (node->attr.tmpl_num == NUM_ALL) {
1588                                                 DEBUG("%.*s[*]", lvl + 1, xlat_tabs);
1589                                         } else {
1590                                                 DEBUG("%.*s[%d]", lvl + 1, xlat_tabs, node->attr.tmpl_num);
1591                                         }
1592                                 }
1593
1594                                 DEBUG("%.*s}", lvl, xlat_tabs);
1595                         }
1596                         break;
1597
1598                 case XLAT_VIRTUAL:
1599                         rad_assert(node->fmt != NULL);
1600                         DEBUG("%.*svirtual --> %s", lvl, xlat_tabs, node->fmt);
1601                         break;
1602
1603                 case XLAT_MODULE:
1604                         rad_assert(node->xlat != NULL);
1605                         DEBUG("%.*sxlat --> %s", lvl, xlat_tabs, node->xlat->name);
1606                         if (node->child) {
1607                                 DEBUG("%.*s{", lvl, xlat_tabs);
1608                                 xlat_tokenize_debug(node->child, lvl + 1);
1609                                 DEBUG("%.*s}", lvl, xlat_tabs);
1610                         }
1611                         break;
1612
1613 #ifdef HAVE_REGEX
1614                 case XLAT_REGEX:
1615                         DEBUG("%.*sregex-var --> %d", lvl, xlat_tabs, node->attr.tmpl_num);
1616                         break;
1617 #endif
1618
1619                 case XLAT_ALTERNATE:
1620                         DEBUG("%.*sif {", lvl, xlat_tabs);
1621                         xlat_tokenize_debug(node->child, lvl + 1);
1622                         DEBUG("%.*s}", lvl, xlat_tabs);
1623                         DEBUG("%.*selse {", lvl, xlat_tabs);
1624                         xlat_tokenize_debug(node->alternate, lvl + 1);
1625                         DEBUG("%.*s}", lvl, xlat_tabs);
1626                         break;
1627                 }
1628                 node = node->next;
1629         }
1630 }
1631
1632 size_t xlat_sprint(char *buffer, size_t bufsize, xlat_exp_t const *node)
1633 {
1634         size_t len;
1635         char *p, *end;
1636
1637         if (!node) {
1638                 *buffer = '\0';
1639                 return 0;
1640         }
1641
1642         p = buffer;
1643         end = buffer + bufsize;
1644
1645         while (node) {
1646                 switch (node->type) {
1647                 case XLAT_LITERAL:
1648                         strlcpy(p, node->fmt, end - p);
1649                         p += strlen(p);
1650                         break;
1651
1652                 case XLAT_PERCENT:
1653                         p[0] = '%';
1654                         p[1] = node->fmt[0];
1655                         p += 2;
1656                         break;
1657
1658                 case XLAT_ATTRIBUTE:
1659                         *(p++) = '%';
1660                         *(p++) = '{';
1661
1662                         if (node->attr.tmpl_request != REQUEST_CURRENT) {
1663                                 strlcpy(p, fr_int2str(request_refs, node->attr.tmpl_request, "??"), end - p);
1664                                 p += strlen(p);
1665                                 *(p++) = '.';
1666                         }
1667
1668                         if ((node->attr.tmpl_request != REQUEST_CURRENT) ||
1669                             (node->attr.tmpl_list != PAIR_LIST_REQUEST)) {
1670                                 strlcpy(p, fr_int2str(pair_lists, node->attr.tmpl_list, "??"), end - p);
1671                                 p += strlen(p);
1672                                 *(p++) = ':';
1673                         }
1674
1675                         strlcpy(p, node->attr.tmpl_da->name, end - p);
1676                         p += strlen(p);
1677
1678                         if (node->attr.tmpl_tag != TAG_ANY) {
1679                                 *(p++) = ':';
1680                                 snprintf(p, end - p, "%u", node->attr.tmpl_tag);
1681                                 p += strlen(p);
1682                         }
1683
1684                         if (node->attr.tmpl_num != NUM_ANY) {
1685                                 *(p++) = '[';
1686                                 switch (node->attr.tmpl_num) {
1687                                 case NUM_COUNT:
1688                                         *(p++) = '#';
1689                                         break;
1690
1691                                 case NUM_ALL:
1692                                         *(p++) = '*';
1693                                         break;
1694
1695                                 default:
1696                                         snprintf(p, end - p, "%i", node->attr.tmpl_num);
1697                                         p += strlen(p);
1698                                 }
1699                                 *(p++) = ']';
1700                         }
1701                         *(p++) = '}';
1702                         break;
1703 #ifdef HAVE_REGEX
1704                 case XLAT_REGEX:
1705                         snprintf(p, end - p, "%%{%i}", node->attr.tmpl_num);
1706                         p += strlen(p);
1707                         break;
1708 #endif
1709                 case XLAT_VIRTUAL:
1710                         *(p++) = '%';
1711                         *(p++) = '{';
1712                         strlcpy(p, node->fmt, end - p);
1713                         p += strlen(p);
1714                         *(p++) = '}';
1715                         break;
1716
1717                 case XLAT_MODULE:
1718                         *(p++) = '%';
1719                         *(p++) = '{';
1720                         strlcpy(p, node->xlat->name, end - p);
1721                         p += strlen(p);
1722                         *(p++) = ':';
1723                         rad_assert(node->child != NULL);
1724                         len = xlat_sprint(p, end - p, node->child);
1725                         p += len;
1726                         *(p++) = '}';
1727                         break;
1728
1729                 case XLAT_ALTERNATE:
1730                         *(p++) = '%';
1731                         *(p++) = '{';
1732
1733                         len = xlat_sprint(p, end - p, node->child);
1734                         p += len;
1735
1736                         *(p++) = ':';
1737                         *(p++) = '-';
1738
1739                         len = xlat_sprint(p, end - p, node->alternate);
1740                         p += len;
1741
1742                         *(p++) = '}';
1743                         break;
1744                 }
1745
1746
1747                 if (p == end) break;
1748
1749                 node = node->next;
1750         }
1751
1752         *p = '\0';
1753
1754         return p - buffer;
1755 }
1756
1757 ssize_t xlat_tokenize(TALLOC_CTX *ctx, char *fmt, xlat_exp_t **head,
1758                       char const **error)
1759 {
1760         return xlat_tokenize_literal(ctx, fmt, head, false, error);
1761 }
1762
1763
1764 /** Tokenize an xlat expansion
1765  *
1766  * @param[in] request the input request.  Memory will be attached here.
1767  * @param[in] fmt the format string to expand
1768  * @param[out] head the head of the xlat list / tree structure.
1769  */
1770 static ssize_t xlat_tokenize_request(REQUEST *request, char const *fmt, xlat_exp_t **head)
1771 {
1772         ssize_t slen;
1773         char *tokens;
1774         char const *error = NULL;
1775
1776         *head = NULL;
1777
1778         /*
1779          *      Copy the original format string to a buffer so that
1780          *      the later functions can mangle it in-place, which is
1781          *      much faster.
1782          */
1783         tokens = talloc_typed_strdup(request, fmt);
1784         if (!tokens) return -1;
1785
1786         slen = xlat_tokenize_literal(request, tokens, head, false, &error);
1787
1788         /*
1789          *      Zero length expansion, return a zero length node.
1790          */
1791         if (slen == 0) {
1792                 *head = talloc_zero(request, xlat_exp_t);
1793         }
1794
1795         /*
1796          *      Output something like:
1797          *
1798          *      "format string"
1799          *      "       ^ error was here"
1800          */
1801         if (slen < 0) {
1802                 talloc_free(tokens);
1803                 rad_assert(error != NULL);
1804
1805                 REMARKER(fmt, -slen, error);
1806                 return slen;
1807         }
1808
1809         if (*head && (rad_debug_lvl > 2)) {
1810                 DEBUG("%s", fmt);
1811                 DEBUG("Parsed xlat tree:");
1812                 xlat_tokenize_debug(*head, 0);
1813         }
1814
1815         /*
1816          *      All of the nodes point to offsets in the "tokens"
1817          *      string.  Let's ensure that free'ing head will free
1818          *      "tokens", too.
1819          */
1820         (void) talloc_steal(*head, tokens);
1821
1822         return slen;
1823 }
1824
1825
1826 static char *xlat_getvp(TALLOC_CTX *ctx, REQUEST *request, vp_tmpl_t const *vpt,
1827                         bool escape, bool return_null)
1828 {
1829         VALUE_PAIR *vp = NULL, *virtual = NULL;
1830         RADIUS_PACKET *packet = NULL;
1831         DICT_VALUE *dv;
1832         char *ret = NULL;
1833
1834         vp_cursor_t cursor;
1835         char quote = escape ? '"' : '\0';
1836
1837         rad_assert((vpt->type == TMPL_TYPE_ATTR) || (vpt->type == TMPL_TYPE_LIST));
1838
1839         /*
1840          *      We only support count and concatenate operations on lists.
1841          */
1842         if (vpt->type == TMPL_TYPE_LIST) {
1843                 vp = tmpl_cursor_init(NULL, &cursor, request, vpt);
1844                 goto do_print;
1845         }
1846
1847         /*
1848          *      See if we're dealing with an attribute in the request
1849          *
1850          *      This allows users to manipulate virtual attributes as if
1851          *      they were real ones.
1852          */
1853         vp = tmpl_cursor_init(NULL, &cursor, request, vpt);
1854         if (vp) goto do_print;
1855
1856         /*
1857          *      We didn't find the VP in a list.
1858          *      If it's not a virtual one, and we're not meant to
1859          *      be counting it, return.
1860          */
1861         if (!vpt->tmpl_da->flags.virtual) {
1862                 if (vpt->tmpl_num == NUM_COUNT) goto do_print;
1863                 return NULL;
1864         }
1865
1866         /*
1867          *      Switch out the request to the one specified by the template
1868          */
1869         if (radius_request(&request, vpt->tmpl_request) < 0) return NULL;
1870
1871         /*
1872          *      Some non-packet expansions
1873          */
1874         switch (vpt->tmpl_da->attr) {
1875         default:
1876                 break;          /* ignore them */
1877
1878         case PW_CLIENT_SHORTNAME:
1879                 if (vpt->tmpl_num == NUM_COUNT) goto count_virtual;
1880                 if (request->client && request->client->shortname) {
1881                         return talloc_typed_strdup(ctx, request->client->shortname);
1882                 }
1883                 return talloc_typed_strdup(ctx, "<UNKNOWN-CLIENT>");
1884
1885         case PW_REQUEST_PROCESSING_STAGE:
1886                 if (vpt->tmpl_num == NUM_COUNT) goto count_virtual;
1887                 if (request->component) {
1888                         return talloc_typed_strdup(ctx, request->component);
1889                 }
1890                 return talloc_typed_strdup(ctx, "server_core");
1891
1892         case PW_VIRTUAL_SERVER:
1893                 if (vpt->tmpl_num == NUM_COUNT) goto count_virtual;
1894                 if (!request->server) return NULL;
1895                 return talloc_typed_strdup(ctx, request->server);
1896
1897         case PW_MODULE_RETURN_CODE:
1898                 if (vpt->tmpl_num == NUM_COUNT) goto count_virtual;
1899                 if (!request->rcode) return NULL;
1900                 return talloc_typed_strdup(ctx, fr_int2str(modreturn_table, request->rcode, ""));
1901         }
1902
1903         /*
1904          *      All of the attributes must now refer to a packet.
1905          *      If there's no packet, we can't print any attribute
1906          *      referencing it.
1907          */
1908         packet = radius_packet(request, vpt->tmpl_list);
1909         if (!packet) {
1910                 if (return_null) return NULL;
1911                 return vp_aprints_type(ctx, vpt->tmpl_da->type);
1912         }
1913
1914         vp = NULL;
1915         switch (vpt->tmpl_da->attr) {
1916         default:
1917                 break;
1918
1919         case PW_PACKET_TYPE:
1920                 dv = dict_valbyattr(PW_PACKET_TYPE, 0, packet->code);
1921                 if (dv) return talloc_typed_strdup(ctx, dv->name);
1922                 return talloc_typed_asprintf(ctx, "%d", packet->code);
1923
1924         case PW_RESPONSE_PACKET_TYPE:
1925         {
1926                 int code = 0;
1927
1928 #ifdef WITH_PROXY
1929                 if (request->proxy_reply && (!request->reply || !request->reply->code)) {
1930                         code = request->proxy_reply->code;
1931                 } else
1932 #endif
1933                         if (request->reply) {
1934                                 code = request->reply->code;
1935                         }
1936
1937                 return talloc_typed_strdup(ctx, fr_packet_codes[code]);
1938         }
1939
1940         /*
1941          *      Virtual attributes which require a temporary VALUE_PAIR
1942          *      to be allocated. We can't use stack allocated memory
1943          *      because of the talloc checks sprinkled throughout the
1944          *      various VP functions.
1945          */
1946         case PW_PACKET_AUTHENTICATION_VECTOR:
1947                 virtual = fr_pair_afrom_da(ctx, vpt->tmpl_da);
1948                 fr_pair_value_memcpy(virtual, packet->vector, sizeof(packet->vector));
1949                 vp = virtual;
1950                 break;
1951
1952         case PW_CLIENT_IP_ADDRESS:
1953         case PW_PACKET_SRC_IP_ADDRESS:
1954                 if (packet->src_ipaddr.af == AF_INET) {
1955                         virtual = fr_pair_afrom_da(ctx, vpt->tmpl_da);
1956                         virtual->vp_ipaddr = packet->src_ipaddr.ipaddr.ip4addr.s_addr;
1957                         vp = virtual;
1958                 }
1959                 break;
1960
1961         case PW_PACKET_DST_IP_ADDRESS:
1962                 if (packet->dst_ipaddr.af == AF_INET) {
1963                         virtual = fr_pair_afrom_da(ctx, vpt->tmpl_da);
1964                         virtual->vp_ipaddr = packet->dst_ipaddr.ipaddr.ip4addr.s_addr;
1965                         vp = virtual;
1966                 }
1967                 break;
1968
1969         case PW_PACKET_SRC_IPV6_ADDRESS:
1970                 if (packet->src_ipaddr.af == AF_INET6) {
1971                         virtual = fr_pair_afrom_da(ctx, vpt->tmpl_da);
1972                         memcpy(&virtual->vp_ipv6addr,
1973                                &packet->src_ipaddr.ipaddr.ip6addr,
1974                                sizeof(packet->src_ipaddr.ipaddr.ip6addr));
1975                         vp = virtual;
1976                 }
1977                 break;
1978
1979         case PW_PACKET_DST_IPV6_ADDRESS:
1980                 if (packet->dst_ipaddr.af == AF_INET6) {
1981                         virtual = fr_pair_afrom_da(ctx, vpt->tmpl_da);
1982                         memcpy(&virtual->vp_ipv6addr,
1983                                &packet->dst_ipaddr.ipaddr.ip6addr,
1984                                sizeof(packet->dst_ipaddr.ipaddr.ip6addr));
1985                         vp = virtual;
1986                 }
1987                 break;
1988
1989         case PW_PACKET_SRC_PORT:
1990                 virtual = fr_pair_afrom_da(ctx, vpt->tmpl_da);
1991                 virtual->vp_integer = packet->src_port;
1992                 vp = virtual;
1993                 break;
1994
1995         case PW_PACKET_DST_PORT:
1996                 virtual = fr_pair_afrom_da(ctx, vpt->tmpl_da);
1997                 virtual->vp_integer = packet->dst_port;
1998                 vp = virtual;
1999                 break;
2000         }
2001
2002         /*
2003          *      Fake various operations for virtual attributes.
2004          */
2005         if (virtual) {
2006                 if (vpt->tmpl_num != NUM_ANY) switch (vpt->tmpl_num) {
2007                 /*
2008                  *      [n] is NULL (we only have [0])
2009                  */
2010                 default:
2011                         goto finish;
2012                 /*
2013                  *      [*] means only one.
2014                  */
2015                 case NUM_ALL:
2016                         break;
2017
2018                 /*
2019                  *      [#] means 1 (as there's only one)
2020                  */
2021                 case NUM_COUNT:
2022                 count_virtual:
2023                         ret = talloc_strdup(ctx, "1");
2024                         goto finish;
2025
2026                 /*
2027                  *      [0] is fine (get the first instance)
2028                  */
2029                 case 0:
2030                         break;
2031                 }
2032                 goto print;
2033         }
2034
2035 do_print:
2036         switch (vpt->tmpl_num) {
2037         /*
2038          *      Return a count of the VPs.
2039          */
2040         case NUM_COUNT:
2041         {
2042                 int count = 0;
2043
2044                 for (vp = tmpl_cursor_init(NULL, &cursor, request, vpt);
2045                      vp;
2046                      vp = tmpl_cursor_next(&cursor, vpt)) count++;
2047
2048                 return talloc_typed_asprintf(ctx, "%d", count);
2049         }
2050
2051
2052         /*
2053          *      Concatenate all values together,
2054          *      separated by commas.
2055          */
2056         case NUM_ALL:
2057         {
2058                 char *p, *q;
2059
2060                 if (!fr_cursor_current(&cursor)) return NULL;
2061                 p = vp_aprints_value(ctx, vp, quote);
2062                 if (!p) return NULL;
2063
2064                 while ((vp = tmpl_cursor_next(&cursor, vpt)) != NULL) {
2065                         q = vp_aprints_value(ctx, vp, quote);
2066                         if (!q) return NULL;
2067                         p = talloc_strdup_append(p, ",");
2068                         p = talloc_strdup_append(p, q);
2069                 }
2070
2071                 return p;
2072         }
2073
2074         default:
2075                 /*
2076                  *      The cursor was set to the correct
2077                  *      position above by tmpl_cursor_init.
2078                  */
2079                 vp = fr_cursor_current(&cursor);
2080                 break;
2081         }
2082
2083         if (!vp) {
2084                 if (return_null) return NULL;
2085                 return vp_aprints_type(ctx, vpt->tmpl_da->type);
2086         }
2087
2088 print:
2089         ret = vp_aprints_value(ctx, vp, quote);
2090
2091 finish:
2092         talloc_free(virtual);
2093         return ret;
2094 }
2095
2096 #ifdef DEBUG_XLAT
2097 static const char xlat_spaces[] = "                                                                                                                                                                                                                                                                ";
2098 #endif
2099
2100 static char *xlat_aprint(TALLOC_CTX *ctx, REQUEST *request, xlat_exp_t const * const node,
2101                          xlat_escape_t escape, void *escape_ctx, int lvl)
2102 {
2103         ssize_t rcode;
2104         char *str = NULL, *child;
2105         char const *p;
2106
2107         XLAT_DEBUG("%.*sxlat aprint %d %s", lvl, xlat_spaces, node->type, node->fmt);
2108
2109         switch (node->type) {
2110                 /*
2111                  *      Don't escape this.
2112                  */
2113         case XLAT_LITERAL:
2114                 XLAT_DEBUG("xlat_aprint LITERAL");
2115                 return talloc_typed_strdup(ctx, node->fmt);
2116
2117                 /*
2118                  *      Do a one-character expansion.
2119                  */
2120         case XLAT_PERCENT:
2121         {
2122                 char *nl;
2123                 size_t freespace = 256;
2124                 struct tm ts;
2125                 time_t when;
2126
2127                 XLAT_DEBUG("xlat_aprint PERCENT");
2128
2129                 str = talloc_array(ctx, char, freespace); /* @todo do better allocation */
2130                 p = node->fmt;
2131
2132                 when = request->timestamp;
2133                 if (request->packet) {
2134                         when = request->packet->timestamp.tv_sec;
2135                 }
2136
2137                 switch (*p) {
2138                 case '%':
2139                         str[0] = '%';
2140                         str[1] = '\0';
2141                         break;
2142
2143                 case 'd': /* request day */
2144                         if (!localtime_r(&when, &ts)) goto error;
2145                         strftime(str, freespace, "%d", &ts);
2146                         break;
2147
2148                 case 'l': /* request timestamp */
2149                         snprintf(str, freespace, "%lu",
2150                                  (unsigned long) when);
2151                         break;
2152
2153                 case 'm': /* request month */
2154                         if (!localtime_r(&when, &ts)) goto error;
2155                         strftime(str, freespace, "%m", &ts);
2156                         break;
2157
2158                 case 'n': /* Request Number*/
2159                         snprintf(str, freespace, "%u", request->number);
2160                         break;
2161
2162                 case 't': /* request timestamp */
2163                         CTIME_R(&when, str, freespace);
2164                         nl = strchr(str, '\n');
2165                         if (nl) *nl = '\0';
2166                         break;
2167
2168                 case 'D': /* request date */
2169                         if (!localtime_r(&when, &ts)) goto error;
2170                         strftime(str, freespace, "%Y%m%d", &ts);
2171                         break;
2172
2173                 case 'G': /* request minute */
2174                         if (!localtime_r(&when, &ts)) goto error;
2175                         strftime(str, freespace, "%M", &ts);
2176                         break;
2177
2178                 case 'H': /* request hour */
2179                         if (!localtime_r(&when, &ts)) goto error;
2180                         strftime(str, freespace, "%H", &ts);
2181                         break;
2182
2183                 case 'I': /* Request ID */
2184                         if (request->packet) {
2185                                 snprintf(str, freespace, "%i", request->packet->id);
2186                         }
2187                         break;
2188
2189                 case 'S': /* request timestamp in SQL format*/
2190                         if (!localtime_r(&when, &ts)) goto error;
2191                         strftime(str, freespace, "%Y-%m-%d %H:%M:%S", &ts);
2192                         break;
2193
2194                 case 'T': /* request timestamp */
2195                         if (!localtime_r(&when, &ts)) goto error;
2196                         strftime(str, freespace, "%Y-%m-%d-%H.%M.%S.000000", &ts);
2197                         break;
2198
2199                 case 'Y': /* request year */
2200                         if (!localtime_r(&when, &ts)) {
2201                                 error:
2202                                 REDEBUG("Failed converting packet timestamp to localtime: %s", fr_syserror(errno));
2203                                 talloc_free(str);
2204                                 return NULL;
2205                         }
2206                         strftime(str, freespace, "%Y", &ts);
2207                         break;
2208
2209                 case 'v': /* Version of code */
2210                         RWDEBUG("%%v is deprecated and will be removed.  Use ${version.freeradius-server}");
2211                         snprintf(str, freespace, "%s", radiusd_version_short);
2212                         break;
2213
2214                 default:
2215                         rad_assert(0 == 1);
2216                         break;
2217                 }
2218         }
2219                 break;
2220
2221         case XLAT_ATTRIBUTE:
2222                 XLAT_DEBUG("xlat_aprint ATTRIBUTE");
2223
2224                 /*
2225                  *      Some attributes are virtual <sigh>
2226                  */
2227                 str = xlat_getvp(ctx, request, &node->attr, escape ? false : true, true);
2228                 if (str) {
2229                         XLAT_DEBUG("EXPAND attr %s", node->attr.tmpl_da->name);
2230                         XLAT_DEBUG("       ---> %s", str);
2231                 }
2232                 break;
2233
2234         case XLAT_VIRTUAL:
2235                 XLAT_DEBUG("xlat_aprint VIRTUAL");
2236                 str = talloc_array(ctx, char, 2048); /* FIXME: have the module call talloc_typed_asprintf */
2237                 rcode = node->xlat->func(node->xlat->instance, request, NULL, str, 2048);
2238                 if (rcode < 0) {
2239                         talloc_free(str);
2240                         return NULL;
2241                 }
2242                 RDEBUG2("EXPAND %s", node->xlat->name);
2243                 RDEBUG2("   --> %s", str);
2244                 break;
2245
2246         case XLAT_MODULE:
2247                 XLAT_DEBUG("xlat_aprint MODULE");
2248
2249                 if (node->child) {
2250                         if (xlat_process(&child, request, node->child, node->xlat->escape, node->xlat->instance) == 0) {
2251                                 return NULL;
2252                         }
2253
2254                         XLAT_DEBUG("%.*sEXPAND mod %s %s", lvl, xlat_spaces, node->fmt, node->child->fmt);
2255                 } else {
2256                         XLAT_DEBUG("%.*sEXPAND mod %s", lvl, xlat_spaces, node->fmt);
2257                         child = talloc_typed_strdup(ctx, "");
2258                 }
2259
2260                 XLAT_DEBUG("%.*s      ---> %s", lvl, xlat_spaces, child);
2261
2262                 /*
2263                  *      Smash \n --> CR.
2264                  *
2265                  *      The OUTPUT of xlat is a printable string.  The INPUT might not be...
2266                  *
2267                  *      This is really the reverse of fr_prints().
2268                  */
2269                 if (cf_new_escape && *child) {
2270                         ssize_t slen;
2271                         PW_TYPE type;
2272                         value_data_t data;
2273
2274                         type = PW_TYPE_STRING;
2275                         slen = value_data_from_str(request, &data, &type, NULL, child, talloc_array_length(child) - 1, '"');
2276                         if (slen <= 0) {
2277                                 talloc_free(child);
2278                                 return NULL;
2279                         }
2280
2281                         talloc_free(child);
2282                         child = data.ptr;
2283
2284                 } else {
2285                         char *q;
2286
2287                         p = q = child;
2288                         while (*p) {
2289                                 if (*p == '\\') switch (p[1]) {
2290                                         default:
2291                                                 *(q++) = p[1];
2292                                                 p += 2;
2293                                                 continue;
2294
2295                                         case 'n':
2296                                                 *(q++) = '\n';
2297                                                 p += 2;
2298                                                 continue;
2299
2300                                         case 't':
2301                                                 *(q++) = '\t';
2302                                                 p += 2;
2303                                                 continue;
2304                                         }
2305
2306                                 *(q++) = *(p++);
2307                         }
2308                         *q = '\0';
2309                 }
2310
2311                 str = talloc_array(ctx, char, 2048); /* FIXME: have the module call talloc_typed_asprintf */
2312                 *str = '\0';    /* Be sure the string is NULL terminated, we now only free on error */
2313
2314                 rcode = node->xlat->func(node->xlat->instance, request, child, str, 2048);
2315                 talloc_free(child);
2316                 if (rcode < 0) {
2317                         talloc_free(str);
2318                         return NULL;
2319                 }
2320                 break;
2321
2322 #ifdef HAVE_REGEX
2323         case XLAT_REGEX:
2324                 XLAT_DEBUG("xlat_aprint REGEX");
2325                 if (regex_request_to_sub(ctx, &str, request, node->attr.tmpl_num) < 0) return NULL;
2326
2327                 break;
2328 #endif
2329
2330         case XLAT_ALTERNATE:
2331                 XLAT_DEBUG("xlat_aprint ALTERNATE");
2332                 rad_assert(node->child != NULL);
2333                 rad_assert(node->alternate != NULL);
2334
2335                 str = xlat_aprint(ctx, request, node->child, escape, escape_ctx, lvl);
2336                 if (str) {
2337                         XLAT_DEBUG("ALTERNATE got string: %s", str);
2338                         break;
2339                 }
2340
2341                 XLAT_DEBUG("ALTERNATE going to alternate");
2342                 str = xlat_aprint(ctx, request, node->alternate, escape, escape_ctx, lvl);
2343                 break;
2344
2345         }
2346
2347         /*
2348          *      If there's no data, return that, instead of an empty string.
2349          */
2350         if (str && !str[0]) {
2351                 talloc_free(str);
2352                 return NULL;
2353         }
2354
2355         /*
2356          *      Escape the non-literals we found above.
2357          */
2358         if (str && escape) {
2359                 char *escaped;
2360
2361                 escaped = talloc_array(ctx, char, 2048); /* FIXME: do something intelligent */
2362                 escape(request, escaped, 2038, str, escape_ctx);
2363                 talloc_free(str);
2364                 str = escaped;
2365         }
2366
2367         return str;
2368 }
2369
2370
2371 static size_t xlat_process(char **out, REQUEST *request, xlat_exp_t const * const head,
2372                            xlat_escape_t escape, void *escape_ctx)
2373 {
2374         int i, list;
2375         size_t total;
2376         char **array, *answer;
2377         xlat_exp_t const *node;
2378
2379         *out = NULL;
2380
2381         /*
2382          *      There are no nodes to process, so the result is a zero
2383          *      length string.
2384          */
2385         if (!head) {
2386                 *out = talloc_zero_array(request, char, 1);
2387                 return 0;
2388         }
2389
2390         /*
2391          *      Hack for speed.  If it's one expansion, just allocate
2392          *      that and return, instead of allocating an intermediary
2393          *      array.
2394          */
2395         if (!head->next) {
2396                 /*
2397                  *      Pass the MAIN escape function.  Recursive
2398                  *      calls will call node-specific escape
2399                  *      functions.
2400                  */
2401                 answer = xlat_aprint(request, request, head, escape, escape_ctx, 0);
2402                 if (!answer) {
2403                         *out = talloc_zero_array(request, char, 1);
2404                         return 0;
2405                 }
2406                 *out = answer;
2407                 return strlen(answer);
2408         }
2409
2410         list = 0;               /* FIXME: calculate this once */
2411         for (node = head; node != NULL; node = node->next) {
2412                 list++;
2413         }
2414
2415         array = talloc_array(request, char *, list);
2416         if (!array) return -1;
2417
2418         for (node = head, i = 0; node != NULL; node = node->next, i++) {
2419                 array[i] = xlat_aprint(array, request, node, escape, escape_ctx, 0); /* may be NULL */
2420         }
2421
2422         total = 0;
2423         for (i = 0; i < list; i++) {
2424                 if (array[i]) total += strlen(array[i]); /* FIXME: calculate strlen once */
2425         }
2426
2427         if (!total) {
2428                 talloc_free(array);
2429                 *out = talloc_zero_array(request, char, 1);
2430                 return 0;
2431         }
2432
2433         answer = talloc_array(request, char, total + 1);
2434
2435         total = 0;
2436         for (i = 0; i < list; i++) {
2437                 size_t len;
2438
2439                 if (array[i]) {
2440                         len = strlen(array[i]);
2441                         memcpy(answer + total, array[i], len);
2442                         total += len;
2443                 }
2444         }
2445         answer[total] = '\0';
2446         talloc_free(array);     /* and child entries */
2447
2448         *out = answer;
2449         return total;
2450 }
2451
2452
2453 /** Replace %whatever in a string.
2454  *
2455  * See 'doc/configuration/variables.rst' for more information.
2456  *
2457  * @param[out] out Where to write pointer to output buffer.
2458  * @param[in] outlen Size of out.
2459  * @param[in] request current request.
2460  * @param[in] node the xlat structure to expand
2461  * @param[in] escape function to escape final value e.g. SQL quoting.
2462  * @param[in] escape_ctx pointer to pass to escape function.
2463  * @return length of string written @bug should really have -1 for failure
2464  */
2465 static ssize_t xlat_expand_struct(char **out, size_t outlen, REQUEST *request, xlat_exp_t const *node,
2466                                   xlat_escape_t escape, void *escape_ctx)
2467 {
2468         char *buff;
2469         ssize_t len;
2470
2471         rad_assert(node != NULL);
2472
2473         len = xlat_process(&buff, request, node, escape, escape_ctx);
2474         if ((len < 0) || !buff) {
2475                 rad_assert(buff == NULL);
2476                 if (*out) *out[0] = '\0';
2477                 return len;
2478         }
2479
2480         len = strlen(buff);
2481
2482         /*
2483          *      If out doesn't point to an existing buffer
2484          *      copy the pointer to our buffer over.
2485          */
2486         if (!*out) {
2487                 *out = buff;
2488                 return len;
2489         }
2490
2491         /*
2492          *      Otherwise copy the malloced buffer to the fixed one.
2493          */
2494         strlcpy(*out, buff, outlen);
2495         talloc_free(buff);
2496         return len;
2497 }
2498
2499 static ssize_t xlat_expand(char **out, size_t outlen, REQUEST *request, char const *fmt,
2500                            xlat_escape_t escape, void *escape_ctx) CC_HINT(nonnull (1, 3, 4));
2501
2502 /** Replace %whatever in a string.
2503  *
2504  * See 'doc/configuration/variables.rst' for more information.
2505  *
2506  * @param[out] out Where to write pointer to output buffer.
2507  * @param[in] outlen Size of out.
2508  * @param[in] request current request.
2509  * @param[in] fmt string to expand.
2510  * @param[in] escape function to escape final value e.g. SQL quoting.
2511  * @param[in] escape_ctx pointer to pass to escape function.
2512  * @return length of string written @bug should really have -1 for failure
2513  */
2514 static ssize_t xlat_expand(char **out, size_t outlen, REQUEST *request, char const *fmt,
2515                            xlat_escape_t escape, void *escape_ctx)
2516 {
2517         ssize_t len;
2518         xlat_exp_t *node;
2519
2520         /*
2521          *      Give better errors than the old code.
2522          */
2523         len = xlat_tokenize_request(request, fmt, &node);
2524         if (len == 0) {
2525                 if (*out) {
2526                         *out[0] = '\0';
2527                 } else {
2528                         *out = talloc_zero_array(request, char, 1);
2529                 }
2530                 return 0;
2531         }
2532
2533         if (len < 0) {
2534                 if (*out) *out[0] = '\0';
2535                 return -1;
2536         }
2537
2538         len = xlat_expand_struct(out, outlen, request, node, escape, escape_ctx);
2539         talloc_free(node);
2540
2541         RDEBUG2("EXPAND %s", fmt);
2542         RDEBUG2("   --> %s", *out);
2543
2544         return len;
2545 }
2546
2547 /** Try to convert an xlat to a tmpl for efficiency
2548  *
2549  * @param ctx to allocate new vp_tmpl_t in.
2550  * @param node to convert.
2551  * @return NULL if unable to convert (not necessarily error), or a new vp_tmpl_t.
2552  */
2553 vp_tmpl_t *xlat_to_tmpl_attr(TALLOC_CTX *ctx, xlat_exp_t *node)
2554 {
2555         vp_tmpl_t *vpt;
2556
2557         if (node->next || (node->type != XLAT_ATTRIBUTE) || (node->attr.type != TMPL_TYPE_ATTR)) return NULL;
2558
2559         /*
2560          *   Concat means something completely different as an attribute reference
2561          *   Count isn't implemented.
2562          */
2563         if ((node->attr.tmpl_num == NUM_COUNT) || (node->attr.tmpl_num == NUM_ALL)) return NULL;
2564
2565         vpt = tmpl_alloc(ctx, TMPL_TYPE_ATTR, node->fmt, -1);
2566         if (!vpt) return NULL;
2567         memcpy(&vpt->data, &node->attr.data, sizeof(vpt->data));
2568
2569         VERIFY_TMPL(vpt);
2570
2571         return vpt;
2572 }
2573
2574 /** Try to convert attr tmpl to an xlat for &attr[*] and artificially constructing expansions
2575  *
2576  * @param ctx to allocate new xlat_expt_t in.
2577  * @param vpt to convert.
2578  * @return NULL if unable to convert (not necessarily error), or a new vp_tmpl_t.
2579  */
2580 xlat_exp_t *xlat_from_tmpl_attr(TALLOC_CTX *ctx, vp_tmpl_t *vpt)
2581 {
2582         xlat_exp_t *node;
2583
2584         if (vpt->type != TMPL_TYPE_ATTR) return NULL;
2585
2586         node = talloc_zero(ctx, xlat_exp_t);
2587         node->type = XLAT_ATTRIBUTE;
2588         node->fmt = talloc_bstrndup(node, vpt->name, vpt->len);
2589         tmpl_init(&node->attr, TMPL_TYPE_ATTR, node->fmt, talloc_array_length(node->fmt) - 1);
2590         memcpy(&node->attr.data, &vpt->data, sizeof(vpt->data));
2591
2592         return node;
2593 }
2594
2595 ssize_t radius_xlat(char *out, size_t outlen, REQUEST *request, char const *fmt, xlat_escape_t escape, void *ctx)
2596 {
2597         return xlat_expand(&out, outlen, request, fmt, escape, ctx);
2598 }
2599
2600 ssize_t radius_xlat_struct(char *out, size_t outlen, REQUEST *request, xlat_exp_t const *xlat, xlat_escape_t escape, void *ctx)
2601 {
2602         return xlat_expand_struct(&out, outlen, request, xlat, escape, ctx);
2603 }
2604
2605 ssize_t radius_axlat(char **out, REQUEST *request, char const *fmt, xlat_escape_t escape, void *ctx)
2606 {
2607         return xlat_expand(out, 0, request, fmt, escape, ctx);
2608 }
2609
2610 ssize_t radius_axlat_struct(char **out, REQUEST *request, xlat_exp_t const *xlat, xlat_escape_t escape, void *ctx)
2611 {
2612         return xlat_expand_struct(out, 0, request, xlat, escape, ctx);
2613 }