Coverage Report

Created: 2025-07-29 07:18

/src/quickjs/libregexp.c
Line
Count
Source (jump to first uncovered line)
1
/*
2
 * Regular Expression Engine
3
 *
4
 * Copyright (c) 2017-2018 Fabrice Bellard
5
 *
6
 * Permission is hereby granted, free of charge, to any person obtaining a copy
7
 * of this software and associated documentation files (the "Software"), to deal
8
 * in the Software without restriction, including without limitation the rights
9
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
10
 * copies of the Software, and to permit persons to whom the Software is
11
 * furnished to do so, subject to the following conditions:
12
 *
13
 * The above copyright notice and this permission notice shall be included in
14
 * all copies or substantial portions of the Software.
15
 *
16
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
17
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
18
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
19
 * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
20
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
21
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
22
 * THE SOFTWARE.
23
 */
24
#include <stdlib.h>
25
#include <stdio.h>
26
#include <stdarg.h>
27
#include <inttypes.h>
28
#include <string.h>
29
#include <assert.h>
30
31
#include "cutils.h"
32
#include "libregexp.h"
33
#include "libunicode.h"
34
35
/*
36
  TODO:
37
38
  - Add a lock step execution mode (=linear time execution guaranteed)
39
    when the regular expression is "simple" i.e. no backreference nor
40
    complicated lookahead. The opcodes are designed for this execution
41
    model.
42
*/
43
44
#if defined(TEST)
45
#define DUMP_REOP
46
#endif
47
48
typedef enum {
49
#define DEF(id, size) REOP_ ## id,
50
#include "libregexp-opcode.h"
51
#undef DEF
52
    REOP_COUNT,
53
} REOPCodeEnum;
54
55
111k
#define CAPTURE_COUNT_MAX 255
56
5.39k
#define STACK_SIZE_MAX 255
57
/* must be large enough to have a negligible runtime cost and small
58
   enough to call the interrupt callback often. */
59
1.01M
#define INTERRUPT_COUNTER_INIT 10000
60
61
/* unicode code points */
62
1.54M
#define CP_LS   0x2028
63
264k
#define CP_PS   0x2029
64
65
#define TMP_BUF_SIZE 128
66
67
typedef struct {
68
    DynBuf byte_code;
69
    const uint8_t *buf_ptr;
70
    const uint8_t *buf_end;
71
    const uint8_t *buf_start;
72
    int re_flags;
73
    BOOL is_unicode;
74
    BOOL unicode_sets; /* if set, is_unicode is also set */
75
    BOOL ignore_case;
76
    BOOL multi_line;
77
    BOOL dotall;
78
    int capture_count;
79
    int total_capture_count; /* -1 = not computed yet */
80
    int has_named_captures; /* -1 = don't know, 0 = no, 1 = yes */
81
    void *opaque;
82
    DynBuf group_names;
83
    union {
84
        char error_msg[TMP_BUF_SIZE];
85
        char tmp_buf[TMP_BUF_SIZE];
86
    } u;
87
} REParseState;
88
89
typedef struct {
90
#ifdef DUMP_REOP
91
    const char *name;
92
#endif
93
    uint8_t size;
94
} REOpCode;
95
96
static const REOpCode reopcode_info[REOP_COUNT] = {
97
#ifdef DUMP_REOP
98
#define DEF(id, size) { #id, size },
99
#else
100
#define DEF(id, size) { size },
101
#endif
102
#include "libregexp-opcode.h"
103
#undef DEF
104
};
105
106
2.02M
#define RE_HEADER_FLAGS         0
107
2.02M
#define RE_HEADER_CAPTURE_COUNT 2
108
1.01M
#define RE_HEADER_STACK_SIZE    3
109
3.29k
#define RE_HEADER_BYTECODE_LEN  4
110
111
1.02M
#define RE_HEADER_LEN 8
112
113
34.3k
static inline int is_digit(int c) {
114
34.3k
    return c >= '0' && c <= '9';
115
34.3k
}
116
117
/* insert 'len' bytes at position 'pos'. Return < 0 if error. */
118
static int dbuf_insert(DynBuf *s, int pos, int len)
119
12.6k
{
120
12.6k
    if (dbuf_realloc(s, s->size + len))
121
0
        return -1;
122
12.6k
    memmove(s->buf + pos + len, s->buf + pos, s->size - pos);
123
12.6k
    s->size += len;
124
12.6k
    return 0;
125
12.6k
}
126
127
typedef struct REString {
128
    struct REString *next;
129
    uint32_t hash;
130
    uint32_t len;
131
    uint32_t buf[];
132
} REString;
133
134
typedef struct {
135
    /* the string list is the union of 'char_range' and of the strings
136
       in hash_table[]. The strings in hash_table[] have a length !=
137
       1. */
138
    CharRange cr;
139
    uint32_t n_strings;
140
    uint32_t hash_size;
141
    int hash_bits;
142
    REString **hash_table;
143
} REStringList;
144
145
static uint32_t re_string_hash(int len, const uint32_t *buf)
146
0
{
147
0
    int i;
148
0
    uint32_t h;
149
0
    h = 1;
150
0
    for(i = 0; i < len; i++)
151
0
        h = h * 263 + buf[i];
152
0
    return h * 0x61C88647;
153
0
}
154
155
static void re_string_list_init(REParseState *s1, REStringList *s)
156
7.06k
{
157
7.06k
    cr_init(&s->cr, s1->opaque, lre_realloc);
158
7.06k
    s->n_strings = 0;
159
7.06k
    s->hash_size = 0;
160
7.06k
    s->hash_bits = 0;
161
7.06k
    s->hash_table = NULL;
162
7.06k
}
163
164
static void re_string_list_free(REStringList *s)
165
7.06k
{
166
7.06k
    REString *p, *p_next;
167
7.06k
    int i;
168
7.06k
    for(i = 0; i < s->hash_size; i++) {
169
0
        for(p = s->hash_table[i]; p != NULL; p = p_next) {
170
0
            p_next = p->next;
171
0
            lre_realloc(s->cr.mem_opaque, p, 0);
172
0
        }
173
0
    }
174
7.06k
    lre_realloc(s->cr.mem_opaque, s->hash_table, 0);
175
176
7.06k
    cr_free(&s->cr);
177
7.06k
}
178
179
static void lre_print_char(int c, BOOL is_range)
180
0
{
181
0
    if (c == '\'' || c == '\\' ||
182
0
        (is_range && (c == '-' || c == ']'))) {
183
0
        printf("\\%c", c);
184
0
    } else if (c >= ' ' && c <= 126) {
185
0
        printf("%c", c);
186
0
    } else {
187
0
        printf("\\u{%04x}", c);
188
0
    }
189
0
}
190
191
static __maybe_unused void re_string_list_dump(const char *str, const REStringList *s)
192
0
{
193
0
    REString *p;
194
0
    const CharRange *cr;
195
0
    int i, j, k;
196
0
197
0
    printf("%s:\n", str);
198
0
    printf("  ranges: [");
199
0
    cr = &s->cr;
200
0
    for(i = 0; i < cr->len; i += 2) {
201
0
        lre_print_char(cr->points[i], TRUE);
202
0
        if (cr->points[i] != cr->points[i + 1] - 1) {
203
0
            printf("-");
204
0
            lre_print_char(cr->points[i + 1] - 1, TRUE);
205
0
        }
206
0
    }
207
0
    printf("]\n");
208
0
    
209
0
    j = 0;
210
0
    for(i = 0; i < s->hash_size; i++) {
211
0
        for(p = s->hash_table[i]; p != NULL; p = p->next) {
212
0
            printf("  %d/%d: '", j, s->n_strings);
213
0
            for(k = 0; k < p->len; k++) {
214
0
                lre_print_char(p->buf[k], FALSE);
215
0
            }
216
0
            printf("'\n");
217
0
            j++;
218
0
        }
219
0
    }
220
0
}
221
222
static int re_string_find2(REStringList *s, int len, const uint32_t *buf,
223
                           uint32_t h0, BOOL add_flag)
224
0
{
225
0
    uint32_t h = 0; /* avoid warning */
226
0
    REString *p;
227
0
    if (s->n_strings != 0) {
228
0
        h = h0 >> (32 - s->hash_bits);
229
0
        for(p = s->hash_table[h]; p != NULL; p = p->next) {
230
0
            if (p->hash == h0 && p->len == len &&
231
0
                !memcmp(p->buf, buf, len * sizeof(buf[0]))) {
232
0
                return 1;
233
0
            }
234
0
        }
235
0
    }
236
    /* not found */
237
0
    if (!add_flag)
238
0
        return 0;
239
    /* increase the size of the hash table if needed */
240
0
    if (unlikely((s->n_strings + 1) > s->hash_size)) {
241
0
        REString **new_hash_table, *p_next;
242
0
        int new_hash_bits, i;
243
0
        uint32_t new_hash_size;
244
0
        new_hash_bits = max_int(s->hash_bits + 1, 4);
245
0
        new_hash_size = 1 << new_hash_bits;
246
0
        new_hash_table = lre_realloc(s->cr.mem_opaque, NULL,
247
0
                                     sizeof(new_hash_table[0]) * new_hash_size);
248
0
        if (!new_hash_table)
249
0
            return -1;
250
0
        memset(new_hash_table, 0, sizeof(new_hash_table[0]) * new_hash_size);
251
0
        for(i = 0; i < s->hash_size; i++) {
252
0
            for(p = s->hash_table[i]; p != NULL; p = p_next) {
253
0
                p_next = p->next;
254
0
                h = p->hash >> (32 - new_hash_bits);
255
0
                p->next = new_hash_table[h];
256
0
                new_hash_table[h] = p;
257
0
            }
258
0
        }
259
0
        lre_realloc(s->cr.mem_opaque, s->hash_table, 0);
260
0
        s->hash_bits = new_hash_bits;
261
0
        s->hash_size = new_hash_size;
262
0
        s->hash_table = new_hash_table;
263
0
        h = h0 >> (32 - s->hash_bits);
264
0
    }
265
266
0
    p = lre_realloc(s->cr.mem_opaque, NULL, sizeof(REString) + len * sizeof(buf[0]));
267
0
    if (!p)
268
0
        return -1;
269
0
    p->next = s->hash_table[h];
270
0
    s->hash_table[h] = p;
271
0
    s->n_strings++;
272
0
    p->hash = h0;
273
0
    p->len = len;
274
0
    memcpy(p->buf, buf, sizeof(buf[0]) * len);
275
0
    return 1;
276
0
}
277
278
static int re_string_find(REStringList *s, int len, const uint32_t *buf,
279
                          BOOL add_flag)
280
0
{
281
0
    uint32_t h0;
282
0
    h0 = re_string_hash(len, buf);
283
0
    return re_string_find2(s, len, buf, h0, add_flag);
284
0
}
285
286
/* return -1 if memory error, 0 if OK */
287
static int re_string_add(REStringList *s, int len, const uint32_t *buf)
288
0
{
289
0
    if (len == 1) {
290
0
        return cr_union_interval(&s->cr, buf[0], buf[0]);
291
0
    }
292
0
    if (re_string_find(s, len, buf, TRUE) < 0)
293
0
        return -1;
294
0
    return 0;
295
0
}
296
297
/* a = a op b */
298
static int re_string_list_op(REStringList *a, REStringList *b, int op)
299
1.38k
{
300
1.38k
    int i, ret;
301
1.38k
    REString *p, **pp;
302
303
1.38k
    if (cr_op1(&a->cr, b->cr.points, b->cr.len, op))
304
0
        return -1;
305
306
1.38k
    switch(op) {
307
1.38k
    case CR_OP_UNION:
308
1.38k
        if (b->n_strings != 0) {
309
0
            for(i = 0; i < b->hash_size; i++) {
310
0
                for(p = b->hash_table[i]; p != NULL; p = p->next) {
311
0
                    if (re_string_find2(a, p->len, p->buf, p->hash, TRUE) < 0)
312
0
                        return -1;
313
0
                }
314
0
            }
315
0
        }
316
1.38k
        break;
317
1.38k
    case CR_OP_INTER:
318
0
    case CR_OP_SUB:
319
0
        for(i = 0; i < a->hash_size; i++) {
320
0
            pp = &a->hash_table[i];
321
0
            for(;;) {
322
0
                p = *pp;
323
0
                if (p == NULL)
324
0
                    break;
325
0
                ret = re_string_find2(b, p->len, p->buf, p->hash, FALSE);
326
0
                if (op == CR_OP_SUB)
327
0
                    ret = !ret;
328
0
                if (!ret) {
329
                    /* remove it */
330
0
                    *pp = p->next;
331
0
                    a->n_strings--;
332
0
                    lre_realloc(a->cr.mem_opaque, p, 0);
333
0
                } else {
334
                    /* keep it */
335
0
                    pp = &p->next;
336
0
                }
337
0
            }
338
0
        }
339
0
        break;
340
0
    default:
341
0
        abort();
342
1.38k
    }
343
1.38k
    return 0;
344
1.38k
}
345
346
static int re_string_list_canonicalize(REParseState *s1,
347
                                       REStringList *s, BOOL is_unicode)
348
0
{
349
0
    if (cr_regexp_canonicalize(&s->cr, is_unicode))
350
0
        return -1;
351
0
    if (s->n_strings != 0) {
352
0
        REStringList a_s, *a = &a_s;
353
0
        int i, j;
354
0
        REString *p;
355
        
356
        /* XXX: simplify */
357
0
        re_string_list_init(s1, a);
358
359
0
        a->n_strings = s->n_strings;
360
0
        a->hash_size = s->hash_size;
361
0
        a->hash_bits = s->hash_bits;
362
0
        a->hash_table = s->hash_table;
363
        
364
0
        s->n_strings = 0;
365
0
        s->hash_size = 0;
366
0
        s->hash_bits = 0;
367
0
        s->hash_table = NULL;
368
369
0
        for(i = 0; i < a->hash_size; i++) {
370
0
            for(p = a->hash_table[i]; p != NULL; p = p->next) {
371
0
                for(j = 0; j < p->len; j++) {
372
0
                    p->buf[j] = lre_canonicalize(p->buf[j], is_unicode);
373
0
                }
374
0
                if (re_string_add(s, p->len, p->buf)) {
375
0
                    re_string_list_free(a);
376
0
                    return -1;
377
0
                }
378
0
            }
379
0
        }
380
0
        re_string_list_free(a);
381
0
    }
382
0
    return 0;
383
0
}
384
385
static const uint16_t char_range_d[] = {
386
    1,
387
    0x0030, 0x0039 + 1,
388
};
389
390
/* code point ranges for Zs,Zl or Zp property */
391
static const uint16_t char_range_s[] = {
392
    10,
393
    0x0009, 0x000D + 1,
394
    0x0020, 0x0020 + 1,
395
    0x00A0, 0x00A0 + 1,
396
    0x1680, 0x1680 + 1,
397
    0x2000, 0x200A + 1,
398
    /* 2028;LINE SEPARATOR;Zl;0;WS;;;;;N;;;;; */
399
    /* 2029;PARAGRAPH SEPARATOR;Zp;0;B;;;;;N;;;;; */
400
    0x2028, 0x2029 + 1,
401
    0x202F, 0x202F + 1,
402
    0x205F, 0x205F + 1,
403
    0x3000, 0x3000 + 1,
404
    /* FEFF;ZERO WIDTH NO-BREAK SPACE;Cf;0;BN;;;;;N;BYTE ORDER MARK;;;; */
405
    0xFEFF, 0xFEFF + 1,
406
};
407
408
static const uint16_t char_range_w[] = {
409
    4,
410
    0x0030, 0x0039 + 1,
411
    0x0041, 0x005A + 1,
412
    0x005F, 0x005F + 1,
413
    0x0061, 0x007A + 1,
414
};
415
416
209k
#define CLASS_RANGE_BASE 0x40000000
417
418
typedef enum {
419
    CHAR_RANGE_d,
420
    CHAR_RANGE_D,
421
    CHAR_RANGE_s,
422
    CHAR_RANGE_S,
423
    CHAR_RANGE_w,
424
    CHAR_RANGE_W,
425
} CharRangeEnum;
426
427
static const uint16_t * const char_range_table[] = {
428
    char_range_d,
429
    char_range_s,
430
    char_range_w,
431
};
432
433
static int cr_init_char_range(REParseState *s, REStringList *cr, uint32_t c)
434
4.13k
{
435
4.13k
    BOOL invert;
436
4.13k
    const uint16_t *c_pt;
437
4.13k
    int len, i;
438
439
4.13k
    invert = c & 1;
440
4.13k
    c_pt = char_range_table[c >> 1];
441
4.13k
    len = *c_pt++;
442
4.13k
    re_string_list_init(s, cr);
443
63.6k
    for(i = 0; i < len * 2; i++) {
444
59.5k
        if (cr_add_point(&cr->cr, c_pt[i]))
445
0
            goto fail;
446
59.5k
    }
447
4.13k
    if (invert) {
448
2.49k
        if (cr_invert(&cr->cr))
449
0
            goto fail;
450
2.49k
    }
451
4.13k
    return 0;
452
0
 fail:
453
0
    re_string_list_free(cr);
454
0
    return -1;
455
4.13k
}
456
457
#ifdef DUMP_REOP
458
static __maybe_unused void lre_dump_bytecode(const uint8_t *buf,
459
                                                     int buf_len)
460
{
461
    int pos, len, opcode, bc_len, re_flags, i;
462
    uint32_t val;
463
464
    assert(buf_len >= RE_HEADER_LEN);
465
466
    re_flags = lre_get_flags(buf);
467
    bc_len = get_u32(buf + RE_HEADER_BYTECODE_LEN);
468
    assert(bc_len + RE_HEADER_LEN <= buf_len);
469
    printf("flags: 0x%x capture_count=%d stack_size=%d\n",
470
           re_flags, buf[RE_HEADER_CAPTURE_COUNT], buf[RE_HEADER_STACK_SIZE]);
471
    if (re_flags & LRE_FLAG_NAMED_GROUPS) {
472
        const char *p;
473
        p = (char *)buf + RE_HEADER_LEN + bc_len;
474
        printf("named groups: ");
475
        for(i = 1; i < buf[RE_HEADER_CAPTURE_COUNT]; i++) {
476
            if (i != 1)
477
                printf(",");
478
            printf("<%s>", p);
479
            p += strlen(p) + 1;
480
        }
481
        printf("\n");
482
        assert(p == (char *)(buf + buf_len));
483
    }
484
    printf("bytecode_len=%d\n", bc_len);
485
486
    buf += RE_HEADER_LEN;
487
    pos = 0;
488
    while (pos < bc_len) {
489
        printf("%5u: ", pos);
490
        opcode = buf[pos];
491
        len = reopcode_info[opcode].size;
492
        if (opcode >= REOP_COUNT) {
493
            printf(" invalid opcode=0x%02x\n", opcode);
494
            break;
495
        }
496
        if ((pos + len) > bc_len) {
497
            printf(" buffer overflow (opcode=0x%02x)\n", opcode);
498
            break;
499
        }
500
        printf("%s", reopcode_info[opcode].name);
501
        switch(opcode) {
502
        case REOP_char:
503
        case REOP_char_i:
504
            val = get_u16(buf + pos + 1);
505
            if (val >= ' ' && val <= 126)
506
                printf(" '%c'", val);
507
            else
508
                printf(" 0x%04x", val);
509
            break;
510
        case REOP_char32:
511
        case REOP_char32_i:
512
            val = get_u32(buf + pos + 1);
513
            if (val >= ' ' && val <= 126)
514
                printf(" '%c'", val);
515
            else
516
                printf(" 0x%08x", val);
517
            break;
518
        case REOP_goto:
519
        case REOP_split_goto_first:
520
        case REOP_split_next_first:
521
        case REOP_loop:
522
        case REOP_lookahead:
523
        case REOP_negative_lookahead:
524
            val = get_u32(buf + pos + 1);
525
            val += (pos + 5);
526
            printf(" %u", val);
527
            break;
528
        case REOP_simple_greedy_quant:
529
            printf(" %u %u %u %u",
530
                   get_u32(buf + pos + 1) + (pos + 17),
531
                   get_u32(buf + pos + 1 + 4),
532
                   get_u32(buf + pos + 1 + 8),
533
                   get_u32(buf + pos + 1 + 12));
534
            break;
535
        case REOP_save_start:
536
        case REOP_save_end:
537
        case REOP_back_reference:
538
        case REOP_back_reference_i:
539
        case REOP_backward_back_reference:
540
        case REOP_backward_back_reference_i:
541
            printf(" %u", buf[pos + 1]);
542
            break;
543
        case REOP_save_reset:
544
            printf(" %u %u", buf[pos + 1], buf[pos + 2]);
545
            break;
546
        case REOP_push_i32:
547
            val = get_u32(buf + pos + 1);
548
            printf(" %d", val);
549
            break;
550
        case REOP_range:
551
        case REOP_range_i:
552
            {
553
                int n, i;
554
                n = get_u16(buf + pos + 1);
555
                len += n * 4;
556
                for(i = 0; i < n * 2; i++) {
557
                    val = get_u16(buf + pos + 3 + i * 2);
558
                    printf(" 0x%04x", val);
559
                }
560
            }
561
            break;
562
        case REOP_range32:
563
        case REOP_range32_i:
564
            {
565
                int n, i;
566
                n = get_u16(buf + pos + 1);
567
                len += n * 8;
568
                for(i = 0; i < n * 2; i++) {
569
                    val = get_u32(buf + pos + 3 + i * 4);
570
                    printf(" 0x%08x", val);
571
                }
572
            }
573
            break;
574
        default:
575
            break;
576
        }
577
        printf("\n");
578
        pos += len;
579
    }
580
}
581
#endif
582
583
static void re_emit_op(REParseState *s, int op)
584
35.8k
{
585
35.8k
    dbuf_putc(&s->byte_code, op);
586
35.8k
}
587
588
/* return the offset of the u32 value */
589
static int re_emit_op_u32(REParseState *s, int op, uint32_t val)
590
30.4k
{
591
30.4k
    int pos;
592
30.4k
    dbuf_putc(&s->byte_code, op);
593
30.4k
    pos = s->byte_code.size;
594
30.4k
    dbuf_put_u32(&s->byte_code, val);
595
30.4k
    return pos;
596
30.4k
}
597
598
static int re_emit_goto(REParseState *s, int op, uint32_t val)
599
7.51k
{
600
7.51k
    int pos;
601
7.51k
    dbuf_putc(&s->byte_code, op);
602
7.51k
    pos = s->byte_code.size;
603
7.51k
    dbuf_put_u32(&s->byte_code, val - (pos + 4));
604
7.51k
    return pos;
605
7.51k
}
606
607
static void re_emit_op_u8(REParseState *s, int op, uint32_t val)
608
56.0k
{
609
56.0k
    dbuf_putc(&s->byte_code, op);
610
56.0k
    dbuf_putc(&s->byte_code, val);
611
56.0k
}
612
613
static void re_emit_op_u16(REParseState *s, int op, uint32_t val)
614
169k
{
615
169k
    dbuf_putc(&s->byte_code, op);
616
169k
    dbuf_put_u16(&s->byte_code, val);
617
169k
}
618
619
static int __attribute__((format(printf, 2, 3))) re_parse_error(REParseState *s, const char *fmt, ...)
620
2.63k
{
621
2.63k
    va_list ap;
622
2.63k
    va_start(ap, fmt);
623
2.63k
    vsnprintf(s->u.error_msg, sizeof(s->u.error_msg), fmt, ap);
624
2.63k
    va_end(ap);
625
2.63k
    return -1;
626
2.63k
}
627
628
static int re_parse_out_of_memory(REParseState *s)
629
0
{
630
0
    return re_parse_error(s, "out of memory");
631
0
}
632
633
/* If allow_overflow is false, return -1 in case of
634
   overflow. Otherwise return INT32_MAX. */
635
static int parse_digits(const uint8_t **pp, BOOL allow_overflow)
636
22.2k
{
637
22.2k
    const uint8_t *p;
638
22.2k
    uint64_t v;
639
22.2k
    int c;
640
641
22.2k
    p = *pp;
642
22.2k
    v = 0;
643
54.1k
    for(;;) {
644
54.1k
        c = *p;
645
54.1k
        if (c < '0' || c > '9')
646
21.9k
            break;
647
32.2k
        v = v * 10 + c - '0';
648
32.2k
        if (v >= INT32_MAX) {
649
1.45k
            if (allow_overflow)
650
1.19k
                v = INT32_MAX;
651
260
            else
652
260
                return -1;
653
1.45k
        }
654
31.9k
        p++;
655
31.9k
    }
656
21.9k
    *pp = p;
657
21.9k
    return v;
658
22.2k
}
659
660
static int re_parse_expect(REParseState *s, const uint8_t **pp, int c)
661
16.9k
{
662
16.9k
    const uint8_t *p;
663
16.9k
    p = *pp;
664
16.9k
    if (*p != c)
665
582
        return re_parse_error(s, "expecting '%c'", c);
666
16.3k
    p++;
667
16.3k
    *pp = p;
668
16.3k
    return 0;
669
16.9k
}
670
671
/* Parse an escape sequence, *pp points after the '\':
672
   allow_utf16 value:
673
   0 : no UTF-16 escapes allowed
674
   1 : UTF-16 escapes allowed
675
   2 : UTF-16 escapes allowed and escapes of surrogate pairs are
676
   converted to a unicode character (unicode regexp case).
677
678
   Return the unicode char and update *pp if recognized,
679
   return -1 if malformed escape,
680
   return -2 otherwise. */
681
int lre_parse_escape(const uint8_t **pp, int allow_utf16)
682
112k
{
683
112k
    const uint8_t *p;
684
112k
    uint32_t c;
685
686
112k
    p = *pp;
687
112k
    c = *p++;
688
112k
    switch(c) {
689
4.39k
    case 'b':
690
4.39k
        c = '\b';
691
4.39k
        break;
692
252
    case 'f':
693
252
        c = '\f';
694
252
        break;
695
219
    case 'n':
696
219
        c = '\n';
697
219
        break;
698
200
    case 'r':
699
200
        c = '\r';
700
200
        break;
701
201
    case 't':
702
201
        c = '\t';
703
201
        break;
704
1.10k
    case 'v':
705
1.10k
        c = '\v';
706
1.10k
        break;
707
468
    case 'x':
708
14.2k
    case 'u':
709
14.2k
        {
710
14.2k
            int h, n, i;
711
14.2k
            uint32_t c1;
712
713
14.2k
            if (*p == '{' && allow_utf16) {
714
1.32k
                p++;
715
1.32k
                c = 0;
716
3.10k
                for(;;) {
717
3.10k
                    h = from_hex(*p++);
718
3.10k
                    if (h < 0)
719
889
                        return -1;
720
2.21k
                    c = (c << 4) | h;
721
2.21k
                    if (c > 0x10FFFF)
722
241
                        return -1;
723
1.97k
                    if (*p == '}')
724
199
                        break;
725
1.97k
                }
726
199
                p++;
727
12.8k
            } else {
728
12.8k
                if (c == 'x') {
729
468
                    n = 2;
730
12.4k
                } else {
731
12.4k
                    n = 4;
732
12.4k
                }
733
734
12.8k
                c = 0;
735
51.4k
                for(i = 0; i < n; i++) {
736
43.5k
                    h = from_hex(*p++);
737
43.5k
                    if (h < 0) {
738
4.96k
                        return -1;
739
4.96k
                    }
740
38.5k
                    c = (c << 4) | h;
741
38.5k
                }
742
7.92k
                if (is_hi_surrogate(c) &&
743
7.92k
                    allow_utf16 == 2 && p[0] == '\\' && p[1] == 'u') {
744
                    /* convert an escaped surrogate pair into a
745
                       unicode char */
746
5.08k
                    c1 = 0;
747
15.5k
                    for(i = 0; i < 4; i++) {
748
14.5k
                        h = from_hex(p[2 + i]);
749
14.5k
                        if (h < 0)
750
4.13k
                            break;
751
10.4k
                        c1 = (c1 << 4) | h;
752
10.4k
                    }
753
5.08k
                    if (i == 4 && is_lo_surrogate(c1)) {
754
441
                        p += 6;
755
441
                        c = from_surrogate(c, c1);
756
441
                    }
757
5.08k
                }
758
7.92k
            }
759
14.2k
        }
760
8.12k
        break;
761
8.12k
    case '0': case '1': case '2': case '3':
762
3.71k
    case '4': case '5': case '6': case '7':
763
3.71k
        c -= '0';
764
3.71k
        if (allow_utf16 == 2) {
765
            /* only accept \0 not followed by digit */
766
0
            if (c != 0 || is_digit(*p))
767
0
                return -1;
768
3.71k
        } else {
769
            /* parse a legacy octal sequence */
770
3.71k
            uint32_t v;
771
3.71k
            v = *p - '0';
772
3.71k
            if (v > 7)
773
3.06k
                break;
774
652
            c = (c << 3) | v;
775
652
            p++;
776
652
            if (c >= 32)
777
224
                break;
778
428
            v = *p - '0';
779
428
            if (v > 7)
780
212
                break;
781
216
            c = (c << 3) | v;
782
216
            p++;
783
216
        }
784
216
        break;
785
87.9k
    default:
786
87.9k
        return -2;
787
112k
    }
788
18.2k
    *pp = p;
789
18.2k
    return c;
790
112k
}
791
792
#ifdef CONFIG_ALL_UNICODE
793
/* XXX: we use the same chars for name and value */
794
static BOOL is_unicode_char(int c)
795
0
{
796
0
    return ((c >= '0' && c <= '9') ||
797
0
            (c >= 'A' && c <= 'Z') ||
798
0
            (c >= 'a' && c <= 'z') ||
799
0
            (c == '_'));
800
0
}
801
802
/* XXX: memory error test */
803
static void seq_prop_cb(void *opaque, const uint32_t *seq, int seq_len)
804
0
{
805
0
    REStringList *sl = opaque;
806
0
    re_string_add(sl, seq_len, seq);
807
0
}
808
809
static int parse_unicode_property(REParseState *s, REStringList *cr,
810
                                  const uint8_t **pp, BOOL is_inv,
811
                                  BOOL allow_sequence_prop)
812
0
{
813
0
    const uint8_t *p;
814
0
    char name[64], value[64];
815
0
    char *q;
816
0
    BOOL script_ext;
817
0
    int ret;
818
819
0
    p = *pp;
820
0
    if (*p != '{')
821
0
        return re_parse_error(s, "expecting '{' after \\p");
822
0
    p++;
823
0
    q = name;
824
0
    while (is_unicode_char(*p)) {
825
0
        if ((q - name) >= sizeof(name) - 1)
826
0
            goto unknown_property_name;
827
0
        *q++ = *p++;
828
0
    }
829
0
    *q = '\0';
830
0
    q = value;
831
0
    if (*p == '=') {
832
0
        p++;
833
0
        while (is_unicode_char(*p)) {
834
0
            if ((q - value) >= sizeof(value) - 1)
835
0
                return re_parse_error(s, "unknown unicode property value");
836
0
            *q++ = *p++;
837
0
        }
838
0
    }
839
0
    *q = '\0';
840
0
    if (*p != '}')
841
0
        return re_parse_error(s, "expecting '}'");
842
0
    p++;
843
    //    printf("name=%s value=%s\n", name, value);
844
845
0
    if (!strcmp(name, "Script") || !strcmp(name, "sc")) {
846
0
        script_ext = FALSE;
847
0
        goto do_script;
848
0
    } else if (!strcmp(name, "Script_Extensions") || !strcmp(name, "scx")) {
849
0
        script_ext = TRUE;
850
0
    do_script:
851
0
        re_string_list_init(s, cr);
852
0
        ret = unicode_script(&cr->cr, value, script_ext);
853
0
        if (ret) {
854
0
            re_string_list_free(cr);
855
0
            if (ret == -2)
856
0
                return re_parse_error(s, "unknown unicode script");
857
0
            else
858
0
                goto out_of_memory;
859
0
        }
860
0
    } else if (!strcmp(name, "General_Category") || !strcmp(name, "gc")) {
861
0
        re_string_list_init(s, cr);
862
0
        ret = unicode_general_category(&cr->cr, value);
863
0
        if (ret) {
864
0
            re_string_list_free(cr);
865
0
            if (ret == -2)
866
0
                return re_parse_error(s, "unknown unicode general category");
867
0
            else
868
0
                goto out_of_memory;
869
0
        }
870
0
    } else if (value[0] == '\0') {
871
0
        re_string_list_init(s, cr);
872
0
        ret = unicode_general_category(&cr->cr, name);
873
0
        if (ret == -1) {
874
0
            re_string_list_free(cr);
875
0
            goto out_of_memory;
876
0
        }
877
0
        if (ret < 0) {
878
0
            ret = unicode_prop(&cr->cr, name);
879
0
            if (ret == -1) {
880
0
                re_string_list_free(cr);
881
0
                goto out_of_memory;
882
0
            }
883
0
        }
884
0
        if (ret < 0 && !is_inv && allow_sequence_prop) {
885
0
            CharRange cr_tmp;
886
0
            cr_init(&cr_tmp, s->opaque, lre_realloc);
887
0
            ret = unicode_sequence_prop(name, seq_prop_cb, cr, &cr_tmp);
888
0
            cr_free(&cr_tmp);
889
0
            if (ret == -1) {
890
0
                re_string_list_free(cr);
891
0
                goto out_of_memory;
892
0
            }
893
0
        }
894
0
        if (ret < 0)
895
0
            goto unknown_property_name;
896
0
    } else {
897
0
    unknown_property_name:
898
0
        return re_parse_error(s, "unknown unicode property name");
899
0
    }
900
901
    /* the ordering of case folding and inversion  differs with
902
       unicode_sets. 'unicode_sets' ordering is more consistent */
903
    /* XXX: the spec seems incorrect, we do it as the other engines
904
       seem to do it. */
905
0
    if (s->ignore_case && s->unicode_sets) {
906
0
        if (re_string_list_canonicalize(s, cr, s->is_unicode)) {
907
0
            re_string_list_free(cr);
908
0
            goto out_of_memory;
909
0
        }
910
0
    }
911
0
    if (is_inv) {
912
0
        if (cr_invert(&cr->cr)) {
913
0
            re_string_list_free(cr);
914
0
            goto out_of_memory;
915
0
        }
916
0
    }
917
0
    if (s->ignore_case && !s->unicode_sets) {
918
0
        if (re_string_list_canonicalize(s, cr, s->is_unicode)) {
919
0
            re_string_list_free(cr);
920
0
            goto out_of_memory;
921
0
        }
922
0
    }
923
0
    *pp = p;
924
0
    return 0;
925
0
 out_of_memory:
926
0
    return re_parse_out_of_memory(s);
927
0
}
928
#endif /* CONFIG_ALL_UNICODE */
929
930
static int get_class_atom(REParseState *s, REStringList *cr,
931
                          const uint8_t **pp, BOOL inclass);
932
933
static int parse_class_string_disjunction(REParseState *s, REStringList *cr,
934
                                          const uint8_t **pp)
935
0
{
936
0
    const uint8_t *p;
937
0
    DynBuf str;
938
0
    int c;
939
    
940
0
    p = *pp;
941
0
    if (*p != '{')
942
0
        return re_parse_error(s, "expecting '{' after \\q");
943
944
0
    dbuf_init2(&str, s->opaque, lre_realloc);
945
0
    re_string_list_init(s, cr);
946
    
947
0
    p++;
948
0
    for(;;) {
949
0
        str.size = 0;
950
0
        while (*p != '}' && *p != '|') {
951
0
            c = get_class_atom(s, NULL, &p, FALSE);
952
0
            if (c < 0)
953
0
                goto fail;
954
0
            if (dbuf_put_u32(&str, c)) {
955
0
                re_parse_out_of_memory(s);
956
0
                goto fail;
957
0
            }
958
0
        }
959
0
        if (re_string_add(cr, str.size / 4, (uint32_t *)str.buf)) {
960
0
            re_parse_out_of_memory(s);
961
0
            goto fail;
962
0
        }
963
0
        if (*p == '}')
964
0
            break;
965
0
        p++;
966
0
    }
967
0
    if (s->ignore_case) {
968
0
        if (re_string_list_canonicalize(s, cr, TRUE))
969
0
            goto fail;
970
0
    }
971
0
    p++; /* skip the '}' */
972
0
    dbuf_free(&str);
973
0
    *pp = p;
974
0
    return 0;
975
0
 fail:
976
0
    dbuf_free(&str);
977
0
    re_string_list_free(cr);
978
0
    return -1;
979
0
}
980
981
/* return -1 if error otherwise the character or a class range
982
   (CLASS_RANGE_BASE) if cr != NULL. In case of class range, 'cr' is
983
   initialized. Otherwise, it is ignored. */
984
static int get_class_atom(REParseState *s, REStringList *cr,
985
                          const uint8_t **pp, BOOL inclass)
986
200k
{
987
200k
    const uint8_t *p;
988
200k
    uint32_t c;
989
200k
    int ret;
990
991
200k
    p = *pp;
992
993
200k
    c = *p;
994
200k
    switch(c) {
995
21.6k
    case '\\':
996
21.6k
        p++;
997
21.6k
        if (p >= s->buf_end)
998
86
            goto unexpected_end;
999
21.5k
        c = *p++;
1000
21.5k
        switch(c) {
1001
381
        case 'd':
1002
381
            c = CHAR_RANGE_d;
1003
381
            goto class_range;
1004
283
        case 'D':
1005
283
            c = CHAR_RANGE_D;
1006
283
            goto class_range;
1007
804
        case 's':
1008
804
            c = CHAR_RANGE_s;
1009
804
            goto class_range;
1010
1.73k
        case 'S':
1011
1.73k
            c = CHAR_RANGE_S;
1012
1.73k
            goto class_range;
1013
451
        case 'w':
1014
451
            c = CHAR_RANGE_w;
1015
451
            goto class_range;
1016
479
        case 'W':
1017
479
            c = CHAR_RANGE_W;
1018
4.13k
        class_range:
1019
4.13k
            if (!cr)
1020
0
                goto default_escape;
1021
4.13k
            if (cr_init_char_range(s, cr, c))
1022
0
                return -1;
1023
4.13k
            c = CLASS_RANGE_BASE;
1024
4.13k
            break;
1025
2.51k
        case 'c':
1026
2.51k
            c = *p;
1027
2.51k
            if ((c >= 'a' && c <= 'z') ||
1028
2.51k
                (c >= 'A' && c <= 'Z') ||
1029
2.51k
                (((c >= '0' && c <= '9') || c == '_') &&
1030
2.11k
                 inclass && !s->is_unicode)) {   /* Annex B.1.4 */
1031
578
                c &= 0x1f;
1032
578
                p++;
1033
1.93k
            } else if (s->is_unicode) {
1034
0
                goto invalid_escape;
1035
1.93k
            } else {
1036
                /* otherwise return '\' and 'c' */
1037
1.93k
                p--;
1038
1.93k
                c = '\\';
1039
1.93k
            }
1040
2.51k
            break;
1041
2.51k
        case '-':
1042
543
            if (!inclass && s->is_unicode)
1043
0
                goto invalid_escape;
1044
543
            break;
1045
543
        case '^':
1046
392
        case '$':
1047
900
        case '\\':
1048
989
        case '.':
1049
1.21k
        case '*':
1050
1.42k
        case '+':
1051
1.63k
        case '?':
1052
1.87k
        case '(':
1053
2.03k
        case ')':
1054
2.22k
        case '[':
1055
2.29k
        case ']':
1056
2.51k
        case '{':
1057
2.92k
        case '}':
1058
3.11k
        case '|':
1059
3.27k
        case '/':
1060
            /* always valid to escape these characters */
1061
3.27k
            break;
1062
0
#ifdef CONFIG_ALL_UNICODE
1063
209
        case 'p':
1064
418
        case 'P':
1065
418
            if (s->is_unicode && cr) {
1066
0
                if (parse_unicode_property(s, cr, &p, (c == 'P'), s->unicode_sets))
1067
0
                    return -1;
1068
0
                c = CLASS_RANGE_BASE;
1069
0
                break;
1070
0
            }
1071
418
            goto default_escape;
1072
418
#endif
1073
418
        case 'q':
1074
177
            if (s->unicode_sets && cr && inclass) {
1075
0
                if (parse_class_string_disjunction(s, cr, &p))
1076
0
                    return -1;
1077
0
                c = CLASS_RANGE_BASE;
1078
0
                break;
1079
0
            }
1080
177
            goto default_escape;
1081
10.5k
        default:
1082
11.1k
        default_escape:
1083
11.1k
            p--;
1084
11.1k
            ret = lre_parse_escape(&p, s->is_unicode * 2);
1085
11.1k
            if (ret >= 0) {
1086
5.81k
                c = ret;
1087
5.81k
            } else {
1088
5.29k
                if (s->is_unicode) {
1089
0
                invalid_escape:
1090
0
                    return re_parse_error(s, "invalid escape sequence in regular expression");
1091
5.29k
                } else {
1092
                    /* just ignore the '\' */
1093
5.29k
                    goto normal_char;
1094
5.29k
                }
1095
5.29k
            }
1096
5.81k
            break;
1097
21.5k
        }
1098
16.2k
        break;
1099
16.2k
    case '\0':
1100
573
        if (p >= s->buf_end) {
1101
659
        unexpected_end:
1102
659
            return re_parse_error(s, "unexpected end");
1103
573
        }
1104
        /* fall thru */
1105
0
        goto normal_char;
1106
1107
560
    case '&':
1108
1.33k
    case '!':
1109
2.84k
    case '#':
1110
3.28k
    case '$':
1111
3.69k
    case '%':
1112
4.19k
    case '*':
1113
4.57k
    case '+':
1114
10.8k
    case ',':
1115
11.1k
    case '.':
1116
15.0k
    case ':':
1117
15.4k
    case ';':
1118
18.6k
    case '<':
1119
19.9k
    case '=':
1120
21.5k
    case '>':
1121
22.2k
    case '?':
1122
22.6k
    case '@':
1123
22.8k
    case '^':
1124
23.2k
    case '`':
1125
23.6k
    case '~':
1126
23.6k
        if (s->unicode_sets && p[1] == c) {
1127
            /* forbidden double characters */
1128
0
            return re_parse_error(s, "invalid class set operation in regular expression");
1129
0
        }
1130
23.6k
        goto normal_char;
1131
1132
23.6k
    case '(':
1133
1.78k
    case ')':
1134
2.96k
    case '[':
1135
3.41k
    case ']':
1136
14.0k
    case '{':
1137
14.7k
    case '}':
1138
16.8k
    case '/':
1139
24.4k
    case '-':
1140
24.6k
    case '|':
1141
24.6k
        if (s->unicode_sets) {
1142
            /* invalid characters in unicode sets */
1143
0
            return re_parse_error(s, "invalid character in class in regular expression");
1144
0
        }
1145
24.6k
        goto normal_char;
1146
1147
129k
    default:
1148
183k
    normal_char:
1149
        /* normal char */
1150
183k
        if (c >= 128) {
1151
6.00k
            c = unicode_from_utf8(p, UTF8_CHAR_LEN_MAX, &p);
1152
6.00k
            if ((unsigned)c > 0xffff && !s->is_unicode) {
1153
                /* XXX: should handle non BMP-1 code points */
1154
391
                return re_parse_error(s, "malformed unicode char");
1155
391
            }
1156
177k
        } else {
1157
177k
            p++;
1158
177k
        }
1159
182k
        break;
1160
200k
    }
1161
199k
    *pp = p;
1162
199k
    return c;
1163
200k
}
1164
1165
static int re_emit_range(REParseState *s, const CharRange *cr)
1166
4.47k
{
1167
4.47k
    int len, i;
1168
4.47k
    uint32_t high;
1169
1170
4.47k
    len = (unsigned)cr->len / 2;
1171
4.47k
    if (len >= 65535)
1172
0
        return re_parse_error(s, "too many ranges");
1173
4.47k
    if (len == 0) {
1174
1.05k
        re_emit_op_u32(s, REOP_char32, -1);
1175
3.41k
    } else {
1176
3.41k
        high = cr->points[cr->len - 1];
1177
3.41k
        if (high == UINT32_MAX)
1178
1.68k
            high = cr->points[cr->len - 2];
1179
3.41k
        if (high <= 0xffff) {
1180
            /* can use 16 bit ranges with the conversion that 0xffff =
1181
               infinity */
1182
3.03k
            re_emit_op_u16(s, s->ignore_case ? REOP_range_i : REOP_range, len);
1183
25.9k
            for(i = 0; i < cr->len; i += 2) {
1184
22.9k
                dbuf_put_u16(&s->byte_code, cr->points[i]);
1185
22.9k
                high = cr->points[i + 1] - 1;
1186
22.9k
                if (high == UINT32_MAX - 1)
1187
1.66k
                    high = 0xffff;
1188
22.9k
                dbuf_put_u16(&s->byte_code, high);
1189
22.9k
            }
1190
3.03k
        } else {
1191
381
            re_emit_op_u16(s, s->ignore_case ? REOP_range32_i : REOP_range32, len);
1192
4.96k
            for(i = 0; i < cr->len; i += 2) {
1193
4.58k
                dbuf_put_u32(&s->byte_code, cr->points[i]);
1194
4.58k
                dbuf_put_u32(&s->byte_code, cr->points[i + 1] - 1);
1195
4.58k
            }
1196
381
        }
1197
3.41k
    }
1198
4.47k
    return 0;
1199
4.47k
}
1200
1201
static int re_string_cmp_len(const void *a, const void *b, void *arg)
1202
0
{
1203
0
    REString *p1 = *(REString **)a;
1204
0
    REString *p2 = *(REString **)b;
1205
0
    return (p1->len < p2->len) - (p1->len > p2->len);
1206
0
}
1207
1208
static void re_emit_char(REParseState *s, int c)
1209
166k
{
1210
166k
    if (c <= 0xffff)
1211
166k
        re_emit_op_u16(s, s->ignore_case ? REOP_char_i : REOP_char, c);
1212
0
    else
1213
0
        re_emit_op_u32(s, s->ignore_case ? REOP_char32_i : REOP_char32, c);
1214
166k
}
1215
1216
static int re_emit_string_list(REParseState *s, const REStringList *sl)
1217
4.47k
{
1218
4.47k
    REString **tab, *p;
1219
4.47k
    int i, j, split_pos, last_match_pos, n;
1220
4.47k
    BOOL has_empty_string, is_last;
1221
    
1222
    //    re_string_list_dump("sl", sl);
1223
4.47k
    if (sl->n_strings == 0) {
1224
        /* simple case: only characters */
1225
4.47k
        if (re_emit_range(s, &sl->cr))
1226
0
            return -1;
1227
4.47k
    } else {
1228
        /* at least one string list is present : match the longest ones first */
1229
        /* XXX: add a new op_switch opcode to compile as a trie */
1230
0
        tab = lre_realloc(s->opaque, NULL, sizeof(tab[0]) * sl->n_strings);
1231
0
        if (!tab) {
1232
0
            re_parse_out_of_memory(s);
1233
0
            return -1;
1234
0
        }
1235
0
        has_empty_string = FALSE;
1236
0
        n = 0;
1237
0
        for(i = 0; i < sl->hash_size; i++) {
1238
0
            for(p = sl->hash_table[i]; p != NULL; p = p->next) {
1239
0
                if (p->len == 0) {
1240
0
                    has_empty_string = TRUE;
1241
0
                } else {
1242
0
                    tab[n++] = p;
1243
0
                }
1244
0
            }
1245
0
        }
1246
0
        assert(n <= sl->n_strings);
1247
        
1248
0
        rqsort(tab, n, sizeof(tab[0]), re_string_cmp_len, NULL);
1249
1250
0
        last_match_pos = -1;
1251
0
        for(i = 0; i < n; i++) {
1252
0
            p = tab[i];
1253
0
            is_last = !has_empty_string && sl->cr.len == 0 && i == (n - 1);
1254
0
            if (!is_last)
1255
0
                split_pos = re_emit_op_u32(s, REOP_split_next_first, 0);
1256
0
            else
1257
0
                split_pos = 0;
1258
0
            for(j = 0; j < p->len; j++) {
1259
0
                re_emit_char(s, p->buf[j]);
1260
0
            }
1261
0
            if (!is_last) {
1262
0
                last_match_pos = re_emit_op_u32(s, REOP_goto, last_match_pos);
1263
0
                put_u32(s->byte_code.buf + split_pos, s->byte_code.size - (split_pos + 4));
1264
0
            }
1265
0
        }
1266
1267
0
        if (sl->cr.len != 0) {
1268
            /* char range */
1269
0
            is_last = !has_empty_string;
1270
0
            if (!is_last)
1271
0
                split_pos = re_emit_op_u32(s, REOP_split_next_first, 0);
1272
0
            else
1273
0
                split_pos = 0; /* not used */
1274
0
            if (re_emit_range(s, &sl->cr)) {
1275
0
                lre_realloc(s->opaque, tab, 0);
1276
0
                return -1;
1277
0
            }
1278
0
            if (!is_last)
1279
0
                put_u32(s->byte_code.buf + split_pos, s->byte_code.size - (split_pos + 4));
1280
0
        }
1281
1282
        /* patch the 'goto match' */
1283
0
        while (last_match_pos != -1) {
1284
0
            int next_pos = get_u32(s->byte_code.buf + last_match_pos);
1285
0
            put_u32(s->byte_code.buf + last_match_pos, s->byte_code.size - (last_match_pos + 4));
1286
0
            last_match_pos = next_pos;
1287
0
        }
1288
        
1289
0
        lre_realloc(s->opaque, tab, 0);
1290
0
    }
1291
4.47k
    return 0;
1292
4.47k
}
1293
1294
static int re_parse_nested_class(REParseState *s, REStringList *cr, const uint8_t **pp);
1295
1296
static int re_parse_class_set_operand(REParseState *s, REStringList *cr, const uint8_t **pp)
1297
0
{
1298
0
    int c1;
1299
0
    const uint8_t *p = *pp;
1300
    
1301
0
    if (*p == '[') {
1302
0
        if (re_parse_nested_class(s, cr, pp))
1303
0
            return -1;
1304
0
    } else {
1305
0
        c1 = get_class_atom(s, cr, pp, TRUE);
1306
0
        if (c1 < 0)
1307
0
            return -1;
1308
0
        if (c1 < CLASS_RANGE_BASE) {
1309
            /* create a range with a single character */
1310
0
            re_string_list_init(s, cr);
1311
0
            if (s->ignore_case)
1312
0
                c1 = lre_canonicalize(c1, s->is_unicode);
1313
0
            if (cr_union_interval(&cr->cr, c1, c1)) {
1314
0
                re_string_list_free(cr);
1315
0
                return -1;
1316
0
            }
1317
0
        }
1318
0
    }
1319
0
    return 0;
1320
0
}
1321
1322
static int re_parse_nested_class(REParseState *s, REStringList *cr, const uint8_t **pp)
1323
2.92k
{
1324
2.92k
    const uint8_t *p;
1325
2.92k
    uint32_t c1, c2;
1326
2.92k
    int ret;
1327
2.92k
    REStringList cr1_s, *cr1 = &cr1_s;
1328
2.92k
    BOOL invert, is_first;
1329
1330
2.92k
    if (lre_check_stack_overflow(s->opaque, 0))
1331
0
        return re_parse_error(s, "stack overflow");
1332
1333
2.92k
    re_string_list_init(s, cr);
1334
2.92k
    p = *pp;
1335
2.92k
    p++;    /* skip '[' */
1336
1337
2.92k
    invert = FALSE;
1338
2.92k
    if (*p == '^') {
1339
367
        p++;
1340
367
        invert = TRUE;
1341
367
    }
1342
    
1343
    /* handle unions */
1344
2.92k
    is_first = TRUE;
1345
32.3k
    for(;;) {
1346
32.3k
        if (*p == ']')
1347
2.25k
            break;
1348
30.0k
        if (*p == '[' && s->unicode_sets) {
1349
0
            if (re_parse_nested_class(s, cr1, &p))
1350
0
                goto fail;
1351
0
            goto class_union;
1352
30.0k
        } else {
1353
30.0k
            c1 = get_class_atom(s, cr1, &p, TRUE);
1354
30.0k
            if ((int)c1 < 0)
1355
636
                goto fail;
1356
29.4k
            if (*p == '-' && p[1] != ']') {
1357
6.68k
                const uint8_t *p0 = p + 1;
1358
6.68k
                if (p[1] == '-' && s->unicode_sets && is_first)
1359
0
                    goto class_atom; /* first character class followed by '--' */
1360
6.68k
                if (c1 >= CLASS_RANGE_BASE) {
1361
195
                    if (s->is_unicode) {
1362
0
                        re_string_list_free(cr1);
1363
0
                        goto invalid_class_range;
1364
0
                    }
1365
                    /* Annex B: match '-' character */
1366
195
                    goto class_atom;
1367
195
                }
1368
6.48k
                c2 = get_class_atom(s, cr1, &p0, TRUE);
1369
6.48k
                if ((int)c2 < 0)
1370
24
                    goto fail;
1371
6.46k
                if (c2 >= CLASS_RANGE_BASE) {
1372
533
                    re_string_list_free(cr1);
1373
533
                    if (s->is_unicode) {
1374
0
                        goto invalid_class_range;
1375
0
                    }
1376
                    /* Annex B: match '-' character */
1377
533
                    goto class_atom;
1378
533
                }
1379
5.92k
                p = p0;
1380
5.92k
                if (c2 < c1) {
1381
11
                invalid_class_range:
1382
11
                    re_parse_error(s, "invalid class range");
1383
11
                    goto fail;
1384
11
                }
1385
5.91k
                if (s->ignore_case) {
1386
5.66k
                    CharRange cr2_s, *cr2 = &cr2_s;
1387
5.66k
                    cr_init(cr2, s->opaque, lre_realloc);
1388
5.66k
                    if (cr_add_interval(cr2, c1, c2 + 1) ||
1389
5.66k
                        cr_regexp_canonicalize(cr2, s->is_unicode) ||
1390
5.66k
                        cr_op1(&cr->cr, cr2->points, cr2->len, CR_OP_UNION)) {
1391
0
                        cr_free(cr2);
1392
0
                        goto memory_error;
1393
0
                    }
1394
5.66k
                    cr_free(cr2);
1395
5.66k
                } else {
1396
256
                    if (cr_union_interval(&cr->cr, c1, c2))
1397
0
                        goto memory_error;
1398
256
                }
1399
5.91k
                is_first = FALSE; /* union operation */
1400
22.7k
            } else {
1401
23.4k
            class_atom:
1402
23.4k
                if (c1 >= CLASS_RANGE_BASE) {
1403
1.38k
                class_union:
1404
1.38k
                    ret = re_string_list_op(cr, cr1, CR_OP_UNION);
1405
1.38k
                    re_string_list_free(cr1);
1406
1.38k
                    if (ret)
1407
0
                        goto memory_error;
1408
22.0k
                } else {
1409
22.0k
                    if (s->ignore_case)
1410
10.7k
                        c1 = lre_canonicalize(c1, s->is_unicode);
1411
22.0k
                    if (cr_union_interval(&cr->cr, c1, c1))
1412
0
                        goto memory_error;
1413
22.0k
                }
1414
23.4k
            }
1415
29.4k
        }
1416
29.3k
        if (s->unicode_sets && is_first) {
1417
0
            if (*p == '&' && p[1] == '&' && p[2] != '&') {
1418
                /* handle '&&' */
1419
0
                for(;;) {
1420
0
                    if (*p == ']') {
1421
0
                        break;
1422
0
                    } else if (*p == '&' && p[1] == '&' && p[2] != '&') {
1423
0
                        p += 2;
1424
0
                    } else {
1425
0
                        goto invalid_operation;
1426
0
                    }
1427
0
                    if (re_parse_class_set_operand(s, cr1, &p))
1428
0
                        goto fail;
1429
0
                    ret = re_string_list_op(cr, cr1, CR_OP_INTER);
1430
0
                    re_string_list_free(cr1);
1431
0
                    if (ret)
1432
0
                        goto memory_error;
1433
0
                }
1434
0
            } else if (*p == '-' && p[1] == '-') {
1435
                /* handle '--' */
1436
0
                for(;;) {
1437
0
                    if (*p == ']') {
1438
0
                        break;
1439
0
                    } else if (*p == '-' && p[1] == '-') {
1440
0
                        p += 2;
1441
0
                    } else {
1442
0
                    invalid_operation:
1443
0
                        re_parse_error(s, "invalid operation in regular expression");
1444
0
                        goto fail;
1445
0
                    }
1446
0
                    if (re_parse_class_set_operand(s, cr1, &p))
1447
0
                        goto fail;
1448
0
                    ret = re_string_list_op(cr, cr1, CR_OP_SUB);
1449
0
                    re_string_list_free(cr1);
1450
0
                    if (ret)
1451
0
                        goto memory_error;
1452
0
                }
1453
0
            }
1454
0
        }
1455
29.3k
        is_first = FALSE;
1456
29.3k
    }
1457
1458
2.25k
    p++;    /* skip ']' */
1459
2.25k
    *pp = p;
1460
2.25k
    if (invert) {
1461
        /* XXX: add may_contain_string syntax check to be fully
1462
           compliant. The test here accepts more input than the
1463
           spec. */
1464
358
        if (cr->n_strings != 0) {
1465
0
            re_parse_error(s, "negated character class with strings in regular expression debugger eval code");
1466
0
            goto fail;
1467
0
        }
1468
358
        if (cr_invert(&cr->cr))
1469
0
            goto memory_error;
1470
358
    }
1471
2.25k
    return 0;
1472
0
 memory_error:
1473
0
    re_parse_out_of_memory(s);
1474
671
 fail:
1475
671
    re_string_list_free(cr);
1476
671
    return -1;
1477
0
}
1478
1479
static int re_parse_char_class(REParseState *s, const uint8_t **pp)
1480
2.92k
{
1481
2.92k
    REStringList cr_s, *cr = &cr_s;
1482
1483
2.92k
    if (re_parse_nested_class(s, cr, pp))
1484
671
        return -1;
1485
2.25k
    if (re_emit_string_list(s, cr))
1486
0
        goto fail;
1487
2.25k
    re_string_list_free(cr);
1488
2.25k
    return 0;
1489
0
 fail:
1490
0
    re_string_list_free(cr);
1491
0
    return -1;
1492
2.25k
}
1493
1494
/* Return:
1495
   - true if the opcodes may not advance the char pointer
1496
   - false if the opcodes always advance the char pointer
1497
*/
1498
static BOOL re_need_check_advance(const uint8_t *bc_buf, int bc_buf_len)
1499
9.45k
{
1500
9.45k
    int pos, opcode, len;
1501
9.45k
    uint32_t val;
1502
9.45k
    BOOL ret;
1503
1504
9.45k
    ret = TRUE;
1505
9.45k
    pos = 0;
1506
63.8k
    while (pos < bc_buf_len) {
1507
60.2k
        opcode = bc_buf[pos];
1508
60.2k
        len = reopcode_info[opcode].size;
1509
60.2k
        switch(opcode) {
1510
431
        case REOP_range:
1511
746
        case REOP_range_i:
1512
746
            val = get_u16(bc_buf + pos + 1);
1513
746
            len += val * 4;
1514
746
            goto simple_char;
1515
265
        case REOP_range32:
1516
332
        case REOP_range32_i:
1517
332
            val = get_u16(bc_buf + pos + 1);
1518
332
            len += val * 8;
1519
332
            goto simple_char;
1520
3.67k
        case REOP_char:
1521
8.29k
        case REOP_char_i:
1522
8.95k
        case REOP_char32:
1523
8.95k
        case REOP_char32_i:
1524
9.75k
        case REOP_dot:
1525
10.0k
        case REOP_any:
1526
11.0k
        simple_char:
1527
11.0k
            ret = FALSE;
1528
11.0k
            break;
1529
828
        case REOP_line_start:
1530
1.33k
        case REOP_line_start_m:
1531
1.85k
        case REOP_line_end:
1532
2.19k
        case REOP_line_end_m:
1533
9.13k
        case REOP_push_i32:
1534
9.13k
        case REOP_push_char_pos:
1535
9.13k
        case REOP_drop:
1536
9.53k
        case REOP_word_boundary:
1537
9.76k
        case REOP_word_boundary_i:
1538
9.99k
        case REOP_not_word_boundary:
1539
10.3k
        case REOP_not_word_boundary_i:
1540
12.0k
        case REOP_prev:
1541
            /* no effect */
1542
12.0k
            break;
1543
25.2k
        case REOP_save_start:
1544
27.5k
        case REOP_save_end:
1545
30.2k
        case REOP_save_reset:
1546
30.4k
        case REOP_back_reference:
1547
30.7k
        case REOP_back_reference_i:
1548
30.9k
        case REOP_backward_back_reference:
1549
31.2k
        case REOP_backward_back_reference_i:
1550
31.2k
            break;
1551
5.83k
        default:
1552
            /* safe behavior: we cannot predict the outcome */
1553
5.83k
            return TRUE;
1554
60.2k
        }
1555
54.3k
        pos += len;
1556
54.3k
    }
1557
3.62k
    return ret;
1558
9.45k
}
1559
1560
/* return -1 if a simple quantifier cannot be used. Otherwise return
1561
   the number of characters in the atom. */
1562
static int re_is_simple_quantifier(const uint8_t *bc_buf, int bc_buf_len)
1563
10.4k
{
1564
10.4k
    int pos, opcode, len, count;
1565
10.4k
    uint32_t val;
1566
1567
10.4k
    count = 0;
1568
10.4k
    pos = 0;
1569
92.0k
    while (pos < bc_buf_len) {
1570
88.8k
        opcode = bc_buf[pos];
1571
88.8k
        len = reopcode_info[opcode].size;
1572
88.8k
        switch(opcode) {
1573
459
        case REOP_range:
1574
728
        case REOP_range_i:
1575
728
            val = get_u16(bc_buf + pos + 1);
1576
728
            len += val * 4;
1577
728
            goto simple_char;
1578
218
        case REOP_range32:
1579
261
        case REOP_range32_i:
1580
261
            val = get_u16(bc_buf + pos + 1);
1581
261
            len += val * 8;
1582
261
            goto simple_char;
1583
73.2k
        case REOP_char:
1584
76.3k
        case REOP_char_i:
1585
76.8k
        case REOP_char32:
1586
76.8k
        case REOP_char32_i:
1587
77.9k
        case REOP_dot:
1588
78.1k
        case REOP_any:
1589
79.1k
        simple_char:
1590
79.1k
            count++;
1591
79.1k
            break;
1592
561
        case REOP_line_start:
1593
926
        case REOP_line_start_m:
1594
1.24k
        case REOP_line_end:
1595
1.57k
        case REOP_line_end_m:
1596
1.77k
        case REOP_word_boundary:
1597
1.98k
        case REOP_word_boundary_i:
1598
2.19k
        case REOP_not_word_boundary:
1599
2.36k
        case REOP_not_word_boundary_i:
1600
2.36k
            break;
1601
7.28k
        default:
1602
7.28k
            return -1;
1603
88.8k
        }
1604
81.5k
        pos += len;
1605
81.5k
    }
1606
3.19k
    return count;
1607
10.4k
}
1608
1609
/* '*pp' is the first char after '<' */
1610
static int re_parse_group_name(char *buf, int buf_size, const uint8_t **pp)
1611
21.6k
{
1612
21.6k
    const uint8_t *p, *p1;
1613
21.6k
    uint32_t c, d;
1614
21.6k
    char *q;
1615
1616
21.6k
    p = *pp;
1617
21.6k
    q = buf;
1618
65.1k
    for(;;) {
1619
65.1k
        c = *p;
1620
65.1k
        if (c == '\\') {
1621
12.7k
            p++;
1622
12.7k
            if (*p != 'u')
1623
497
                return -1;
1624
12.2k
            c = lre_parse_escape(&p, 2); // accept surrogate pairs
1625
52.3k
        } else if (c == '>') {
1626
8.22k
            break;
1627
44.1k
        } else if (c >= 128) {
1628
3.86k
            c = unicode_from_utf8(p, UTF8_CHAR_LEN_MAX, &p);
1629
3.86k
            if (is_hi_surrogate(c)) {
1630
463
                d = unicode_from_utf8(p, UTF8_CHAR_LEN_MAX, &p1);
1631
463
                if (is_lo_surrogate(d)) {
1632
198
                    c = from_surrogate(c, d);
1633
198
                    p = p1;
1634
198
                }
1635
463
            }
1636
40.2k
        } else {
1637
40.2k
            p++;
1638
40.2k
        }
1639
56.3k
        if (c > 0x10FFFF)
1640
5.75k
            return -1;
1641
50.6k
        if (q == buf) {
1642
15.4k
            if (!lre_js_is_ident_first(c))
1643
5.62k
                return -1;
1644
35.1k
        } else {
1645
35.1k
            if (!lre_js_is_ident_next(c))
1646
1.32k
                return -1;
1647
35.1k
        }
1648
43.6k
        if ((q - buf + UTF8_CHAR_LEN_MAX + 1) > buf_size)
1649
195
            return -1;
1650
43.4k
        if (c < 128) {
1651
38.8k
            *q++ = c;
1652
38.8k
        } else {
1653
4.60k
            q += unicode_to_utf8((uint8_t*)q, c);
1654
4.60k
        }
1655
43.4k
    }
1656
8.22k
    if (q == buf)
1657
212
        return -1;
1658
8.01k
    *q = '\0';
1659
8.01k
    p++;
1660
8.01k
    *pp = p;
1661
8.01k
    return 0;
1662
8.22k
}
1663
1664
/* if capture_name = NULL: return the number of captures + 1.
1665
   Otherwise, return the capture index corresponding to capture_name
1666
   or -1 if none */
1667
static int re_parse_captures(REParseState *s, int *phas_named_captures,
1668
                             const char *capture_name)
1669
5.08k
{
1670
5.08k
    const uint8_t *p;
1671
5.08k
    int capture_index;
1672
5.08k
    char name[TMP_BUF_SIZE];
1673
1674
5.08k
    capture_index = 1;
1675
5.08k
    *phas_named_captures = 0;
1676
973k
    for (p = s->buf_start; p < s->buf_end; p++) {
1677
971k
        switch (*p) {
1678
115k
        case '(':
1679
115k
            if (p[1] == '?') {
1680
50.7k
                if (p[2] == '<' && p[3] != '=' && p[3] != '!') {
1681
16.3k
                    *phas_named_captures = 1;
1682
                    /* potential named capture */
1683
16.3k
                    if (capture_name) {
1684
15.5k
                        p += 3;
1685
15.5k
                        if (re_parse_group_name(name, sizeof(name), &p) == 0) {
1686
3.44k
                            if (!strcmp(name, capture_name))
1687
2.74k
                                return capture_index;
1688
3.44k
                        }
1689
15.5k
                    }
1690
13.6k
                    capture_index++;
1691
13.6k
                    if (capture_index >= CAPTURE_COUNT_MAX)
1692
4
                        goto done;
1693
13.6k
                }
1694
65.0k
            } else {
1695
65.0k
                capture_index++;
1696
65.0k
                if (capture_index >= CAPTURE_COUNT_MAX)
1697
208
                    goto done;
1698
65.0k
            }
1699
112k
            break;
1700
150k
        case '\\':
1701
150k
            p++;
1702
150k
            break;
1703
1.32k
        case '[':
1704
10.6k
            for (p += 1 + (*p == ']'); p < s->buf_end && *p != ']'; p++) {
1705
9.27k
                if (*p == '\\')
1706
3.47k
                    p++;
1707
9.27k
            }
1708
1.32k
            break;
1709
971k
        }
1710
971k
    }
1711
2.33k
 done:
1712
2.33k
    if (capture_name)
1713
1.25k
        return -1;
1714
1.08k
    else
1715
1.08k
        return capture_index;
1716
2.33k
}
1717
1718
static int re_count_captures(REParseState *s)
1719
4.85k
{
1720
4.85k
    if (s->total_capture_count < 0) {
1721
1.08k
        s->total_capture_count = re_parse_captures(s, &s->has_named_captures,
1722
1.08k
                                                   NULL);
1723
1.08k
    }
1724
4.85k
    return s->total_capture_count;
1725
4.85k
}
1726
1727
static BOOL re_has_named_captures(REParseState *s)
1728
2.75k
{
1729
2.75k
    if (s->has_named_captures < 0)
1730
711
        re_count_captures(s);
1731
2.75k
    return s->has_named_captures;
1732
2.75k
}
1733
1734
static int find_group_name(REParseState *s, const char *name)
1735
4.57k
{
1736
4.57k
    const char *p, *buf_end;
1737
4.57k
    size_t len, name_len;
1738
4.57k
    int capture_index;
1739
1740
4.57k
    p = (char *)s->group_names.buf;
1741
4.57k
    if (!p) return -1;
1742
926
    buf_end = (char *)s->group_names.buf + s->group_names.size;
1743
926
    name_len = strlen(name);
1744
926
    capture_index = 1;
1745
9.27k
    while (p < buf_end) {
1746
8.55k
        len = strlen(p);
1747
8.55k
        if (len == name_len && memcmp(name, p, name_len) == 0)
1748
205
            return capture_index;
1749
8.34k
        p += len + 1;
1750
8.34k
        capture_index++;
1751
8.34k
    }
1752
721
    return -1;
1753
926
}
1754
1755
static int re_parse_disjunction(REParseState *s, BOOL is_backward_dir);
1756
1757
static int re_parse_modifiers(REParseState *s, const uint8_t **pp)
1758
3.54k
{
1759
3.54k
    const uint8_t *p = *pp;
1760
3.54k
    int mask = 0;
1761
3.54k
    int val;
1762
1763
6.18k
    for(;;) {
1764
6.18k
        if (*p == 'i') {
1765
1.49k
            val = LRE_FLAG_IGNORECASE;
1766
4.69k
        } else if (*p == 'm') {
1767
581
            val = LRE_FLAG_MULTILINE;
1768
4.10k
        } else if (*p == 's') {
1769
575
            val = LRE_FLAG_DOTALL;
1770
3.53k
        } else {
1771
3.53k
            break;
1772
3.53k
        }
1773
2.65k
        if (mask & val)
1774
7
            return re_parse_error(s, "duplicate modifier: '%c'", *p);
1775
2.64k
        mask |= val;
1776
2.64k
        p++;
1777
2.64k
    }
1778
3.53k
    *pp = p;
1779
3.53k
    return mask;
1780
3.54k
}
1781
1782
static BOOL update_modifier(BOOL val, int add_mask, int remove_mask,
1783
                            int mask)
1784
7.38k
{
1785
7.38k
    if (add_mask & mask)
1786
1.72k
        val = TRUE;
1787
7.38k
    if (remove_mask & mask)
1788
870
        val = FALSE;
1789
7.38k
    return val;
1790
7.38k
}
1791
1792
static int re_parse_term(REParseState *s, BOOL is_backward_dir)
1793
410k
{
1794
410k
    const uint8_t *p;
1795
410k
    int c, last_atom_start, quant_min, quant_max, last_capture_count;
1796
410k
    BOOL greedy, add_zero_advance_check, is_neg, is_backward_lookahead;
1797
410k
    REStringList cr_s, *cr = &cr_s;
1798
1799
410k
    last_atom_start = -1;
1800
410k
    last_capture_count = 0;
1801
410k
    p = s->buf_ptr;
1802
410k
    c = *p;
1803
410k
    switch(c) {
1804
1.46k
    case '^':
1805
1.46k
        p++;
1806
1.46k
        re_emit_op(s, s->multi_line ? REOP_line_start_m : REOP_line_start);
1807
1.46k
        break;
1808
1.18k
    case '$':
1809
1.18k
        p++;
1810
1.18k
        re_emit_op(s, s->multi_line ? REOP_line_end_m : REOP_line_end);
1811
1.18k
        break;
1812
2.74k
    case '.':
1813
2.74k
        p++;
1814
2.74k
        last_atom_start = s->byte_code.size;
1815
2.74k
        last_capture_count = s->capture_count;
1816
2.74k
        if (is_backward_dir)
1817
330
            re_emit_op(s, REOP_prev);
1818
2.74k
        re_emit_op(s, s->dotall ? REOP_any : REOP_dot);
1819
2.74k
        if (is_backward_dir)
1820
330
            re_emit_op(s, REOP_prev);
1821
2.74k
        break;
1822
10.0k
    case '{':
1823
10.0k
        if (s->is_unicode) {
1824
0
            return re_parse_error(s, "syntax error");
1825
10.0k
        } else if (!is_digit(p[1])) {
1826
            /* Annex B: we accept '{' not followed by digits as a
1827
               normal atom */
1828
3.87k
            goto parse_class_atom;
1829
6.21k
        } else {
1830
6.21k
            const uint8_t *p1 = p + 1;
1831
            /* Annex B: error if it is like a repetition count */
1832
6.21k
            parse_digits(&p1, TRUE);
1833
6.21k
            if (*p1 == ',') {
1834
5.29k
                p1++;
1835
5.29k
                if (is_digit(*p1)) {
1836
608
                    parse_digits(&p1, TRUE);
1837
608
                }
1838
5.29k
            }
1839
6.21k
            if (*p1 != '}') {
1840
6.20k
                goto parse_class_atom;
1841
6.20k
            }
1842
6.21k
        }
1843
        /* fall thru */
1844
7
    case '*':
1845
17
    case '+':
1846
24
    case '?':
1847
24
        return re_parse_error(s, "nothing to repeat");
1848
228k
    case '(':
1849
228k
        if (p[1] == '?') {
1850
195k
            if (p[2] == ':') {
1851
180k
                p += 3;
1852
180k
                last_atom_start = s->byte_code.size;
1853
180k
                last_capture_count = s->capture_count;
1854
180k
                s->buf_ptr = p;
1855
180k
                if (re_parse_disjunction(s, is_backward_dir))
1856
179k
                    return -1;
1857
811
                p = s->buf_ptr;
1858
811
                if (re_parse_expect(s, &p, ')'))
1859
15
                    return -1;
1860
14.7k
            } else if (p[2] == 'i' || p[2] == 'm' || p[2] == 's' || p[2] == '-') {
1861
2.50k
                BOOL saved_ignore_case, saved_multi_line, saved_dotall;
1862
2.50k
                int add_mask, remove_mask;
1863
2.50k
                p += 2;
1864
2.50k
                remove_mask = 0;
1865
2.50k
                add_mask = re_parse_modifiers(s, &p);
1866
2.50k
                if (add_mask < 0)
1867
3
                    return -1;
1868
2.49k
                if (*p == '-') {
1869
1.03k
                    p++;
1870
1.03k
                    remove_mask = re_parse_modifiers(s, &p);
1871
1.03k
                    if (remove_mask < 0)
1872
4
                        return -1;
1873
1.03k
                }
1874
2.49k
                if ((add_mask == 0 && remove_mask == 0) ||
1875
2.49k
                    (add_mask & remove_mask) != 0) {
1876
6
                    return re_parse_error(s, "invalid modifiers");
1877
6
                }
1878
2.48k
                if (re_parse_expect(s, &p, ':'))
1879
28
                    return -1;
1880
2.46k
                saved_ignore_case = s->ignore_case;
1881
2.46k
                saved_multi_line = s->multi_line;
1882
2.46k
                saved_dotall = s->dotall;
1883
2.46k
                s->ignore_case = update_modifier(s->ignore_case, add_mask, remove_mask, LRE_FLAG_IGNORECASE);
1884
2.46k
                s->multi_line = update_modifier(s->multi_line, add_mask, remove_mask, LRE_FLAG_MULTILINE);
1885
2.46k
                s->dotall = update_modifier(s->dotall, add_mask, remove_mask, LRE_FLAG_DOTALL);
1886
1887
2.46k
                last_atom_start = s->byte_code.size;
1888
2.46k
                last_capture_count = s->capture_count;
1889
2.46k
                s->buf_ptr = p;
1890
2.46k
                if (re_parse_disjunction(s, is_backward_dir))
1891
1.66k
                    return -1;
1892
797
                p = s->buf_ptr;
1893
797
                if (re_parse_expect(s, &p, ')'))
1894
116
                    return -1;
1895
681
                s->ignore_case = saved_ignore_case;
1896
681
                s->multi_line = saved_multi_line;
1897
681
                s->dotall = saved_dotall;
1898
12.2k
            } else if ((p[2] == '=' || p[2] == '!')) {
1899
9.97k
                is_neg = (p[2] == '!');
1900
9.97k
                is_backward_lookahead = FALSE;
1901
9.97k
                p += 3;
1902
9.97k
                goto lookahead;
1903
9.97k
            } else if (p[2] == '<' &&
1904
2.31k
                       (p[3] == '=' || p[3] == '!')) {
1905
1.31k
                int pos;
1906
1.31k
                is_neg = (p[3] == '!');
1907
1.31k
                is_backward_lookahead = TRUE;
1908
1.31k
                p += 4;
1909
                /* lookahead */
1910
11.2k
            lookahead:
1911
                /* Annex B allows lookahead to be used as an atom for
1912
                   the quantifiers */
1913
11.2k
                if (!s->is_unicode && !is_backward_lookahead)  {
1914
9.97k
                    last_atom_start = s->byte_code.size;
1915
9.97k
                    last_capture_count = s->capture_count;
1916
9.97k
                }
1917
11.2k
                pos = re_emit_op_u32(s, REOP_lookahead + is_neg, 0);
1918
11.2k
                s->buf_ptr = p;
1919
11.2k
                if (re_parse_disjunction(s, is_backward_lookahead))
1920
10.3k
                    return -1;
1921
919
                p = s->buf_ptr;
1922
919
                if (re_parse_expect(s, &p, ')'))
1923
134
                    return -1;
1924
785
                re_emit_op(s, REOP_match);
1925
                /* jump after the 'match' after the lookahead is successful */
1926
785
                if (dbuf_error(&s->byte_code))
1927
0
                    return -1;
1928
785
                put_u32(s->byte_code.buf + pos, s->byte_code.size - (pos + 4));
1929
997
            } else if (p[2] == '<') {
1930
913
                p += 3;
1931
913
                if (re_parse_group_name(s->u.tmp_buf, sizeof(s->u.tmp_buf),
1932
913
                                        &p)) {
1933
538
                    return re_parse_error(s, "invalid group name");
1934
538
                }
1935
375
                if (find_group_name(s, s->u.tmp_buf) > 0) {
1936
9
                    return re_parse_error(s, "duplicate group name");
1937
9
                }
1938
                /* group name with a trailing zero */
1939
366
                dbuf_put(&s->group_names, (uint8_t *)s->u.tmp_buf,
1940
366
                         strlen(s->u.tmp_buf) + 1);
1941
366
                s->has_named_captures = 1;
1942
366
                goto parse_capture;
1943
375
            } else {
1944
84
                return re_parse_error(s, "invalid group");
1945
84
            }
1946
195k
        } else {
1947
32.8k
            int capture_index;
1948
32.8k
            p++;
1949
            /* capture without group name */
1950
32.8k
            dbuf_putc(&s->group_names, 0);
1951
33.1k
        parse_capture:
1952
33.1k
            if (s->capture_count >= CAPTURE_COUNT_MAX)
1953
20
                return re_parse_error(s, "too many captures");
1954
33.1k
            last_atom_start = s->byte_code.size;
1955
33.1k
            last_capture_count = s->capture_count;
1956
33.1k
            capture_index = s->capture_count++;
1957
33.1k
            re_emit_op_u8(s, REOP_save_start + is_backward_dir,
1958
33.1k
                          capture_index);
1959
1960
33.1k
            s->buf_ptr = p;
1961
33.1k
            if (re_parse_disjunction(s, is_backward_dir))
1962
23.7k
                return -1;
1963
9.40k
            p = s->buf_ptr;
1964
1965
9.40k
            re_emit_op_u8(s, REOP_save_start + 1 - is_backward_dir,
1966
9.40k
                          capture_index);
1967
1968
9.40k
            if (re_parse_expect(s, &p, ')'))
1969
289
                return -1;
1970
9.40k
        }
1971
11.3k
        break;
1972
24.9k
    case '\\':
1973
24.9k
        switch(p[1]) {
1974
496
        case 'b':
1975
974
        case 'B':
1976
974
            if (p[1] != 'b') {
1977
478
                re_emit_op(s, s->ignore_case ? REOP_not_word_boundary_i : REOP_not_word_boundary);
1978
496
            } else {
1979
496
                re_emit_op(s, s->ignore_case ? REOP_word_boundary_i : REOP_word_boundary);
1980
496
            }
1981
974
            p += 2;
1982
974
            break;
1983
5.70k
        case 'k':
1984
5.70k
            {
1985
5.70k
                const uint8_t *p1;
1986
5.70k
                int dummy_res;
1987
1988
5.70k
                p1 = p;
1989
5.70k
                if (p1[2] != '<') {
1990
                    /* annex B: we tolerate invalid group names in non
1991
                       unicode mode if there is no named capture
1992
                       definition */
1993
584
                    if (s->is_unicode || re_has_named_captures(s))
1994
23
                        return re_parse_error(s, "expecting group name");
1995
561
                    else
1996
561
                        goto parse_class_atom;
1997
584
                }
1998
5.11k
                p1 += 3;
1999
5.11k
                if (re_parse_group_name(s->u.tmp_buf, sizeof(s->u.tmp_buf),
2000
5.11k
                                        &p1)) {
2001
920
                    if (s->is_unicode || re_has_named_captures(s))
2002
7
                        return re_parse_error(s, "invalid group name");
2003
913
                    else
2004
913
                        goto parse_class_atom;
2005
920
                }
2006
4.19k
                c = find_group_name(s, s->u.tmp_buf);
2007
4.19k
                if (c < 0) {
2008
                    /* no capture name parsed before, try to look
2009
                       after (inefficient, but hopefully not common */
2010
4.00k
                    c = re_parse_captures(s, &dummy_res, s->u.tmp_buf);
2011
4.00k
                    if (c < 0) {
2012
1.25k
                        if (s->is_unicode || re_has_named_captures(s))
2013
242
                            return re_parse_error(s, "group name not defined");
2014
1.01k
                        else
2015
1.01k
                            goto parse_class_atom;
2016
1.25k
                    }
2017
4.00k
                }
2018
2.94k
                p = p1;
2019
2.94k
            }
2020
0
            goto emit_back_reference;
2021
1.50k
        case '0':
2022
1.50k
            p += 2;
2023
1.50k
            c = 0;
2024
1.50k
            if (s->is_unicode) {
2025
0
                if (is_digit(*p)) {
2026
0
                    return re_parse_error(s, "invalid decimal escape in regular expression");
2027
0
                }
2028
1.50k
            } else {
2029
                /* Annex B.1.4: accept legacy octal */
2030
1.50k
                if (*p >= '0' && *p <= '7') {
2031
830
                    c = *p++ - '0';
2032
830
                    if (*p >= '0' && *p <= '7') {
2033
352
                        c = (c << 3) + *p++ - '0';
2034
352
                    }
2035
830
                }
2036
1.50k
            }
2037
1.50k
            goto normal_char;
2038
3.14k
        case '1': case '2': case '3': case '4':
2039
4.78k
        case '5': case '6': case '7': case '8':
2040
5.01k
        case '9':
2041
5.01k
            {
2042
5.01k
                const uint8_t *q = ++p;
2043
2044
5.01k
                c = parse_digits(&p, FALSE);
2045
5.01k
                if (c < 0 || (c >= s->capture_count && c >= re_count_captures(s))) {
2046
3.72k
                    if (!s->is_unicode) {
2047
                        /* Annex B.1.4: accept legacy octal */
2048
3.72k
                        p = q;
2049
3.72k
                        if (*p <= '7') {
2050
3.16k
                            c = 0;
2051
3.16k
                            if (*p <= '3')
2052
1.42k
                                c = *p++ - '0';
2053
3.16k
                            if (*p >= '0' && *p <= '7') {
2054
2.08k
                                c = (c << 3) + *p++ - '0';
2055
2.08k
                                if (*p >= '0' && *p <= '7') {
2056
692
                                    c = (c << 3) + *p++ - '0';
2057
692
                                }
2058
2.08k
                            }
2059
3.16k
                        } else {
2060
564
                            c = *p++;
2061
564
                        }
2062
3.72k
                        goto normal_char;
2063
3.72k
                    }
2064
0
                    return re_parse_error(s, "back reference out of range in regular expression");
2065
3.72k
                }
2066
4.22k
            emit_back_reference:
2067
4.22k
                last_atom_start = s->byte_code.size;
2068
4.22k
                last_capture_count = s->capture_count;
2069
                
2070
4.22k
                re_emit_op_u8(s, REOP_back_reference + 2 * is_backward_dir + s->ignore_case, c);
2071
4.22k
            }
2072
0
            break;
2073
11.7k
        default:
2074
11.7k
            goto parse_class_atom;
2075
24.9k
        }
2076
5.20k
        break;
2077
5.20k
    case '[':
2078
2.92k
        last_atom_start = s->byte_code.size;
2079
2.92k
        last_capture_count = s->capture_count;
2080
2.92k
        if (is_backward_dir)
2081
258
            re_emit_op(s, REOP_prev);
2082
2.92k
        if (re_parse_char_class(s, &p))
2083
671
            return -1;
2084
2.25k
        if (is_backward_dir)
2085
252
            re_emit_op(s, REOP_prev);
2086
2.25k
        break;
2087
446
    case ']':
2088
1.02k
    case '}':
2089
1.02k
        if (s->is_unicode)
2090
0
            return re_parse_error(s, "syntax error");
2091
1.02k
        goto parse_class_atom;
2092
138k
    default:
2093
163k
    parse_class_atom:
2094
163k
        c = get_class_atom(s, cr, &p, FALSE);
2095
163k
        if ((int)c < 0)
2096
390
            return -1;
2097
168k
    normal_char:
2098
168k
        last_atom_start = s->byte_code.size;
2099
168k
        last_capture_count = s->capture_count;
2100
168k
        if (is_backward_dir)
2101
1.69k
            re_emit_op(s, REOP_prev);
2102
168k
        if (c >= CLASS_RANGE_BASE) {
2103
2.21k
            int ret;
2104
2.21k
            ret = re_emit_string_list(s, cr);
2105
2.21k
            re_string_list_free(cr);
2106
2.21k
            if (ret)
2107
0
                return -1;
2108
166k
        } else {
2109
166k
            if (s->ignore_case)
2110
7.37k
                c = lre_canonicalize(c, s->is_unicode);
2111
166k
            re_emit_char(s, c);
2112
166k
        }
2113
168k
        if (is_backward_dir)
2114
1.69k
            re_emit_op(s, REOP_prev);
2115
168k
        break;
2116
410k
    }
2117
2118
    /* quantifier */
2119
192k
    if (last_atom_start >= 0) {
2120
188k
        c = *p;
2121
188k
        switch(c) {
2122
2.53k
        case '*':
2123
2.53k
            p++;
2124
2.53k
            quant_min = 0;
2125
2.53k
            quant_max = INT32_MAX;
2126
2.53k
            goto quantifier;
2127
4.00k
        case '+':
2128
4.00k
            p++;
2129
4.00k
            quant_min = 1;
2130
4.00k
            quant_max = INT32_MAX;
2131
4.00k
            goto quantifier;
2132
3.03k
        case '?':
2133
3.03k
            p++;
2134
3.03k
            quant_min = 0;
2135
3.03k
            quant_max = 1;
2136
3.03k
            goto quantifier;
2137
12.2k
        case '{':
2138
12.2k
            {
2139
12.2k
                const uint8_t *p1 = p;
2140
                /* As an extension (see ES6 annex B), we accept '{' not
2141
                   followed by digits as a normal atom */
2142
12.2k
                if (!is_digit(p[1])) {
2143
3.72k
                    if (s->is_unicode)
2144
0
                        goto invalid_quant_count;
2145
3.72k
                    break;
2146
3.72k
                }
2147
8.56k
                p++;
2148
8.56k
                quant_min = parse_digits(&p, TRUE);
2149
8.56k
                quant_max = quant_min;
2150
8.56k
                if (*p == ',') {
2151
6.70k
                    p++;
2152
6.70k
                    if (is_digit(*p)) {
2153
1.84k
                        quant_max = parse_digits(&p, TRUE);
2154
1.84k
                        if (quant_max < quant_min) {
2155
3
                        invalid_quant_count:
2156
3
                            return re_parse_error(s, "invalid repetition count");
2157
3
                        }
2158
4.86k
                    } else {
2159
4.86k
                        quant_max = INT32_MAX; /* infinity */
2160
4.86k
                    }
2161
6.70k
                }
2162
8.56k
                if (*p != '}' && !s->is_unicode) {
2163
                    /* Annex B: normal atom if invalid '{' syntax */
2164
6.02k
                    p = p1;
2165
6.02k
                    break;
2166
6.02k
                }
2167
2.54k
                if (re_parse_expect(s, &p, '}'))
2168
0
                    return -1;
2169
2.54k
            }
2170
12.1k
        quantifier:
2171
12.1k
            greedy = TRUE;
2172
12.1k
            if (*p == '?') {
2173
1.39k
                p++;
2174
1.39k
                greedy = FALSE;
2175
1.39k
            }
2176
12.1k
            if (last_atom_start < 0) {
2177
0
                return re_parse_error(s, "nothing to repeat");
2178
0
            }
2179
12.1k
            if (greedy) {
2180
10.7k
                int len, pos;
2181
2182
10.7k
                if (quant_max > 0) {
2183
                    /* specific optimization for simple quantifiers */
2184
10.4k
                    if (dbuf_error(&s->byte_code))
2185
0
                        goto out_of_memory;
2186
10.4k
                    len = re_is_simple_quantifier(s->byte_code.buf + last_atom_start,
2187
10.4k
                                                 s->byte_code.size - last_atom_start);
2188
10.4k
                    if (len > 0) {
2189
2.65k
                        re_emit_op(s, REOP_match);
2190
2191
2.65k
                        if (dbuf_insert(&s->byte_code, last_atom_start, 17))
2192
0
                            goto out_of_memory;
2193
2.65k
                        pos = last_atom_start;
2194
2.65k
                        s->byte_code.buf[pos++] = REOP_simple_greedy_quant;
2195
2.65k
                        put_u32(&s->byte_code.buf[pos],
2196
2.65k
                                s->byte_code.size - last_atom_start - 17);
2197
2.65k
                        pos += 4;
2198
2.65k
                        put_u32(&s->byte_code.buf[pos], quant_min);
2199
2.65k
                        pos += 4;
2200
2.65k
                        put_u32(&s->byte_code.buf[pos], quant_max);
2201
2.65k
                        pos += 4;
2202
2.65k
                        put_u32(&s->byte_code.buf[pos], len);
2203
2.65k
                        pos += 4;
2204
2.65k
                        goto done;
2205
2.65k
                    }
2206
10.4k
                }
2207
2208
8.06k
                if (dbuf_error(&s->byte_code))
2209
0
                    goto out_of_memory;
2210
8.06k
            }
2211
            /* the spec tells that if there is no advance when
2212
               running the atom after the first quant_min times,
2213
               then there is no match. We remove this test when we
2214
               are sure the atom always advances the position. */
2215
9.45k
            add_zero_advance_check = re_need_check_advance(s->byte_code.buf + last_atom_start,
2216
9.45k
                                                           s->byte_code.size - last_atom_start);
2217
2218
9.45k
            {
2219
9.45k
                int len, pos;
2220
9.45k
                len = s->byte_code.size - last_atom_start;
2221
9.45k
                if (quant_min == 0) {
2222
                    /* need to reset the capture in case the atom is
2223
                       not executed */
2224
4.50k
                    if (last_capture_count != s->capture_count) {
2225
2.87k
                        if (dbuf_insert(&s->byte_code, last_atom_start, 3))
2226
0
                            goto out_of_memory;
2227
2.87k
                        s->byte_code.buf[last_atom_start++] = REOP_save_reset;
2228
2.87k
                        s->byte_code.buf[last_atom_start++] = last_capture_count;
2229
2.87k
                        s->byte_code.buf[last_atom_start++] = s->capture_count - 1;
2230
2.87k
                    }
2231
4.50k
                    if (quant_max == 0) {
2232
242
                        s->byte_code.size = last_atom_start;
2233
4.26k
                    } else if (quant_max == 1 || quant_max == INT32_MAX) {
2234
3.62k
                        BOOL has_goto = (quant_max == INT32_MAX);
2235
3.62k
                        if (dbuf_insert(&s->byte_code, last_atom_start, 5 + add_zero_advance_check))
2236
0
                            goto out_of_memory;
2237
3.62k
                        s->byte_code.buf[last_atom_start] = REOP_split_goto_first +
2238
3.62k
                            greedy;
2239
3.62k
                        put_u32(s->byte_code.buf + last_atom_start + 1,
2240
3.62k
                                len + 5 * has_goto + add_zero_advance_check * 2);
2241
3.62k
                        if (add_zero_advance_check) {
2242
3.11k
                            s->byte_code.buf[last_atom_start + 1 + 4] = REOP_push_char_pos;
2243
3.11k
                            re_emit_op(s, REOP_check_advance);
2244
3.11k
                        }
2245
3.62k
                        if (has_goto)
2246
1.65k
                            re_emit_goto(s, REOP_goto, last_atom_start);
2247
3.62k
                    } else {
2248
639
                        if (dbuf_insert(&s->byte_code, last_atom_start, 10 + add_zero_advance_check))
2249
0
                            goto out_of_memory;
2250
639
                        pos = last_atom_start;
2251
639
                        s->byte_code.buf[pos++] = REOP_push_i32;
2252
639
                        put_u32(s->byte_code.buf + pos, quant_max);
2253
639
                        pos += 4;
2254
639
                        s->byte_code.buf[pos++] = REOP_split_goto_first + greedy;
2255
639
                        put_u32(s->byte_code.buf + pos, len + 5 + add_zero_advance_check * 2);
2256
639
                        pos += 4;
2257
639
                        if (add_zero_advance_check) {
2258
392
                            s->byte_code.buf[pos++] = REOP_push_char_pos;
2259
392
                            re_emit_op(s, REOP_check_advance);
2260
392
                        }
2261
639
                        re_emit_goto(s, REOP_loop, last_atom_start + 5);
2262
639
                        re_emit_op(s, REOP_drop);
2263
639
                    }
2264
4.95k
                } else if (quant_min == 1 && quant_max == INT32_MAX &&
2265
4.95k
                           !add_zero_advance_check) {
2266
429
                    re_emit_goto(s, REOP_split_next_first - greedy,
2267
429
                                 last_atom_start);
2268
4.52k
                } else {
2269
4.52k
                    if (quant_min == 1) {
2270
                        /* nothing to add */
2271
3.49k
                    } else {
2272
1.03k
                        if (dbuf_insert(&s->byte_code, last_atom_start, 5))
2273
0
                            goto out_of_memory;
2274
1.03k
                        s->byte_code.buf[last_atom_start] = REOP_push_i32;
2275
1.03k
                        put_u32(s->byte_code.buf + last_atom_start + 1,
2276
1.03k
                                quant_min);
2277
1.03k
                        last_atom_start += 5;
2278
1.03k
                        re_emit_goto(s, REOP_loop, last_atom_start);
2279
1.03k
                        re_emit_op(s, REOP_drop);
2280
1.03k
                    }
2281
4.52k
                    if (quant_max == INT32_MAX) {
2282
3.15k
                        pos = s->byte_code.size;
2283
3.15k
                        re_emit_op_u32(s, REOP_split_goto_first + greedy,
2284
3.15k
                                       len + 5 + add_zero_advance_check * 2);
2285
3.15k
                        if (add_zero_advance_check)
2286
2.96k
                            re_emit_op(s, REOP_push_char_pos);
2287
                        /* copy the atom */
2288
3.15k
                        dbuf_put_self(&s->byte_code, last_atom_start, len);
2289
3.15k
                        if (add_zero_advance_check)
2290
2.96k
                            re_emit_op(s, REOP_check_advance);
2291
3.15k
                        re_emit_goto(s, REOP_goto, pos);
2292
3.15k
                    } else if (quant_max > quant_min) {
2293
607
                        re_emit_op_u32(s, REOP_push_i32, quant_max - quant_min);
2294
607
                        pos = s->byte_code.size;
2295
607
                        re_emit_op_u32(s, REOP_split_goto_first + greedy,
2296
607
                                       len + 5 + add_zero_advance_check * 2);
2297
607
                        if (add_zero_advance_check)
2298
247
                            re_emit_op(s, REOP_push_char_pos);
2299
                        /* copy the atom */
2300
607
                        dbuf_put_self(&s->byte_code, last_atom_start, len);
2301
607
                        if (add_zero_advance_check)
2302
247
                            re_emit_op(s, REOP_check_advance);
2303
607
                        re_emit_goto(s, REOP_loop, pos);
2304
607
                        re_emit_op(s, REOP_drop);
2305
607
                    }
2306
4.52k
                }
2307
9.45k
                last_atom_start = -1;
2308
9.45k
            }
2309
0
            break;
2310
166k
        default:
2311
166k
            break;
2312
188k
        }
2313
188k
    }
2314
192k
 done:
2315
192k
    s->buf_ptr = p;
2316
192k
    return 0;
2317
0
 out_of_memory:
2318
0
    return re_parse_out_of_memory(s);
2319
192k
}
2320
2321
static int re_parse_alternative(REParseState *s, BOOL is_backward_dir)
2322
235k
{
2323
235k
    const uint8_t *p;
2324
235k
    int ret;
2325
235k
    size_t start, term_start, end, term_size;
2326
2327
235k
    start = s->byte_code.size;
2328
427k
    for(;;) {
2329
427k
        p = s->buf_ptr;
2330
427k
        if (p >= s->buf_end)
2331
3.85k
            break;
2332
424k
        if (*p == '|' || *p == ')')
2333
13.2k
            break;
2334
410k
        term_start = s->byte_code.size;
2335
410k
        ret = re_parse_term(s, is_backward_dir);
2336
410k
        if (ret)
2337
218k
            return ret;
2338
192k
        if (is_backward_dir) {
2339
            /* reverse the order of the terms (XXX: inefficient, but
2340
               speed is not really critical here) */
2341
2.70k
            end = s->byte_code.size;
2342
2.70k
            term_size = end - term_start;
2343
2.70k
            if (dbuf_realloc(&s->byte_code, end + term_size))
2344
0
                return -1;
2345
2.70k
            memmove(s->byte_code.buf + start + term_size,
2346
2.70k
                    s->byte_code.buf + start,
2347
2.70k
                    end - start);
2348
2.70k
            memcpy(s->byte_code.buf + start, s->byte_code.buf + end,
2349
2.70k
                   term_size);
2350
2.70k
        }
2351
192k
    }
2352
17.1k
    return 0;
2353
235k
}
2354
2355
static int re_parse_disjunction(REParseState *s, BOOL is_backward_dir)
2356
233k
{
2357
233k
    int start, len, pos;
2358
2359
233k
    if (lre_check_stack_overflow(s->opaque, 0))
2360
0
        return re_parse_error(s, "stack overflow");
2361
2362
233k
    start = s->byte_code.size;
2363
233k
    if (re_parse_alternative(s, is_backward_dir))
2364
217k
        return -1;
2365
17.1k
    while (*s->buf_ptr == '|') {
2366
1.86k
        s->buf_ptr++;
2367
2368
1.86k
        len = s->byte_code.size - start;
2369
2370
        /* insert a split before the first alternative */
2371
1.86k
        if (dbuf_insert(&s->byte_code, start, 5)) {
2372
0
            return re_parse_out_of_memory(s);
2373
0
        }
2374
1.86k
        s->byte_code.buf[start] = REOP_split_next_first;
2375
1.86k
        put_u32(s->byte_code.buf + start + 1, len + 5);
2376
2377
1.86k
        pos = re_emit_op_u32(s, REOP_goto, 0);
2378
2379
1.86k
        if (re_parse_alternative(s, is_backward_dir))
2380
400
            return -1;
2381
2382
        /* patch the goto */
2383
1.46k
        len = s->byte_code.size - (pos + 4);
2384
1.46k
        put_u32(s->byte_code.buf + pos, len);
2385
1.46k
    }
2386
15.2k
    return 0;
2387
15.6k
}
2388
2389
/* the control flow is recursive so the analysis can be linear */
2390
static int compute_stack_size(const uint8_t *bc_buf, int bc_buf_len)
2391
3.29k
{
2392
3.29k
    int stack_size, stack_size_max, pos, opcode, len;
2393
3.29k
    uint32_t val;
2394
2395
3.29k
    stack_size = 0;
2396
3.29k
    stack_size_max = 0;
2397
3.29k
    bc_buf += RE_HEADER_LEN;
2398
3.29k
    bc_buf_len -= RE_HEADER_LEN;
2399
3.29k
    pos = 0;
2400
700M
    while (pos < bc_buf_len) {
2401
700M
        opcode = bc_buf[pos];
2402
700M
        len = reopcode_info[opcode].size;
2403
700M
        assert(opcode < REOP_COUNT);
2404
700M
        assert((pos + len) <= bc_buf_len);
2405
700M
        switch(opcode) {
2406
5.36M
        case REOP_push_i32:
2407
58.5M
        case REOP_push_char_pos:
2408
58.5M
            stack_size++;
2409
58.5M
            if (stack_size > stack_size_max) {
2410
5.39k
                if (stack_size > STACK_SIZE_MAX)
2411
1
                    return -1;
2412
5.39k
                stack_size_max = stack_size;
2413
5.39k
            }
2414
58.5M
            break;
2415
58.5M
        case REOP_drop:
2416
58.5M
        case REOP_check_advance:
2417
58.5M
            assert(stack_size > 0);
2418
58.5M
            stack_size--;
2419
58.5M
            break;
2420
57.2M
        case REOP_range:
2421
57.2M
        case REOP_range_i:
2422
57.2M
            val = get_u16(bc_buf + pos + 1);
2423
57.2M
            len += val * 4;
2424
57.2M
            break;
2425
1.67k
        case REOP_range32:
2426
2.03k
        case REOP_range32_i:
2427
2.03k
            val = get_u16(bc_buf + pos + 1);
2428
2.03k
            len += val * 8;
2429
2.03k
            break;
2430
700M
        }
2431
700M
        pos += len;
2432
700M
    }
2433
3.29k
    return stack_size_max;
2434
3.29k
}
2435
2436
/* 'buf' must be a zero terminated UTF-8 string of length buf_len.
2437
   Return NULL if error and allocate an error message in *perror_msg,
2438
   otherwise the compiled bytecode and its length in plen.
2439
*/
2440
uint8_t *lre_compile(int *plen, char *error_msg, int error_msg_size,
2441
                     const char *buf, size_t buf_len, int re_flags,
2442
                     void *opaque)
2443
5.93k
{
2444
5.93k
    REParseState s_s, *s = &s_s;
2445
5.93k
    int stack_size;
2446
5.93k
    BOOL is_sticky;
2447
2448
5.93k
    memset(s, 0, sizeof(*s));
2449
5.93k
    s->opaque = opaque;
2450
5.93k
    s->buf_ptr = (const uint8_t *)buf;
2451
5.93k
    s->buf_end = s->buf_ptr + buf_len;
2452
5.93k
    s->buf_start = s->buf_ptr;
2453
5.93k
    s->re_flags = re_flags;
2454
5.93k
    s->is_unicode = ((re_flags & (LRE_FLAG_UNICODE | LRE_FLAG_UNICODE_SETS)) != 0);
2455
5.93k
    is_sticky = ((re_flags & LRE_FLAG_STICKY) != 0);
2456
5.93k
    s->ignore_case = ((re_flags & LRE_FLAG_IGNORECASE) != 0);
2457
5.93k
    s->multi_line = ((re_flags & LRE_FLAG_MULTILINE) != 0);
2458
5.93k
    s->dotall = ((re_flags & LRE_FLAG_DOTALL) != 0);
2459
5.93k
    s->unicode_sets = ((re_flags & LRE_FLAG_UNICODE_SETS) != 0);
2460
5.93k
    s->capture_count = 1;
2461
5.93k
    s->total_capture_count = -1;
2462
5.93k
    s->has_named_captures = -1;
2463
2464
5.93k
    dbuf_init2(&s->byte_code, opaque, lre_realloc);
2465
5.93k
    dbuf_init2(&s->group_names, opaque, lre_realloc);
2466
2467
5.93k
    dbuf_put_u16(&s->byte_code, re_flags); /* first element is the flags */
2468
5.93k
    dbuf_putc(&s->byte_code, 0); /* second element is the number of captures */
2469
5.93k
    dbuf_putc(&s->byte_code, 0); /* stack size */
2470
5.93k
    dbuf_put_u32(&s->byte_code, 0); /* bytecode length */
2471
2472
5.93k
    if (!is_sticky) {
2473
        /* iterate thru all positions (about the same as .*?( ... ) )
2474
           .  We do it without an explicit loop so that lock step
2475
           thread execution will be possible in an optimized
2476
           implementation */
2477
5.93k
        re_emit_op_u32(s, REOP_split_goto_first, 1 + 5);
2478
5.93k
        re_emit_op(s, REOP_any);
2479
5.93k
        re_emit_op_u32(s, REOP_goto, -(5 + 1 + 5));
2480
5.93k
    }
2481
5.93k
    re_emit_op_u8(s, REOP_save_start, 0);
2482
2483
5.93k
    if (re_parse_disjunction(s, FALSE)) {
2484
2.63k
    error:
2485
2.63k
        dbuf_free(&s->byte_code);
2486
2.63k
        dbuf_free(&s->group_names);
2487
2.63k
        pstrcpy(error_msg, error_msg_size, s->u.error_msg);
2488
2.63k
        *plen = 0;
2489
2.63k
        return NULL;
2490
2.60k
    }
2491
2492
3.32k
    re_emit_op_u8(s, REOP_save_end, 0);
2493
2494
3.32k
    re_emit_op(s, REOP_match);
2495
2496
3.32k
    if (*s->buf_ptr != '\0') {
2497
31
        re_parse_error(s, "extraneous characters at the end");
2498
31
        goto error;
2499
31
    }
2500
2501
3.29k
    if (dbuf_error(&s->byte_code)) {
2502
0
        re_parse_out_of_memory(s);
2503
0
        goto error;
2504
0
    }
2505
2506
3.29k
    stack_size = compute_stack_size(s->byte_code.buf, s->byte_code.size);
2507
3.29k
    if (stack_size < 0) {
2508
1
        re_parse_error(s, "too many imbricated quantifiers");
2509
1
        goto error;
2510
1
    }
2511
2512
3.29k
    s->byte_code.buf[RE_HEADER_CAPTURE_COUNT] = s->capture_count;
2513
3.29k
    s->byte_code.buf[RE_HEADER_STACK_SIZE] = stack_size;
2514
3.29k
    put_u32(s->byte_code.buf + RE_HEADER_BYTECODE_LEN,
2515
3.29k
            s->byte_code.size - RE_HEADER_LEN);
2516
2517
    /* add the named groups if needed */
2518
3.29k
    if (s->group_names.size > (s->capture_count - 1)) {
2519
16
        dbuf_put(&s->byte_code, s->group_names.buf, s->group_names.size);
2520
16
        put_u16(s->byte_code.buf + RE_HEADER_FLAGS,
2521
16
                lre_get_flags(s->byte_code.buf) | LRE_FLAG_NAMED_GROUPS);
2522
16
    }
2523
3.29k
    dbuf_free(&s->group_names);
2524
2525
#ifdef DUMP_REOP
2526
    lre_dump_bytecode(s->byte_code.buf, s->byte_code.size);
2527
#endif
2528
2529
3.29k
    error_msg[0] = '\0';
2530
3.29k
    *plen = s->byte_code.size;
2531
3.29k
    return s->byte_code.buf;
2532
3.29k
}
2533
2534
static BOOL is_line_terminator(uint32_t c)
2535
1.27M
{
2536
1.27M
    return (c == '\n' || c == '\r' || c == CP_LS || c == CP_PS);
2537
1.27M
}
2538
2539
static BOOL is_word_char(uint32_t c)
2540
10.7k
{
2541
10.7k
    return ((c >= '0' && c <= '9') ||
2542
10.7k
            (c >= 'a' && c <= 'z') ||
2543
10.7k
            (c >= 'A' && c <= 'Z') ||
2544
10.7k
            (c == '_'));
2545
10.7k
}
2546
2547
#define GET_CHAR(c, cptr, cbuf_end, cbuf_type)                          \
2548
1.50M
    do {                                                                \
2549
1.50M
        if (cbuf_type == 0) {                                           \
2550
491k
            c = *cptr++;                                                \
2551
1.01M
        } else {                                                        \
2552
1.01M
            const uint16_t *_p = (const uint16_t *)cptr;                \
2553
1.01M
            const uint16_t *_end = (const uint16_t *)cbuf_end;          \
2554
1.01M
            c = *_p++;                                                  \
2555
1.01M
            if (is_hi_surrogate(c) && cbuf_type == 2) {                 \
2556
2
                if (_p < _end && is_lo_surrogate(*_p)) {                \
2557
0
                    c = from_surrogate(c, *_p++);                       \
2558
0
                }                                                       \
2559
2
            }                                                           \
2560
1.01M
            cptr = (const void *)_p;                                    \
2561
1.01M
        }                                                               \
2562
1.50M
    } while (0)
2563
2564
#define PEEK_CHAR(c, cptr, cbuf_end, cbuf_type)                         \
2565
16.7k
    do {                                                                \
2566
16.7k
        if (cbuf_type == 0) {                                           \
2567
16.7k
            c = cptr[0];                                                \
2568
16.7k
        } else {                                                        \
2569
0
            const uint16_t *_p = (const uint16_t *)cptr;                \
2570
0
            const uint16_t *_end = (const uint16_t *)cbuf_end;          \
2571
0
            c = *_p++;                                                  \
2572
0
            if (is_hi_surrogate(c) && cbuf_type == 2) {                 \
2573
0
                if (_p < _end && is_lo_surrogate(*_p)) {                \
2574
0
                    c = from_surrogate(c, *_p);                         \
2575
0
                }                                                       \
2576
0
            }                                                           \
2577
0
        }                                                               \
2578
16.7k
    } while (0)
2579
2580
#define PEEK_PREV_CHAR(c, cptr, cbuf_start, cbuf_type)                  \
2581
12.6k
    do {                                                                \
2582
12.6k
        if (cbuf_type == 0) {                                           \
2583
12.6k
            c = cptr[-1];                                               \
2584
12.6k
        } else {                                                        \
2585
0
            const uint16_t *_p = (const uint16_t *)cptr - 1;            \
2586
0
            const uint16_t *_start = (const uint16_t *)cbuf_start;      \
2587
0
            c = *_p;                                                    \
2588
0
            if (is_lo_surrogate(c) && cbuf_type == 2) {                 \
2589
0
                if (_p > _start && is_hi_surrogate(_p[-1])) {           \
2590
0
                    c = from_surrogate(*--_p, c);                       \
2591
0
                }                                                       \
2592
0
            }                                                           \
2593
0
        }                                                               \
2594
12.6k
    } while (0)
2595
2596
#define GET_PREV_CHAR(c, cptr, cbuf_start, cbuf_type)                   \
2597
768k
    do {                                                                \
2598
768k
        if (cbuf_type == 0) {                                           \
2599
768k
            cptr--;                                                     \
2600
768k
            c = cptr[0];                                                \
2601
768k
        } else {                                                        \
2602
0
            const uint16_t *_p = (const uint16_t *)cptr - 1;            \
2603
0
            const uint16_t *_start = (const uint16_t *)cbuf_start;      \
2604
0
            c = *_p;                                                    \
2605
0
            if (is_lo_surrogate(c) && cbuf_type == 2) {                 \
2606
0
                if (_p > _start && is_hi_surrogate(_p[-1])) {           \
2607
0
                    c = from_surrogate(*--_p, c);                       \
2608
0
                }                                                       \
2609
0
            }                                                           \
2610
0
            cptr = (const void *)_p;                                    \
2611
0
        }                                                               \
2612
768k
    } while (0)
2613
2614
#define PREV_CHAR(cptr, cbuf_start, cbuf_type)                          \
2615
144k
    do {                                                                \
2616
144k
        if (cbuf_type == 0) {                                           \
2617
144k
            cptr--;                                                     \
2618
144k
        } else {                                                        \
2619
0
            const uint16_t *_p = (const uint16_t *)cptr - 1;            \
2620
0
            const uint16_t *_start = (const uint16_t *)cbuf_start;      \
2621
0
            if (is_lo_surrogate(*_p) && cbuf_type == 2) {               \
2622
0
                if (_p > _start && is_hi_surrogate(_p[-1])) {           \
2623
0
                    --_p;                                               \
2624
0
                }                                                       \
2625
0
            }                                                           \
2626
0
            cptr = (const void *)_p;                                    \
2627
0
        }                                                               \
2628
144k
    } while (0)
2629
2630
typedef uintptr_t StackInt;
2631
2632
typedef enum {
2633
    RE_EXEC_STATE_SPLIT,
2634
    RE_EXEC_STATE_LOOKAHEAD,
2635
    RE_EXEC_STATE_NEGATIVE_LOOKAHEAD,
2636
    RE_EXEC_STATE_GREEDY_QUANT,
2637
} REExecStateEnum;
2638
2639
typedef struct REExecState {
2640
    REExecStateEnum type : 8;
2641
    uint8_t stack_len;
2642
    size_t count; /* only used for RE_EXEC_STATE_GREEDY_QUANT */
2643
    const uint8_t *cptr;
2644
    const uint8_t *pc;
2645
    void *buf[0];
2646
} REExecState;
2647
2648
typedef struct {
2649
    const uint8_t *cbuf;
2650
    const uint8_t *cbuf_end;
2651
    /* 0 = 8 bit chars, 1 = 16 bit chars, 2 = 16 bit chars, UTF-16 */
2652
    int cbuf_type;
2653
    int capture_count;
2654
    int stack_size_max;
2655
    BOOL is_unicode;
2656
    int interrupt_counter;
2657
    void *opaque; /* used for stack overflow check */
2658
2659
    size_t state_size;
2660
    uint8_t *state_stack;
2661
    size_t state_stack_size;
2662
    size_t state_stack_len;
2663
} REExecContext;
2664
2665
static int push_state(REExecContext *s,
2666
                      uint8_t **capture,
2667
                      StackInt *stack, size_t stack_len,
2668
                      const uint8_t *pc, const uint8_t *cptr,
2669
                      REExecStateEnum type, size_t count)
2670
1.29M
{
2671
1.29M
    REExecState *rs;
2672
1.29M
    uint8_t *new_stack;
2673
1.29M
    size_t new_size, i, n;
2674
1.29M
    StackInt *stack_buf;
2675
2676
1.29M
    if (unlikely((s->state_stack_len + 1) > s->state_stack_size)) {
2677
        /* reallocate the stack */
2678
4.75k
        new_size = s->state_stack_size * 3 / 2;
2679
4.75k
        if (new_size < 8)
2680
3.29k
            new_size = 8;
2681
4.75k
        new_stack = lre_realloc(s->opaque, s->state_stack, new_size * s->state_size);
2682
4.75k
        if (!new_stack)
2683
0
            return -1;
2684
4.75k
        s->state_stack_size = new_size;
2685
4.75k
        s->state_stack = new_stack;
2686
4.75k
    }
2687
1.29M
    rs = (REExecState *)(s->state_stack + s->state_stack_len * s->state_size);
2688
1.29M
    s->state_stack_len++;
2689
1.29M
    rs->type = type;
2690
1.29M
    rs->count = count;
2691
1.29M
    rs->stack_len = stack_len;
2692
1.29M
    rs->cptr = cptr;
2693
1.29M
    rs->pc = pc;
2694
1.29M
    n = 2 * s->capture_count;
2695
43.9M
    for(i = 0; i < n; i++)
2696
42.6M
        rs->buf[i] = capture[i];
2697
1.29M
    stack_buf = (StackInt *)(rs->buf + n);
2698
10.7M
    for(i = 0; i < stack_len; i++)
2699
9.42M
        stack_buf[i] = stack[i];
2700
1.29M
    return 0;
2701
1.29M
}
2702
2703
static int lre_poll_timeout(REExecContext *s)
2704
3.71M
{
2705
3.71M
    if (unlikely(--s->interrupt_counter <= 0)) {
2706
240
        s->interrupt_counter = INTERRUPT_COUNTER_INIT;
2707
240
        if (lre_check_timeout(s->opaque))
2708
140
            return LRE_RET_TIMEOUT;
2709
240
    }
2710
3.71M
    return 0;
2711
3.71M
}
2712
2713
/* return 1 if match, 0 if not match or < 0 if error. */
2714
static intptr_t lre_exec_backtrack(REExecContext *s, uint8_t **capture,
2715
                                   StackInt *stack, int stack_len,
2716
                                   const uint8_t *pc, const uint8_t *cptr,
2717
                                   BOOL no_recurse)
2718
1.36M
{
2719
1.36M
    int opcode, ret;
2720
1.36M
    int cbuf_type;
2721
1.36M
    uint32_t val, c;
2722
1.36M
    const uint8_t *cbuf_end;
2723
2724
1.36M
    cbuf_type = s->cbuf_type;
2725
1.36M
    cbuf_end = s->cbuf_end;
2726
2727
11.6M
    for(;;) {
2728
        //        printf("top=%p: pc=%d\n", th_list.top, (int)(pc - (bc_buf + RE_HEADER_LEN)));
2729
11.6M
        opcode = *pc++;
2730
11.6M
        switch(opcode) {
2731
288k
        case REOP_match:
2732
288k
            {
2733
288k
                REExecState *rs;
2734
288k
                if (no_recurse)
2735
221k
                    return (intptr_t)cptr;
2736
66.9k
                ret = 1;
2737
66.9k
                goto recurse;
2738
2.14M
            no_match:
2739
2.14M
                if (no_recurse)
2740
133k
                    return 0;
2741
2.01M
                ret = 0;
2742
2.08M
            recurse:
2743
2.12M
                for(;;) {
2744
2.12M
                    if (lre_poll_timeout(s))
2745
72
                        return LRE_RET_TIMEOUT;
2746
2.12M
                    if (s->state_stack_len == 0)
2747
1.01M
                        return ret;
2748
1.10M
                    rs = (REExecState *)(s->state_stack +
2749
1.10M
                                         (s->state_stack_len - 1) * s->state_size);
2750
1.10M
                    if (rs->type == RE_EXEC_STATE_SPLIT) {
2751
874k
                        if (!ret) {
2752
875k
                        pop_state:
2753
875k
                            memcpy(capture, rs->buf,
2754
875k
                                   sizeof(capture[0]) * 2 * s->capture_count);
2755
931k
                        pop_state1:
2756
931k
                            pc = rs->pc;
2757
931k
                            cptr = rs->cptr;
2758
931k
                            stack_len = rs->stack_len;
2759
931k
                            memcpy(stack, rs->buf + 2 * s->capture_count,
2760
931k
                                   stack_len * sizeof(stack[0]));
2761
931k
                            s->state_stack_len--;
2762
931k
                            break;
2763
875k
                        }
2764
874k
                    } else if (rs->type == RE_EXEC_STATE_GREEDY_QUANT) {
2765
138k
                        if (!ret) {
2766
137k
                            uint32_t char_count, i;
2767
137k
                            memcpy(capture, rs->buf,
2768
137k
                                   sizeof(capture[0]) * 2 * s->capture_count);
2769
137k
                            stack_len = rs->stack_len;
2770
137k
                            memcpy(stack, rs->buf + 2 * s->capture_count,
2771
137k
                                   stack_len * sizeof(stack[0]));
2772
137k
                            pc = rs->pc;
2773
137k
                            cptr = rs->cptr;
2774
                            /* go backward */
2775
137k
                            char_count = get_u32(pc + 12);
2776
276k
                            for(i = 0; i < char_count; i++) {
2777
139k
                                PREV_CHAR(cptr, s->cbuf, cbuf_type);
2778
139k
                            }
2779
137k
                            pc = (pc + 16) + (int)get_u32(pc);
2780
137k
                            rs->cptr = cptr;
2781
137k
                            rs->count--;
2782
137k
                            if (rs->count == 0) {
2783
61.0k
                                s->state_stack_len--;
2784
61.0k
                            }
2785
137k
                            break;
2786
137k
                        }
2787
138k
                    } else {
2788
96.0k
                        ret = ((rs->type == RE_EXEC_STATE_LOOKAHEAD && ret) ||
2789
96.0k
                               (rs->type == RE_EXEC_STATE_NEGATIVE_LOOKAHEAD && !ret));
2790
96.0k
                        if (ret) {
2791
                            /* keep the capture in case of positive lookahead */
2792
79.5k
                            if (rs->type == RE_EXEC_STATE_LOOKAHEAD)
2793
56.8k
                                goto pop_state1;
2794
22.6k
                            else
2795
22.6k
                                goto pop_state;
2796
79.5k
                        }
2797
96.0k
                    }
2798
40.4k
                    s->state_stack_len--;
2799
40.4k
                }
2800
2.08M
            }
2801
1.06M
            break;
2802
1.06M
        case REOP_char32:
2803
353
        case REOP_char32_i:
2804
353
            val = get_u32(pc);
2805
353
            pc += 4;
2806
353
            goto test_char;
2807
591k
        case REOP_char:
2808
599k
        case REOP_char_i:
2809
599k
            val = get_u16(pc);
2810
599k
            pc += 2;
2811
599k
        test_char:
2812
599k
            if (cptr >= cbuf_end)
2813
393k
                goto no_match;
2814
205k
            GET_CHAR(c, cptr, cbuf_end, cbuf_type);
2815
205k
            if (opcode == REOP_char_i || opcode == REOP_char32_i) {
2816
3.31k
                c = lre_canonicalize(c, s->is_unicode);
2817
3.31k
            }
2818
205k
            if (val != c)
2819
201k
                goto no_match;
2820
4.37k
            break;
2821
96.9k
        case REOP_split_goto_first:
2822
1.12M
        case REOP_split_next_first:
2823
1.12M
            {
2824
1.12M
                const uint8_t *pc1;
2825
2826
1.12M
                val = get_u32(pc);
2827
1.12M
                pc += 4;
2828
1.12M
                if (opcode == REOP_split_next_first) {
2829
1.03M
                    pc1 = pc + (int)val;
2830
1.03M
                } else {
2831
96.9k
                    pc1 = pc;
2832
96.9k
                    pc = pc + (int)val;
2833
96.9k
                }
2834
1.12M
                ret = push_state(s, capture, stack, stack_len,
2835
1.12M
                                 pc1, cptr, RE_EXEC_STATE_SPLIT, 0);
2836
1.12M
                if (ret < 0)
2837
0
                    return LRE_RET_MEMORY_ERROR;
2838
1.12M
                break;
2839
1.12M
            }
2840
1.12M
        case REOP_lookahead:
2841
96.0k
        case REOP_negative_lookahead:
2842
96.0k
            val = get_u32(pc);
2843
96.0k
            pc += 4;
2844
96.0k
            ret = push_state(s, capture, stack, stack_len,
2845
96.0k
                             pc + (int)val, cptr,
2846
96.0k
                             RE_EXEC_STATE_LOOKAHEAD + opcode - REOP_lookahead,
2847
96.0k
                             0);
2848
96.0k
            if (ret < 0)
2849
0
                return LRE_RET_MEMORY_ERROR;
2850
96.0k
            break;
2851
2852
608k
        case REOP_goto:
2853
608k
            val = get_u32(pc);
2854
608k
            pc += 4 + (int)val;
2855
608k
            if (lre_poll_timeout(s))
2856
12
                return LRE_RET_TIMEOUT;
2857
608k
            break;
2858
608k
        case REOP_line_start:
2859
28.9k
        case REOP_line_start_m:
2860
28.9k
            if (cptr == s->cbuf)
2861
21.0k
                break;
2862
7.89k
            if (opcode == REOP_line_start)
2863
365
                goto no_match;
2864
7.52k
            PEEK_PREV_CHAR(c, cptr, s->cbuf, cbuf_type);
2865
7.52k
            if (!is_line_terminator(c))
2866
6.06k
                goto no_match;
2867
1.46k
            break;
2868
1.60k
        case REOP_line_end:
2869
13.2k
        case REOP_line_end_m:
2870
13.2k
            if (cptr == cbuf_end)
2871
1.88k
                break;
2872
11.4k
            if (opcode == REOP_line_end)
2873
295
                goto no_match;
2874
11.1k
            PEEK_CHAR(c, cptr, cbuf_end, cbuf_type);
2875
11.1k
            if (!is_line_terminator(c))
2876
10.6k
                goto no_match;
2877
452
            break;
2878
1.32M
        case REOP_dot:
2879
1.32M
            if (cptr == cbuf_end)
2880
69.6k
                goto no_match;
2881
1.25M
            GET_CHAR(c, cptr, cbuf_end, cbuf_type);
2882
1.25M
            if (is_line_terminator(c))
2883
1.01M
                goto no_match;
2884
247k
            break;
2885
247k
        case REOP_any:
2886
11.3k
            if (cptr == cbuf_end)
2887
4.72k
                goto no_match;
2888
6.59k
            GET_CHAR(c, cptr, cbuf_end, cbuf_type);
2889
6.59k
            break;
2890
2.31M
        case REOP_save_start:
2891
4.18M
        case REOP_save_end:
2892
4.18M
            val = *pc++;
2893
4.18M
            assert(val < s->capture_count);
2894
4.18M
            capture[2 * val + opcode - REOP_save_start] = (uint8_t *)cptr;
2895
4.18M
            break;
2896
346k
        case REOP_save_reset:
2897
346k
            {
2898
346k
                uint32_t val2;
2899
346k
                val = pc[0];
2900
346k
                val2 = pc[1];
2901
346k
                pc += 2;
2902
346k
                assert(val2 < s->capture_count);
2903
3.25M
                while (val <= val2) {
2904
2.90M
                    capture[2 * val] = NULL;
2905
2.90M
                    capture[2 * val + 1] = NULL;
2906
2.90M
                    val++;
2907
2.90M
                }
2908
346k
            }
2909
0
            break;
2910
38.9k
        case REOP_push_i32:
2911
38.9k
            val = get_u32(pc);
2912
38.9k
            pc += 4;
2913
38.9k
            stack[stack_len++] = val;
2914
38.9k
            break;
2915
402k
        case REOP_drop:
2916
402k
            stack_len--;
2917
402k
            break;
2918
1.00M
        case REOP_loop:
2919
1.00M
            val = get_u32(pc);
2920
1.00M
            pc += 4;
2921
1.00M
            if (--stack[stack_len - 1] != 0) {
2922
631k
                pc += (int)val;
2923
631k
                if (lre_poll_timeout(s))
2924
33
                    return LRE_RET_TIMEOUT;
2925
631k
            }
2926
1.00M
            break;
2927
1.00M
        case REOP_push_char_pos:
2928
543k
            stack[stack_len++] = (uintptr_t)cptr;
2929
543k
            break;
2930
714k
        case REOP_check_advance:
2931
714k
            if (stack[--stack_len] == (uintptr_t)cptr)
2932
377k
                goto no_match;
2933
337k
            break;
2934
337k
        case REOP_word_boundary:
2935
4.81k
        case REOP_word_boundary_i:
2936
10.8k
        case REOP_not_word_boundary:
2937
11.7k
        case REOP_not_word_boundary_i:
2938
11.7k
            {
2939
11.7k
                BOOL v1, v2;
2940
11.7k
                int ignore_case = (opcode == REOP_word_boundary_i || opcode == REOP_not_word_boundary_i);
2941
11.7k
                BOOL is_boundary = (opcode == REOP_word_boundary || opcode == REOP_word_boundary_i);
2942
                /* char before */
2943
11.7k
                if (cptr == s->cbuf) {
2944
6.56k
                    v1 = FALSE;
2945
6.56k
                } else {
2946
5.16k
                    PEEK_PREV_CHAR(c, cptr, s->cbuf, cbuf_type);
2947
5.16k
                    if (ignore_case)
2948
1.40k
                        c = lre_canonicalize(c, s->is_unicode);
2949
5.16k
                    v1 = is_word_char(c);
2950
5.16k
                }
2951
                /* current char */
2952
11.7k
                if (cptr >= cbuf_end) {
2953
6.13k
                    v2 = FALSE;
2954
6.13k
                } else {
2955
5.59k
                    PEEK_CHAR(c, cptr, cbuf_end, cbuf_type);
2956
5.59k
                    if (ignore_case)
2957
1.23k
                        c = lre_canonicalize(c, s->is_unicode);
2958
5.59k
                    v2 = is_word_char(c);
2959
5.59k
                }
2960
11.7k
                if (v1 ^ v2 ^ is_boundary)
2961
7.52k
                    goto no_match;
2962
11.7k
            }
2963
4.21k
            break;
2964
10.2k
        case REOP_back_reference:
2965
11.4k
        case REOP_back_reference_i:
2966
35.9k
        case REOP_backward_back_reference:
2967
36.1k
        case REOP_backward_back_reference_i:
2968
36.1k
            {
2969
36.1k
                const uint8_t *cptr1, *cptr1_end, *cptr1_start;
2970
36.1k
                uint32_t c1, c2;
2971
2972
36.1k
                val = *pc++;
2973
36.1k
                if (val >= s->capture_count)
2974
238
                    goto no_match;
2975
35.9k
                cptr1_start = capture[2 * val];
2976
35.9k
                cptr1_end = capture[2 * val + 1];
2977
35.9k
                if (!cptr1_start || !cptr1_end)
2978
4.77k
                    break;
2979
31.1k
                if (opcode == REOP_back_reference ||
2980
31.1k
                    opcode == REOP_back_reference_i) {
2981
6.42k
                    cptr1 = cptr1_start;
2982
6.68k
                    while (cptr1 < cptr1_end) {
2983
2.65k
                        if (cptr >= cbuf_end)
2984
1.59k
                            goto no_match;
2985
1.06k
                        GET_CHAR(c1, cptr1, cptr1_end, cbuf_type);
2986
1.06k
                        GET_CHAR(c2, cptr, cbuf_end, cbuf_type);
2987
1.06k
                        if (opcode == REOP_back_reference_i) {
2988
411
                            c1 = lre_canonicalize(c1, s->is_unicode);
2989
411
                            c2 = lre_canonicalize(c2, s->is_unicode);
2990
411
                        }
2991
1.06k
                        if (c1 != c2)
2992
793
                            goto no_match;
2993
1.06k
                    }
2994
24.7k
                } else {
2995
24.7k
                    cptr1 = cptr1_end;
2996
400k
                    while (cptr1 > cptr1_start) {
2997
384k
                        if (cptr == s->cbuf)
2998
695
                            goto no_match;
2999
384k
                        GET_PREV_CHAR(c1, cptr1, cptr1_start, cbuf_type);
3000
384k
                        GET_PREV_CHAR(c2, cptr, s->cbuf, cbuf_type);
3001
384k
                        if (opcode == REOP_backward_back_reference_i) {
3002
185
                            c1 = lre_canonicalize(c1, s->is_unicode);
3003
185
                            c2 = lre_canonicalize(c2, s->is_unicode);
3004
185
                        }
3005
384k
                        if (c1 != c2)
3006
8.39k
                            goto no_match;
3007
384k
                    }
3008
24.7k
                }
3009
31.1k
            }
3010
19.6k
            break;
3011
48.0k
        case REOP_range:
3012
51.5k
        case REOP_range_i:
3013
51.5k
            {
3014
51.5k
                int n;
3015
51.5k
                uint32_t low, high, idx_min, idx_max, idx;
3016
3017
51.5k
                n = get_u16(pc); /* n must be >= 1 */
3018
51.5k
                pc += 2;
3019
51.5k
                if (cptr >= cbuf_end)
3020
32.4k
                    goto no_match;
3021
19.0k
                GET_CHAR(c, cptr, cbuf_end, cbuf_type);
3022
19.0k
                if (opcode == REOP_range_i) {
3023
1.04k
                    c = lre_canonicalize(c, s->is_unicode);
3024
1.04k
                }
3025
19.0k
                idx_min = 0;
3026
19.0k
                low = get_u16(pc + 0 * 4);
3027
19.0k
                if (c < low)
3028
402
                    goto no_match;
3029
18.6k
                idx_max = n - 1;
3030
18.6k
                high = get_u16(pc + idx_max * 4 + 2);
3031
                /* 0xffff in for last value means +infinity */
3032
18.6k
                if (unlikely(c >= 0xffff) && high == 0xffff)
3033
0
                    goto range_match;
3034
18.6k
                if (c > high)
3035
283
                    goto no_match;
3036
44.8k
                while (idx_min <= idx_max) {
3037
44.0k
                    idx = (idx_min + idx_max) / 2;
3038
44.0k
                    low = get_u16(pc + idx * 4);
3039
44.0k
                    high = get_u16(pc + idx * 4 + 2);
3040
44.0k
                    if (c < low)
3041
18.8k
                        idx_max = idx - 1;
3042
25.2k
                    else if (c > high)
3043
7.71k
                        idx_min = idx + 1;
3044
17.5k
                    else
3045
17.5k
                        goto range_match;
3046
44.0k
                }
3047
772
                goto no_match;
3048
17.5k
            range_match:
3049
17.5k
                pc += 4 * n;
3050
17.5k
            }
3051
0
            break;
3052
11.3k
        case REOP_range32:
3053
11.6k
        case REOP_range32_i:
3054
11.6k
            {
3055
11.6k
                int n;
3056
11.6k
                uint32_t low, high, idx_min, idx_max, idx;
3057
3058
11.6k
                n = get_u16(pc); /* n must be >= 1 */
3059
11.6k
                pc += 2;
3060
11.6k
                if (cptr >= cbuf_end)
3061
2.92k
                    goto no_match;
3062
8.76k
                GET_CHAR(c, cptr, cbuf_end, cbuf_type);
3063
8.76k
                if (opcode == REOP_range32_i) {
3064
268
                    c = lre_canonicalize(c, s->is_unicode);
3065
268
                }
3066
8.76k
                idx_min = 0;
3067
8.76k
                low = get_u32(pc + 0 * 8);
3068
8.76k
                if (c < low)
3069
3.04k
                    goto no_match;
3070
5.71k
                idx_max = n - 1;
3071
5.71k
                high = get_u32(pc + idx_max * 8 + 4);
3072
5.71k
                if (c > high)
3073
0
                    goto no_match;
3074
18.7k
                while (idx_min <= idx_max) {
3075
15.5k
                    idx = (idx_min + idx_max) / 2;
3076
15.5k
                    low = get_u32(pc + idx * 8);
3077
15.5k
                    high = get_u32(pc + idx * 8 + 4);
3078
15.5k
                    if (c < low)
3079
3.46k
                        idx_max = idx - 1;
3080
12.0k
                    else if (c > high)
3081
9.54k
                        idx_min = idx + 1;
3082
2.50k
                    else
3083
2.50k
                        goto range32_match;
3084
15.5k
                }
3085
3.20k
                goto no_match;
3086
3.20k
            range32_match:
3087
2.50k
                pc += 8 * n;
3088
2.50k
            }
3089
0
            break;
3090
5.32k
        case REOP_prev:
3091
            /* go to the previous char */
3092
5.32k
            if (cptr == s->cbuf)
3093
494
                goto no_match;
3094
4.82k
            PREV_CHAR(cptr, s->cbuf, cbuf_type);
3095
4.82k
            break;
3096
180k
        case REOP_simple_greedy_quant:
3097
180k
            {
3098
180k
                uint32_t next_pos, quant_min, quant_max;
3099
180k
                size_t q;
3100
180k
                intptr_t res;
3101
180k
                const uint8_t *pc1;
3102
3103
180k
                next_pos = get_u32(pc);
3104
180k
                quant_min = get_u32(pc + 4);
3105
180k
                quant_max = get_u32(pc + 8);
3106
180k
                pc += 16;
3107
180k
                pc1 = pc;
3108
180k
                pc += (int)next_pos;
3109
3110
180k
                q = 0;
3111
354k
                for(;;) {
3112
354k
                    if (lre_poll_timeout(s))
3113
23
                        return LRE_RET_TIMEOUT;
3114
354k
                    res = lre_exec_backtrack(s, capture, stack, stack_len,
3115
354k
                                             pc1, cptr, TRUE);
3116
354k
                    if (res == LRE_RET_MEMORY_ERROR ||
3117
354k
                        res == LRE_RET_TIMEOUT)
3118
0
                        return res;
3119
354k
                    if (!res)
3120
133k
                        break;
3121
221k
                    cptr = (uint8_t *)res;
3122
221k
                    q++;
3123
221k
                    if (q >= quant_max && quant_max != INT32_MAX)
3124
47.3k
                        break;
3125
221k
                }
3126
180k
                if (q < quant_min)
3127
9.94k
                    goto no_match;
3128
170k
                if (q > quant_min) {
3129
                    /* will examine all matches down to quant_min */
3130
68.7k
                    ret = push_state(s, capture, stack, stack_len,
3131
68.7k
                                     pc1 - 16, cptr,
3132
68.7k
                                     RE_EXEC_STATE_GREEDY_QUANT,
3133
68.7k
                                     q - quant_min);
3134
68.7k
                    if (ret < 0)
3135
0
                        return LRE_RET_MEMORY_ERROR;
3136
68.7k
                }
3137
170k
            }
3138
170k
            break;
3139
170k
        default:
3140
0
            abort();
3141
11.6M
        }
3142
11.6M
    }
3143
1.36M
}
3144
3145
/* Return 1 if match, 0 if not match or < 0 if error (see LRE_RET_x). cindex is the
3146
   starting position of the match and must be such as 0 <= cindex <=
3147
   clen. */
3148
int lre_exec(uint8_t **capture,
3149
             const uint8_t *bc_buf, const uint8_t *cbuf, int cindex, int clen,
3150
             int cbuf_type, void *opaque)
3151
1.01M
{
3152
1.01M
    REExecContext s_s, *s = &s_s;
3153
1.01M
    int re_flags, i, alloca_size, ret;
3154
1.01M
    StackInt *stack_buf;
3155
3156
1.01M
    re_flags = lre_get_flags(bc_buf);
3157
1.01M
    s->is_unicode = (re_flags & (LRE_FLAG_UNICODE | LRE_FLAG_UNICODE_SETS)) != 0;
3158
1.01M
    s->capture_count = bc_buf[RE_HEADER_CAPTURE_COUNT];
3159
1.01M
    s->stack_size_max = bc_buf[RE_HEADER_STACK_SIZE];
3160
1.01M
    s->cbuf = cbuf;
3161
1.01M
    s->cbuf_end = cbuf + (clen << cbuf_type);
3162
1.01M
    s->cbuf_type = cbuf_type;
3163
1.01M
    if (s->cbuf_type == 1 && s->is_unicode)
3164
1.00M
        s->cbuf_type = 2;
3165
1.01M
    s->interrupt_counter = INTERRUPT_COUNTER_INIT;
3166
1.01M
    s->opaque = opaque;
3167
3168
1.01M
    s->state_size = sizeof(REExecState) +
3169
1.01M
        s->capture_count * sizeof(capture[0]) * 2 +
3170
1.01M
        s->stack_size_max * sizeof(stack_buf[0]);
3171
1.01M
    s->state_stack = NULL;
3172
1.01M
    s->state_stack_len = 0;
3173
1.01M
    s->state_stack_size = 0;
3174
3175
3.05M
    for(i = 0; i < s->capture_count * 2; i++)
3176
2.04M
        capture[i] = NULL;
3177
1.01M
    alloca_size = s->stack_size_max * sizeof(stack_buf[0]);
3178
1.01M
    stack_buf = alloca(alloca_size);
3179
1.01M
    ret = lre_exec_backtrack(s, capture, stack_buf, 0, bc_buf + RE_HEADER_LEN,
3180
1.01M
                             cbuf + (cindex << cbuf_type), FALSE);
3181
1.01M
    lre_realloc(s->opaque, s->state_stack, 0);
3182
1.01M
    return ret;
3183
1.01M
}
3184
3185
int lre_get_capture_count(const uint8_t *bc_buf)
3186
1.01M
{
3187
1.01M
    return bc_buf[RE_HEADER_CAPTURE_COUNT];
3188
1.01M
}
3189
3190
int lre_get_flags(const uint8_t *bc_buf)
3191
2.02M
{
3192
2.02M
    return get_u16(bc_buf + RE_HEADER_FLAGS);
3193
2.02M
}
3194
3195
/* Return NULL if no group names. Otherwise, return a pointer to
3196
   'capture_count - 1' zero terminated UTF-8 strings. */
3197
const char *lre_get_groupnames(const uint8_t *bc_buf)
3198
0
{
3199
0
    uint32_t re_bytecode_len;
3200
0
    if ((lre_get_flags(bc_buf) & LRE_FLAG_NAMED_GROUPS) == 0)
3201
0
        return NULL;
3202
0
    re_bytecode_len = get_u32(bc_buf + RE_HEADER_BYTECODE_LEN);
3203
0
    return (const char *)(bc_buf + RE_HEADER_LEN + re_bytecode_len);
3204
0
}
3205
3206
#ifdef TEST
3207
3208
BOOL lre_check_stack_overflow(void *opaque, size_t alloca_size)
3209
{
3210
    return FALSE;
3211
}
3212
3213
void *lre_realloc(void *opaque, void *ptr, size_t size)
3214
{
3215
    return realloc(ptr, size);
3216
}
3217
3218
int main(int argc, char **argv)
3219
{
3220
    int len, flags, ret, i;
3221
    uint8_t *bc;
3222
    char error_msg[64];
3223
    uint8_t *capture[CAPTURE_COUNT_MAX * 2];
3224
    const char *input;
3225
    int input_len, capture_count;
3226
3227
    if (argc < 4) {
3228
        printf("usage: %s regexp flags input\n", argv[0]);
3229
        return 1;
3230
    }
3231
    flags = atoi(argv[2]);
3232
    bc = lre_compile(&len, error_msg, sizeof(error_msg), argv[1],
3233
                     strlen(argv[1]), flags, NULL);
3234
    if (!bc) {
3235
        fprintf(stderr, "error: %s\n", error_msg);
3236
        exit(1);
3237
    }
3238
3239
    input = argv[3];
3240
    input_len = strlen(input);
3241
3242
    ret = lre_exec(capture, bc, (uint8_t *)input, 0, input_len, 0, NULL);
3243
    printf("ret=%d\n", ret);
3244
    if (ret == 1) {
3245
        capture_count = lre_get_capture_count(bc);
3246
        for(i = 0; i < 2 * capture_count; i++) {
3247
            uint8_t *ptr;
3248
            ptr = capture[i];
3249
            printf("%d: ", i);
3250
            if (!ptr)
3251
                printf("<nil>");
3252
            else
3253
                printf("%u", (int)(ptr - (uint8_t *)input));
3254
            printf("\n");
3255
        }
3256
    }
3257
    return 0;
3258
}
3259
#endif