regexec.c 26.7 KB
Newer Older
1 2
/*
 * re_*exec and friends - match REs
3
 *
B
Bruce Momjian 已提交
4 5
 * Copyright (c) 1998, 1999 Henry Spencer.	All rights reserved.
 *
6 7 8
 * Development of this software was funded, in part, by Cray Research Inc.,
 * UUNET Communications Services Inc., Sun Microsystems Inc., and Scriptics
 * Corporation, none of whom are responsible for the results.  The author
B
Bruce Momjian 已提交
9 10
 * thanks all of them.
 *
11 12 13 14
 * Redistribution and use in source and binary forms -- with or without
 * modification -- are permitted for any purpose, provided that
 * redistributions in source form retain this entire copyright notice and
 * indicate the origin and nature of any modifications.
B
Bruce Momjian 已提交
15
 *
16 17
 * I'd appreciate being given credit for this package in the documentation
 * of software which uses it, but that is not a requirement.
B
Bruce Momjian 已提交
18
 *
19 20 21 22 23 24 25 26 27 28
 * THIS SOFTWARE IS PROVIDED ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES,
 * INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY
 * AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL
 * HENRY SPENCER BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
 * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
 * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS;
 * OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
 * WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR
 * OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
 * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
29
 *
30
 * src/backend/regex/regexec.c
31 32 33
 *
 */

34 35 36 37 38
#include "regex/regguts.h"



/* lazy-DFA representation */
B
Bruce Momjian 已提交
39 40
struct arcp
{								/* "pointer" to an outarc */
41
	struct sset *ss;
B
Bruce Momjian 已提交
42
	color		co;
43 44
};

B
Bruce Momjian 已提交
45 46 47 48 49 50
struct sset
{								/* state set */
	unsigned   *states;			/* pointer to bitvector */
	unsigned	hash;			/* hash of bitvector */
#define  HASH(bv, nw)	 (((nw) == 1) ? *(bv) : hash(bv, nw))
#define  HIT(h,bv,ss,nw) ((ss)->hash == (h) && ((nw) == 1 || \
51
		memcmp(VS(bv), VS((ss)->states), (nw)*sizeof(unsigned)) == 0))
B
Bruce Momjian 已提交
52 53 54 55 56 57 58 59 60
	int			flags;
#define  STARTER	 01			/* the initial state set */
#define  POSTSTATE	 02			/* includes the goal state */
#define  LOCKED		 04			/* locked in cache */
#define  NOPROGRESS  010		/* zero-progress state set */
	struct arcp ins;			/* chain of inarcs pointing here */
	chr		   *lastseen;		/* last entered on arrival here */
	struct sset **outs;			/* outarc vector indexed by color */
	struct arcp *inchain;		/* chain-pointer vector for outarcs */
61 62
};

B
Bruce Momjian 已提交
63 64 65 66 67 68 69 70 71 72 73 74
struct dfa
{
	int			nssets;			/* size of cache */
	int			nssused;		/* how many entries occupied yet */
	int			nstates;		/* number of states */
	int			ncolors;		/* length of outarc and inchain vectors */
	int			wordsper;		/* length of state-set bitvectors */
	struct sset *ssets;			/* state-set cache */
	unsigned   *statesarea;		/* bitvector storage */
	unsigned   *work;			/* pointer to work area within statesarea */
	struct sset **outsarea;		/* outarc-vector storage */
	struct arcp *incarea;		/* inchain storage */
75 76
	struct cnfa *cnfa;
	struct colormap *cm;
B
Bruce Momjian 已提交
77
	chr		   *lastpost;		/* location of last cache-flushed success */
B
Bruce Momjian 已提交
78
	chr		   *lastnopr;		/* location of last cache-flushed NOPROGRESS */
B
Bruce Momjian 已提交
79 80 81
	struct sset *search;		/* replacement-search-pointer memory */
	int			cptsmalloced;	/* were the areas individually malloced? */
	char	   *mallocarea;		/* self, or master malloced area, or NULL */
82 83
};

B
Bruce Momjian 已提交
84
#define WORK	1				/* number of work bitvectors needed */
85 86

/* setup for non-malloc allocation for small cases */
B
Bruce Momjian 已提交
87 88 89 90 91 92 93 94 95
#define FEWSTATES	20			/* must be less than UBITS */
#define FEWCOLORS	15
struct smalldfa
{
	struct dfa	dfa;
	struct sset ssets[FEWSTATES * 2];
	unsigned	statesarea[FEWSTATES * 2 + WORK];
	struct sset *outsarea[FEWSTATES * 2 * FEWCOLORS];
	struct arcp incarea[FEWSTATES * 2 * FEWCOLORS];
96
};
B
Bruce Momjian 已提交
97 98

#define DOMALLOC	((struct smalldfa *)NULL)	/* force malloc */
99 100 101 102



/* internal variables, bundled for easy passing around */
B
Bruce Momjian 已提交
103 104 105
struct vars
{
	regex_t    *re;
106
	struct guts *g;
B
Bruce Momjian 已提交
107 108
	int			eflags;			/* copies of arguments */
	size_t		nmatch;
109 110
	regmatch_t *pmatch;
	rm_detail_t *details;
B
Bruce Momjian 已提交
111
	chr		   *start;			/* start of string */
112
	chr		   *search_start;	/* search start of string */
B
Bruce Momjian 已提交
113 114 115
	chr		   *stop;			/* just past end of string */
	int			err;			/* error code if any (0 none) */
	regoff_t   *mem;			/* memory vector for backtracking */
116 117 118 119
	struct smalldfa dfa1;
	struct smalldfa dfa2;
};

B
Bruce Momjian 已提交
120 121
#define VISERR(vv)	((vv)->err != 0)	/* have we seen an error yet? */
#define ISERR() VISERR(v)
122
#define VERR(vv,e)	((vv)->err = ((vv)->err ? (vv)->err : (e)))
B
Bruce Momjian 已提交
123
#define ERR(e)	VERR(v, e)		/* record an error */
B
Bruce Momjian 已提交
124
#define NOERR() {if (ISERR()) return v->err;}	/* if error seen, return it */
B
Bruce Momjian 已提交
125 126
#define OFF(p)	((p) - v->start)
#define LOFF(p) ((long)OFF(p))
127 128


129

130
/*
131 132 133
 * forward declarations
 */
/* === regexec.c === */
B
Bruce Momjian 已提交
134 135 136 137 138 139 140 141 142 143 144 145 146 147 148
static int	find(struct vars *, struct cnfa *, struct colormap *);
static int	cfind(struct vars *, struct cnfa *, struct colormap *);
static int	cfindloop(struct vars *, struct cnfa *, struct colormap *, struct dfa *, struct dfa *, chr **);
static void zapsubs(regmatch_t *, size_t);
static void zapmem(struct vars *, struct subre *);
static void subset(struct vars *, struct subre *, chr *, chr *);
static int	dissect(struct vars *, struct subre *, chr *, chr *);
static int	condissect(struct vars *, struct subre *, chr *, chr *);
static int	altdissect(struct vars *, struct subre *, chr *, chr *);
static int	cdissect(struct vars *, struct subre *, chr *, chr *);
static int	ccondissect(struct vars *, struct subre *, chr *, chr *);
static int	crevdissect(struct vars *, struct subre *, chr *, chr *);
static int	cbrdissect(struct vars *, struct subre *, chr *, chr *);
static int	caltdissect(struct vars *, struct subre *, chr *, chr *);

149
/* === rege_dfa.c === */
B
Bruce Momjian 已提交
150 151 152 153 154 155 156 157 158 159 160
static chr *longest(struct vars *, struct dfa *, chr *, chr *, int *);
static chr *shortest(struct vars *, struct dfa *, chr *, chr *, chr *, chr **, int *);
static chr *lastcold(struct vars *, struct dfa *);
static struct dfa *newdfa(struct vars *, struct cnfa *, struct colormap *, struct smalldfa *);
static void freedfa(struct dfa *);
static unsigned hash(unsigned *, int);
static struct sset *initialize(struct vars *, struct dfa *, chr *);
static struct sset *miss(struct vars *, struct dfa *, struct sset *, pcolor, chr *, chr *);
static int	lacon(struct vars *, struct cnfa *, chr *, pcolor);
static struct sset *getvacant(struct vars *, struct dfa *, chr *, chr *);
static struct sset *pickss(struct vars *, struct dfa *, chr *, chr *);
161 162 163 164 165 166 167


/*
 * pg_regexec - match regular expression
 */
int
pg_regexec(regex_t *re,
168
		   const chr *string,
169
		   size_t len,
170
		   size_t search_start,
171
		   rm_detail_t *details,
172 173 174 175 176 177
		   size_t nmatch,
		   regmatch_t pmatch[],
		   int flags)
{
	struct vars var;
	register struct vars *v = &var;
B
Bruce Momjian 已提交
178 179 180 181 182 183 184 185 186
	int			st;
	size_t		n;
	int			backref;

#define  LOCALMAT	 20
	regmatch_t	mat[LOCALMAT];

#define  LOCALMEM	 40
	regoff_t	mem[LOCALMEM];
187 188 189 190 191 192 193

	/* sanity checks */
	if (re == NULL || string == NULL || re->re_magic != REMAGIC)
		return REG_INVARG;
	if (re->re_csize != sizeof(chr))
		return REG_MIXED;

194 195 196
	/* Initialize locale-dependent support */
	pg_set_regex_collation(re->re_collation);

197 198
	/* setup */
	v->re = re;
B
Bruce Momjian 已提交
199 200
	v->g = (struct guts *) re->re_guts;
	if ((v->g->cflags & REG_EXPECT) && details == NULL)
201
		return REG_INVARG;
B
Bruce Momjian 已提交
202
	if (v->g->info & REG_UIMPOSSIBLE)
203
		return REG_NOMATCH;
B
Bruce Momjian 已提交
204
	backref = (v->g->info & REG_UBACKREF) ? 1 : 0;
205
	v->eflags = flags;
B
Bruce Momjian 已提交
206 207
	if (v->g->cflags & REG_NOSUB)
		nmatch = 0;				/* override client */
208
	v->nmatch = nmatch;
B
Bruce Momjian 已提交
209 210
	if (backref)
	{
211 212 213 214
		/* need work area */
		if (v->g->nsub + 1 <= LOCALMAT)
			v->pmatch = mat;
		else
B
Bruce Momjian 已提交
215 216
			v->pmatch = (regmatch_t *) MALLOC((v->g->nsub + 1) *
											  sizeof(regmatch_t));
217 218 219
		if (v->pmatch == NULL)
			return REG_ESPACE;
		v->nmatch = v->g->nsub + 1;
B
Bruce Momjian 已提交
220 221
	}
	else
222 223
		v->pmatch = pmatch;
	v->details = details;
B
Bruce Momjian 已提交
224
	v->start = (chr *) string;
225
	v->search_start = (chr *) string + search_start;
B
Bruce Momjian 已提交
226
	v->stop = (chr *) string + len;
227
	v->err = 0;
B
Bruce Momjian 已提交
228 229
	if (backref)
	{
230 231
		/* need retry memory */
		assert(v->g->ntree >= 0);
B
Bruce Momjian 已提交
232
		n = (size_t) v->g->ntree;
233 234 235
		if (n <= LOCALMEM)
			v->mem = mem;
		else
B
Bruce Momjian 已提交
236 237 238
			v->mem = (regoff_t *) MALLOC(n * sizeof(regoff_t));
		if (v->mem == NULL)
		{
239 240 241 242
			if (v->pmatch != pmatch && v->pmatch != mat)
				FREE(v->pmatch);
			return REG_ESPACE;
		}
B
Bruce Momjian 已提交
243 244
	}
	else
245 246 247 248 249 250 251 252 253 254
		v->mem = NULL;

	/* do it */
	assert(v->g->tree != NULL);
	if (backref)
		st = cfind(v, &v->g->tree->cnfa, &v->g->cmap);
	else
		st = find(v, &v->g->tree->cnfa, &v->g->cmap);

	/* copy (portion of) match vector over if necessary */
B
Bruce Momjian 已提交
255 256
	if (st == REG_OKAY && v->pmatch != pmatch && nmatch > 0)
	{
257 258
		zapsubs(pmatch, nmatch);
		n = (nmatch < v->nmatch) ? nmatch : v->nmatch;
B
Bruce Momjian 已提交
259
		memcpy(VS(pmatch), VS(v->pmatch), n * sizeof(regmatch_t));
260 261 262 263 264 265 266 267 268 269 270 271 272 273
	}

	/* clean up */
	if (v->pmatch != pmatch && v->pmatch != mat)
		FREE(v->pmatch);
	if (v->mem != NULL && v->mem != mem)
		FREE(v->mem);
	return st;
}

/*
 * find - find a match for the main NFA (no-complications case)
 */
static int
B
Bruce Momjian 已提交
274 275 276
find(struct vars * v,
	 struct cnfa * cnfa,
	 struct colormap * cm)
277 278 279
{
	struct dfa *s;
	struct dfa *d;
B
Bruce Momjian 已提交
280 281 282
	chr		   *begin;
	chr		   *end = NULL;
	chr		   *cold;
B
Bruce Momjian 已提交
283
	chr		   *open;			/* open and close of range of possible starts */
B
Bruce Momjian 已提交
284 285 286
	chr		   *close;
	int			hitend;
	int			shorter = (v->g->tree->flags & SHORTER) ? 1 : 0;
287 288 289 290 291 292 293

	/* first, a shot with the search RE */
	s = newdfa(v, &v->g->search, cm, &v->dfa1);
	assert(!(ISERR() && s != NULL));
	NOERR();
	MDEBUG(("\nsearch at %ld\n", LOFF(v->start)));
	cold = NULL;
294 295
	close = shortest(v, s, v->search_start, v->search_start, v->stop,
					 &cold, (int *) NULL);
296 297
	freedfa(s);
	NOERR();
B
Bruce Momjian 已提交
298 299
	if (v->g->cflags & REG_EXPECT)
	{
300 301 302 303 304
		assert(v->details != NULL);
		if (cold != NULL)
			v->details->rm_extend.rm_so = OFF(cold);
		else
			v->details->rm_extend.rm_so = OFF(v->stop);
B
Bruce Momjian 已提交
305
		v->details->rm_extend.rm_eo = OFF(v->stop);		/* unknown */
306
	}
B
Bruce Momjian 已提交
307
	if (close == NULL)			/* not found */
308
		return REG_NOMATCH;
B
Bruce Momjian 已提交
309
	if (v->nmatch == 0)			/* found, don't need exact location */
310 311 312 313 314 315 316 317 318 319
		return REG_OKAY;

	/* find starting point and match */
	assert(cold != NULL);
	open = cold;
	cold = NULL;
	MDEBUG(("between %ld and %ld\n", LOFF(open), LOFF(close)));
	d = newdfa(v, cnfa, cm, &v->dfa1);
	assert(!(ISERR() && d != NULL));
	NOERR();
B
Bruce Momjian 已提交
320 321
	for (begin = open; begin <= close; begin++)
	{
322 323 324
		MDEBUG(("\nfind trying at %ld\n", LOFF(begin)));
		if (shorter)
			end = shortest(v, d, begin, begin, v->stop,
B
Bruce Momjian 已提交
325
						   (chr **) NULL, &hitend);
326 327 328 329 330 331
		else
			end = longest(v, d, begin, v->stop, &hitend);
		NOERR();
		if (hitend && cold == NULL)
			cold = begin;
		if (end != NULL)
B
Bruce Momjian 已提交
332
			break;				/* NOTE BREAK OUT */
333 334 335 336 337 338 339 340
	}
	assert(end != NULL);		/* search RE succeeded so loop should */
	freedfa(d);

	/* and pin down details */
	assert(v->nmatch > 0);
	v->pmatch[0].rm_so = OFF(begin);
	v->pmatch[0].rm_eo = OFF(end);
B
Bruce Momjian 已提交
341 342
	if (v->g->cflags & REG_EXPECT)
	{
343 344 345 346
		if (cold != NULL)
			v->details->rm_extend.rm_so = OFF(cold);
		else
			v->details->rm_extend.rm_so = OFF(v->stop);
B
Bruce Momjian 已提交
347
		v->details->rm_extend.rm_eo = OFF(v->stop);		/* unknown */
348
	}
B
Bruce Momjian 已提交
349
	if (v->nmatch == 1)			/* no need for submatches */
350 351 352 353 354 355 356 357 358 359 360
		return REG_OKAY;

	/* submatches */
	zapsubs(v->pmatch, v->nmatch);
	return dissect(v, v->g->tree, begin, end);
}

/*
 * cfind - find a match for the main NFA (with complications)
 */
static int
B
Bruce Momjian 已提交
361 362 363
cfind(struct vars * v,
	  struct cnfa * cnfa,
	  struct colormap * cm)
364 365 366
{
	struct dfa *s;
	struct dfa *d;
B
Bruce Momjian 已提交
367 368
	chr		   *cold;
	int			ret;
369 370 371 372

	s = newdfa(v, &v->g->search, cm, &v->dfa1);
	NOERR();
	d = newdfa(v, cnfa, cm, &v->dfa2);
B
Bruce Momjian 已提交
373 374
	if (ISERR())
	{
375 376 377 378 379 380 381 382 383 384
		assert(d == NULL);
		freedfa(s);
		return v->err;
	}

	ret = cfindloop(v, cnfa, cm, d, s, &cold);

	freedfa(d);
	freedfa(s);
	NOERR();
B
Bruce Momjian 已提交
385 386
	if (v->g->cflags & REG_EXPECT)
	{
387 388 389 390 391
		assert(v->details != NULL);
		if (cold != NULL)
			v->details->rm_extend.rm_so = OFF(cold);
		else
			v->details->rm_extend.rm_so = OFF(v->stop);
B
Bruce Momjian 已提交
392
		v->details->rm_extend.rm_eo = OFF(v->stop);		/* unknown */
393 394 395 396 397 398 399 400
	}
	return ret;
}

/*
 * cfindloop - the heart of cfind
 */
static int
B
Bruce Momjian 已提交
401 402 403 404 405
cfindloop(struct vars * v,
		  struct cnfa * cnfa,
		  struct colormap * cm,
		  struct dfa * d,
		  struct dfa * s,
406
		  chr **coldp)			/* where to put coldstart pointer */
407
{
B
Bruce Momjian 已提交
408 409 410
	chr		   *begin;
	chr		   *end;
	chr		   *cold;
B
Bruce Momjian 已提交
411
	chr		   *open;			/* open and close of range of possible starts */
B
Bruce Momjian 已提交
412 413 414 415 416 417
	chr		   *close;
	chr		   *estart;
	chr		   *estop;
	int			er;
	int			shorter = v->g->tree->flags & SHORTER;
	int			hitend;
418 419 420

	assert(d != NULL && s != NULL);
	cold = NULL;
421
	close = v->search_start;
B
Bruce Momjian 已提交
422 423
	do
	{
424
		MDEBUG(("\ncsearch at %ld\n", LOFF(close)));
B
Bruce Momjian 已提交
425
		close = shortest(v, s, close, close, v->stop, &cold, (int *) NULL);
426 427 428 429 430 431
		if (close == NULL)
			break;				/* NOTE BREAK */
		assert(cold != NULL);
		open = cold;
		cold = NULL;
		MDEBUG(("cbetween %ld and %ld\n", LOFF(open), LOFF(close)));
B
Bruce Momjian 已提交
432 433
		for (begin = open; begin <= close; begin++)
		{
434 435 436
			MDEBUG(("\ncfind trying at %ld\n", LOFF(begin)));
			estart = begin;
			estop = v->stop;
B
Bruce Momjian 已提交
437 438
			for (;;)
			{
439 440
				if (shorter)
					end = shortest(v, d, begin, estart,
B
Bruce Momjian 已提交
441
								   estop, (chr **) NULL, &hitend);
442 443
				else
					end = longest(v, d, begin, estop,
B
Bruce Momjian 已提交
444
								  &hitend);
445 446 447 448 449 450 451 452
				if (hitend && cold == NULL)
					cold = begin;
				if (end == NULL)
					break;		/* NOTE BREAK OUT */
				MDEBUG(("tentative end %ld\n", LOFF(end)));
				zapsubs(v->pmatch, v->nmatch);
				zapmem(v, v->g->tree);
				er = cdissect(v, v->g->tree, begin, end);
B
Bruce Momjian 已提交
453 454 455 456
				if (er == REG_OKAY)
				{
					if (v->nmatch > 0)
					{
457 458 459 460 461 462
						v->pmatch[0].rm_so = OFF(begin);
						v->pmatch[0].rm_eo = OFF(end);
					}
					*coldp = cold;
					return REG_OKAY;
				}
B
Bruce Momjian 已提交
463 464
				if (er != REG_NOMATCH)
				{
465
					ERR(er);
466
					*coldp = cold;
467 468
					return er;
				}
B
Bruce Momjian 已提交
469 470
				if ((shorter) ? end == estop : end == begin)
				{
471 472 473 474 475 476 477 478 479 480 481 482 483 484 485 486 487 488 489 490 491 492 493 494
					/* no point in trying again */
					*coldp = cold;
					return REG_NOMATCH;
				}
				/* go around and try again */
				if (shorter)
					estart = end + 1;
				else
					estop = end - 1;
			}
		}
	} while (close < v->stop);

	*coldp = cold;
	return REG_NOMATCH;
}

/*
 * zapsubs - initialize the subexpression matches to "no match"
 */
static void
zapsubs(regmatch_t *p,
		size_t n)
{
B
Bruce Momjian 已提交
495
	size_t		i;
496

B
Bruce Momjian 已提交
497 498
	for (i = n - 1; i > 0; i--)
	{
499 500 501 502 503 504 505 506 507
		p[i].rm_so = -1;
		p[i].rm_eo = -1;
	}
}

/*
 * zapmem - initialize the retry memory of a subtree to zeros
 */
static void
B
Bruce Momjian 已提交
508 509
zapmem(struct vars * v,
	   struct subre * t)
510 511 512 513 514 515
{
	if (t == NULL)
		return;

	assert(v->mem != NULL);
	v->mem[t->retry] = 0;
B
Bruce Momjian 已提交
516 517
	if (t->op == '(')
	{
518 519 520 521 522 523 524 525 526 527 528 529 530 531 532
		assert(t->subno > 0);
		v->pmatch[t->subno].rm_so = -1;
		v->pmatch[t->subno].rm_eo = -1;
	}

	if (t->left != NULL)
		zapmem(v, t->left);
	if (t->right != NULL)
		zapmem(v, t->right);
}

/*
 * subset - set any subexpression relevant to a successful subre
 */
static void
B
Bruce Momjian 已提交
533 534
subset(struct vars * v,
	   struct subre * sub,
535 536
	   chr *begin,
	   chr *end)
537
{
B
Bruce Momjian 已提交
538
	int			n = sub->subno;
539 540

	assert(n > 0);
B
Bruce Momjian 已提交
541
	if ((size_t) n >= v->nmatch)
542 543 544 545 546 547 548 549 550 551
		return;

	MDEBUG(("setting %d\n", n));
	v->pmatch[n].rm_so = OFF(begin);
	v->pmatch[n].rm_eo = OFF(end);
}

/*
 * dissect - determine subexpression matches (uncomplicated case)
 */
B
Bruce Momjian 已提交
552 553 554
static int						/* regexec return code */
dissect(struct vars * v,
		struct subre * t,
555 556
		chr *begin,				/* beginning of relevant substring */
		chr *end)				/* end of same */
557 558 559 560
{
	assert(t != NULL);
	MDEBUG(("dissect %ld-%ld\n", LOFF(begin), LOFF(end)));

B
Bruce Momjian 已提交
561 562 563 564 565 566 567 568 569 570 571 572 573 574 575 576 577 578 579 580
	switch (t->op)
	{
		case '=':				/* terminal node */
			assert(t->left == NULL && t->right == NULL);
			return REG_OKAY;	/* no action, parent did the work */
		case '|':				/* alternation */
			assert(t->left != NULL);
			return altdissect(v, t, begin, end);
		case 'b':				/* back ref -- shouldn't be calling us! */
			return REG_ASSERT;
		case '.':				/* concatenation */
			assert(t->left != NULL && t->right != NULL);
			return condissect(v, t, begin, end);
		case '(':				/* capturing */
			assert(t->left != NULL && t->right == NULL);
			assert(t->subno > 0);
			subset(v, t, begin, end);
			return dissect(v, t->left, begin, end);
		default:
			return REG_ASSERT;
581 582 583 584 585
	}
}

/*
 * condissect - determine concatenation subexpression matches (uncomplicated)
586
 */
B
Bruce Momjian 已提交
587 588 589
static int						/* regexec return code */
condissect(struct vars * v,
		   struct subre * t,
590 591
		   chr *begin,			/* beginning of relevant substring */
		   chr *end)			/* end of same */
592
{
593 594
	struct dfa *d;
	struct dfa *d2;
B
Bruce Momjian 已提交
595 596 597 598
	chr		   *mid;
	int			i;
	int			shorter = (t->left->flags & SHORTER) ? 1 : 0;
	chr		   *stop = (shorter) ? end : begin;
599 600 601 602 603 604 605 606

	assert(t->op == '.');
	assert(t->left != NULL && t->left->cnfa.nstates > 0);
	assert(t->right != NULL && t->right->cnfa.nstates > 0);

	d = newdfa(v, &t->left->cnfa, &v->g->cmap, &v->dfa1);
	NOERR();
	d2 = newdfa(v, &t->right->cnfa, &v->g->cmap, &v->dfa2);
B
Bruce Momjian 已提交
607 608
	if (ISERR())
	{
609 610 611 612 613 614 615
		assert(d2 == NULL);
		freedfa(d);
		return v->err;
	}

	/* pick a tentative midpoint */
	if (shorter)
B
Bruce Momjian 已提交
616 617
		mid = shortest(v, d, begin, begin, end, (chr **) NULL,
					   (int *) NULL);
618
	else
B
Bruce Momjian 已提交
619 620 621
		mid = longest(v, d, begin, end, (int *) NULL);
	if (mid == NULL)
	{
622 623 624 625 626 627 628
		freedfa(d);
		freedfa(d2);
		return REG_ASSERT;
	}
	MDEBUG(("tentative midpoint %ld\n", LOFF(mid)));

	/* iterate until satisfaction or failure */
B
Bruce Momjian 已提交
629 630
	while (longest(v, d2, mid, end, (int *) NULL) != end)
	{
631
		/* that midpoint didn't work, find a new one */
B
Bruce Momjian 已提交
632 633
		if (mid == stop)
		{
634 635 636 637 638 639 640
			/* all possibilities exhausted! */
			MDEBUG(("no midpoint!\n"));
			freedfa(d);
			freedfa(d2);
			return REG_ASSERT;
		}
		if (shorter)
B
Bruce Momjian 已提交
641 642
			mid = shortest(v, d, begin, mid + 1, end, (chr **) NULL,
						   (int *) NULL);
643
		else
B
Bruce Momjian 已提交
644 645 646
			mid = longest(v, d, begin, mid - 1, (int *) NULL);
		if (mid == NULL)
		{
647 648 649 650 651 652 653 654 655 656 657 658 659 660 661 662 663 664 665 666 667 668
			/* failed to find a new one! */
			MDEBUG(("failed midpoint!\n"));
			freedfa(d);
			freedfa(d2);
			return REG_ASSERT;
		}
		MDEBUG(("new midpoint %ld\n", LOFF(mid)));
	}

	/* satisfaction */
	MDEBUG(("successful\n"));
	freedfa(d);
	freedfa(d2);
	i = dissect(v, t->left, begin, mid);
	if (i != REG_OKAY)
		return i;
	return dissect(v, t->right, mid, end);
}

/*
 * altdissect - determine alternative subexpression matches (uncomplicated)
 */
B
Bruce Momjian 已提交
669 670 671
static int						/* regexec return code */
altdissect(struct vars * v,
		   struct subre * t,
672 673
		   chr *begin,			/* beginning of relevant substring */
		   chr *end)			/* end of same */
674 675
{
	struct dfa *d;
B
Bruce Momjian 已提交
676
	int			i;
677 678 679 680

	assert(t != NULL);
	assert(t->op == '|');

B
Bruce Momjian 已提交
681 682
	for (i = 0; t != NULL; t = t->right, i++)
	{
683 684 685 686 687
		MDEBUG(("trying %dth\n", i));
		assert(t->left != NULL && t->left->cnfa.nstates > 0);
		d = newdfa(v, &t->left->cnfa, &v->g->cmap, &v->dfa1);
		if (ISERR())
			return v->err;
B
Bruce Momjian 已提交
688 689
		if (longest(v, d, begin, end, (int *) NULL) == end)
		{
690 691 692 693 694 695
			MDEBUG(("success\n"));
			freedfa(d);
			return dissect(v, t->left, begin, end);
		}
		freedfa(d);
	}
B
Bruce Momjian 已提交
696
	return REG_ASSERT;			/* none of them matched?!? */
697 698 699 700
}

/*
 * cdissect - determine subexpression matches (with complications)
B
Bruce Momjian 已提交
701
 * The retry memory stores the offset of the trial midpoint from begin,
702 703
 * plus 1 so that 0 uniquely means "clean slate".
 */
B
Bruce Momjian 已提交
704 705 706
static int						/* regexec return code */
cdissect(struct vars * v,
		 struct subre * t,
707 708
		 chr *begin,			/* beginning of relevant substring */
		 chr *end)				/* end of same */
709
{
T
Tom Lane 已提交
710 711
	int			er;

712 713 714
	assert(t != NULL);
	MDEBUG(("cdissect %ld-%ld %c\n", LOFF(begin), LOFF(end), t->op));

B
Bruce Momjian 已提交
715 716 717 718 719 720 721 722 723 724 725 726 727 728 729 730
	switch (t->op)
	{
		case '=':				/* terminal node */
			assert(t->left == NULL && t->right == NULL);
			return REG_OKAY;	/* no action, parent did the work */
		case '|':				/* alternation */
			assert(t->left != NULL);
			return caltdissect(v, t, begin, end);
		case 'b':				/* back ref -- shouldn't be calling us! */
			assert(t->left == NULL && t->right == NULL);
			return cbrdissect(v, t, begin, end);
		case '.':				/* concatenation */
			assert(t->left != NULL && t->right != NULL);
			return ccondissect(v, t, begin, end);
		case '(':				/* capturing */
			assert(t->left != NULL && t->right == NULL);
T
Tom Lane 已提交
731 732 733 734 735
			assert(t->subno > 0);
			er = cdissect(v, t->left, begin, end);
			if (er == REG_OKAY)
				subset(v, t, begin, end);
			return er;
B
Bruce Momjian 已提交
736 737
		default:
			return REG_ASSERT;
738 739 740 741 742
	}
}

/*
 * ccondissect - concatenation subexpression matches (with complications)
B
Bruce Momjian 已提交
743
 * The retry memory stores the offset of the trial midpoint from begin,
744 745
 * plus 1 so that 0 uniquely means "clean slate".
 */
B
Bruce Momjian 已提交
746 747 748
static int						/* regexec return code */
ccondissect(struct vars * v,
			struct subre * t,
749 750
			chr *begin,			/* beginning of relevant substring */
			chr *end)			/* end of same */
751 752 753
{
	struct dfa *d;
	struct dfa *d2;
B
Bruce Momjian 已提交
754 755
	chr		   *mid;
	int			er;
756 757 758 759 760

	assert(t->op == '.');
	assert(t->left != NULL && t->left->cnfa.nstates > 0);
	assert(t->right != NULL && t->right->cnfa.nstates > 0);

B
Bruce Momjian 已提交
761
	if (t->left->flags & SHORTER)		/* reverse scan */
762 763 764 765 766 767
		return crevdissect(v, t, begin, end);

	d = newdfa(v, &t->left->cnfa, &v->g->cmap, DOMALLOC);
	if (ISERR())
		return v->err;
	d2 = newdfa(v, &t->right->cnfa, &v->g->cmap, DOMALLOC);
B
Bruce Momjian 已提交
768 769
	if (ISERR())
	{
770 771 772 773 774 775
		freedfa(d);
		return v->err;
	}
	MDEBUG(("cconcat %d\n", t->retry));

	/* pick a tentative midpoint */
B
Bruce Momjian 已提交
776 777 778 779 780
	if (v->mem[t->retry] == 0)
	{
		mid = longest(v, d, begin, end, (int *) NULL);
		if (mid == NULL)
		{
781 782 783 784 785 786
			freedfa(d);
			freedfa(d2);
			return REG_NOMATCH;
		}
		MDEBUG(("tentative midpoint %ld\n", LOFF(mid)));
		v->mem[t->retry] = (mid - begin) + 1;
B
Bruce Momjian 已提交
787 788 789
	}
	else
	{
790 791 792 793 794
		mid = begin + (v->mem[t->retry] - 1);
		MDEBUG(("working midpoint %ld\n", LOFF(mid)));
	}

	/* iterate until satisfaction or failure */
B
Bruce Momjian 已提交
795 796
	for (;;)
	{
797
		/* try this midpoint on for size */
798
		if (longest(v, d2, mid, end, (int *) NULL) == end)
B
Bruce Momjian 已提交
799
		{
800 801 802 803 804 805 806 807 808 809 810 811 812 813 814 815 816 817 818
			er = cdissect(v, t->left, begin, mid);
			if (er == REG_OKAY)
			{
				er = cdissect(v, t->right, mid, end);
				if (er == REG_OKAY)
				{
					/* satisfaction */
					MDEBUG(("successful\n"));
					freedfa(d);
					freedfa(d2);
					return REG_OKAY;
				}
			}
			if (er != REG_OKAY && er != REG_NOMATCH)
			{
				freedfa(d);
				freedfa(d2);
				return er;
			}
819 820 821
		}

		/* that midpoint didn't work, find a new one */
B
Bruce Momjian 已提交
822 823
		if (mid == begin)
		{
824 825 826 827 828 829
			/* all possibilities exhausted */
			MDEBUG(("%d no midpoint\n", t->retry));
			freedfa(d);
			freedfa(d2);
			return REG_NOMATCH;
		}
B
Bruce Momjian 已提交
830 831 832
		mid = longest(v, d, begin, mid - 1, (int *) NULL);
		if (mid == NULL)
		{
833 834 835 836 837 838 839 840 841 842 843 844
			/* failed to find a new one */
			MDEBUG(("%d failed midpoint\n", t->retry));
			freedfa(d);
			freedfa(d2);
			return REG_NOMATCH;
		}
		MDEBUG(("%d: new midpoint %ld\n", t->retry, LOFF(mid)));
		v->mem[t->retry] = (mid - begin) + 1;
		zapmem(v, t->left);
		zapmem(v, t->right);
	}

845 846
	/* can't get here */
	return REG_ASSERT;
847
}
848 849 850

/*
 * crevdissect - determine backref shortest-first subexpression matches
B
Bruce Momjian 已提交
851
 * The retry memory stores the offset of the trial midpoint from begin,
852 853
 * plus 1 so that 0 uniquely means "clean slate".
 */
B
Bruce Momjian 已提交
854 855 856
static int						/* regexec return code */
crevdissect(struct vars * v,
			struct subre * t,
857 858
			chr *begin,			/* beginning of relevant substring */
			chr *end)			/* end of same */
859 860 861
{
	struct dfa *d;
	struct dfa *d2;
B
Bruce Momjian 已提交
862 863
	chr		   *mid;
	int			er;
864 865 866 867

	assert(t->op == '.');
	assert(t->left != NULL && t->left->cnfa.nstates > 0);
	assert(t->right != NULL && t->right->cnfa.nstates > 0);
B
Bruce Momjian 已提交
868
	assert(t->left->flags & SHORTER);
869 870 871 872 873 874

	/* concatenation -- need to split the substring between parts */
	d = newdfa(v, &t->left->cnfa, &v->g->cmap, DOMALLOC);
	if (ISERR())
		return v->err;
	d2 = newdfa(v, &t->right->cnfa, &v->g->cmap, DOMALLOC);
B
Bruce Momjian 已提交
875 876
	if (ISERR())
	{
877 878 879 880 881 882
		freedfa(d);
		return v->err;
	}
	MDEBUG(("crev %d\n", t->retry));

	/* pick a tentative midpoint */
B
Bruce Momjian 已提交
883 884 885 886 887
	if (v->mem[t->retry] == 0)
	{
		mid = shortest(v, d, begin, begin, end, (chr **) NULL, (int *) NULL);
		if (mid == NULL)
		{
888 889 890 891 892 893
			freedfa(d);
			freedfa(d2);
			return REG_NOMATCH;
		}
		MDEBUG(("tentative midpoint %ld\n", LOFF(mid)));
		v->mem[t->retry] = (mid - begin) + 1;
B
Bruce Momjian 已提交
894 895 896
	}
	else
	{
897 898 899 900 901
		mid = begin + (v->mem[t->retry] - 1);
		MDEBUG(("working midpoint %ld\n", LOFF(mid)));
	}

	/* iterate until satisfaction or failure */
B
Bruce Momjian 已提交
902 903
	for (;;)
	{
904
		/* try this midpoint on for size */
905
		if (longest(v, d2, mid, end, (int *) NULL) == end)
B
Bruce Momjian 已提交
906
		{
907 908 909 910 911 912 913 914 915 916 917 918 919 920 921 922 923 924 925
			er = cdissect(v, t->left, begin, mid);
			if (er == REG_OKAY)
			{
				er = cdissect(v, t->right, mid, end);
				if (er == REG_OKAY)
				{
					/* satisfaction */
					MDEBUG(("successful\n"));
					freedfa(d);
					freedfa(d2);
					return REG_OKAY;
				}
			}
			if (er != REG_OKAY && er != REG_NOMATCH)
			{
				freedfa(d);
				freedfa(d2);
				return er;
			}
926 927 928
		}

		/* that midpoint didn't work, find a new one */
B
Bruce Momjian 已提交
929 930
		if (mid == end)
		{
931 932 933 934 935 936
			/* all possibilities exhausted */
			MDEBUG(("%d no midpoint\n", t->retry));
			freedfa(d);
			freedfa(d2);
			return REG_NOMATCH;
		}
B
Bruce Momjian 已提交
937 938 939
		mid = shortest(v, d, begin, mid + 1, end, (chr **) NULL, (int *) NULL);
		if (mid == NULL)
		{
940 941 942 943 944 945 946 947 948 949 950 951
			/* failed to find a new one */
			MDEBUG(("%d failed midpoint\n", t->retry));
			freedfa(d);
			freedfa(d2);
			return REG_NOMATCH;
		}
		MDEBUG(("%d: new midpoint %ld\n", t->retry, LOFF(mid)));
		v->mem[t->retry] = (mid - begin) + 1;
		zapmem(v, t->left);
		zapmem(v, t->right);
	}

952 953
	/* can't get here */
	return REG_ASSERT;
954 955 956 957 958
}

/*
 * cbrdissect - determine backref subexpression matches
 */
B
Bruce Momjian 已提交
959 960 961
static int						/* regexec return code */
cbrdissect(struct vars * v,
		   struct subre * t,
962 963
		   chr *begin,			/* beginning of relevant substring */
		   chr *end)			/* end of same */
964
{
B
Bruce Momjian 已提交
965 966 967 968 969 970 971 972
	int			i;
	int			n = t->subno;
	size_t		len;
	chr		   *paren;
	chr		   *p;
	chr		   *stop;
	int			min = t->min;
	int			max = t->max;
973 974 975 976

	assert(t != NULL);
	assert(t->op == 'b');
	assert(n >= 0);
B
Bruce Momjian 已提交
977
	assert((size_t) n < v->nmatch);
978 979 980 981 982 983 984 985 986 987 988 989 990 991

	MDEBUG(("cbackref n%d %d{%d-%d}\n", t->retry, n, min, max));

	if (v->pmatch[n].rm_so == -1)
		return REG_NOMATCH;
	paren = v->start + v->pmatch[n].rm_so;
	len = v->pmatch[n].rm_eo - v->pmatch[n].rm_so;

	/* no room to maneuver -- retries are pointless */
	if (v->mem[t->retry])
		return REG_NOMATCH;
	v->mem[t->retry] = 1;

	/* special-case zero-length string */
B
Bruce Momjian 已提交
992 993
	if (len == 0)
	{
994 995 996 997 998 999 1000
		if (begin == end)
			return REG_OKAY;
		return REG_NOMATCH;
	}

	/* and too-short string */
	assert(end >= begin);
B
Bruce Momjian 已提交
1001
	if ((size_t) (end - begin) < len)
1002 1003 1004 1005 1006
		return REG_NOMATCH;
	stop = end - len;

	/* count occurrences */
	i = 0;
B
Bruce Momjian 已提交
1007 1008 1009 1010
	for (p = begin; p <= stop && (i < max || max == INFINITY); p += len)
	{
		if ((*v->g->compare) (paren, p, len) != 0)
			break;
1011 1012 1013 1014 1015
		i++;
	}
	MDEBUG(("cbackref found %d\n", i));

	/* and sort it out */
B
Bruce Momjian 已提交
1016
	if (p != end)				/* didn't consume all of it */
1017 1018 1019
		return REG_NOMATCH;
	if (min <= i && (i <= max || max == INFINITY))
		return REG_OKAY;
B
Bruce Momjian 已提交
1020
	return REG_NOMATCH;			/* out of range */
1021 1022 1023 1024 1025
}

/*
 * caltdissect - determine alternative subexpression matches (w. complications)
 */
B
Bruce Momjian 已提交
1026 1027 1028
static int						/* regexec return code */
caltdissect(struct vars * v,
			struct subre * t,
1029 1030
			chr *begin,			/* beginning of relevant substring */
			chr *end)			/* end of same */
1031 1032
{
	struct dfa *d;
B
Bruce Momjian 已提交
1033 1034 1035 1036
	int			er;

#define  UNTRIED 0				/* not yet tried at all */
#define  TRYING  1				/* top matched, trying submatches */
B
Bruce Momjian 已提交
1037
#define  TRIED	 2				/* top didn't match or submatches exhausted */
1038 1039 1040 1041 1042 1043 1044 1045 1046 1047

	if (t == NULL)
		return REG_NOMATCH;
	assert(t->op == '|');
	if (v->mem[t->retry] == TRIED)
		return caltdissect(v, t->right, begin, end);

	MDEBUG(("calt n%d\n", t->retry));
	assert(t->left != NULL);

B
Bruce Momjian 已提交
1048 1049
	if (v->mem[t->retry] == UNTRIED)
	{
1050 1051 1052
		d = newdfa(v, &t->left->cnfa, &v->g->cmap, DOMALLOC);
		if (ISERR())
			return v->err;
B
Bruce Momjian 已提交
1053 1054
		if (longest(v, d, begin, end, (int *) NULL) != end)
		{
1055 1056 1057 1058 1059 1060 1061 1062 1063 1064 1065 1066 1067 1068 1069 1070 1071 1072 1073 1074
			freedfa(d);
			v->mem[t->retry] = TRIED;
			return caltdissect(v, t->right, begin, end);
		}
		freedfa(d);
		MDEBUG(("calt matched\n"));
		v->mem[t->retry] = TRYING;
	}

	er = cdissect(v, t->left, begin, end);
	if (er != REG_NOMATCH)
		return er;

	v->mem[t->retry] = TRIED;
	return caltdissect(v, t->right, begin, end);
}



#include "rege_dfa.c"