1 #ifndef lint
2 static char *rcsid = "$Id: mapselector.c,v 1.22 2003/01/20 13:39:58 m-kasahr Exp $";
3 #endif
4 
5 /*
6  * Copyright (c) 2001,2002 Japan Network Information Center.
7  * All rights reserved.
8  *
9  * By using this file, you agree to the terms and conditions set forth bellow.
10  *
11  * 			LICENSE TERMS AND CONDITIONS
12  *
13  * The following License Terms and Conditions apply, unless a different
14  * license is obtained from Japan Network Information Center ("JPNIC"),
15  * a Japanese association, Kokusai-Kougyou-Kanda Bldg 6F, 2-3-4 Uchi-Kanda,
16  * Chiyoda-ku, Tokyo 101-0047, Japan.
17  *
18  * 1. Use, Modification and Redistribution (including distribution of any
19  *    modified or derived work) in source and/or binary forms is permitted
20  *    under this License Terms and Conditions.
21  *
22  * 2. Redistribution of source code must retain the copyright notices as they
23  *    appear in each source code file, this License Terms and Conditions.
24  *
25  * 3. Redistribution in binary form must reproduce the Copyright Notice,
26  *    this License Terms and Conditions, in the documentation and/or other
27  *    materials provided with the distribution.  For the purposes of binary
28  *    distribution the "Copyright Notice" refers to the following language:
29  *    "Copyright (c) 2000-2002 Japan Network Information Center.  All rights reserved."
30  *
31  * 4. The name of JPNIC may not be used to endorse or promote products
32  *    derived from this Software without specific prior written approval of
33  *    JPNIC.
34  *
35  * 5. Disclaimer/Limitation of Liability: THIS SOFTWARE IS PROVIDED BY JPNIC
36  *    "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
37  *    LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A
38  *    PARTICULAR PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL JPNIC BE LIABLE
39  *    FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
40  *    CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
41  *    SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR
42  *    BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
43  *    WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR
44  *    OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
45  *    ADVISED OF THE POSSIBILITY OF SUCH DAMAGES.
46  */
47 
48 #include <config.h>
49 
50 #include <stddef.h>
51 #include <stdlib.h>
52 #include <string.h>
53 
54 #include <idn/assert.h>
55 #include <idn/logmacro.h>
56 #include <idn/result.h>
57 #include <idn/mapselector.h>
58 #include <idn/strhash.h>
59 #include <idn/debug.h>
60 #include <idn/util.h>
61 #include <idn/ucs4.h>
62 
63 struct idn_mapselector {
64 	idn__strhash_t maphash;
65 	int reference_count;
66 };
67 
68 /*
69  * Maximum length of a top level domain name. (e.g. `com', `jp', ...)
70  */
71 #define MAPSELECTOR_MAX_TLD_LENGTH	63
72 
73 static void string_ascii_tolower(char *string);
74 
75 
76 const unsigned long *
idn_mapselector_getnotld(void)77 idn_mapselector_getnotld(void) {
78 	static const unsigned long notld[] = {0x002d, 0x0000};  /* "-" */
79 	return (notld);
80 }
81 
82 const unsigned long *
idn_mapselector_getdefaulttld(void)83 idn_mapselector_getdefaulttld(void) {
84 	const static unsigned long defaulttld[] = {0x002e, 0x0000};  /* "." */
85 	return (defaulttld);
86 }
87 
88 idn_result_t
idn_mapselector_initialize(void)89 idn_mapselector_initialize(void) {
90 	idn_result_t r;
91 
92 	TRACE(("idn_mapselector_initialize()\n"));
93 
94 	r = idn_mapper_initialize();
95 
96 	TRACE(("idn_mapselector_initialize(): %s\n", idn_result_tostring(r)));
97 	return (r);
98 }
99 
100 idn_result_t
idn_mapselector_create(idn_mapselector_t * ctxp)101 idn_mapselector_create(idn_mapselector_t *ctxp) {
102 	idn_mapselector_t ctx = NULL;
103 	idn_result_t r;
104 
105 	assert(ctxp != NULL);
106 	TRACE(("idn_mapselector_create()\n"));
107 
108 	ctx = (idn_mapselector_t)malloc(sizeof(struct idn_mapselector));
109 	if (ctx == NULL) {
110 		r = idn_nomemory;
111 		goto ret;
112 	}
113 
114 	ctx->maphash = NULL;
115 	ctx->reference_count = 1;
116 
117 	r = idn__strhash_create(&(ctx->maphash));
118 	if (r != idn_success)
119 		goto ret;
120 
121 	*ctxp = ctx;
122 	r = idn_success;
123 
124 ret:
125 	if (r != idn_success) {
126 		if (ctx != NULL)
127 			free(ctx->maphash);
128 		free(ctx);
129 	}
130 	TRACE(("idn_mapselector_create(): %s\n", idn_result_tostring(r)));
131 	return (r);
132 }
133 
134 void
idn_mapselector_destroy(idn_mapselector_t ctx)135 idn_mapselector_destroy(idn_mapselector_t ctx) {
136 	assert(ctx != NULL);
137 
138 	TRACE(("idn_mapselector_destroy()\n"));
139 
140 	ctx->reference_count--;
141 	if (ctx->reference_count <= 0) {
142 		TRACE(("idn_mapselector_destroy(): "
143 		       "the object is destroyed\n"));
144 		idn__strhash_destroy(ctx->maphash,
145 			(idn__strhash_freeproc_t)&idn_mapper_destroy);
146 		free(ctx);
147 	} else {
148 		TRACE(("idn_mapselector_destroy(): "
149 		       "update reference count (%d->%d)\n",
150 		       ctx->reference_count + 1, ctx->reference_count));
151 	}
152 }
153 
154 void
idn_mapselector_incrref(idn_mapselector_t ctx)155 idn_mapselector_incrref(idn_mapselector_t ctx) {
156 	assert(ctx != NULL);
157 
158 	TRACE(("idn_mapselector_incrref()\n"));
159 	TRACE(("idn_mapselector_incrref: update reference count (%d->%d)\n",
160 		ctx->reference_count, ctx->reference_count + 1));
161 
162 	ctx->reference_count++;
163 }
164 
165 idn_result_t
idn_mapselector_add(idn_mapselector_t ctx,const char * tld,const char * name)166 idn_mapselector_add(idn_mapselector_t ctx, const char *tld, const char *name) {
167 	idn_result_t r;
168 	idn_mapper_t mapper;
169 	char hash_key[MAPSELECTOR_MAX_TLD_LENGTH + 1];
170 
171 	assert(ctx != NULL && tld != NULL);
172 
173 	TRACE(("idn_mapselector_add(tld=%s, name=%s)\n", tld, name));
174 
175 	if (!(tld[0] == '.' && tld[1] == '\0')) {
176 		if (tld[0] == '.')
177 			tld++;
178 		if (strchr(tld, '.') != NULL) {
179 			ERROR(("idn_mapselector_add: "
180 			       "invalid TLD \"%-.100s\"\n", tld));
181 			r = idn_invalid_name;
182 			goto ret;
183 		}
184 	}
185 	if (strlen(tld) > MAPSELECTOR_MAX_TLD_LENGTH) {
186 		ERROR(("idn_mapselector_add: "
187 		       "too long TLD \"%-.100s\"\n", tld));
188 		r = idn_invalid_name;
189 		goto ret;
190 	}
191 	strcpy(hash_key, tld);
192 	string_ascii_tolower(hash_key);
193 
194 	if (idn__strhash_get(ctx->maphash, hash_key, (void **)&mapper)
195 		!= idn_success) {
196 		r = idn_mapper_create(&mapper);
197 		if (r != idn_success)
198 			goto ret;
199 
200 		r = idn__strhash_put(ctx->maphash, hash_key, mapper);
201 		if (r != idn_success)
202 			goto ret;
203 	}
204 
205 	r = idn_mapper_add(mapper, name);
206 ret:
207 	TRACE(("idn_mapselector_add(): %s\n", idn_result_tostring(r)));
208 	return (r);
209 }
210 
211 idn_result_t
idn_mapselector_addall(idn_mapselector_t ctx,const char * tld,const char ** scheme_names,int nschemes)212 idn_mapselector_addall(idn_mapselector_t ctx, const char *tld,
213 		       const char **scheme_names, int nschemes) {
214 	idn_result_t r;
215 	int i;
216 
217 	assert(ctx != NULL && tld != NULL && scheme_names != NULL);
218 
219 	TRACE(("idn_mapselector_addall(tld=%s, nschemes=%d)\n",
220 	      tld, nschemes));
221 
222 	for (i = 0; i < nschemes; i++) {
223 		r = idn_mapselector_add(ctx, tld, (const char *)*scheme_names);
224 		if (r != idn_success)
225 			goto ret;
226 		scheme_names++;
227 	}
228 
229 	r = idn_success;
230 ret:
231 	TRACE(("idn_mapselector_addall(): %s\n", idn_result_tostring(r)));
232 	return (r);
233 }
234 
235 idn_mapper_t
idn_mapselector_mapper(idn_mapselector_t ctx,const char * tld)236 idn_mapselector_mapper(idn_mapselector_t ctx, const char *tld) {
237 	idn_result_t r;
238 	idn_mapper_t mapper;
239 	char hash_key[MAPSELECTOR_MAX_TLD_LENGTH + 1];
240 
241 	assert(ctx != NULL && tld != NULL);
242 
243 	TRACE(("idn_mapselector_mapper(tld=%s)\n", tld));
244 
245 	if (!(tld[0] == '.' && tld[1] == '\0')) {
246 		if (tld[0] == '.')
247 			tld++;
248 		if (strchr(tld, '.') != NULL)
249 			return (NULL);
250 	}
251 	if (strlen(tld) > MAPSELECTOR_MAX_TLD_LENGTH)
252 		return (NULL);
253 	strcpy(hash_key, tld);
254 	string_ascii_tolower(hash_key);
255 
256 	mapper = NULL;
257 	r = idn__strhash_get(ctx->maphash, hash_key, (void **)&mapper);
258 	if (r != idn_success)
259 		return (NULL);
260 
261 	idn_mapper_incrref(mapper);
262 
263 	return (mapper);
264 }
265 
266 idn_result_t
idn_mapselector_map(idn_mapselector_t ctx,const unsigned long * from,const char * tld,unsigned long * to,size_t tolen)267 idn_mapselector_map(idn_mapselector_t ctx, const unsigned long *from,
268 		    const char *tld, unsigned long *to, size_t tolen) {
269 	idn_result_t r;
270 	idn_mapper_t mapper = NULL;
271 	char hash_key[MAPSELECTOR_MAX_TLD_LENGTH + 1];
272 	size_t fromlen;
273 
274 	assert(ctx != NULL && from != NULL && to != NULL);
275 
276 	TRACE(("idn_mapselector_map(from=\"%s\", tld=\"%s\", tolen=%d)\n",
277 	       idn__debug_ucs4xstring(from, 50), idn__debug_xstring(tld, 50),
278 	       (int)tolen));
279 
280 	if (!(tld[0] == '.' && tld[1] == '\0')) {
281 		if (tld[0] == '.')
282 			tld++;
283 		if (strchr(tld, '.') != NULL) {
284 			r = idn_invalid_name;
285 			goto ret;
286 		}
287 	}
288 	if (strlen(tld) > MAPSELECTOR_MAX_TLD_LENGTH) {
289 		r = idn_invalid_name;
290 		goto ret;
291 	}
292 	strcpy(hash_key, tld);
293 	string_ascii_tolower(hash_key);
294 
295 	fromlen = idn_ucs4_strlen(from);
296 
297 	/*
298 	 * Get mapper for the TLD.
299 	 */
300 	if (idn__strhash_get(ctx->maphash, hash_key, (void **)&mapper)
301 	    != idn_success) {
302 		strcpy(hash_key, IDN_MAPSELECTOR_DEFAULTTLD);
303 		idn__strhash_get(ctx->maphash, hash_key, (void **)&mapper);
304 	}
305 
306 	/*
307 	 * Map.
308 	 * If default mapper has not been registered, copy the string.
309 	 */
310 	if (mapper == NULL) {
311 		TRACE(("idn_mapselector_map(): no mapper\n"));
312 		if (fromlen + 1 > tolen) {
313 			r = idn_buffer_overflow;
314 			goto ret;
315 		}
316 		memcpy(to, from, (fromlen + 1) * sizeof(*from));
317 	} else {
318 		TRACE(("idn_mapselector_map(): tld=%s\n", tld));
319 		r = idn_mapper_map(mapper, from, to, tolen);
320 		if (r != idn_success)
321 			goto ret;
322 	}
323 
324 	r = idn_success;
325 ret:
326 	if (r == idn_success) {
327 		TRACE(("idn_mapselector_map(): succcess (to=\"%s\")\n",
328 		       idn__debug_ucs4xstring(to, 50)));
329 	} else {
330 		TRACE(("idn_mapselector_map(): %s\n", idn_result_tostring(r)));
331 	}
332 	return (r);
333 }
334 
335 idn_result_t
idn_mapselector_map2(idn_mapselector_t ctx,const unsigned long * from,const unsigned long * tld,unsigned long * to,size_t tolen)336 idn_mapselector_map2(idn_mapselector_t ctx, const unsigned long *from,
337 		     const unsigned long *tld, unsigned long *to,
338 		     size_t tolen) {
339 	char tld_utf8[MAPSELECTOR_MAX_TLD_LENGTH + 1];
340 	idn_result_t r;
341 
342 	assert(ctx != NULL && from != NULL && to != NULL);
343 
344 	TRACE(("idn_mapselector_map2(from=\"%s\", tld=\"%s\")\n",
345 	       idn__debug_ucs4xstring(from, 50),
346 	       idn__debug_ucs4xstring(tld, 50)));
347 
348 	r = idn_ucs4_ucs4toutf8(tld, tld_utf8, sizeof(tld_utf8));
349 	if (r == idn_buffer_overflow) {
350 		r = idn_invalid_name;
351 		goto ret;
352 	} else if (r != idn_success) {
353 		goto ret;
354 	}
355 
356 	r = idn_mapselector_map(ctx, from, tld_utf8, to, tolen);
357 ret:
358 	if (r == idn_success) {
359 		TRACE(("idn_mapselector_map2(): success (to=\"%s\")\n",
360 		       idn__debug_ucs4xstring(to, 50)));
361 	} else {
362 	    TRACE(("idn_mapselector_map2(): %s\n", idn_result_tostring(r)));
363 	}
364 	return (r);
365 }
366 
367 static void
string_ascii_tolower(char * string)368 string_ascii_tolower(char *string) {
369 	unsigned char *p;
370 
371 	for (p = (unsigned char *) string; *p != '\0'; p++) {
372 		if ('A' <= *p && *p <= 'Z')
373 			*p = *p - 'A' + 'a';
374 	}
375 }
376