Coverage Report

Created: 2024-01-18 20:17

/src/libxslt/libexslt/strings.c
Line
Count
Source (jump to first uncovered line)
1
#define IN_LIBEXSLT
2
#include "libexslt/libexslt.h"
3
4
#include <libxml/tree.h>
5
#include <libxml/xpath.h>
6
#include <libxml/xpathInternals.h>
7
#include <libxml/parser.h>
8
#include <libxml/encoding.h>
9
#include <libxml/uri.h>
10
11
#include <libxslt/xsltutils.h>
12
#include <libxslt/xsltInternals.h>
13
#include <libxslt/extensions.h>
14
15
#include "exslt.h"
16
17
/**
18
 * exsltStrTokenizeFunction:
19
 * @ctxt: an XPath parser context
20
 * @nargs: the number of arguments
21
 *
22
 * Splits up a string on the characters of the delimiter string and returns a
23
 * node set of token elements, each containing one token from the string.
24
 */
25
static void
26
exsltStrTokenizeFunction(xmlXPathParserContextPtr ctxt, int nargs)
27
59.1k
{
28
59.1k
    xsltTransformContextPtr tctxt;
29
59.1k
    xmlChar *str, *delimiters, *cur;
30
59.1k
    const xmlChar *token, *delimiter;
31
59.1k
    xmlNodePtr node;
32
59.1k
    xmlDocPtr container;
33
59.1k
    xmlXPathObjectPtr ret = NULL;
34
59.1k
    int clen;
35
36
59.1k
    if ((nargs < 1) || (nargs > 2)) {
37
153
        xmlXPathSetArityError(ctxt);
38
153
        return;
39
153
    }
40
41
58.9k
    if (nargs == 2) {
42
9.27k
        delimiters = xmlXPathPopString(ctxt);
43
9.27k
        if (xmlXPathCheckError(ctxt))
44
0
            return;
45
49.7k
    } else {
46
49.7k
        delimiters = xmlStrdup((const xmlChar *) "\t\r\n ");
47
49.7k
    }
48
58.9k
    if (delimiters == NULL)
49
0
        return;
50
51
58.9k
    str = xmlXPathPopString(ctxt);
52
58.9k
    if (xmlXPathCheckError(ctxt) || (str == NULL)) {
53
0
        xmlFree(delimiters);
54
0
        return;
55
0
    }
56
57
    /* Return a result tree fragment */
58
58.9k
    tctxt = xsltXPathGetTransformContext(ctxt);
59
58.9k
    if (tctxt == NULL) {
60
0
        xsltTransformError(xsltXPathGetTransformContext(ctxt), NULL, NULL,
61
0
        "exslt:tokenize : internal error tctxt == NULL\n");
62
0
  goto fail;
63
0
    }
64
65
58.9k
    container = xsltCreateRVT(tctxt);
66
58.9k
    if (container != NULL) {
67
58.9k
        xsltRegisterLocalRVT(tctxt, container);
68
58.9k
        ret = xmlXPathNewNodeSet(NULL);
69
58.9k
        if (ret != NULL) {
70
33.6M
            for (cur = str, token = str; *cur != 0; cur += clen) {
71
33.5M
          clen = xmlUTF8Strsize(cur, 1);
72
33.5M
    if (*delimiters == 0) { /* empty string case */
73
283k
        xmlChar ctmp;
74
283k
        ctmp = *(cur+clen);
75
283k
        *(cur+clen) = 0;
76
283k
                    node = xmlNewDocRawNode(container, NULL,
77
283k
                                       (const xmlChar *) "token", cur);
78
283k
        xmlAddChild((xmlNodePtr) container, node);
79
283k
        xmlXPathNodeSetAddUnique(ret->nodesetval, node);
80
283k
                    *(cur+clen) = ctmp; /* restore the changed byte */
81
283k
                    token = cur + clen;
82
151M
                } else for (delimiter = delimiters; *delimiter != 0;
83
124M
        delimiter += xmlUTF8Strsize(delimiter, 1)) {
84
124M
                    if (!xmlUTF8Charcmp(cur, delimiter)) {
85
5.66M
                        if (cur == token) {
86
                            /* discard empty tokens */
87
5.52M
                            token = cur + clen;
88
5.52M
                            break;
89
5.52M
                        }
90
142k
                        *cur = 0; /* terminate the token */
91
142k
                        node = xmlNewDocRawNode(container, NULL,
92
142k
                                           (const xmlChar *) "token", token);
93
142k
      xmlAddChild((xmlNodePtr) container, node);
94
142k
      xmlXPathNodeSetAddUnique(ret->nodesetval, node);
95
142k
                        *cur = *delimiter; /* restore the changed byte */
96
142k
                        token = cur + clen;
97
142k
                        break;
98
5.66M
                    }
99
124M
                }
100
33.5M
            }
101
58.9k
            if (token != cur) {
102
20.9k
    node = xmlNewDocRawNode(container, NULL,
103
20.9k
            (const xmlChar *) "token", token);
104
20.9k
                xmlAddChild((xmlNodePtr) container, node);
105
20.9k
          xmlXPathNodeSetAddUnique(ret->nodesetval, node);
106
20.9k
            }
107
58.9k
        }
108
58.9k
    }
109
110
58.9k
fail:
111
58.9k
    if (str != NULL)
112
58.9k
        xmlFree(str);
113
58.9k
    if (delimiters != NULL)
114
58.9k
        xmlFree(delimiters);
115
58.9k
    if (ret != NULL)
116
58.9k
        valuePush(ctxt, ret);
117
0
    else
118
0
        valuePush(ctxt, xmlXPathNewNodeSet(NULL));
119
58.9k
}
120
121
/**
122
 * exsltStrSplitFunction:
123
 * @ctxt: an XPath parser context
124
 * @nargs: the number of arguments
125
 *
126
 * Splits up a string on a delimiting string and returns a node set of token
127
 * elements, each containing one token from the string.
128
 */
129
static void
130
105k
exsltStrSplitFunction(xmlXPathParserContextPtr ctxt, int nargs) {
131
105k
    xsltTransformContextPtr tctxt;
132
105k
    xmlChar *str, *delimiter, *cur;
133
105k
    const xmlChar *token;
134
105k
    xmlNodePtr node;
135
105k
    xmlDocPtr container;
136
105k
    xmlXPathObjectPtr ret = NULL;
137
105k
    int delimiterLength;
138
139
105k
    if ((nargs < 1) || (nargs > 2)) {
140
536
        xmlXPathSetArityError(ctxt);
141
536
        return;
142
536
    }
143
144
105k
    if (nargs == 2) {
145
93.2k
        delimiter = xmlXPathPopString(ctxt);
146
93.2k
        if (xmlXPathCheckError(ctxt))
147
0
            return;
148
93.2k
    } else {
149
11.8k
        delimiter = xmlStrdup((const xmlChar *) " ");
150
11.8k
    }
151
105k
    if (delimiter == NULL)
152
0
        return;
153
105k
    delimiterLength = xmlStrlen (delimiter);
154
155
105k
    str = xmlXPathPopString(ctxt);
156
105k
    if (xmlXPathCheckError(ctxt) || (str == NULL)) {
157
0
        xmlFree(delimiter);
158
0
        return;
159
0
    }
160
161
    /* Return a result tree fragment */
162
105k
    tctxt = xsltXPathGetTransformContext(ctxt);
163
105k
    if (tctxt == NULL) {
164
0
        xsltTransformError(xsltXPathGetTransformContext(ctxt), NULL, NULL,
165
0
        "exslt:tokenize : internal error tctxt == NULL\n");
166
0
  goto fail;
167
0
    }
168
169
    /*
170
    * OPTIMIZE TODO: We are creating an xmlDoc for every split!
171
    */
172
105k
    container = xsltCreateRVT(tctxt);
173
105k
    if (container != NULL) {
174
105k
        xsltRegisterLocalRVT(tctxt, container);
175
105k
        ret = xmlXPathNewNodeSet(NULL);
176
105k
        if (ret != NULL) {
177
352M
            for (cur = str, token = str; *cur != 0; cur++) {
178
352M
    if (delimiterLength == 0) {
179
39.5M
        if (cur != token) {
180
39.4M
      xmlChar tmp = *cur;
181
39.4M
      *cur = 0;
182
39.4M
                        node = xmlNewDocRawNode(container, NULL,
183
39.4M
                                           (const xmlChar *) "token", token);
184
39.4M
      xmlAddChild((xmlNodePtr) container, node);
185
39.4M
      xmlXPathNodeSetAddUnique(ret->nodesetval, node);
186
39.4M
      *cur = tmp;
187
39.4M
      token++;
188
39.4M
        }
189
39.5M
    }
190
312M
    else if (!xmlStrncasecmp(cur, delimiter, delimiterLength)) {
191
16.5M
        if (cur == token) {
192
      /* discard empty tokens */
193
11.7M
      cur = cur + delimiterLength - 1;
194
11.7M
      token = cur + 1;
195
11.7M
      continue;
196
11.7M
        }
197
4.86M
        *cur = 0;
198
4.86M
        node = xmlNewDocRawNode(container, NULL,
199
4.86M
               (const xmlChar *) "token", token);
200
4.86M
        xmlAddChild((xmlNodePtr) container, node);
201
4.86M
        xmlXPathNodeSetAddUnique(ret->nodesetval, node);
202
4.86M
        *cur = *delimiter;
203
4.86M
        cur = cur + delimiterLength - 1;
204
4.86M
        token = cur + 1;
205
4.86M
                }
206
352M
            }
207
105k
      if (token != cur) {
208
101k
    node = xmlNewDocRawNode(container, NULL,
209
101k
           (const xmlChar *) "token", token);
210
101k
    xmlAddChild((xmlNodePtr) container, node);
211
101k
    xmlXPathNodeSetAddUnique(ret->nodesetval, node);
212
101k
      }
213
105k
        }
214
105k
    }
215
216
105k
fail:
217
105k
    if (str != NULL)
218
105k
        xmlFree(str);
219
105k
    if (delimiter != NULL)
220
105k
        xmlFree(delimiter);
221
105k
    if (ret != NULL)
222
105k
        valuePush(ctxt, ret);
223
0
    else
224
0
        valuePush(ctxt, xmlXPathNewNodeSet(NULL));
225
105k
}
226
227
/**
228
 * exsltStrEncodeUriFunction:
229
 * @ctxt: an XPath parser context
230
 * @nargs: the number of arguments
231
 *
232
 * URI-Escapes a string
233
 */
234
static void
235
16.0k
exsltStrEncodeUriFunction (xmlXPathParserContextPtr ctxt, int nargs) {
236
16.0k
    int escape_all = 1, str_len = 0;
237
16.0k
    xmlChar *str = NULL, *ret = NULL, *tmp;
238
239
16.0k
    if ((nargs < 2) || (nargs > 3)) {
240
170
  xmlXPathSetArityError(ctxt);
241
170
  return;
242
170
    }
243
244
15.8k
    if (nargs >= 3) {
245
        /* check for UTF-8 if encoding was explicitly given;
246
           we don't support anything else yet */
247
60
        tmp = xmlXPathPopString(ctxt);
248
60
        if (xmlUTF8Strlen(tmp) != 5 || xmlStrcmp((const xmlChar *)"UTF-8",tmp)) {
249
58
      xmlXPathReturnEmptyString(ctxt);
250
58
      xmlFree(tmp);
251
58
      return;
252
58
  }
253
2
  xmlFree(tmp);
254
2
    }
255
256
15.8k
    escape_all = xmlXPathPopBoolean(ctxt);
257
258
15.8k
    str = xmlXPathPopString(ctxt);
259
15.8k
    str_len = xmlUTF8Strlen(str);
260
261
15.8k
    if (str_len <= 0) {
262
3.07k
        if (str_len < 0)
263
2.97k
            xsltGenericError(xsltGenericErrorContext,
264
2.97k
                             "exsltStrEncodeUriFunction: invalid UTF-8\n");
265
3.07k
  xmlXPathReturnEmptyString(ctxt);
266
3.07k
  xmlFree(str);
267
3.07k
  return;
268
3.07k
    }
269
270
12.7k
    ret = xmlURIEscapeStr(str,(const xmlChar *)(escape_all?"-_.!~*'()":"-_.!~*'();/?:@&=+$,[]"));
271
12.7k
    xmlXPathReturnString(ctxt, ret);
272
273
12.7k
    if (str != NULL)
274
12.7k
  xmlFree(str);
275
12.7k
}
276
277
/**
278
 * exsltStrDecodeUriFunction:
279
 * @ctxt: an XPath parser context
280
 * @nargs: the number of arguments
281
 *
282
 * reverses URI-Escaping of a string
283
 */
284
static void
285
18.5k
exsltStrDecodeUriFunction (xmlXPathParserContextPtr ctxt, int nargs) {
286
18.5k
    int str_len = 0;
287
18.5k
    xmlChar *str = NULL, *ret = NULL, *tmp;
288
289
18.5k
    if ((nargs < 1) || (nargs > 2)) {
290
99
  xmlXPathSetArityError(ctxt);
291
99
  return;
292
99
    }
293
294
18.4k
    if (nargs >= 2) {
295
        /* check for UTF-8 if encoding was explicitly given;
296
           we don't support anything else yet */
297
48
        tmp = xmlXPathPopString(ctxt);
298
48
        if (xmlUTF8Strlen(tmp) != 5 || xmlStrcmp((const xmlChar *)"UTF-8",tmp)) {
299
43
      xmlXPathReturnEmptyString(ctxt);
300
43
      xmlFree(tmp);
301
43
      return;
302
43
  }
303
5
  xmlFree(tmp);
304
5
    }
305
306
18.4k
    str = xmlXPathPopString(ctxt);
307
18.4k
    str_len = xmlUTF8Strlen(str);
308
309
18.4k
    if (str_len <= 0) {
310
3.67k
        if (str_len < 0)
311
3.52k
            xsltGenericError(xsltGenericErrorContext,
312
3.52k
                             "exsltStrDecodeUriFunction: invalid UTF-8\n");
313
3.67k
  xmlXPathReturnEmptyString(ctxt);
314
3.67k
  xmlFree(str);
315
3.67k
  return;
316
3.67k
    }
317
318
14.7k
    ret = (xmlChar *) xmlURIUnescapeString((const char *)str,0,NULL);
319
14.7k
    if (!xmlCheckUTF8(ret)) {
320
  /* FIXME: instead of throwing away the whole URI, we should
321
        only discard the invalid sequence(s). How to do that? */
322
10.6k
  xmlXPathReturnEmptyString(ctxt);
323
10.6k
  xmlFree(str);
324
10.6k
  xmlFree(ret);
325
10.6k
  return;
326
10.6k
    }
327
328
4.13k
    xmlXPathReturnString(ctxt, ret);
329
330
4.13k
    if (str != NULL)
331
4.13k
  xmlFree(str);
332
4.13k
}
333
334
/**
335
 * exsltStrPaddingFunction:
336
 * @ctxt: an XPath parser context
337
 * @nargs: the number of arguments
338
 *
339
 * Creates a padding string of a certain length.
340
 */
341
static void
342
7.18k
exsltStrPaddingFunction (xmlXPathParserContextPtr ctxt, int nargs) {
343
7.18k
    int number, str_len = 0, str_size = 0;
344
7.18k
    double floatval;
345
7.18k
    xmlChar *str = NULL;
346
7.18k
    xmlBufferPtr buf;
347
348
7.18k
    if ((nargs < 1) || (nargs > 2)) {
349
74
  xmlXPathSetArityError(ctxt);
350
74
  return;
351
74
    }
352
353
7.10k
    if (nargs == 2) {
354
6.51k
  str = xmlXPathPopString(ctxt);
355
6.51k
  str_len = xmlUTF8Strlen(str);
356
6.51k
  str_size = xmlStrlen(str);
357
6.51k
    }
358
359
7.10k
    floatval = xmlXPathPopNumber(ctxt);
360
361
7.10k
    if (str_len <= 0) {
362
4.09k
        if (str_len < 0) {
363
3.44k
            xsltGenericError(xsltGenericErrorContext,
364
3.44k
                             "exsltStrPaddingFunction: invalid UTF-8\n");
365
3.44k
            xmlXPathReturnEmptyString(ctxt);
366
3.44k
            xmlFree(str);
367
3.44k
            return;
368
3.44k
        }
369
644
  if (str != NULL) xmlFree(str);
370
644
  str = xmlStrdup((const xmlChar *) " ");
371
644
  str_len = 1;
372
644
  str_size = 1;
373
644
    }
374
375
3.66k
    if (xmlXPathIsNaN(floatval) || floatval < 0.0) {
376
575
        number = 0;
377
3.08k
    } else if (floatval >= 100000.0) {
378
40
        number = 100000;
379
40
    }
380
3.04k
    else {
381
3.04k
        number = (int) floatval;
382
3.04k
    }
383
384
3.66k
    if (number <= 0) {
385
729
  xmlXPathReturnEmptyString(ctxt);
386
729
  xmlFree(str);
387
729
  return;
388
729
    }
389
390
2.93k
    buf = xmlBufferCreateSize(number);
391
2.93k
    if (buf == NULL) {
392
0
        xmlXPathSetError(ctxt, XPATH_MEMORY_ERROR);
393
0
  xmlFree(str);
394
0
  return;
395
0
    }
396
2.93k
    xmlBufferSetAllocationScheme(buf, XML_BUFFER_ALLOC_DOUBLEIT);
397
398
3.99M
    while (number >= str_len) {
399
3.98M
        xmlBufferAdd(buf, str, str_size);
400
3.98M
  number -= str_len;
401
3.98M
    }
402
2.93k
    if (number > 0) {
403
1.62k
  str_size = xmlUTF8Strsize(str, number);
404
1.62k
        xmlBufferAdd(buf, str, str_size);
405
1.62k
    }
406
407
2.93k
    xmlXPathReturnString(ctxt, xmlBufferDetach(buf));
408
409
2.93k
    xmlBufferFree(buf);
410
2.93k
    if (str != NULL)
411
2.93k
  xmlFree(str);
412
2.93k
}
413
414
/**
415
 * exsltStrAlignFunction:
416
 * @ctxt: an XPath parser context
417
 * @nargs: the number of arguments
418
 *
419
 * Aligns a string within another string.
420
 */
421
static void
422
31.5k
exsltStrAlignFunction (xmlXPathParserContextPtr ctxt, int nargs) {
423
31.5k
    xmlChar *str, *padding, *alignment, *ret;
424
31.5k
    int str_l, padding_l;
425
426
31.5k
    if ((nargs < 2) || (nargs > 3)) {
427
367
  xmlXPathSetArityError(ctxt);
428
367
  return;
429
367
    }
430
431
31.2k
    if (nargs == 3)
432
8.48k
  alignment = xmlXPathPopString(ctxt);
433
22.7k
    else
434
22.7k
  alignment = NULL;
435
436
31.2k
    padding = xmlXPathPopString(ctxt);
437
31.2k
    str = xmlXPathPopString(ctxt);
438
439
31.2k
    str_l = xmlUTF8Strlen (str);
440
31.2k
    padding_l = xmlUTF8Strlen (padding);
441
442
31.2k
    if (str_l < 0 || padding_l < 0) {
443
12.5k
        xsltGenericError(xsltGenericErrorContext,
444
12.5k
                         "exsltStrAlignFunction: invalid UTF-8\n");
445
12.5k
        xmlXPathReturnEmptyString(ctxt);
446
12.5k
        xmlFree(str);
447
12.5k
        xmlFree(padding);
448
12.5k
        xmlFree(alignment);
449
12.5k
        return;
450
12.5k
    }
451
452
18.6k
    if (str_l == padding_l) {
453
2.72k
  xmlXPathReturnString (ctxt, str);
454
2.72k
  xmlFree(padding);
455
2.72k
  xmlFree(alignment);
456
2.72k
  return;
457
2.72k
    }
458
459
15.9k
    if (str_l > padding_l) {
460
7.09k
  ret = xmlUTF8Strndup (str, padding_l);
461
8.83k
    } else {
462
8.83k
  if (xmlStrEqual(alignment, (const xmlChar *) "right")) {
463
292
      ret = xmlUTF8Strndup (padding, padding_l - str_l);
464
292
      ret = xmlStrcat (ret, str);
465
8.53k
  } else if (xmlStrEqual(alignment, (const xmlChar *) "center")) {
466
784
      int left = (padding_l - str_l) / 2;
467
784
      int right_start;
468
469
784
      ret = xmlUTF8Strndup (padding, left);
470
784
      ret = xmlStrcat (ret, str);
471
472
784
      right_start = xmlUTF8Strsize (padding, left + str_l);
473
784
      ret = xmlStrcat (ret, padding + right_start);
474
7.75k
  } else {
475
7.75k
      int str_s;
476
477
7.75k
      str_s = xmlUTF8Strsize(padding, str_l);
478
7.75k
      ret = xmlStrdup (str);
479
7.75k
      ret = xmlStrcat (ret, padding + str_s);
480
7.75k
  }
481
8.83k
    }
482
483
15.9k
    xmlXPathReturnString (ctxt, ret);
484
485
15.9k
    xmlFree(str);
486
15.9k
    xmlFree(padding);
487
15.9k
    xmlFree(alignment);
488
15.9k
}
489
490
/**
491
 * exsltStrConcatFunction:
492
 * @ctxt: an XPath parser context
493
 * @nargs: the number of arguments
494
 *
495
 * Takes a node set and returns the concatenation of the string values
496
 * of the nodes in that node set.  If the node set is empty, it
497
 * returns an empty string.
498
 */
499
static void
500
787
exsltStrConcatFunction (xmlXPathParserContextPtr ctxt, int nargs) {
501
787
    xmlXPathObjectPtr obj;
502
787
    xmlBufferPtr buf;
503
787
    int i;
504
505
787
    if (nargs  != 1) {
506
111
  xmlXPathSetArityError(ctxt);
507
111
  return;
508
111
    }
509
510
676
    if (!xmlXPathStackIsNodeSet(ctxt)) {
511
24
  xmlXPathSetTypeError(ctxt);
512
24
  return;
513
24
    }
514
515
652
    obj = valuePop (ctxt);
516
517
652
    if (xmlXPathNodeSetIsEmpty(obj->nodesetval)) {
518
102
        xmlXPathFreeObject(obj);
519
102
  xmlXPathReturnEmptyString(ctxt);
520
102
  return;
521
102
    }
522
523
550
    buf = xmlBufferCreate();
524
550
    if (buf == NULL) {
525
0
        xmlXPathSetError(ctxt, XPATH_MEMORY_ERROR);
526
0
        xmlXPathFreeObject(obj);
527
0
  return;
528
0
    }
529
550
    xmlBufferSetAllocationScheme(buf, XML_BUFFER_ALLOC_DOUBLEIT);
530
531
57.3k
    for (i = 0; i < obj->nodesetval->nodeNr; i++) {
532
56.7k
  xmlChar *tmp;
533
56.7k
  tmp = xmlXPathCastNodeToString(obj->nodesetval->nodeTab[i]);
534
535
56.7k
        xmlBufferCat(buf, tmp);
536
537
56.7k
  xmlFree(tmp);
538
56.7k
    }
539
540
550
    xmlXPathFreeObject (obj);
541
542
550
    xmlXPathReturnString(ctxt, xmlBufferDetach(buf));
543
550
    xmlBufferFree(buf);
544
550
}
545
546
/**
547
 * exsltStrReturnString:
548
 * @ctxt: an XPath parser context
549
 * @str: a string
550
 * @len: length of string
551
 *
552
 * Returns a string as a node set.
553
 */
554
static int
555
exsltStrReturnString(xmlXPathParserContextPtr ctxt, const xmlChar *str,
556
                     int len)
557
10.7k
{
558
10.7k
    xsltTransformContextPtr tctxt = xsltXPathGetTransformContext(ctxt);
559
10.7k
    xmlDocPtr container;
560
10.7k
    xmlNodePtr text_node;
561
10.7k
    xmlXPathObjectPtr ret;
562
563
10.7k
    container = xsltCreateRVT(tctxt);
564
10.7k
    if (container == NULL) {
565
0
        xmlXPathSetError(ctxt, XPATH_MEMORY_ERROR);
566
0
        return(-1);
567
0
    }
568
10.7k
    xsltRegisterLocalRVT(tctxt, container);
569
570
10.7k
    text_node = xmlNewTextLen(str, len);
571
10.7k
    if (text_node == NULL) {
572
0
        xmlXPathSetError(ctxt, XPATH_MEMORY_ERROR);
573
0
        return(-1);
574
0
    }
575
10.7k
    xmlAddChild((xmlNodePtr) container, text_node);
576
577
10.7k
    ret = xmlXPathNewNodeSet(text_node);
578
10.7k
    if (ret == NULL) {
579
0
        xmlXPathSetError(ctxt, XPATH_MEMORY_ERROR);
580
0
        return(-1);
581
0
    }
582
583
10.7k
    valuePush(ctxt, ret);
584
585
10.7k
    return(0);
586
10.7k
}
587
588
/**
589
 * exsltStrReplaceFunction:
590
 * @ctxt: an XPath parser context
591
 * @nargs: the number of arguments
592
 *
593
 * Takes a string, and two node sets and returns the string with all strings in
594
 * the first node set replaced by all strings in the second node set.
595
 */
596
static void
597
11.2k
exsltStrReplaceFunction (xmlXPathParserContextPtr ctxt, int nargs) {
598
11.2k
    int i, i_empty, n, slen0, rlen0, *slen, *rlen;
599
11.2k
    void *mem = NULL;
600
11.2k
    const xmlChar *src, *start;
601
11.2k
    xmlChar *string, *search_str = NULL, *replace_str = NULL;
602
11.2k
    xmlChar **search, **replace;
603
11.2k
    xmlNodeSetPtr search_set = NULL, replace_set = NULL;
604
11.2k
    xmlBufferPtr buf;
605
606
11.2k
    if (nargs  != 3) {
607
504
        xmlXPathSetArityError(ctxt);
608
504
        return;
609
504
    }
610
611
    /* get replace argument */
612
613
10.7k
    if (!xmlXPathStackIsNodeSet(ctxt))
614
1.81k
        replace_str = xmlXPathPopString(ctxt);
615
8.94k
    else
616
8.94k
        replace_set = xmlXPathPopNodeSet(ctxt);
617
618
10.7k
    if (xmlXPathCheckError(ctxt))
619
0
        goto fail_replace;
620
621
    /* get search argument */
622
623
10.7k
    if (!xmlXPathStackIsNodeSet(ctxt)) {
624
845
        search_str = xmlXPathPopString(ctxt);
625
845
        n = 1;
626
845
    }
627
9.91k
    else {
628
9.91k
        search_set = xmlXPathPopNodeSet(ctxt);
629
9.91k
        n = search_set != NULL ? search_set->nodeNr : 0;
630
9.91k
    }
631
632
10.7k
    if (xmlXPathCheckError(ctxt))
633
0
        goto fail_search;
634
635
    /* get string argument */
636
637
10.7k
    string = xmlXPathPopString(ctxt);
638
10.7k
    if (xmlXPathCheckError(ctxt))
639
0
        goto fail_string;
640
641
    /* check for empty search node list */
642
643
10.7k
    if (n <= 0) {
644
1.05k
        exsltStrReturnString(ctxt, string, xmlStrlen(string));
645
1.05k
        goto done_empty_search;
646
1.05k
    }
647
648
    /* allocate memory for string pointer and length arrays */
649
650
9.70k
    if (n == 1) {
651
3.79k
        search = &search_str;
652
3.79k
        replace = &replace_str;
653
3.79k
        slen = &slen0;
654
3.79k
        rlen = &rlen0;
655
3.79k
    }
656
5.91k
    else {
657
5.91k
        mem = xmlMalloc(2 * n * (sizeof(const xmlChar *) + sizeof(int)));
658
5.91k
        if (mem == NULL) {
659
0
            xmlXPathSetError(ctxt, XPATH_MEMORY_ERROR);
660
0
            goto fail_malloc;
661
0
        }
662
5.91k
        search = (xmlChar **) mem;
663
5.91k
        replace = search + n;
664
5.91k
        slen = (int *) (replace + n);
665
5.91k
        rlen = slen + n;
666
5.91k
    }
667
668
    /* process arguments */
669
670
9.70k
    i_empty = -1;
671
672
3.81M
    for (i=0; i<n; ++i) {
673
3.80M
        if (search_set != NULL) {
674
3.80M
            search[i] = xmlXPathCastNodeToString(search_set->nodeTab[i]);
675
3.80M
            if (search[i] == NULL) {
676
0
                n = i;
677
0
                goto fail_process_args;
678
0
            }
679
3.80M
        }
680
681
3.80M
        slen[i] = xmlStrlen(search[i]);
682
3.80M
        if (i_empty < 0 && slen[i] == 0)
683
416
            i_empty = i;
684
685
3.80M
        if (replace_set != NULL) {
686
3.78M
            if (i < replace_set->nodeNr) {
687
27.0k
                replace[i] = xmlXPathCastNodeToString(replace_set->nodeTab[i]);
688
27.0k
                if (replace[i] == NULL) {
689
0
                    n = i + 1;
690
0
                    goto fail_process_args;
691
0
                }
692
27.0k
            }
693
3.76M
            else
694
3.76M
                replace[i] = NULL;
695
3.78M
        }
696
16.7k
        else {
697
16.7k
            if (i == 0)
698
1.37k
                replace[i] = replace_str;
699
15.3k
            else
700
15.3k
                replace[i] = NULL;
701
16.7k
        }
702
703
3.80M
        if (replace[i] == NULL)
704
3.77M
            rlen[i] = 0;
705
27.9k
        else
706
27.9k
            rlen[i] = xmlStrlen(replace[i]);
707
3.80M
    }
708
709
9.70k
    if (i_empty >= 0 && rlen[i_empty] == 0)
710
101
        i_empty = -1;
711
712
    /* replace operation */
713
714
9.70k
    buf = xmlBufferCreate();
715
9.70k
    if (buf == NULL) {
716
0
        xmlXPathSetError(ctxt, XPATH_MEMORY_ERROR);
717
0
        goto fail_buffer;
718
0
    }
719
9.70k
    xmlBufferSetAllocationScheme(buf, XML_BUFFER_ALLOC_DOUBLEIT);
720
9.70k
    src = string;
721
9.70k
    start = string;
722
723
8.04M
    while (*src != 0) {
724
8.03M
        int max_len = 0, i_match = 0;
725
726
225M
        for (i=0; i<n; ++i) {
727
217M
            if (*src == search[i][0] &&
728
217M
                slen[i] > max_len &&
729
217M
                xmlStrncmp(src, search[i], slen[i]) == 0)
730
626k
            {
731
626k
                i_match = i;
732
626k
                max_len = slen[i];
733
626k
            }
734
217M
        }
735
736
8.03M
        if (max_len == 0) {
737
7.41M
            if (i_empty >= 0 && start < src) {
738
349k
                if (xmlBufferAdd(buf, start, src - start) ||
739
349k
                    xmlBufferAdd(buf, replace[i_empty], rlen[i_empty]))
740
0
                {
741
0
                    xmlXPathSetError(ctxt, XPATH_MEMORY_ERROR);
742
0
                    goto fail_buffer_add;
743
0
                }
744
349k
                start = src;
745
349k
            }
746
747
7.41M
            src += xmlUTF8Strsize(src, 1);
748
7.41M
        }
749
626k
        else {
750
626k
            if ((start < src &&
751
626k
                 xmlBufferAdd(buf, start, src - start)) ||
752
626k
                (rlen[i_match] &&
753
626k
                 xmlBufferAdd(buf, replace[i_match], rlen[i_match])))
754
0
            {
755
0
                xmlXPathSetError(ctxt, XPATH_MEMORY_ERROR);
756
0
                goto fail_buffer_add;
757
0
            }
758
759
626k
            src += slen[i_match];
760
626k
            start = src;
761
626k
        }
762
8.03M
    }
763
764
9.70k
    if (start < src && xmlBufferAdd(buf, start, src - start)) {
765
0
        xmlXPathSetError(ctxt, XPATH_MEMORY_ERROR);
766
0
        goto fail_buffer_add;
767
0
    }
768
769
    /* create result node set */
770
771
9.70k
    exsltStrReturnString(ctxt, xmlBufferContent(buf), xmlBufferLength(buf));
772
773
    /* clean up */
774
775
9.70k
fail_buffer_add:
776
9.70k
    xmlBufferFree(buf);
777
778
9.70k
fail_buffer:
779
9.70k
fail_process_args:
780
9.70k
    if (search_set != NULL) {
781
3.81M
        for (i=0; i<n; ++i)
782
3.80M
            xmlFree(search[i]);
783
8.85k
    }
784
9.70k
    if (replace_set != NULL) {
785
3.79M
        for (i=0; i<n; ++i) {
786
3.78M
            if (replace[i] != NULL)
787
27.0k
                xmlFree(replace[i]);
788
3.78M
        }
789
8.32k
    }
790
791
9.70k
    if (mem != NULL)
792
5.91k
        xmlFree(mem);
793
794
9.70k
fail_malloc:
795
10.7k
done_empty_search:
796
10.7k
    xmlFree(string);
797
798
10.7k
fail_string:
799
10.7k
    if (search_set != NULL)
800
9.00k
        xmlXPathFreeNodeSet(search_set);
801
1.75k
    else
802
1.75k
        xmlFree(search_str);
803
804
10.7k
fail_search:
805
10.7k
    if (replace_set != NULL)
806
8.48k
        xmlXPathFreeNodeSet(replace_set);
807
2.28k
    else
808
2.28k
        xmlFree(replace_str);
809
810
10.7k
fail_replace:
811
10.7k
    return;
812
10.7k
}
813
814
/**
815
 * exsltStrRegister:
816
 *
817
 * Registers the EXSLT - Strings module
818
 */
819
820
void
821
3.07k
exsltStrRegister (void) {
822
3.07k
    xsltRegisterExtModuleFunction ((const xmlChar *) "tokenize",
823
3.07k
           EXSLT_STRINGS_NAMESPACE,
824
3.07k
           exsltStrTokenizeFunction);
825
3.07k
    xsltRegisterExtModuleFunction ((const xmlChar *) "split",
826
3.07k
           EXSLT_STRINGS_NAMESPACE,
827
3.07k
           exsltStrSplitFunction);
828
3.07k
    xsltRegisterExtModuleFunction ((const xmlChar *) "encode-uri",
829
3.07k
           EXSLT_STRINGS_NAMESPACE,
830
3.07k
           exsltStrEncodeUriFunction);
831
3.07k
    xsltRegisterExtModuleFunction ((const xmlChar *) "decode-uri",
832
3.07k
           EXSLT_STRINGS_NAMESPACE,
833
3.07k
           exsltStrDecodeUriFunction);
834
3.07k
    xsltRegisterExtModuleFunction ((const xmlChar *) "padding",
835
3.07k
           EXSLT_STRINGS_NAMESPACE,
836
3.07k
           exsltStrPaddingFunction);
837
3.07k
    xsltRegisterExtModuleFunction ((const xmlChar *) "align",
838
3.07k
           EXSLT_STRINGS_NAMESPACE,
839
3.07k
           exsltStrAlignFunction);
840
3.07k
    xsltRegisterExtModuleFunction ((const xmlChar *) "concat",
841
3.07k
           EXSLT_STRINGS_NAMESPACE,
842
3.07k
           exsltStrConcatFunction);
843
3.07k
    xsltRegisterExtModuleFunction ((const xmlChar *) "replace",
844
3.07k
           EXSLT_STRINGS_NAMESPACE,
845
3.07k
           exsltStrReplaceFunction);
846
3.07k
}
847
848
/**
849
 * exsltStrXpathCtxtRegister:
850
 *
851
 * Registers the EXSLT - Strings module for use outside XSLT
852
 */
853
int
854
exsltStrXpathCtxtRegister (xmlXPathContextPtr ctxt, const xmlChar *prefix)
855
0
{
856
0
    if (ctxt
857
0
        && prefix
858
0
        && !xmlXPathRegisterNs(ctxt,
859
0
                               prefix,
860
0
                               (const xmlChar *) EXSLT_STRINGS_NAMESPACE)
861
0
        && !xmlXPathRegisterFuncNS(ctxt,
862
0
                                   (const xmlChar *) "encode-uri",
863
0
                                   (const xmlChar *) EXSLT_STRINGS_NAMESPACE,
864
0
                                   exsltStrEncodeUriFunction)
865
0
        && !xmlXPathRegisterFuncNS(ctxt,
866
0
                                   (const xmlChar *) "decode-uri",
867
0
                                   (const xmlChar *) EXSLT_STRINGS_NAMESPACE,
868
0
                                   exsltStrDecodeUriFunction)
869
0
        && !xmlXPathRegisterFuncNS(ctxt,
870
0
                                   (const xmlChar *) "padding",
871
0
                                   (const xmlChar *) EXSLT_STRINGS_NAMESPACE,
872
0
                                   exsltStrPaddingFunction)
873
0
        && !xmlXPathRegisterFuncNS(ctxt,
874
0
                                   (const xmlChar *) "align",
875
0
                                   (const xmlChar *) EXSLT_STRINGS_NAMESPACE,
876
0
                                   exsltStrAlignFunction)
877
0
        && !xmlXPathRegisterFuncNS(ctxt,
878
0
                                   (const xmlChar *) "concat",
879
0
                                   (const xmlChar *) EXSLT_STRINGS_NAMESPACE,
880
0
                                   exsltStrConcatFunction)) {
881
0
        return 0;
882
0
    }
883
0
    return -1;
884
0
}