#ifndef lint
static char *rcsid = "$Id: mapselector.c,v 1.1 2003/06/04 00:25:56 marka Exp $";
#endif
#include <config.h>
#include <stddef.h>
#include <stdlib.h>
#include <string.h>
#include <idn/assert.h>
#include <idn/logmacro.h>
#include <idn/result.h>
#include <idn/mapselector.h>
#include <idn/strhash.h>
#include <idn/debug.h>
#include <idn/util.h>
#include <idn/ucs4.h>
struct idn_mapselector {
idn__strhash_t maphash;
int reference_count;
};
#define MAPSELECTOR_MAX_TLD_LENGTH 63
static void string_ascii_tolower(char *string);
const unsigned long *
idn_mapselector_getnotld(void) {
static const unsigned long notld[] = {0x002d, 0x0000};
return (notld);
}
const unsigned long *
idn_mapselector_getdefaulttld(void) {
const static unsigned long defaulttld[] = {0x002e, 0x0000};
return (defaulttld);
}
idn_result_t
idn_mapselector_initialize(void) {
idn_result_t r;
TRACE(("idn_mapselector_initialize()\n"));
r = idn_mapper_initialize();
TRACE(("idn_mapselector_initialize(): %s\n", idn_result_tostring(r)));
return (r);
}
idn_result_t
idn_mapselector_create(idn_mapselector_t *ctxp) {
idn_mapselector_t ctx = NULL;
idn_result_t r;
assert(ctxp != NULL);
TRACE(("idn_mapselector_create()\n"));
ctx = (idn_mapselector_t)malloc(sizeof(struct idn_mapselector));
if (ctx == NULL) {
r = idn_nomemory;
goto ret;
}
ctx->maphash = NULL;
ctx->reference_count = 1;
r = idn__strhash_create(&(ctx->maphash));
if (r != idn_success)
goto ret;
*ctxp = ctx;
r = idn_success;
ret:
if (r != idn_success) {
if (ctx != NULL)
free(ctx->maphash);
free(ctx);
}
TRACE(("idn_mapselector_create(): %s\n", idn_result_tostring(r)));
return (r);
}
void
idn_mapselector_destroy(idn_mapselector_t ctx) {
assert(ctx != NULL);
TRACE(("idn_mapselector_destroy()\n"));
ctx->reference_count--;
if (ctx->reference_count <= 0) {
TRACE(("idn_mapselector_destroy(): "
"the object is destroyed\n"));
idn__strhash_destroy(ctx->maphash,
(idn__strhash_freeproc_t)&idn_mapper_destroy);
free(ctx);
} else {
TRACE(("idn_mapselector_destroy(): "
"update reference count (%d->%d)\n",
ctx->reference_count + 1, ctx->reference_count));
}
}
void
idn_mapselector_incrref(idn_mapselector_t ctx) {
assert(ctx != NULL);
TRACE(("idn_mapselector_incrref()\n"));
TRACE(("idn_mapselector_incrref: update reference count (%d->%d)\n",
ctx->reference_count, ctx->reference_count + 1));
ctx->reference_count++;
}
idn_result_t
idn_mapselector_add(idn_mapselector_t ctx, const char *tld, const char *name) {
idn_result_t r;
idn_mapper_t mapper;
char hash_key[MAPSELECTOR_MAX_TLD_LENGTH + 1];
assert(ctx != NULL && tld != NULL);
TRACE(("idn_mapselector_add(tld=%s, name=%s)\n", tld, name));
if (!(tld[0] == '.' && tld[1] == '\0')) {
if (tld[0] == '.')
tld++;
if (strchr(tld, '.') != NULL) {
ERROR(("idn_mapselector_add: "
"invalid TLD \"%-.100s\"\n", tld));
r = idn_invalid_name;
goto ret;
}
}
if (strlen(tld) > MAPSELECTOR_MAX_TLD_LENGTH) {
ERROR(("idn_mapselector_add: "
"too long TLD \"%-.100s\"\n", tld));
r = idn_invalid_name;
goto ret;
}
strcpy(hash_key, tld);
string_ascii_tolower(hash_key);
if (idn__strhash_get(ctx->maphash, hash_key, (void **)&mapper)
!= idn_success) {
r = idn_mapper_create(&mapper);
if (r != idn_success)
goto ret;
r = idn__strhash_put(ctx->maphash, hash_key, mapper);
if (r != idn_success)
goto ret;
}
r = idn_mapper_add(mapper, name);
ret:
TRACE(("idn_mapselector_add(): %s\n", idn_result_tostring(r)));
return (r);
}
idn_result_t
idn_mapselector_addall(idn_mapselector_t ctx, const char *tld,
const char **scheme_names, int nschemes) {
idn_result_t r;
int i;
assert(ctx != NULL && tld != NULL && scheme_names != NULL);
TRACE(("idn_mapselector_addall(tld=%s, nschemes=%d)\n",
tld, nschemes));
for (i = 0; i < nschemes; i++) {
r = idn_mapselector_add(ctx, tld, (const char *)*scheme_names);
if (r != idn_success)
goto ret;
scheme_names++;
}
r = idn_success;
ret:
TRACE(("idn_mapselector_addall(): %s\n", idn_result_tostring(r)));
return (r);
}
idn_mapper_t
idn_mapselector_mapper(idn_mapselector_t ctx, const char *tld) {
idn_result_t r;
idn_mapper_t mapper;
char hash_key[MAPSELECTOR_MAX_TLD_LENGTH + 1];
assert(ctx != NULL && tld != NULL);
TRACE(("idn_mapselector_mapper(tld=%s)\n", tld));
if (!(tld[0] == '.' && tld[1] == '\0')) {
if (tld[0] == '.')
tld++;
if (strchr(tld, '.') != NULL)
return (NULL);
}
if (strlen(tld) > MAPSELECTOR_MAX_TLD_LENGTH)
return (NULL);
strcpy(hash_key, tld);
string_ascii_tolower(hash_key);
mapper = NULL;
r = idn__strhash_get(ctx->maphash, hash_key, (void **)&mapper);
if (r != idn_success)
return (NULL);
idn_mapper_incrref(mapper);
return (mapper);
}
idn_result_t
idn_mapselector_map(idn_mapselector_t ctx, const unsigned long *from,
const char *tld, unsigned long *to, size_t tolen) {
idn_result_t r;
idn_mapper_t mapper = NULL;
char hash_key[MAPSELECTOR_MAX_TLD_LENGTH + 1];
size_t fromlen;
assert(ctx != NULL && from != NULL && to != NULL);
TRACE(("idn_mapselector_map(from=\"%s\", tld=\"%s\", tolen=%d)\n",
idn__debug_ucs4xstring(from, 50), idn__debug_xstring(tld, 50),
(int)tolen));
if (!(tld[0] == '.' && tld[1] == '\0')) {
if (tld[0] == '.')
tld++;
if (strchr(tld, '.') != NULL) {
r = idn_invalid_name;
goto ret;
}
}
if (strlen(tld) > MAPSELECTOR_MAX_TLD_LENGTH) {
r = idn_invalid_name;
goto ret;
}
strcpy(hash_key, tld);
string_ascii_tolower(hash_key);
fromlen = idn_ucs4_strlen(from);
if (idn__strhash_get(ctx->maphash, hash_key, (void **)&mapper)
!= idn_success) {
strcpy(hash_key, IDN_MAPSELECTOR_DEFAULTTLD);
idn__strhash_get(ctx->maphash, hash_key, (void **)&mapper);
}
if (mapper == NULL) {
TRACE(("idn_mapselector_map(): no mapper\n"));
if (fromlen + 1 > tolen) {
r = idn_buffer_overflow;
goto ret;
}
memcpy(to, from, (fromlen + 1) * sizeof(*from));
} else {
TRACE(("idn_mapselector_map(): tld=%s\n", tld));
r = idn_mapper_map(mapper, from, to, tolen);
if (r != idn_success)
goto ret;
}
r = idn_success;
ret:
if (r == idn_success) {
TRACE(("idn_mapselector_map(): succcess (to=\"%s\")\n",
idn__debug_ucs4xstring(to, 50)));
} else {
TRACE(("idn_mapselector_map(): %s\n", idn_result_tostring(r)));
}
return (r);
}
idn_result_t
idn_mapselector_map2(idn_mapselector_t ctx, const unsigned long *from,
const unsigned long *tld, unsigned long *to,
size_t tolen) {
char tld_utf8[MAPSELECTOR_MAX_TLD_LENGTH + 1];
idn_result_t r;
assert(ctx != NULL && from != NULL && to != NULL);
TRACE(("idn_mapselector_map2(from=\"%s\", tld=\"%s\")\n",
idn__debug_ucs4xstring(from, 50),
idn__debug_ucs4xstring(tld, 50)));
r = idn_ucs4_ucs4toutf8(tld, tld_utf8, sizeof(tld_utf8));
if (r == idn_buffer_overflow) {
r = idn_invalid_name;
goto ret;
} else if (r != idn_success) {
goto ret;
}
r = idn_mapselector_map(ctx, from, tld_utf8, to, tolen);
ret:
if (r == idn_success) {
TRACE(("idn_mapselector_map2(): success (to=\"%s\")\n",
idn__debug_ucs4xstring(to, 50)));
} else {
TRACE(("idn_mapselector_map2(): %s\n", idn_result_tostring(r)));
}
return (r);
}
static void
string_ascii_tolower(char *string) {
unsigned char *p;
for (p = (unsigned char *) string; *p != '\0'; p++) {
if ('A' <= *p && *p <= 'Z')
*p = *p - 'A' + 'a';
}
}