regis.c 3.4 KB
Newer Older
1 2 3
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
B
Bruce Momjian 已提交
4

5
#include "regis.h"
6
#include "ts_locale.h"
7 8
#include "common.h"

9
bool
B
Bruce Momjian 已提交
10 11
RS_isRegis(const char *str)
{
T
Tom Lane 已提交
12 13 14 15 16 17 18
	while (str && *str)
	{
		if (t_isalpha(str) ||
			t_iseq(str, '[') ||
			t_iseq(str,']') ||
			t_iseq(str, '^'))
			str += pg_mblen(str);
19
		else
20
			return false;
T
Tom Lane 已提交
21
	}
22
	return true;
23 24
}

B
Bruce Momjian 已提交
25
#define RS_IN_ONEOF 1
26 27 28 29
#define RS_IN_ONEOF_IN	2
#define RS_IN_NONEOF	3
#define RS_IN_WAIT	4

B
Bruce Momjian 已提交
30 31 32 33 34 35
static RegisNode *
newRegisNode(RegisNode * prev, int len)
{
	RegisNode  *ptr;

	ptr = (RegisNode *) malloc(RNHDRSZ + len + 1);
36
	if (!ptr)
B
Bruce Momjian 已提交
37 38
		ts_error(ERROR, "No memory");
	memset(ptr, 0, RNHDRSZ + len + 1);
39
	if (prev)
B
Bruce Momjian 已提交
40
		prev->next = ptr;
41 42 43
	return ptr;
}

44 45
void
RS_compile(Regis * r, bool issuffix, char *str)
B
Bruce Momjian 已提交
46
{
47
	int			len = strlen(str);
B
Bruce Momjian 已提交
48
	int			state = RS_IN_WAIT;
49
	char			*c = (char*)str;
B
Bruce Momjian 已提交
50 51 52
	RegisNode  *ptr = NULL;

	memset(r, 0, sizeof(Regis));
53 54
	r->issuffix = (issuffix) ? 1 : 0;

55
	while(*c)
B
Bruce Momjian 已提交
56 57 58
	{
		if (state == RS_IN_WAIT)
		{
59
			if (t_isalpha(c))
B
Bruce Momjian 已提交
60 61 62
			{
				if (ptr)
					ptr = newRegisNode(ptr, len);
63
				else
B
Bruce Momjian 已提交
64
					ptr = r->node = newRegisNode(NULL, len);
65
				COPYCHAR(ptr->data, c);
66
				ptr->type = RSF_ONEOF;
67
				ptr->len = pg_mblen(c);
B
Bruce Momjian 已提交
68
			}
69
			else if (t_iseq(c,'['))
B
Bruce Momjian 已提交
70 71 72
			{
				if (ptr)
					ptr = newRegisNode(ptr, len);
73
				else
B
Bruce Momjian 已提交
74
					ptr = r->node = newRegisNode(NULL, len);
75
				ptr->type = RSF_ONEOF;
B
Bruce Momjian 已提交
76 77 78
				state = RS_IN_ONEOF;
			}
			else
79
				ts_error(ERROR, "Error in regis: %s", str );
B
Bruce Momjian 已提交
80 81 82
		}
		else if (state == RS_IN_ONEOF)
		{
83
			if (t_iseq(c,'^'))
B
Bruce Momjian 已提交
84
			{
85
				ptr->type = RSF_NONEOF;
B
Bruce Momjian 已提交
86 87
				state = RS_IN_NONEOF;
			}
88
			else if (t_isalpha(c))
B
Bruce Momjian 已提交
89
			{
90 91
				COPYCHAR(ptr->data, c);
				ptr->len = pg_mblen(c);
B
Bruce Momjian 已提交
92 93 94
				state = RS_IN_ONEOF_IN;
			}
			else
95
				ts_error(ERROR, "Error in regis: %s", str);
B
Bruce Momjian 已提交
96 97 98
		}
		else if (state == RS_IN_ONEOF_IN || state == RS_IN_NONEOF)
		{
99
			if (t_isalpha(c))
B
Bruce Momjian 已提交
100
			{
101 102
				COPYCHAR(ptr->data+ptr->len,  c);
				ptr->len+=pg_mblen(c);
B
Bruce Momjian 已提交
103
			}
104
			else if (t_iseq(c,']'))
B
Bruce Momjian 已提交
105 106
				state = RS_IN_WAIT;
			else
107
				ts_error(ERROR, "Error in regis: %s", str);
B
Bruce Momjian 已提交
108 109
		}
		else
110 111
			ts_error(ERROR, "Internal error in RS_compile: %d", state);
		c += pg_mblen(c);
112 113 114
	}

	ptr = r->node;
B
Bruce Momjian 已提交
115 116
	while (ptr)
	{
117
		r->nchar++;
B
Bruce Momjian 已提交
118
		ptr = ptr->next;
119 120 121
	}
}

B
Bruce Momjian 已提交
122 123 124 125 126
void
RS_free(Regis * r)
{
	RegisNode  *ptr = r->node,
			   *tmp;
127

B
Bruce Momjian 已提交
128 129 130
	while (ptr)
	{
		tmp = ptr->next;
131 132 133 134 135 136 137
		free(ptr);
		ptr = tmp;
	}

	r->node = NULL;
}

138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169
#ifdef TS_USE_WIDE
static bool
mb_strchr(char *str, char *c) {
	int clen = pg_mblen(c), plen,i;
	char 	*ptr =str;
	bool	res=false;

	clen = pg_mblen(c);
	while( *ptr && !res) {
		plen = pg_mblen(ptr);
		if ( plen == clen ) {
			i=plen;
			res = true;
			while(i--)
				if ( *(ptr+i) != *(c+i) ) {
					res = false;
					break; 
				}
		}
		
		ptr += plen;
	}	 

	return res;	
}
#else
#define mb_strchr(s,c)	( (strchr((s),*(c)) == NULL) ? false : true )
#endif


bool
RS_execute(Regis * r, char *str)
B
Bruce Momjian 已提交
170 171
{
	RegisNode  *ptr = r->node;
172 173
	char *c = str;
	int len=0;
174

175 176 177 178
	while(*c) {
		len++;
		c += pg_mblen(c);
	}	
179

B
Bruce Momjian 已提交
180
	if (len < r->nchar)
181 182
		return 0;

183 184 185 186 187 188 189
	c = str;
	if (r->issuffix) {
		len -= r->nchar;
		while(len-- > 0)
			c += pg_mblen(c);
	}

190

B
Bruce Momjian 已提交
191 192 193 194
	while (ptr)
	{
		switch (ptr->type)
		{
195
			case RSF_ONEOF:
196 197
				if ( mb_strchr((char *) ptr->data, c) != true )
					return false;
198 199
				break;
			case RSF_NONEOF:
200 201
				if ( mb_strchr((char *) ptr->data, c) == true )
					return false;
202 203
				break;
			default:
B
Bruce Momjian 已提交
204
				ts_error(ERROR, "RS_execute: Unknown type node: %d\n", ptr->type);
205
		}
B
Bruce Momjian 已提交
206
		ptr = ptr->next;
207
		c+=pg_mblen(c);
208 209
	}

210
	return true;
211
}