Coverage Report

Created: 2025-08-08 06:35

/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
118k
#define CAPTURE_COUNT_MAX 255
56
5.50k
#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
3.87k
#define INTERRUPT_COUNTER_INIT 10000
60
61
/* unicode code points */
62
467k
#define CP_LS   0x2028
63
231k
#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
3.67k
#define RE_HEADER_FLAGS         0
107
8.05k
#define RE_HEADER_CAPTURE_COUNT 2
108
7.28k
#define RE_HEADER_STACK_SIZE    3
109
3.64k
#define RE_HEADER_BYTECODE_LEN  4
110
111
14.5k
#define RE_HEADER_LEN 8
112
113
18.5k
static inline int is_digit(int c) {
114
18.5k
    return c >= '0' && c <= '9';
115
18.5k
}
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.9k
{
120
12.9k
    if (dbuf_realloc(s, s->size + len))
121
0
        return -1;
122
12.9k
    memmove(s->buf + pos + len, s->buf + pos, s->size - pos);
123
12.9k
    s->size += len;
124
12.9k
    return 0;
125
12.9k
}
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
6.22k
{
157
6.22k
    cr_init(&s->cr, s1->opaque, lre_realloc);
158
6.22k
    s->n_strings = 0;
159
6.22k
    s->hash_size = 0;
160
6.22k
    s->hash_bits = 0;
161
6.22k
    s->hash_table = NULL;
162
6.22k
}
163
164
static void re_string_list_free(REStringList *s)
165
6.22k
{
166
6.22k
    REString *p, *p_next;
167
6.22k
    int i;
168
6.22k
    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
6.22k
    lre_realloc(s->cr.mem_opaque, s->hash_table, 0);
175
176
6.22k
    cr_free(&s->cr);
177
6.22k
}
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
937
{
300
937
    int i, ret;
301
937
    REString *p, **pp;
302
303
937
    if (cr_op1(&a->cr, b->cr.points, b->cr.len, op))
304
0
        return -1;
305
306
937
    switch(op) {
307
937
    case CR_OP_UNION:
308
937
        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
937
        break;
317
937
    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
937
    }
343
937
    return 0;
344
937
}
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
176k
#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
3.15k
{
435
3.15k
    BOOL invert;
436
3.15k
    const uint16_t *c_pt;
437
3.15k
    int len, i;
438
439
3.15k
    invert = c & 1;
440
3.15k
    c_pt = char_range_table[c >> 1];
441
3.15k
    len = *c_pt++;
442
3.15k
    re_string_list_init(s, cr);
443
41.7k
    for(i = 0; i < len * 2; i++) {
444
38.6k
        if (cr_add_point(&cr->cr, c_pt[i]))
445
0
            goto fail;
446
38.6k
    }
447
3.15k
    if (invert) {
448
1.34k
        if (cr_invert(&cr->cr))
449
0
            goto fail;
450
1.34k
    }
451
3.15k
    return 0;
452
0
 fail:
453
0
    re_string_list_free(cr);
454
0
    return -1;
455
3.15k
}
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
36.0k
{
585
36.0k
    dbuf_putc(&s->byte_code, op);
586
36.0k
}
587
588
/* return the offset of the u32 value */
589
static int re_emit_op_u32(REParseState *s, int op, uint32_t val)
590
28.8k
{
591
28.8k
    int pos;
592
28.8k
    dbuf_putc(&s->byte_code, op);
593
28.8k
    pos = s->byte_code.size;
594
28.8k
    dbuf_put_u32(&s->byte_code, val);
595
28.8k
    return pos;
596
28.8k
}
597
598
static int re_emit_goto(REParseState *s, int op, uint32_t val)
599
8.07k
{
600
8.07k
    int pos;
601
8.07k
    dbuf_putc(&s->byte_code, op);
602
8.07k
    pos = s->byte_code.size;
603
8.07k
    dbuf_put_u32(&s->byte_code, val - (pos + 4));
604
8.07k
    return pos;
605
8.07k
}
606
607
static void re_emit_op_u8(REParseState *s, int op, uint32_t val)
608
61.8k
{
609
61.8k
    dbuf_putc(&s->byte_code, op);
610
61.8k
    dbuf_putc(&s->byte_code, val);
611
61.8k
}
612
613
static void re_emit_op_u16(REParseState *s, int op, uint32_t val)
614
149k
{
615
149k
    dbuf_putc(&s->byte_code, op);
616
149k
    dbuf_put_u16(&s->byte_code, val);
617
149k
}
618
619
static int __attribute__((format(printf, 2, 3))) re_parse_error(REParseState *s, const char *fmt, ...)
620
2.96k
{
621
2.96k
    va_list ap;
622
2.96k
    va_start(ap, fmt);
623
2.96k
    vsnprintf(s->u.error_msg, sizeof(s->u.error_msg), fmt, ap);
624
2.96k
    va_end(ap);
625
2.96k
    return -1;
626
2.96k
}
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
14.4k
{
637
14.4k
    const uint8_t *p;
638
14.4k
    uint64_t v;
639
14.4k
    int c;
640
641
14.4k
    p = *pp;
642
14.4k
    v = 0;
643
38.4k
    for(;;) {
644
38.4k
        c = *p;
645
38.4k
        if (c < '0' || c > '9')
646
14.2k
            break;
647
24.2k
        v = v * 10 + c - '0';
648
24.2k
        if (v >= INT32_MAX) {
649
1.75k
            if (allow_overflow)
650
1.55k
                v = INT32_MAX;
651
201
            else
652
201
                return -1;
653
1.75k
        }
654
24.0k
        p++;
655
24.0k
    }
656
14.2k
    *pp = p;
657
14.2k
    return v;
658
14.4k
}
659
660
static int re_parse_expect(REParseState *s, const uint8_t **pp, int c)
661
17.4k
{
662
17.4k
    const uint8_t *p;
663
17.4k
    p = *pp;
664
17.4k
    if (*p != c)
665
699
        return re_parse_error(s, "expecting '%c'", c);
666
16.7k
    p++;
667
16.7k
    *pp = p;
668
16.7k
    return 0;
669
17.4k
}
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
22.4k
{
683
22.4k
    const uint8_t *p;
684
22.4k
    uint32_t c;
685
686
22.4k
    p = *pp;
687
22.4k
    c = *p++;
688
22.4k
    switch(c) {
689
301
    case 'b':
690
301
        c = '\b';
691
301
        break;
692
251
    case 'f':
693
251
        c = '\f';
694
251
        break;
695
215
    case 'n':
696
215
        c = '\n';
697
215
        break;
698
199
    case 'r':
699
199
        c = '\r';
700
199
        break;
701
200
    case 't':
702
200
        c = '\t';
703
200
        break;
704
194
    case 'v':
705
194
        c = '\v';
706
194
        break;
707
472
    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.33k
                p++;
715
1.33k
                c = 0;
716
3.14k
                for(;;) {
717
3.14k
                    h = from_hex(*p++);
718
3.14k
                    if (h < 0)
719
894
                        return -1;
720
2.25k
                    c = (c << 4) | h;
721
2.25k
                    if (c > 0x10FFFF)
722
245
                        return -1;
723
2.00k
                    if (*p == '}')
724
199
                        break;
725
2.00k
                }
726
199
                p++;
727
12.9k
            } else {
728
12.9k
                if (c == 'x') {
729
472
                    n = 2;
730
12.4k
                } else {
731
12.4k
                    n = 4;
732
12.4k
                }
733
734
12.9k
                c = 0;
735
51.7k
                for(i = 0; i < n; i++) {
736
43.5k
                    h = from_hex(*p++);
737
43.5k
                    if (h < 0) {
738
4.78k
                        return -1;
739
4.78k
                    }
740
38.8k
                    c = (c << 4) | h;
741
38.8k
                }
742
8.11k
                if (is_hi_surrogate(c) &&
743
8.11k
                    allow_utf16 == 2 && p[0] == '\\' && p[1] == 'u') {
744
                    /* convert an escaped surrogate pair into a
745
                       unicode char */
746
5.02k
                    c1 = 0;
747
15.3k
                    for(i = 0; i < 4; i++) {
748
14.3k
                        h = from_hex(p[2 + i]);
749
14.3k
                        if (h < 0)
750
4.06k
                            break;
751
10.3k
                        c1 = (c1 << 4) | h;
752
10.3k
                    }
753
5.02k
                    if (i == 4 && is_lo_surrogate(c1)) {
754
438
                        p += 6;
755
438
                        c = from_surrogate(c, c1);
756
438
                    }
757
5.02k
                }
758
8.11k
            }
759
14.2k
        }
760
8.31k
        break;
761
8.31k
    case '0': case '1': case '2': case '3':
762
3.48k
    case '4': case '5': case '6': case '7':
763
3.48k
        c -= '0';
764
3.48k
        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.48k
        } else {
769
            /* parse a legacy octal sequence */
770
3.48k
            uint32_t v;
771
3.48k
            v = *p - '0';
772
3.48k
            if (v > 7)
773
2.85k
                break;
774
632
            c = (c << 3) | v;
775
632
            p++;
776
632
            if (c >= 32)
777
219
                break;
778
413
            v = *p - '0';
779
413
            if (v > 7)
780
211
                break;
781
202
            c = (c << 3) | v;
782
202
            p++;
783
202
        }
784
202
        break;
785
3.36k
    default:
786
3.36k
        return -2;
787
22.4k
    }
788
13.1k
    *pp = p;
789
13.1k
    return c;
790
22.4k
}
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
169k
{
987
169k
    const uint8_t *p;
988
169k
    uint32_t c;
989
169k
    int ret;
990
991
169k
    p = *pp;
992
993
169k
    c = *p;
994
169k
    switch(c) {
995
20.2k
    case '\\':
996
20.2k
        p++;
997
20.2k
        if (p >= s->buf_end)
998
85
            goto unexpected_end;
999
20.1k
        c = *p++;
1000
20.1k
        switch(c) {
1001
529
        case 'd':
1002
529
            c = CHAR_RANGE_d;
1003
529
            goto class_range;
1004
294
        case 'D':
1005
294
            c = CHAR_RANGE_D;
1006
294
            goto class_range;
1007
824
        case 's':
1008
824
            c = CHAR_RANGE_s;
1009
824
            goto class_range;
1010
702
        case 'S':
1011
702
            c = CHAR_RANGE_S;
1012
702
            goto class_range;
1013
461
        case 'w':
1014
461
            c = CHAR_RANGE_w;
1015
461
            goto class_range;
1016
346
        case 'W':
1017
346
            c = CHAR_RANGE_W;
1018
3.15k
        class_range:
1019
3.15k
            if (!cr)
1020
0
                goto default_escape;
1021
3.15k
            if (cr_init_char_range(s, cr, c))
1022
0
                return -1;
1023
3.15k
            c = CLASS_RANGE_BASE;
1024
3.15k
            break;
1025
2.44k
        case 'c':
1026
2.44k
            c = *p;
1027
2.44k
            if ((c >= 'a' && c <= 'z') ||
1028
2.44k
                (c >= 'A' && c <= 'Z') ||
1029
2.44k
                (((c >= '0' && c <= '9') || c == '_') &&
1030
2.00k
                 inclass && !s->is_unicode)) {   /* Annex B.1.4 */
1031
634
                c &= 0x1f;
1032
634
                p++;
1033
1.80k
            } else if (s->is_unicode) {
1034
0
                goto invalid_escape;
1035
1.80k
            } else {
1036
                /* otherwise return '\' and 'c' */
1037
1.80k
                p--;
1038
1.80k
                c = '\\';
1039
1.80k
            }
1040
2.44k
            break;
1041
2.44k
        case '-':
1042
426
            if (!inclass && s->is_unicode)
1043
0
                goto invalid_escape;
1044
426
            break;
1045
426
        case '^':
1046
451
        case '$':
1047
1.36k
        case '\\':
1048
1.56k
        case '.':
1049
1.77k
        case '*':
1050
1.97k
        case '+':
1051
2.27k
        case '?':
1052
2.53k
        case '(':
1053
2.74k
        case ')':
1054
2.94k
        case '[':
1055
3.14k
        case ']':
1056
3.34k
        case '{':
1057
3.54k
        case '}':
1058
3.74k
        case '|':
1059
3.97k
        case '/':
1060
            /* always valid to escape these characters */
1061
3.97k
            break;
1062
0
#ifdef CONFIG_ALL_UNICODE
1063
197
        case 'p':
1064
391
        case 'P':
1065
391
            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
391
            goto default_escape;
1072
391
#endif
1073
391
        case 'q':
1074
207
            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
207
            goto default_escape;
1081
9.58k
        default:
1082
10.1k
        default_escape:
1083
10.1k
            p--;
1084
10.1k
            ret = lre_parse_escape(&p, s->is_unicode * 2);
1085
10.1k
            if (ret >= 0) {
1086
5.65k
                c = ret;
1087
5.65k
            } else {
1088
4.52k
                if (s->is_unicode) {
1089
0
                invalid_escape:
1090
0
                    return re_parse_error(s, "invalid escape sequence in regular expression");
1091
4.52k
                } else {
1092
                    /* just ignore the '\' */
1093
4.52k
                    goto normal_char;
1094
4.52k
                }
1095
4.52k
            }
1096
5.65k
            break;
1097
20.1k
        }
1098
15.6k
        break;
1099
15.6k
    case '\0':
1100
664
        if (p >= s->buf_end) {
1101
749
        unexpected_end:
1102
749
            return re_parse_error(s, "unexpected end");
1103
664
        }
1104
        /* fall thru */
1105
0
        goto normal_char;
1106
1107
490
    case '&':
1108
1.20k
    case '!':
1109
1.66k
    case '#':
1110
2.01k
    case '$':
1111
2.39k
    case '%':
1112
2.59k
    case '*':
1113
2.87k
    case '+':
1114
5.19k
    case ',':
1115
5.47k
    case '.':
1116
9.06k
    case ':':
1117
9.53k
    case ';':
1118
12.1k
    case '<':
1119
12.8k
    case '=':
1120
14.5k
    case '>':
1121
14.9k
    case '?':
1122
15.4k
    case '@':
1123
15.6k
    case '^':
1124
16.0k
    case '`':
1125
16.5k
    case '~':
1126
16.5k
        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
16.5k
        goto normal_char;
1131
1132
16.5k
    case '(':
1133
670
    case ')':
1134
1.38k
    case '[':
1135
1.88k
    case ']':
1136
7.93k
    case '{':
1137
8.56k
    case '}':
1138
9.79k
    case '/':
1139
18.7k
    case '-':
1140
18.9k
    case '|':
1141
18.9k
        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
18.9k
        goto normal_char;
1146
1147
112k
    default:
1148
152k
    normal_char:
1149
        /* normal char */
1150
152k
        if (c >= 128) {
1151
4.61k
            c = unicode_from_utf8(p, UTF8_CHAR_LEN_MAX, &p);
1152
4.61k
            if ((unsigned)c > 0xffff && !s->is_unicode) {
1153
                /* XXX: should handle non BMP-1 code points */
1154
411
                return re_parse_error(s, "malformed unicode char");
1155
411
            }
1156
148k
        } else {
1157
148k
            p++;
1158
148k
        }
1159
152k
        break;
1160
169k
    }
1161
167k
    *pp = p;
1162
167k
    return c;
1163
169k
}
1164
1165
static int re_emit_range(REParseState *s, const CharRange *cr)
1166
4.25k
{
1167
4.25k
    int len, i;
1168
4.25k
    uint32_t high;
1169
1170
4.25k
    len = (unsigned)cr->len / 2;
1171
4.25k
    if (len >= 65535)
1172
0
        return re_parse_error(s, "too many ranges");
1173
4.25k
    if (len == 0) {
1174
1.06k
        re_emit_op_u32(s, REOP_char32, -1);
1175
3.19k
    } else {
1176
3.19k
        high = cr->points[cr->len - 1];
1177
3.19k
        if (high == UINT32_MAX)
1178
1.38k
            high = cr->points[cr->len - 2];
1179
3.19k
        if (high <= 0xffff) {
1180
            /* can use 16 bit ranges with the conversion that 0xffff =
1181
               infinity */
1182
2.76k
            re_emit_op_u16(s, s->ignore_case ? REOP_range_i : REOP_range, len);
1183
21.9k
            for(i = 0; i < cr->len; i += 2) {
1184
19.1k
                dbuf_put_u16(&s->byte_code, cr->points[i]);
1185
19.1k
                high = cr->points[i + 1] - 1;
1186
19.1k
                if (high == UINT32_MAX - 1)
1187
1.36k
                    high = 0xffff;
1188
19.1k
                dbuf_put_u16(&s->byte_code, high);
1189
19.1k
            }
1190
2.76k
        } else {
1191
434
            re_emit_op_u16(s, s->ignore_case ? REOP_range32_i : REOP_range32, len);
1192
6.16k
            for(i = 0; i < cr->len; i += 2) {
1193
5.73k
                dbuf_put_u32(&s->byte_code, cr->points[i]);
1194
5.73k
                dbuf_put_u32(&s->byte_code, cr->points[i + 1] - 1);
1195
5.73k
            }
1196
434
        }
1197
3.19k
    }
1198
4.25k
    return 0;
1199
4.25k
}
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
146k
{
1210
146k
    if (c <= 0xffff)
1211
146k
        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
146k
}
1215
1216
static int re_emit_string_list(REParseState *s, const REStringList *sl)
1217
4.25k
{
1218
4.25k
    REString **tab, *p;
1219
4.25k
    int i, j, split_pos, last_match_pos, n;
1220
4.25k
    BOOL has_empty_string, is_last;
1221
    
1222
    //    re_string_list_dump("sl", sl);
1223
4.25k
    if (sl->n_strings == 0) {
1224
        /* simple case: only characters */
1225
4.25k
        if (re_emit_range(s, &sl->cr))
1226
0
            return -1;
1227
4.25k
    } 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.25k
    return 0;
1292
4.25k
}
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
3.06k
{
1324
3.06k
    const uint8_t *p;
1325
3.06k
    uint32_t c1, c2;
1326
3.06k
    int ret;
1327
3.06k
    REStringList cr1_s, *cr1 = &cr1_s;
1328
3.06k
    BOOL invert, is_first;
1329
1330
3.06k
    if (lre_check_stack_overflow(s->opaque, 0))
1331
0
        return re_parse_error(s, "stack overflow");
1332
1333
3.06k
    re_string_list_init(s, cr);
1334
3.06k
    p = *pp;
1335
3.06k
    p++;    /* skip '[' */
1336
1337
3.06k
    invert = FALSE;
1338
3.06k
    if (*p == '^') {
1339
367
        p++;
1340
367
        invert = TRUE;
1341
367
    }
1342
    
1343
    /* handle unions */
1344
3.06k
    is_first = TRUE;
1345
21.8k
    for(;;) {
1346
21.8k
        if (*p == ']')
1347
2.30k
            break;
1348
19.5k
        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
19.5k
        } else {
1353
19.5k
            c1 = get_class_atom(s, cr1, &p, TRUE);
1354
19.5k
            if ((int)c1 < 0)
1355
731
                goto fail;
1356
18.8k
            if (*p == '-' && p[1] != ']') {
1357
5.92k
                const uint8_t *p0 = p + 1;
1358
5.92k
                if (p[1] == '-' && s->unicode_sets && is_first)
1359
0
                    goto class_atom; /* first character class followed by '--' */
1360
5.92k
                if (c1 >= CLASS_RANGE_BASE) {
1361
323
                    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
323
                    goto class_atom;
1367
323
                }
1368
5.60k
                c2 = get_class_atom(s, cr1, &p0, TRUE);
1369
5.60k
                if ((int)c2 < 0)
1370
21
                    goto fail;
1371
5.58k
                if (c2 >= CLASS_RANGE_BASE) {
1372
264
                    re_string_list_free(cr1);
1373
264
                    if (s->is_unicode) {
1374
0
                        goto invalid_class_range;
1375
0
                    }
1376
                    /* Annex B: match '-' character */
1377
264
                    goto class_atom;
1378
264
                }
1379
5.31k
                p = p0;
1380
5.31k
                if (c2 < c1) {
1381
13
                invalid_class_range:
1382
13
                    re_parse_error(s, "invalid class range");
1383
13
                    goto fail;
1384
13
                }
1385
5.30k
                if (s->ignore_case) {
1386
5.04k
                    CharRange cr2_s, *cr2 = &cr2_s;
1387
5.04k
                    cr_init(cr2, s->opaque, lre_realloc);
1388
5.04k
                    if (cr_add_interval(cr2, c1, c2 + 1) ||
1389
5.04k
                        cr_regexp_canonicalize(cr2, s->is_unicode) ||
1390
5.04k
                        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.04k
                    cr_free(cr2);
1395
5.04k
                } else {
1396
263
                    if (cr_union_interval(&cr->cr, c1, c2))
1397
0
                        goto memory_error;
1398
263
                }
1399
5.30k
                is_first = FALSE; /* union operation */
1400
12.8k
            } else {
1401
13.4k
            class_atom:
1402
13.4k
                if (c1 >= CLASS_RANGE_BASE) {
1403
937
                class_union:
1404
937
                    ret = re_string_list_op(cr, cr1, CR_OP_UNION);
1405
937
                    re_string_list_free(cr1);
1406
937
                    if (ret)
1407
0
                        goto memory_error;
1408
12.5k
                } else {
1409
12.5k
                    if (s->ignore_case)
1410
2.24k
                        c1 = lre_canonicalize(c1, s->is_unicode);
1411
12.5k
                    if (cr_union_interval(&cr->cr, c1, c1))
1412
0
                        goto memory_error;
1413
12.5k
                }
1414
13.4k
            }
1415
18.8k
        }
1416
18.7k
        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
18.7k
        is_first = FALSE;
1456
18.7k
    }
1457
1458
2.30k
    p++;    /* skip ']' */
1459
2.30k
    *pp = p;
1460
2.30k
    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.30k
    return 0;
1472
0
 memory_error:
1473
0
    re_parse_out_of_memory(s);
1474
765
 fail:
1475
765
    re_string_list_free(cr);
1476
765
    return -1;
1477
0
}
1478
1479
static int re_parse_char_class(REParseState *s, const uint8_t **pp)
1480
3.06k
{
1481
3.06k
    REStringList cr_s, *cr = &cr_s;
1482
1483
3.06k
    if (re_parse_nested_class(s, cr, pp))
1484
765
        return -1;
1485
2.30k
    if (re_emit_string_list(s, cr))
1486
0
        goto fail;
1487
2.30k
    re_string_list_free(cr);
1488
2.30k
    return 0;
1489
0
 fail:
1490
0
    re_string_list_free(cr);
1491
0
    return -1;
1492
2.30k
}
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.73k
{
1500
9.73k
    int pos, opcode, len;
1501
9.73k
    uint32_t val;
1502
9.73k
    BOOL ret;
1503
1504
9.73k
    ret = TRUE;
1505
9.73k
    pos = 0;
1506
63.2k
    while (pos < bc_buf_len) {
1507
59.7k
        opcode = bc_buf[pos];
1508
59.7k
        len = reopcode_info[opcode].size;
1509
59.7k
        switch(opcode) {
1510
457
        case REOP_range:
1511
780
        case REOP_range_i:
1512
780
            val = get_u16(bc_buf + pos + 1);
1513
780
            len += val * 4;
1514
780
            goto simple_char;
1515
264
        case REOP_range32:
1516
557
        case REOP_range32_i:
1517
557
            val = get_u16(bc_buf + pos + 1);
1518
557
            len += val * 8;
1519
557
            goto simple_char;
1520
3.58k
        case REOP_char:
1521
5.32k
        case REOP_char_i:
1522
5.97k
        case REOP_char32:
1523
5.97k
        case REOP_char32_i:
1524
6.79k
        case REOP_dot:
1525
7.03k
        case REOP_any:
1526
8.37k
        simple_char:
1527
8.37k
            ret = FALSE;
1528
8.37k
            break;
1529
823
        case REOP_line_start:
1530
1.23k
        case REOP_line_start_m:
1531
1.69k
        case REOP_line_end:
1532
1.95k
        case REOP_line_end_m:
1533
7.46k
        case REOP_push_i32:
1534
7.46k
        case REOP_push_char_pos:
1535
7.46k
        case REOP_drop:
1536
7.86k
        case REOP_word_boundary:
1537
8.24k
        case REOP_word_boundary_i:
1538
8.48k
        case REOP_not_word_boundary:
1539
8.86k
        case REOP_not_word_boundary_i:
1540
11.3k
        case REOP_prev:
1541
            /* no effect */
1542
11.3k
            break;
1543
26.9k
        case REOP_save_start:
1544
29.8k
        case REOP_save_end:
1545
32.3k
        case REOP_save_reset:
1546
32.5k
        case REOP_back_reference:
1547
32.8k
        case REOP_back_reference_i:
1548
33.0k
        case REOP_backward_back_reference:
1549
33.8k
        case REOP_backward_back_reference_i:
1550
33.8k
            break;
1551
6.16k
        default:
1552
            /* safe behavior: we cannot predict the outcome */
1553
6.16k
            return TRUE;
1554
59.7k
        }
1555
53.5k
        pos += len;
1556
53.5k
    }
1557
3.56k
    return ret;
1558
9.73k
}
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
90.0k
    while (pos < bc_buf_len) {
1570
87.1k
        opcode = bc_buf[pos];
1571
87.1k
        len = reopcode_info[opcode].size;
1572
87.1k
        switch(opcode) {
1573
411
        case REOP_range:
1574
661
        case REOP_range_i:
1575
661
            val = get_u16(bc_buf + pos + 1);
1576
661
            len += val * 4;
1577
661
            goto simple_char;
1578
218
        case REOP_range32:
1579
467
        case REOP_range32_i:
1580
467
            val = get_u16(bc_buf + pos + 1);
1581
467
            len += val * 8;
1582
467
            goto simple_char;
1583
73.4k
        case REOP_char:
1584
74.6k
        case REOP_char_i:
1585
75.1k
        case REOP_char32:
1586
75.1k
        case REOP_char32_i:
1587
76.1k
        case REOP_dot:
1588
76.3k
        case REOP_any:
1589
77.4k
        simple_char:
1590
77.4k
            count++;
1591
77.4k
            break;
1592
556
        case REOP_line_start:
1593
797
        case REOP_line_start_m:
1594
1.02k
        case REOP_line_end:
1595
1.22k
        case REOP_line_end_m:
1596
1.42k
        case REOP_word_boundary:
1597
1.63k
        case REOP_word_boundary_i:
1598
1.83k
        case REOP_not_word_boundary:
1599
2.04k
        case REOP_not_word_boundary_i:
1600
2.04k
            break;
1601
7.66k
        default:
1602
7.66k
            return -1;
1603
87.1k
        }
1604
79.5k
        pos += len;
1605
79.5k
    }
1606
2.81k
    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.5k
{
1612
21.5k
    const uint8_t *p, *p1;
1613
21.5k
    uint32_t c, d;
1614
21.5k
    char *q;
1615
1616
21.5k
    p = *pp;
1617
21.5k
    q = buf;
1618
66.2k
    for(;;) {
1619
66.2k
        c = *p;
1620
66.2k
        if (c == '\\') {
1621
12.7k
            p++;
1622
12.7k
            if (*p != 'u')
1623
470
                return -1;
1624
12.2k
            c = lre_parse_escape(&p, 2); // accept surrogate pairs
1625
53.4k
        } else if (c == '>') {
1626
8.44k
            break;
1627
45.0k
        } else if (c >= 128) {
1628
3.69k
            c = unicode_from_utf8(p, UTF8_CHAR_LEN_MAX, &p);
1629
3.69k
            if (is_hi_surrogate(c)) {
1630
467
                d = unicode_from_utf8(p, UTF8_CHAR_LEN_MAX, &p1);
1631
467
                if (is_lo_surrogate(d)) {
1632
207
                    c = from_surrogate(c, d);
1633
207
                    p = p1;
1634
207
                }
1635
467
            }
1636
41.3k
        } else {
1637
41.3k
            p++;
1638
41.3k
        }
1639
57.3k
        if (c > 0x10FFFF)
1640
5.63k
            return -1;
1641
51.6k
        if (q == buf) {
1642
15.5k
            if (!lre_js_is_ident_first(c))
1643
5.52k
                return -1;
1644
36.1k
        } else {
1645
36.1k
            if (!lre_js_is_ident_next(c))
1646
1.29k
                return -1;
1647
36.1k
        }
1648
44.8k
        if ((q - buf + UTF8_CHAR_LEN_MAX + 1) > buf_size)
1649
195
            return -1;
1650
44.6k
        if (c < 128) {
1651
40.0k
            *q++ = c;
1652
40.0k
        } else {
1653
4.62k
            q += unicode_to_utf8((uint8_t*)q, c);
1654
4.62k
        }
1655
44.6k
    }
1656
8.44k
    if (q == buf)
1657
212
        return -1;
1658
8.23k
    *q = '\0';
1659
8.23k
    p++;
1660
8.23k
    *pp = p;
1661
8.23k
    return 0;
1662
8.44k
}
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.24k
{
1670
5.24k
    const uint8_t *p;
1671
5.24k
    int capture_index;
1672
5.24k
    char name[TMP_BUF_SIZE];
1673
1674
5.24k
    capture_index = 1;
1675
5.24k
    *phas_named_captures = 0;
1676
954k
    for (p = s->buf_start; p < s->buf_end; p++) {
1677
952k
        switch (*p) {
1678
117k
        case '(':
1679
117k
            if (p[1] == '?') {
1680
50.4k
                if (p[2] == '<' && p[3] != '=' && p[3] != '!') {
1681
16.2k
                    *phas_named_captures = 1;
1682
                    /* potential named capture */
1683
16.2k
                    if (capture_name) {
1684
15.4k
                        p += 3;
1685
15.4k
                        if (re_parse_group_name(name, sizeof(name), &p) == 0) {
1686
3.51k
                            if (!strcmp(name, capture_name))
1687
2.79k
                                return capture_index;
1688
3.51k
                        }
1689
15.4k
                    }
1690
13.4k
                    capture_index++;
1691
13.4k
                    if (capture_index >= CAPTURE_COUNT_MAX)
1692
4
                        goto done;
1693
13.4k
                }
1694
66.8k
            } else {
1695
66.8k
                capture_index++;
1696
66.8k
                if (capture_index >= CAPTURE_COUNT_MAX)
1697
208
                    goto done;
1698
66.8k
            }
1699
114k
            break;
1700
149k
        case '\\':
1701
149k
            p++;
1702
149k
            break;
1703
1.18k
        case '[':
1704
7.25k
            for (p += 1 + (*p == ']'); p < s->buf_end && *p != ']'; p++) {
1705
6.06k
                if (*p == '\\')
1706
3.56k
                    p++;
1707
6.06k
            }
1708
1.18k
            break;
1709
952k
        }
1710
952k
    }
1711
2.45k
 done:
1712
2.45k
    if (capture_name)
1713
1.31k
        return -1;
1714
1.13k
    else
1715
1.13k
        return capture_index;
1716
2.45k
}
1717
1718
static int re_count_captures(REParseState *s)
1719
4.70k
{
1720
4.70k
    if (s->total_capture_count < 0) {
1721
1.13k
        s->total_capture_count = re_parse_captures(s, &s->has_named_captures,
1722
1.13k
                                                   NULL);
1723
1.13k
    }
1724
4.70k
    return s->total_capture_count;
1725
4.70k
}
1726
1727
static BOOL re_has_named_captures(REParseState *s)
1728
2.68k
{
1729
2.68k
    if (s->has_named_captures < 0)
1730
747
        re_count_captures(s);
1731
2.68k
    return s->has_named_captures;
1732
2.68k
}
1733
1734
static int find_group_name(REParseState *s, const char *name)
1735
4.71k
{
1736
4.71k
    const char *p, *buf_end;
1737
4.71k
    size_t len, name_len;
1738
4.71k
    int capture_index;
1739
1740
4.71k
    p = (char *)s->group_names.buf;
1741
4.71k
    if (!p) return -1;
1742
992
    buf_end = (char *)s->group_names.buf + s->group_names.size;
1743
992
    name_len = strlen(name);
1744
992
    capture_index = 1;
1745
9.50k
    while (p < buf_end) {
1746
8.71k
        len = strlen(p);
1747
8.71k
        if (len == name_len && memcmp(name, p, name_len) == 0)
1748
207
            return capture_index;
1749
8.51k
        p += len + 1;
1750
8.51k
        capture_index++;
1751
8.51k
    }
1752
785
    return -1;
1753
992
}
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.57k
{
1759
3.57k
    const uint8_t *p = *pp;
1760
3.57k
    int mask = 0;
1761
3.57k
    int val;
1762
1763
6.41k
    for(;;) {
1764
6.41k
        if (*p == 'i') {
1765
1.35k
            val = LRE_FLAG_IGNORECASE;
1766
5.06k
        } else if (*p == 'm') {
1767
628
            val = LRE_FLAG_MULTILINE;
1768
4.43k
        } else if (*p == 's') {
1769
876
            val = LRE_FLAG_DOTALL;
1770
3.56k
        } else {
1771
3.56k
            break;
1772
3.56k
        }
1773
2.85k
        if (mask & val)
1774
8
            return re_parse_error(s, "duplicate modifier: '%c'", *p);
1775
2.84k
        mask |= val;
1776
2.84k
        p++;
1777
2.84k
    }
1778
3.56k
    *pp = p;
1779
3.56k
    return mask;
1780
3.57k
}
1781
1782
static BOOL update_modifier(BOOL val, int add_mask, int remove_mask,
1783
                            int mask)
1784
7.66k
{
1785
7.66k
    if (add_mask & mask)
1786
1.81k
        val = TRUE;
1787
7.66k
    if (remove_mask & mask)
1788
976
        val = FALSE;
1789
7.66k
    return val;
1790
7.66k
}
1791
1792
static int re_parse_term(REParseState *s, BOOL is_backward_dir)
1793
422k
{
1794
422k
    const uint8_t *p;
1795
422k
    int c, last_atom_start, quant_min, quant_max, last_capture_count;
1796
422k
    BOOL greedy, add_zero_advance_check, is_neg, is_backward_lookahead;
1797
422k
    REStringList cr_s, *cr = &cr_s;
1798
1799
422k
    last_atom_start = -1;
1800
422k
    last_capture_count = 0;
1801
422k
    p = s->buf_ptr;
1802
422k
    c = *p;
1803
422k
    switch(c) {
1804
1.05k
    case '^':
1805
1.05k
        p++;
1806
1.05k
        re_emit_op(s, s->multi_line ? REOP_line_start_m : REOP_line_start);
1807
1.05k
        break;
1808
912
    case '$':
1809
912
        p++;
1810
912
        re_emit_op(s, s->multi_line ? REOP_line_end_m : REOP_line_end);
1811
912
        break;
1812
2.40k
    case '.':
1813
2.40k
        p++;
1814
2.40k
        last_atom_start = s->byte_code.size;
1815
2.40k
        last_capture_count = s->capture_count;
1816
2.40k
        if (is_backward_dir)
1817
336
            re_emit_op(s, REOP_prev);
1818
2.40k
        re_emit_op(s, s->dotall ? REOP_any : REOP_dot);
1819
2.40k
        if (is_backward_dir)
1820
336
            re_emit_op(s, REOP_prev);
1821
2.40k
        break;
1822
6.03k
    case '{':
1823
6.03k
        if (s->is_unicode) {
1824
0
            return re_parse_error(s, "syntax error");
1825
6.03k
        } else if (!is_digit(p[1])) {
1826
            /* Annex B: we accept '{' not followed by digits as a
1827
               normal atom */
1828
3.82k
            goto parse_class_atom;
1829
3.82k
        } else {
1830
2.21k
            const uint8_t *p1 = p + 1;
1831
            /* Annex B: error if it is like a repetition count */
1832
2.21k
            parse_digits(&p1, TRUE);
1833
2.21k
            if (*p1 == ',') {
1834
1.38k
                p1++;
1835
1.38k
                if (is_digit(*p1)) {
1836
767
                    parse_digits(&p1, TRUE);
1837
767
                }
1838
1.38k
            }
1839
2.21k
            if (*p1 != '}') {
1840
2.21k
                goto parse_class_atom;
1841
2.21k
            }
1842
2.21k
        }
1843
        /* fall thru */
1844
11
    case '*':
1845
22
    case '+':
1846
28
    case '?':
1847
28
        return re_parse_error(s, "nothing to repeat");
1848
260k
    case '(':
1849
260k
        if (p[1] == '?') {
1850
223k
            if (p[2] == ':') {
1851
212k
                p += 3;
1852
212k
                last_atom_start = s->byte_code.size;
1853
212k
                last_capture_count = s->capture_count;
1854
212k
                s->buf_ptr = p;
1855
212k
                if (re_parse_disjunction(s, is_backward_dir))
1856
211k
                    return -1;
1857
909
                p = s->buf_ptr;
1858
909
                if (re_parse_expect(s, &p, ')'))
1859
21
                    return -1;
1860
10.6k
            } else if (p[2] == 'i' || p[2] == 'm' || p[2] == 's' || p[2] == '-') {
1861
2.60k
                BOOL saved_ignore_case, saved_multi_line, saved_dotall;
1862
2.60k
                int add_mask, remove_mask;
1863
2.60k
                p += 2;
1864
2.60k
                remove_mask = 0;
1865
2.60k
                add_mask = re_parse_modifiers(s, &p);
1866
2.60k
                if (add_mask < 0)
1867
4
                    return -1;
1868
2.60k
                if (*p == '-') {
1869
962
                    p++;
1870
962
                    remove_mask = re_parse_modifiers(s, &p);
1871
962
                    if (remove_mask < 0)
1872
4
                        return -1;
1873
962
                }
1874
2.60k
                if ((add_mask == 0 && remove_mask == 0) ||
1875
2.60k
                    (add_mask & remove_mask) != 0) {
1876
7
                    return re_parse_error(s, "invalid modifiers");
1877
7
                }
1878
2.59k
                if (re_parse_expect(s, &p, ':'))
1879
38
                    return -1;
1880
2.55k
                saved_ignore_case = s->ignore_case;
1881
2.55k
                saved_multi_line = s->multi_line;
1882
2.55k
                saved_dotall = s->dotall;
1883
2.55k
                s->ignore_case = update_modifier(s->ignore_case, add_mask, remove_mask, LRE_FLAG_IGNORECASE);
1884
2.55k
                s->multi_line = update_modifier(s->multi_line, add_mask, remove_mask, LRE_FLAG_MULTILINE);
1885
2.55k
                s->dotall = update_modifier(s->dotall, add_mask, remove_mask, LRE_FLAG_DOTALL);
1886
1887
2.55k
                last_atom_start = s->byte_code.size;
1888
2.55k
                last_capture_count = s->capture_count;
1889
2.55k
                s->buf_ptr = p;
1890
2.55k
                if (re_parse_disjunction(s, is_backward_dir))
1891
1.80k
                    return -1;
1892
746
                p = s->buf_ptr;
1893
746
                if (re_parse_expect(s, &p, ')'))
1894
171
                    return -1;
1895
575
                s->ignore_case = saved_ignore_case;
1896
575
                s->multi_line = saved_multi_line;
1897
575
                s->dotall = saved_dotall;
1898
8.04k
            } else if ((p[2] == '=' || p[2] == '!')) {
1899
5.63k
                is_neg = (p[2] == '!');
1900
5.63k
                is_backward_lookahead = FALSE;
1901
5.63k
                p += 3;
1902
5.63k
                goto lookahead;
1903
5.63k
            } else if (p[2] == '<' &&
1904
2.41k
                       (p[3] == '=' || p[3] == '!')) {
1905
1.33k
                int pos;
1906
1.33k
                is_neg = (p[3] == '!');
1907
1.33k
                is_backward_lookahead = TRUE;
1908
1.33k
                p += 4;
1909
                /* lookahead */
1910
6.96k
            lookahead:
1911
                /* Annex B allows lookahead to be used as an atom for
1912
                   the quantifiers */
1913
6.96k
                if (!s->is_unicode && !is_backward_lookahead)  {
1914
5.63k
                    last_atom_start = s->byte_code.size;
1915
5.63k
                    last_capture_count = s->capture_count;
1916
5.63k
                }
1917
6.96k
                pos = re_emit_op_u32(s, REOP_lookahead + is_neg, 0);
1918
6.96k
                s->buf_ptr = p;
1919
6.96k
                if (re_parse_disjunction(s, is_backward_lookahead))
1920
5.87k
                    return -1;
1921
1.09k
                p = s->buf_ptr;
1922
1.09k
                if (re_parse_expect(s, &p, ')'))
1923
146
                    return -1;
1924
946
                re_emit_op(s, REOP_match);
1925
                /* jump after the 'match' after the lookahead is successful */
1926
946
                if (dbuf_error(&s->byte_code))
1927
0
                    return -1;
1928
946
                put_u32(s->byte_code.buf + pos, s->byte_code.size - (pos + 4));
1929
1.08k
            } else if (p[2] == '<') {
1930
999
                p += 3;
1931
999
                if (re_parse_group_name(s->u.tmp_buf, sizeof(s->u.tmp_buf),
1932
999
                                        &p)) {
1933
584
                    return re_parse_error(s, "invalid group name");
1934
584
                }
1935
415
                if (find_group_name(s, s->u.tmp_buf) > 0) {
1936
11
                    return re_parse_error(s, "duplicate group name");
1937
11
                }
1938
                /* group name with a trailing zero */
1939
404
                dbuf_put(&s->group_names, (uint8_t *)s->u.tmp_buf,
1940
404
                         strlen(s->u.tmp_buf) + 1);
1941
404
                s->has_named_captures = 1;
1942
404
                goto parse_capture;
1943
415
            } else {
1944
82
                return re_parse_error(s, "invalid group");
1945
82
            }
1946
223k
        } else {
1947
37.3k
            int capture_index;
1948
37.3k
            p++;
1949
            /* capture without group name */
1950
37.3k
            dbuf_putc(&s->group_names, 0);
1951
37.7k
        parse_capture:
1952
37.7k
            if (s->capture_count >= CAPTURE_COUNT_MAX)
1953
27
                return re_parse_error(s, "too many captures");
1954
37.6k
            last_atom_start = s->byte_code.size;
1955
37.6k
            last_capture_count = s->capture_count;
1956
37.6k
            capture_index = s->capture_count++;
1957
37.6k
            re_emit_op_u8(s, REOP_save_start + is_backward_dir,
1958
37.6k
                          capture_index);
1959
1960
37.6k
            s->buf_ptr = p;
1961
37.6k
            if (re_parse_disjunction(s, is_backward_dir))
1962
28.1k
                return -1;
1963
9.56k
            p = s->buf_ptr;
1964
1965
9.56k
            re_emit_op_u8(s, REOP_save_start + 1 - is_backward_dir,
1966
9.56k
                          capture_index);
1967
1968
9.56k
            if (re_parse_expect(s, &p, ')'))
1969
323
                return -1;
1970
9.56k
        }
1971
11.6k
        break;
1972
24.9k
    case '\\':
1973
24.9k
        switch(p[1]) {
1974
565
        case 'b':
1975
1.22k
        case 'B':
1976
1.22k
            if (p[1] != 'b') {
1977
655
                re_emit_op(s, s->ignore_case ? REOP_not_word_boundary_i : REOP_not_word_boundary);
1978
655
            } else {
1979
565
                re_emit_op(s, s->ignore_case ? REOP_word_boundary_i : REOP_word_boundary);
1980
565
            }
1981
1.22k
            p += 2;
1982
1.22k
            break;
1983
5.67k
        case 'k':
1984
5.67k
            {
1985
5.67k
                const uint8_t *p1;
1986
5.67k
                int dummy_res;
1987
1988
5.67k
                p1 = p;
1989
5.67k
                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
587
                    if (s->is_unicode || re_has_named_captures(s))
1994
22
                        return re_parse_error(s, "expecting group name");
1995
565
                    else
1996
565
                        goto parse_class_atom;
1997
587
                }
1998
5.08k
                p1 += 3;
1999
5.08k
                if (re_parse_group_name(s->u.tmp_buf, sizeof(s->u.tmp_buf),
2000
5.08k
                                        &p1)) {
2001
781
                    if (s->is_unicode || re_has_named_captures(s))
2002
7
                        return re_parse_error(s, "invalid group name");
2003
774
                    else
2004
774
                        goto parse_class_atom;
2005
781
                }
2006
4.30k
                c = find_group_name(s, s->u.tmp_buf);
2007
4.30k
                if (c < 0) {
2008
                    /* no capture name parsed before, try to look
2009
                       after (inefficient, but hopefully not common */
2010
4.10k
                    c = re_parse_captures(s, &dummy_res, s->u.tmp_buf);
2011
4.10k
                    if (c < 0) {
2012
1.31k
                        if (s->is_unicode || re_has_named_captures(s))
2013
264
                            return re_parse_error(s, "group name not defined");
2014
1.05k
                        else
2015
1.05k
                            goto parse_class_atom;
2016
1.31k
                    }
2017
4.10k
                }
2018
2.98k
                p = p1;
2019
2.98k
            }
2020
0
            goto emit_back_reference;
2021
1.38k
        case '0':
2022
1.38k
            p += 2;
2023
1.38k
            c = 0;
2024
1.38k
            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.38k
            } else {
2029
                /* Annex B.1.4: accept legacy octal */
2030
1.38k
                if (*p >= '0' && *p <= '7') {
2031
819
                    c = *p++ - '0';
2032
819
                    if (*p >= '0' && *p <= '7') {
2033
347
                        c = (c << 3) + *p++ - '0';
2034
347
                    }
2035
819
                }
2036
1.38k
            }
2037
1.38k
            goto normal_char;
2038
2.93k
        case '1': case '2': case '3': case '4':
2039
4.57k
        case '5': case '6': case '7': case '8':
2040
4.79k
        case '9':
2041
4.79k
            {
2042
4.79k
                const uint8_t *q = ++p;
2043
2044
4.79k
                c = parse_digits(&p, FALSE);
2045
4.79k
                if (c < 0 || (c >= s->capture_count && c >= re_count_captures(s))) {
2046
3.47k
                    if (!s->is_unicode) {
2047
                        /* Annex B.1.4: accept legacy octal */
2048
3.47k
                        p = q;
2049
3.47k
                        if (*p <= '7') {
2050
2.92k
                            c = 0;
2051
2.92k
                            if (*p <= '3')
2052
1.26k
                                c = *p++ - '0';
2053
2.92k
                            if (*p >= '0' && *p <= '7') {
2054
1.80k
                                c = (c << 3) + *p++ - '0';
2055
1.80k
                                if (*p >= '0' && *p <= '7') {
2056
538
                                    c = (c << 3) + *p++ - '0';
2057
538
                                }
2058
1.80k
                            }
2059
2.92k
                        } else {
2060
552
                            c = *p++;
2061
552
                        }
2062
3.47k
                        goto normal_char;
2063
3.47k
                    }
2064
0
                    return re_parse_error(s, "back reference out of range in regular expression");
2065
3.47k
                }
2066
4.30k
            emit_back_reference:
2067
4.30k
                last_atom_start = s->byte_code.size;
2068
4.30k
                last_capture_count = s->capture_count;
2069
                
2070
4.30k
                re_emit_op_u8(s, REOP_back_reference + 2 * is_backward_dir + s->ignore_case, c);
2071
4.30k
            }
2072
0
            break;
2073
11.9k
        default:
2074
11.9k
            goto parse_class_atom;
2075
24.9k
        }
2076
5.52k
        break;
2077
5.52k
    case '[':
2078
3.06k
        last_atom_start = s->byte_code.size;
2079
3.06k
        last_capture_count = s->capture_count;
2080
3.06k
        if (is_backward_dir)
2081
260
            re_emit_op(s, REOP_prev);
2082
3.06k
        if (re_parse_char_class(s, &p))
2083
765
            return -1;
2084
2.30k
        if (is_backward_dir)
2085
253
            re_emit_op(s, REOP_prev);
2086
2.30k
        break;
2087
496
    case ']':
2088
1.00k
    case '}':
2089
1.00k
        if (s->is_unicode)
2090
0
            return re_parse_error(s, "syntax error");
2091
1.00k
        goto parse_class_atom;
2092
122k
    default:
2093
143k
    parse_class_atom:
2094
143k
        c = get_class_atom(s, cr, &p, FALSE);
2095
143k
        if ((int)c < 0)
2096
408
            return -1;
2097
148k
    normal_char:
2098
148k
        last_atom_start = s->byte_code.size;
2099
148k
        last_capture_count = s->capture_count;
2100
148k
        if (is_backward_dir)
2101
1.31k
            re_emit_op(s, REOP_prev);
2102
148k
        if (c >= CLASS_RANGE_BASE) {
2103
1.95k
            int ret;
2104
1.95k
            ret = re_emit_string_list(s, cr);
2105
1.95k
            re_string_list_free(cr);
2106
1.95k
            if (ret)
2107
0
                return -1;
2108
146k
        } else {
2109
146k
            if (s->ignore_case)
2110
4.86k
                c = lre_canonicalize(c, s->is_unicode);
2111
146k
            re_emit_char(s, c);
2112
146k
        }
2113
148k
        if (is_backward_dir)
2114
1.31k
            re_emit_op(s, REOP_prev);
2115
148k
        break;
2116
422k
    }
2117
2118
    /* quantifier */
2119
172k
    if (last_atom_start >= 0) {
2120
168k
        c = *p;
2121
168k
        switch(c) {
2122
2.47k
        case '*':
2123
2.47k
            p++;
2124
2.47k
            quant_min = 0;
2125
2.47k
            quant_max = INT32_MAX;
2126
2.47k
            goto quantifier;
2127
4.47k
        case '+':
2128
4.47k
            p++;
2129
4.47k
            quant_min = 1;
2130
4.47k
            quant_max = INT32_MAX;
2131
4.47k
            goto quantifier;
2132
2.54k
        case '?':
2133
2.54k
            p++;
2134
2.54k
            quant_min = 0;
2135
2.54k
            quant_max = 1;
2136
2.54k
            goto quantifier;
2137
8.31k
        case '{':
2138
8.31k
            {
2139
8.31k
                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
8.31k
                if (!is_digit(p[1])) {
2143
3.68k
                    if (s->is_unicode)
2144
0
                        goto invalid_quant_count;
2145
3.68k
                    break;
2146
3.68k
                }
2147
4.62k
                p++;
2148
4.62k
                quant_min = parse_digits(&p, TRUE);
2149
4.62k
                quant_max = quant_min;
2150
4.62k
                if (*p == ',') {
2151
2.85k
                    p++;
2152
2.85k
                    if (is_digit(*p)) {
2153
2.04k
                        quant_max = parse_digits(&p, TRUE);
2154
2.04k
                        if (quant_max < quant_min) {
2155
4
                        invalid_quant_count:
2156
4
                            return re_parse_error(s, "invalid repetition count");
2157
4
                        }
2158
2.04k
                    } else {
2159
809
                        quant_max = INT32_MAX; /* infinity */
2160
809
                    }
2161
2.85k
                }
2162
4.62k
                if (*p != '}' && !s->is_unicode) {
2163
                    /* Annex B: normal atom if invalid '{' syntax */
2164
2.03k
                    p = p1;
2165
2.03k
                    break;
2166
2.03k
                }
2167
2.59k
                if (re_parse_expect(s, &p, '}'))
2168
0
                    return -1;
2169
2.59k
            }
2170
12.0k
        quantifier:
2171
12.0k
            greedy = TRUE;
2172
12.0k
            if (*p == '?') {
2173
1.38k
                p++;
2174
1.38k
                greedy = FALSE;
2175
1.38k
            }
2176
12.0k
            if (last_atom_start < 0) {
2177
0
                return re_parse_error(s, "nothing to repeat");
2178
0
            }
2179
12.0k
            if (greedy) {
2180
10.6k
                int len, pos;
2181
2182
10.6k
                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.34k
                        re_emit_op(s, REOP_match);
2190
2191
2.34k
                        if (dbuf_insert(&s->byte_code, last_atom_start, 17))
2192
0
                            goto out_of_memory;
2193
2.34k
                        pos = last_atom_start;
2194
2.34k
                        s->byte_code.buf[pos++] = REOP_simple_greedy_quant;
2195
2.34k
                        put_u32(&s->byte_code.buf[pos],
2196
2.34k
                                s->byte_code.size - last_atom_start - 17);
2197
2.34k
                        pos += 4;
2198
2.34k
                        put_u32(&s->byte_code.buf[pos], quant_min);
2199
2.34k
                        pos += 4;
2200
2.34k
                        put_u32(&s->byte_code.buf[pos], quant_max);
2201
2.34k
                        pos += 4;
2202
2.34k
                        put_u32(&s->byte_code.buf[pos], len);
2203
2.34k
                        pos += 4;
2204
2.34k
                        goto done;
2205
2.34k
                    }
2206
10.4k
                }
2207
2208
8.35k
                if (dbuf_error(&s->byte_code))
2209
0
                    goto out_of_memory;
2210
8.35k
            }
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.73k
            add_zero_advance_check = re_need_check_advance(s->byte_code.buf + last_atom_start,
2216
9.73k
                                                           s->byte_code.size - last_atom_start);
2217
2218
9.73k
            {
2219
9.73k
                int len, pos;
2220
9.73k
                len = s->byte_code.size - last_atom_start;
2221
9.73k
                if (quant_min == 0) {
2222
                    /* need to reset the capture in case the atom is
2223
                       not executed */
2224
4.24k
                    if (last_capture_count != s->capture_count) {
2225
2.75k
                        if (dbuf_insert(&s->byte_code, last_atom_start, 3))
2226
0
                            goto out_of_memory;
2227
2.75k
                        s->byte_code.buf[last_atom_start++] = REOP_save_reset;
2228
2.75k
                        s->byte_code.buf[last_atom_start++] = last_capture_count;
2229
2.75k
                        s->byte_code.buf[last_atom_start++] = s->capture_count - 1;
2230
2.75k
                    }
2231
4.24k
                    if (quant_max == 0) {
2232
218
                        s->byte_code.size = last_atom_start;
2233
4.03k
                    } else if (quant_max == 1 || quant_max == INT32_MAX) {
2234
3.35k
                        BOOL has_goto = (quant_max == INT32_MAX);
2235
3.35k
                        if (dbuf_insert(&s->byte_code, last_atom_start, 5 + add_zero_advance_check))
2236
0
                            goto out_of_memory;
2237
3.35k
                        s->byte_code.buf[last_atom_start] = REOP_split_goto_first +
2238
3.35k
                            greedy;
2239
3.35k
                        put_u32(s->byte_code.buf + last_atom_start + 1,
2240
3.35k
                                len + 5 * has_goto + add_zero_advance_check * 2);
2241
3.35k
                        if (add_zero_advance_check) {
2242
2.89k
                            s->byte_code.buf[last_atom_start + 1 + 4] = REOP_push_char_pos;
2243
2.89k
                            re_emit_op(s, REOP_check_advance);
2244
2.89k
                        }
2245
3.35k
                        if (has_goto)
2246
1.64k
                            re_emit_goto(s, REOP_goto, last_atom_start);
2247
3.35k
                    } else {
2248
680
                        if (dbuf_insert(&s->byte_code, last_atom_start, 10 + add_zero_advance_check))
2249
0
                            goto out_of_memory;
2250
680
                        pos = last_atom_start;
2251
680
                        s->byte_code.buf[pos++] = REOP_push_i32;
2252
680
                        put_u32(s->byte_code.buf + pos, quant_max);
2253
680
                        pos += 4;
2254
680
                        s->byte_code.buf[pos++] = REOP_split_goto_first + greedy;
2255
680
                        put_u32(s->byte_code.buf + pos, len + 5 + add_zero_advance_check * 2);
2256
680
                        pos += 4;
2257
680
                        if (add_zero_advance_check) {
2258
419
                            s->byte_code.buf[pos++] = REOP_push_char_pos;
2259
419
                            re_emit_op(s, REOP_check_advance);
2260
419
                        }
2261
680
                        re_emit_goto(s, REOP_loop, last_atom_start + 5);
2262
680
                        re_emit_op(s, REOP_drop);
2263
680
                    }
2264
5.48k
                } else if (quant_min == 1 && quant_max == INT32_MAX &&
2265
5.48k
                           !add_zero_advance_check) {
2266
467
                    re_emit_goto(s, REOP_split_next_first - greedy,
2267
467
                                 last_atom_start);
2268
5.02k
                } else {
2269
5.02k
                    if (quant_min == 1) {
2270
                        /* nothing to add */
2271
3.98k
                    } 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
5.02k
                    if (quant_max == INT32_MAX) {
2282
3.63k
                        pos = s->byte_code.size;
2283
3.63k
                        re_emit_op_u32(s, REOP_split_goto_first + greedy,
2284
3.63k
                                       len + 5 + add_zero_advance_check * 2);
2285
3.63k
                        if (add_zero_advance_check)
2286
3.44k
                            re_emit_op(s, REOP_push_char_pos);
2287
                        /* copy the atom */
2288
3.63k
                        dbuf_put_self(&s->byte_code, last_atom_start, len);
2289
3.63k
                        if (add_zero_advance_check)
2290
3.44k
                            re_emit_op(s, REOP_check_advance);
2291
3.63k
                        re_emit_goto(s, REOP_goto, pos);
2292
3.63k
                    } else if (quant_max > quant_min) {
2293
614
                        re_emit_op_u32(s, REOP_push_i32, quant_max - quant_min);
2294
614
                        pos = s->byte_code.size;
2295
614
                        re_emit_op_u32(s, REOP_split_goto_first + greedy,
2296
614
                                       len + 5 + add_zero_advance_check * 2);
2297
614
                        if (add_zero_advance_check)
2298
247
                            re_emit_op(s, REOP_push_char_pos);
2299
                        /* copy the atom */
2300
614
                        dbuf_put_self(&s->byte_code, last_atom_start, len);
2301
614
                        if (add_zero_advance_check)
2302
247
                            re_emit_op(s, REOP_check_advance);
2303
614
                        re_emit_goto(s, REOP_loop, pos);
2304
614
                        re_emit_op(s, REOP_drop);
2305
614
                    }
2306
5.02k
                }
2307
9.73k
                last_atom_start = -1;
2308
9.73k
            }
2309
0
            break;
2310
151k
        default:
2311
151k
            break;
2312
168k
        }
2313
168k
    }
2314
172k
 done:
2315
172k
    s->buf_ptr = p;
2316
172k
    return 0;
2317
0
 out_of_memory:
2318
0
    return re_parse_out_of_memory(s);
2319
172k
}
2320
2321
static int re_parse_alternative(REParseState *s, BOOL is_backward_dir)
2322
269k
{
2323
269k
    const uint8_t *p;
2324
269k
    int ret;
2325
269k
    size_t start, term_start, end, term_size;
2326
2327
269k
    start = s->byte_code.size;
2328
441k
    for(;;) {
2329
441k
        p = s->buf_ptr;
2330
441k
        if (p >= s->buf_end)
2331
4.30k
            break;
2332
437k
        if (*p == '|' || *p == ')')
2333
14.4k
            break;
2334
422k
        term_start = s->byte_code.size;
2335
422k
        ret = re_parse_term(s, is_backward_dir);
2336
422k
        if (ret)
2337
250k
            return ret;
2338
172k
        if (is_backward_dir) {
2339
            /* reverse the order of the terms (XXX: inefficient, but
2340
               speed is not really critical here) */
2341
2.48k
            end = s->byte_code.size;
2342
2.48k
            term_size = end - term_start;
2343
2.48k
            if (dbuf_realloc(&s->byte_code, end + term_size))
2344
0
                return -1;
2345
2.48k
            memmove(s->byte_code.buf + start + term_size,
2346
2.48k
                    s->byte_code.buf + start,
2347
2.48k
                    end - start);
2348
2.48k
            memcpy(s->byte_code.buf + start, s->byte_code.buf + end,
2349
2.48k
                   term_size);
2350
2.48k
        }
2351
172k
    }
2352
18.7k
    return 0;
2353
269k
}
2354
2355
static int re_parse_disjunction(REParseState *s, BOOL is_backward_dir)
2356
266k
{
2357
266k
    int start, len, pos;
2358
2359
266k
    if (lre_check_stack_overflow(s->opaque, 0))
2360
0
        return re_parse_error(s, "stack overflow");
2361
2362
266k
    start = s->byte_code.size;
2363
266k
    if (re_parse_alternative(s, is_backward_dir))
2364
250k
        return -1;
2365
18.7k
    while (*s->buf_ptr == '|') {
2366
2.77k
        s->buf_ptr++;
2367
2368
2.77k
        len = s->byte_code.size - start;
2369
2370
        /* insert a split before the first alternative */
2371
2.77k
        if (dbuf_insert(&s->byte_code, start, 5)) {
2372
0
            return re_parse_out_of_memory(s);
2373
0
        }
2374
2.77k
        s->byte_code.buf[start] = REOP_split_next_first;
2375
2.77k
        put_u32(s->byte_code.buf + start + 1, len + 5);
2376
2377
2.77k
        pos = re_emit_op_u32(s, REOP_goto, 0);
2378
2379
2.77k
        if (re_parse_alternative(s, is_backward_dir))
2380
405
            return -1;
2381
2382
        /* patch the goto */
2383
2.36k
        len = s->byte_code.size - (pos + 4);
2384
2.36k
        put_u32(s->byte_code.buf + pos, len);
2385
2.36k
    }
2386
15.9k
    return 0;
2387
16.4k
}
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.64k
{
2392
3.64k
    int stack_size, stack_size_max, pos, opcode, len;
2393
3.64k
    uint32_t val;
2394
2395
3.64k
    stack_size = 0;
2396
3.64k
    stack_size_max = 0;
2397
3.64k
    bc_buf += RE_HEADER_LEN;
2398
3.64k
    bc_buf_len -= RE_HEADER_LEN;
2399
3.64k
    pos = 0;
2400
1.03G
    while (pos < bc_buf_len) {
2401
1.03G
        opcode = bc_buf[pos];
2402
1.03G
        len = reopcode_info[opcode].size;
2403
1.03G
        assert(opcode < REOP_COUNT);
2404
1.03G
        assert((pos + len) <= bc_buf_len);
2405
1.03G
        switch(opcode) {
2406
12.4M
        case REOP_push_i32:
2407
88.7M
        case REOP_push_char_pos:
2408
88.7M
            stack_size++;
2409
88.7M
            if (stack_size > stack_size_max) {
2410
5.50k
                if (stack_size > STACK_SIZE_MAX)
2411
1
                    return -1;
2412
5.50k
                stack_size_max = stack_size;
2413
5.50k
            }
2414
88.7M
            break;
2415
88.7M
        case REOP_drop:
2416
88.7M
        case REOP_check_advance:
2417
88.7M
            assert(stack_size > 0);
2418
88.7M
            stack_size--;
2419
88.7M
            break;
2420
82.2M
        case REOP_range:
2421
82.2M
        case REOP_range_i:
2422
82.2M
            val = get_u16(bc_buf + pos + 1);
2423
82.2M
            len += val * 4;
2424
82.2M
            break;
2425
1.64k
        case REOP_range32:
2426
4.47k
        case REOP_range32_i:
2427
4.47k
            val = get_u16(bc_buf + pos + 1);
2428
4.47k
            len += val * 8;
2429
4.47k
            break;
2430
1.03G
        }
2431
1.03G
        pos += len;
2432
1.03G
    }
2433
3.64k
    return stack_size_max;
2434
3.64k
}
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
6.60k
{
2444
6.60k
    REParseState s_s, *s = &s_s;
2445
6.60k
    int stack_size;
2446
6.60k
    BOOL is_sticky;
2447
2448
6.60k
    memset(s, 0, sizeof(*s));
2449
6.60k
    s->opaque = opaque;
2450
6.60k
    s->buf_ptr = (const uint8_t *)buf;
2451
6.60k
    s->buf_end = s->buf_ptr + buf_len;
2452
6.60k
    s->buf_start = s->buf_ptr;
2453
6.60k
    s->re_flags = re_flags;
2454
6.60k
    s->is_unicode = ((re_flags & (LRE_FLAG_UNICODE | LRE_FLAG_UNICODE_SETS)) != 0);
2455
6.60k
    is_sticky = ((re_flags & LRE_FLAG_STICKY) != 0);
2456
6.60k
    s->ignore_case = ((re_flags & LRE_FLAG_IGNORECASE) != 0);
2457
6.60k
    s->multi_line = ((re_flags & LRE_FLAG_MULTILINE) != 0);
2458
6.60k
    s->dotall = ((re_flags & LRE_FLAG_DOTALL) != 0);
2459
6.60k
    s->unicode_sets = ((re_flags & LRE_FLAG_UNICODE_SETS) != 0);
2460
6.60k
    s->capture_count = 1;
2461
6.60k
    s->total_capture_count = -1;
2462
6.60k
    s->has_named_captures = -1;
2463
2464
6.60k
    dbuf_init2(&s->byte_code, opaque, lre_realloc);
2465
6.60k
    dbuf_init2(&s->group_names, opaque, lre_realloc);
2466
2467
6.60k
    dbuf_put_u16(&s->byte_code, re_flags); /* first element is the flags */
2468
6.60k
    dbuf_putc(&s->byte_code, 0); /* second element is the number of captures */
2469
6.60k
    dbuf_putc(&s->byte_code, 0); /* stack size */
2470
6.60k
    dbuf_put_u32(&s->byte_code, 0); /* bytecode length */
2471
2472
6.60k
    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
6.60k
        re_emit_op_u32(s, REOP_split_goto_first, 1 + 5);
2478
6.60k
        re_emit_op(s, REOP_any);
2479
6.60k
        re_emit_op_u32(s, REOP_goto, -(5 + 1 + 5));
2480
6.60k
    }
2481
6.60k
    re_emit_op_u8(s, REOP_save_start, 0);
2482
2483
6.60k
    if (re_parse_disjunction(s, FALSE)) {
2484
2.96k
    error:
2485
2.96k
        dbuf_free(&s->byte_code);
2486
2.96k
        dbuf_free(&s->group_names);
2487
2.96k
        pstrcpy(error_msg, error_msg_size, s->u.error_msg);
2488
2.96k
        *plen = 0;
2489
2.96k
        return NULL;
2490
2.91k
    }
2491
2492
3.68k
    re_emit_op_u8(s, REOP_save_end, 0);
2493
2494
3.68k
    re_emit_op(s, REOP_match);
2495
2496
3.68k
    if (*s->buf_ptr != '\0') {
2497
44
        re_parse_error(s, "extraneous characters at the end");
2498
44
        goto error;
2499
44
    }
2500
2501
3.64k
    if (dbuf_error(&s->byte_code)) {
2502
0
        re_parse_out_of_memory(s);
2503
0
        goto error;
2504
0
    }
2505
2506
3.64k
    stack_size = compute_stack_size(s->byte_code.buf, s->byte_code.size);
2507
3.64k
    if (stack_size < 0) {
2508
1
        re_parse_error(s, "too many imbricated quantifiers");
2509
1
        goto error;
2510
1
    }
2511
2512
3.64k
    s->byte_code.buf[RE_HEADER_CAPTURE_COUNT] = s->capture_count;
2513
3.64k
    s->byte_code.buf[RE_HEADER_STACK_SIZE] = stack_size;
2514
3.64k
    put_u32(s->byte_code.buf + RE_HEADER_BYTECODE_LEN,
2515
3.64k
            s->byte_code.size - RE_HEADER_LEN);
2516
2517
    /* add the named groups if needed */
2518
3.64k
    if (s->group_names.size > (s->capture_count - 1)) {
2519
17
        dbuf_put(&s->byte_code, s->group_names.buf, s->group_names.size);
2520
17
        put_u16(s->byte_code.buf + RE_HEADER_FLAGS,
2521
17
                lre_get_flags(s->byte_code.buf) | LRE_FLAG_NAMED_GROUPS);
2522
17
    }
2523
3.64k
    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.64k
    error_msg[0] = '\0';
2530
3.64k
    *plen = s->byte_code.size;
2531
3.64k
    return s->byte_code.buf;
2532
3.64k
}
2533
2534
static BOOL is_line_terminator(uint32_t c)
2535
236k
{
2536
236k
    return (c == '\n' || c == '\r' || c == CP_LS || c == CP_PS);
2537
236k
}
2538
2539
static BOOL is_word_char(uint32_t c)
2540
24.1k
{
2541
24.1k
    return ((c >= '0' && c <= '9') ||
2542
24.1k
            (c >= 'a' && c <= 'z') ||
2543
24.1k
            (c >= 'A' && c <= 'Z') ||
2544
24.1k
            (c == '_'));
2545
24.1k
}
2546
2547
#define GET_CHAR(c, cptr, cbuf_end, cbuf_type)                          \
2548
458k
    do {                                                                \
2549
458k
        if (cbuf_type == 0) {                                           \
2550
458k
            c = *cptr++;                                                \
2551
458k
        } else {                                                        \
2552
0
            const uint16_t *_p = (const uint16_t *)cptr;                \
2553
0
            const uint16_t *_end = (const uint16_t *)cbuf_end;          \
2554
0
            c = *_p++;                                                  \
2555
0
            if (is_hi_surrogate(c) && cbuf_type == 2) {                 \
2556
0
                if (_p < _end && is_lo_surrogate(*_p)) {                \
2557
0
                    c = from_surrogate(c, *_p++);                       \
2558
0
                }                                                       \
2559
0
            }                                                           \
2560
0
            cptr = (const void *)_p;                                    \
2561
0
        }                                                               \
2562
458k
    } while (0)
2563
2564
#define PEEK_CHAR(c, cptr, cbuf_end, cbuf_type)                         \
2565
14.4k
    do {                                                                \
2566
14.4k
        if (cbuf_type == 0) {                                           \
2567
14.4k
            c = cptr[0];                                                \
2568
14.4k
        } 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
14.4k
    } 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
10.3M
    do {                                                                \
2598
10.3M
        if (cbuf_type == 0) {                                           \
2599
10.3M
            cptr--;                                                     \
2600
10.3M
            c = cptr[0];                                                \
2601
10.3M
        } 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
10.3M
    } while (0)
2613
2614
#define PREV_CHAR(cptr, cbuf_start, cbuf_type)                          \
2615
118k
    do {                                                                \
2616
118k
        if (cbuf_type == 0) {                                           \
2617
118k
            cptr--;                                                     \
2618
118k
        } 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
118k
    } 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.16M
{
2671
1.16M
    REExecState *rs;
2672
1.16M
    uint8_t *new_stack;
2673
1.16M
    size_t new_size, i, n;
2674
1.16M
    StackInt *stack_buf;
2675
2676
1.16M
    if (unlikely((s->state_stack_len + 1) > s->state_stack_size)) {
2677
        /* reallocate the stack */
2678
4.99k
        new_size = s->state_stack_size * 3 / 2;
2679
4.99k
        if (new_size < 8)
2680
3.64k
            new_size = 8;
2681
4.99k
        new_stack = lre_realloc(s->opaque, s->state_stack, new_size * s->state_size);
2682
4.99k
        if (!new_stack)
2683
0
            return -1;
2684
4.99k
        s->state_stack_size = new_size;
2685
4.99k
        s->state_stack = new_stack;
2686
4.99k
    }
2687
1.16M
    rs = (REExecState *)(s->state_stack + s->state_stack_len * s->state_size);
2688
1.16M
    s->state_stack_len++;
2689
1.16M
    rs->type = type;
2690
1.16M
    rs->count = count;
2691
1.16M
    rs->stack_len = stack_len;
2692
1.16M
    rs->cptr = cptr;
2693
1.16M
    rs->pc = pc;
2694
1.16M
    n = 2 * s->capture_count;
2695
30.9M
    for(i = 0; i < n; i++)
2696
29.7M
        rs->buf[i] = capture[i];
2697
1.16M
    stack_buf = (StackInt *)(rs->buf + n);
2698
7.67M
    for(i = 0; i < stack_len; i++)
2699
6.51M
        stack_buf[i] = stack[i];
2700
1.16M
    return 0;
2701
1.16M
}
2702
2703
static int lre_poll_timeout(REExecContext *s)
2704
2.65M
{
2705
2.65M
    if (unlikely(--s->interrupt_counter <= 0)) {
2706
234
        s->interrupt_counter = INTERRUPT_COUNTER_INIT;
2707
234
        if (lre_check_timeout(s->opaque))
2708
134
            return LRE_RET_TIMEOUT;
2709
234
    }
2710
2.65M
    return 0;
2711
2.65M
}
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
335k
{
2719
335k
    int opcode, ret;
2720
335k
    int cbuf_type;
2721
335k
    uint32_t val, c;
2722
335k
    const uint8_t *cbuf_end;
2723
2724
335k
    cbuf_type = s->cbuf_type;
2725
335k
    cbuf_end = s->cbuf_end;
2726
2727
8.89M
    for(;;) {
2728
        //        printf("top=%p: pc=%d\n", th_list.top, (int)(pc - (bc_buf + RE_HEADER_LEN)));
2729
8.89M
        opcode = *pc++;
2730
8.89M
        switch(opcode) {
2731
295k
        case REOP_match:
2732
295k
            {
2733
295k
                REExecState *rs;
2734
295k
                if (no_recurse)
2735
211k
                    return (intptr_t)cptr;
2736
83.4k
                ret = 1;
2737
83.4k
                goto recurse;
2738
1.06M
            no_match:
2739
1.06M
                if (no_recurse)
2740
120k
                    return 0;
2741
939k
                ret = 0;
2742
1.02M
            recurse:
2743
1.07M
                for(;;) {
2744
1.07M
                    if (lre_poll_timeout(s))
2745
63
                        return LRE_RET_TIMEOUT;
2746
1.07M
                    if (s->state_stack_len == 0)
2747
3.50k
                        return ret;
2748
1.06M
                    rs = (REExecState *)(s->state_stack +
2749
1.06M
                                         (s->state_stack_len - 1) * s->state_size);
2750
1.06M
                    if (rs->type == RE_EXEC_STATE_SPLIT) {
2751
858k
                        if (!ret) {
2752
836k
                        pop_state:
2753
836k
                            memcpy(capture, rs->buf,
2754
836k
                                   sizeof(capture[0]) * 2 * s->capture_count);
2755
905k
                        pop_state1:
2756
905k
                            pc = rs->pc;
2757
905k
                            cptr = rs->cptr;
2758
905k
                            stack_len = rs->stack_len;
2759
905k
                            memcpy(stack, rs->buf + 2 * s->capture_count,
2760
905k
                                   stack_len * sizeof(stack[0]));
2761
905k
                            s->state_stack_len--;
2762
905k
                            break;
2763
836k
                        }
2764
858k
                    } else if (rs->type == RE_EXEC_STATE_GREEDY_QUANT) {
2765
115k
                        if (!ret) {
2766
114k
                            uint32_t char_count, i;
2767
114k
                            memcpy(capture, rs->buf,
2768
114k
                                   sizeof(capture[0]) * 2 * s->capture_count);
2769
114k
                            stack_len = rs->stack_len;
2770
114k
                            memcpy(stack, rs->buf + 2 * s->capture_count,
2771
114k
                                   stack_len * sizeof(stack[0]));
2772
114k
                            pc = rs->pc;
2773
114k
                            cptr = rs->cptr;
2774
                            /* go backward */
2775
114k
                            char_count = get_u32(pc + 12);
2776
231k
                            for(i = 0; i < char_count; i++) {
2777
116k
                                PREV_CHAR(cptr, s->cbuf, cbuf_type);
2778
116k
                            }
2779
114k
                            pc = (pc + 16) + (int)get_u32(pc);
2780
114k
                            rs->cptr = cptr;
2781
114k
                            rs->count--;
2782
114k
                            if (rs->count == 0) {
2783
42.5k
                                s->state_stack_len--;
2784
42.5k
                            }
2785
114k
                            break;
2786
114k
                        }
2787
115k
                    } else {
2788
94.4k
                        ret = ((rs->type == RE_EXEC_STATE_LOOKAHEAD && ret) ||
2789
94.4k
                               (rs->type == RE_EXEC_STATE_NEGATIVE_LOOKAHEAD && !ret));
2790
94.4k
                        if (ret) {
2791
                            /* keep the capture in case of positive lookahead */
2792
76.6k
                            if (rs->type == RE_EXEC_STATE_LOOKAHEAD)
2793
68.3k
                                goto pop_state1;
2794
8.26k
                            else
2795
8.26k
                                goto pop_state;
2796
76.6k
                        }
2797
94.4k
                    }
2798
48.2k
                    s->state_stack_len--;
2799
48.2k
                }
2800
1.02M
            }
2801
1.01M
            break;
2802
1.01M
        case REOP_char32:
2803
286
        case REOP_char32_i:
2804
286
            val = get_u32(pc);
2805
286
            pc += 4;
2806
286
            goto test_char;
2807
564k
        case REOP_char:
2808
569k
        case REOP_char_i:
2809
569k
            val = get_u16(pc);
2810
569k
            pc += 2;
2811
569k
        test_char:
2812
569k
            if (cptr >= cbuf_end)
2813
377k
                goto no_match;
2814
192k
            GET_CHAR(c, cptr, cbuf_end, cbuf_type);
2815
192k
            if (opcode == REOP_char_i || opcode == REOP_char32_i) {
2816
4.04k
                c = lre_canonicalize(c, s->is_unicode);
2817
4.04k
            }
2818
192k
            if (val != c)
2819
188k
                goto no_match;
2820
3.99k
            break;
2821
76.3k
        case REOP_split_goto_first:
2822
1.01M
        case REOP_split_next_first:
2823
1.01M
            {
2824
1.01M
                const uint8_t *pc1;
2825
2826
1.01M
                val = get_u32(pc);
2827
1.01M
                pc += 4;
2828
1.01M
                if (opcode == REOP_split_next_first) {
2829
937k
                    pc1 = pc + (int)val;
2830
937k
                } else {
2831
76.3k
                    pc1 = pc;
2832
76.3k
                    pc = pc + (int)val;
2833
76.3k
                }
2834
1.01M
                ret = push_state(s, capture, stack, stack_len,
2835
1.01M
                                 pc1, cptr, RE_EXEC_STATE_SPLIT, 0);
2836
1.01M
                if (ret < 0)
2837
0
                    return LRE_RET_MEMORY_ERROR;
2838
1.01M
                break;
2839
1.01M
            }
2840
1.01M
        case REOP_lookahead:
2841
94.4k
        case REOP_negative_lookahead:
2842
94.4k
            val = get_u32(pc);
2843
94.4k
            pc += 4;
2844
94.4k
            ret = push_state(s, capture, stack, stack_len,
2845
94.4k
                             pc + (int)val, cptr,
2846
94.4k
                             RE_EXEC_STATE_LOOKAHEAD + opcode - REOP_lookahead,
2847
94.4k
                             0);
2848
94.4k
            if (ret < 0)
2849
0
                return LRE_RET_MEMORY_ERROR;
2850
94.4k
            break;
2851
2852
601k
        case REOP_goto:
2853
601k
            val = get_u32(pc);
2854
601k
            pc += 4 + (int)val;
2855
601k
            if (lre_poll_timeout(s))
2856
11
                return LRE_RET_TIMEOUT;
2857
601k
            break;
2858
601k
        case REOP_line_start:
2859
22.7k
        case REOP_line_start_m:
2860
22.7k
            if (cptr == s->cbuf)
2861
21.1k
                break;
2862
1.63k
            if (opcode == REOP_line_start)
2863
228
                goto no_match;
2864
1.40k
            PEEK_PREV_CHAR(c, cptr, s->cbuf, cbuf_type);
2865
1.40k
            if (!is_line_terminator(c))
2866
883
                goto no_match;
2867
523
            break;
2868
1.45k
        case REOP_line_end:
2869
3.37k
        case REOP_line_end_m:
2870
3.37k
            if (cptr == cbuf_end)
2871
1.52k
                break;
2872
1.85k
            if (opcode == REOP_line_end)
2873
204
                goto no_match;
2874
1.65k
            PEEK_CHAR(c, cptr, cbuf_end, cbuf_type);
2875
1.65k
            if (!is_line_terminator(c))
2876
1.24k
                goto no_match;
2877
412
            break;
2878
293k
        case REOP_dot:
2879
293k
            if (cptr == cbuf_end)
2880
59.9k
                goto no_match;
2881
233k
            GET_CHAR(c, cptr, cbuf_end, cbuf_type);
2882
233k
            if (is_line_terminator(c))
2883
4.12k
                goto no_match;
2884
229k
            break;
2885
229k
        case REOP_any:
2886
17.4k
            if (cptr == cbuf_end)
2887
3.67k
                goto no_match;
2888
13.7k
            GET_CHAR(c, cptr, cbuf_end, cbuf_type);
2889
13.7k
            break;
2890
1.22M
        case REOP_save_start:
2891
2.99M
        case REOP_save_end:
2892
2.99M
            val = *pc++;
2893
2.99M
            assert(val < s->capture_count);
2894
2.99M
            capture[2 * val + opcode - REOP_save_start] = (uint8_t *)cptr;
2895
2.99M
            break;
2896
236k
        case REOP_save_reset:
2897
236k
            {
2898
236k
                uint32_t val2;
2899
236k
                val = pc[0];
2900
236k
                val2 = pc[1];
2901
236k
                pc += 2;
2902
236k
                assert(val2 < s->capture_count);
2903
2.32M
                while (val <= val2) {
2904
2.08M
                    capture[2 * val] = NULL;
2905
2.08M
                    capture[2 * val + 1] = NULL;
2906
2.08M
                    val++;
2907
2.08M
                }
2908
236k
            }
2909
0
            break;
2910
26.2k
        case REOP_push_i32:
2911
26.2k
            val = get_u32(pc);
2912
26.2k
            pc += 4;
2913
26.2k
            stack[stack_len++] = val;
2914
26.2k
            break;
2915
395k
        case REOP_drop:
2916
395k
            stack_len--;
2917
395k
            break;
2918
1.02M
        case REOP_loop:
2919
1.02M
            val = get_u32(pc);
2920
1.02M
            pc += 4;
2921
1.02M
            if (--stack[stack_len - 1] != 0) {
2922
647k
                pc += (int)val;
2923
647k
                if (lre_poll_timeout(s))
2924
36
                    return LRE_RET_TIMEOUT;
2925
647k
            }
2926
1.02M
            break;
2927
1.02M
        case REOP_push_char_pos:
2928
444k
            stack[stack_len++] = (uintptr_t)cptr;
2929
444k
            break;
2930
602k
        case REOP_check_advance:
2931
602k
            if (stack[--stack_len] == (uintptr_t)cptr)
2932
362k
                goto no_match;
2933
239k
            break;
2934
239k
        case REOP_word_boundary:
2935
3.16k
        case REOP_word_boundary_i:
2936
17.3k
        case REOP_not_word_boundary:
2937
18.0k
        case REOP_not_word_boundary_i:
2938
18.0k
            {
2939
18.0k
                BOOL v1, v2;
2940
18.0k
                int ignore_case = (opcode == REOP_word_boundary_i || opcode == REOP_not_word_boundary_i);
2941
18.0k
                BOOL is_boundary = (opcode == REOP_word_boundary || opcode == REOP_word_boundary_i);
2942
                /* char before */
2943
18.0k
                if (cptr == s->cbuf) {
2944
6.80k
                    v1 = FALSE;
2945
11.2k
                } else {
2946
11.2k
                    PEEK_PREV_CHAR(c, cptr, s->cbuf, cbuf_type);
2947
11.2k
                    if (ignore_case)
2948
1.30k
                        c = lre_canonicalize(c, s->is_unicode);
2949
11.2k
                    v1 = is_word_char(c);
2950
11.2k
                }
2951
                /* current char */
2952
18.0k
                if (cptr >= cbuf_end) {
2953
5.25k
                    v2 = FALSE;
2954
12.8k
                } else {
2955
12.8k
                    PEEK_CHAR(c, cptr, cbuf_end, cbuf_type);
2956
12.8k
                    if (ignore_case)
2957
1.18k
                        c = lre_canonicalize(c, s->is_unicode);
2958
12.8k
                    v2 = is_word_char(c);
2959
12.8k
                }
2960
18.0k
                if (v1 ^ v2 ^ is_boundary)
2961
5.77k
                    goto no_match;
2962
18.0k
            }
2963
12.3k
            break;
2964
12.3k
        case REOP_back_reference:
2965
13.0k
        case REOP_back_reference_i:
2966
41.0k
        case REOP_backward_back_reference:
2967
41.3k
        case REOP_backward_back_reference_i:
2968
41.3k
            {
2969
41.3k
                const uint8_t *cptr1, *cptr1_end, *cptr1_start;
2970
41.3k
                uint32_t c1, c2;
2971
2972
41.3k
                val = *pc++;
2973
41.3k
                if (val >= s->capture_count)
2974
238
                    goto no_match;
2975
41.0k
                cptr1_start = capture[2 * val];
2976
41.0k
                cptr1_end = capture[2 * val + 1];
2977
41.0k
                if (!cptr1_start || !cptr1_end)
2978
4.27k
                    break;
2979
36.8k
                if (opcode == REOP_back_reference ||
2980
36.8k
                    opcode == REOP_back_reference_i) {
2981
8.49k
                    cptr1 = cptr1_start;
2982
9.46k
                    while (cptr1 < cptr1_end) {
2983
8.60k
                        if (cptr >= cbuf_end)
2984
5.00k
                            goto no_match;
2985
3.59k
                        GET_CHAR(c1, cptr1, cptr1_end, cbuf_type);
2986
3.59k
                        GET_CHAR(c2, cptr, cbuf_end, cbuf_type);
2987
3.59k
                        if (opcode == REOP_back_reference_i) {
2988
2.28k
                            c1 = lre_canonicalize(c1, s->is_unicode);
2989
2.28k
                            c2 = lre_canonicalize(c2, s->is_unicode);
2990
2.28k
                        }
2991
3.59k
                        if (c1 != c2)
2992
2.62k
                            goto no_match;
2993
3.59k
                    }
2994
28.3k
                } else {
2995
28.3k
                    cptr1 = cptr1_end;
2996
5.18M
                    while (cptr1 > cptr1_start) {
2997
5.17M
                        if (cptr == s->cbuf)
2998
2.64k
                            goto no_match;
2999
5.17M
                        GET_PREV_CHAR(c1, cptr1, cptr1_start, cbuf_type);
3000
5.17M
                        GET_PREV_CHAR(c2, cptr, s->cbuf, cbuf_type);
3001
5.17M
                        if (opcode == REOP_backward_back_reference_i) {
3002
522
                            c1 = lre_canonicalize(c1, s->is_unicode);
3003
522
                            c2 = lre_canonicalize(c2, s->is_unicode);
3004
522
                        }
3005
5.17M
                        if (c1 != c2)
3006
10.1k
                            goto no_match;
3007
5.17M
                    }
3008
28.3k
                }
3009
36.8k
            }
3010
16.3k
            break;
3011
34.9k
        case REOP_range:
3012
35.6k
        case REOP_range_i:
3013
35.6k
            {
3014
35.6k
                int n;
3015
35.6k
                uint32_t low, high, idx_min, idx_max, idx;
3016
3017
35.6k
                n = get_u16(pc); /* n must be >= 1 */
3018
35.6k
                pc += 2;
3019
35.6k
                if (cptr >= cbuf_end)
3020
25.0k
                    goto no_match;
3021
10.6k
                GET_CHAR(c, cptr, cbuf_end, cbuf_type);
3022
10.6k
                if (opcode == REOP_range_i) {
3023
410
                    c = lre_canonicalize(c, s->is_unicode);
3024
410
                }
3025
10.6k
                idx_min = 0;
3026
10.6k
                low = get_u16(pc + 0 * 4);
3027
10.6k
                if (c < low)
3028
404
                    goto no_match;
3029
10.2k
                idx_max = n - 1;
3030
10.2k
                high = get_u16(pc + idx_max * 4 + 2);
3031
                /* 0xffff in for last value means +infinity */
3032
10.2k
                if (unlikely(c >= 0xffff) && high == 0xffff)
3033
0
                    goto range_match;
3034
10.2k
                if (c > high)
3035
387
                    goto no_match;
3036
24.4k
                while (idx_min <= idx_max) {
3037
24.2k
                    idx = (idx_min + idx_max) / 2;
3038
24.2k
                    low = get_u16(pc + idx * 4);
3039
24.2k
                    high = get_u16(pc + idx * 4 + 2);
3040
24.2k
                    if (c < low)
3041
6.51k
                        idx_max = idx - 1;
3042
17.7k
                    else if (c > high)
3043
8.16k
                        idx_min = idx + 1;
3044
9.54k
                    else
3045
9.54k
                        goto range_match;
3046
24.2k
                }
3047
276
                goto no_match;
3048
9.54k
            range_match:
3049
9.54k
                pc += 4 * n;
3050
9.54k
            }
3051
0
            break;
3052
1.76k
        case REOP_range32:
3053
2.24k
        case REOP_range32_i:
3054
2.24k
            {
3055
2.24k
                int n;
3056
2.24k
                uint32_t low, high, idx_min, idx_max, idx;
3057
3058
2.24k
                n = get_u16(pc); /* n must be >= 1 */
3059
2.24k
                pc += 2;
3060
2.24k
                if (cptr >= cbuf_end)
3061
811
                    goto no_match;
3062
1.42k
                GET_CHAR(c, cptr, cbuf_end, cbuf_type);
3063
1.42k
                if (opcode == REOP_range32_i) {
3064
397
                    c = lre_canonicalize(c, s->is_unicode);
3065
397
                }
3066
1.42k
                idx_min = 0;
3067
1.42k
                low = get_u32(pc + 0 * 8);
3068
1.42k
                if (c < low)
3069
721
                    goto no_match;
3070
708
                idx_max = n - 1;
3071
708
                high = get_u32(pc + idx_max * 8 + 4);
3072
708
                if (c > high)
3073
0
                    goto no_match;
3074
1.74k
                while (idx_min <= idx_max) {
3075
1.36k
                    idx = (idx_min + idx_max) / 2;
3076
1.36k
                    low = get_u32(pc + idx * 8);
3077
1.36k
                    high = get_u32(pc + idx * 8 + 4);
3078
1.36k
                    if (c < low)
3079
407
                        idx_max = idx - 1;
3080
957
                    else if (c > high)
3081
627
                        idx_min = idx + 1;
3082
330
                    else
3083
330
                        goto range32_match;
3084
1.36k
                }
3085
378
                goto no_match;
3086
378
            range32_match:
3087
330
                pc += 8 * n;
3088
330
            }
3089
0
            break;
3090
2.26k
        case REOP_prev:
3091
            /* go to the previous char */
3092
2.26k
            if (cptr == s->cbuf)
3093
488
                goto no_match;
3094
1.77k
            PREV_CHAR(cptr, s->cbuf, cbuf_type);
3095
1.77k
            break;
3096
156k
        case REOP_simple_greedy_quant:
3097
156k
            {
3098
156k
                uint32_t next_pos, quant_min, quant_max;
3099
156k
                size_t q;
3100
156k
                intptr_t res;
3101
156k
                const uint8_t *pc1;
3102
3103
156k
                next_pos = get_u32(pc);
3104
156k
                quant_min = get_u32(pc + 4);
3105
156k
                quant_max = get_u32(pc + 8);
3106
156k
                pc += 16;
3107
156k
                pc1 = pc;
3108
156k
                pc += (int)next_pos;
3109
3110
156k
                q = 0;
3111
331k
                for(;;) {
3112
331k
                    if (lre_poll_timeout(s))
3113
24
                        return LRE_RET_TIMEOUT;
3114
331k
                    res = lre_exec_backtrack(s, capture, stack, stack_len,
3115
331k
                                             pc1, cptr, TRUE);
3116
331k
                    if (res == LRE_RET_MEMORY_ERROR ||
3117
331k
                        res == LRE_RET_TIMEOUT)
3118
0
                        return res;
3119
331k
                    if (!res)
3120
120k
                        break;
3121
211k
                    cptr = (uint8_t *)res;
3122
211k
                    q++;
3123
211k
                    if (q >= quant_max && quant_max != INT32_MAX)
3124
36.7k
                        break;
3125
211k
                }
3126
156k
                if (q < quant_min)
3127
5.96k
                    goto no_match;
3128
150k
                if (q > quant_min) {
3129
                    /* will examine all matches down to quant_min */
3130
52.2k
                    ret = push_state(s, capture, stack, stack_len,
3131
52.2k
                                     pc1 - 16, cptr,
3132
52.2k
                                     RE_EXEC_STATE_GREEDY_QUANT,
3133
52.2k
                                     q - quant_min);
3134
52.2k
                    if (ret < 0)
3135
0
                        return LRE_RET_MEMORY_ERROR;
3136
52.2k
                }
3137
150k
            }
3138
150k
            break;
3139
150k
        default:
3140
0
            abort();
3141
8.89M
        }
3142
8.89M
    }
3143
335k
}
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
3.64k
{
3152
3.64k
    REExecContext s_s, *s = &s_s;
3153
3.64k
    int re_flags, i, alloca_size, ret;
3154
3.64k
    StackInt *stack_buf;
3155
3156
3.64k
    re_flags = lre_get_flags(bc_buf);
3157
3.64k
    s->is_unicode = (re_flags & (LRE_FLAG_UNICODE | LRE_FLAG_UNICODE_SETS)) != 0;
3158
3.64k
    s->capture_count = bc_buf[RE_HEADER_CAPTURE_COUNT];
3159
3.64k
    s->stack_size_max = bc_buf[RE_HEADER_STACK_SIZE];
3160
3.64k
    s->cbuf = cbuf;
3161
3.64k
    s->cbuf_end = cbuf + (clen << cbuf_type);
3162
3.64k
    s->cbuf_type = cbuf_type;
3163
3.64k
    if (s->cbuf_type == 1 && s->is_unicode)
3164
0
        s->cbuf_type = 2;
3165
3.64k
    s->interrupt_counter = INTERRUPT_COUNTER_INIT;
3166
3.64k
    s->opaque = opaque;
3167
3168
3.64k
    s->state_size = sizeof(REExecState) +
3169
3.64k
        s->capture_count * sizeof(capture[0]) * 2 +
3170
3.64k
        s->stack_size_max * sizeof(stack_buf[0]);
3171
3.64k
    s->state_stack = NULL;
3172
3.64k
    s->state_stack_len = 0;
3173
3.64k
    s->state_stack_size = 0;
3174
3175
25.0k
    for(i = 0; i < s->capture_count * 2; i++)
3176
21.3k
        capture[i] = NULL;
3177
3.64k
    alloca_size = s->stack_size_max * sizeof(stack_buf[0]);
3178
3.64k
    stack_buf = alloca(alloca_size);
3179
3.64k
    ret = lre_exec_backtrack(s, capture, stack_buf, 0, bc_buf + RE_HEADER_LEN,
3180
3.64k
                             cbuf + (cindex << cbuf_type), FALSE);
3181
3.64k
    lre_realloc(s->opaque, s->state_stack, 0);
3182
3.64k
    return ret;
3183
3.64k
}
3184
3185
int lre_get_capture_count(const uint8_t *bc_buf)
3186
766
{
3187
766
    return bc_buf[RE_HEADER_CAPTURE_COUNT];
3188
766
}
3189
3190
int lre_get_flags(const uint8_t *bc_buf)
3191
3.66k
{
3192
3.66k
    return get_u16(bc_buf + RE_HEADER_FLAGS);
3193
3.66k
}
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