• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1 /*
2 **********************************************************************
3 *   Copyright (C) 2001-2010 IBM and others. All rights reserved.
4 **********************************************************************
5 *   Date        Name        Description
6 *  07/02/2001   synwee      Creation.
7 **********************************************************************
8 */
9 
10 #include "unicode/utypes.h"
11 
12 #if !UCONFIG_NO_COLLATION && !UCONFIG_NO_BREAK_ITERATION
13 
14 #include "unicode/usearch.h"
15 #include "unicode/ustring.h"
16 #include "unicode/uchar.h"
17 #include "normalizer2impl.h"
18 #include "ucol_imp.h"
19 #include "usrchimp.h"
20 #include "cmemory.h"
21 #include "ucln_in.h"
22 #include "uassert.h"
23 #include "ustr_imp.h"
24 
25 U_NAMESPACE_USE
26 
27 // don't use Boyer-Moore
28 // (and if we decide to turn this on again there are several new TODOs that will need to be addressed)
29 #define BOYER_MOORE 0
30 
31 #define LENGTHOF(array) (int32_t)(sizeof(array)/sizeof((array)[0]))
32 
33 // internal definition ---------------------------------------------------
34 
35 #define LAST_BYTE_MASK_          0xFF
36 #define SECOND_LAST_BYTE_SHIFT_  8
37 #define SUPPLEMENTARY_MIN_VALUE_ 0x10000
38 
39 static const uint16_t *fcdTrieIndex = NULL;
40 static UChar32 fcdHighStart = 0;
41 
42 // internal methods -------------------------------------------------
43 
44 /**
45 * Fast collation element iterator setOffset.
46 * This function does not check for bounds.
47 * @param coleiter collation element iterator
48 * @param offset to set
49 */
50 static
setColEIterOffset(UCollationElements * elems,int32_t offset)51 inline void setColEIterOffset(UCollationElements *elems,
52                       int32_t             offset)
53 {
54     collIterate *ci = &(elems->iteratordata_);
55     ci->pos         = ci->string + offset;
56     ci->CEpos       = ci->toReturn = ci->extendCEs ? ci->extendCEs : ci->CEs;
57     if (ci->flags & UCOL_ITER_INNORMBUF) {
58         ci->flags = ci->origFlags;
59     }
60     ci->fcdPosition = NULL;
61 
62 	ci->offsetReturn = NULL;
63     ci->offsetStore  = ci->offsetBuffer;
64 	ci->offsetRepeatCount = ci->offsetRepeatValue = 0;
65 }
66 
67 /**
68 * Getting the mask for collation strength
69 * @param strength collation strength
70 * @return collation element mask
71 */
72 static
getMask(UCollationStrength strength)73 inline uint32_t getMask(UCollationStrength strength)
74 {
75     switch (strength)
76     {
77     case UCOL_PRIMARY:
78         return UCOL_PRIMARYORDERMASK;
79     case UCOL_SECONDARY:
80         return UCOL_SECONDARYORDERMASK | UCOL_PRIMARYORDERMASK;
81     default:
82         return UCOL_TERTIARYORDERMASK | UCOL_SECONDARYORDERMASK |
83                UCOL_PRIMARYORDERMASK;
84     }
85 }
86 
87 /**
88 * This is to squeeze the 21bit ces into a 256 table
89 * @param ce collation element
90 * @return collapsed version of the collation element
91 */
92 static
hash(uint32_t ce)93 inline int hash(uint32_t ce)
94 {
95     // the old value UCOL_PRIMARYORDER(ce) % MAX_TABLE_SIZE_ does not work
96     // well with the new collation where most of the latin 1 characters
97     // are of the value xx000xxx. their hashes will most of the time be 0
98     // to be discussed on the hash algo.
99     return UCOL_PRIMARYORDER(ce) % MAX_TABLE_SIZE_;
100 }
101 
102 U_CDECL_BEGIN
103 static UBool U_CALLCONV
usearch_cleanup(void)104 usearch_cleanup(void) {
105     fcdTrieIndex = NULL;
106     return TRUE;
107 }
108 U_CDECL_END
109 
110 /**
111 * Initializing the fcd tables.
112 * Internal method, status assumed to be a success.
113 * @param status output error if any, caller to check status before calling
114 *               method, status assumed to be success when passed in.
115 */
116 static
initializeFCD(UErrorCode * status)117 inline void initializeFCD(UErrorCode *status)
118 {
119     if (fcdTrieIndex == NULL) {
120         fcdTrieIndex = unorm_getFCDTrieIndex(fcdHighStart, status);
121         ucln_i18n_registerCleanup(UCLN_I18N_USEARCH, usearch_cleanup);
122     }
123 }
124 
125 /**
126 * Gets the fcd value for a character at the argument index.
127 * This method takes into accounts of the supplementary characters.
128 * @param str UTF16 string where character for fcd retrieval resides
129 * @param offset position of the character whose fcd is to be retrieved, to be
130 *               overwritten with the next character position, taking
131 *               surrogate characters into consideration.
132 * @param strlength length of the argument string
133 * @return fcd value
134 */
135 static
getFCD(const UChar * str,int32_t * offset,int32_t strlength)136 uint16_t getFCD(const UChar   *str, int32_t *offset,
137                              int32_t  strlength)
138 {
139     const UChar *temp = str + *offset;
140     uint16_t    result = unorm_nextFCD16(fcdTrieIndex, fcdHighStart, temp, str + strlength);
141     *offset = (int32_t)(temp - str);
142     return result;
143 }
144 
145 /**
146 * Getting the modified collation elements taking into account the collation
147 * attributes
148 * @param strsrch string search data
149 * @param sourcece
150 * @return the modified collation element
151 */
152 static
getCE(const UStringSearch * strsrch,uint32_t sourcece)153 inline int32_t getCE(const UStringSearch *strsrch, uint32_t sourcece)
154 {
155     // note for tertiary we can't use the collator->tertiaryMask, that
156     // is a preprocessed mask that takes into account case options. since
157     // we are only concerned with exact matches, we don't need that.
158     sourcece &= strsrch->ceMask;
159 
160     if (strsrch->toShift) {
161         // alternate handling here, since only the 16 most significant digits
162         // is only used, we can safely do a compare without masking
163         // if the ce is a variable, we mask and get only the primary values
164         // no shifting to quartenary is required since all primary values
165         // less than variabletop will need to be masked off anyway.
166         if (strsrch->variableTop > sourcece) {
167             if (strsrch->strength >= UCOL_QUATERNARY) {
168                 sourcece &= UCOL_PRIMARYORDERMASK;
169             }
170             else {
171                 sourcece = UCOL_IGNORABLE;
172             }
173         }
174     } else if (strsrch->strength >= UCOL_QUATERNARY && sourcece == UCOL_IGNORABLE) {
175         sourcece = 0xFFFF;
176     }
177 
178     return sourcece;
179 }
180 
181 /**
182 * Allocate a memory and returns NULL if it failed.
183 * Internal method, status assumed to be a success.
184 * @param size to allocate
185 * @param status output error if any, caller to check status before calling
186 *               method, status assumed to be success when passed in.
187 * @return newly allocated array, NULL otherwise
188 */
189 static
allocateMemory(uint32_t size,UErrorCode * status)190 inline void * allocateMemory(uint32_t size, UErrorCode *status)
191 {
192     uint32_t *result = (uint32_t *)uprv_malloc(size);
193     if (result == NULL) {
194         *status = U_MEMORY_ALLOCATION_ERROR;
195     }
196     return result;
197 }
198 
199 /**
200 * Adds a uint32_t value to a destination array.
201 * Creates a new array if we run out of space. The caller will have to
202 * manually deallocate the newly allocated array.
203 * Internal method, status assumed to be success, caller has to check status
204 * before calling this method. destination not to be NULL and has at least
205 * size destinationlength.
206 * @param destination target array
207 * @param offset destination offset to add value
208 * @param destinationlength target array size, return value for the new size
209 * @param value to be added
210 * @param increments incremental size expected
211 * @param status output error if any, caller to check status before calling
212 *               method, status assumed to be success when passed in.
213 * @return new destination array, destination if there was no new allocation
214 */
215 static
addTouint32_tArray(int32_t * destination,uint32_t offset,uint32_t * destinationlength,uint32_t value,uint32_t increments,UErrorCode * status)216 inline int32_t * addTouint32_tArray(int32_t    *destination,
217                                     uint32_t    offset,
218                                     uint32_t   *destinationlength,
219                                     uint32_t    value,
220                                     uint32_t    increments,
221                                     UErrorCode *status)
222 {
223     uint32_t newlength = *destinationlength;
224     if (offset + 1 == newlength) {
225         newlength += increments;
226         int32_t *temp = (int32_t *)allocateMemory(
227                                          sizeof(int32_t) * newlength, status);
228         if (U_FAILURE(*status)) {
229             return NULL;
230         }
231         uprv_memcpy(temp, destination, sizeof(int32_t) * offset);
232         *destinationlength = newlength;
233         destination        = temp;
234     }
235     destination[offset] = value;
236     return destination;
237 }
238 
239 /**
240 * Adds a uint64_t value to a destination array.
241 * Creates a new array if we run out of space. The caller will have to
242 * manually deallocate the newly allocated array.
243 * Internal method, status assumed to be success, caller has to check status
244 * before calling this method. destination not to be NULL and has at least
245 * size destinationlength.
246 * @param destination target array
247 * @param offset destination offset to add value
248 * @param destinationlength target array size, return value for the new size
249 * @param value to be added
250 * @param increments incremental size expected
251 * @param status output error if any, caller to check status before calling
252 *               method, status assumed to be success when passed in.
253 * @return new destination array, destination if there was no new allocation
254 */
255 static
addTouint64_tArray(int64_t * destination,uint32_t offset,uint32_t * destinationlength,uint64_t value,uint32_t increments,UErrorCode * status)256 inline int64_t * addTouint64_tArray(int64_t    *destination,
257                                     uint32_t    offset,
258                                     uint32_t   *destinationlength,
259                                     uint64_t    value,
260                                     uint32_t    increments,
261                                     UErrorCode *status)
262 {
263     uint32_t newlength = *destinationlength;
264     if (offset + 1 == newlength) {
265         newlength += increments;
266         int64_t *temp = (int64_t *)allocateMemory(
267                                          sizeof(int64_t) * newlength, status);
268 
269         if (U_FAILURE(*status)) {
270             return NULL;
271         }
272 
273         uprv_memcpy(temp, destination, sizeof(int64_t) * offset);
274         *destinationlength = newlength;
275         destination        = temp;
276     }
277 
278     destination[offset] = value;
279 
280     return destination;
281 }
282 
283 /**
284 * Initializing the ce table for a pattern.
285 * Stores non-ignorable collation keys.
286 * Table size will be estimated by the size of the pattern text. Table
287 * expansion will be perform as we go along. Adding 1 to ensure that the table
288 * size definitely increases.
289 * Internal method, status assumed to be a success.
290 * @param strsrch string search data
291 * @param status output error if any, caller to check status before calling
292 *               method, status assumed to be success when passed in.
293 * @return total number of expansions
294 */
295 static
initializePatternCETable(UStringSearch * strsrch,UErrorCode * status)296 inline uint16_t initializePatternCETable(UStringSearch *strsrch,
297                                          UErrorCode    *status)
298 {
299     UPattern *pattern            = &(strsrch->pattern);
300     uint32_t  cetablesize        = INITIAL_ARRAY_SIZE_;
301     int32_t  *cetable            = pattern->CEBuffer;
302     uint32_t  patternlength      = pattern->textLength;
303     UCollationElements *coleiter = strsrch->utilIter;
304 
305     if (coleiter == NULL) {
306         coleiter = ucol_openElements(strsrch->collator, pattern->text,
307                                      patternlength, status);
308         // status will be checked in ucol_next(..) later and if it is an
309         // error UCOL_NULLORDER the result of ucol_next(..) and 0 will be
310         // returned.
311         strsrch->utilIter = coleiter;
312     }
313     else {
314         uprv_init_collIterate(strsrch->collator, pattern->text,
315                          pattern->textLength,
316                          &coleiter->iteratordata_,
317                          status);
318     }
319     if(U_FAILURE(*status)) {
320         return 0;
321     }
322 
323     if (pattern->CE != cetable && pattern->CE) {
324         uprv_free(pattern->CE);
325     }
326 
327     uint16_t  offset      = 0;
328     uint16_t  result      = 0;
329     int32_t   ce;
330 
331     while ((ce = ucol_next(coleiter, status)) != UCOL_NULLORDER &&
332            U_SUCCESS(*status)) {
333         uint32_t newce = getCE(strsrch, ce);
334         if (newce) {
335             int32_t *temp = addTouint32_tArray(cetable, offset, &cetablesize,
336                                   newce,
337                                   patternlength - ucol_getOffset(coleiter) + 1,
338                                   status);
339             if (U_FAILURE(*status)) {
340                 return 0;
341             }
342             offset ++;
343             if (cetable != temp && cetable != pattern->CEBuffer) {
344                 uprv_free(cetable);
345             }
346             cetable = temp;
347         }
348         result += (uint16_t)(ucol_getMaxExpansion(coleiter, ce) - 1);
349     }
350 
351     cetable[offset]   = 0;
352     pattern->CE       = cetable;
353     pattern->CELength = offset;
354 
355     return result;
356 }
357 
358 /**
359 * Initializing the pce table for a pattern.
360 * Stores non-ignorable collation keys.
361 * Table size will be estimated by the size of the pattern text. Table
362 * expansion will be perform as we go along. Adding 1 to ensure that the table
363 * size definitely increases.
364 * Internal method, status assumed to be a success.
365 * @param strsrch string search data
366 * @param status output error if any, caller to check status before calling
367 *               method, status assumed to be success when passed in.
368 * @return total number of expansions
369 */
370 static
initializePatternPCETable(UStringSearch * strsrch,UErrorCode * status)371 inline uint16_t initializePatternPCETable(UStringSearch *strsrch,
372                                           UErrorCode    *status)
373 {
374     UPattern *pattern            = &(strsrch->pattern);
375     uint32_t  pcetablesize       = INITIAL_ARRAY_SIZE_;
376     int64_t  *pcetable           = pattern->PCEBuffer;
377     uint32_t  patternlength      = pattern->textLength;
378     UCollationElements *coleiter = strsrch->utilIter;
379 
380     if (coleiter == NULL) {
381         coleiter = ucol_openElements(strsrch->collator, pattern->text,
382                                      patternlength, status);
383         // status will be checked in ucol_next(..) later and if it is an
384         // error UCOL_NULLORDER the result of ucol_next(..) and 0 will be
385         // returned.
386         strsrch->utilIter = coleiter;
387     } else {
388         uprv_init_collIterate(strsrch->collator, pattern->text,
389                               pattern->textLength,
390                               &coleiter->iteratordata_,
391                               status);
392     }
393     if(U_FAILURE(*status)) {
394         return 0;
395     }
396 
397     if (pattern->PCE != pcetable && pattern->PCE != NULL) {
398         uprv_free(pattern->PCE);
399     }
400 
401     uint16_t  offset = 0;
402     uint16_t  result = 0;
403     int64_t   pce;
404 
405     uprv_init_pce(coleiter);
406 
407     // ** Should processed CEs be signed or unsigned?
408     // ** (the rest of the code in this file seems to play fast-and-loose with
409     // **  whether a CE is signed or unsigned. For example, look at routine above this one.)
410     while ((pce = ucol_nextProcessed(coleiter, NULL, NULL, status)) != UCOL_PROCESSED_NULLORDER &&
411            U_SUCCESS(*status)) {
412         int64_t *temp = addTouint64_tArray(pcetable, offset, &pcetablesize,
413                               pce,
414                               patternlength - ucol_getOffset(coleiter) + 1,
415                               status);
416 
417         if (U_FAILURE(*status)) {
418             return 0;
419         }
420 
421         offset += 1;
422 
423         if (pcetable != temp && pcetable != pattern->PCEBuffer) {
424             uprv_free(pcetable);
425         }
426 
427         pcetable = temp;
428         //result += (uint16_t)(ucol_getMaxExpansion(coleiter, ce) - 1);
429     }
430 
431     pcetable[offset]   = 0;
432     pattern->PCE       = pcetable;
433     pattern->PCELength = offset;
434 
435     return result;
436 }
437 
438 /**
439 * Initializes the pattern struct.
440 * Internal method, status assumed to be success.
441 * @param strsrch UStringSearch data storage
442 * @param status output error if any, caller to check status before calling
443 *               method, status assumed to be success when passed in.
444 * @return expansionsize the total expansion size of the pattern
445 */
446 static
initializePattern(UStringSearch * strsrch,UErrorCode * status)447 inline int16_t initializePattern(UStringSearch *strsrch, UErrorCode *status)
448 {
449           UPattern   *pattern     = &(strsrch->pattern);
450     const UChar      *patterntext = pattern->text;
451           int32_t     length      = pattern->textLength;
452           int32_t index       = 0;
453 
454     // Since the strength is primary, accents are ignored in the pattern.
455     if (strsrch->strength == UCOL_PRIMARY) {
456     	pattern->hasPrefixAccents = 0;
457     	pattern->hasSuffixAccents = 0;
458     } else {
459 	    pattern->hasPrefixAccents = getFCD(patterntext, &index, length) >>
460 	                                                     SECOND_LAST_BYTE_SHIFT_;
461 	    index = length;
462 	    UTF_BACK_1(patterntext, 0, index);
463 	    pattern->hasSuffixAccents = getFCD(patterntext, &index, length) &
464 	                                                             LAST_BYTE_MASK_;
465     }
466 
467     // ** HACK **
468     if (strsrch->pattern.PCE != NULL) {
469         if (strsrch->pattern.PCE != strsrch->pattern.PCEBuffer) {
470             uprv_free(strsrch->pattern.PCE);
471         }
472 
473         strsrch->pattern.PCE = NULL;
474     }
475 
476     // since intializePattern is an internal method status is a success.
477     return initializePatternCETable(strsrch, status);
478 }
479 
480 /**
481 * Initializing shift tables, with the default values.
482 * If a corresponding default value is 0, the shift table is not set.
483 * @param shift table for forwards shift
484 * @param backshift table for backwards shift
485 * @param cetable table containing pattern ce
486 * @param cesize size of the pattern ces
487 * @param expansionsize total size of the expansions
488 * @param defaultforward the default forward value
489 * @param defaultbackward the default backward value
490 */
491 static
setShiftTable(int16_t shift[],int16_t backshift[],int32_t * cetable,int32_t cesize,int16_t expansionsize,int16_t defaultforward,int16_t defaultbackward)492 inline void setShiftTable(int16_t   shift[], int16_t backshift[],
493                           int32_t  *cetable, int32_t cesize,
494                           int16_t   expansionsize,
495                           int16_t   defaultforward,
496                           int16_t   defaultbackward)
497 {
498     // estimate the value to shift. to do that we estimate the smallest
499     // number of characters to give the relevant ces, ie approximately
500     // the number of ces minus their expansion, since expansions can come
501     // from a character.
502     int32_t count;
503     for (count = 0; count < MAX_TABLE_SIZE_; count ++) {
504         shift[count] = defaultforward;
505     }
506     cesize --; // down to the last index
507     for (count = 0; count < cesize; count ++) {
508         // number of ces from right of array to the count
509         int temp = defaultforward - count - 1;
510         shift[hash(cetable[count])] = temp > 1 ? temp : 1;
511     }
512     shift[hash(cetable[cesize])] = 1;
513     // for ignorables we just shift by one. see test examples.
514     shift[hash(0)] = 1;
515 
516     for (count = 0; count < MAX_TABLE_SIZE_; count ++) {
517         backshift[count] = defaultbackward;
518     }
519     for (count = cesize; count > 0; count --) {
520         // the original value count does not seem to work
521         backshift[hash(cetable[count])] = count > expansionsize ?
522                                           (int16_t)(count - expansionsize) : 1;
523     }
524     backshift[hash(cetable[0])] = 1;
525     backshift[hash(0)] = 1;
526 }
527 
528 /**
529 * Building of the pattern collation element list and the boyer moore strsrch
530 * table.
531 * The canonical match will only be performed after the default match fails.
532 * For both cases we need to remember the size of the composed and decomposed
533 * versions of the string. Since the Boyer-Moore shift calculations shifts by
534 * a number of characters in the text and tries to match the pattern from that
535 * offset, the shift value can not be too large in case we miss some
536 * characters. To choose a right shift size, we estimate the NFC form of the
537 * and use its size as a shift guide. The NFC form should be the small
538 * possible representation of the pattern. Anyways, we'll err on the smaller
539 * shift size. Hence the calculation for minlength.
540 * Canonical match will be performed slightly differently. We'll split the
541 * pattern into 3 parts, the prefix accents (PA), the middle string bounded by
542 * the first and last base character (MS), the ending accents (EA). Matches
543 * will be done on MS first, and only when we match MS then some processing
544 * will be required for the prefix and end accents in order to determine if
545 * they match PA and EA. Hence the default shift values
546 * for the canonical match will take the size of either end's accent into
547 * consideration. Forwards search will take the end accents into consideration
548 * for the default shift values and the backwards search will take the prefix
549 * accents into consideration.
550 * If pattern has no non-ignorable ce, we return a illegal argument error.
551 * Internal method, status assumed to be success.
552 * @param strsrch UStringSearch data storage
553 * @param status  for output errors if it occurs, status is assumed to be a
554 *                success when it is passed in.
555 */
556 static
initialize(UStringSearch * strsrch,UErrorCode * status)557 inline void initialize(UStringSearch *strsrch, UErrorCode *status)
558 {
559     int16_t expandlength  = initializePattern(strsrch, status);
560     if (U_SUCCESS(*status) && strsrch->pattern.CELength > 0) {
561         UPattern *pattern = &strsrch->pattern;
562         int32_t   cesize  = pattern->CELength;
563 
564         int16_t minlength = cesize > expandlength
565                             ? (int16_t)cesize - expandlength : 1;
566         pattern->defaultShiftSize    = minlength;
567         setShiftTable(pattern->shift, pattern->backShift, pattern->CE,
568                       cesize, expandlength, minlength, minlength);
569         return;
570     }
571     strsrch->pattern.defaultShiftSize = 0;
572 }
573 
574 #if BOYER_MOORE
575 /**
576 * Check to make sure that the match length is at the end of the character by
577 * using the breakiterator.
578 * @param strsrch string search data
579 * @param start target text start offset
580 * @param end target text end offset
581 */
582 static
checkBreakBoundary(const UStringSearch * strsrch,int32_t *,int32_t * end)583 void checkBreakBoundary(const UStringSearch *strsrch, int32_t * /*start*/,
584                                int32_t *end)
585 {
586 #if !UCONFIG_NO_BREAK_ITERATION
587     UBreakIterator *breakiterator = strsrch->search->internalBreakIter;
588     if (breakiterator) {
589 	    int32_t matchend = *end;
590 	    //int32_t matchstart = *start;
591 
592 	    if (!ubrk_isBoundary(breakiterator, matchend)) {
593 	    	*end = ubrk_following(breakiterator, matchend);
594         }
595 
596 	    /* Check the start of the matched text to make sure it doesn't have any accents
597 	     * before it.  This code may not be necessary and so it is commented out */
598 	    /*if (!ubrk_isBoundary(breakiterator, matchstart) && !ubrk_isBoundary(breakiterator, matchstart-1)) {
599 	    	*start = ubrk_preceding(breakiterator, matchstart);
600 	    }*/
601     }
602 #endif
603 }
604 
605 /**
606 * Determine whether the target text in UStringSearch bounded by the offset
607 * start and end is one or more whole units of text as
608 * determined by the breakiterator in UStringSearch.
609 * @param strsrch string search data
610 * @param start target text start offset
611 * @param end target text end offset
612 */
613 static
isBreakUnit(const UStringSearch * strsrch,int32_t start,int32_t end)614 UBool isBreakUnit(const UStringSearch *strsrch, int32_t start,
615                                int32_t    end)
616 {
617 #if !UCONFIG_NO_BREAK_ITERATION
618     UBreakIterator *breakiterator = strsrch->search->breakIter;
619     //TODO: Add here.
620     if (breakiterator) {
621         int32_t startindex = ubrk_first(breakiterator);
622         int32_t endindex   = ubrk_last(breakiterator);
623 
624         // out-of-range indexes are never boundary positions
625         if (start < startindex || start > endindex ||
626             end < startindex || end > endindex) {
627             return FALSE;
628         }
629         // otherwise, we can use following() on the position before the
630         // specified one and return true of the position we get back is the
631         // one the user specified
632         UBool result = (start == startindex ||
633                 ubrk_following(breakiterator, start - 1) == start) &&
634                (end == endindex ||
635                 ubrk_following(breakiterator, end - 1) == end);
636         if (result) {
637             // iterates the individual ces
638                   UCollationElements *coleiter  = strsrch->utilIter;
639             const UChar              *text      = strsrch->search->text +
640                                                                       start;
641                   UErrorCode          status    = U_ZERO_ERROR;
642             ucol_setText(coleiter, text, end - start, &status);
643             for (int32_t count = 0; count < strsrch->pattern.CELength;
644                  count ++) {
645                 int32_t ce = getCE(strsrch, ucol_next(coleiter, &status));
646                 if (ce == UCOL_IGNORABLE) {
647                     count --;
648                     continue;
649                 }
650                 if (U_FAILURE(status) || ce != strsrch->pattern.CE[count]) {
651                     return FALSE;
652                 }
653             }
654             int32_t nextce = ucol_next(coleiter, &status);
655             while (ucol_getOffset(coleiter) == (end - start)
656                    && getCE(strsrch, nextce) == UCOL_IGNORABLE) {
657                 nextce = ucol_next(coleiter, &status);
658             }
659             if (ucol_getOffset(coleiter) == (end - start)
660                 && nextce != UCOL_NULLORDER) {
661                 // extra collation elements at the end of the match
662                 return FALSE;
663             }
664         }
665         return result;
666     }
667 #endif
668     return TRUE;
669 }
670 
671 /**
672 * Getting the next base character offset if current offset is an accent,
673 * or the current offset if the current character contains a base character.
674 * accents the following base character will be returned
675 * @param text string
676 * @param textoffset current offset
677 * @param textlength length of text string
678 * @return the next base character or the current offset
679 *         if the current character is contains a base character.
680 */
681 static
getNextBaseOffset(const UChar * text,int32_t textoffset,int32_t textlength)682 inline int32_t getNextBaseOffset(const UChar       *text,
683                                            int32_t  textoffset,
684                                            int32_t      textlength)
685 {
686     if (textoffset < textlength) {
687         int32_t temp = textoffset;
688         if (getFCD(text, &temp, textlength) >> SECOND_LAST_BYTE_SHIFT_) {
689             while (temp < textlength) {
690                 int32_t result = temp;
691                 if ((getFCD(text, &temp, textlength) >>
692                      SECOND_LAST_BYTE_SHIFT_) == 0) {
693                     return result;
694                 }
695             }
696             return textlength;
697         }
698     }
699     return textoffset;
700 }
701 
702 /**
703 * Gets the next base character offset depending on the string search pattern
704 * data
705 * @param strsrch string search data
706 * @param textoffset current offset, one offset away from the last character
707 *                   to search for.
708 * @return start index of the next base character or the current offset
709 *         if the current character is contains a base character.
710 */
711 static
getNextUStringSearchBaseOffset(UStringSearch * strsrch,int32_t textoffset)712 inline int32_t getNextUStringSearchBaseOffset(UStringSearch *strsrch,
713                                                   int32_t    textoffset)
714 {
715     int32_t textlength = strsrch->search->textLength;
716     if (strsrch->pattern.hasSuffixAccents &&
717         textoffset < textlength) {
718               int32_t  temp       = textoffset;
719         const UChar       *text       = strsrch->search->text;
720         UTF_BACK_1(text, 0, temp);
721         if (getFCD(text, &temp, textlength) & LAST_BYTE_MASK_) {
722             return getNextBaseOffset(text, textoffset, textlength);
723         }
724     }
725     return textoffset;
726 }
727 
728 /**
729 * Shifting the collation element iterator position forward to prepare for
730 * a following match. If the last character is a unsafe character, we'll only
731 * shift by 1 to capture contractions, normalization etc.
732 * Internal method, status assumed to be success.
733 * @param text strsrch string search data
734 * @param textoffset start text position to do search
735 * @param ce the text ce which failed the match.
736 * @param patternceindex index of the ce within the pattern ce buffer which
737 *        failed the match
738 * @return final offset
739 */
740 static
shiftForward(UStringSearch * strsrch,int32_t textoffset,int32_t ce,int32_t patternceindex)741 inline int32_t shiftForward(UStringSearch *strsrch,
742                                 int32_t    textoffset,
743                                 int32_t       ce,
744                                 int32_t        patternceindex)
745 {
746     UPattern *pattern = &(strsrch->pattern);
747     if (ce != UCOL_NULLORDER) {
748         int32_t shift = pattern->shift[hash(ce)];
749         // this is to adjust for characters in the middle of the
750         // substring for matching that failed.
751         int32_t adjust = pattern->CELength - patternceindex;
752         if (adjust > 1 && shift >= adjust) {
753             shift -= adjust - 1;
754         }
755         textoffset += shift;
756     }
757     else {
758         textoffset += pattern->defaultShiftSize;
759     }
760 
761     textoffset = getNextUStringSearchBaseOffset(strsrch, textoffset);
762     // check for unsafe characters
763     // * if it is the start or middle of a contraction: to be done after
764     //   a initial match is found
765     // * thai or lao base consonant character: similar to contraction
766     // * high surrogate character: similar to contraction
767     // * next character is a accent: shift to the next base character
768     return textoffset;
769 }
770 #endif // #if BOYER_MOORE
771 
772 /**
773 * sets match not found
774 * @param strsrch string search data
775 */
776 static
setMatchNotFound(UStringSearch * strsrch)777 inline void setMatchNotFound(UStringSearch *strsrch)
778 {
779     // this method resets the match result regardless of the error status.
780     strsrch->search->matchedIndex = USEARCH_DONE;
781     strsrch->search->matchedLength = 0;
782     if (strsrch->search->isForwardSearching) {
783         setColEIterOffset(strsrch->textIter, strsrch->search->textLength);
784     }
785     else {
786         setColEIterOffset(strsrch->textIter, 0);
787     }
788 }
789 
790 #if BOYER_MOORE
791 /**
792 * Gets the offset to the next safe point in text.
793 * ie. not the middle of a contraction, swappable characters or supplementary
794 * characters.
795 * @param collator collation sata
796 * @param text string to work with
797 * @param textoffset offset in string
798 * @param textlength length of text string
799 * @return offset to the next safe character
800 */
801 static
getNextSafeOffset(const UCollator * collator,const UChar * text,int32_t textoffset,int32_t textlength)802 inline int32_t getNextSafeOffset(const UCollator   *collator,
803                                      const UChar       *text,
804                                            int32_t  textoffset,
805                                            int32_t      textlength)
806 {
807     int32_t result = textoffset; // first contraction character
808     while (result != textlength && ucol_unsafeCP(text[result], collator)) {
809         result ++;
810     }
811     return result;
812 }
813 
814 /**
815 * This checks for accents in the potential match started with a .
816 * composite character.
817 * This is really painful... we have to check that composite character do not
818 * have any extra accents. We have to normalize the potential match and find
819 * the immediate decomposed character before the match.
820 * The first composite character would have been taken care of by the fcd
821 * checks in checkForwardExactMatch.
822 * This is the slow path after the fcd of the first character and
823 * the last character has been checked by checkForwardExactMatch and we
824 * determine that the potential match has extra non-ignorable preceding
825 * ces.
826 * E.g. looking for \u0301 acute in \u01FA A ring above and acute,
827 * checkExtraMatchAccent should fail since there is a middle ring in \u01FA
828 * Note here that accents checking are slow and cautioned in the API docs.
829 * Internal method, status assumed to be a success, caller should check status
830 * before calling this method
831 * @param strsrch string search data
832 * @param start index of the potential unfriendly composite character
833 * @param end index of the potential unfriendly composite character
834 * @param status output error status if any.
835 * @return TRUE if there is non-ignorable accents before at the beginning
836 *              of the match, FALSE otherwise.
837 */
838 
839 static
checkExtraMatchAccents(const UStringSearch * strsrch,int32_t start,int32_t end,UErrorCode * status)840 UBool checkExtraMatchAccents(const UStringSearch *strsrch, int32_t start,
841                                    int32_t    end,
842                                    UErrorCode    *status)
843 {
844     UBool result = FALSE;
845     if (strsrch->pattern.hasPrefixAccents) {
846               int32_t  length = end - start;
847               int32_t  offset = 0;
848         const UChar       *text   = strsrch->search->text + start;
849 
850         UTF_FWD_1(text, offset, length);
851         // we are only concerned with the first composite character
852         if (unorm_quickCheck(text, offset, UNORM_NFD, status) == UNORM_NO) {
853             int32_t safeoffset = getNextSafeOffset(strsrch->collator,
854                                                        text, 0, length);
855             if (safeoffset != length) {
856                 safeoffset ++;
857             }
858             UChar   *norm = NULL;
859             UChar    buffer[INITIAL_ARRAY_SIZE_];
860             int32_t  size = unorm_normalize(text, safeoffset, UNORM_NFD, 0,
861                                             buffer, INITIAL_ARRAY_SIZE_,
862                                             status);
863             if (U_FAILURE(*status)) {
864                 return FALSE;
865             }
866             if (size >= INITIAL_ARRAY_SIZE_) {
867                 norm = (UChar *)allocateMemory((size + 1) * sizeof(UChar),
868                                                status);
869                 // if allocation failed, status will be set to
870                 // U_MEMORY_ALLOCATION_ERROR and unorm_normalize internally
871                 // checks for it.
872                 size = unorm_normalize(text, safeoffset, UNORM_NFD, 0, norm,
873                                        size, status);
874                 if (U_FAILURE(*status) && norm != NULL) {
875                     uprv_free(norm);
876                     return FALSE;
877                 }
878             }
879             else {
880                 norm = buffer;
881             }
882 
883             UCollationElements *coleiter  = strsrch->utilIter;
884             ucol_setText(coleiter, norm, size, status);
885             uint32_t            firstce   = strsrch->pattern.CE[0];
886             UBool               ignorable = TRUE;
887             uint32_t            ce        = UCOL_IGNORABLE;
888             while (U_SUCCESS(*status) && ce != firstce && ce != (uint32_t)UCOL_NULLORDER) {
889                 offset = ucol_getOffset(coleiter);
890                 if (ce != firstce && ce != UCOL_IGNORABLE) {
891                     ignorable = FALSE;
892                 }
893                 ce = ucol_next(coleiter, status);
894             }
895             UChar32 codepoint;
896             UTF_PREV_CHAR(norm, 0, offset, codepoint);
897             result = !ignorable && (u_getCombiningClass(codepoint) != 0);
898 
899             if (norm != buffer) {
900                 uprv_free(norm);
901             }
902         }
903     }
904 
905     return result;
906 }
907 
908 /**
909 * Used by exact matches, checks if there are accents before the match.
910 * This is really painful... we have to check that composite characters at
911 * the start of the matches have to not have any extra accents.
912 * We check the FCD of the character first, if it starts with an accent and
913 * the first pattern ce does not match the first ce of the character, we bail.
914 * Otherwise we try normalizing the first composite
915 * character and find the immediate decomposed character before the match to
916 * see if it is an non-ignorable accent.
917 * Now normalizing the first composite character is enough because we ensure
918 * that when the match is passed in here with extra beginning ces, the
919 * first or last ce that match has to occur within the first character.
920 * E.g. looking for \u0301 acute in \u01FA A ring above and acute,
921 * checkExtraMatchAccent should fail since there is a middle ring in \u01FA
922 * Note here that accents checking are slow and cautioned in the API docs.
923 * @param strsrch string search data
924 * @param start offset
925 * @param end offset
926 * @return TRUE if there are accents on either side of the match,
927 *         FALSE otherwise
928 */
929 static
hasAccentsBeforeMatch(const UStringSearch * strsrch,int32_t start,int32_t end)930 UBool hasAccentsBeforeMatch(const UStringSearch *strsrch, int32_t start,
931                                   int32_t    end)
932 {
933     if (strsrch->pattern.hasPrefixAccents) {
934         UCollationElements *coleiter  = strsrch->textIter;
935         UErrorCode          status    = U_ZERO_ERROR;
936         // we have been iterating forwards previously
937         uint32_t            ignorable = TRUE;
938         int32_t             firstce   = strsrch->pattern.CE[0];
939 
940         setColEIterOffset(coleiter, start);
941         int32_t ce  = getCE(strsrch, ucol_next(coleiter, &status));
942         if (U_FAILURE(status)) {
943             return TRUE;
944         }
945         while (ce != firstce) {
946             if (ce != UCOL_IGNORABLE) {
947                 ignorable = FALSE;
948             }
949             ce = getCE(strsrch, ucol_next(coleiter, &status));
950             if (U_FAILURE(status) || ce == UCOL_NULLORDER) {
951                 return TRUE;
952             }
953         }
954         if (!ignorable && inNormBuf(coleiter)) {
955             // within normalization buffer, discontiguous handled here
956             return TRUE;
957         }
958 
959         // within text
960         int32_t temp = start;
961         // original code
962         // accent = (getFCD(strsrch->search->text, &temp,
963         //                  strsrch->search->textLength)
964         //            >> SECOND_LAST_BYTE_SHIFT_);
965         // however this code does not work well with VC7 .net in release mode.
966         // maybe the inlines for getFCD combined with shifting has bugs in
967         // VC7. anyways this is a work around.
968         UBool accent = getFCD(strsrch->search->text, &temp,
969                               strsrch->search->textLength) > 0xFF;
970         if (!accent) {
971             return checkExtraMatchAccents(strsrch, start, end, &status);
972         }
973         if (!ignorable) {
974             return TRUE;
975         }
976         if (start > 0) {
977             temp = start;
978             UTF_BACK_1(strsrch->search->text, 0, temp);
979             if (getFCD(strsrch->search->text, &temp,
980                        strsrch->search->textLength) & LAST_BYTE_MASK_) {
981                 setColEIterOffset(coleiter, start);
982                 ce = ucol_previous(coleiter, &status);
983                 if (U_FAILURE(status) ||
984                     (ce != UCOL_NULLORDER && ce != UCOL_IGNORABLE)) {
985                     return TRUE;
986                 }
987             }
988         }
989     }
990 
991     return FALSE;
992 }
993 
994 /**
995 * Used by exact matches, checks if there are accents bounding the match.
996 * Note this is the initial boundary check. If the potential match
997 * starts or ends with composite characters, the accents in those
998 * characters will be determined later.
999 * Not doing backwards iteration here, since discontiguos contraction for
1000 * backwards collation element iterator, use up too many characters.
1001 * E.g. looking for \u030A ring in \u01FA A ring above and acute,
1002 * should fail since there is a acute at the end of \u01FA
1003 * Note here that accents checking are slow and cautioned in the API docs.
1004 * @param strsrch string search data
1005 * @param start offset of match
1006 * @param end end offset of the match
1007 * @return TRUE if there are accents on either side of the match,
1008 *         FALSE otherwise
1009 */
1010 static
hasAccentsAfterMatch(const UStringSearch * strsrch,int32_t start,int32_t end)1011 UBool hasAccentsAfterMatch(const UStringSearch *strsrch, int32_t start,
1012                                  int32_t    end)
1013 {
1014     if (strsrch->pattern.hasSuffixAccents) {
1015         const UChar       *text       = strsrch->search->text;
1016               int32_t  temp       = end;
1017               int32_t      textlength = strsrch->search->textLength;
1018         UTF_BACK_1(text, 0, temp);
1019         if (getFCD(text, &temp, textlength) & LAST_BYTE_MASK_) {
1020             int32_t             firstce  = strsrch->pattern.CE[0];
1021             UCollationElements *coleiter = strsrch->textIter;
1022             UErrorCode          status   = U_ZERO_ERROR;
1023 			int32_t ce;
1024             setColEIterOffset(coleiter, start);
1025             while ((ce = getCE(strsrch, ucol_next(coleiter, &status))) != firstce) {
1026                 if (U_FAILURE(status) || ce == UCOL_NULLORDER) {
1027                     return TRUE;
1028                 }
1029             }
1030             int32_t count = 1;
1031             while (count < strsrch->pattern.CELength) {
1032                 if (getCE(strsrch, ucol_next(coleiter, &status))
1033                     == UCOL_IGNORABLE) {
1034                     // Thai can give an ignorable here.
1035                     count --;
1036                 }
1037                 if (U_FAILURE(status)) {
1038                     return TRUE;
1039                 }
1040                 count ++;
1041             }
1042 
1043 			ce = ucol_next(coleiter, &status);
1044             if (U_FAILURE(status)) {
1045                 return TRUE;
1046             }
1047             if (ce != UCOL_NULLORDER && ce != UCOL_IGNORABLE) {
1048             	ce = getCE(strsrch, ce);
1049             }
1050             if (ce != UCOL_NULLORDER && ce != UCOL_IGNORABLE) {
1051                 if (ucol_getOffset(coleiter) <= end) {
1052                     return TRUE;
1053                 }
1054                 if (getFCD(text, &end, textlength) >> SECOND_LAST_BYTE_SHIFT_) {
1055                     return TRUE;
1056                 }
1057             }
1058         }
1059     }
1060     return FALSE;
1061 }
1062 #endif // #if BOYER_MOORE
1063 
1064 /**
1065 * Checks if the offset runs out of the text string
1066 * @param offset
1067 * @param textlength of the text string
1068 * @return TRUE if offset is out of bounds, FALSE otherwise
1069 */
1070 static
isOutOfBounds(int32_t textlength,int32_t offset)1071 inline UBool isOutOfBounds(int32_t textlength, int32_t offset)
1072 {
1073     return offset < 0 || offset > textlength;
1074 }
1075 
1076 /**
1077 * Checks for identical match
1078 * @param strsrch string search data
1079 * @param start offset of possible match
1080 * @param end offset of possible match
1081 * @return TRUE if identical match is found
1082 */
1083 static
checkIdentical(const UStringSearch * strsrch,int32_t start,int32_t end)1084 inline UBool checkIdentical(const UStringSearch *strsrch, int32_t start,
1085                                   int32_t    end)
1086 {
1087     if (strsrch->strength != UCOL_IDENTICAL) {
1088         return TRUE;
1089     }
1090 
1091     // Note: We could use Normalizer::compare() or similar, but for short strings
1092     // which may not be in FCD it might be faster to just NFD them.
1093     UErrorCode status = U_ZERO_ERROR;
1094     UnicodeString t2, p2;
1095     strsrch->nfd->normalize(
1096         UnicodeString(FALSE, strsrch->search->text + start, end - start), t2, status);
1097     strsrch->nfd->normalize(
1098         UnicodeString(FALSE, strsrch->pattern.text, strsrch->pattern.textLength), p2, status);
1099     // return FALSE if NFD failed
1100     return U_SUCCESS(status) && t2 == p2;
1101 }
1102 
1103 #if BOYER_MOORE
1104 /**
1105 * Checks to see if the match is repeated
1106 * @param strsrch string search data
1107 * @param start new match start index
1108 * @param end new match end index
1109 * @return TRUE if the the match is repeated, FALSE otherwise
1110 */
1111 static
checkRepeatedMatch(UStringSearch * strsrch,int32_t start,int32_t end)1112 inline UBool checkRepeatedMatch(UStringSearch *strsrch,
1113                                 int32_t    start,
1114                                 int32_t    end)
1115 {
1116     int32_t lastmatchindex = strsrch->search->matchedIndex;
1117     UBool       result;
1118     if (lastmatchindex == USEARCH_DONE) {
1119         return FALSE;
1120     }
1121     if (strsrch->search->isForwardSearching) {
1122         result = start <= lastmatchindex;
1123     }
1124     else {
1125         result = start >= lastmatchindex;
1126     }
1127     if (!result && !strsrch->search->isOverlap) {
1128         if (strsrch->search->isForwardSearching) {
1129             result = start < lastmatchindex + strsrch->search->matchedLength;
1130         }
1131         else {
1132             result = end > lastmatchindex;
1133         }
1134     }
1135     return result;
1136 }
1137 
1138 /**
1139 * Gets the collation element iterator's current offset.
1140 * @param coleiter collation element iterator
1141 * @param forwards flag TRUE if we are moving in th forwards direction
1142 * @return current offset
1143 */
1144 static
getColElemIterOffset(const UCollationElements * coleiter,UBool forwards)1145 inline int32_t getColElemIterOffset(const UCollationElements *coleiter,
1146                                               UBool               forwards)
1147 {
1148     int32_t result = ucol_getOffset(coleiter);
1149     // intricacies of the the backwards collation element iterator
1150     if (FALSE && !forwards && inNormBuf(coleiter) && !isFCDPointerNull(coleiter)) {
1151         result ++;
1152     }
1153     return result;
1154 }
1155 
1156 /**
1157 * Checks match for contraction.
1158 * If the match ends with a partial contraction we fail.
1159 * If the match starts too far off (because of backwards iteration) we try to
1160 * chip off the extra characters depending on whether a breakiterator has
1161 * been used.
1162 * Internal method, error assumed to be success, caller has to check status
1163 * before calling this method.
1164 * @param strsrch string search data
1165 * @param start offset of potential match, to be modified if necessary
1166 * @param end offset of potential match, to be modified if necessary
1167 * @param status output error status if any
1168 * @return TRUE if match passes the contraction test, FALSE otherwise
1169 */
1170 
1171 static
checkNextExactContractionMatch(UStringSearch * strsrch,int32_t * start,int32_t * end,UErrorCode * status)1172 UBool checkNextExactContractionMatch(UStringSearch *strsrch,
1173                                      int32_t   *start,
1174                                      int32_t   *end, UErrorCode  *status)
1175 {
1176           UCollationElements *coleiter   = strsrch->textIter;
1177           int32_t             textlength = strsrch->search->textLength;
1178           int32_t             temp       = *start;
1179     const UCollator          *collator   = strsrch->collator;
1180     const UChar              *text       = strsrch->search->text;
1181     // This part checks if either ends of the match contains potential
1182     // contraction. If so we'll have to iterate through them
1183     // The start contraction needs to be checked since ucol_previous dumps
1184     // all characters till the first safe character into the buffer.
1185     // *start + 1 is used to test for the unsafe characters instead of *start
1186     // because ucol_prev takes all unsafe characters till the first safe
1187     // character ie *start. so by testing *start + 1, we can estimate if
1188     // excess prefix characters has been included in the potential search
1189     // results.
1190     if ((*end < textlength && ucol_unsafeCP(text[*end], collator)) ||
1191         (*start + 1 < textlength
1192          && ucol_unsafeCP(text[*start + 1], collator))) {
1193         int32_t expansion  = getExpansionPrefix(coleiter);
1194         UBool   expandflag = expansion > 0;
1195         setColEIterOffset(coleiter, *start);
1196         while (expansion > 0) {
1197             // getting rid of the redundant ce, caused by setOffset.
1198             // since backward contraction/expansion may have extra ces if we
1199             // are in the normalization buffer, hasAccentsBeforeMatch would
1200             // have taken care of it.
1201             // E.g. the character \u01FA will have an expansion of 3, but if
1202             // we are only looking for acute and ring \u030A and \u0301, we'll
1203             // have to skip the first ce in the expansion buffer.
1204             ucol_next(coleiter, status);
1205             if (U_FAILURE(*status)) {
1206                 return FALSE;
1207             }
1208             if (ucol_getOffset(coleiter) != temp) {
1209                 *start = temp;
1210                 temp  = ucol_getOffset(coleiter);
1211             }
1212             expansion --;
1213         }
1214 
1215         int32_t  *patternce       = strsrch->pattern.CE;
1216         int32_t   patterncelength = strsrch->pattern.CELength;
1217         int32_t   count           = 0;
1218         while (count < patterncelength) {
1219             int32_t ce = getCE(strsrch, ucol_next(coleiter, status));
1220             if (ce == UCOL_IGNORABLE) {
1221                 continue;
1222             }
1223             if (expandflag && count == 0 && ucol_getOffset(coleiter) != temp) {
1224                 *start = temp;
1225                 temp   = ucol_getOffset(coleiter);
1226             }
1227             if (U_FAILURE(*status) || ce != patternce[count]) {
1228                 (*end) ++;
1229                 *end = getNextUStringSearchBaseOffset(strsrch, *end);
1230                 return FALSE;
1231             }
1232             count ++;
1233         }
1234     }
1235     return TRUE;
1236 }
1237 
1238 /**
1239 * Checks and sets the match information if found.
1240 * Checks
1241 * <ul>
1242 * <li> the potential match does not repeat the previous match
1243 * <li> boundaries are correct
1244 * <li> exact matches has no extra accents
1245 * <li> identical matchesb
1246 * <li> potential match does not end in the middle of a contraction
1247 * <\ul>
1248 * Otherwise the offset will be shifted to the next character.
1249 * Internal method, status assumed to be success, caller has to check status
1250 * before calling this method.
1251 * @param strsrch string search data
1252 * @param textoffset offset in the collation element text. the returned value
1253 *        will be the truncated end offset of the match or the new start
1254 *        search offset.
1255 * @param status output error status if any
1256 * @return TRUE if the match is valid, FALSE otherwise
1257 */
1258 static
checkNextExactMatch(UStringSearch * strsrch,int32_t * textoffset,UErrorCode * status)1259 inline UBool checkNextExactMatch(UStringSearch *strsrch,
1260                                  int32_t   *textoffset, UErrorCode *status)
1261 {
1262     UCollationElements *coleiter = strsrch->textIter;
1263     int32_t         start    = getColElemIterOffset(coleiter, FALSE);
1264 
1265     if (!checkNextExactContractionMatch(strsrch, &start, textoffset, status)) {
1266         return FALSE;
1267     }
1268 
1269     // this totally matches, however we need to check if it is repeating
1270     if (!isBreakUnit(strsrch, start, *textoffset) ||
1271         checkRepeatedMatch(strsrch, start, *textoffset) ||
1272         hasAccentsBeforeMatch(strsrch, start, *textoffset) ||
1273         !checkIdentical(strsrch, start, *textoffset) ||
1274         hasAccentsAfterMatch(strsrch, start, *textoffset)) {
1275 
1276         (*textoffset) ++;
1277         *textoffset = getNextUStringSearchBaseOffset(strsrch, *textoffset);
1278         return FALSE;
1279     }
1280 
1281     //Add breakiterator boundary check for primary strength search.
1282     if (!strsrch->search->breakIter && strsrch->strength == UCOL_PRIMARY) {
1283     	checkBreakBoundary(strsrch, &start, textoffset);
1284     }
1285 
1286     // totally match, we will get rid of the ending ignorables.
1287     strsrch->search->matchedIndex  = start;
1288     strsrch->search->matchedLength = *textoffset - start;
1289     return TRUE;
1290 }
1291 
1292 /**
1293 * Getting the previous base character offset, or the current offset if the
1294 * current character is a base character
1295 * @param text string
1296 * @param textoffset one offset after the current character
1297 * @return the offset of the next character after the base character or the first
1298 *         composed character with accents
1299 */
1300 static
getPreviousBaseOffset(const UChar * text,int32_t textoffset)1301 inline int32_t getPreviousBaseOffset(const UChar       *text,
1302                                                int32_t  textoffset)
1303 {
1304     if (textoffset > 0) {
1305         for (;;) {
1306             int32_t result = textoffset;
1307             UTF_BACK_1(text, 0, textoffset);
1308             int32_t temp = textoffset;
1309             uint16_t fcd = getFCD(text, &temp, result);
1310             if ((fcd >> SECOND_LAST_BYTE_SHIFT_) == 0) {
1311                 if (fcd & LAST_BYTE_MASK_) {
1312                     return textoffset;
1313                 }
1314                 return result;
1315             }
1316             if (textoffset == 0) {
1317                 return 0;
1318             }
1319         }
1320     }
1321     return textoffset;
1322 }
1323 
1324 /**
1325 * Getting the indexes of the accents that are not blocked in the argument
1326 * accent array
1327 * @param accents array of accents in nfd terminated by a 0.
1328 * @param accentsindex array of indexes of the accents that are not blocked
1329 */
1330 static
getUnblockedAccentIndex(UChar * accents,int32_t * accentsindex)1331 inline int getUnblockedAccentIndex(UChar *accents, int32_t *accentsindex)
1332 {
1333     int32_t index     = 0;
1334     int32_t     length    = u_strlen(accents);
1335     UChar32     codepoint = 0;
1336     int         cclass    = 0;
1337     int         result    = 0;
1338     int32_t temp;
1339     while (index < length) {
1340         temp = index;
1341         UTF_NEXT_CHAR(accents, index, length, codepoint);
1342         if (u_getCombiningClass(codepoint) != cclass) {
1343             cclass        = u_getCombiningClass(codepoint);
1344             accentsindex[result] = temp;
1345             result ++;
1346         }
1347     }
1348     accentsindex[result] = length;
1349     return result;
1350 }
1351 
1352 /**
1353 * Appends 3 UChar arrays to a destination array.
1354 * Creates a new array if we run out of space. The caller will have to
1355 * manually deallocate the newly allocated array.
1356 * Internal method, status assumed to be success, caller has to check status
1357 * before calling this method. destination not to be NULL and has at least
1358 * size destinationlength.
1359 * @param destination target array
1360 * @param destinationlength target array size, returning the appended length
1361 * @param source1 null-terminated first array
1362 * @param source2 second array
1363 * @param source2length length of seond array
1364 * @param source3 null-terminated third array
1365 * @param status error status if any
1366 * @return new destination array, destination if there was no new allocation
1367 */
1368 static
addToUCharArray(UChar * destination,int32_t * destinationlength,const UChar * source1,const UChar * source2,int32_t source2length,const UChar * source3,UErrorCode * status)1369 inline UChar * addToUCharArray(      UChar      *destination,
1370                                      int32_t    *destinationlength,
1371                                const UChar      *source1,
1372                                const UChar      *source2,
1373                                      int32_t     source2length,
1374                                const UChar      *source3,
1375                                      UErrorCode *status)
1376 {
1377     int32_t source1length = source1 ? u_strlen(source1) : 0;
1378     int32_t source3length = source3 ? u_strlen(source3) : 0;
1379     if (*destinationlength < source1length + source2length + source3length +
1380                                                                            1)
1381     {
1382         destination = (UChar *)allocateMemory(
1383           (source1length + source2length + source3length + 1) * sizeof(UChar),
1384           status);
1385         // if error allocating memory, status will be
1386         // U_MEMORY_ALLOCATION_ERROR
1387         if (U_FAILURE(*status)) {
1388             *destinationlength = 0;
1389             return NULL;
1390         }
1391     }
1392     if (source1length != 0) {
1393         uprv_memcpy(destination, source1, sizeof(UChar) * source1length);
1394     }
1395     if (source2length != 0) {
1396         uprv_memcpy(destination + source1length, source2,
1397                     sizeof(UChar) * source2length);
1398     }
1399     if (source3length != 0) {
1400         uprv_memcpy(destination + source1length + source2length, source3,
1401                     sizeof(UChar) * source3length);
1402     }
1403     *destinationlength = source1length + source2length + source3length;
1404     return destination;
1405 }
1406 
1407 /**
1408 * Running through a collation element iterator to see if the contents matches
1409 * pattern in string search data
1410 * @param strsrch string search data
1411 * @param coleiter collation element iterator
1412 * @return TRUE if a match if found, FALSE otherwise
1413 */
1414 static
checkCollationMatch(const UStringSearch * strsrch,UCollationElements * coleiter)1415 inline UBool checkCollationMatch(const UStringSearch      *strsrch,
1416                                        UCollationElements *coleiter)
1417 {
1418     int         patternceindex = strsrch->pattern.CELength;
1419     int32_t    *patternce      = strsrch->pattern.CE;
1420     UErrorCode  status = U_ZERO_ERROR;
1421     while (patternceindex > 0) {
1422         int32_t ce = getCE(strsrch, ucol_next(coleiter, &status));
1423         if (ce == UCOL_IGNORABLE) {
1424             continue;
1425         }
1426         if (U_FAILURE(status) || ce != *patternce) {
1427             return FALSE;
1428         }
1429         patternce ++;
1430         patternceindex --;
1431     }
1432     return TRUE;
1433 }
1434 
1435 /**
1436 * Rearranges the front accents to try matching.
1437 * Prefix accents in the text will be grouped according to their combining
1438 * class and the groups will be mixed and matched to try find the perfect
1439 * match with the pattern.
1440 * So for instance looking for "\u0301" in "\u030A\u0301\u0325"
1441 * step 1: split "\u030A\u0301" into 6 other type of potential accent substrings
1442 *         "\u030A", "\u0301", "\u0325", "\u030A\u0301", "\u030A\u0325",
1443 *         "\u0301\u0325".
1444 * step 2: check if any of the generated substrings matches the pattern.
1445 * Internal method, status is assumed to be success, caller has to check status
1446 * before calling this method.
1447 * @param strsrch string search match
1448 * @param start first offset of the accents to start searching
1449 * @param end start of the last accent set
1450 * @param status output error status if any
1451 * @return USEARCH_DONE if a match is not found, otherwise return the starting
1452 *         offset of the match. Note this start includes all preceding accents.
1453 */
1454 static
doNextCanonicalPrefixMatch(UStringSearch * strsrch,int32_t start,int32_t end,UErrorCode * status)1455 int32_t doNextCanonicalPrefixMatch(UStringSearch *strsrch,
1456                                        int32_t    start,
1457                                        int32_t    end,
1458                                        UErrorCode    *status)
1459 {
1460     const UChar       *text       = strsrch->search->text;
1461           int32_t      textlength = strsrch->search->textLength;
1462           int32_t  tempstart  = start;
1463 
1464     if ((getFCD(text, &tempstart, textlength) & LAST_BYTE_MASK_) == 0) {
1465         // die... failed at a base character
1466         return USEARCH_DONE;
1467     }
1468 
1469     int32_t offset = getNextBaseOffset(text, tempstart, textlength);
1470     start = getPreviousBaseOffset(text, tempstart);
1471 
1472     UChar       accents[INITIAL_ARRAY_SIZE_];
1473     // normalizing the offensive string
1474     unorm_normalize(text + start, offset - start, UNORM_NFD, 0, accents,
1475                     INITIAL_ARRAY_SIZE_, status);
1476     if (U_FAILURE(*status)) {
1477         return USEARCH_DONE;
1478     }
1479 
1480     int32_t         accentsindex[INITIAL_ARRAY_SIZE_];
1481     int32_t         accentsize = getUnblockedAccentIndex(accents,
1482                                                                  accentsindex);
1483     int32_t         count      = (2 << (accentsize - 1)) - 1;
1484     UChar               buffer[INITIAL_ARRAY_SIZE_];
1485     UCollationElements *coleiter   = strsrch->utilIter;
1486     while (U_SUCCESS(*status) && count > 0) {
1487         UChar *rearrange = strsrch->canonicalPrefixAccents;
1488         // copy the base characters
1489         for (int k = 0; k < accentsindex[0]; k ++) {
1490             *rearrange ++ = accents[k];
1491         }
1492         // forming all possible canonical rearrangement by dropping
1493         // sets of accents
1494         for (int i = 0; i <= accentsize - 1; i ++) {
1495             int32_t mask = 1 << (accentsize - i - 1);
1496             if (count & mask) {
1497                 for (int j = accentsindex[i]; j < accentsindex[i + 1]; j ++) {
1498                     *rearrange ++ = accents[j];
1499                 }
1500             }
1501         }
1502         *rearrange = 0;
1503         int32_t  matchsize = INITIAL_ARRAY_SIZE_;
1504         UChar   *match     = addToUCharArray(buffer, &matchsize,
1505                                            strsrch->canonicalPrefixAccents,
1506                                            strsrch->search->text + offset,
1507                                            end - offset,
1508                                            strsrch->canonicalSuffixAccents,
1509                                            status);
1510 
1511         // if status is a failure, ucol_setText does nothing.
1512         // run the collator iterator through this match
1513         ucol_setText(coleiter, match, matchsize, status);
1514         if (U_SUCCESS(*status)) {
1515             if (checkCollationMatch(strsrch, coleiter)) {
1516                 if (match != buffer) {
1517                     uprv_free(match);
1518                 }
1519                 return start;
1520             }
1521         }
1522         count --;
1523     }
1524     return USEARCH_DONE;
1525 }
1526 
1527 /**
1528 * Gets the offset to the safe point in text before textoffset.
1529 * ie. not the middle of a contraction, swappable characters or supplementary
1530 * characters.
1531 * @param collator collation sata
1532 * @param text string to work with
1533 * @param textoffset offset in string
1534 * @param textlength length of text string
1535 * @return offset to the previous safe character
1536 */
1537 static
getPreviousSafeOffset(const UCollator * collator,const UChar * text,int32_t textoffset)1538 inline uint32_t getPreviousSafeOffset(const UCollator   *collator,
1539                                       const UChar       *text,
1540                                             int32_t  textoffset)
1541 {
1542     int32_t result = textoffset; // first contraction character
1543     while (result != 0 && ucol_unsafeCP(text[result - 1], collator)) {
1544         result --;
1545     }
1546     if (result != 0) {
1547         // the first contraction character is consider unsafe here
1548         result --;
1549     }
1550     return result;
1551 }
1552 
1553 /**
1554 * Cleaning up after we passed the safe zone
1555 * @param strsrch string search data
1556 * @param safetext safe text array
1557 * @param safebuffer safe text buffer
1558 * @param coleiter collation element iterator for safe text
1559 */
1560 static
cleanUpSafeText(const UStringSearch * strsrch,UChar * safetext,UChar * safebuffer)1561 inline void cleanUpSafeText(const UStringSearch *strsrch, UChar *safetext,
1562                                   UChar         *safebuffer)
1563 {
1564     if (safetext != safebuffer && safetext != strsrch->canonicalSuffixAccents)
1565     {
1566        uprv_free(safetext);
1567     }
1568 }
1569 
1570 /**
1571 * Take the rearranged end accents and tries matching. If match failed at
1572 * a seperate preceding set of accents (seperated from the rearranged on by
1573 * at least a base character) then we rearrange the preceding accents and
1574 * tries matching again.
1575 * We allow skipping of the ends of the accent set if the ces do not match.
1576 * However if the failure is found before the accent set, it fails.
1577 * Internal method, status assumed to be success, caller has to check status
1578 * before calling this method.
1579 * @param strsrch string search data
1580 * @param textoffset of the start of the rearranged accent
1581 * @param status output error status if any
1582 * @return USEARCH_DONE if a match is not found, otherwise return the starting
1583 *         offset of the match. Note this start includes all preceding accents.
1584 */
1585 static
doNextCanonicalSuffixMatch(UStringSearch * strsrch,int32_t textoffset,UErrorCode * status)1586 int32_t doNextCanonicalSuffixMatch(UStringSearch *strsrch,
1587                                        int32_t    textoffset,
1588                                        UErrorCode    *status)
1589 {
1590     const UChar              *text           = strsrch->search->text;
1591     const UCollator          *collator       = strsrch->collator;
1592           int32_t             safelength     = 0;
1593           UChar              *safetext;
1594           int32_t             safetextlength;
1595           UChar               safebuffer[INITIAL_ARRAY_SIZE_];
1596           UCollationElements *coleiter       = strsrch->utilIter;
1597           int32_t         safeoffset     = textoffset;
1598 
1599     if (textoffset != 0 && ucol_unsafeCP(strsrch->canonicalSuffixAccents[0],
1600                                          collator)) {
1601         safeoffset     = getPreviousSafeOffset(collator, text, textoffset);
1602         safelength     = textoffset - safeoffset;
1603         safetextlength = INITIAL_ARRAY_SIZE_;
1604         safetext       = addToUCharArray(safebuffer, &safetextlength, NULL,
1605                                          text + safeoffset, safelength,
1606                                          strsrch->canonicalSuffixAccents,
1607                                          status);
1608     }
1609     else {
1610         safetextlength = u_strlen(strsrch->canonicalSuffixAccents);
1611         safetext       = strsrch->canonicalSuffixAccents;
1612     }
1613 
1614     // if status is a failure, ucol_setText does nothing
1615     ucol_setText(coleiter, safetext, safetextlength, status);
1616     // status checked in loop below
1617 
1618     int32_t  *ce        = strsrch->pattern.CE;
1619     int32_t   celength  = strsrch->pattern.CELength;
1620     int       ceindex   = celength - 1;
1621     UBool     isSafe    = TRUE; // indication flag for position in safe zone
1622 
1623     while (ceindex >= 0) {
1624         int32_t textce = ucol_previous(coleiter, status);
1625         if (U_FAILURE(*status)) {
1626             if (isSafe) {
1627                 cleanUpSafeText(strsrch, safetext, safebuffer);
1628             }
1629             return USEARCH_DONE;
1630         }
1631         if (textce == UCOL_NULLORDER) {
1632             // check if we have passed the safe buffer
1633             if (coleiter == strsrch->textIter) {
1634                 cleanUpSafeText(strsrch, safetext, safebuffer);
1635                 return USEARCH_DONE;
1636             }
1637             cleanUpSafeText(strsrch, safetext, safebuffer);
1638             safetext = safebuffer;
1639             coleiter = strsrch->textIter;
1640             setColEIterOffset(coleiter, safeoffset);
1641             // status checked at the start of the loop
1642             isSafe = FALSE;
1643             continue;
1644         }
1645         textce = getCE(strsrch, textce);
1646         if (textce != UCOL_IGNORABLE && textce != ce[ceindex]) {
1647             // do the beginning stuff
1648             int32_t failedoffset = getColElemIterOffset(coleiter, FALSE);
1649             if (isSafe && failedoffset >= safelength) {
1650                 // alas... no hope. failed at rearranged accent set
1651                 cleanUpSafeText(strsrch, safetext, safebuffer);
1652                 return USEARCH_DONE;
1653             }
1654             else {
1655                 if (isSafe) {
1656                     failedoffset += safeoffset;
1657                     cleanUpSafeText(strsrch, safetext, safebuffer);
1658                 }
1659 
1660                 // try rearranging the front accents
1661                 int32_t result = doNextCanonicalPrefixMatch(strsrch,
1662                                         failedoffset, textoffset, status);
1663                 if (result != USEARCH_DONE) {
1664                     // if status is a failure, ucol_setOffset does nothing
1665                     setColEIterOffset(strsrch->textIter, result);
1666                 }
1667                 if (U_FAILURE(*status)) {
1668                     return USEARCH_DONE;
1669                 }
1670                 return result;
1671             }
1672         }
1673         if (textce == ce[ceindex]) {
1674             ceindex --;
1675         }
1676     }
1677     // set offset here
1678     if (isSafe) {
1679         int32_t result     = getColElemIterOffset(coleiter, FALSE);
1680         // sets the text iterator here with the correct expansion and offset
1681         int32_t    leftoverces = getExpansionPrefix(coleiter);
1682         cleanUpSafeText(strsrch, safetext, safebuffer);
1683         if (result >= safelength) {
1684             result = textoffset;
1685         }
1686         else {
1687             result += safeoffset;
1688         }
1689         setColEIterOffset(strsrch->textIter, result);
1690         strsrch->textIter->iteratordata_.toReturn =
1691                        setExpansionPrefix(strsrch->textIter, leftoverces);
1692         return result;
1693     }
1694 
1695     return ucol_getOffset(coleiter);
1696 }
1697 
1698 /**
1699 * Trying out the substring and sees if it can be a canonical match.
1700 * This will try normalizing the end accents and arranging them into canonical
1701 * equivalents and check their corresponding ces with the pattern ce.
1702 * Suffix accents in the text will be grouped according to their combining
1703 * class and the groups will be mixed and matched to try find the perfect
1704 * match with the pattern.
1705 * So for instance looking for "\u0301" in "\u030A\u0301\u0325"
1706 * step 1: split "\u030A\u0301" into 6 other type of potential accent substrings
1707 *         "\u030A", "\u0301", "\u0325", "\u030A\u0301", "\u030A\u0325",
1708 *         "\u0301\u0325".
1709 * step 2: check if any of the generated substrings matches the pattern.
1710 * Internal method, status assumed to be success, caller has to check status
1711 * before calling this method.
1712 * @param strsrch string search data
1713 * @param textoffset end offset in the collation element text that ends with
1714 *                   the accents to be rearranged
1715 * @param status error status if any
1716 * @return TRUE if the match is valid, FALSE otherwise
1717 */
1718 static
doNextCanonicalMatch(UStringSearch * strsrch,int32_t textoffset,UErrorCode * status)1719 UBool doNextCanonicalMatch(UStringSearch *strsrch,
1720                            int32_t    textoffset,
1721                            UErrorCode    *status)
1722 {
1723     const UChar       *text = strsrch->search->text;
1724           int32_t  temp = textoffset;
1725     UTF_BACK_1(text, 0, temp);
1726     if ((getFCD(text, &temp, textoffset) & LAST_BYTE_MASK_) == 0) {
1727         UCollationElements *coleiter = strsrch->textIter;
1728         int32_t         offset   = getColElemIterOffset(coleiter, FALSE);
1729         if (strsrch->pattern.hasPrefixAccents) {
1730             offset = doNextCanonicalPrefixMatch(strsrch, offset, textoffset,
1731                                                 status);
1732             if (U_SUCCESS(*status) && offset != USEARCH_DONE) {
1733                 setColEIterOffset(coleiter, offset);
1734                 return TRUE;
1735             }
1736         }
1737         return FALSE;
1738     }
1739 
1740     if (!strsrch->pattern.hasSuffixAccents) {
1741         return FALSE;
1742     }
1743 
1744     UChar       accents[INITIAL_ARRAY_SIZE_];
1745     // offset to the last base character in substring to search
1746     int32_t baseoffset = getPreviousBaseOffset(text, textoffset);
1747     // normalizing the offensive string
1748     unorm_normalize(text + baseoffset, textoffset - baseoffset, UNORM_NFD,
1749                                0, accents, INITIAL_ARRAY_SIZE_, status);
1750     // status checked in loop below
1751 
1752     int32_t accentsindex[INITIAL_ARRAY_SIZE_];
1753     int32_t size = getUnblockedAccentIndex(accents, accentsindex);
1754 
1755     // 2 power n - 1 plus the full set of accents
1756     int32_t  count = (2 << (size - 1)) - 1;
1757     while (U_SUCCESS(*status) && count > 0) {
1758         UChar *rearrange = strsrch->canonicalSuffixAccents;
1759         // copy the base characters
1760         for (int k = 0; k < accentsindex[0]; k ++) {
1761             *rearrange ++ = accents[k];
1762         }
1763         // forming all possible canonical rearrangement by dropping
1764         // sets of accents
1765         for (int i = 0; i <= size - 1; i ++) {
1766             int32_t mask = 1 << (size - i - 1);
1767             if (count & mask) {
1768                 for (int j = accentsindex[i]; j < accentsindex[i + 1]; j ++) {
1769                     *rearrange ++ = accents[j];
1770                 }
1771             }
1772         }
1773         *rearrange = 0;
1774         int32_t offset = doNextCanonicalSuffixMatch(strsrch, baseoffset,
1775                                                         status);
1776         if (offset != USEARCH_DONE) {
1777             return TRUE; // match found
1778         }
1779         count --;
1780     }
1781     return FALSE;
1782 }
1783 
1784 /**
1785 * Gets the previous base character offset depending on the string search
1786 * pattern data
1787 * @param strsrch string search data
1788 * @param textoffset current offset, current character
1789 * @return the offset of the next character after this base character or itself
1790 *         if it is a composed character with accents
1791 */
1792 static
getPreviousUStringSearchBaseOffset(UStringSearch * strsrch,int32_t textoffset)1793 inline int32_t getPreviousUStringSearchBaseOffset(UStringSearch *strsrch,
1794                                                       int32_t textoffset)
1795 {
1796     if (strsrch->pattern.hasPrefixAccents && textoffset > 0) {
1797         const UChar       *text = strsrch->search->text;
1798               int32_t  offset = textoffset;
1799         if (getFCD(text, &offset, strsrch->search->textLength) >>
1800                                                    SECOND_LAST_BYTE_SHIFT_) {
1801             return getPreviousBaseOffset(text, textoffset);
1802         }
1803     }
1804     return textoffset;
1805 }
1806 
1807 /**
1808 * Checks match for contraction.
1809 * If the match ends with a partial contraction we fail.
1810 * If the match starts too far off (because of backwards iteration) we try to
1811 * chip off the extra characters
1812 * Internal method, status assumed to be success, caller has to check status
1813 * before calling this method.
1814 * @param strsrch string search data
1815 * @param start offset of potential match, to be modified if necessary
1816 * @param end offset of potential match, to be modified if necessary
1817 * @param status output error status if any
1818 * @return TRUE if match passes the contraction test, FALSE otherwise
1819 */
1820 static
checkNextCanonicalContractionMatch(UStringSearch * strsrch,int32_t * start,int32_t * end,UErrorCode * status)1821 UBool checkNextCanonicalContractionMatch(UStringSearch *strsrch,
1822                                          int32_t   *start,
1823                                          int32_t   *end,
1824                                          UErrorCode    *status)
1825 {
1826           UCollationElements *coleiter   = strsrch->textIter;
1827           int32_t             textlength = strsrch->search->textLength;
1828           int32_t         temp       = *start;
1829     const UCollator          *collator   = strsrch->collator;
1830     const UChar              *text       = strsrch->search->text;
1831     // This part checks if either ends of the match contains potential
1832     // contraction. If so we'll have to iterate through them
1833     if ((*end < textlength && ucol_unsafeCP(text[*end], collator)) ||
1834         (*start + 1 < textlength
1835          && ucol_unsafeCP(text[*start + 1], collator))) {
1836         int32_t expansion  = getExpansionPrefix(coleiter);
1837         UBool   expandflag = expansion > 0;
1838         setColEIterOffset(coleiter, *start);
1839         while (expansion > 0) {
1840             // getting rid of the redundant ce, caused by setOffset.
1841             // since backward contraction/expansion may have extra ces if we
1842             // are in the normalization buffer, hasAccentsBeforeMatch would
1843             // have taken care of it.
1844             // E.g. the character \u01FA will have an expansion of 3, but if
1845             // we are only looking for acute and ring \u030A and \u0301, we'll
1846             // have to skip the first ce in the expansion buffer.
1847             ucol_next(coleiter, status);
1848             if (U_FAILURE(*status)) {
1849                 return FALSE;
1850             }
1851             if (ucol_getOffset(coleiter) != temp) {
1852                 *start = temp;
1853                 temp  = ucol_getOffset(coleiter);
1854             }
1855             expansion --;
1856         }
1857 
1858         int32_t  *patternce       = strsrch->pattern.CE;
1859         int32_t   patterncelength = strsrch->pattern.CELength;
1860         int32_t   count           = 0;
1861         int32_t   textlength      = strsrch->search->textLength;
1862         while (count < patterncelength) {
1863             int32_t ce = getCE(strsrch, ucol_next(coleiter, status));
1864             // status checked below, note that if status is a failure
1865             // ucol_next returns UCOL_NULLORDER
1866             if (ce == UCOL_IGNORABLE) {
1867                 continue;
1868             }
1869             if (expandflag && count == 0 && ucol_getOffset(coleiter) != temp) {
1870                 *start = temp;
1871                 temp   = ucol_getOffset(coleiter);
1872             }
1873 
1874             if (count == 0 && ce != patternce[0]) {
1875                 // accents may have extra starting ces, this occurs when a
1876                 // pure accent pattern is matched without rearrangement
1877                 // text \u0325\u0300 and looking for \u0300
1878                 int32_t expected = patternce[0];
1879                 if (getFCD(text, start, textlength) & LAST_BYTE_MASK_) {
1880                     ce = getCE(strsrch, ucol_next(coleiter, status));
1881                     while (U_SUCCESS(*status) && ce != expected &&
1882                            ce != UCOL_NULLORDER &&
1883                            ucol_getOffset(coleiter) <= *end) {
1884                         ce = getCE(strsrch, ucol_next(coleiter, status));
1885                     }
1886                 }
1887             }
1888             if (U_FAILURE(*status) || ce != patternce[count]) {
1889                 (*end) ++;
1890                 *end = getNextUStringSearchBaseOffset(strsrch, *end);
1891                 return FALSE;
1892             }
1893             count ++;
1894         }
1895     }
1896     return TRUE;
1897 }
1898 
1899 /**
1900 * Checks and sets the match information if found.
1901 * Checks
1902 * <ul>
1903 * <li> the potential match does not repeat the previous match
1904 * <li> boundaries are correct
1905 * <li> potential match does not end in the middle of a contraction
1906 * <li> identical matches
1907 * <\ul>
1908 * Otherwise the offset will be shifted to the next character.
1909 * Internal method, status assumed to be success, caller has to check the
1910 * status before calling this method.
1911 * @param strsrch string search data
1912 * @param textoffset offset in the collation element text. the returned value
1913 *        will be the truncated end offset of the match or the new start
1914 *        search offset.
1915 * @param status output error status if any
1916 * @return TRUE if the match is valid, FALSE otherwise
1917 */
1918 static
checkNextCanonicalMatch(UStringSearch * strsrch,int32_t * textoffset,UErrorCode * status)1919 inline UBool checkNextCanonicalMatch(UStringSearch *strsrch,
1920                                      int32_t   *textoffset,
1921                                      UErrorCode    *status)
1922 {
1923     // to ensure that the start and ends are not composite characters
1924     UCollationElements *coleiter = strsrch->textIter;
1925     // if we have a canonical accent match
1926     if ((strsrch->pattern.hasSuffixAccents &&
1927         strsrch->canonicalSuffixAccents[0]) ||
1928         (strsrch->pattern.hasPrefixAccents &&
1929         strsrch->canonicalPrefixAccents[0])) {
1930         strsrch->search->matchedIndex  = getPreviousUStringSearchBaseOffset(
1931                                                     strsrch,
1932                                                     ucol_getOffset(coleiter));
1933         strsrch->search->matchedLength = *textoffset -
1934                                                 strsrch->search->matchedIndex;
1935         return TRUE;
1936     }
1937 
1938     int32_t start = getColElemIterOffset(coleiter, FALSE);
1939     if (!checkNextCanonicalContractionMatch(strsrch, &start, textoffset,
1940                                             status) || U_FAILURE(*status)) {
1941         return FALSE;
1942     }
1943 
1944     start = getPreviousUStringSearchBaseOffset(strsrch, start);
1945     // this totally matches, however we need to check if it is repeating
1946     if (checkRepeatedMatch(strsrch, start, *textoffset) ||
1947         !isBreakUnit(strsrch, start, *textoffset) ||
1948         !checkIdentical(strsrch, start, *textoffset)) {
1949         (*textoffset) ++;
1950         *textoffset = getNextBaseOffset(strsrch->search->text, *textoffset,
1951                                         strsrch->search->textLength);
1952         return FALSE;
1953     }
1954 
1955     strsrch->search->matchedIndex  = start;
1956     strsrch->search->matchedLength = *textoffset - start;
1957     return TRUE;
1958 }
1959 
1960 /**
1961 * Shifting the collation element iterator position forward to prepare for
1962 * a preceding match. If the first character is a unsafe character, we'll only
1963 * shift by 1 to capture contractions, normalization etc.
1964 * Internal method, status assumed to be success, caller has to check status
1965 * before calling this method.
1966 * @param text strsrch string search data
1967 * @param textoffset start text position to do search
1968 * @param ce the text ce which failed the match.
1969 * @param patternceindex index of the ce within the pattern ce buffer which
1970 *        failed the match
1971 * @return final offset
1972 */
1973 static
reverseShift(UStringSearch * strsrch,int32_t textoffset,int32_t ce,int32_t patternceindex)1974 inline int32_t reverseShift(UStringSearch *strsrch,
1975                                 int32_t    textoffset,
1976                                 int32_t       ce,
1977                                 int32_t        patternceindex)
1978 {
1979     if (strsrch->search->isOverlap) {
1980         if (textoffset != strsrch->search->textLength) {
1981             textoffset --;
1982         }
1983         else {
1984             textoffset -= strsrch->pattern.defaultShiftSize;
1985         }
1986     }
1987     else {
1988         if (ce != UCOL_NULLORDER) {
1989             int32_t shift = strsrch->pattern.backShift[hash(ce)];
1990 
1991             // this is to adjust for characters in the middle of the substring
1992             // for matching that failed.
1993             int32_t adjust = patternceindex;
1994             if (adjust > 1 && shift > adjust) {
1995                 shift -= adjust - 1;
1996             }
1997             textoffset -= shift;
1998         }
1999         else {
2000             textoffset -= strsrch->pattern.defaultShiftSize;
2001         }
2002     }
2003     textoffset = getPreviousUStringSearchBaseOffset(strsrch, textoffset);
2004     return textoffset;
2005 }
2006 
2007 /**
2008 * Checks match for contraction.
2009 * If the match starts with a partial contraction we fail.
2010 * Internal method, status assumed to be success, caller has to check status
2011 * before calling this method.
2012 * @param strsrch string search data
2013 * @param start offset of potential match, to be modified if necessary
2014 * @param end offset of potential match, to be modified if necessary
2015 * @param status output error status if any
2016 * @return TRUE if match passes the contraction test, FALSE otherwise
2017 */
2018 static
checkPreviousExactContractionMatch(UStringSearch * strsrch,int32_t * start,int32_t * end,UErrorCode * status)2019 UBool checkPreviousExactContractionMatch(UStringSearch *strsrch,
2020                                      int32_t   *start,
2021                                      int32_t   *end, UErrorCode  *status)
2022 {
2023           UCollationElements *coleiter   = strsrch->textIter;
2024           int32_t             textlength = strsrch->search->textLength;
2025           int32_t             temp       = *end;
2026     const UCollator          *collator   = strsrch->collator;
2027     const UChar              *text       = strsrch->search->text;
2028     // This part checks if either if the start of the match contains potential
2029     // contraction. If so we'll have to iterate through them
2030     // Since we used ucol_next while previously looking for the potential
2031     // match, this guarantees that our end will not be a partial contraction,
2032     // or a partial supplementary character.
2033     if (*start < textlength && ucol_unsafeCP(text[*start], collator)) {
2034         int32_t expansion  = getExpansionSuffix(coleiter);
2035         UBool   expandflag = expansion > 0;
2036         setColEIterOffset(coleiter, *end);
2037         while (U_SUCCESS(*status) && expansion > 0) {
2038             // getting rid of the redundant ce
2039             // since forward contraction/expansion may have extra ces
2040             // if we are in the normalization buffer, hasAccentsBeforeMatch
2041             // would have taken care of it.
2042             // E.g. the character \u01FA will have an expansion of 3, but if
2043             // we are only looking for A ring A\u030A, we'll have to skip the
2044             // last ce in the expansion buffer
2045             ucol_previous(coleiter, status);
2046             if (U_FAILURE(*status)) {
2047                 return FALSE;
2048             }
2049             if (ucol_getOffset(coleiter) != temp) {
2050                 *end = temp;
2051                 temp  = ucol_getOffset(coleiter);
2052             }
2053             expansion --;
2054         }
2055 
2056         int32_t  *patternce       = strsrch->pattern.CE;
2057         int32_t   patterncelength = strsrch->pattern.CELength;
2058         int32_t   count           = patterncelength;
2059         while (count > 0) {
2060             int32_t ce = getCE(strsrch, ucol_previous(coleiter, status));
2061             // status checked below, note that if status is a failure
2062             // ucol_previous returns UCOL_NULLORDER
2063             if (ce == UCOL_IGNORABLE) {
2064                 continue;
2065             }
2066             if (expandflag && count == 0 &&
2067                 getColElemIterOffset(coleiter, FALSE) != temp) {
2068                 *end = temp;
2069                 temp  = ucol_getOffset(coleiter);
2070             }
2071             if (U_FAILURE(*status) || ce != patternce[count - 1]) {
2072                 (*start) --;
2073                 *start = getPreviousBaseOffset(text, *start);
2074                 return FALSE;
2075             }
2076             count --;
2077         }
2078     }
2079     return TRUE;
2080 }
2081 
2082 /**
2083 * Checks and sets the match information if found.
2084 * Checks
2085 * <ul>
2086 * <li> the current match does not repeat the last match
2087 * <li> boundaries are correct
2088 * <li> exact matches has no extra accents
2089 * <li> identical matches
2090 * <\ul>
2091 * Otherwise the offset will be shifted to the preceding character.
2092 * Internal method, status assumed to be success, caller has to check status
2093 * before calling this method.
2094 * @param strsrch string search data
2095 * @param collator
2096 * @param coleiter collation element iterator
2097 * @param text string
2098 * @param textoffset offset in the collation element text. the returned value
2099 *        will be the truncated start offset of the match or the new start
2100 *        search offset.
2101 * @param status output error status if any
2102 * @return TRUE if the match is valid, FALSE otherwise
2103 */
2104 static
checkPreviousExactMatch(UStringSearch * strsrch,int32_t * textoffset,UErrorCode * status)2105 inline UBool checkPreviousExactMatch(UStringSearch *strsrch,
2106                                      int32_t   *textoffset,
2107                                      UErrorCode    *status)
2108 {
2109     // to ensure that the start and ends are not composite characters
2110     int32_t end = ucol_getOffset(strsrch->textIter);
2111     if (!checkPreviousExactContractionMatch(strsrch, textoffset, &end, status)
2112         || U_FAILURE(*status)) {
2113             return FALSE;
2114     }
2115 
2116     // this totally matches, however we need to check if it is repeating
2117     // the old match
2118     if (checkRepeatedMatch(strsrch, *textoffset, end) ||
2119         !isBreakUnit(strsrch, *textoffset, end) ||
2120         hasAccentsBeforeMatch(strsrch, *textoffset, end) ||
2121         !checkIdentical(strsrch, *textoffset, end) ||
2122         hasAccentsAfterMatch(strsrch, *textoffset, end)) {
2123         (*textoffset) --;
2124         *textoffset = getPreviousBaseOffset(strsrch->search->text,
2125                                             *textoffset);
2126         return FALSE;
2127     }
2128 
2129     //Add breakiterator boundary check for primary strength search.
2130     if (!strsrch->search->breakIter && strsrch->strength == UCOL_PRIMARY) {
2131     	checkBreakBoundary(strsrch, textoffset, &end);
2132     }
2133 
2134     strsrch->search->matchedIndex = *textoffset;
2135     strsrch->search->matchedLength = end - *textoffset;
2136     return TRUE;
2137 }
2138 
2139 /**
2140 * Rearranges the end accents to try matching.
2141 * Suffix accents in the text will be grouped according to their combining
2142 * class and the groups will be mixed and matched to try find the perfect
2143 * match with the pattern.
2144 * So for instance looking for "\u0301" in "\u030A\u0301\u0325"
2145 * step 1: split "\u030A\u0301" into 6 other type of potential accent substrings
2146 *         "\u030A", "\u0301", "\u0325", "\u030A\u0301", "\u030A\u0325",
2147 *         "\u0301\u0325".
2148 * step 2: check if any of the generated substrings matches the pattern.
2149 * Internal method, status assumed to be success, user has to check status
2150 * before calling this method.
2151 * @param strsrch string search match
2152 * @param start offset of the first base character
2153 * @param end start of the last accent set
2154 * @param status only error status if any
2155 * @return USEARCH_DONE if a match is not found, otherwise return the ending
2156 *         offset of the match. Note this start includes all following accents.
2157 */
2158 static
doPreviousCanonicalSuffixMatch(UStringSearch * strsrch,int32_t start,int32_t end,UErrorCode * status)2159 int32_t doPreviousCanonicalSuffixMatch(UStringSearch *strsrch,
2160                                            int32_t    start,
2161                                            int32_t    end,
2162                                            UErrorCode    *status)
2163 {
2164     const UChar       *text       = strsrch->search->text;
2165           int32_t  tempend    = end;
2166 
2167     UTF_BACK_1(text, 0, tempend);
2168     if (!(getFCD(text, &tempend, strsrch->search->textLength) &
2169                                                            LAST_BYTE_MASK_)) {
2170         // die... failed at a base character
2171         return USEARCH_DONE;
2172     }
2173     end = getNextBaseOffset(text, end, strsrch->search->textLength);
2174 
2175     if (U_SUCCESS(*status)) {
2176         UChar       accents[INITIAL_ARRAY_SIZE_];
2177         int32_t offset = getPreviousBaseOffset(text, end);
2178         // normalizing the offensive string
2179         unorm_normalize(text + offset, end - offset, UNORM_NFD, 0, accents,
2180                         INITIAL_ARRAY_SIZE_, status);
2181 
2182         int32_t         accentsindex[INITIAL_ARRAY_SIZE_];
2183         int32_t         accentsize = getUnblockedAccentIndex(accents,
2184                                                          accentsindex);
2185         int32_t         count      = (2 << (accentsize - 1)) - 1;
2186         UChar               buffer[INITIAL_ARRAY_SIZE_];
2187         UCollationElements *coleiter = strsrch->utilIter;
2188         while (U_SUCCESS(*status) && count > 0) {
2189             UChar *rearrange = strsrch->canonicalSuffixAccents;
2190             // copy the base characters
2191             for (int k = 0; k < accentsindex[0]; k ++) {
2192                 *rearrange ++ = accents[k];
2193             }
2194             // forming all possible canonical rearrangement by dropping
2195             // sets of accents
2196             for (int i = 0; i <= accentsize - 1; i ++) {
2197                 int32_t mask = 1 << (accentsize - i - 1);
2198                 if (count & mask) {
2199                     for (int j = accentsindex[i]; j < accentsindex[i + 1]; j ++) {
2200                         *rearrange ++ = accents[j];
2201                     }
2202                 }
2203             }
2204             *rearrange = 0;
2205             int32_t  matchsize = INITIAL_ARRAY_SIZE_;
2206             UChar   *match     = addToUCharArray(buffer, &matchsize,
2207                                            strsrch->canonicalPrefixAccents,
2208                                            strsrch->search->text + start,
2209                                            offset - start,
2210                                            strsrch->canonicalSuffixAccents,
2211                                            status);
2212 
2213             // run the collator iterator through this match
2214             // if status is a failure ucol_setText does nothing
2215             ucol_setText(coleiter, match, matchsize, status);
2216             if (U_SUCCESS(*status)) {
2217                 if (checkCollationMatch(strsrch, coleiter)) {
2218                     if (match != buffer) {
2219                         uprv_free(match);
2220                     }
2221                     return end;
2222                 }
2223             }
2224             count --;
2225         }
2226     }
2227     return USEARCH_DONE;
2228 }
2229 
2230 /**
2231 * Take the rearranged start accents and tries matching. If match failed at
2232 * a seperate following set of accents (seperated from the rearranged on by
2233 * at least a base character) then we rearrange the preceding accents and
2234 * tries matching again.
2235 * We allow skipping of the ends of the accent set if the ces do not match.
2236 * However if the failure is found before the accent set, it fails.
2237 * Internal method, status assumed to be success, caller has to check status
2238 * before calling this method.
2239 * @param strsrch string search data
2240 * @param textoffset of the ends of the rearranged accent
2241 * @param status output error status if any
2242 * @return USEARCH_DONE if a match is not found, otherwise return the ending
2243 *         offset of the match. Note this start includes all following accents.
2244 */
2245 static
doPreviousCanonicalPrefixMatch(UStringSearch * strsrch,int32_t textoffset,UErrorCode * status)2246 int32_t doPreviousCanonicalPrefixMatch(UStringSearch *strsrch,
2247                                            int32_t    textoffset,
2248                                            UErrorCode    *status)
2249 {
2250     const UChar       *text       = strsrch->search->text;
2251     const UCollator   *collator   = strsrch->collator;
2252           int32_t      safelength = 0;
2253           UChar       *safetext;
2254           int32_t      safetextlength;
2255           UChar        safebuffer[INITIAL_ARRAY_SIZE_];
2256           int32_t  safeoffset = textoffset;
2257 
2258     if (textoffset &&
2259         ucol_unsafeCP(strsrch->canonicalPrefixAccents[
2260                                  u_strlen(strsrch->canonicalPrefixAccents) - 1
2261                                          ], collator)) {
2262         safeoffset     = getNextSafeOffset(collator, text, textoffset,
2263                                            strsrch->search->textLength);
2264         safelength     = safeoffset - textoffset;
2265         safetextlength = INITIAL_ARRAY_SIZE_;
2266         safetext       = addToUCharArray(safebuffer, &safetextlength,
2267                                          strsrch->canonicalPrefixAccents,
2268                                          text + textoffset, safelength,
2269                                          NULL, status);
2270     }
2271     else {
2272         safetextlength = u_strlen(strsrch->canonicalPrefixAccents);
2273         safetext       = strsrch->canonicalPrefixAccents;
2274     }
2275 
2276     UCollationElements *coleiter = strsrch->utilIter;
2277      // if status is a failure, ucol_setText does nothing
2278     ucol_setText(coleiter, safetext, safetextlength, status);
2279     // status checked in loop below
2280 
2281     int32_t  *ce           = strsrch->pattern.CE;
2282     int32_t   celength     = strsrch->pattern.CELength;
2283     int       ceindex      = 0;
2284     UBool     isSafe       = TRUE; // safe zone indication flag for position
2285     int32_t   prefixlength = u_strlen(strsrch->canonicalPrefixAccents);
2286 
2287     while (ceindex < celength) {
2288         int32_t textce = ucol_next(coleiter, status);
2289         if (U_FAILURE(*status)) {
2290             if (isSafe) {
2291                 cleanUpSafeText(strsrch, safetext, safebuffer);
2292             }
2293             return USEARCH_DONE;
2294         }
2295         if (textce == UCOL_NULLORDER) {
2296             // check if we have passed the safe buffer
2297             if (coleiter == strsrch->textIter) {
2298                 cleanUpSafeText(strsrch, safetext, safebuffer);
2299                 return USEARCH_DONE;
2300             }
2301             cleanUpSafeText(strsrch, safetext, safebuffer);
2302             safetext = safebuffer;
2303             coleiter = strsrch->textIter;
2304             setColEIterOffset(coleiter, safeoffset);
2305             // status checked at the start of the loop
2306             isSafe = FALSE;
2307             continue;
2308         }
2309         textce = getCE(strsrch, textce);
2310         if (textce != UCOL_IGNORABLE && textce != ce[ceindex]) {
2311             // do the beginning stuff
2312             int32_t failedoffset = ucol_getOffset(coleiter);
2313             if (isSafe && failedoffset <= prefixlength) {
2314                 // alas... no hope. failed at rearranged accent set
2315                 cleanUpSafeText(strsrch, safetext, safebuffer);
2316                 return USEARCH_DONE;
2317             }
2318             else {
2319                 if (isSafe) {
2320                     failedoffset = safeoffset - failedoffset;
2321                     cleanUpSafeText(strsrch, safetext, safebuffer);
2322                 }
2323 
2324                 // try rearranging the end accents
2325                 int32_t result = doPreviousCanonicalSuffixMatch(strsrch,
2326                                         textoffset, failedoffset, status);
2327                 if (result != USEARCH_DONE) {
2328                     // if status is a failure, ucol_setOffset does nothing
2329                     setColEIterOffset(strsrch->textIter, result);
2330                 }
2331                 if (U_FAILURE(*status)) {
2332                     return USEARCH_DONE;
2333                 }
2334                 return result;
2335             }
2336         }
2337         if (textce == ce[ceindex]) {
2338             ceindex ++;
2339         }
2340     }
2341     // set offset here
2342     if (isSafe) {
2343         int32_t result      = ucol_getOffset(coleiter);
2344         // sets the text iterator here with the correct expansion and offset
2345         int32_t     leftoverces = getExpansionSuffix(coleiter);
2346         cleanUpSafeText(strsrch, safetext, safebuffer);
2347         if (result <= prefixlength) {
2348             result = textoffset;
2349         }
2350         else {
2351             result = textoffset + (safeoffset - result);
2352         }
2353         setColEIterOffset(strsrch->textIter, result);
2354         setExpansionSuffix(strsrch->textIter, leftoverces);
2355         return result;
2356     }
2357 
2358     return ucol_getOffset(coleiter);
2359 }
2360 
2361 /**
2362 * Trying out the substring and sees if it can be a canonical match.
2363 * This will try normalizing the starting accents and arranging them into
2364 * canonical equivalents and check their corresponding ces with the pattern ce.
2365 * Prefix accents in the text will be grouped according to their combining
2366 * class and the groups will be mixed and matched to try find the perfect
2367 * match with the pattern.
2368 * So for instance looking for "\u0301" in "\u030A\u0301\u0325"
2369 * step 1: split "\u030A\u0301" into 6 other type of potential accent substrings
2370 *         "\u030A", "\u0301", "\u0325", "\u030A\u0301", "\u030A\u0325",
2371 *         "\u0301\u0325".
2372 * step 2: check if any of the generated substrings matches the pattern.
2373 * Internal method, status assumed to be success, caller has to check status
2374 * before calling this method.
2375 * @param strsrch string search data
2376 * @param textoffset start offset in the collation element text that starts
2377 *                   with the accents to be rearranged
2378 * @param status output error status if any
2379 * @return TRUE if the match is valid, FALSE otherwise
2380 */
2381 static
doPreviousCanonicalMatch(UStringSearch * strsrch,int32_t textoffset,UErrorCode * status)2382 UBool doPreviousCanonicalMatch(UStringSearch *strsrch,
2383                                int32_t    textoffset,
2384                                UErrorCode    *status)
2385 {
2386     const UChar       *text       = strsrch->search->text;
2387           int32_t  temp       = textoffset;
2388           int32_t      textlength = strsrch->search->textLength;
2389     if ((getFCD(text, &temp, textlength) >> SECOND_LAST_BYTE_SHIFT_) == 0) {
2390         UCollationElements *coleiter = strsrch->textIter;
2391         int32_t         offset   = ucol_getOffset(coleiter);
2392         if (strsrch->pattern.hasSuffixAccents) {
2393             offset = doPreviousCanonicalSuffixMatch(strsrch, textoffset,
2394                                                     offset, status);
2395             if (U_SUCCESS(*status) && offset != USEARCH_DONE) {
2396                 setColEIterOffset(coleiter, offset);
2397                 return TRUE;
2398             }
2399         }
2400         return FALSE;
2401     }
2402 
2403     if (!strsrch->pattern.hasPrefixAccents) {
2404         return FALSE;
2405     }
2406 
2407     UChar       accents[INITIAL_ARRAY_SIZE_];
2408     // offset to the last base character in substring to search
2409     int32_t baseoffset = getNextBaseOffset(text, textoffset, textlength);
2410     // normalizing the offensive string
2411     unorm_normalize(text + textoffset, baseoffset - textoffset, UNORM_NFD,
2412                                0, accents, INITIAL_ARRAY_SIZE_, status);
2413     // status checked in loop
2414 
2415     int32_t accentsindex[INITIAL_ARRAY_SIZE_];
2416     int32_t size = getUnblockedAccentIndex(accents, accentsindex);
2417 
2418     // 2 power n - 1 plus the full set of accents
2419     int32_t  count = (2 << (size - 1)) - 1;
2420     while (U_SUCCESS(*status) && count > 0) {
2421         UChar *rearrange = strsrch->canonicalPrefixAccents;
2422         // copy the base characters
2423         for (int k = 0; k < accentsindex[0]; k ++) {
2424             *rearrange ++ = accents[k];
2425         }
2426         // forming all possible canonical rearrangement by dropping
2427         // sets of accents
2428         for (int i = 0; i <= size - 1; i ++) {
2429             int32_t mask = 1 << (size - i - 1);
2430             if (count & mask) {
2431                 for (int j = accentsindex[i]; j < accentsindex[i + 1]; j ++) {
2432                     *rearrange ++ = accents[j];
2433                 }
2434             }
2435         }
2436         *rearrange = 0;
2437         int32_t offset = doPreviousCanonicalPrefixMatch(strsrch,
2438                                                           baseoffset, status);
2439         if (offset != USEARCH_DONE) {
2440             return TRUE; // match found
2441         }
2442         count --;
2443     }
2444     return FALSE;
2445 }
2446 
2447 /**
2448 * Checks match for contraction.
2449 * If the match starts with a partial contraction we fail.
2450 * Internal method, status assumed to be success, caller has to check status
2451 * before calling this method.
2452 * @param strsrch string search data
2453 * @param start offset of potential match, to be modified if necessary
2454 * @param end offset of potential match, to be modified if necessary
2455 * @param status only error status if any
2456 * @return TRUE if match passes the contraction test, FALSE otherwise
2457 */
2458 static
checkPreviousCanonicalContractionMatch(UStringSearch * strsrch,int32_t * start,int32_t * end,UErrorCode * status)2459 UBool checkPreviousCanonicalContractionMatch(UStringSearch *strsrch,
2460                                      int32_t   *start,
2461                                      int32_t   *end, UErrorCode  *status)
2462 {
2463           UCollationElements *coleiter   = strsrch->textIter;
2464           int32_t             textlength = strsrch->search->textLength;
2465           int32_t         temp       = *end;
2466     const UCollator          *collator   = strsrch->collator;
2467     const UChar              *text       = strsrch->search->text;
2468     // This part checks if either if the start of the match contains potential
2469     // contraction. If so we'll have to iterate through them
2470     // Since we used ucol_next while previously looking for the potential
2471     // match, this guarantees that our end will not be a partial contraction,
2472     // or a partial supplementary character.
2473     if (*start < textlength && ucol_unsafeCP(text[*start], collator)) {
2474         int32_t expansion  = getExpansionSuffix(coleiter);
2475         UBool   expandflag = expansion > 0;
2476         setColEIterOffset(coleiter, *end);
2477         while (expansion > 0) {
2478             // getting rid of the redundant ce
2479             // since forward contraction/expansion may have extra ces
2480             // if we are in the normalization buffer, hasAccentsBeforeMatch
2481             // would have taken care of it.
2482             // E.g. the character \u01FA will have an expansion of 3, but if
2483             // we are only looking for A ring A\u030A, we'll have to skip the
2484             // last ce in the expansion buffer
2485             ucol_previous(coleiter, status);
2486             if (U_FAILURE(*status)) {
2487                 return FALSE;
2488             }
2489             if (ucol_getOffset(coleiter) != temp) {
2490                 *end = temp;
2491                 temp  = ucol_getOffset(coleiter);
2492             }
2493             expansion --;
2494         }
2495 
2496         int32_t  *patternce       = strsrch->pattern.CE;
2497         int32_t   patterncelength = strsrch->pattern.CELength;
2498         int32_t   count           = patterncelength;
2499         while (count > 0) {
2500             int32_t ce = getCE(strsrch, ucol_previous(coleiter, status));
2501             // status checked below, note that if status is a failure
2502             // ucol_previous returns UCOL_NULLORDER
2503             if (ce == UCOL_IGNORABLE) {
2504                 continue;
2505             }
2506             if (expandflag && count == 0 &&
2507                 getColElemIterOffset(coleiter, FALSE) != temp) {
2508                 *end = temp;
2509                 temp  = ucol_getOffset(coleiter);
2510             }
2511             if (count == patterncelength &&
2512                 ce != patternce[patterncelength - 1]) {
2513                 // accents may have extra starting ces, this occurs when a
2514                 // pure accent pattern is matched without rearrangement
2515                 int32_t    expected = patternce[patterncelength - 1];
2516                 UTF_BACK_1(text, 0, *end);
2517                 if (getFCD(text, end, textlength) & LAST_BYTE_MASK_) {
2518                     ce = getCE(strsrch, ucol_previous(coleiter, status));
2519                     while (U_SUCCESS(*status) && ce != expected &&
2520                            ce != UCOL_NULLORDER &&
2521                            ucol_getOffset(coleiter) <= *start) {
2522                         ce = getCE(strsrch, ucol_previous(coleiter, status));
2523                     }
2524                 }
2525             }
2526             if (U_FAILURE(*status) || ce != patternce[count - 1]) {
2527                 (*start) --;
2528                 *start = getPreviousBaseOffset(text, *start);
2529                 return FALSE;
2530             }
2531             count --;
2532         }
2533     }
2534     return TRUE;
2535 }
2536 
2537 /**
2538 * Checks and sets the match information if found.
2539 * Checks
2540 * <ul>
2541 * <li> the potential match does not repeat the previous match
2542 * <li> boundaries are correct
2543 * <li> potential match does not end in the middle of a contraction
2544 * <li> identical matches
2545 * <\ul>
2546 * Otherwise the offset will be shifted to the next character.
2547 * Internal method, status assumed to be success, caller has to check status
2548 * before calling this method.
2549 * @param strsrch string search data
2550 * @param textoffset offset in the collation element text. the returned value
2551 *        will be the truncated start offset of the match or the new start
2552 *        search offset.
2553 * @param status only error status if any
2554 * @return TRUE if the match is valid, FALSE otherwise
2555 */
2556 static
checkPreviousCanonicalMatch(UStringSearch * strsrch,int32_t * textoffset,UErrorCode * status)2557 inline UBool checkPreviousCanonicalMatch(UStringSearch *strsrch,
2558                                          int32_t   *textoffset,
2559                                          UErrorCode    *status)
2560 {
2561     // to ensure that the start and ends are not composite characters
2562     UCollationElements *coleiter = strsrch->textIter;
2563     // if we have a canonical accent match
2564     if ((strsrch->pattern.hasSuffixAccents &&
2565         strsrch->canonicalSuffixAccents[0]) ||
2566         (strsrch->pattern.hasPrefixAccents &&
2567         strsrch->canonicalPrefixAccents[0])) {
2568         strsrch->search->matchedIndex  = *textoffset;
2569         strsrch->search->matchedLength =
2570             getNextUStringSearchBaseOffset(strsrch,
2571                                       getColElemIterOffset(coleiter, FALSE))
2572             - *textoffset;
2573         return TRUE;
2574     }
2575 
2576     int32_t end = ucol_getOffset(coleiter);
2577     if (!checkPreviousCanonicalContractionMatch(strsrch, textoffset, &end,
2578                                                 status) ||
2579          U_FAILURE(*status)) {
2580         return FALSE;
2581     }
2582 
2583     end = getNextUStringSearchBaseOffset(strsrch, end);
2584     // this totally matches, however we need to check if it is repeating
2585     if (checkRepeatedMatch(strsrch, *textoffset, end) ||
2586         !isBreakUnit(strsrch, *textoffset, end) ||
2587         !checkIdentical(strsrch, *textoffset, end)) {
2588         (*textoffset) --;
2589         *textoffset = getPreviousBaseOffset(strsrch->search->text,
2590                                             *textoffset);
2591         return FALSE;
2592     }
2593 
2594     strsrch->search->matchedIndex  = *textoffset;
2595     strsrch->search->matchedLength = end - *textoffset;
2596     return TRUE;
2597 }
2598 #endif // #if BOYER_MOORE
2599 
2600 // constructors and destructor -------------------------------------------
2601 
usearch_open(const UChar * pattern,int32_t patternlength,const UChar * text,int32_t textlength,const char * locale,UBreakIterator * breakiter,UErrorCode * status)2602 U_CAPI UStringSearch * U_EXPORT2 usearch_open(const UChar *pattern,
2603                                           int32_t         patternlength,
2604                                     const UChar          *text,
2605                                           int32_t         textlength,
2606                                     const char           *locale,
2607                                           UBreakIterator *breakiter,
2608                                           UErrorCode     *status)
2609 {
2610     if (U_FAILURE(*status)) {
2611         return NULL;
2612     }
2613 #if UCONFIG_NO_BREAK_ITERATION
2614     if (breakiter != NULL) {
2615         *status = U_UNSUPPORTED_ERROR;
2616         return NULL;
2617     }
2618 #endif
2619     if (locale) {
2620         // ucol_open internally checks for status
2621         UCollator     *collator = ucol_open(locale, status);
2622         // pattern, text checks are done in usearch_openFromCollator
2623         UStringSearch *result   = usearch_openFromCollator(pattern,
2624                                               patternlength, text, textlength,
2625                                               collator, breakiter, status);
2626 
2627         if (result == NULL || U_FAILURE(*status)) {
2628             if (collator) {
2629                 ucol_close(collator);
2630             }
2631             return NULL;
2632         }
2633         else {
2634             result->ownCollator = TRUE;
2635         }
2636         return result;
2637     }
2638     *status = U_ILLEGAL_ARGUMENT_ERROR;
2639     return NULL;
2640 }
2641 
usearch_openFromCollator(const UChar * pattern,int32_t patternlength,const UChar * text,int32_t textlength,const UCollator * collator,UBreakIterator * breakiter,UErrorCode * status)2642 U_CAPI UStringSearch * U_EXPORT2 usearch_openFromCollator(
2643                                   const UChar          *pattern,
2644                                         int32_t         patternlength,
2645                                   const UChar          *text,
2646                                         int32_t         textlength,
2647                                   const UCollator      *collator,
2648                                         UBreakIterator *breakiter,
2649                                         UErrorCode     *status)
2650 {
2651     if (U_FAILURE(*status)) {
2652         return NULL;
2653     }
2654 #if UCONFIG_NO_BREAK_ITERATION
2655     if (breakiter != NULL) {
2656         *status = U_UNSUPPORTED_ERROR;
2657         return NULL;
2658     }
2659 #endif
2660     if (pattern == NULL || text == NULL || collator == NULL) {
2661         *status = U_ILLEGAL_ARGUMENT_ERROR;
2662         return NULL;
2663     }
2664 
2665     // string search does not really work when numeric collation is turned on
2666     if(ucol_getAttribute(collator, UCOL_NUMERIC_COLLATION, status) == UCOL_ON) {
2667         *status = U_UNSUPPORTED_ERROR;
2668         return NULL;
2669     }
2670 
2671     if (U_SUCCESS(*status)) {
2672         initializeFCD(status);
2673         if (U_FAILURE(*status)) {
2674             return NULL;
2675         }
2676 
2677         UStringSearch *result;
2678         if (textlength == -1) {
2679             textlength = u_strlen(text);
2680         }
2681         if (patternlength == -1) {
2682             patternlength = u_strlen(pattern);
2683         }
2684         if (textlength <= 0 || patternlength <= 0) {
2685             *status = U_ILLEGAL_ARGUMENT_ERROR;
2686             return NULL;
2687         }
2688 
2689         result = (UStringSearch *)uprv_malloc(sizeof(UStringSearch));
2690         if (result == NULL) {
2691             *status = U_MEMORY_ALLOCATION_ERROR;
2692             return NULL;
2693         }
2694 
2695         result->collator    = collator;
2696         result->strength    = ucol_getStrength(collator);
2697         result->ceMask      = getMask(result->strength);
2698         result->toShift     =
2699              ucol_getAttribute(collator, UCOL_ALTERNATE_HANDLING, status) ==
2700                                                             UCOL_SHIFTED;
2701         result->variableTop = ucol_getVariableTop(collator, status);
2702 
2703         result->nfd         = Normalizer2Factory::getNFDInstance(*status);
2704 
2705         if (U_FAILURE(*status)) {
2706             uprv_free(result);
2707             return NULL;
2708         }
2709 
2710         result->search             = (USearch *)uprv_malloc(sizeof(USearch));
2711         if (result->search == NULL) {
2712             *status = U_MEMORY_ALLOCATION_ERROR;
2713             uprv_free(result);
2714             return NULL;
2715         }
2716 
2717         result->search->text       = text;
2718         result->search->textLength = textlength;
2719 
2720         result->pattern.text       = pattern;
2721         result->pattern.textLength = patternlength;
2722         result->pattern.CE         = NULL;
2723         result->pattern.PCE        = NULL;
2724 
2725         result->search->breakIter  = breakiter;
2726 #if !UCONFIG_NO_BREAK_ITERATION
2727         result->search->internalBreakIter = ubrk_open(UBRK_CHARACTER, ucol_getLocaleByType(result->collator, ULOC_VALID_LOCALE, status), text, textlength, status);
2728         if (breakiter) {
2729         	ubrk_setText(breakiter, text, textlength, status);
2730         }
2731 #endif
2732 
2733         result->ownCollator           = FALSE;
2734         result->search->matchedLength = 0;
2735         result->search->matchedIndex  = USEARCH_DONE;
2736         result->utilIter              = NULL;
2737         result->textIter              = ucol_openElements(collator, text,
2738                                                           textlength, status);
2739         if (U_FAILURE(*status)) {
2740             usearch_close(result);
2741             return NULL;
2742         }
2743 
2744         result->search->isOverlap          = FALSE;
2745         result->search->isCanonicalMatch   = FALSE;
2746         result->search->elementComparisonType = 0;
2747         result->search->isForwardSearching = TRUE;
2748         result->search->reset              = TRUE;
2749 
2750         initialize(result, status);
2751 
2752         if (U_FAILURE(*status)) {
2753             usearch_close(result);
2754             return NULL;
2755         }
2756 
2757         return result;
2758     }
2759     return NULL;
2760 }
2761 
usearch_close(UStringSearch * strsrch)2762 U_CAPI void U_EXPORT2 usearch_close(UStringSearch *strsrch)
2763 {
2764     if (strsrch) {
2765         if (strsrch->pattern.CE != strsrch->pattern.CEBuffer &&
2766             strsrch->pattern.CE) {
2767             uprv_free(strsrch->pattern.CE);
2768         }
2769 
2770         if (strsrch->pattern.PCE != NULL &&
2771             strsrch->pattern.PCE != strsrch->pattern.PCEBuffer) {
2772             uprv_free(strsrch->pattern.PCE);
2773         }
2774 
2775         ucol_closeElements(strsrch->textIter);
2776         ucol_closeElements(strsrch->utilIter);
2777 
2778         if (strsrch->ownCollator && strsrch->collator) {
2779             ucol_close((UCollator *)strsrch->collator);
2780         }
2781 
2782 #if !UCONFIG_NO_BREAK_ITERATION
2783         if (strsrch->search->internalBreakIter) {
2784         	ubrk_close(strsrch->search->internalBreakIter);
2785         }
2786 #endif
2787 
2788         uprv_free(strsrch->search);
2789         uprv_free(strsrch);
2790     }
2791 }
2792 
2793 // set and get methods --------------------------------------------------
2794 
usearch_setOffset(UStringSearch * strsrch,int32_t position,UErrorCode * status)2795 U_CAPI void U_EXPORT2 usearch_setOffset(UStringSearch *strsrch,
2796                                         int32_t    position,
2797                                         UErrorCode    *status)
2798 {
2799     if (U_SUCCESS(*status) && strsrch) {
2800         if (isOutOfBounds(strsrch->search->textLength, position)) {
2801             *status = U_INDEX_OUTOFBOUNDS_ERROR;
2802         }
2803         else {
2804             setColEIterOffset(strsrch->textIter, position);
2805         }
2806         strsrch->search->matchedIndex  = USEARCH_DONE;
2807         strsrch->search->matchedLength = 0;
2808         strsrch->search->reset         = FALSE;
2809     }
2810 }
2811 
usearch_getOffset(const UStringSearch * strsrch)2812 U_CAPI int32_t U_EXPORT2 usearch_getOffset(const UStringSearch *strsrch)
2813 {
2814     if (strsrch) {
2815         int32_t result = ucol_getOffset(strsrch->textIter);
2816         if (isOutOfBounds(strsrch->search->textLength, result)) {
2817             return USEARCH_DONE;
2818         }
2819         return result;
2820     }
2821     return USEARCH_DONE;
2822 }
2823 
usearch_setAttribute(UStringSearch * strsrch,USearchAttribute attribute,USearchAttributeValue value,UErrorCode * status)2824 U_CAPI void U_EXPORT2 usearch_setAttribute(UStringSearch *strsrch,
2825                                  USearchAttribute attribute,
2826                                  USearchAttributeValue value,
2827                                  UErrorCode *status)
2828 {
2829     if (U_SUCCESS(*status) && strsrch) {
2830         switch (attribute)
2831         {
2832         case USEARCH_OVERLAP :
2833             strsrch->search->isOverlap = (value == USEARCH_ON ? TRUE : FALSE);
2834             break;
2835         case USEARCH_CANONICAL_MATCH :
2836             strsrch->search->isCanonicalMatch = (value == USEARCH_ON ? TRUE :
2837                                                                       FALSE);
2838             break;
2839         case USEARCH_ELEMENT_COMPARISON :
2840             if (value == USEARCH_PATTERN_BASE_WEIGHT_IS_WILDCARD || value == USEARCH_ANY_BASE_WEIGHT_IS_WILDCARD) {
2841                 strsrch->search->elementComparisonType = (int16_t)value;
2842             } else {
2843                 strsrch->search->elementComparisonType = 0;
2844             }
2845             break;
2846         case USEARCH_ATTRIBUTE_COUNT :
2847         default:
2848             *status = U_ILLEGAL_ARGUMENT_ERROR;
2849         }
2850     }
2851     if (value == USEARCH_ATTRIBUTE_VALUE_COUNT) {
2852         *status = U_ILLEGAL_ARGUMENT_ERROR;
2853     }
2854 }
2855 
usearch_getAttribute(const UStringSearch * strsrch,USearchAttribute attribute)2856 U_CAPI USearchAttributeValue U_EXPORT2 usearch_getAttribute(
2857                                                 const UStringSearch *strsrch,
2858                                                 USearchAttribute attribute)
2859 {
2860     if (strsrch) {
2861         switch (attribute) {
2862         case USEARCH_OVERLAP :
2863             return (strsrch->search->isOverlap == TRUE ? USEARCH_ON :
2864                                                         USEARCH_OFF);
2865         case USEARCH_CANONICAL_MATCH :
2866             return (strsrch->search->isCanonicalMatch == TRUE ? USEARCH_ON :
2867                                                                USEARCH_OFF);
2868         case USEARCH_ELEMENT_COMPARISON :
2869             {
2870                 int16_t value = strsrch->search->elementComparisonType;
2871                 if (value == USEARCH_PATTERN_BASE_WEIGHT_IS_WILDCARD || value == USEARCH_ANY_BASE_WEIGHT_IS_WILDCARD) {
2872                     return (USearchAttributeValue)value;
2873                 } else {
2874                     return USEARCH_STANDARD_ELEMENT_COMPARISON;
2875                 }
2876             }
2877         case USEARCH_ATTRIBUTE_COUNT :
2878             return USEARCH_DEFAULT;
2879         }
2880     }
2881     return USEARCH_DEFAULT;
2882 }
2883 
usearch_getMatchedStart(const UStringSearch * strsrch)2884 U_CAPI int32_t U_EXPORT2 usearch_getMatchedStart(
2885                                                 const UStringSearch *strsrch)
2886 {
2887     if (strsrch == NULL) {
2888         return USEARCH_DONE;
2889     }
2890     return strsrch->search->matchedIndex;
2891 }
2892 
2893 
usearch_getMatchedText(const UStringSearch * strsrch,UChar * result,int32_t resultCapacity,UErrorCode * status)2894 U_CAPI int32_t U_EXPORT2 usearch_getMatchedText(const UStringSearch *strsrch,
2895                                             UChar         *result,
2896                                             int32_t        resultCapacity,
2897                                             UErrorCode    *status)
2898 {
2899     if (U_FAILURE(*status)) {
2900         return USEARCH_DONE;
2901     }
2902     if (strsrch == NULL || resultCapacity < 0 || (resultCapacity > 0 &&
2903         result == NULL)) {
2904         *status = U_ILLEGAL_ARGUMENT_ERROR;
2905         return USEARCH_DONE;
2906     }
2907 
2908     int32_t     copylength = strsrch->search->matchedLength;
2909     int32_t copyindex  = strsrch->search->matchedIndex;
2910     if (copyindex == USEARCH_DONE) {
2911         u_terminateUChars(result, resultCapacity, 0, status);
2912         return USEARCH_DONE;
2913     }
2914 
2915     if (resultCapacity < copylength) {
2916         copylength = resultCapacity;
2917     }
2918     if (copylength > 0) {
2919         uprv_memcpy(result, strsrch->search->text + copyindex,
2920                     copylength * sizeof(UChar));
2921     }
2922     return u_terminateUChars(result, resultCapacity,
2923                              strsrch->search->matchedLength, status);
2924 }
2925 
usearch_getMatchedLength(const UStringSearch * strsrch)2926 U_CAPI int32_t U_EXPORT2 usearch_getMatchedLength(
2927                                               const UStringSearch *strsrch)
2928 {
2929     if (strsrch) {
2930         return strsrch->search->matchedLength;
2931     }
2932     return USEARCH_DONE;
2933 }
2934 
2935 #if !UCONFIG_NO_BREAK_ITERATION
2936 
usearch_setBreakIterator(UStringSearch * strsrch,UBreakIterator * breakiter,UErrorCode * status)2937 U_CAPI void U_EXPORT2 usearch_setBreakIterator(UStringSearch  *strsrch,
2938                                                UBreakIterator *breakiter,
2939                                                UErrorCode     *status)
2940 {
2941     if (U_SUCCESS(*status) && strsrch) {
2942     	strsrch->search->breakIter = breakiter;
2943         if (breakiter) {
2944             ubrk_setText(breakiter, strsrch->search->text,
2945                          strsrch->search->textLength, status);
2946         }
2947     }
2948 }
2949 
2950 U_CAPI const UBreakIterator* U_EXPORT2
usearch_getBreakIterator(const UStringSearch * strsrch)2951 usearch_getBreakIterator(const UStringSearch *strsrch)
2952 {
2953     if (strsrch) {
2954         return strsrch->search->breakIter;
2955     }
2956     return NULL;
2957 }
2958 
2959 #endif
2960 
usearch_setText(UStringSearch * strsrch,const UChar * text,int32_t textlength,UErrorCode * status)2961 U_CAPI void U_EXPORT2 usearch_setText(      UStringSearch *strsrch,
2962                                       const UChar         *text,
2963                                             int32_t        textlength,
2964                                             UErrorCode    *status)
2965 {
2966     if (U_SUCCESS(*status)) {
2967         if (strsrch == NULL || text == NULL || textlength < -1 ||
2968             textlength == 0) {
2969             *status = U_ILLEGAL_ARGUMENT_ERROR;
2970         }
2971         else {
2972             if (textlength == -1) {
2973                 textlength = u_strlen(text);
2974             }
2975             strsrch->search->text       = text;
2976             strsrch->search->textLength = textlength;
2977             ucol_setText(strsrch->textIter, text, textlength, status);
2978             strsrch->search->matchedIndex  = USEARCH_DONE;
2979             strsrch->search->matchedLength = 0;
2980             strsrch->search->reset         = TRUE;
2981 #if !UCONFIG_NO_BREAK_ITERATION
2982             if (strsrch->search->breakIter != NULL) {
2983                 ubrk_setText(strsrch->search->breakIter, text,
2984                              textlength, status);
2985             }
2986             ubrk_setText(strsrch->search->internalBreakIter, text, textlength, status);
2987 #endif
2988         }
2989     }
2990 }
2991 
usearch_getText(const UStringSearch * strsrch,int32_t * length)2992 U_CAPI const UChar * U_EXPORT2 usearch_getText(const UStringSearch *strsrch,
2993                                                      int32_t       *length)
2994 {
2995     if (strsrch) {
2996         *length = strsrch->search->textLength;
2997         return strsrch->search->text;
2998     }
2999     return NULL;
3000 }
3001 
usearch_setCollator(UStringSearch * strsrch,const UCollator * collator,UErrorCode * status)3002 U_CAPI void U_EXPORT2 usearch_setCollator(      UStringSearch *strsrch,
3003                                           const UCollator     *collator,
3004                                                 UErrorCode    *status)
3005 {
3006     if (U_SUCCESS(*status)) {
3007         if (collator == NULL) {
3008             *status = U_ILLEGAL_ARGUMENT_ERROR;
3009             return;
3010         }
3011 
3012         if (strsrch) {
3013             if (strsrch->ownCollator && (strsrch->collator != collator)) {
3014                 ucol_close((UCollator *)strsrch->collator);
3015                 strsrch->ownCollator = FALSE;
3016             }
3017             strsrch->collator    = collator;
3018             strsrch->strength    = ucol_getStrength(collator);
3019             strsrch->ceMask      = getMask(strsrch->strength);
3020 #if !UCONFIG_NO_BREAK_ITERATION
3021         	ubrk_close(strsrch->search->internalBreakIter);
3022         	strsrch->search->internalBreakIter = ubrk_open(UBRK_CHARACTER, ucol_getLocaleByType(collator, ULOC_VALID_LOCALE, status),
3023         											 strsrch->search->text, strsrch->search->textLength, status);
3024 #endif
3025             // if status is a failure, ucol_getAttribute returns UCOL_DEFAULT
3026             strsrch->toShift     =
3027                ucol_getAttribute(collator, UCOL_ALTERNATE_HANDLING, status) ==
3028                                                                 UCOL_SHIFTED;
3029             // if status is a failure, ucol_getVariableTop returns 0
3030             strsrch->variableTop = ucol_getVariableTop(collator, status);
3031             if (U_SUCCESS(*status)) {
3032                 initialize(strsrch, status);
3033                 if (U_SUCCESS(*status)) {
3034                     /* free offset buffer to avoid memory leak before initializing. */
3035                     ucol_freeOffsetBuffer(&(strsrch->textIter->iteratordata_));
3036                     uprv_init_collIterate(collator, strsrch->search->text,
3037                                           strsrch->search->textLength,
3038                                           &(strsrch->textIter->iteratordata_),
3039                                           status);
3040                     strsrch->utilIter->iteratordata_.coll = collator;
3041                 }
3042             }
3043         }
3044 
3045         // **** are these calls needed?
3046         // **** we call uprv_init_pce in initializePatternPCETable
3047         // **** and the CEBuffer constructor...
3048 #if 0
3049         uprv_init_pce(strsrch->textIter);
3050         uprv_init_pce(strsrch->utilIter);
3051 #endif
3052     }
3053 }
3054 
usearch_getCollator(const UStringSearch * strsrch)3055 U_CAPI UCollator * U_EXPORT2 usearch_getCollator(const UStringSearch *strsrch)
3056 {
3057     if (strsrch) {
3058         return (UCollator *)strsrch->collator;
3059     }
3060     return NULL;
3061 }
3062 
usearch_setPattern(UStringSearch * strsrch,const UChar * pattern,int32_t patternlength,UErrorCode * status)3063 U_CAPI void U_EXPORT2 usearch_setPattern(      UStringSearch *strsrch,
3064                                          const UChar         *pattern,
3065                                                int32_t        patternlength,
3066                                                UErrorCode    *status)
3067 {
3068     if (U_SUCCESS(*status)) {
3069         if (strsrch == NULL || pattern == NULL) {
3070             *status = U_ILLEGAL_ARGUMENT_ERROR;
3071         }
3072         else {
3073             if (patternlength == -1) {
3074                 patternlength = u_strlen(pattern);
3075             }
3076             if (patternlength == 0) {
3077                 *status = U_ILLEGAL_ARGUMENT_ERROR;
3078                 return;
3079             }
3080             strsrch->pattern.text       = pattern;
3081             strsrch->pattern.textLength = patternlength;
3082             initialize(strsrch, status);
3083         }
3084     }
3085 }
3086 
3087 U_CAPI const UChar* U_EXPORT2
usearch_getPattern(const UStringSearch * strsrch,int32_t * length)3088 usearch_getPattern(const UStringSearch *strsrch,
3089                    int32_t       *length)
3090 {
3091     if (strsrch) {
3092         *length = strsrch->pattern.textLength;
3093         return strsrch->pattern.text;
3094     }
3095     return NULL;
3096 }
3097 
3098 // miscellanous methods --------------------------------------------------
3099 
usearch_first(UStringSearch * strsrch,UErrorCode * status)3100 U_CAPI int32_t U_EXPORT2 usearch_first(UStringSearch *strsrch,
3101                                            UErrorCode    *status)
3102 {
3103     if (strsrch && U_SUCCESS(*status)) {
3104         strsrch->search->isForwardSearching = TRUE;
3105         usearch_setOffset(strsrch, 0, status);
3106         if (U_SUCCESS(*status)) {
3107             return usearch_next(strsrch, status);
3108         }
3109     }
3110     return USEARCH_DONE;
3111 }
3112 
usearch_following(UStringSearch * strsrch,int32_t position,UErrorCode * status)3113 U_CAPI int32_t U_EXPORT2 usearch_following(UStringSearch *strsrch,
3114                                                int32_t    position,
3115                                                UErrorCode    *status)
3116 {
3117     if (strsrch && U_SUCCESS(*status)) {
3118         strsrch->search->isForwardSearching = TRUE;
3119         // position checked in usearch_setOffset
3120         usearch_setOffset(strsrch, position, status);
3121         if (U_SUCCESS(*status)) {
3122             return usearch_next(strsrch, status);
3123         }
3124     }
3125     return USEARCH_DONE;
3126 }
3127 
usearch_last(UStringSearch * strsrch,UErrorCode * status)3128 U_CAPI int32_t U_EXPORT2 usearch_last(UStringSearch *strsrch,
3129                                           UErrorCode    *status)
3130 {
3131     if (strsrch && U_SUCCESS(*status)) {
3132         strsrch->search->isForwardSearching = FALSE;
3133         usearch_setOffset(strsrch, strsrch->search->textLength, status);
3134         if (U_SUCCESS(*status)) {
3135             return usearch_previous(strsrch, status);
3136         }
3137     }
3138     return USEARCH_DONE;
3139 }
3140 
usearch_preceding(UStringSearch * strsrch,int32_t position,UErrorCode * status)3141 U_CAPI int32_t U_EXPORT2 usearch_preceding(UStringSearch *strsrch,
3142                                                int32_t    position,
3143                                                UErrorCode    *status)
3144 {
3145     if (strsrch && U_SUCCESS(*status)) {
3146         strsrch->search->isForwardSearching = FALSE;
3147         // position checked in usearch_setOffset
3148         usearch_setOffset(strsrch, position, status);
3149         if (U_SUCCESS(*status)) {
3150             return usearch_previous(strsrch, status);
3151         }
3152     }
3153     return USEARCH_DONE;
3154 }
3155 
3156 /**
3157 * If a direction switch is required, we'll count the number of ces till the
3158 * beginning of the collation element iterator and iterate forwards that
3159 * number of times. This is so that we get to the correct point within the
3160 * string to continue the search in. Imagine when we are in the middle of the
3161 * normalization buffer when the change in direction is request. arrrgghh....
3162 * After searching the offset within the collation element iterator will be
3163 * shifted to the start of the match. If a match is not found, the offset would
3164 * have been set to the end of the text string in the collation element
3165 * iterator.
3166 * Okay, here's my take on normalization buffer. The only time when there can
3167 * be 2 matches within the same normalization is when the pattern is consists
3168 * of all accents. But since the offset returned is from the text string, we
3169 * should not confuse the caller by returning the second match within the
3170 * same normalization buffer. If we do, the 2 results will have the same match
3171 * offsets, and that'll be confusing. I'll return the next match that doesn't
3172 * fall within the same normalization buffer. Note this does not affect the
3173 * results of matches spanning the text and the normalization buffer.
3174 * The position to start searching is taken from the collation element
3175 * iterator. Callers of this API would have to set the offset in the collation
3176 * element iterator before using this method.
3177 */
usearch_next(UStringSearch * strsrch,UErrorCode * status)3178 U_CAPI int32_t U_EXPORT2 usearch_next(UStringSearch *strsrch,
3179                                           UErrorCode    *status)
3180 {
3181     if (U_SUCCESS(*status) && strsrch) {
3182         // note offset is either equivalent to the start of the previous match
3183         // or is set by the user
3184         int32_t      offset       = usearch_getOffset(strsrch);
3185         USearch     *search       = strsrch->search;
3186         search->reset             = FALSE;
3187         int32_t      textlength   = search->textLength;
3188         if (search->isForwardSearching) {
3189 #if BOYER_MOORE
3190             if (offset == textlength
3191                 || (!search->isOverlap &&
3192                     (offset + strsrch->pattern.defaultShiftSize > textlength ||
3193                     (search->matchedIndex != USEARCH_DONE &&
3194                      offset + search->matchedLength >= textlength)))) {
3195                 // not enough characters to match
3196                 setMatchNotFound(strsrch);
3197                 return USEARCH_DONE;
3198             }
3199 #else
3200             if (offset == textlength ||
3201                 (! search->isOverlap &&
3202                 (search->matchedIndex != USEARCH_DONE &&
3203                 offset + search->matchedLength > textlength))) {
3204                     // not enough characters to match
3205                     setMatchNotFound(strsrch);
3206                     return USEARCH_DONE;
3207             }
3208 #endif
3209         }
3210         else {
3211             // switching direction.
3212             // if matchedIndex == USEARCH_DONE, it means that either a
3213             // setOffset has been called or that previous ran off the text
3214             // string. the iterator would have been set to offset 0 if a
3215             // match is not found.
3216             search->isForwardSearching = TRUE;
3217             if (search->matchedIndex != USEARCH_DONE) {
3218                 // there's no need to set the collation element iterator
3219                 // the next call to next will set the offset.
3220                 return search->matchedIndex;
3221             }
3222         }
3223 
3224         if (U_SUCCESS(*status)) {
3225             if (strsrch->pattern.CELength == 0) {
3226                 if (search->matchedIndex == USEARCH_DONE) {
3227                     search->matchedIndex = offset;
3228                 }
3229                 else { // moves by codepoints
3230                     UTF_FWD_1(search->text, search->matchedIndex, textlength);
3231                 }
3232 
3233                 search->matchedLength = 0;
3234                 setColEIterOffset(strsrch->textIter, search->matchedIndex);
3235                 // status checked below
3236                 if (search->matchedIndex == textlength) {
3237                     search->matchedIndex = USEARCH_DONE;
3238                 }
3239             }
3240             else {
3241                 if (search->matchedLength > 0) {
3242                     // if matchlength is 0 we are at the start of the iteration
3243                     if (search->isOverlap) {
3244                         ucol_setOffset(strsrch->textIter, offset + 1, status);
3245                     }
3246                     else {
3247                         ucol_setOffset(strsrch->textIter,
3248                                        offset + search->matchedLength, status);
3249                     }
3250                 }
3251                 else {
3252                     // for boundary check purposes. this will ensure that the
3253                     // next match will not preceed the current offset
3254                     // note search->matchedIndex will always be set to something
3255                     // in the code
3256                     search->matchedIndex = offset - 1;
3257                 }
3258 
3259                 if (search->isCanonicalMatch) {
3260                     // can't use exact here since extra accents are allowed.
3261                     usearch_handleNextCanonical(strsrch, status);
3262                 }
3263                 else {
3264                     usearch_handleNextExact(strsrch, status);
3265                 }
3266             }
3267 
3268             if (U_FAILURE(*status)) {
3269                 return USEARCH_DONE;
3270             }
3271 
3272 #if !BOYER_MOORE
3273             if (search->matchedIndex == USEARCH_DONE) {
3274                 ucol_setOffset(strsrch->textIter, search->textLength, status);
3275             } else {
3276                 ucol_setOffset(strsrch->textIter, search->matchedIndex, status);
3277             }
3278 #endif
3279 
3280             return search->matchedIndex;
3281         }
3282     }
3283     return USEARCH_DONE;
3284 }
3285 
usearch_previous(UStringSearch * strsrch,UErrorCode * status)3286 U_CAPI int32_t U_EXPORT2 usearch_previous(UStringSearch *strsrch,
3287                                               UErrorCode *status)
3288 {
3289     if (U_SUCCESS(*status) && strsrch) {
3290         int32_t offset;
3291         USearch *search = strsrch->search;
3292         if (search->reset) {
3293             offset                     = search->textLength;
3294             search->isForwardSearching = FALSE;
3295             search->reset              = FALSE;
3296             setColEIterOffset(strsrch->textIter, offset);
3297         }
3298         else {
3299             offset = usearch_getOffset(strsrch);
3300         }
3301 
3302         int32_t matchedindex = search->matchedIndex;
3303         if (search->isForwardSearching == TRUE) {
3304             // switching direction.
3305             // if matchedIndex == USEARCH_DONE, it means that either a
3306             // setOffset has been called or that next ran off the text
3307             // string. the iterator would have been set to offset textLength if
3308             // a match is not found.
3309             search->isForwardSearching = FALSE;
3310             if (matchedindex != USEARCH_DONE) {
3311                 return matchedindex;
3312             }
3313         }
3314         else {
3315 #if BOYER_MOORE
3316             if (offset == 0 || matchedindex == 0 ||
3317                 (!search->isOverlap &&
3318                     (offset < strsrch->pattern.defaultShiftSize ||
3319                     (matchedindex != USEARCH_DONE &&
3320                     matchedindex < strsrch->pattern.defaultShiftSize)))) {
3321                 // not enough characters to match
3322                 setMatchNotFound(strsrch);
3323                 return USEARCH_DONE;
3324             }
3325 #else
3326             // Could check pattern length, but the
3327             // linear search will do the right thing
3328             if (offset == 0 || matchedindex == 0) {
3329                 setMatchNotFound(strsrch);
3330                 return USEARCH_DONE;
3331             }
3332 #endif
3333         }
3334 
3335         if (U_SUCCESS(*status)) {
3336             if (strsrch->pattern.CELength == 0) {
3337                 search->matchedIndex =
3338                       (matchedindex == USEARCH_DONE ? offset : matchedindex);
3339                 if (search->matchedIndex == 0) {
3340                     setMatchNotFound(strsrch);
3341                     // status checked below
3342                 }
3343                 else { // move by codepoints
3344                     UTF_BACK_1(search->text, 0, search->matchedIndex);
3345                     setColEIterOffset(strsrch->textIter, search->matchedIndex);
3346                     // status checked below
3347                     search->matchedLength = 0;
3348                 }
3349             }
3350             else {
3351 #if !BOYER_MOORE
3352                 if (search->matchedIndex != USEARCH_DONE) {
3353                     if (search->isOverlap) {
3354                         ucol_setOffset(strsrch->textIter, search->matchedIndex + search->matchedLength - 2, status);
3355                     }
3356                 }
3357 #endif
3358 
3359                 if (strsrch->search->isCanonicalMatch) {
3360                     // can't use exact here since extra accents are allowed.
3361                     usearch_handlePreviousCanonical(strsrch, status);
3362                     // status checked below
3363                 }
3364                 else {
3365                     usearch_handlePreviousExact(strsrch, status);
3366                     // status checked below
3367                 }
3368             }
3369 
3370             if (U_FAILURE(*status)) {
3371                 return USEARCH_DONE;
3372             }
3373 
3374             return search->matchedIndex;
3375         }
3376     }
3377     return USEARCH_DONE;
3378 }
3379 
3380 
3381 
usearch_reset(UStringSearch * strsrch)3382 U_CAPI void U_EXPORT2 usearch_reset(UStringSearch *strsrch)
3383 {
3384     /*
3385     reset is setting the attributes that are already in
3386     string search, hence all attributes in the collator should
3387     be retrieved without any problems
3388     */
3389     if (strsrch) {
3390         UErrorCode status            = U_ZERO_ERROR;
3391         UBool      sameCollAttribute = TRUE;
3392         uint32_t   ceMask;
3393         UBool      shift;
3394         uint32_t   varTop;
3395 
3396         // **** hack to deal w/ how processed CEs encode quaternary ****
3397         UCollationStrength newStrength = ucol_getStrength(strsrch->collator);
3398         if ((strsrch->strength < UCOL_QUATERNARY && newStrength >= UCOL_QUATERNARY) ||
3399             (strsrch->strength >= UCOL_QUATERNARY && newStrength < UCOL_QUATERNARY)) {
3400                 sameCollAttribute = FALSE;
3401         }
3402 
3403         strsrch->strength    = ucol_getStrength(strsrch->collator);
3404         ceMask = getMask(strsrch->strength);
3405         if (strsrch->ceMask != ceMask) {
3406             strsrch->ceMask = ceMask;
3407             sameCollAttribute = FALSE;
3408         }
3409 
3410         // if status is a failure, ucol_getAttribute returns UCOL_DEFAULT
3411         shift = ucol_getAttribute(strsrch->collator, UCOL_ALTERNATE_HANDLING,
3412                                   &status) == UCOL_SHIFTED;
3413         if (strsrch->toShift != shift) {
3414             strsrch->toShift  = shift;
3415             sameCollAttribute = FALSE;
3416         }
3417 
3418         // if status is a failure, ucol_getVariableTop returns 0
3419         varTop = ucol_getVariableTop(strsrch->collator, &status);
3420         if (strsrch->variableTop != varTop) {
3421             strsrch->variableTop = varTop;
3422             sameCollAttribute    = FALSE;
3423         }
3424         if (!sameCollAttribute) {
3425             initialize(strsrch, &status);
3426         }
3427         /* free offset buffer to avoid memory leak before initializing. */
3428         ucol_freeOffsetBuffer(&(strsrch->textIter->iteratordata_));
3429         uprv_init_collIterate(strsrch->collator, strsrch->search->text,
3430                               strsrch->search->textLength,
3431                               &(strsrch->textIter->iteratordata_),
3432                               &status);
3433         strsrch->search->matchedLength      = 0;
3434         strsrch->search->matchedIndex       = USEARCH_DONE;
3435         strsrch->search->isOverlap          = FALSE;
3436         strsrch->search->isCanonicalMatch   = FALSE;
3437         strsrch->search->elementComparisonType = 0;
3438         strsrch->search->isForwardSearching = TRUE;
3439         strsrch->search->reset              = TRUE;
3440     }
3441 }
3442 
3443 //
3444 //  CEI  Collation Element + source text index.
3445 //       These structs are kept in the circular buffer.
3446 //
3447 struct  CEI {
3448     int64_t ce;
3449     int32_t lowIndex;
3450     int32_t highIndex;
3451 };
3452 
3453 U_NAMESPACE_BEGIN
3454 
3455 
3456 //
3457 //  CEBuffer   A circular buffer of CEs from the text being searched.
3458 //
3459 #define   DEFAULT_CEBUFFER_SIZE 50
3460 struct CEBuffer {
3461     CEI                  defBuf[DEFAULT_CEBUFFER_SIZE];
3462     CEI                 *buf;
3463     int32_t              bufSize;
3464     int32_t              firstIx;
3465     int32_t              limitIx;
3466     UCollationElements  *ceIter;
3467     UStringSearch       *strSearch;
3468 
3469 
3470 
3471                CEBuffer(UStringSearch *ss, UErrorCode *status);
3472                ~CEBuffer();
3473    const CEI   *get(int32_t index);
3474    const CEI   *getPrevious(int32_t index);
3475 };
3476 
3477 
CEBuffer(UStringSearch * ss,UErrorCode * status)3478 CEBuffer::CEBuffer(UStringSearch *ss, UErrorCode *status) {
3479     buf = defBuf;
3480     strSearch = ss;
3481     bufSize = ss->pattern.CELength+10;
3482     ceIter    = ss->textIter;
3483     firstIx = 0;
3484     limitIx = 0;
3485 
3486     uprv_init_pce(ceIter);
3487 
3488     if (bufSize>DEFAULT_CEBUFFER_SIZE) {
3489         buf = (CEI *)uprv_malloc(bufSize * sizeof(CEI));
3490         if (buf == NULL) {
3491             *status = U_MEMORY_ALLOCATION_ERROR;
3492         }
3493     }
3494 }
3495 
3496 // TODO: add a reset or init function so that allocated
3497 //       buffers can be retained & reused.
3498 
~CEBuffer()3499 CEBuffer::~CEBuffer() {
3500     if (buf != defBuf) {
3501         uprv_free(buf);
3502     }
3503 }
3504 
3505 
3506 // Get the CE with the specified index.
3507 //   Index must be in the range
3508 //          n-history_size < index < n+1
3509 //   where n is the largest index to have been fetched by some previous call to this function.
3510 //   The CE value will be UCOL__PROCESSED_NULLORDER at end of input.
3511 //
get(int32_t index)3512 const CEI *CEBuffer::get(int32_t index) {
3513     int i = index % bufSize;
3514 
3515     if (index>=firstIx && index<limitIx) {
3516         // The request was for an entry already in our buffer.
3517         //  Just return it.
3518         return &buf[i];
3519     }
3520 
3521     // Caller is requesting a new, never accessed before, CE.
3522     //   Verify that it is the next one in sequence, which is all
3523     //   that is allowed.
3524     if (index != limitIx) {
3525         U_ASSERT(FALSE);
3526 
3527         return NULL;
3528     }
3529 
3530     // Manage the circular CE buffer indexing
3531     limitIx++;
3532 
3533     if (limitIx - firstIx >= bufSize) {
3534         // The buffer is full, knock out the lowest-indexed entry.
3535         firstIx++;
3536     }
3537 
3538     UErrorCode status = U_ZERO_ERROR;
3539 
3540     buf[i].ce = ucol_nextProcessed(ceIter, &buf[i].lowIndex, &buf[i].highIndex, &status);
3541 
3542     return &buf[i];
3543 }
3544 
3545 // Get the CE with the specified index.
3546 //   Index must be in the range
3547 //          n-history_size < index < n+1
3548 //   where n is the largest index to have been fetched by some previous call to this function.
3549 //   The CE value will be UCOL__PROCESSED_NULLORDER at end of input.
3550 //
getPrevious(int32_t index)3551 const CEI *CEBuffer::getPrevious(int32_t index) {
3552     int i = index % bufSize;
3553 
3554     if (index>=firstIx && index<limitIx) {
3555         // The request was for an entry already in our buffer.
3556         //  Just return it.
3557         return &buf[i];
3558     }
3559 
3560     // Caller is requesting a new, never accessed before, CE.
3561     //   Verify that it is the next one in sequence, which is all
3562     //   that is allowed.
3563     if (index != limitIx) {
3564         U_ASSERT(FALSE);
3565 
3566         return NULL;
3567     }
3568 
3569     // Manage the circular CE buffer indexing
3570     limitIx++;
3571 
3572     if (limitIx - firstIx >= bufSize) {
3573         // The buffer is full, knock out the lowest-indexed entry.
3574         firstIx++;
3575     }
3576 
3577     UErrorCode status = U_ZERO_ERROR;
3578 
3579     buf[i].ce = ucol_previousProcessed(ceIter, &buf[i].lowIndex, &buf[i].highIndex, &status);
3580 
3581     return &buf[i];
3582 }
3583 
3584 U_NAMESPACE_END
3585 
3586 
3587 // #define USEARCH_DEBUG
3588 
3589 #ifdef USEARCH_DEBUG
3590 #include <stdio.h>
3591 #include <stdlib.h>
3592 #endif
3593 
3594 /*
3595  * Find the next break boundary after startIndex. If the UStringSearch object
3596  * has an external break iterator, use that. Otherwise use the internal character
3597  * break iterator.
3598  */
nextBoundaryAfter(UStringSearch * strsrch,int32_t startIndex)3599 static int32_t nextBoundaryAfter(UStringSearch *strsrch, int32_t startIndex) {
3600 #if 0
3601     const UChar *text = strsrch->search->text;
3602     int32_t textLen   = strsrch->search->textLength;
3603 
3604     U_ASSERT(startIndex>=0);
3605     U_ASSERT(startIndex<=textLen);
3606 
3607     if (startIndex >= textLen) {
3608         return startIndex;
3609     }
3610 
3611     UChar32  c;
3612     int32_t  i = startIndex;
3613     U16_NEXT(text, i, textLen, c);
3614 
3615     // If we are on a control character, stop without looking for combining marks.
3616     //    Control characters do not combine.
3617     int32_t gcProperty = u_getIntPropertyValue(c, UCHAR_GRAPHEME_CLUSTER_BREAK);
3618     if (gcProperty==U_GCB_CONTROL || gcProperty==U_GCB_LF || gcProperty==U_GCB_CR) {
3619         return i;
3620     }
3621 
3622     // The initial character was not a control, and can thus accept trailing
3623     //   combining characters.  Advance over however many of them there are.
3624     int32_t  indexOfLastCharChecked;
3625     for (;;) {
3626         indexOfLastCharChecked = i;
3627         if (i>=textLen) {
3628             break;
3629         }
3630         U16_NEXT(text, i, textLen, c);
3631         gcProperty = u_getIntPropertyValue(c, UCHAR_GRAPHEME_CLUSTER_BREAK);
3632         if (gcProperty != U_GCB_EXTEND && gcProperty != U_GCB_SPACING_MARK) {
3633             break;
3634         }
3635     }
3636     return indexOfLastCharChecked;
3637 #elif !UCONFIG_NO_BREAK_ITERATION
3638     UBreakIterator *breakiterator = strsrch->search->breakIter;
3639 
3640     if (breakiterator == NULL) {
3641         breakiterator = strsrch->search->internalBreakIter;
3642     }
3643 
3644     if (breakiterator != NULL) {
3645     	return ubrk_following(breakiterator, startIndex);
3646     }
3647 
3648     return startIndex;
3649 #else
3650     // **** or should we use the original code? ****
3651     return startIndex;
3652 #endif
3653 
3654 }
3655 
3656 /*
3657  * Returns TRUE if index is on a break boundary. If the UStringSearch
3658  * has an external break iterator, test using that, otherwise test
3659  * using the internal character break iterator.
3660  */
isBreakBoundary(UStringSearch * strsrch,int32_t index)3661 static UBool isBreakBoundary(UStringSearch *strsrch, int32_t index) {
3662 #if 0
3663     const UChar *text = strsrch->search->text;
3664     int32_t textLen   = strsrch->search->textLength;
3665 
3666     U_ASSERT(index>=0);
3667     U_ASSERT(index<=textLen);
3668 
3669     if (index>=textLen || index<=0) {
3670         return FALSE;
3671     }
3672 
3673     // If the character at the current index is not a GRAPHEME_EXTEND
3674     //    then we can not be within a combining sequence.
3675     UChar32  c;
3676     U16_GET(text, 0, index, textLen, c);
3677     int32_t gcProperty = u_getIntPropertyValue(c, UCHAR_GRAPHEME_CLUSTER_BREAK);
3678     if (gcProperty != U_GCB_EXTEND && gcProperty != U_GCB_SPACING_MARK) {
3679         return FALSE;
3680     }
3681 
3682     // We are at a combining mark.  If the preceding character is anything
3683     //   except a CONTROL, CR or LF, we are in a combining sequence.
3684     U16_PREV(text, 0, index, c);
3685     gcProperty = u_getIntPropertyValue(c, UCHAR_GRAPHEME_CLUSTER_BREAK);
3686     UBool combining =  !(gcProperty==U_GCB_CONTROL || gcProperty==U_GCB_LF || gcProperty==U_GCB_CR);
3687     return combining;
3688 #elif !UCONFIG_NO_BREAK_ITERATION
3689     UBreakIterator *breakiterator = strsrch->search->breakIter;
3690 
3691     if (breakiterator == NULL) {
3692         breakiterator = strsrch->search->internalBreakIter;
3693     }
3694 
3695     return (breakiterator != NULL && ! ubrk_isBoundary(breakiterator, index));
3696 #else
3697     // **** or use the original code? ****
3698     return FALSE;
3699 #endif
3700 }
3701 
3702 #if 0
3703 static UBool onBreakBoundaries(const UStringSearch *strsrch, int32_t start, int32_t end)
3704 {
3705 #if !UCONFIG_NO_BREAK_ITERATION
3706     UBreakIterator *breakiterator = strsrch->search->breakIter;
3707 
3708     if (breakiterator != NULL) {
3709         int32_t startindex = ubrk_first(breakiterator);
3710         int32_t endindex   = ubrk_last(breakiterator);
3711 
3712         // out-of-range indexes are never boundary positions
3713         if (start < startindex || start > endindex ||
3714             end < startindex || end > endindex) {
3715             return FALSE;
3716         }
3717 
3718         return ubrk_isBoundary(breakiterator, start) &&
3719                ubrk_isBoundary(breakiterator, end);
3720     }
3721 #endif
3722 
3723     return TRUE;
3724 }
3725 #endif
3726 
3727 typedef enum {
3728     U_CE_MATCH = -1,
3729     U_CE_NO_MATCH = 0,
3730     U_CE_SKIP_TARG,
3731     U_CE_SKIP_PATN
3732 } UCompareCEsResult;
3733 #define U_CE_LEVEL2_BASE 0x00000005
3734 #define U_CE_LEVEL3_BASE 0x00050000
3735 
compareCE64s(int64_t targCE,int64_t patCE,int16_t compareType)3736 static UCompareCEsResult compareCE64s(int64_t targCE, int64_t patCE, int16_t compareType) {
3737     if (targCE == patCE) {
3738         return U_CE_MATCH;
3739     }
3740     if (compareType == 0) {
3741         return U_CE_NO_MATCH;
3742     }
3743 
3744     int64_t targCEshifted = targCE >> 32;
3745     int64_t patCEshifted = patCE >> 32;
3746     int64_t mask;
3747 
3748     mask = 0xFFFF0000;
3749     int32_t targLev1 = (int32_t)(targCEshifted & mask);
3750     int32_t patLev1 = (int32_t)(patCEshifted & mask);
3751     if ( targLev1 != patLev1 ) {
3752         if ( targLev1 == 0 ) {
3753             return U_CE_SKIP_TARG;
3754         }
3755         if ( patLev1 == 0 && compareType == USEARCH_ANY_BASE_WEIGHT_IS_WILDCARD ) {
3756             return U_CE_SKIP_PATN;
3757         }
3758         return U_CE_NO_MATCH;
3759     }
3760 
3761     mask = 0x0000FFFF;
3762     int32_t targLev2 = (int32_t)(targCEshifted & mask);
3763     int32_t patLev2 = (int32_t)(patCEshifted & mask);
3764     if ( targLev2 != patLev2 ) {
3765         if ( targLev2 == 0 ) {
3766             return U_CE_SKIP_TARG;
3767         }
3768         if ( patLev2 == 0 && compareType == USEARCH_ANY_BASE_WEIGHT_IS_WILDCARD ) {
3769             return U_CE_SKIP_PATN;
3770         }
3771         return (patLev2 == U_CE_LEVEL2_BASE || (compareType == USEARCH_ANY_BASE_WEIGHT_IS_WILDCARD && targLev2 == U_CE_LEVEL2_BASE) )?
3772             U_CE_MATCH: U_CE_NO_MATCH;
3773     }
3774 
3775     mask = 0xFFFF0000;
3776     int32_t targLev3 = (int32_t)(targCE & mask);
3777     int32_t patLev3 = (int32_t)(patCE & mask);
3778     if ( targLev3 != patLev3 ) {
3779         return (patLev3 == U_CE_LEVEL3_BASE || (compareType == USEARCH_ANY_BASE_WEIGHT_IS_WILDCARD && targLev3 == U_CE_LEVEL3_BASE) )?
3780             U_CE_MATCH: U_CE_NO_MATCH;
3781    }
3782 
3783     return U_CE_MATCH;
3784 }
3785 
3786 #if BOYER_MOORE
3787 // TODO: #if BOYER_MOORE, need 32-bit version of compareCE64s
3788 #endif
3789 
usearch_search(UStringSearch * strsrch,int32_t startIdx,int32_t * matchStart,int32_t * matchLimit,UErrorCode * status)3790 U_CAPI UBool U_EXPORT2 usearch_search(UStringSearch  *strsrch,
3791                                        int32_t        startIdx,
3792                                        int32_t        *matchStart,
3793                                        int32_t        *matchLimit,
3794                                        UErrorCode     *status)
3795 {
3796     if (U_FAILURE(*status)) {
3797         return FALSE;
3798     }
3799 
3800     // TODO:  reject search patterns beginning with a combining char.
3801 
3802 #ifdef USEARCH_DEBUG
3803     if (getenv("USEARCH_DEBUG") != NULL) {
3804         printf("Pattern CEs\n");
3805         for (int ii=0; ii<strsrch->pattern.CELength; ii++) {
3806             printf(" %8x", strsrch->pattern.CE[ii]);
3807         }
3808         printf("\n");
3809     }
3810 
3811 #endif
3812     // Input parameter sanity check.
3813     //  TODO:  should input indicies clip to the text length
3814     //         in the same way that UText does.
3815     if(strsrch->pattern.CELength == 0         ||
3816        startIdx < 0                           ||
3817        startIdx > strsrch->search->textLength ||
3818        strsrch->pattern.CE == NULL) {
3819            *status = U_ILLEGAL_ARGUMENT_ERROR;
3820            return FALSE;
3821     }
3822 
3823     if (strsrch->pattern.PCE == NULL) {
3824         initializePatternPCETable(strsrch, status);
3825     }
3826 
3827     ucol_setOffset(strsrch->textIter, startIdx, status);
3828     CEBuffer ceb(strsrch, status);
3829 
3830 
3831     int32_t    targetIx = 0;
3832     const CEI *targetCEI = NULL;
3833     int32_t    patIx;
3834     UBool      found;
3835 
3836     int32_t  mStart = -1;
3837     int32_t  mLimit = -1;
3838     int32_t  minLimit;
3839     int32_t  maxLimit;
3840 
3841 
3842 
3843     // Outer loop moves over match starting positions in the
3844     //      target CE space.
3845     // Here we see the target as a sequence of collation elements, resulting from the following:
3846     // 1. Target characters were decomposed, and (if appropriate) other compressions and expansions are applied
3847     //    (for example, digraphs such as IJ may be broken into two characters).
3848     // 2. An int64_t CE weight is determined for each resulting unit (high 16 bits are primary strength, next
3849     //    16 bits are secondary, next 16 (the high 16 bits of the low 32-bit half) are tertiary. Any of these
3850     //    fields that are for strengths below that of the collator are set to 0. If this makes the int64_t
3851     //    CE weight 0 (as for a combining diacritic with secondary weight when the collator strentgh is primary),
3852     //    then the CE is deleted, so the following code sees only CEs that are relevant.
3853     // For each CE, the lowIndex and highIndex correspond to where this CE begins and ends in the original text.
3854     // If lowIndex==highIndex, either the CE resulted from an expansion/decomposition of one of the original text
3855     // characters, or the CE marks the limit of the target text (in which case the CE weight is UCOL_PROCESSED_NULLORDER).
3856     //
3857     for(targetIx=0; ; targetIx++)
3858     {
3859         found = TRUE;
3860         //  Inner loop checks for a match beginning at each
3861         //  position from the outer loop.
3862         int32_t targetIxOffset = 0;
3863         int64_t patCE = 0;
3864         for (patIx=0; patIx<strsrch->pattern.PCELength; patIx++) {
3865             patCE = strsrch->pattern.PCE[patIx];
3866             targetCEI = ceb.get(targetIx+patIx+targetIxOffset);
3867             //  Compare CE from target string with CE from the pattern.
3868             //    Note that the target CE will be UCOL_PROCESSED_NULLORDER if we reach the end of input,
3869             //    which will fail the compare, below.
3870             UCompareCEsResult ceMatch = compareCE64s(targetCEI->ce, patCE, strsrch->search->elementComparisonType);
3871             if ( ceMatch == U_CE_NO_MATCH ) {
3872                 found = FALSE;
3873                 break;
3874             } else if ( ceMatch > U_CE_NO_MATCH ) {
3875                 if ( ceMatch == U_CE_SKIP_TARG ) {
3876                     // redo with same patCE, next targCE
3877                     patIx--;
3878                     targetIxOffset++;
3879                 } else { // ceMatch == U_CE_SKIP_PATN
3880                     // redo with same targCE, next patCE
3881                     targetIxOffset--;
3882                 }
3883             }
3884         }
3885         targetIxOffset += strsrch->pattern.PCELength; // this is now the offset in target CE space to end of the match so far
3886 
3887         if (!found && ((targetCEI == NULL) || (targetCEI->ce != UCOL_PROCESSED_NULLORDER))) {
3888             // No match at this targetIx.  Try again at the next.
3889             continue;
3890         }
3891 
3892         if (!found) {
3893             // No match at all, we have run off the end of the target text.
3894             break;
3895         }
3896 
3897 
3898         // We have found a match in CE space.
3899         // Now determine the bounds in string index space.
3900         //  There still is a chance of match failure if the CE range not correspond to
3901         //     an acceptable character range.
3902         //
3903         const CEI *firstCEI = ceb.get(targetIx);
3904         const CEI *lastCEI  = ceb.get(targetIx + targetIxOffset - 1);
3905 
3906         mStart   = firstCEI->lowIndex;
3907         minLimit = lastCEI->lowIndex;
3908 
3909         // Look at the CE following the match.  If it is UCOL_NULLORDER the match
3910         //   extended to the end of input, and the match is good.
3911 
3912         // Look at the high and low indices of the CE following the match. If
3913         // they are the same it means one of two things:
3914         //    1. The match extended to the last CE from the target text, which is OK, or
3915         //    2. The last CE that was part of the match is in an expansion that extends
3916         //       to the first CE after the match. In this case, we reject the match.
3917         if (strsrch->search->elementComparisonType == 0) {
3918             const CEI *nextCEI  = ceb.get(targetIx + targetIxOffset);
3919             maxLimit = nextCEI->lowIndex;
3920             if (nextCEI->lowIndex == nextCEI->highIndex && nextCEI->ce != UCOL_PROCESSED_NULLORDER) {
3921                 found = FALSE;
3922             }
3923         } else {
3924             const CEI *nextCEI;
3925             for ( ; ; ++targetIxOffset ) {
3926                 nextCEI = ceb.get(targetIx + targetIxOffset);
3927                 maxLimit = nextCEI->lowIndex;
3928 				// If we are at the end of the target too, match succeeds
3929                 if (  nextCEI->ce == UCOL_PROCESSED_NULLORDER ) {
3930                     break;
3931                 }
3932                 // As long as the next CE has primary weight of 0,
3933                 // it is part of the last target element matched by the pattern;
3934                 // make sure it can be part of a match with the last patCE
3935                 if ( (((nextCEI->ce) >> 32) & 0xFFFF0000UL) == 0 ) {
3936                 	UCompareCEsResult ceMatch = compareCE64s(nextCEI->ce, patCE, strsrch->search->elementComparisonType);
3937                 	if ( ceMatch == U_CE_NO_MATCH || ceMatch == U_CE_SKIP_PATN ) {
3938                 		found = FALSE;
3939                 		break;
3940                 	}
3941                 // If lowIndex == highIndex, this target CE is part of an expansion of the last matched
3942                 // target element, but it has non-zero primary weight => match fails
3943                 } else if ( nextCEI->lowIndex == nextCEI->highIndex ) {
3944                 	found = false;
3945                 	break;
3946                 // Else the target CE is not part of an expansion of the last matched element, match succeeds
3947                 } else {
3948                 	break;
3949                 }
3950             }
3951         }
3952 
3953 
3954         // Check for the start of the match being within a combining sequence.
3955         //   This can happen if the pattern itself begins with a combining char, and
3956         //   the match found combining marks in the target text that were attached
3957         //    to something else.
3958         //   This type of match should be rejected for not completely consuming a
3959         //   combining sequence.
3960         if (isBreakBoundary(strsrch, mStart)) {
3961             found = FALSE;
3962         }
3963 
3964         // Check for the start of the match being within an Collation Element Expansion,
3965         //   meaning that the first char of the match is only partially matched.
3966         //   With exapnsions, the first CE will report the index of the source
3967         //   character, and all subsequent (expansions) CEs will report the source index of the
3968         //    _following_ character.
3969         int32_t secondIx = firstCEI->highIndex;
3970         if (mStart == secondIx) {
3971             found = FALSE;
3972         }
3973 
3974         //  Advance the match end position to the first acceptable match boundary.
3975         //    This advances the index over any combining charcters.
3976         mLimit = maxLimit;
3977         if (minLimit < maxLimit) {
3978             int32_t nba = nextBoundaryAfter(strsrch, minLimit);
3979 
3980             if (nba >= lastCEI->highIndex) {
3981                 mLimit = nba;
3982             }
3983         }
3984 
3985     #ifdef USEARCH_DEBUG
3986         if (getenv("USEARCH_DEBUG") != NULL) {
3987             printf("minLimit, maxLimit, mLimit = %d, %d, %d\n", minLimit, maxLimit, mLimit);
3988         }
3989     #endif
3990 
3991         // If advancing to the end of a combining sequence in character indexing space
3992         //   advanced us beyond the end of the match in CE space, reject this match.
3993         if (mLimit > maxLimit) {
3994             found = FALSE;
3995         }
3996 
3997         if (isBreakBoundary(strsrch, mLimit)) {
3998             found = FALSE;
3999         }
4000 
4001         if (! checkIdentical(strsrch, mStart, mLimit)) {
4002             found = FALSE;
4003         }
4004 
4005         if (found) {
4006             break;
4007         }
4008     }
4009 
4010     #ifdef USEARCH_DEBUG
4011     if (getenv("USEARCH_DEBUG") != NULL) {
4012         printf("Target CEs [%d .. %d]\n", ceb.firstIx, ceb.limitIx);
4013         int32_t  lastToPrint = ceb.limitIx+2;
4014         for (int ii=ceb.firstIx; ii<lastToPrint; ii++) {
4015             printf("%8x@%d ", ceb.get(ii)->ce, ceb.get(ii)->srcIndex);
4016         }
4017         printf("\n%s\n", found? "match found" : "no match");
4018     }
4019     #endif
4020 
4021     // All Done.  Store back the match bounds to the caller.
4022     //
4023     if (found==FALSE) {
4024         mLimit = -1;
4025         mStart = -1;
4026     }
4027 
4028     if (matchStart != NULL) {
4029         *matchStart= mStart;
4030     }
4031 
4032     if (matchLimit != NULL) {
4033         *matchLimit = mLimit;
4034     }
4035 
4036     return found;
4037 }
4038 
4039 
usearch_searchBackwards(UStringSearch * strsrch,int32_t startIdx,int32_t * matchStart,int32_t * matchLimit,UErrorCode * status)4040 U_CAPI UBool U_EXPORT2 usearch_searchBackwards(UStringSearch  *strsrch,
4041                                                 int32_t        startIdx,
4042                                                 int32_t        *matchStart,
4043                                                 int32_t        *matchLimit,
4044                                                 UErrorCode     *status)
4045 {
4046     if (U_FAILURE(*status)) {
4047         return FALSE;
4048     }
4049 
4050     // TODO:  reject search patterns beginning with a combining char.
4051 
4052 #ifdef USEARCH_DEBUG
4053     if (getenv("USEARCH_DEBUG") != NULL) {
4054         printf("Pattern CEs\n");
4055         for (int ii=0; ii<strsrch->pattern.CELength; ii++) {
4056             printf(" %8x", strsrch->pattern.CE[ii]);
4057         }
4058         printf("\n");
4059     }
4060 
4061 #endif
4062     // Input parameter sanity check.
4063     //  TODO:  should input indicies clip to the text length
4064     //         in the same way that UText does.
4065     if(strsrch->pattern.CELength == 0         ||
4066        startIdx < 0                           ||
4067        startIdx > strsrch->search->textLength ||
4068        strsrch->pattern.CE == NULL) {
4069            *status = U_ILLEGAL_ARGUMENT_ERROR;
4070            return FALSE;
4071     }
4072 
4073     if (strsrch->pattern.PCE == NULL) {
4074         initializePatternPCETable(strsrch, status);
4075     }
4076 
4077     CEBuffer ceb(strsrch, status);
4078     int32_t    targetIx = 0;
4079 
4080     /*
4081      * Pre-load the buffer with the CE's for the grapheme
4082      * after our starting position so that we're sure that
4083      * we can look at the CE following the match when we
4084      * check the match boundaries.
4085      *
4086      * This will also pre-fetch the first CE that we'll
4087      * consider for the match.
4088      */
4089     if (startIdx < strsrch->search->textLength) {
4090         UBreakIterator *bi = strsrch->search->internalBreakIter;
4091         int32_t next = ubrk_following(bi, startIdx);
4092 
4093         ucol_setOffset(strsrch->textIter, next, status);
4094 
4095         for (targetIx = 0; ; targetIx += 1) {
4096             if (ceb.getPrevious(targetIx)->lowIndex < startIdx) {
4097                 break;
4098             }
4099         }
4100     } else {
4101         ucol_setOffset(strsrch->textIter, startIdx, status);
4102     }
4103 
4104 
4105     const CEI *targetCEI = NULL;
4106     int32_t    patIx;
4107     UBool      found;
4108 
4109     int32_t  limitIx = targetIx;
4110     int32_t  mStart = -1;
4111     int32_t  mLimit = -1;
4112     int32_t  minLimit;
4113     int32_t  maxLimit;
4114 
4115 
4116 
4117     // Outer loop moves over match starting positions in the
4118     //      target CE space.
4119     // Here, targetIx values increase toward the beginning of the base text (i.e. we get the text CEs in reverse order).
4120     // But  patIx is 0 at the beginning of the pattern and increases toward the end.
4121     // So this loop performs a comparison starting with the end of pattern, and prcessd toward the beginning of the pattern
4122     // and the beginning of the base text.
4123     for(targetIx = limitIx; ; targetIx += 1)
4124     {
4125         found = TRUE;
4126         //  Inner loop checks for a match beginning at each
4127         //  position from the outer loop.
4128         int32_t targetIxOffset = 0;
4129         for (patIx = strsrch->pattern.PCELength - 1; patIx >= 0; patIx -= 1) {
4130             int64_t patCE = strsrch->pattern.PCE[patIx];
4131 
4132             targetCEI = ceb.getPrevious(targetIx + strsrch->pattern.PCELength - 1 - patIx + targetIxOffset);
4133             //  Compare CE from target string with CE from the pattern.
4134             //    Note that the target CE will be UCOL_NULLORDER if we reach the end of input,
4135             //    which will fail the compare, below.
4136             UCompareCEsResult ceMatch = compareCE64s(targetCEI->ce, patCE, strsrch->search->elementComparisonType);
4137             if ( ceMatch == U_CE_NO_MATCH ) {
4138                 found = FALSE;
4139                 break;
4140             } else if ( ceMatch > U_CE_NO_MATCH ) {
4141                 if ( ceMatch == U_CE_SKIP_TARG ) {
4142                     // redo with same patCE, next targCE
4143                     patIx++;
4144                     targetIxOffset++;
4145                 } else { // ceMatch == U_CE_SKIP_PATN
4146                     // redo with same targCE, next patCE
4147                     targetIxOffset--;
4148                 }
4149             }
4150         }
4151 
4152         if (!found && ((targetCEI == NULL) || (targetCEI->ce != UCOL_PROCESSED_NULLORDER))) {
4153             // No match at this targetIx.  Try again at the next.
4154             continue;
4155         }
4156 
4157         if (!found) {
4158             // No match at all, we have run off the end of the target text.
4159             break;
4160         }
4161 
4162 
4163         // We have found a match in CE space.
4164         // Now determine the bounds in string index space.
4165         //  There still is a chance of match failure if the CE range not correspond to
4166         //     an acceptable character range.
4167         //
4168         const CEI *firstCEI = ceb.getPrevious(targetIx + strsrch->pattern.PCELength - 1 + targetIxOffset);
4169         const CEI *lastCEI  = ceb.getPrevious(targetIx);
4170         const CEI *nextCEI  = targetIx > 0? ceb.getPrevious(targetIx - 1) : NULL;
4171 
4172         mStart   = firstCEI->lowIndex;
4173         minLimit = lastCEI->lowIndex;
4174         maxLimit = targetIx > 0? nextCEI->lowIndex : lastCEI->highIndex;
4175 
4176         // Look at the CE following the match.  If it is UCOL_NULLORDER the match
4177         //   extended to the end of input, and the match is good.
4178 
4179         // Look at the high and low indices of the CE following the match. If
4180         // they are the same it means one of two things:
4181         //    1. The match extended to the last CE from the target text, which is OK, or
4182         //    2. The last CE that was part of the match is in an expansion that extends
4183         //       to the first CE after the match. In this case, we reject the match.
4184         if (targetIx >= 1) {
4185             if (nextCEI->lowIndex == nextCEI->highIndex && nextCEI->ce != UCOL_PROCESSED_NULLORDER) {
4186                 found = FALSE;
4187             }
4188         }
4189 
4190 
4191         // Check for the start of the match being within a combining sequence.
4192         //   This can happen if the pattern itself begins with a combining char, and
4193         //   the match found combining marks in the target text that were attached
4194         //    to something else.
4195         //   This type of match should be rejected for not completely consuming a
4196         //   combining sequence.
4197         if (isBreakBoundary(strsrch, mStart)) {
4198             found = FALSE;
4199         }
4200 
4201         // Look at the high index of the first CE in the match. If it's the same as the
4202         // low index, the first CE in the match is in the middle of an expansion.
4203         if (mStart == firstCEI->highIndex) {
4204             found = FALSE;
4205         }
4206 
4207         //  Advance the match end position to the first acceptable match boundary.
4208         //    This advances the index over any combining charcters.
4209         mLimit = maxLimit;
4210         if (/*targetIx > 0 &&*/ minLimit < maxLimit) {
4211             int32_t nba = nextBoundaryAfter(strsrch, minLimit);
4212 
4213             if (nba >= lastCEI->highIndex) {
4214                 mLimit = nba;
4215             }
4216         }
4217 
4218     #ifdef USEARCH_DEBUG
4219         if (getenv("USEARCH_DEBUG") != NULL) {
4220             printf("minLimit, maxLimit, mLimit = %d, %d, %d\n", minLimit, maxLimit, mLimit);
4221         }
4222     #endif
4223 
4224         // If advancing to the end of a combining sequence in character indexing space
4225         //   advanced us beyond the end of the match in CE space, reject this match.
4226         if (mLimit > maxLimit) {
4227             found = FALSE;
4228         }
4229 
4230         // Make sure the end of the match is on a break boundary
4231         if (isBreakBoundary(strsrch, mLimit)) {
4232             found = FALSE;
4233         }
4234 
4235         if (! checkIdentical(strsrch, mStart, mLimit)) {
4236             found = FALSE;
4237         }
4238 
4239         if (found) {
4240             break;
4241         }
4242     }
4243 
4244     #ifdef USEARCH_DEBUG
4245     if (getenv("USEARCH_DEBUG") != NULL) {
4246         printf("Target CEs [%d .. %d]\n", ceb.firstIx, ceb.limitIx);
4247         int32_t  lastToPrint = ceb.limitIx+2;
4248         for (int ii=ceb.firstIx; ii<lastToPrint; ii++) {
4249             printf("%8x@%d ", ceb.get(ii)->ce, ceb.get(ii)->srcIndex);
4250         }
4251         printf("\n%s\n", found? "match found" : "no match");
4252     }
4253     #endif
4254 
4255     // All Done.  Store back the match bounds to the caller.
4256     //
4257     if (found==FALSE) {
4258         mLimit = -1;
4259         mStart = -1;
4260     }
4261 
4262     if (matchStart != NULL) {
4263         *matchStart= mStart;
4264     }
4265 
4266     if (matchLimit != NULL) {
4267         *matchLimit = mLimit;
4268     }
4269 
4270     return found;
4271 }
4272 
4273 
4274 
4275 
4276 // internal use methods declared in usrchimp.h -----------------------------
4277 
usearch_handleNextExact(UStringSearch * strsrch,UErrorCode * status)4278 UBool usearch_handleNextExact(UStringSearch *strsrch, UErrorCode *status)
4279 {
4280     if (U_FAILURE(*status)) {
4281         setMatchNotFound(strsrch);
4282         return FALSE;
4283     }
4284 
4285 #if BOYER_MOORE
4286     UCollationElements *coleiter        = strsrch->textIter;
4287     int32_t             textlength      = strsrch->search->textLength;
4288     int32_t            *patternce       = strsrch->pattern.CE;
4289     int32_t             patterncelength = strsrch->pattern.CELength;
4290     int32_t             textoffset      = ucol_getOffset(coleiter);
4291 
4292     // status used in setting coleiter offset, since offset is checked in
4293     // shiftForward before setting the coleiter offset, status never
4294     // a failure
4295     textoffset = shiftForward(strsrch, textoffset, UCOL_NULLORDER,
4296                               patterncelength);
4297     while (textoffset <= textlength)
4298     {
4299         uint32_t    patternceindex = patterncelength - 1;
4300         int32_t     targetce;
4301         UBool       found          = FALSE;
4302         int32_t    lastce          = UCOL_NULLORDER;
4303 
4304         setColEIterOffset(coleiter, textoffset);
4305 
4306         for (;;) {
4307             // finding the last pattern ce match, imagine composite characters
4308             // for example: search for pattern A in text \u00C0
4309             // we'll have to skip \u0300 the grave first before we get to A
4310             targetce = ucol_previous(coleiter, status);
4311             if (U_FAILURE(*status) || targetce == UCOL_NULLORDER) {
4312                 found = FALSE;
4313                 break;
4314             }
4315             targetce = getCE(strsrch, targetce);
4316             if (targetce == UCOL_IGNORABLE && inNormBuf(coleiter)) {
4317                 // this is for the text \u0315\u0300 that requires
4318                 // normalization and pattern \u0300, where \u0315 is ignorable
4319                 continue;
4320             }
4321             if (lastce == UCOL_NULLORDER || lastce == UCOL_IGNORABLE) {
4322                 lastce = targetce;
4323             }
4324             // TODO: #if BOYER_MOORE, replace with code using 32-bit version of compareCE64s
4325             if (targetce == patternce[patternceindex]) {
4326                 // the first ce can be a contraction
4327                 found = TRUE;
4328                 break;
4329             }
4330             if (!hasExpansion(coleiter)) {
4331                 found = FALSE;
4332                 break;
4333             }
4334         }
4335 
4336         //targetce = lastce;
4337 
4338         while (found && patternceindex > 0) {
4339         	lastce = targetce;
4340             targetce    = ucol_previous(coleiter, status);
4341             if (U_FAILURE(*status) || targetce == UCOL_NULLORDER) {
4342                 found = FALSE;
4343                 break;
4344             }
4345             targetce    = getCE(strsrch, targetce);
4346             if (targetce == UCOL_IGNORABLE) {
4347                 continue;
4348             }
4349 
4350             patternceindex --;
4351             // TODO: #if BOYER_MOORE, replace with code using 32-bit version of compareCE64s
4352             found = found && targetce == patternce[patternceindex];
4353         }
4354 
4355         targetce = lastce;
4356 
4357         if (!found) {
4358             if (U_FAILURE(*status)) {
4359                 break;
4360             }
4361             textoffset = shiftForward(strsrch, textoffset, lastce,
4362                                       patternceindex);
4363             // status checked at loop.
4364             patternceindex = patterncelength;
4365             continue;
4366         }
4367 
4368         if (checkNextExactMatch(strsrch, &textoffset, status)) {
4369             // status checked in ucol_setOffset
4370             setColEIterOffset(coleiter, strsrch->search->matchedIndex);
4371             return TRUE;
4372         }
4373     }
4374     setMatchNotFound(strsrch);
4375     return FALSE;
4376 #else
4377     int32_t textOffset = ucol_getOffset(strsrch->textIter);
4378     int32_t start = -1;
4379     int32_t end = -1;
4380 
4381     if (usearch_search(strsrch, textOffset, &start, &end, status)) {
4382         strsrch->search->matchedIndex  = start;
4383         strsrch->search->matchedLength = end - start;
4384         return TRUE;
4385     } else {
4386         setMatchNotFound(strsrch);
4387         return FALSE;
4388     }
4389 #endif
4390 }
4391 
usearch_handleNextCanonical(UStringSearch * strsrch,UErrorCode * status)4392 UBool usearch_handleNextCanonical(UStringSearch *strsrch, UErrorCode *status)
4393 {
4394     if (U_FAILURE(*status)) {
4395         setMatchNotFound(strsrch);
4396         return FALSE;
4397     }
4398 
4399 #if BOYER_MOORE
4400     UCollationElements *coleiter        = strsrch->textIter;
4401     int32_t             textlength      = strsrch->search->textLength;
4402     int32_t            *patternce       = strsrch->pattern.CE;
4403     int32_t             patterncelength = strsrch->pattern.CELength;
4404     int32_t             textoffset      = ucol_getOffset(coleiter);
4405     UBool               hasPatternAccents =
4406        strsrch->pattern.hasSuffixAccents || strsrch->pattern.hasPrefixAccents;
4407 
4408     textoffset = shiftForward(strsrch, textoffset, UCOL_NULLORDER,
4409                               patterncelength);
4410     strsrch->canonicalPrefixAccents[0] = 0;
4411     strsrch->canonicalSuffixAccents[0] = 0;
4412 
4413     while (textoffset <= textlength)
4414     {
4415         int32_t     patternceindex = patterncelength - 1;
4416         int32_t     targetce;
4417         UBool       found          = FALSE;
4418         int32_t     lastce         = UCOL_NULLORDER;
4419 
4420         setColEIterOffset(coleiter, textoffset);
4421 
4422         for (;;) {
4423             // finding the last pattern ce match, imagine composite characters
4424             // for example: search for pattern A in text \u00C0
4425             // we'll have to skip \u0300 the grave first before we get to A
4426             targetce = ucol_previous(coleiter, status);
4427             if (U_FAILURE(*status) || targetce == UCOL_NULLORDER) {
4428                 found = FALSE;
4429                 break;
4430             }
4431             targetce = getCE(strsrch, targetce);
4432             if (lastce == UCOL_NULLORDER || lastce == UCOL_IGNORABLE) {
4433                 lastce = targetce;
4434             }
4435             // TODO: #if BOYER_MOORE, replace with code using 32-bit version of compareCE64s
4436             if (targetce == patternce[patternceindex]) {
4437                 // the first ce can be a contraction
4438                 found = TRUE;
4439                 break;
4440             }
4441             if (!hasExpansion(coleiter)) {
4442                 found = FALSE;
4443                 break;
4444             }
4445         }
4446 
4447         while (found && patternceindex > 0) {
4448             targetce    = ucol_previous(coleiter, status);
4449             if (U_FAILURE(*status) || targetce == UCOL_NULLORDER) {
4450                 found = FALSE;
4451                 break;
4452             }
4453             targetce    = getCE(strsrch, targetce);
4454             if (targetce == UCOL_IGNORABLE) {
4455                 continue;
4456             }
4457 
4458             patternceindex --;
4459             // TODO: #if BOYER_MOORE, replace with code using 32-bit version of compareCE64s
4460             found = found && targetce == patternce[patternceindex];
4461         }
4462 
4463         // initializing the rearranged accent array
4464         if (hasPatternAccents && !found) {
4465             strsrch->canonicalPrefixAccents[0] = 0;
4466             strsrch->canonicalSuffixAccents[0] = 0;
4467             if (U_FAILURE(*status)) {
4468                 break;
4469             }
4470             found = doNextCanonicalMatch(strsrch, textoffset, status);
4471         }
4472 
4473         if (!found) {
4474             if (U_FAILURE(*status)) {
4475                 break;
4476             }
4477             textoffset = shiftForward(strsrch, textoffset, lastce,
4478                                       patternceindex);
4479             // status checked at loop
4480             patternceindex = patterncelength;
4481             continue;
4482         }
4483 
4484         if (checkNextCanonicalMatch(strsrch, &textoffset, status)) {
4485             setColEIterOffset(coleiter, strsrch->search->matchedIndex);
4486             return TRUE;
4487         }
4488     }
4489     setMatchNotFound(strsrch);
4490     return FALSE;
4491 #else
4492     int32_t textOffset = ucol_getOffset(strsrch->textIter);
4493     int32_t start = -1;
4494     int32_t end = -1;
4495 
4496     if (usearch_search(strsrch, textOffset, &start, &end, status)) {
4497         strsrch->search->matchedIndex  = start;
4498         strsrch->search->matchedLength = end - start;
4499         return TRUE;
4500     } else {
4501         setMatchNotFound(strsrch);
4502         return FALSE;
4503     }
4504 #endif
4505 }
4506 
usearch_handlePreviousExact(UStringSearch * strsrch,UErrorCode * status)4507 UBool usearch_handlePreviousExact(UStringSearch *strsrch, UErrorCode *status)
4508 {
4509     if (U_FAILURE(*status)) {
4510         setMatchNotFound(strsrch);
4511         return FALSE;
4512     }
4513 
4514 #if BOYER_MOORE
4515     UCollationElements *coleiter        = strsrch->textIter;
4516     int32_t            *patternce       = strsrch->pattern.CE;
4517     int32_t             patterncelength = strsrch->pattern.CELength;
4518     int32_t             textoffset      = ucol_getOffset(coleiter);
4519 
4520     // shifting it check for setting offset
4521     // if setOffset is called previously or there was no previous match, we
4522     // leave the offset as it is.
4523     if (strsrch->search->matchedIndex != USEARCH_DONE) {
4524         textoffset = strsrch->search->matchedIndex;
4525     }
4526 
4527     textoffset = reverseShift(strsrch, textoffset, UCOL_NULLORDER,
4528                               patterncelength);
4529 
4530     while (textoffset >= 0)
4531     {
4532         int32_t     patternceindex = 1;
4533         int32_t     targetce;
4534         UBool       found          = FALSE;
4535         int32_t     firstce        = UCOL_NULLORDER;
4536 
4537         // if status is a failure, ucol_setOffset does nothing
4538         setColEIterOffset(coleiter, textoffset);
4539 
4540         for (;;) {
4541             // finding the first pattern ce match, imagine composite
4542             // characters. for example: search for pattern \u0300 in text
4543             // \u00C0, we'll have to skip A first before we get to
4544             // \u0300 the grave accent
4545             targetce = ucol_next(coleiter, status);
4546             if (U_FAILURE(*status) || targetce == UCOL_NULLORDER) {
4547                 found = FALSE;
4548                 break;
4549             }
4550             targetce = getCE(strsrch, targetce);
4551             if (firstce == UCOL_NULLORDER || firstce == UCOL_IGNORABLE) {
4552                 firstce = targetce;
4553             }
4554             if (targetce == UCOL_IGNORABLE && strsrch->strength != UCOL_PRIMARY) {
4555                 continue;
4556             }
4557             // TODO: #if BOYER_MOORE, replace with code using 32-bit version of compareCE64s
4558             if (targetce == patternce[0]) {
4559                 found = TRUE;
4560                 break;
4561             }
4562             if (!hasExpansion(coleiter)) {
4563                 // checking for accents in composite character
4564                 found = FALSE;
4565                 break;
4566             }
4567         }
4568 
4569         //targetce = firstce;
4570 
4571         while (found && (patternceindex < patterncelength)) {
4572         	firstce = targetce;
4573             targetce    = ucol_next(coleiter, status);
4574             if (U_FAILURE(*status) || targetce == UCOL_NULLORDER) {
4575                 found = FALSE;
4576                 break;
4577             }
4578             targetce    = getCE(strsrch, targetce);
4579             if (targetce == UCOL_IGNORABLE) {
4580                 continue;
4581             }
4582 
4583             // TODO: #if BOYER_MOORE, replace with code using 32-bit version of compareCE64s
4584             found = found && targetce == patternce[patternceindex];
4585             patternceindex ++;
4586         }
4587 
4588         targetce = firstce;
4589 
4590         if (!found) {
4591             if (U_FAILURE(*status)) {
4592                 break;
4593             }
4594 
4595             textoffset = reverseShift(strsrch, textoffset, targetce,
4596                                       patternceindex);
4597             patternceindex = 0;
4598             continue;
4599         }
4600 
4601         if (checkPreviousExactMatch(strsrch, &textoffset, status)) {
4602             setColEIterOffset(coleiter, textoffset);
4603             return TRUE;
4604         }
4605     }
4606     setMatchNotFound(strsrch);
4607     return FALSE;
4608 #else
4609     int32_t textOffset = ucol_getOffset(strsrch->textIter);
4610     int32_t start = -1;
4611     int32_t end = -1;
4612 
4613     if (usearch_searchBackwards(strsrch, textOffset, &start, &end, status)) {
4614         strsrch->search->matchedIndex = start;
4615         strsrch->search->matchedLength = end - start;
4616         return TRUE;
4617     } else {
4618         setMatchNotFound(strsrch);
4619         return FALSE;
4620     }
4621 #endif
4622 }
4623 
usearch_handlePreviousCanonical(UStringSearch * strsrch,UErrorCode * status)4624 UBool usearch_handlePreviousCanonical(UStringSearch *strsrch,
4625                                       UErrorCode    *status)
4626 {
4627     if (U_FAILURE(*status)) {
4628         setMatchNotFound(strsrch);
4629         return FALSE;
4630     }
4631 
4632 #if BOYER_MOORE
4633     UCollationElements *coleiter        = strsrch->textIter;
4634     int32_t            *patternce       = strsrch->pattern.CE;
4635     int32_t             patterncelength = strsrch->pattern.CELength;
4636     int32_t             textoffset      = ucol_getOffset(coleiter);
4637     UBool               hasPatternAccents =
4638        strsrch->pattern.hasSuffixAccents || strsrch->pattern.hasPrefixAccents;
4639 
4640     // shifting it check for setting offset
4641     // if setOffset is called previously or there was no previous match, we
4642     // leave the offset as it is.
4643     if (strsrch->search->matchedIndex != USEARCH_DONE) {
4644         textoffset = strsrch->search->matchedIndex;
4645     }
4646 
4647     textoffset = reverseShift(strsrch, textoffset, UCOL_NULLORDER,
4648                               patterncelength);
4649     strsrch->canonicalPrefixAccents[0] = 0;
4650     strsrch->canonicalSuffixAccents[0] = 0;
4651 
4652     while (textoffset >= 0)
4653     {
4654         int32_t     patternceindex = 1;
4655         int32_t     targetce;
4656         UBool       found          = FALSE;
4657         int32_t     firstce        = UCOL_NULLORDER;
4658 
4659         setColEIterOffset(coleiter, textoffset);
4660         for (;;) {
4661             // finding the first pattern ce match, imagine composite
4662             // characters. for example: search for pattern \u0300 in text
4663             // \u00C0, we'll have to skip A first before we get to
4664             // \u0300 the grave accent
4665             targetce = ucol_next(coleiter, status);
4666             if (U_FAILURE(*status) || targetce == UCOL_NULLORDER) {
4667                 found = FALSE;
4668                 break;
4669             }
4670             targetce = getCE(strsrch, targetce);
4671             if (firstce == UCOL_NULLORDER || firstce == UCOL_IGNORABLE) {
4672                 firstce = targetce;
4673             }
4674 
4675             // TODO: #if BOYER_MOORE, replace with code using 32-bit version of compareCE64s
4676             if (targetce == patternce[0]) {
4677                 // the first ce can be a contraction
4678                 found = TRUE;
4679                 break;
4680             }
4681             if (!hasExpansion(coleiter)) {
4682                 // checking for accents in composite character
4683                 found = FALSE;
4684                 break;
4685             }
4686         }
4687 
4688         targetce = firstce;
4689 
4690         while (found && patternceindex < patterncelength) {
4691             targetce    = ucol_next(coleiter, status);
4692             if (U_FAILURE(*status) || targetce == UCOL_NULLORDER) {
4693                 found = FALSE;
4694                 break;
4695             }
4696             targetce = getCE(strsrch, targetce);
4697             if (targetce == UCOL_IGNORABLE) {
4698                 continue;
4699             }
4700 
4701             // TODO: #if BOYER_MOORE, replace with code using 32-bit version of compareCE64s
4702             found = found && targetce == patternce[patternceindex];
4703             patternceindex ++;
4704         }
4705 
4706         // initializing the rearranged accent array
4707         if (hasPatternAccents && !found) {
4708             strsrch->canonicalPrefixAccents[0] = 0;
4709             strsrch->canonicalSuffixAccents[0] = 0;
4710             if (U_FAILURE(*status)) {
4711                 break;
4712             }
4713             found = doPreviousCanonicalMatch(strsrch, textoffset, status);
4714         }
4715 
4716         if (!found) {
4717             if (U_FAILURE(*status)) {
4718                 break;
4719             }
4720             textoffset = reverseShift(strsrch, textoffset, targetce,
4721                                       patternceindex);
4722             patternceindex = 0;
4723             continue;
4724         }
4725 
4726         if (checkPreviousCanonicalMatch(strsrch, &textoffset, status)) {
4727             setColEIterOffset(coleiter, textoffset);
4728             return TRUE;
4729         }
4730     }
4731     setMatchNotFound(strsrch);
4732     return FALSE;
4733 #else
4734     int32_t textOffset = ucol_getOffset(strsrch->textIter);
4735     int32_t start = -1;
4736     int32_t end = -1;
4737 
4738     if (usearch_searchBackwards(strsrch, textOffset, &start, &end, status)) {
4739         strsrch->search->matchedIndex = start;
4740         strsrch->search->matchedLength = end - start;
4741         return TRUE;
4742     } else {
4743         setMatchNotFound(strsrch);
4744         return FALSE;
4745     }
4746 #endif
4747 }
4748 
4749 #endif /* #if !UCONFIG_NO_COLLATION */
4750