From 92abd82f0d1d42da7c68a3bb3d2d6079073120ae Mon Sep 17 00:00:00 2001 From: Norihiro Tanaka Date: Wed, 26 Mar 2014 22:14:19 +0900 Subject: [PATCH] grep: optimization of DFA by reuse of multi-byte buffers in non-UTF8 locales * src/dfa.c (struct dfa): New members `mblen_buf', `nmblen_buf', `inputwcs', `ninputwcs', `mb_follows' and `mb_match_lens'. (SKIP_REMAINS_MB_IF_INITIAL_STATE): Use them instead of global variables. (match_anychar): Use them instead of global variables. (match_mb_charset): Use them instead of global variables. (check_matching_with_multibyte_ops): Use it instead of new allocation. (transit_state_consume_1char): Use them instead of global variables. (transit_state): Use them instead of global variables. (prepare_wc_buf): Use them instead of global variables. (dfaexec): Initializations of new members. (free_mbdata): Memory Deallocations for new members. --- src/dfa.c | 133 ++++++++++++++++++++++++++++++++------------------------------ 1 file changed, 68 insertions(+), 65 deletions(-) diff --git a/src/dfa.c b/src/dfa.c index 4ed2189..39ff197 100644 --- a/src/dfa.c +++ b/src/dfa.c @@ -436,6 +436,30 @@ struct dfa struct dfamust *musts; /* List of strings, at least one of which is known to appear in any r.e. matching the dfa. */ + unsigned char *mblen_buf; /* Correspond to the input buffer in dfaexec. + Each element stores the number of remaining + bytes of the corresponding multibyte + character in the input string. A element's + value is 0 if the corresponding character is + single-byte. + e.g., input : 'a', , , + mblen_buf : 0, 3, 2, 1 + */ + size_t nmblen_buf; /* Length of the mblen buffer currently + allocated. */ + wchar_t *inputwcs; /* Wide character representation of the input + string in dfaexec. + The length of this array is the same as + the length of input string (char array). + inputstring[i] is a single-byte char, + or the first byte of a multibyte char; + inputwcs[i] is the codepoint. */ + size_t ninputwcs; /* Length of the input wide characters + currently allocated. */ + position_set *mb_follows; /* Follow set added by ANYCHAR and/or MBCSET + on demand. */ + int *mb_match_lens; /* Array of length reduced by ANYCHAR and/or + MBCSET. */ }; /* Some macros for user access to dfa internals. */ @@ -876,22 +900,6 @@ static int cur_mb_len = 1; /* Length of the multibyte representation of static mbstate_t mbs; /* mbstate for mbrtowc. */ static wchar_t wctok; /* Wide character representation of the current multibyte character. */ -static unsigned char *mblen_buf;/* Correspond to the input buffer in dfaexec. - Each element stores the number of remaining - bytes of the corresponding multibyte - character in the input string. A element's - value is 0 if the corresponding character is - single-byte. - e.g., input : 'a', , , - mblen_buf : 0, 3, 2, 1 - */ -static wchar_t *inputwcs; /* Wide character representation of the input - string in dfaexec. - The length of this array is the same as - the length of input string (char array). - inputstring[i] is a single-byte char, - or the first byte of a multibyte char; - inputwcs[i] is the codepoint. */ static unsigned char const *buf_begin; /* reference to begin in dfaexec. */ static unsigned char const *buf_end; /* reference to end in dfaexec. */ @@ -2959,14 +2967,12 @@ build_state_zero (struct dfa *d) #define SKIP_REMAINS_MB_IF_INITIAL_STATE(s, p) \ if (s == 0) \ { \ - while (inputwcs[p - buf_begin] == 0 \ - && mblen_buf[p - buf_begin] > 0 \ + while (d->inputwcs[p - buf_begin] == 0 \ + && d->mblen_buf[p - buf_begin] > 0 \ && (unsigned char const *) p < buf_end) \ ++p; \ if ((char *) p >= end) \ { \ - free (mblen_buf); \ - free (inputwcs); \ *end = saved_end; \ return NULL; \ } \ @@ -3060,8 +3066,8 @@ match_anychar (struct dfa *d, state_num s, position pos, size_t idx) wchar_t wc; int mbclen; - wc = inputwcs[idx]; - mbclen = (mblen_buf[idx] == 0) ? 1 : mblen_buf[idx]; + wc = d->inputwcs[idx]; + mbclen = (d->mblen_buf[idx] == 0) ? 1 : d->mblen_buf[idx]; /* Check syntax bits. */ if (wc == (wchar_t) eolbyte) @@ -3102,7 +3108,7 @@ match_mb_charset (struct dfa *d, state_num s, position pos, size_t idx) int context; wchar_t wc; /* Current referring character. */ - wc = inputwcs[idx]; + wc = d->inputwcs[idx]; /* Check syntax bits. */ if (wc == (wchar_t) eolbyte) @@ -3123,7 +3129,7 @@ match_mb_charset (struct dfa *d, state_num s, position pos, size_t idx) /* Assign the current referring operator to work_mbc. */ work_mbc = &(d->mbcsets[(d->multibyte_prop[pos.index]) >> 2]); match = !work_mbc->invert; - match_len = (mblen_buf[idx] == 0) ? 1 : mblen_buf[idx]; + match_len = (d->mblen_buf[idx] == 0) ? 1 : d->mblen_buf[idx]; /* Match in range 0-255? */ if (wc < NOTCHAR && work_mbc->cset != -1 @@ -3201,7 +3207,7 @@ check_matching_with_multibyte_ops (struct dfa *d, state_num s, size_t idx) size_t i; int *rarray; - MALLOC (rarray, d->states[s].mbps.nelem); + rarray = d->mb_match_lens; for (i = 0; i < d->states[s].mbps.nelem; ++i) { position pos = d->states[s].mbps.elems[i]; @@ -3231,7 +3237,7 @@ check_matching_with_multibyte_ops (struct dfa *d, state_num s, size_t idx) static status_transit_state transit_state_consume_1char (struct dfa *d, state_num s, unsigned char const **pp, - int *match_lens, int *mbclen, position_set * pps) + int *match_lens, int *mbclen) { size_t i, j; int k; @@ -3241,7 +3247,7 @@ transit_state_consume_1char (struct dfa *d, state_num s, /* Calculate the length of the (single/multi byte) character to which p points. */ - *mbclen = (mblen_buf[*pp - buf_begin] == 0) ? 1 : mblen_buf[*pp - buf_begin]; + *mbclen = (d->mblen_buf[*pp - buf_begin] == 0) ? 1 : d->mblen_buf[*pp - buf_begin]; /* Calculate the state which can be reached from the state 's' by consuming '*mbclen' single bytes from the buffer. */ @@ -3251,8 +3257,8 @@ transit_state_consume_1char (struct dfa *d, state_num s, s2 = s1; rs = transit_state_singlebyte (d, s2, (*pp)++, &s1); } - /* Copy the positions contained by 's1' to the set 'pps'. */ - copy (&(d->states[s1].elems), pps); + /* Copy the positions contained by 's1' to the set 'd->mb_follows'. */ + copy (&(d->states[s1].elems), d->mb_follows); /* Check (input) match_lens, and initialize if it is NULL. */ if (match_lens == NULL && d->states[s].mbps.nelem != 0) @@ -3267,12 +3273,10 @@ transit_state_consume_1char (struct dfa *d, state_num s, if (work_mbls[i] == *mbclen) for (j = 0; j < d->follows[d->states[s].mbps.elems[i].index].nelem; j++) - insert (d->follows[d->states[s].mbps.elems[i].index].elems[j], pps); + insert (d->follows[d->states[s].mbps.elems[i].index].elems[j], + d->mb_follows); } - if (match_lens == NULL && work_mbls != NULL) - free (work_mbls); - /* FIXME: this return value is always ignored. */ return rs; } @@ -3289,7 +3293,6 @@ transit_state (struct dfa *d, state_num s, unsigned char const **pp) size_t i, j; int *match_lens = NULL; size_t nelem = d->states[s].mbps.nelem; /* Just a alias. */ - position_set follows; unsigned char const *p1 = *pp; wchar_t wc; @@ -3320,26 +3323,25 @@ transit_state (struct dfa *d, state_num s, unsigned char const **pp) if (rs == TRANSIT_STATE_DONE) ++*pp; - free (match_lens); return s1; } /* This state has some operators which can match a multibyte character. */ - alloc_position_set (&follows, d->nleaves); + d->mb_follows->nelem = 0; /* 'maxlen' may be longer than the length of a character, because it may not be a character but a (multi character) collating element. We enumerate all of the positions which 's' can reach by consuming 'maxlen' bytes. */ - transit_state_consume_1char (d, s, pp, match_lens, &mbclen, &follows); + transit_state_consume_1char (d, s, pp, match_lens, &mbclen); - wc = inputwcs[*pp - mbclen - buf_begin]; - s1 = state_index (d, &follows, wchar_context (wc)); + wc = d->inputwcs[*pp - mbclen - buf_begin]; + s1 = state_index (d, d->mb_follows, wchar_context (wc)); realloc_trans_if_necessary (d, s1); while (*pp - p1 < maxlen) { - transit_state_consume_1char (d, s1, pp, NULL, &mbclen, &follows); + transit_state_consume_1char (d, s1, pp, NULL, &mbclen); for (i = 0; i < nelem; i++) { @@ -3347,15 +3349,13 @@ transit_state (struct dfa *d, state_num s, unsigned char const **pp) for (j = 0; j < d->follows[d->states[s1].mbps.elems[i].index].nelem; j++) insert (d->follows[d->states[s1].mbps.elems[i].index].elems[j], - &follows); + d->mb_follows); } - wc = inputwcs[*pp - mbclen - buf_begin]; - s1 = state_index (d, &follows, wchar_context (wc)); + wc = d->inputwcs[*pp - mbclen - buf_begin]; + s1 = state_index (d, d->mb_follows, wchar_context (wc)); realloc_trans_if_necessary (d, s1); } - free (match_lens); - free (follows.elems); return s1; } @@ -3374,21 +3374,21 @@ prepare_wc_buf (struct dfa *d, const char *begin, const char *end) for (i = 0; i < ilim; i++) { - size_t nbytes = mbs_to_wchar (d, inputwcs + i, begin + i, ilim - i, &mbs); - mblen_buf[i] = nbytes - (nbytes == 1); + size_t nbytes = mbs_to_wchar (d, d->inputwcs + i, begin + i, ilim - i, &mbs); + d->mblen_buf[i] = nbytes - (nbytes == 1); if (begin[i] == eol) break; while (--nbytes != 0) { i++; - mblen_buf[i] = nbytes; - inputwcs[i] = 0; + d->mblen_buf[i] = nbytes; + d->inputwcs[i] = 0; } } buf_end = (unsigned char *) (begin + i); - mblen_buf[i] = 0; - inputwcs[i] = 0; /* sentinel */ + d->mblen_buf[i] = 0; + d->inputwcs[i] = 0; /* sentinel */ #endif /* MBS_SUPPORT */ } @@ -3426,10 +3426,18 @@ dfaexec (struct dfa *d, char const *begin, char *end, if (d->mb_cur_max > 1) { - MALLOC (mblen_buf, end - begin + 2); - MALLOC (inputwcs, end - begin + 2); + static bool mb_alloc = false; + REALLOC_IF_NECESSARY (d->mblen_buf, d->nmblen_buf, end - begin + 2); + REALLOC_IF_NECESSARY (d->inputwcs, d->ninputwcs, end - begin + 2); memset (&mbs, 0, sizeof (mbstate_t)); prepare_wc_buf (d, (const char *) p, end); + if (!mb_alloc) + { + MALLOC (d->mb_match_lens, d->nleaves); + MALLOC (d->mb_follows, 1); + alloc_position_set (d->mb_follows, d->nleaves); + mb_alloc = true; + } } for (;;) @@ -3456,8 +3464,6 @@ dfaexec (struct dfa *d, char const *begin, char *end, if (backref) { *backref = 1; - free (mblen_buf); - free (inputwcs); *end = saved_end; return (char *) p; } @@ -3490,11 +3496,6 @@ dfaexec (struct dfa *d, char const *begin, char *end, { if (backref) *backref = (d->states[s].backref != 0); - if (d->mb_cur_max > 1) - { - free (mblen_buf); - free (inputwcs); - } *end = saved_end; return (char *) p; } @@ -3525,11 +3526,6 @@ dfaexec (struct dfa *d, char const *begin, char *end, /* Check if we've run off the end of the buffer. */ if ((char *) p > end) { - if (d->mb_cur_max > 1) - { - free (mblen_buf); - free (inputwcs); - } *end = saved_end; return NULL; } @@ -3580,6 +3576,13 @@ free_mbdata (struct dfa *d) free (d->mbcsets); d->mbcsets = NULL; d->nmbcsets = 0; + + free (d->mblen_buf); + free (d->inputwcs); + if (d->mb_follows != NULL) + free (d->mb_follows->elems); + free (d->mb_follows); + free (d->mb_match_lens); } /* Initialize the components of a dfa that the other routines don't -- 1.9.1