1 /*
2 * count.c
3 */
4 #include <stdlib.h>
5 #include <stdio.h>
6 #include <string.h>
7 #include "oniguruma.h"
8
9 #define ulen(enc, p) onigenc_str_bytelen_null(enc, (UChar* )p)
10
11 static int
test(OnigEncoding enc,OnigMatchParam * mp,char * in_pattern,char * in_str)12 test(OnigEncoding enc, OnigMatchParam* mp, char* in_pattern, char* in_str)
13 {
14 int r;
15 unsigned char *start, *range, *end;
16 regex_t* reg;
17 OnigErrorInfo einfo;
18 OnigRegion *region;
19 UChar* pattern;
20 UChar* str;
21
22 pattern = (UChar* )in_pattern;
23 str = (UChar* )in_str;
24
25 r = onig_new(®, pattern, pattern + ulen(enc, pattern),
26 ONIG_OPTION_DEFAULT, enc, ONIG_SYNTAX_DEFAULT, &einfo);
27 if (r != ONIG_NORMAL) {
28 char s[ONIG_MAX_ERROR_MESSAGE_LEN];
29 onig_error_code_to_str((UChar* )s, r, &einfo);
30 fprintf(stderr, "COMPILE ERROR: %d: %s\n", r, s);
31 return -1;
32 }
33
34 region = onig_region_new();
35
36 end = str + ulen(enc, str);
37 start = str;
38 range = end;
39 r = onig_search_with_param(reg, str, end, start, range, region,
40 ONIG_OPTION_NONE, mp);
41 if (r >= 0) {
42 int slot;
43 OnigValue val;
44 char* tag;
45 int tag_len;
46
47 fprintf(stdout, "match at %d\n", r);
48
49 show_count:
50 if (enc == ONIG_ENCODING_UTF16_BE) {
51 tag = "\000x\000\000";
52 }
53 else if (enc == ONIG_ENCODING_UTF16_LE) {
54 tag = "x\000\000\000";
55 }
56 else {
57 tag = "x";
58 }
59 tag_len = ulen(enc, tag);
60
61 slot = 0;
62 r = onig_get_callout_data_by_tag(reg, mp, (UChar* )tag, (UChar* )tag + tag_len,
63 slot, 0, &val);
64 if (r < ONIG_NORMAL) goto err;
65 else if (r > ONIG_NORMAL) {
66 fprintf(stdout, "COUNT[x]: NO DATA\n");
67 }
68 else {
69 fprintf(stdout, "COUNT[x]: %ld\n", val.l);
70 }
71 }
72 else if (r == ONIG_MISMATCH) {
73 fprintf(stdout, "search fail\n");
74 goto show_count;
75 }
76 else { /* error */
77 char s[ONIG_MAX_ERROR_MESSAGE_LEN];
78 err:
79 onig_error_code_to_str((UChar* )s, r);
80 fprintf(stdout, "SEARCH ERROR: %d: %s\n", r, s);
81 }
82
83 onig_region_free(region, 1 /* 1:free self, 0:free contents only */);
84 onig_free(reg);
85 return r;
86 }
87
main(int argc,char * argv[])88 extern int main(int argc, char* argv[])
89 {
90 int r;
91 OnigMatchParam* mp;
92 OnigEncoding encs[3];
93
94 encs[0] = ONIG_ENCODING_UTF8;
95 encs[1] = ONIG_ENCODING_UTF16_BE;
96 encs[2] = ONIG_ENCODING_UTF16_LE;
97
98 r = onig_initialize(encs, sizeof(encs)/sizeof(encs[0]));
99 if (r != ONIG_NORMAL) {
100 fprintf(stderr, "FAIL: onig_initialize(): %d\n", r);
101 return -1;
102 }
103
104 mp = onig_new_match_param();
105
106 test(encs[0], mp, "abc(.(*COUNT[x]))*(*FAIL)", "abcdefg");
107 test(encs[0], mp, "abc(.(*COUNT[_any_]))*(.(*COUNT[x]))*d", "abcdefg");
108 /* fail count */
109 test(encs[0], mp, "abc(.(*COUNT[x]{<}))*f", "abcdefg");
110 /* success count */
111 test(encs[0], mp, "abc(.(*COUNT[x]{X}))*f", "abcdefg");
112 /* passed count */
113 test(encs[0], mp, "abc(.(*COUNT[x]))*f", "abcdefg");
114 test(encs[0], mp, "a(.(*COUNT[x]))*z", "abcd\nabcdz");
115 /* total count */
116 test(encs[0], mp, "a(.(*TOTAL_COUNT[x]))*z", "abcd\nabcdz");
117
118 test(encs[1], mp, "\000a\000b\000c\000(\000.\000(\000*\000C\000O\000U\000N\000T\000[\000x\000]\000)\000)\000*\000(\000*\000F\000A\000I\000L\000)\000\000", "\000a\000b\000c\000d\000e\000f\000g\000\000");
119
120 test(encs[2], mp, "a\000b\000c\000(\000.\000(\000*\000C\000O\000U\000N\000T\000[\000x\000]\000)\000)\000*\000(\000*\000F\000A\000I\000L\000)\000\000\000", "a\000b\000c\000d\000e\000f\000g\000\000\000");
121
122 onig_free_match_param(mp);
123 onig_end();
124 return 0;
125 }
126