blob: 64e199d9bef6722acf2f318d688df3cc0de25962 [file] [log] [blame]
Guido van Rossum004c1e11997-05-09 02:35:58 +00001/* regexpr.c
2 *
3 * Author: Tatu Ylonen <ylo@ngs.fi>
4 *
5 * Copyright (c) 1991 Tatu Ylonen, Espoo, Finland
6 *
7 * Permission to use, copy, modify, distribute, and sell this software
8 * and its documentation for any purpose is hereby granted without
9 * fee, provided that the above copyright notice appear in all copies.
10 * This software is provided "as is" without express or implied
11 * warranty.
12 *
13 * Created: Thu Sep 26 17:14:05 1991 ylo
14 * Last modified: Mon Nov 4 17:06:48 1991 ylo
15 * Ported to Think C: 19 Jan 1992 guido@cwi.nl
16 *
17 * This code draws many ideas from the regular expression packages by
18 * Henry Spencer of the University of Toronto and Richard Stallman of
19 * the Free Software Foundation.
20 *
21 * Emacs-specific code and syntax table code is almost directly borrowed
22 * from GNU regexp.
23 *
24 * Bugs fixed and lots of reorganization by Jeffrey C. Ollie, April
25 * 1997 Thanks for bug reports and ideas from Andrew Kuchling, Tim
26 * Peters, Guido van Rossum, Ka-Ping Yee, Sjoerd Mullender, and
27 * probably one or two others that I'm forgetting.
28 *
29 * $Id$ */
Guido van Rossumb674c3b1992-01-19 16:32:47 +000030
Guido van Rossum95e80531997-08-13 22:34:14 +000031#include "Python.h"
Guido van Rossumb674c3b1992-01-19 16:32:47 +000032#include "regexpr.h"
Guido van Rossum8102c001997-09-05 01:48:48 +000033#include <assert.h>
Guido van Rossumb674c3b1992-01-19 16:32:47 +000034
Guido van Rossumdb25f321997-07-10 14:31:32 +000035/* The original code blithely assumed that sizeof(short) == 2. Not
36 * always true. Original instances of "(short)x" were replaced by
37 * SHORT(x), where SHORT is #defined below. */
38
39#define SHORT(x) ((x) & 0x8000 ? (x) - 0x10000 : (x))
40
Guido van Rossum004c1e11997-05-09 02:35:58 +000041/* The stack implementation is taken from an idea by Andrew Kuchling.
42 * It's a doubly linked list of arrays. The advantages of this over a
43 * simple linked list are that the number of mallocs required are
44 * reduced. It also makes it possible to statically allocate enough
45 * space so that small patterns don't ever need to call malloc.
46 *
47 * The advantages over a single array is that is periodically
48 * realloced when more space is needed is that we avoid ever copying
49 * the stack. */
50
51/* item_t is the basic stack element. Defined as a union of
52 * structures so that both registers, failure points, and counters can
53 * be pushed/popped from the stack. There's nothing built into the
54 * item to keep track of whether a certain stack item is a register, a
55 * failure point, or a counter. */
56
57typedef union item_t
58{
Guido van Rossumdb25f321997-07-10 14:31:32 +000059 struct
60 {
61 int num;
62 int level;
Guido van Rossum95e80531997-08-13 22:34:14 +000063 unsigned char *start;
64 unsigned char *end;
Guido van Rossumdb25f321997-07-10 14:31:32 +000065 } reg;
66 struct
67 {
68 int count;
69 int level;
70 int phantom;
Guido van Rossum95e80531997-08-13 22:34:14 +000071 unsigned char *code;
72 unsigned char *text;
Guido van Rossumdb25f321997-07-10 14:31:32 +000073 } fail;
74 struct
75 {
76 int num;
77 int level;
78 int count;
79 } cntr;
Guido van Rossum004c1e11997-05-09 02:35:58 +000080} item_t;
81
82#define STACK_PAGE_SIZE 256
83#define NUM_REGISTERS 256
84
85/* A 'page' of stack items. */
86
87typedef struct item_page_t
88{
Guido van Rossumdb25f321997-07-10 14:31:32 +000089 item_t items[STACK_PAGE_SIZE];
90 struct item_page_t *prev;
91 struct item_page_t *next;
Guido van Rossum004c1e11997-05-09 02:35:58 +000092} item_page_t;
93
94
95typedef struct match_state
96{
Guido van Rossumdb25f321997-07-10 14:31:32 +000097 /* The number of registers that have been pushed onto the stack
98 * since the last failure point. */
Guido van Rossum004c1e11997-05-09 02:35:58 +000099
Guido van Rossumdb25f321997-07-10 14:31:32 +0000100 int count;
101
102 /* Used to control when registers need to be pushed onto the
103 * stack. */
104
105 int level;
106
107 /* The number of failure points on the stack. */
108
109 int point;
110
111 /* Storage for the registers. Each register consists of two
112 * pointers to characters. So register N is represented as
113 * start[N] and end[N]. The pointers must be converted to
114 * offsets from the beginning of the string before returning the
115 * registers to the calling program. */
116
Guido van Rossum95e80531997-08-13 22:34:14 +0000117 unsigned char *start[NUM_REGISTERS];
118 unsigned char *end[NUM_REGISTERS];
Guido van Rossumdb25f321997-07-10 14:31:32 +0000119
120 /* Keeps track of whether a register has changed recently. */
121
122 int changed[NUM_REGISTERS];
123
124 /* Structure to encapsulate the stack. */
125 struct
126 {
127 /* index into the curent page. If index == 0 and you need
128 * to pop an item, move to the previous page and set index
129 * = STACK_PAGE_SIZE - 1. Otherwise decrement index to
130 * push a page. If index == STACK_PAGE_SIZE and you need
131 * to push a page move to the next page and set index =
132 * 0. If there is no new next page, allocate a new page
133 * and link it in. Otherwise, increment index to push a
134 * page. */
135
136 int index;
137 item_page_t *current; /* Pointer to the current page. */
138 item_page_t first; /* First page is statically allocated. */
139 } stack;
Guido van Rossum004c1e11997-05-09 02:35:58 +0000140} match_state;
141
Guido van Rossumdb25f321997-07-10 14:31:32 +0000142/* Initialize a state object */
143
144/* #define NEW_STATE(state) \ */
145/* memset(&state, 0, (void *)(&state.stack) - (void *)(&state)); \ */
146/* state.stack.current = &state.stack.first; \ */
147/* state.stack.first.prev = NULL; \ */
148/* state.stack.first.next = NULL; \ */
149/* state.stack.index = 0; \ */
150/* state.level = 1 */
151
152#define NEW_STATE(state, nregs) \
153{ \
154 int i; \
155 for (i = 0; i < nregs; i++) \
156 { \
157 state.start[i] = NULL; \
158 state.end[i] = NULL; \
159 state.changed[i] = 0; \
160 } \
161 state.stack.current = &state.stack.first; \
162 state.stack.first.prev = NULL; \
163 state.stack.first.next = NULL; \
164 state.stack.index = 0; \
165 state.level = 1; \
166 state.count = 0; \
167 state.level = 0; \
168 state.point = 0; \
169}
170
171/* Free any memory that might have been malloc'd */
172
173#define FREE_STATE(state) \
174while(state.stack.first.next != NULL) \
175{ \
176 state.stack.current = state.stack.first.next; \
177 state.stack.first.next = state.stack.current->next; \
178 free(state.stack.current); \
179}
180
Guido van Rossum004c1e11997-05-09 02:35:58 +0000181/* Discard the top 'count' stack items. */
182
183#define STACK_DISCARD(stack, count, on_error) \
184stack.index -= count; \
185while (stack.index < 0) \
186{ \
Guido van Rossumfaf49081997-07-15 01:47:08 +0000187 if (stack.current->prev == NULL) \
188 on_error; \
189 stack.current = stack.current->prev; \
190 stack.index += STACK_PAGE_SIZE; \
Guido van Rossum004c1e11997-05-09 02:35:58 +0000191}
192
193/* Store a pointer to the previous item on the stack. Used to pop an
194 * item off of the stack. */
195
196#define STACK_PREV(stack, top, on_error) \
197if (stack.index == 0) \
198{ \
Guido van Rossumfaf49081997-07-15 01:47:08 +0000199 if (stack.current->prev == NULL) \
200 on_error; \
201 stack.current = stack.current->prev; \
202 stack.index = STACK_PAGE_SIZE - 1; \
Guido van Rossum004c1e11997-05-09 02:35:58 +0000203} \
204else \
Guido van Rossumfaf49081997-07-15 01:47:08 +0000205{ \
206 stack.index--; \
207} \
Guido van Rossum004c1e11997-05-09 02:35:58 +0000208top = &(stack.current->items[stack.index])
209
210/* Store a pointer to the next item on the stack. Used to push an item
211 * on to the stack. */
212
213#define STACK_NEXT(stack, top, on_error) \
214if (stack.index == STACK_PAGE_SIZE) \
215{ \
Guido van Rossumfaf49081997-07-15 01:47:08 +0000216 if (stack.current->next == NULL) \
217 { \
218 stack.current->next = (item_page_t *)malloc(sizeof(item_page_t)); \
219 if (stack.current->next == NULL) \
220 on_error; \
221 stack.current->next->prev = stack.current; \
222 stack.current->next->next = NULL; \
223 } \
224 stack.current = stack.current->next; \
225 stack.index = 0; \
Guido van Rossum004c1e11997-05-09 02:35:58 +0000226} \
227top = &(stack.current->items[stack.index++])
228
229/* Store a pointer to the item that is 'count' items back in the
230 * stack. STACK_BACK(stack, top, 1, on_error) is equivalent to
231 * STACK_TOP(stack, top, on_error). */
232
233#define STACK_BACK(stack, top, count, on_error) \
234{ \
Guido van Rossumfaf49081997-07-15 01:47:08 +0000235 int index; \
236 item_page_t *current; \
237 current = stack.current; \
238 index = stack.index - (count); \
239 while (index < 0) \
240 { \
241 if (current->prev == NULL) \
242 on_error; \
243 current = current->prev; \
244 index += STACK_PAGE_SIZE; \
245 } \
246 top = &(current->items[index]); \
Guido van Rossum004c1e11997-05-09 02:35:58 +0000247}
248
249/* Store a pointer to the top item on the stack. Execute the
250 * 'on_error' code if there are no items on the stack. */
251
252#define STACK_TOP(stack, top, on_error) \
253if (stack.index == 0) \
254{ \
Guido van Rossumfaf49081997-07-15 01:47:08 +0000255 if (stack.current->prev == NULL) \
256 on_error; \
257 top = &(stack.current->prev->items[STACK_PAGE_SIZE - 1]); \
Guido van Rossum004c1e11997-05-09 02:35:58 +0000258} \
259else \
Guido van Rossumfaf49081997-07-15 01:47:08 +0000260{ \
261 top = &(stack.current->items[stack.index - 1]); \
262}
Guido van Rossum004c1e11997-05-09 02:35:58 +0000263
264/* Test to see if the stack is empty */
265
266#define STACK_EMPTY(stack) ((stack.index == 0) && \
267 (stack.current->prev == NULL))
268
Guido van Rossum004c1e11997-05-09 02:35:58 +0000269/* Return the start of register 'reg' */
270
271#define GET_REG_START(state, reg) (state.start[reg])
272
273/* Return the end of register 'reg' */
274
275#define GET_REG_END(state, reg) (state.end[reg])
276
277/* Set the start of register 'reg'. If the state of the register needs
278 * saving, push it on the stack. */
279
280#define SET_REG_START(state, reg, text, on_error) \
281if(state.changed[reg] < state.level) \
282{ \
Guido van Rossumfaf49081997-07-15 01:47:08 +0000283 item_t *item; \
284 STACK_NEXT(state.stack, item, on_error); \
285 item->reg.num = reg; \
286 item->reg.start = state.start[reg]; \
287 item->reg.end = state.end[reg]; \
288 item->reg.level = state.changed[reg]; \
289 state.changed[reg] = state.level; \
290 state.count++; \
Guido van Rossum004c1e11997-05-09 02:35:58 +0000291} \
292state.start[reg] = text
293
294/* Set the end of register 'reg'. If the state of the register needs
295 * saving, push it on the stack. */
296
297#define SET_REG_END(state, reg, text, on_error) \
298if(state.changed[reg] < state.level) \
299{ \
Guido van Rossumfaf49081997-07-15 01:47:08 +0000300 item_t *item; \
301 STACK_NEXT(state.stack, item, on_error); \
302 item->reg.num = reg; \
303 item->reg.start = state.start[reg]; \
304 item->reg.end = state.end[reg]; \
305 item->reg.level = state.changed[reg]; \
306 state.changed[reg] = state.level; \
307 state.count++; \
Guido van Rossum004c1e11997-05-09 02:35:58 +0000308} \
309state.end[reg] = text
310
311#define PUSH_FAILURE(state, xcode, xtext, on_error) \
312{ \
Guido van Rossumfaf49081997-07-15 01:47:08 +0000313 item_t *item; \
314 STACK_NEXT(state.stack, item, on_error); \
315 item->fail.code = xcode; \
316 item->fail.text = xtext; \
317 item->fail.count = state.count; \
318 item->fail.level = state.level; \
319 item->fail.phantom = 0; \
320 state.count = 0; \
321 state.level++; \
322 state.point++; \
Guido van Rossum004c1e11997-05-09 02:35:58 +0000323}
324
325/* Update the last failure point with a new position in the text. */
326
Guido van Rossum004c1e11997-05-09 02:35:58 +0000327#define UPDATE_FAILURE(state, xtext, on_error) \
328{ \
Guido van Rossumfaf49081997-07-15 01:47:08 +0000329 item_t *item; \
330 STACK_BACK(state.stack, item, state.count + 1, on_error); \
331 if (!item->fail.phantom) \
332 { \
333 item_t *item2; \
334 STACK_NEXT(state.stack, item2, on_error); \
335 item2->fail.code = item->fail.code; \
336 item2->fail.text = xtext; \
337 item2->fail.count = state.count; \
338 item2->fail.level = state.level; \
339 item2->fail.phantom = 1; \
340 state.count = 0; \
341 state.level++; \
342 state.point++; \
343 } \
344 else \
345 { \
346 STACK_DISCARD(state.stack, state.count, on_error); \
347 STACK_TOP(state.stack, item, on_error); \
348 item->fail.text = xtext; \
349 state.count = 0; \
350 state.level++; \
351 } \
Guido van Rossum004c1e11997-05-09 02:35:58 +0000352}
353
354#define POP_FAILURE(state, xcode, xtext, on_empty, on_error) \
355{ \
Guido van Rossumfaf49081997-07-15 01:47:08 +0000356 item_t *item; \
357 do \
358 { \
359 while(state.count > 0) \
360 { \
361 STACK_PREV(state.stack, item, on_error); \
362 state.start[item->reg.num] = item->reg.start; \
363 state.end[item->reg.num] = item->reg.end; \
364 state.changed[item->reg.num] = item->reg.level; \
365 state.count--; \
366 } \
367 STACK_PREV(state.stack, item, on_empty); \
368 xcode = item->fail.code; \
369 xtext = item->fail.text; \
370 state.count = item->fail.count; \
371 state.level = item->fail.level; \
372 state.point--; \
373 } \
374 while (item->fail.text == NULL); \
Guido van Rossum004c1e11997-05-09 02:35:58 +0000375}
Guido van Rossumb674c3b1992-01-19 16:32:47 +0000376
377enum regexp_compiled_ops /* opcodes for compiled regexp */
378{
Guido van Rossumfaf49081997-07-15 01:47:08 +0000379 Cend, /* end of pattern reached */
380 Cbol, /* beginning of line */
381 Ceol, /* end of line */
382 Cset, /* character set. Followed by 32 bytes of set. */
383 Cexact, /* followed by a byte to match */
384 Canychar, /* matches any character except newline */
385 Cstart_memory, /* set register start addr (followed by reg number) */
386 Cend_memory, /* set register end addr (followed by reg number) */
387 Cmatch_memory, /* match a duplicate of reg contents (regnum follows)*/
388 Cjump, /* followed by two bytes (lsb,msb) of displacement. */
389 Cstar_jump, /* will change to jump/update_failure_jump at runtime */
390 Cfailure_jump, /* jump to addr on failure */
391 Cupdate_failure_jump, /* update topmost failure point and jump */
392 Cdummy_failure_jump, /* push a dummy failure point and jump */
393 Cbegbuf, /* match at beginning of buffer */
394 Cendbuf, /* match at end of buffer */
395 Cwordbeg, /* match at beginning of word */
396 Cwordend, /* match at end of word */
397 Cwordbound, /* match if at word boundary */
398 Cnotwordbound, /* match if not at word boundary */
399 Csyntaxspec, /* matches syntax code (1 byte follows) */
Guido van Rossum95e80531997-08-13 22:34:14 +0000400 Cnotsyntaxspec, /* matches if syntax code does not match (1 byte follows) */
Guido van Rossumfaf49081997-07-15 01:47:08 +0000401 Crepeat1
Guido van Rossumb674c3b1992-01-19 16:32:47 +0000402};
403
404enum regexp_syntax_op /* syntax codes for plain and quoted characters */
405{
Guido van Rossumfaf49081997-07-15 01:47:08 +0000406 Rend, /* special code for end of regexp */
407 Rnormal, /* normal character */
408 Ranychar, /* any character except newline */
409 Rquote, /* the quote character */
410 Rbol, /* match beginning of line */
411 Reol, /* match end of line */
412 Roptional, /* match preceding expression optionally */
413 Rstar, /* match preceding expr zero or more times */
414 Rplus, /* match preceding expr one or more times */
415 Ror, /* match either of alternatives */
416 Ropenpar, /* opening parenthesis */
417 Rclosepar, /* closing parenthesis */
418 Rmemory, /* match memory register */
419 Rextended_memory, /* \vnn to match registers 10-99 */
420 Ropenset, /* open set. Internal syntax hard-coded below. */
421 /* the following are gnu extensions to "normal" regexp syntax */
422 Rbegbuf, /* beginning of buffer */
423 Rendbuf, /* end of buffer */
424 Rwordchar, /* word character */
425 Rnotwordchar, /* not word character */
426 Rwordbeg, /* beginning of word */
427 Rwordend, /* end of word */
428 Rwordbound, /* word bound */
429 Rnotwordbound, /* not word bound */
430 Rnum_ops
Guido van Rossumb674c3b1992-01-19 16:32:47 +0000431};
432
433static int re_compile_initialized = 0;
434static int regexp_syntax = 0;
Guido van Rossumb6775db1994-08-01 11:34:53 +0000435int re_syntax = 0; /* Exported copy of regexp_syntax */
Guido van Rossumb674c3b1992-01-19 16:32:47 +0000436static unsigned char regexp_plain_ops[256];
437static unsigned char regexp_quoted_ops[256];
438static unsigned char regexp_precedences[Rnum_ops];
439static int regexp_context_indep_ops;
440static int regexp_ansi_sequences;
441
442#define NUM_LEVELS 5 /* number of precedence levels in use */
443#define MAX_NESTING 100 /* max nesting level of operators */
444
Guido van Rossumb674c3b1992-01-19 16:32:47 +0000445#define SYNTAX(ch) re_syntax_table[(unsigned char)(ch)]
Guido van Rossumb674c3b1992-01-19 16:32:47 +0000446
Guido van Rossum95e80531997-08-13 22:34:14 +0000447unsigned char re_syntax_table[256];
Guido van Rossumb674c3b1992-01-19 16:32:47 +0000448
Guido van Rossum8102c001997-09-05 01:48:48 +0000449void re_compile_initialize()
Guido van Rossumb674c3b1992-01-19 16:32:47 +0000450{
Guido van Rossumfaf49081997-07-15 01:47:08 +0000451 int a;
Guido van Rossumb674c3b1992-01-19 16:32:47 +0000452
Guido van Rossumfaf49081997-07-15 01:47:08 +0000453 static int syntax_table_inited = 0;
Guido van Rossum74fb3031997-07-17 22:41:38 +0000454
Guido van Rossumfaf49081997-07-15 01:47:08 +0000455 if (!syntax_table_inited)
456 {
457 syntax_table_inited = 1;
458 memset(re_syntax_table, 0, 256);
459 for (a = 'a'; a <= 'z'; a++)
460 re_syntax_table[a] = Sword;
461 for (a = 'A'; a <= 'Z'; a++)
462 re_syntax_table[a] = Sword;
463 for (a = '0'; a <= '9'; a++)
Guido van Rossum52d68321997-08-13 03:21:14 +0000464 re_syntax_table[a] = Sword | Sdigit | Shexdigit;
465 for (a = '0'; a <= '7'; a++)
466 re_syntax_table[a] |= Soctaldigit;
467 for (a = 'A'; a <= 'F'; a++)
468 re_syntax_table[a] |= Shexdigit;
469 for (a = 'a'; a <= 'f'; a++)
470 re_syntax_table[a] |= Shexdigit;
Guido van Rossum74fb3031997-07-17 22:41:38 +0000471 re_syntax_table['_'] = Sword;
472 for (a = 9; a <= 13; a++)
473 re_syntax_table[a] = Swhitespace;
474 re_syntax_table[' '] = Swhitespace;
Guido van Rossumfaf49081997-07-15 01:47:08 +0000475 }
476 re_compile_initialized = 1;
477 for (a = 0; a < 256; a++)
478 {
479 regexp_plain_ops[a] = Rnormal;
480 regexp_quoted_ops[a] = Rnormal;
481 }
482 for (a = '0'; a <= '9'; a++)
483 regexp_quoted_ops[a] = Rmemory;
484 regexp_plain_ops['\134'] = Rquote;
485 if (regexp_syntax & RE_NO_BK_PARENS)
486 {
487 regexp_plain_ops['('] = Ropenpar;
488 regexp_plain_ops[')'] = Rclosepar;
489 }
490 else
491 {
492 regexp_quoted_ops['('] = Ropenpar;
493 regexp_quoted_ops[')'] = Rclosepar;
494 }
495 if (regexp_syntax & RE_NO_BK_VBAR)
496 regexp_plain_ops['\174'] = Ror;
497 else
498 regexp_quoted_ops['\174'] = Ror;
499 regexp_plain_ops['*'] = Rstar;
500 if (regexp_syntax & RE_BK_PLUS_QM)
501 {
502 regexp_quoted_ops['+'] = Rplus;
503 regexp_quoted_ops['?'] = Roptional;
504 }
505 else
506 {
507 regexp_plain_ops['+'] = Rplus;
508 regexp_plain_ops['?'] = Roptional;
509 }
510 if (regexp_syntax & RE_NEWLINE_OR)
511 regexp_plain_ops['\n'] = Ror;
512 regexp_plain_ops['\133'] = Ropenset;
513 regexp_plain_ops['\136'] = Rbol;
514 regexp_plain_ops['$'] = Reol;
515 regexp_plain_ops['.'] = Ranychar;
516 if (!(regexp_syntax & RE_NO_GNU_EXTENSIONS))
517 {
518 regexp_quoted_ops['w'] = Rwordchar;
519 regexp_quoted_ops['W'] = Rnotwordchar;
520 regexp_quoted_ops['<'] = Rwordbeg;
521 regexp_quoted_ops['>'] = Rwordend;
522 regexp_quoted_ops['b'] = Rwordbound;
523 regexp_quoted_ops['B'] = Rnotwordbound;
524 regexp_quoted_ops['`'] = Rbegbuf;
525 regexp_quoted_ops['\''] = Rendbuf;
526 }
527 if (regexp_syntax & RE_ANSI_HEX)
528 regexp_quoted_ops['v'] = Rextended_memory;
529 for (a = 0; a < Rnum_ops; a++)
530 regexp_precedences[a] = 4;
531 if (regexp_syntax & RE_TIGHT_VBAR)
532 {
533 regexp_precedences[Ror] = 3;
534 regexp_precedences[Rbol] = 2;
535 regexp_precedences[Reol] = 2;
536 }
537 else
538 {
539 regexp_precedences[Ror] = 2;
540 regexp_precedences[Rbol] = 3;
541 regexp_precedences[Reol] = 3;
542 }
543 regexp_precedences[Rclosepar] = 1;
544 regexp_precedences[Rend] = 0;
545 regexp_context_indep_ops = (regexp_syntax & RE_CONTEXT_INDEP_OPS) != 0;
546 regexp_ansi_sequences = (regexp_syntax & RE_ANSI_HEX) != 0;
Guido van Rossumb674c3b1992-01-19 16:32:47 +0000547}
548
Guido van Rossum8102c001997-09-05 01:48:48 +0000549int re_set_syntax(syntax)
550 int syntax;
Guido van Rossumb674c3b1992-01-19 16:32:47 +0000551{
Guido van Rossumfaf49081997-07-15 01:47:08 +0000552 int ret;
553
554 ret = regexp_syntax;
555 regexp_syntax = syntax;
556 re_syntax = syntax; /* Exported copy */
557 re_compile_initialize();
558 return ret;
Guido van Rossumb674c3b1992-01-19 16:32:47 +0000559}
560
Guido van Rossum8102c001997-09-05 01:48:48 +0000561static int hex_char_to_decimal(ch)
562 int ch;
Guido van Rossumb674c3b1992-01-19 16:32:47 +0000563{
Guido van Rossumfaf49081997-07-15 01:47:08 +0000564 if (ch >= '0' && ch <= '9')
565 return ch - '0';
566 if (ch >= 'a' && ch <= 'f')
567 return ch - 'a' + 10;
568 if (ch >= 'A' && ch <= 'F')
569 return ch - 'A' + 10;
570 return 16;
Guido van Rossumb674c3b1992-01-19 16:32:47 +0000571}
572
Guido van Rossum8102c001997-09-05 01:48:48 +0000573static void re_compile_fastmap_aux(code,
574 pos,
575 visited,
576 can_be_null,
577 fastmap)
578 unsigned char *code;
579 int pos;
580 unsigned char *visited;
581 unsigned char *can_be_null;
582 unsigned char *fastmap;
Guido van Rossumb674c3b1992-01-19 16:32:47 +0000583{
Guido van Rossumfaf49081997-07-15 01:47:08 +0000584 int a;
585 int b;
586 int syntaxcode;
587
588 if (visited[pos])
589 return; /* we have already been here */
590 visited[pos] = 1;
591 for (;;)
Guido van Rossum74fb3031997-07-17 22:41:38 +0000592 switch (code[pos++]) {
Guido van Rossumfaf49081997-07-15 01:47:08 +0000593 case Cend:
Guido van Rossum74fb3031997-07-17 22:41:38 +0000594 {
595 *can_be_null = 1;
596 return;
597 }
Guido van Rossumfaf49081997-07-15 01:47:08 +0000598 case Cbol:
599 case Cbegbuf:
600 case Cendbuf:
601 case Cwordbeg:
602 case Cwordend:
603 case Cwordbound:
604 case Cnotwordbound:
605 {
606 for (a = 0; a < 256; a++)
607 fastmap[a] = 1;
608 break;
609 }
610 case Csyntaxspec:
611 {
612 syntaxcode = code[pos++];
613 for (a = 0; a < 256; a++)
Guido van Rossume59d3f81997-12-02 20:39:23 +0000614 if (SYNTAX(a) & syntaxcode)
Guido van Rossumfaf49081997-07-15 01:47:08 +0000615 fastmap[a] = 1;
616 return;
617 }
618 case Cnotsyntaxspec:
619 {
620 syntaxcode = code[pos++];
621 for (a = 0; a < 256; a++)
Guido van Rossume59d3f81997-12-02 20:39:23 +0000622 if (!(SYNTAX(a) & syntaxcode) )
Guido van Rossumfaf49081997-07-15 01:47:08 +0000623 fastmap[a] = 1;
624 return;
625 }
626 case Ceol:
627 {
628 fastmap['\n'] = 1;
629 if (*can_be_null == 0)
630 *can_be_null = 2; /* can match null, but only at end of buffer*/
631 return;
632 }
633 case Cset:
634 {
635 for (a = 0; a < 256/8; a++)
636 if (code[pos + a] != 0)
637 for (b = 0; b < 8; b++)
638 if (code[pos + a] & (1 << b))
639 fastmap[(a << 3) + b] = 1;
640 pos += 256/8;
641 return;
642 }
643 case Cexact:
644 {
645 fastmap[(unsigned char)code[pos]] = 1;
646 return;
647 }
648 case Canychar:
649 {
650 for (a = 0; a < 256; a++)
651 if (a != '\n')
652 fastmap[a] = 1;
653 return;
654 }
655 case Cstart_memory:
656 case Cend_memory:
657 {
658 pos++;
659 break;
660 }
661 case Cmatch_memory:
662 {
663 for (a = 0; a < 256; a++)
664 fastmap[a] = 1;
665 *can_be_null = 1;
666 return;
667 }
668 case Cjump:
669 case Cdummy_failure_jump:
670 case Cupdate_failure_jump:
671 case Cstar_jump:
672 {
673 a = (unsigned char)code[pos++];
674 a |= (unsigned char)code[pos++] << 8;
675 pos += (int)SHORT(a);
676 if (visited[pos])
677 {
678 /* argh... the regexp contains empty loops. This is not
679 good, as this may cause a failure stack overflow when
680 matching. Oh well. */
681 /* this path leads nowhere; pursue other paths. */
682 return;
683 }
684 visited[pos] = 1;
685 break;
686 }
687 case Cfailure_jump:
688 {
689 a = (unsigned char)code[pos++];
690 a |= (unsigned char)code[pos++] << 8;
691 a = pos + (int)SHORT(a);
692 re_compile_fastmap_aux(code, a, visited, can_be_null, fastmap);
693 break;
694 }
695 case Crepeat1:
696 {
697 pos += 2;
698 break;
699 }
700 default:
701 {
Guido van Rossum95e80531997-08-13 22:34:14 +0000702 PyErr_SetString(PyExc_SystemError, "Unknown regex opcode: memory corrupted?");
703 return;
Guido van Rossumfaf49081997-07-15 01:47:08 +0000704 /*NOTREACHED*/
705 }
706 }
Guido van Rossum004c1e11997-05-09 02:35:58 +0000707}
Guido van Rossumb674c3b1992-01-19 16:32:47 +0000708
Guido van Rossum8102c001997-09-05 01:48:48 +0000709static int re_do_compile_fastmap(buffer,
710 used,
711 pos,
712 can_be_null,
713 fastmap)
714 unsigned char *buffer;
715 int used;
716 int pos;
717 unsigned char *can_be_null;
718 unsigned char *fastmap;
Guido van Rossum004c1e11997-05-09 02:35:58 +0000719{
Guido van Rossum95e80531997-08-13 22:34:14 +0000720 unsigned char small_visited[512], *visited;
Guido van Rossum004c1e11997-05-09 02:35:58 +0000721
Guido van Rossumfaf49081997-07-15 01:47:08 +0000722 if (used <= sizeof(small_visited))
723 visited = small_visited;
724 else
725 {
726 visited = malloc(used);
727 if (!visited)
728 return 0;
729 }
730 *can_be_null = 0;
731 memset(fastmap, 0, 256);
732 memset(visited, 0, used);
733 re_compile_fastmap_aux(buffer, pos, visited, can_be_null, fastmap);
734 if (visited != small_visited)
735 free(visited);
736 return 1;
Guido van Rossum004c1e11997-05-09 02:35:58 +0000737}
Guido van Rossumb674c3b1992-01-19 16:32:47 +0000738
Guido van Rossum8102c001997-09-05 01:48:48 +0000739void re_compile_fastmap(bufp)
740 regexp_t bufp;
Guido van Rossum004c1e11997-05-09 02:35:58 +0000741{
Guido van Rossumfaf49081997-07-15 01:47:08 +0000742 if (!bufp->fastmap || bufp->fastmap_accurate)
743 return;
744 assert(bufp->used > 0);
745 if (!re_do_compile_fastmap(bufp->buffer,
746 bufp->used,
747 0,
748 &bufp->can_be_null,
749 bufp->fastmap))
750 return;
Guido van Rossum95e80531997-08-13 22:34:14 +0000751 if (PyErr_Occurred()) return;
Guido van Rossumfaf49081997-07-15 01:47:08 +0000752 if (bufp->buffer[0] == Cbol)
753 bufp->anchor = 1; /* begline */
754 else
755 if (bufp->buffer[0] == Cbegbuf)
756 bufp->anchor = 2; /* begbuf */
757 else
758 bufp->anchor = 0; /* none */
759 bufp->fastmap_accurate = 1;
Guido van Rossum004c1e11997-05-09 02:35:58 +0000760}
761
762/*
763 * star is coded as:
764 * 1: failure_jump 2
765 * ... code for operand of star
766 * star_jump 1
767 * 2: ... code after star
768 *
769 * We change the star_jump to update_failure_jump if we can determine
770 * that it is safe to do so; otherwise we change it to an ordinary
771 * jump.
772 *
773 * plus is coded as
774 *
775 * jump 2
776 * 1: failure_jump 3
777 * 2: ... code for operand of plus
778 * star_jump 1
779 * 3: ... code after plus
780 *
781 * For star_jump considerations this is processed identically to star.
782 *
783 */
784
Guido van Rossum8102c001997-09-05 01:48:48 +0000785static int re_optimize_star_jump(bufp, code)
786 regexp_t bufp;
787 unsigned char *code;
Guido van Rossum004c1e11997-05-09 02:35:58 +0000788{
Guido van Rossum95e80531997-08-13 22:34:14 +0000789 unsigned char map[256];
790 unsigned char can_be_null;
791 unsigned char *p1;
792 unsigned char *p2;
793 unsigned char ch;
Guido van Rossumfaf49081997-07-15 01:47:08 +0000794 int a;
795 int b;
796 int num_instructions = 0;
Guido van Rossum95e80531997-08-13 22:34:14 +0000797
Guido van Rossumfaf49081997-07-15 01:47:08 +0000798 a = (unsigned char)*code++;
799 a |= (unsigned char)*code++ << 8;
800 a = (int)SHORT(a);
801
802 p1 = code + a + 3; /* skip the failure_jump */
Guido van Rossum95e80531997-08-13 22:34:14 +0000803 /* Check that the jump is within the pattern */
804 if (p1<bufp->buffer || bufp->buffer+bufp->used<p1)
805 {
806 PyErr_SetString(PyExc_SystemError, "Regex VM jump out of bounds (failure_jump opt)");
807 return 0;
808 }
809
Guido van Rossumfaf49081997-07-15 01:47:08 +0000810 assert(p1[-3] == Cfailure_jump);
811 p2 = code;
812 /* p1 points inside loop, p2 points to after loop */
813 if (!re_do_compile_fastmap(bufp->buffer, bufp->used,
814 p2 - bufp->buffer, &can_be_null, map))
815 goto make_normal_jump;
816
817 /* If we might introduce a new update point inside the
818 * loop, we can't optimize because then update_jump would
819 * update a wrong failure point. Thus we have to be
820 * quite careful here.
821 */
822
823 /* loop until we find something that consumes a character */
Guido van Rossum004c1e11997-05-09 02:35:58 +0000824 loop_p1:
Guido van Rossumfaf49081997-07-15 01:47:08 +0000825 num_instructions++;
826 switch (*p1++)
827 {
828 case Cbol:
829 case Ceol:
830 case Cbegbuf:
831 case Cendbuf:
832 case Cwordbeg:
833 case Cwordend:
834 case Cwordbound:
835 case Cnotwordbound:
836 {
837 goto loop_p1;
838 }
839 case Cstart_memory:
840 case Cend_memory:
841 {
842 p1++;
843 goto loop_p1;
844 }
845 case Cexact:
846 {
847 ch = (unsigned char)*p1++;
848 if (map[(int)ch])
849 goto make_normal_jump;
850 break;
851 }
852 case Canychar:
853 {
854 for (b = 0; b < 256; b++)
855 if (b != '\n' && map[b])
856 goto make_normal_jump;
857 break;
858 }
859 case Cset:
860 {
861 for (b = 0; b < 256; b++)
862 if ((p1[b >> 3] & (1 << (b & 7))) && map[b])
863 goto make_normal_jump;
864 p1 += 256/8;
865 break;
866 }
867 default:
868 {
869 goto make_normal_jump;
870 }
871 }
872 /* now we know that we can't backtrack. */
873 while (p1 != p2 - 3)
874 {
875 num_instructions++;
876 switch (*p1++)
877 {
878 case Cend:
879 {
880 return 0;
881 }
882 case Cbol:
883 case Ceol:
884 case Canychar:
885 case Cbegbuf:
886 case Cendbuf:
887 case Cwordbeg:
888 case Cwordend:
889 case Cwordbound:
890 case Cnotwordbound:
891 {
892 break;
893 }
894 case Cset:
895 {
896 p1 += 256/8;
897 break;
898 }
899 case Cexact:
900 case Cstart_memory:
901 case Cend_memory:
902 case Cmatch_memory:
903 case Csyntaxspec:
904 case Cnotsyntaxspec:
905 {
906 p1++;
907 break;
908 }
909 case Cjump:
910 case Cstar_jump:
911 case Cfailure_jump:
912 case Cupdate_failure_jump:
913 case Cdummy_failure_jump:
914 {
915 goto make_normal_jump;
916 }
917 default:
918 {
919 return 0;
920 break;
921 }
922 }
923 }
924
Guido van Rossum95e80531997-08-13 22:34:14 +0000925 /* make_update_jump: */
Guido van Rossumfaf49081997-07-15 01:47:08 +0000926 code -= 3;
927 a += 3; /* jump to after the Cfailure_jump */
928 code[0] = Cupdate_failure_jump;
929 code[1] = a & 0xff;
930 code[2] = a >> 8;
931 if (num_instructions > 1)
932 return 1;
933 assert(num_instructions == 1);
934 /* if the only instruction matches a single character, we can do
935 * better */
936 p1 = code + 3 + a; /* start of sole instruction */
937 if (*p1 == Cset || *p1 == Cexact || *p1 == Canychar ||
938 *p1 == Csyntaxspec || *p1 == Cnotsyntaxspec)
939 code[0] = Crepeat1;
940 return 1;
941
Guido van Rossum004c1e11997-05-09 02:35:58 +0000942 make_normal_jump:
Guido van Rossumfaf49081997-07-15 01:47:08 +0000943 code -= 3;
944 *code = Cjump;
945 return 1;
Guido van Rossum004c1e11997-05-09 02:35:58 +0000946}
947
Guido van Rossum8102c001997-09-05 01:48:48 +0000948static int re_optimize(bufp)
949 regexp_t bufp;
Guido van Rossum004c1e11997-05-09 02:35:58 +0000950{
Guido van Rossum95e80531997-08-13 22:34:14 +0000951 unsigned char *code;
Guido van Rossumfaf49081997-07-15 01:47:08 +0000952
953 code = bufp->buffer;
954
955 while(1)
956 {
957 switch (*code++)
958 {
959 case Cend:
960 {
961 return 1;
962 }
963 case Canychar:
964 case Cbol:
965 case Ceol:
966 case Cbegbuf:
967 case Cendbuf:
968 case Cwordbeg:
969 case Cwordend:
970 case Cwordbound:
971 case Cnotwordbound:
972 {
973 break;
974 }
975 case Cset:
976 {
977 code += 256/8;
978 break;
979 }
980 case Cexact:
981 case Cstart_memory:
982 case Cend_memory:
983 case Cmatch_memory:
984 case Csyntaxspec:
985 case Cnotsyntaxspec:
986 {
987 code++;
988 break;
989 }
990 case Cstar_jump:
991 {
992 if (!re_optimize_star_jump(bufp, code))
993 {
994 return 0;
995 }
996 /* fall through */
997 }
998 case Cupdate_failure_jump:
999 case Cjump:
1000 case Cdummy_failure_jump:
1001 case Cfailure_jump:
1002 case Crepeat1:
1003 {
1004 code += 2;
1005 break;
1006 }
1007 default:
1008 {
1009 return 0;
1010 }
1011 }
1012 }
Guido van Rossum004c1e11997-05-09 02:35:58 +00001013}
1014
1015#define NEXTCHAR(var) \
1016{ \
Guido van Rossumfaf49081997-07-15 01:47:08 +00001017 if (pos >= size) \
1018 goto ends_prematurely; \
1019 (var) = regex[pos]; \
1020 pos++; \
Guido van Rossum004c1e11997-05-09 02:35:58 +00001021}
1022
1023#define ALLOC(amount) \
1024{ \
Guido van Rossumfaf49081997-07-15 01:47:08 +00001025 if (pattern_offset+(amount) > alloc) \
1026 { \
1027 alloc += 256 + (amount); \
1028 pattern = realloc(pattern, alloc); \
1029 if (!pattern) \
1030 goto out_of_memory; \
1031 } \
Guido van Rossum004c1e11997-05-09 02:35:58 +00001032}
Guido van Rossumb674c3b1992-01-19 16:32:47 +00001033
1034#define STORE(ch) pattern[pattern_offset++] = (ch)
1035
1036#define CURRENT_LEVEL_START (starts[starts_base + current_level])
1037
1038#define SET_LEVEL_START starts[starts_base + current_level] = pattern_offset
1039
Guido van Rossum004c1e11997-05-09 02:35:58 +00001040#define PUSH_LEVEL_STARTS \
Guido van Rossumfaf49081997-07-15 01:47:08 +00001041if (starts_base < (MAX_NESTING-1)*NUM_LEVELS) \
1042 starts_base += NUM_LEVELS; \
1043else \
1044 goto too_complex \
Guido van Rossumb674c3b1992-01-19 16:32:47 +00001045
1046#define POP_LEVEL_STARTS starts_base -= NUM_LEVELS
1047
Guido van Rossum004c1e11997-05-09 02:35:58 +00001048#define PUT_ADDR(offset,addr) \
1049{ \
Guido van Rossumfaf49081997-07-15 01:47:08 +00001050 int disp = (addr) - (offset) - 2; \
1051 pattern[(offset)] = disp & 0xff; \
1052 pattern[(offset)+1] = (disp>>8) & 0xff; \
Guido van Rossum004c1e11997-05-09 02:35:58 +00001053}
Guido van Rossumb674c3b1992-01-19 16:32:47 +00001054
Guido van Rossum004c1e11997-05-09 02:35:58 +00001055#define INSERT_JUMP(pos,type,addr) \
1056{ \
Guido van Rossumfaf49081997-07-15 01:47:08 +00001057 int a, p = (pos), t = (type), ad = (addr); \
1058 for (a = pattern_offset - 1; a >= p; a--) \
1059 pattern[a + 3] = pattern[a]; \
1060 pattern[p] = t; \
1061 PUT_ADDR(p+1,ad); \
1062 pattern_offset += 3; \
Guido van Rossum004c1e11997-05-09 02:35:58 +00001063}
Guido van Rossumfaf49081997-07-15 01:47:08 +00001064
Guido van Rossumb674c3b1992-01-19 16:32:47 +00001065#define SETBIT(buf,offset,bit) (buf)[(offset)+(bit)/8] |= (1<<((bit) & 7))
1066
Guido van Rossum004c1e11997-05-09 02:35:58 +00001067#define SET_FIELDS \
1068{ \
Guido van Rossumfaf49081997-07-15 01:47:08 +00001069 bufp->allocated = alloc; \
1070 bufp->buffer = pattern; \
1071 bufp->used = pattern_offset; \
Guido van Rossum004c1e11997-05-09 02:35:58 +00001072}
Guido van Rossumb674c3b1992-01-19 16:32:47 +00001073
Guido van Rossum004c1e11997-05-09 02:35:58 +00001074#define GETHEX(var) \
1075{ \
Guido van Rossum95e80531997-08-13 22:34:14 +00001076 unsigned char gethex_ch, gethex_value; \
Guido van Rossumfaf49081997-07-15 01:47:08 +00001077 NEXTCHAR(gethex_ch); \
1078 gethex_value = hex_char_to_decimal(gethex_ch); \
1079 if (gethex_value == 16) \
1080 goto hex_error; \
1081 NEXTCHAR(gethex_ch); \
1082 gethex_ch = hex_char_to_decimal(gethex_ch); \
1083 if (gethex_ch == 16) \
1084 goto hex_error; \
1085 (var) = gethex_value * 16 + gethex_ch; \
Guido van Rossum004c1e11997-05-09 02:35:58 +00001086}
Guido van Rossumb674c3b1992-01-19 16:32:47 +00001087
Guido van Rossumfaf49081997-07-15 01:47:08 +00001088#define ANSI_TRANSLATE(ch) \
Guido van Rossum004c1e11997-05-09 02:35:58 +00001089{ \
Guido van Rossumfaf49081997-07-15 01:47:08 +00001090 switch (ch) \
1091 { \
1092 case 'a': \
1093 case 'A': \
1094 { \
1095 ch = 7; /* audible bell */ \
1096 break; \
1097 } \
1098 case 'b': \
1099 case 'B': \
1100 { \
1101 ch = 8; /* backspace */ \
1102 break; \
1103 } \
1104 case 'f': \
1105 case 'F': \
1106 { \
1107 ch = 12; /* form feed */ \
1108 break; \
1109 } \
1110 case 'n': \
1111 case 'N': \
1112 { \
1113 ch = 10; /* line feed */ \
1114 break; \
1115 } \
1116 case 'r': \
1117 case 'R': \
1118 { \
1119 ch = 13; /* carriage return */ \
1120 break; \
1121 } \
1122 case 't': \
1123 case 'T': \
1124 { \
1125 ch = 9; /* tab */ \
1126 break; \
1127 } \
1128 case 'v': \
1129 case 'V': \
1130 { \
1131 ch = 11; /* vertical tab */ \
1132 break; \
1133 } \
1134 case 'x': /* hex code */ \
1135 case 'X': \
1136 { \
1137 GETHEX(ch); \
1138 break; \
1139 } \
1140 default: \
1141 { \
1142 /* other characters passed through */ \
1143 if (translate) \
1144 ch = translate[(unsigned char)ch]; \
1145 break; \
1146 } \
1147 } \
Guido van Rossum004c1e11997-05-09 02:35:58 +00001148}
Guido van Rossumb674c3b1992-01-19 16:32:47 +00001149
Guido van Rossum8102c001997-09-05 01:48:48 +00001150char *re_compile_pattern(regex, size, bufp)
1151 unsigned char *regex;
1152 int size;
1153 regexp_t bufp;
Guido van Rossum004c1e11997-05-09 02:35:58 +00001154{
Guido van Rossumfaf49081997-07-15 01:47:08 +00001155 int a;
1156 int pos;
1157 int op;
1158 int current_level;
1159 int level;
1160 int opcode;
1161 int pattern_offset = 0, alloc;
1162 int starts[NUM_LEVELS * MAX_NESTING];
1163 int starts_base;
1164 int future_jumps[MAX_NESTING];
1165 int num_jumps;
1166 unsigned char ch = '\0';
Guido van Rossum95e80531997-08-13 22:34:14 +00001167 unsigned char *pattern;
1168 unsigned char *translate;
Guido van Rossumfaf49081997-07-15 01:47:08 +00001169 int next_register;
1170 int paren_depth;
1171 int num_open_registers;
1172 int open_registers[RE_NREGS];
1173 int beginning_context;
1174
1175 if (!re_compile_initialized)
1176 re_compile_initialize();
1177 bufp->used = 0;
1178 bufp->fastmap_accurate = 0;
1179 bufp->uses_registers = 1;
1180 bufp->num_registers = 1;
1181 translate = bufp->translate;
1182 pattern = bufp->buffer;
1183 alloc = bufp->allocated;
1184 if (alloc == 0 || pattern == NULL)
1185 {
1186 alloc = 256;
1187 pattern = malloc(alloc);
1188 if (!pattern)
1189 goto out_of_memory;
1190 }
1191 pattern_offset = 0;
1192 starts_base = 0;
1193 num_jumps = 0;
1194 current_level = 0;
1195 SET_LEVEL_START;
1196 num_open_registers = 0;
1197 next_register = 1;
1198 paren_depth = 0;
1199 beginning_context = 1;
1200 op = -1;
1201 /* we use Rend dummy to ensure that pending jumps are updated
1202 (due to low priority of Rend) before exiting the loop. */
1203 pos = 0;
1204 while (op != Rend)
1205 {
1206 if (pos >= size)
1207 op = Rend;
1208 else
1209 {
1210 NEXTCHAR(ch);
1211 if (translate)
1212 ch = translate[(unsigned char)ch];
1213 op = regexp_plain_ops[(unsigned char)ch];
1214 if (op == Rquote)
1215 {
1216 NEXTCHAR(ch);
1217 op = regexp_quoted_ops[(unsigned char)ch];
1218 if (op == Rnormal && regexp_ansi_sequences)
1219 ANSI_TRANSLATE(ch);
1220 }
1221 }
1222 level = regexp_precedences[op];
1223 /* printf("ch='%c' op=%d level=%d current_level=%d
1224 curlevstart=%d\n", ch, op, level, current_level,
1225 CURRENT_LEVEL_START); */
1226 if (level > current_level)
1227 {
1228 for (current_level++; current_level < level; current_level++)
1229 SET_LEVEL_START;
1230 SET_LEVEL_START;
1231 }
1232 else
1233 if (level < current_level)
1234 {
1235 current_level = level;
1236 for (;num_jumps > 0 &&
1237 future_jumps[num_jumps-1] >= CURRENT_LEVEL_START;
1238 num_jumps--)
1239 PUT_ADDR(future_jumps[num_jumps-1], pattern_offset);
1240 }
1241 switch (op)
1242 {
1243 case Rend:
1244 {
1245 break;
1246 }
1247 case Rnormal:
1248 {
1249 normal_char:
1250 opcode = Cexact;
1251 store_opcode_and_arg: /* opcode & ch must be set */
1252 SET_LEVEL_START;
1253 ALLOC(2);
1254 STORE(opcode);
1255 STORE(ch);
1256 break;
1257 }
1258 case Ranychar:
1259 {
1260 opcode = Canychar;
1261 store_opcode:
1262 SET_LEVEL_START;
1263 ALLOC(1);
1264 STORE(opcode);
1265 break;
1266 }
1267 case Rquote:
1268 {
1269 abort();
1270 /*NOTREACHED*/
1271 }
1272 case Rbol:
1273 {
1274 if (!beginning_context)
1275 if (regexp_context_indep_ops)
1276 goto op_error;
1277 else
1278 goto normal_char;
1279 opcode = Cbol;
1280 goto store_opcode;
1281 }
1282 case Reol:
1283 {
1284 if (!((pos >= size) ||
1285 ((regexp_syntax & RE_NO_BK_VBAR) ?
1286 (regex[pos] == '\174') :
1287 (pos+1 < size && regex[pos] == '\134' &&
1288 regex[pos+1] == '\174')) ||
1289 ((regexp_syntax & RE_NO_BK_PARENS)?
1290 (regex[pos] == ')'):
1291 (pos+1 < size && regex[pos] == '\134' &&
1292 regex[pos+1] == ')'))))
1293 if (regexp_context_indep_ops)
1294 goto op_error;
1295 else
1296 goto normal_char;
1297 opcode = Ceol;
1298 goto store_opcode;
1299 /* NOTREACHED */
1300 break;
1301 }
1302 case Roptional:
1303 {
1304 if (beginning_context)
1305 if (regexp_context_indep_ops)
1306 goto op_error;
1307 else
1308 goto normal_char;
1309 if (CURRENT_LEVEL_START == pattern_offset)
1310 break; /* ignore empty patterns for ? */
1311 ALLOC(3);
1312 INSERT_JUMP(CURRENT_LEVEL_START, Cfailure_jump,
1313 pattern_offset + 3);
1314 break;
1315 }
1316 case Rstar:
1317 case Rplus:
1318 {
1319 if (beginning_context)
1320 if (regexp_context_indep_ops)
1321 goto op_error;
1322 else
1323 goto normal_char;
1324 if (CURRENT_LEVEL_START == pattern_offset)
1325 break; /* ignore empty patterns for + and * */
1326 ALLOC(9);
1327 INSERT_JUMP(CURRENT_LEVEL_START, Cfailure_jump,
1328 pattern_offset + 6);
1329 INSERT_JUMP(pattern_offset, Cstar_jump, CURRENT_LEVEL_START);
1330 if (op == Rplus) /* jump over initial failure_jump */
1331 INSERT_JUMP(CURRENT_LEVEL_START, Cdummy_failure_jump,
1332 CURRENT_LEVEL_START + 6);
1333 break;
1334 }
1335 case Ror:
1336 {
1337 ALLOC(6);
1338 INSERT_JUMP(CURRENT_LEVEL_START, Cfailure_jump,
1339 pattern_offset + 6);
1340 if (num_jumps >= MAX_NESTING)
1341 goto too_complex;
1342 STORE(Cjump);
1343 future_jumps[num_jumps++] = pattern_offset;
1344 STORE(0);
1345 STORE(0);
1346 SET_LEVEL_START;
1347 break;
1348 }
1349 case Ropenpar:
1350 {
1351 SET_LEVEL_START;
1352 if (next_register < RE_NREGS)
1353 {
1354 bufp->uses_registers = 1;
1355 ALLOC(2);
1356 STORE(Cstart_memory);
1357 STORE(next_register);
1358 open_registers[num_open_registers++] = next_register;
1359 bufp->num_registers++;
1360 next_register++;
1361 }
1362 paren_depth++;
1363 PUSH_LEVEL_STARTS;
1364 current_level = 0;
1365 SET_LEVEL_START;
1366 break;
1367 }
1368 case Rclosepar:
1369 {
1370 if (paren_depth <= 0)
1371 goto parenthesis_error;
1372 POP_LEVEL_STARTS;
1373 current_level = regexp_precedences[Ropenpar];
1374 paren_depth--;
1375 if (paren_depth < num_open_registers)
1376 {
1377 bufp->uses_registers = 1;
1378 ALLOC(2);
1379 STORE(Cend_memory);
1380 num_open_registers--;
1381 STORE(open_registers[num_open_registers]);
1382 }
1383 break;
1384 }
1385 case Rmemory:
1386 {
1387 if (ch == '0')
1388 goto bad_match_register;
1389 assert(ch >= '0' && ch <= '9');
1390 bufp->uses_registers = 1;
1391 opcode = Cmatch_memory;
1392 ch -= '0';
1393 goto store_opcode_and_arg;
1394 }
1395 case Rextended_memory:
1396 {
1397 NEXTCHAR(ch);
1398 if (ch < '0' || ch > '9')
1399 goto bad_match_register;
1400 NEXTCHAR(a);
1401 if (a < '0' || a > '9')
1402 goto bad_match_register;
1403 ch = 10 * (a - '0') + ch - '0';
1404 if (ch <= 0 || ch >= RE_NREGS)
1405 goto bad_match_register;
1406 bufp->uses_registers = 1;
1407 opcode = Cmatch_memory;
1408 goto store_opcode_and_arg;
1409 }
1410 case Ropenset:
1411 {
1412 int complement;
1413 int prev;
1414 int offset;
1415 int range;
1416 int firstchar;
Guido van Rossumb674c3b1992-01-19 16:32:47 +00001417
Guido van Rossumfaf49081997-07-15 01:47:08 +00001418 SET_LEVEL_START;
1419 ALLOC(1+256/8);
1420 STORE(Cset);
1421 offset = pattern_offset;
1422 for (a = 0; a < 256/8; a++)
1423 STORE(0);
1424 NEXTCHAR(ch);
1425 if (translate)
1426 ch = translate[(unsigned char)ch];
1427 if (ch == '\136')
1428 {
1429 complement = 1;
1430 NEXTCHAR(ch);
1431 if (translate)
1432 ch = translate[(unsigned char)ch];
1433 }
1434 else
1435 complement = 0;
1436 prev = -1;
1437 range = 0;
1438 firstchar = 1;
1439 while (ch != '\135' || firstchar)
1440 {
1441 firstchar = 0;
1442 if (regexp_ansi_sequences && ch == '\134')
1443 {
1444 NEXTCHAR(ch);
1445 ANSI_TRANSLATE(ch);
1446 }
1447 if (range)
1448 {
1449 for (a = prev; a <= (int)ch; a++)
1450 SETBIT(pattern, offset, a);
1451 prev = -1;
1452 range = 0;
1453 }
1454 else
1455 if (prev != -1 && ch == '-')
1456 range = 1;
1457 else
1458 {
1459 SETBIT(pattern, offset, ch);
1460 prev = ch;
1461 }
1462 NEXTCHAR(ch);
1463 if (translate)
1464 ch = translate[(unsigned char)ch];
1465 }
1466 if (range)
1467 SETBIT(pattern, offset, '-');
1468 if (complement)
1469 {
1470 for (a = 0; a < 256/8; a++)
1471 pattern[offset+a] ^= 0xff;
1472 }
1473 break;
1474 }
1475 case Rbegbuf:
1476 {
1477 opcode = Cbegbuf;
1478 goto store_opcode;
1479 }
1480 case Rendbuf:
1481 {
1482 opcode = Cendbuf;
1483 goto store_opcode;
1484 }
1485 case Rwordchar:
1486 {
1487 opcode = Csyntaxspec;
1488 ch = Sword;
1489 goto store_opcode_and_arg;
1490 }
1491 case Rnotwordchar:
1492 {
1493 opcode = Cnotsyntaxspec;
1494 ch = Sword;
1495 goto store_opcode_and_arg;
1496 }
1497 case Rwordbeg:
1498 {
1499 opcode = Cwordbeg;
1500 goto store_opcode;
1501 }
1502 case Rwordend:
1503 {
1504 opcode = Cwordend;
1505 goto store_opcode;
1506 }
1507 case Rwordbound:
1508 {
1509 opcode = Cwordbound;
1510 goto store_opcode;
1511 }
1512 case Rnotwordbound:
1513 {
1514 opcode = Cnotwordbound;
1515 goto store_opcode;
1516 }
1517 default:
1518 {
1519 abort();
1520 }
1521 }
1522 beginning_context = (op == Ropenpar || op == Ror);
1523 }
1524 if (starts_base != 0)
1525 goto parenthesis_error;
1526 assert(num_jumps == 0);
1527 ALLOC(1);
1528 STORE(Cend);
1529 SET_FIELDS;
1530 if(!re_optimize(bufp))
Guido van Rossumd19c04a1997-09-03 00:47:36 +00001531 return "Optimization error";
Guido van Rossumfaf49081997-07-15 01:47:08 +00001532 return NULL;
Guido van Rossumb674c3b1992-01-19 16:32:47 +00001533
Guido van Rossum004c1e11997-05-09 02:35:58 +00001534 op_error:
Guido van Rossumfaf49081997-07-15 01:47:08 +00001535 SET_FIELDS;
Guido van Rossumd19c04a1997-09-03 00:47:36 +00001536 return "Badly placed special character";
Guido van Rossumb674c3b1992-01-19 16:32:47 +00001537
Guido van Rossum004c1e11997-05-09 02:35:58 +00001538 bad_match_register:
Guido van Rossumfaf49081997-07-15 01:47:08 +00001539 SET_FIELDS;
Guido van Rossumd19c04a1997-09-03 00:47:36 +00001540 return "Bad match register number";
Guido van Rossum004c1e11997-05-09 02:35:58 +00001541
1542 hex_error:
Guido van Rossumfaf49081997-07-15 01:47:08 +00001543 SET_FIELDS;
Guido van Rossumd19c04a1997-09-03 00:47:36 +00001544 return "Bad hexadecimal number";
Guido van Rossum004c1e11997-05-09 02:35:58 +00001545
1546 parenthesis_error:
Guido van Rossumfaf49081997-07-15 01:47:08 +00001547 SET_FIELDS;
Guido van Rossumd19c04a1997-09-03 00:47:36 +00001548 return "Badly placed parenthesis";
Guido van Rossum004c1e11997-05-09 02:35:58 +00001549
1550 out_of_memory:
Guido van Rossumfaf49081997-07-15 01:47:08 +00001551 SET_FIELDS;
Guido van Rossumd19c04a1997-09-03 00:47:36 +00001552 return "Out of memory";
Guido van Rossum004c1e11997-05-09 02:35:58 +00001553
1554 ends_prematurely:
Guido van Rossumfaf49081997-07-15 01:47:08 +00001555 SET_FIELDS;
Guido van Rossumd19c04a1997-09-03 00:47:36 +00001556 return "Regular expression ends prematurely";
Guido van Rossumb674c3b1992-01-19 16:32:47 +00001557
Guido van Rossum004c1e11997-05-09 02:35:58 +00001558 too_complex:
Guido van Rossumfaf49081997-07-15 01:47:08 +00001559 SET_FIELDS;
Guido van Rossumd19c04a1997-09-03 00:47:36 +00001560 return "Regular expression too complex";
Guido van Rossumb674c3b1992-01-19 16:32:47 +00001561}
Guido van Rossum004c1e11997-05-09 02:35:58 +00001562
Guido van Rossumb674c3b1992-01-19 16:32:47 +00001563#undef CHARAT
1564#undef NEXTCHAR
1565#undef GETHEX
1566#undef ALLOC
1567#undef STORE
1568#undef CURRENT_LEVEL_START
1569#undef SET_LEVEL_START
1570#undef PUSH_LEVEL_STARTS
1571#undef POP_LEVEL_STARTS
1572#undef PUT_ADDR
1573#undef INSERT_JUMP
1574#undef SETBIT
1575#undef SET_FIELDS
1576
Guido van Rossum004c1e11997-05-09 02:35:58 +00001577#define PREFETCH if (text == textend) goto fail
1578
1579#define NEXTCHAR(var) \
1580PREFETCH; \
1581var = (unsigned char)*text++; \
1582if (translate) \
Guido van Rossumfaf49081997-07-15 01:47:08 +00001583 var = translate[var]
Guido van Rossum004c1e11997-05-09 02:35:58 +00001584
Guido van Rossum8102c001997-09-05 01:48:48 +00001585int re_match(bufp,
1586 string,
1587 size,
1588 pos,
1589 old_regs)
1590 regexp_t bufp;
1591 unsigned char *string;
1592 int size;
1593 int pos;
1594 regexp_registers_t old_regs;
Guido van Rossumb674c3b1992-01-19 16:32:47 +00001595{
Guido van Rossum95e80531997-08-13 22:34:14 +00001596 unsigned char *code;
1597 unsigned char *translate;
1598 unsigned char *text;
1599 unsigned char *textstart;
1600 unsigned char *textend;
Guido van Rossumfaf49081997-07-15 01:47:08 +00001601 int a;
1602 int b;
1603 int ch;
1604 int reg;
1605 int match_end;
Guido van Rossum95e80531997-08-13 22:34:14 +00001606 unsigned char *regstart;
1607 unsigned char *regend;
Guido van Rossumfaf49081997-07-15 01:47:08 +00001608 int regsize;
1609 match_state state;
Guido van Rossum004c1e11997-05-09 02:35:58 +00001610
Guido van Rossumfaf49081997-07-15 01:47:08 +00001611 assert(pos >= 0 && size >= 0);
1612 assert(pos <= size);
Guido van Rossum004c1e11997-05-09 02:35:58 +00001613
Guido van Rossumfaf49081997-07-15 01:47:08 +00001614 text = string + pos;
1615 textstart = string;
1616 textend = string + size;
Guido van Rossum004c1e11997-05-09 02:35:58 +00001617
Guido van Rossumfaf49081997-07-15 01:47:08 +00001618 code = bufp->buffer;
Guido van Rossum004c1e11997-05-09 02:35:58 +00001619
Guido van Rossumfaf49081997-07-15 01:47:08 +00001620 translate = bufp->translate;
Guido van Rossum004c1e11997-05-09 02:35:58 +00001621
Guido van Rossumfaf49081997-07-15 01:47:08 +00001622 NEW_STATE(state, bufp->num_registers);
1623
Guido van Rossum004c1e11997-05-09 02:35:58 +00001624 continue_matching:
Guido van Rossumfaf49081997-07-15 01:47:08 +00001625 switch (*code++)
Guido van Rossumb674c3b1992-01-19 16:32:47 +00001626 {
Guido van Rossumfaf49081997-07-15 01:47:08 +00001627 case Cend:
Guido van Rossum004c1e11997-05-09 02:35:58 +00001628 {
Guido van Rossumfaf49081997-07-15 01:47:08 +00001629 match_end = text - textstart;
1630 if (old_regs)
1631 {
1632 old_regs->start[0] = pos;
1633 old_regs->end[0] = match_end;
1634 if (!bufp->uses_registers)
1635 {
1636 for (a = 1; a < RE_NREGS; a++)
1637 {
1638 old_regs->start[a] = -1;
1639 old_regs->end[a] = -1;
1640 }
1641 }
1642 else
1643 {
1644 for (a = 1; a < bufp->num_registers; a++)
1645 {
1646 if ((GET_REG_START(state, a) == NULL) ||
1647 (GET_REG_END(state, a) == NULL))
1648 {
1649 old_regs->start[a] = -1;
1650 old_regs->end[a] = -1;
1651 continue;
1652 }
1653 old_regs->start[a] = GET_REG_START(state, a) - textstart;
1654 old_regs->end[a] = GET_REG_END(state, a) - textstart;
1655 }
1656 for (; a < RE_NREGS; a++)
1657 {
1658 old_regs->start[a] = -1;
1659 old_regs->end[a] = -1;
1660 }
1661 }
1662 }
1663 FREE_STATE(state);
1664 return match_end - pos;
Guido van Rossum004c1e11997-05-09 02:35:58 +00001665 }
Guido van Rossumfaf49081997-07-15 01:47:08 +00001666 case Cbol:
1667 {
1668 if (text == textstart || text[-1] == '\n')
1669 goto continue_matching;
1670 goto fail;
1671 }
1672 case Ceol:
1673 {
1674 if (text == textend || *text == '\n')
1675 goto continue_matching;
1676 goto fail;
1677 }
1678 case Cset:
1679 {
1680 NEXTCHAR(ch);
1681 if (code[ch/8] & (1<<(ch & 7)))
1682 {
1683 code += 256/8;
1684 goto continue_matching;
1685 }
1686 goto fail;
1687 }
1688 case Cexact:
1689 {
1690 NEXTCHAR(ch);
1691 if (ch != (unsigned char)*code++)
1692 goto fail;
1693 goto continue_matching;
1694 }
1695 case Canychar:
1696 {
1697 NEXTCHAR(ch);
1698 if (ch == '\n')
1699 goto fail;
1700 goto continue_matching;
1701 }
1702 case Cstart_memory:
1703 {
1704 reg = *code++;
1705 SET_REG_START(state, reg, text, goto error);
1706 goto continue_matching;
1707 }
1708 case Cend_memory:
1709 {
1710 reg = *code++;
1711 SET_REG_END(state, reg, text, goto error);
1712 goto continue_matching;
1713 }
1714 case Cmatch_memory:
1715 {
1716 reg = *code++;
1717 regstart = GET_REG_START(state, reg);
1718 regend = GET_REG_END(state, reg);
1719 if ((regstart == NULL) || (regend == NULL))
1720 goto fail; /* or should we just match nothing? */
1721 regsize = regend - regstart;
Guido van Rossum004c1e11997-05-09 02:35:58 +00001722
Guido van Rossumfaf49081997-07-15 01:47:08 +00001723 if (regsize > (textend - text))
1724 goto fail;
1725 if(translate)
1726 {
1727 for (; regstart < regend; regstart++, text++)
1728 if (translate[*regstart] != translate[*text])
1729 goto fail;
1730 }
1731 else
1732 for (; regstart < regend; regstart++, text++)
1733 if (*regstart != *text)
1734 goto fail;
1735 goto continue_matching;
Guido van Rossum004c1e11997-05-09 02:35:58 +00001736 }
Guido van Rossumfaf49081997-07-15 01:47:08 +00001737 case Cupdate_failure_jump:
Guido van Rossumdb25f321997-07-10 14:31:32 +00001738 {
Guido van Rossumfaf49081997-07-15 01:47:08 +00001739 UPDATE_FAILURE(state, text, goto error);
1740 /* fall to next case */
Guido van Rossumdb25f321997-07-10 14:31:32 +00001741 }
Guido van Rossumfaf49081997-07-15 01:47:08 +00001742 /* treat Cstar_jump just like Cjump if it hasn't been optimized */
1743 case Cstar_jump:
1744 case Cjump:
1745 {
1746 a = (unsigned char)*code++;
1747 a |= (unsigned char)*code++ << 8;
1748 code += (int)SHORT(a);
Guido van Rossum95e80531997-08-13 22:34:14 +00001749 if (code<bufp->buffer || bufp->buffer+bufp->used<code) {
1750 PyErr_SetString(PyExc_SystemError, "Regex VM jump out of bounds (Cjump)");
1751 FREE_STATE(state);
1752 return -2;
1753 }
Guido van Rossumfaf49081997-07-15 01:47:08 +00001754 goto continue_matching;
1755 }
1756 case Cdummy_failure_jump:
1757 {
Guido van Rossum95e80531997-08-13 22:34:14 +00001758 unsigned char *failuredest;
1759
Guido van Rossumfaf49081997-07-15 01:47:08 +00001760 a = (unsigned char)*code++;
1761 a |= (unsigned char)*code++ << 8;
1762 a = (int)SHORT(a);
1763 assert(*code == Cfailure_jump);
1764 b = (unsigned char)code[1];
1765 b |= (unsigned char)code[2] << 8;
Guido van Rossum95e80531997-08-13 22:34:14 +00001766 failuredest = code + (int)SHORT(b) + 3;
1767 if (failuredest<bufp->buffer || bufp->buffer+bufp->used < failuredest) {
1768 PyErr_SetString(PyExc_SystemError, "Regex VM jump out of bounds (Cdummy_failure_jump failuredest)");
1769 FREE_STATE(state);
1770 return -2;
1771 }
1772 PUSH_FAILURE(state, failuredest, NULL, goto error);
Guido van Rossumfaf49081997-07-15 01:47:08 +00001773 code += a;
Guido van Rossum95e80531997-08-13 22:34:14 +00001774 if (code<bufp->buffer || bufp->buffer+bufp->used < code) {
1775 PyErr_SetString(PyExc_SystemError, "Regex VM jump out of bounds (Cdummy_failure_jump code)");
1776 FREE_STATE(state);
1777 return -2;
1778 }
Guido van Rossumfaf49081997-07-15 01:47:08 +00001779 goto continue_matching;
1780 }
1781 case Cfailure_jump:
1782 {
1783 a = (unsigned char)*code++;
1784 a |= (unsigned char)*code++ << 8;
1785 a = (int)SHORT(a);
Guido van Rossum95e80531997-08-13 22:34:14 +00001786 if (code+a<bufp->buffer || bufp->buffer+bufp->used < code+a) {
1787 PyErr_SetString(PyExc_SystemError, "Regex VM jump out of bounds (Cfailure_jump)");
1788 FREE_STATE(state);
1789 return -2;
1790 }
Guido van Rossumfaf49081997-07-15 01:47:08 +00001791 PUSH_FAILURE(state, code + a, text, goto error);
1792 goto continue_matching;
1793 }
1794 case Crepeat1:
1795 {
Guido van Rossum95e80531997-08-13 22:34:14 +00001796 unsigned char *pinst;
Guido van Rossumfaf49081997-07-15 01:47:08 +00001797 a = (unsigned char)*code++;
1798 a |= (unsigned char)*code++ << 8;
1799 a = (int)SHORT(a);
1800 pinst = code + a;
Guido van Rossum95e80531997-08-13 22:34:14 +00001801 if (pinst<bufp->buffer || bufp->buffer+bufp->used<pinst) {
1802 PyErr_SetString(PyExc_SystemError, "Regex VM jump out of bounds (Crepeat1)");
1803 FREE_STATE(state);
1804 return -2;
1805 }
Guido van Rossumfaf49081997-07-15 01:47:08 +00001806 /* pinst is sole instruction in loop, and it matches a
1807 * single character. Since Crepeat1 was originally a
1808 * Cupdate_failure_jump, we also know that backtracking
1809 * is useless: so long as the single-character
1810 * expression matches, it must be used. Also, in the
1811 * case of +, we've already matched one character, so +
1812 * can't fail: nothing here can cause a failure. */
1813 switch (*pinst++)
1814 {
1815 case Cset:
Guido van Rossum95e80531997-08-13 22:34:14 +00001816 {
1817 if (translate)
Guido van Rossumfaf49081997-07-15 01:47:08 +00001818 {
1819 while (text < textend)
1820 {
1821 ch = translate[(unsigned char)*text];
1822 if (pinst[ch/8] & (1<<(ch & 7)))
1823 text++;
1824 else
1825 break;
1826 }
1827 }
1828 else
1829 {
1830 while (text < textend)
1831 {
1832 ch = (unsigned char)*text;
1833 if (pinst[ch/8] & (1<<(ch & 7)))
1834 text++;
1835 else
1836 break;
1837 }
1838 }
1839 break;
Guido van Rossum95e80531997-08-13 22:34:14 +00001840 }
Guido van Rossumfaf49081997-07-15 01:47:08 +00001841 case Cexact:
1842 {
1843 ch = (unsigned char)*pinst;
1844 if (translate)
1845 {
1846 while (text < textend &&
1847 translate[(unsigned char)*text] == ch)
1848 text++;
1849 }
1850 else
1851 {
1852 while (text < textend && (unsigned char)*text == ch)
1853 text++;
1854 }
1855 break;
1856 }
1857 case Canychar:
1858 {
1859 while (text < textend && (unsigned char)*text != '\n')
1860 text++;
1861 break;
1862 }
1863 case Csyntaxspec:
1864 {
1865 a = (unsigned char)*pinst;
1866 if (translate)
1867 {
1868 while (text < textend &&
Guido van Rossume59d3f81997-12-02 20:39:23 +00001869 (SYNTAX(translate[*text]) & a) )
Guido van Rossumfaf49081997-07-15 01:47:08 +00001870 text++;
1871 }
1872 else
1873 {
Guido van Rossume59d3f81997-12-02 20:39:23 +00001874 while (text < textend && (SYNTAX(*text) & a) )
Guido van Rossumfaf49081997-07-15 01:47:08 +00001875 text++;
1876 }
1877 break;
1878 }
1879 case Cnotsyntaxspec:
1880 {
1881 a = (unsigned char)*pinst;
1882 if (translate)
1883 {
1884 while (text < textend &&
Guido van Rossume59d3f81997-12-02 20:39:23 +00001885 !(SYNTAX(translate[*text]) & a) )
Guido van Rossumfaf49081997-07-15 01:47:08 +00001886 text++;
1887 }
1888 else
1889 {
Guido van Rossume59d3f81997-12-02 20:39:23 +00001890 while (text < textend && !(SYNTAX(*text) & a) )
Guido van Rossumfaf49081997-07-15 01:47:08 +00001891 text++;
1892 }
1893 break;
1894 }
1895 default:
1896 {
Guido van Rossum95e80531997-08-13 22:34:14 +00001897 FREE_STATE(state);
1898 PyErr_SetString(PyExc_SystemError, "Unknown regex opcode: memory corrupted?");
1899 return -2;
Guido van Rossumfaf49081997-07-15 01:47:08 +00001900 /*NOTREACHED*/
1901 }
1902 }
1903 /* due to the funky way + and * are compiled, the top
1904 * failure- stack entry at this point is actually a
1905 * success entry -- update it & pop it */
1906 UPDATE_FAILURE(state, text, goto error);
1907 goto fail; /* i.e., succeed <wink/sigh> */
1908 }
1909 case Cbegbuf:
1910 {
1911 if (text == textstart)
1912 goto continue_matching;
1913 goto fail;
1914 }
1915 case Cendbuf:
1916 {
1917 if (text == textend)
1918 goto continue_matching;
1919 goto fail;
1920 }
1921 case Cwordbeg:
1922 {
1923 if (text == textend)
1924 goto fail;
Guido van Rossum95e80531997-08-13 22:34:14 +00001925 if (!(SYNTAX(*text) & Sword))
Guido van Rossumfaf49081997-07-15 01:47:08 +00001926 goto fail;
1927 if (text == textstart)
1928 goto continue_matching;
Guido van Rossum74fb3031997-07-17 22:41:38 +00001929 if (!(SYNTAX(text[-1]) & Sword))
Guido van Rossumfaf49081997-07-15 01:47:08 +00001930 goto continue_matching;
1931 goto fail;
1932 }
1933 case Cwordend:
1934 {
1935 if (text == textstart)
1936 goto fail;
Guido van Rossum74fb3031997-07-17 22:41:38 +00001937 if (!(SYNTAX(text[-1]) & Sword))
Guido van Rossumfaf49081997-07-15 01:47:08 +00001938 goto fail;
1939 if (text == textend)
1940 goto continue_matching;
Guido van Rossum95e80531997-08-13 22:34:14 +00001941 if (!(SYNTAX(*text) & Sword))
1942 goto continue_matching;
1943 goto fail;
Guido van Rossumfaf49081997-07-15 01:47:08 +00001944 }
1945 case Cwordbound:
1946 {
1947 /* Note: as in gnu regexp, this also matches at the
1948 * beginning and end of buffer. */
Guido van Rossum004c1e11997-05-09 02:35:58 +00001949
Guido van Rossumfaf49081997-07-15 01:47:08 +00001950 if (text == textstart || text == textend)
1951 goto continue_matching;
Guido van Rossum74fb3031997-07-17 22:41:38 +00001952 if ((SYNTAX(text[-1]) & Sword) ^ (SYNTAX(*text) & Sword))
Guido van Rossumfaf49081997-07-15 01:47:08 +00001953 goto continue_matching;
1954 goto fail;
1955 }
1956 case Cnotwordbound:
1957 {
1958 /* Note: as in gnu regexp, this never matches at the
1959 * beginning and end of buffer. */
1960 if (text == textstart || text == textend)
1961 goto fail;
Guido van Rossum74fb3031997-07-17 22:41:38 +00001962 if (!((SYNTAX(text[-1]) & Sword) ^ (SYNTAX(*text) & Sword)))
Guido van Rossum53665e51997-08-15 15:45:25 +00001963 goto continue_matching;
1964 goto fail;
Guido van Rossumfaf49081997-07-15 01:47:08 +00001965 }
1966 case Csyntaxspec:
1967 {
1968 NEXTCHAR(ch);
Guido van Rossum74fb3031997-07-17 22:41:38 +00001969 if (!(SYNTAX(ch) & (unsigned char)*code++))
Guido van Rossumfaf49081997-07-15 01:47:08 +00001970 goto fail;
1971 goto continue_matching;
1972 }
1973 case Cnotsyntaxspec:
1974 {
1975 NEXTCHAR(ch);
Guido van Rossum74fb3031997-07-17 22:41:38 +00001976 if (SYNTAX(ch) & (unsigned char)*code++)
Guido van Rossum95e80531997-08-13 22:34:14 +00001977 goto fail;
Guido van Rossumfaf49081997-07-15 01:47:08 +00001978 goto continue_matching;
1979 }
1980 default:
1981 {
Guido van Rossum95e80531997-08-13 22:34:14 +00001982 FREE_STATE(state);
1983 PyErr_SetString(PyExc_SystemError, "Unknown regex opcode: memory corrupted?");
1984 return -2;
Guido van Rossumfaf49081997-07-15 01:47:08 +00001985 /*NOTREACHED*/
1986 }
1987 }
Guido van Rossum95e80531997-08-13 22:34:14 +00001988
1989
Guido van Rossum004c1e11997-05-09 02:35:58 +00001990
Guido van Rossum3b1a57a1992-01-27 16:47:46 +00001991#if 0 /* This line is never reached --Guido */
Guido van Rossumfaf49081997-07-15 01:47:08 +00001992 abort();
Guido van Rossum5f21dd11992-01-19 16:49:14 +00001993#endif
Guido van Rossumfaf49081997-07-15 01:47:08 +00001994 /*
1995 *NOTREACHED
1996 */
Guido van Rossum95e80531997-08-13 22:34:14 +00001997
1998 /* Using "break;" in the above switch statement is equivalent to "goto fail;" */
Guido van Rossum004c1e11997-05-09 02:35:58 +00001999 fail:
Guido van Rossumfaf49081997-07-15 01:47:08 +00002000 POP_FAILURE(state, code, text, goto done_matching, goto error);
2001 goto continue_matching;
Guido van Rossum004c1e11997-05-09 02:35:58 +00002002
2003 done_matching:
2004/* if(translated != NULL) */
2005/* free(translated); */
Guido van Rossumfaf49081997-07-15 01:47:08 +00002006 FREE_STATE(state);
2007 return -1;
Guido van Rossumb674c3b1992-01-19 16:32:47 +00002008
Guido van Rossum004c1e11997-05-09 02:35:58 +00002009 error:
2010/* if (translated != NULL) */
2011/* free(translated); */
Guido van Rossumfaf49081997-07-15 01:47:08 +00002012 FREE_STATE(state);
2013 return -2;
Guido van Rossumb674c3b1992-01-19 16:32:47 +00002014}
Guido van Rossum95e80531997-08-13 22:34:14 +00002015
Guido van Rossumb674c3b1992-01-19 16:32:47 +00002016
2017#undef PREFETCH
2018#undef NEXTCHAR
Guido van Rossumb674c3b1992-01-19 16:32:47 +00002019
Guido van Rossum8102c001997-09-05 01:48:48 +00002020int re_search(bufp,
2021 string,
2022 size,
2023 pos,
2024 range,
2025 regs)
2026 regexp_t bufp;
2027 unsigned char *string;
2028 int size;
2029 int pos;
2030 int range;
2031 regexp_registers_t regs;
Guido van Rossumb674c3b1992-01-19 16:32:47 +00002032{
Guido van Rossum95e80531997-08-13 22:34:14 +00002033 unsigned char *fastmap;
2034 unsigned char *translate;
2035 unsigned char *text;
2036 unsigned char *partstart;
2037 unsigned char *partend;
Guido van Rossumfaf49081997-07-15 01:47:08 +00002038 int dir;
2039 int ret;
Guido van Rossum95e80531997-08-13 22:34:14 +00002040 unsigned char anchor;
Guido van Rossumb674c3b1992-01-19 16:32:47 +00002041
Guido van Rossumfaf49081997-07-15 01:47:08 +00002042 assert(size >= 0 && pos >= 0);
2043 assert(pos + range >= 0 && pos + range <= size); /* Bugfix by ylo */
Guido van Rossumb674c3b1992-01-19 16:32:47 +00002044
Guido van Rossumfaf49081997-07-15 01:47:08 +00002045 fastmap = bufp->fastmap;
2046 translate = bufp->translate;
Guido van Rossum95e80531997-08-13 22:34:14 +00002047 if (fastmap && !bufp->fastmap_accurate) {
2048 re_compile_fastmap(bufp);
2049 if (PyErr_Occurred()) return -2;
2050 }
2051
Guido van Rossumfaf49081997-07-15 01:47:08 +00002052 anchor = bufp->anchor;
2053 if (bufp->can_be_null == 1) /* can_be_null == 2: can match null at eob */
2054 fastmap = NULL;
Guido van Rossum004c1e11997-05-09 02:35:58 +00002055
Guido van Rossumfaf49081997-07-15 01:47:08 +00002056 if (range < 0)
2057 {
2058 dir = -1;
2059 range = -range;
Guido van Rossumb674c3b1992-01-19 16:32:47 +00002060 }
Guido van Rossum004c1e11997-05-09 02:35:58 +00002061 else
Guido van Rossumfaf49081997-07-15 01:47:08 +00002062 dir = 1;
2063
2064 if (anchor == 2)
2065 if (pos != 0)
2066 return -1;
2067 else
2068 range = 0;
2069
2070 for (; range >= 0; range--, pos += dir)
2071 {
2072 if (fastmap)
2073 {
2074 if (dir == 1)
2075 { /* searching forwards */
2076
2077 text = string + pos;
2078 partend = string + size;
2079 partstart = text;
2080 if (translate)
2081 while (text != partend &&
2082 !fastmap[(unsigned char) translate[(unsigned char)*text]])
2083 text++;
2084 else
2085 while (text != partend && !fastmap[(unsigned char)*text])
2086 text++;
2087 pos += text - partstart;
2088 range -= text - partstart;
2089 if (pos == size && bufp->can_be_null == 0)
2090 return -1;
2091 }
2092 else
2093 { /* searching backwards */
2094 text = string + pos;
2095 partstart = string + pos - range;
2096 partend = text;
2097 if (translate)
2098 while (text != partstart &&
2099 !fastmap[(unsigned char)
2100 translate[(unsigned char)*text]])
2101 text--;
2102 else
2103 while (text != partstart &&
2104 !fastmap[(unsigned char)*text])
2105 text--;
2106 pos -= partend - text;
2107 range -= partend - text;
2108 }
2109 }
2110 if (anchor == 1)
2111 { /* anchored to begline */
2112 if (pos > 0 && (string[pos - 1] != '\n'))
2113 continue;
2114 }
2115 assert(pos >= 0 && pos <= size);
2116 ret = re_match(bufp, string, size, pos, regs);
2117 if (ret >= 0)
2118 return pos;
2119 if (ret == -2)
2120 return -2;
Guido van Rossumb674c3b1992-01-19 16:32:47 +00002121 }
Guido van Rossumfaf49081997-07-15 01:47:08 +00002122 return -1;
Guido van Rossumb674c3b1992-01-19 16:32:47 +00002123}
Guido van Rossum74fb3031997-07-17 22:41:38 +00002124
2125/*
2126** Local Variables:
2127** mode: c
2128** c-file-style: "python"
2129** End:
2130*/