punycode.c File Reference

#include <rz_types.h>
#include <rz_util.h>

Go to the source code of this file.

Macros
#define	BASE   36
#define	TMIN   1
#define	TMAX   26
#define	SKEW   38
#define	DAMP   700
#define	INITIAL_N   128
#define	INITIAL_BIAS   72

Functions
int	utf32len (ut32 *input)
ut8 *	utf32toutf8 (ut32 *input)
ut32 *	utf8toutf32 (const ut8 *input, int len)
ut32	adapt_bias (ut32 delta, unsigned n_points, int is_first)
char	encode_digit (int c)
static ut32	encode_var_int (const ut32 bias, const ut32 delta, char *dst)
static ut32	decode_digit (ut32 v)
RZ_API char *	rz_punycode_encode (const ut8 *src, int srclen, int *dstlen)
RZ_API char *	rz_punycode_decode (const char *src, int srclen, int *dstlen)

Macro Definition Documentation

BASE

#define BASE   36

Definition at line 7 of file punycode.c.

DAMP

#define DAMP   700

Definition at line 11 of file punycode.c.

INITIAL_BIAS

#define INITIAL_BIAS   72

Definition at line 13 of file punycode.c.

INITIAL_N

#define INITIAL_N   128

Definition at line 12 of file punycode.c.

SKEW

#define SKEW   38

Definition at line 10 of file punycode.c.

TMAX

#define TMAX   26

Definition at line 9 of file punycode.c.

TMIN

#define TMIN   1

Definition at line 8 of file punycode.c.

Function Documentation

adapt_bias()

ut32 adapt_bias	(	ut32	delta,
		unsigned	n_points,
		int	is_first
	)

Definition at line 112 of file punycode.c.

                                                             {
    ut32 k = 0;
    delta /= is_first ? DAMP : 2;
    delta += delta / n_points;
 
    while (delta > ((BASE - TMIN) * TMAX) / 2) {
        delta /= (BASE - TMIN);
        k += BASE;
    }
 
    return k + (((BASE - TMIN + 1) * delta) / (delta + SKEW));
}

References BASE, DAMP, delta, k, SKEW, TMAX, and TMIN.

Referenced by rz_punycode_decode(), and rz_punycode_encode().

decode_digit()

static ut32 decode_digit ( ut32  v )

static

Definition at line 163 of file punycode.c.

                                 {
    if (IS_DIGIT(v)) {
        return v - 22;
    }
    if (v >= 'a' && v <= 'z') {
        return v - 'a';
    }
    if (v >= 'A' && v <= 'Z') {
        return v - 'A';
    }
    return UT32_MAX;
}

References IS_DIGIT, UT32_MAX, and v.

Referenced by rz_punycode_decode().

encode_digit()

char encode_digit

(

int

c

)

Definition at line 125 of file punycode.c.

                         {
    //  assert (c >= 0 && c <= BASE - TMIN);
    if (c > 25) {
        return c + 22;
    }
    return c + 'a';
}

References c.

Referenced by encode_var_int().

encode_var_int()

static ut32 encode_var_int ( const ut32  bias, const ut32  delta, char *  dst  )

static

Definition at line 133 of file punycode.c.

                                                                         {
    ut32 i, k, q, t;
    i = 0;
    k = BASE;
    q = delta;
 
    while (true) {
        if (k <= bias) {
            t = TMIN;
        } else if (k >= bias + TMAX) {
            t = TMAX;
        } else {
            t = k - bias;
        }
 
        if (q < t) {
            break;
        }
 
        dst[i++] = encode_digit(t + (q - t) % (BASE - t));
 
        q = (q - t) / (BASE - t);
        k += BASE;
    }
 
    dst[i++] = encode_digit(q);
 
    return i;
}

References BASE, delta, dst, encode_digit(), i, k, TMAX, and TMIN.

Referenced by rz_punycode_encode().

rz_punycode_decode()

RZ_API char* rz_punycode_decode	(	const char *	src,
		int	srclen,
		int *	dstlen
	)

Definition at line 255 of file punycode.c.

                                                                          {
    const char *p;
    ut32 si, di;
    ut32 b, n, t, i, k, w;
    ut32 digit, org_i, bias;
    ut32 *dst = NULL;
    ut8 *finaldst = NULL;
 
    *dstlen = 0;
    if (!src || srclen < 1) {
        return NULL;
    }
 
    dst = calloc(2 * srclen + 10, 4);
    if (!dst) {
        return NULL;
    }
 
    for (si = 0; si < srclen; si++) {
        if (src[si] & 0x80) {
            free(dst);
            return NULL; /*just return it*/
        }
    }
 
    for (p = src + srclen - 1; p > src && *p != '-'; p--) {
        ;
    }
    b = p - src;
 
    di = b;
    for (i = 0; i < di; i++) {
        dst[i] = src[i];
    }
 
    i = 0;
    n = INITIAL_N;
    bias = INITIAL_BIAS;
 
    for (si = b + (b > 0); si < srclen; di++) {
        org_i = i;
 
        for (w = 1, k = BASE;; k += BASE) {
            digit = decode_digit(src[si++]);
 
            if (digit == UT32_MAX) {
                free(dst);
                return NULL;
            }
 
            if (digit > (UT32_MAX - i) / w) {
                free(dst);
                return NULL;
            }
 
            i += digit * w;
 
            if (k <= bias) {
                t = TMIN;
            } else if (k >= bias + TMAX) {
                t = TMAX;
            } else {
                t = k - bias;
            }
 
            if (digit < t) {
                break;
            }
 
            if (w > UT32_MAX / (BASE - t)) {
                free(dst);
                return NULL;
            }
 
            w *= BASE - t;
        }
 
        bias = adapt_bias(i - org_i, di + 1, org_i == 0);
 
        if (i / (di + 1) > UT32_MAX - n) {
            free(dst);
            return NULL;
        }
 
        n += i / (di + 1);
        i %= (di + 1);
 
        memmove(dst + i + 1, dst + i, (di - i) * sizeof(ut32));
        dst[i++] = n;
    }
 
    finaldst = utf32toutf8(dst);
    free(dst);
    if (finaldst) {
        *dstlen = strlen((const char *)finaldst);
    } else {
        eprintf("ERROR: finaldst is null\n");
        return NULL;
    }
    return (char *)finaldst;
}

References adapt_bias(), b, BASE, calloc(), decode_digit(), dst, eprintf, free(), i, INITIAL_BIAS, INITIAL_N, k, n, NULL, p, si, src, TMAX, TMIN, UT32_MAX, utf32toutf8(), and w.

Referenced by update().

rz_punycode_encode()

RZ_API char* rz_punycode_encode	(	const ut8 *	src,
		int	srclen,
		int *	dstlen
	)

Definition at line 176 of file punycode.c.

                                                                         {
    ut32 m, n;
    ut32 b, h;
    ut32 si, di;
    ut32 delta, bias;
    ut32 *actualsrc = NULL;
    ut32 len = 0;
    char *dst = NULL;
 
    *dstlen = 0;
 
    if (!src || srclen < 1) {
        return NULL;
    }
 
    actualsrc = utf8toutf32(src, srclen);
    if (!actualsrc) {
        return NULL;
    }
 
    len = utf32len(actualsrc);
 
    dst = calloc(2 * len + 10, 1);
    if (!dst) {
        free(actualsrc);
        return NULL;
    }
 
    for (si = 0, di = 0; si < len; si++) {
        if (actualsrc[si] < 128) {
            dst[di++] = actualsrc[si];
        }
    }
 
    b = h = di;
 
    if (di > 0) {
        dst[di++] = '-';
    }
 
    n = INITIAL_N;
    bias = INITIAL_BIAS;
 
    for (delta = 0; h < len; n++, delta++) {
        for (m = UT32_MAX, si = 0; si < len; si++) {
            if (actualsrc[si] >= n && actualsrc[si] < m) {
                m = actualsrc[si];
            }
        }
 
        if ((m - n) > (UT32_MAX - delta) / (h + 1)) {
            free(actualsrc);
            free(dst);
            return NULL;
        }
 
        delta += (m - n) * (h + 1);
        n = m;
 
        for (si = 0; si < len; si++) {
            if (actualsrc[si] < n) {
                if (++delta == 0) {
                    free(actualsrc);
                    free(dst);
                    return NULL;
                }
            } else if (actualsrc[si] == n) {
                di += encode_var_int(bias, delta, &dst[di]);
                bias = adapt_bias(delta, h + 1, h == b);
                delta = 0;
                h++;
            }
        }
    }
    *dstlen = di;
    free(actualsrc);
    return dst;
}

References adapt_bias(), b, calloc(), delta, dst, encode_var_int(), free(), h, INITIAL_BIAS, INITIAL_N, len, regress::m, n, NULL, si, src, UT32_MAX, utf32len(), and utf8toutf32().

Referenced by update().

utf32len()

int utf32len

(

ut32 *

input

)

Definition at line 15 of file punycode.c.

                          {
    int i = 0;
    while (*(input + i)) {
        i++;
    }
    return i;
}

References i, and input().

Referenced by rz_punycode_encode(), and utf32toutf8().

utf32toutf8()

ut8* utf32toutf8

(

ut32 *

input

)

Definition at line 23 of file punycode.c.

                              {
    if (!input) {
        eprintf("ERROR input is null\n");
        return NULL;
    }
 
    int i = 0;
    int j = 0;
    int len = utf32len(input);
    ut8 *result = calloc(4, len + 1);
    if (!result) {
        eprintf("ERROR: out of memory\n");
        return NULL;
    }
 
    for (i = 0; i < len; i++) {
        if (input[i] < 0x80) {
            result[j] = input[i];
            j++;
        } else if (input[i] < 0x800) {
            result[j + 1] = 0x80 | (input[i] & 0x3f);
            result[j] = 0xc0 | ((input[i] >> 6) & 0x1f);
            j += 2;
        } else if (input[i] < 0x10000) {
            result[j + 2] = 0x80 | (input[i] & 0x3f);
            result[j + 1] = 0x80 | ((input[i] >> 6) & 0x3f);
            result[j] = 0xe0 | ((input[i] >> 12) & 0xf);
            j += 3;
        } else if (input[i] < 0x200000) {
            result[j + 3] = 0x80 | (input[i] & 0x3f);
            result[j + 2] = 0x80 | ((input[i] >> 6) & 0x3f);
            result[j + 1] = 0x80 | ((input[i] >> 12) & 0x3f);
            result[j] = 0xf0 | ((input[i] >> 18) & 0x7);
            j += 4;
        } else {
            eprintf("ERROR in toutf8. Seems like input is invalid\n");
            free(result);
            return NULL;
        }
    }
 
    result[j] = 0;
    return result;
}

References calloc(), eprintf, free(), i, input(), len, NULL, and utf32len().

Referenced by rz_punycode_decode().

utf8toutf32()

ut32* utf8toutf32	(	const ut8 *	input,
		int	len
	)

Definition at line 68 of file punycode.c.

                                             {
    if (!input) {
        eprintf("ERROR input is null\n");
        return NULL;
    }
 
    int val = 0;
    ut32 *result = calloc(len + 1, 4);
 
    if (!result) {
        eprintf("ERROR: out of memory\n");
        return NULL;
    }
 
    for (int i = 0, j = 0; i < len; j++) {
        if (input[i] >> 7 == 0) {
            val = input[i];
            i += 1;
        } else if (input[i] >> 5 == 0x6 && (len - i) > 1) {
            val = (((input[i] & 0x1f) << 6) & 0xfc0) |
                (input[i + 1] & 0x3f);
            i += 2;
        } else if (input[i] >> 4 == 0xe && (len - i) > 2) {
            val = (((input[i] & 0xf) << 12) & 0xf000) |
                (((input[i + 1] & 0x3f) << 6) & 0xffc0) |
                (input[i + 2] & 0x3f);
            i += 3;
        } else if (input[i] >> 3 == 0x1e && (len - i) > 3) {
            val = (((input[i] & 0xf) << 18) & 0x1c0000) |
                (((input[i + 1] & 0x3f) << 12) & 0x1ff000) |
                (((input[i + 2] & 0x3f) << 6) & 0x1fffc0) |
                (input[i + 3] & 0x3f);
            i += 4;
        } else {
            eprintf("ERROR in toutf32. Seems like input is invalid.\n");
            free(result);
            return NULL;
        }
        result[j] = val;
    }
 
    return result;
}

References calloc(), eprintf, free(), i, input(), len, NULL, and val.

Referenced by rz_punycode_encode().