1 7957cbd9 2021-01-27 op /*
  2 7957cbd9 2021-01-27 op  * Copyright (c) 2021 Omar Polo <op@omarpolo.com>
  3 7957cbd9 2021-01-27 op  *
  4 7957cbd9 2021-01-27 op  * Permission to use, copy, modify, and distribute this software for any
  5 7957cbd9 2021-01-27 op  * purpose with or without fee is hereby granted, provided that the above
  6 7957cbd9 2021-01-27 op  * copyright notice and this permission notice appear in all copies.
  7 7957cbd9 2021-01-27 op  *
  8 7957cbd9 2021-01-27 op  * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
  9 7957cbd9 2021-01-27 op  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
 10 7957cbd9 2021-01-27 op  * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
 11 7957cbd9 2021-01-27 op  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
 12 7957cbd9 2021-01-27 op  * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
 13 7957cbd9 2021-01-27 op  * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
 14 7957cbd9 2021-01-27 op  * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
 15 7957cbd9 2021-01-27 op  */
 16 7957cbd9 2021-01-27 op 
 17 7957cbd9 2021-01-27 op #include <stddef.h>
 18 7957cbd9 2021-01-27 op #include <stdint.h>
 19 7957cbd9 2021-01-27 op #include <string.h>
 20 7957cbd9 2021-01-27 op 
 21 7957cbd9 2021-01-27 op #include "gmid.h"
 22 7957cbd9 2021-01-27 op 
 23 7957cbd9 2021-01-27 op #define BASE	36
 24 7957cbd9 2021-01-27 op #define TMIN	1
 25 7957cbd9 2021-01-27 op #define TMAX	26
 26 7957cbd9 2021-01-27 op #define SKEW	38
 27 7957cbd9 2021-01-27 op #define DAMP	700
 28 7957cbd9 2021-01-27 op #define IBIAS	72
 29 7957cbd9 2021-01-27 op #define IN	128
 30 7957cbd9 2021-01-27 op 
 31 7957cbd9 2021-01-27 op static int
 32 7957cbd9 2021-01-27 op adapt(int delta, int numpoints, int firsttime)
 33 7957cbd9 2021-01-27 op {
 34 7957cbd9 2021-01-27 op 	int k;
 35 7957cbd9 2021-01-27 op 
 36 7957cbd9 2021-01-27 op 	if (firsttime)
 37 7957cbd9 2021-01-27 op 		delta = delta / DAMP;
 38 7957cbd9 2021-01-27 op 	else
 39 7957cbd9 2021-01-27 op 		delta = delta / 2;
 40 7957cbd9 2021-01-27 op 
 41 7957cbd9 2021-01-27 op 	delta += (delta / numpoints);
 42 7957cbd9 2021-01-27 op 
 43 7957cbd9 2021-01-27 op 	k = 0;
 44 7957cbd9 2021-01-27 op 	while (delta > ((BASE - TMIN) * TMAX) / 2) {
 45 7957cbd9 2021-01-27 op 		delta = delta / (BASE - TMIN);
 46 7957cbd9 2021-01-27 op 		k += BASE;
 47 7957cbd9 2021-01-27 op 	}
 48 7957cbd9 2021-01-27 op 	return k + (((BASE - TMIN + 1) * delta) / (delta + SKEW));
 49 7957cbd9 2021-01-27 op }
 50 7957cbd9 2021-01-27 op 
 51 7957cbd9 2021-01-27 op static const char *
 52 7957cbd9 2021-01-27 op copy_until_delimiter(const char *s, char *out, size_t len)
 53 7957cbd9 2021-01-27 op {
 54 7957cbd9 2021-01-27 op 	char *end, *t;
 55 0cd66754 2021-01-27 op 	size_t l;
 56 7957cbd9 2021-01-27 op 
 57 7957cbd9 2021-01-27 op 	end = strchr(s, '\0');
 58 0cd66754 2021-01-27 op 	l = end - s;
 59 0cd66754 2021-01-27 op 	if (l > len)
 60 7957cbd9 2021-01-27 op 		return NULL;
 61 7957cbd9 2021-01-27 op 
 62 7957cbd9 2021-01-27 op 	for (t = end; t >= s; --t)
 63 7957cbd9 2021-01-27 op 		if (*t == '-')
 64 7957cbd9 2021-01-27 op 			break;
 65 7957cbd9 2021-01-27 op 
 66 7957cbd9 2021-01-27 op 	if (t < s)
 67 7957cbd9 2021-01-27 op 		t = end;
 68 7957cbd9 2021-01-27 op 
 69 7957cbd9 2021-01-27 op 	for (; s < t; ++s, ++out) {
 70 7957cbd9 2021-01-27 op 		if (*s > 'z')
 71 7957cbd9 2021-01-27 op 			return NULL;
 72 7957cbd9 2021-01-27 op 		*out = *s;
 73 7957cbd9 2021-01-27 op 	}
 74 7957cbd9 2021-01-27 op 
 75 7957cbd9 2021-01-27 op 	return s;
 76 7957cbd9 2021-01-27 op }
 77 7957cbd9 2021-01-27 op 
 78 7957cbd9 2021-01-27 op static unsigned int
 79 7957cbd9 2021-01-27 op digit_value(char c)
 80 7957cbd9 2021-01-27 op {
 81 7957cbd9 2021-01-27 op 	if ('A' <= c && c <= 'Z')
 82 7957cbd9 2021-01-27 op 		return c - 'A';
 83 7957cbd9 2021-01-27 op 
 84 7957cbd9 2021-01-27 op 	if ('a' <= c && c <= 'z')
 85 7957cbd9 2021-01-27 op 		return c - 'a';
 86 7957cbd9 2021-01-27 op 
 87 7957cbd9 2021-01-27 op 	if ('0' <= c && c <= '9')
 88 7957cbd9 2021-01-27 op 		return 26 + c - '0';
 89 7957cbd9 2021-01-27 op 
 90 7957cbd9 2021-01-27 op 	return c;
 91 7957cbd9 2021-01-27 op }
 92 7957cbd9 2021-01-27 op 
 93 7957cbd9 2021-01-27 op static int
 94 7957cbd9 2021-01-27 op insert(char *out, size_t len, int codepoint, size_t i)
 95 7957cbd9 2021-01-27 op {
 96 7957cbd9 2021-01-27 op 	int l;
 97 7957cbd9 2021-01-27 op 	char *t;
 98 7957cbd9 2021-01-27 op 
 99 7957cbd9 2021-01-27 op 	if (codepoint <= 0x7F)
100 7957cbd9 2021-01-27 op 		return 0;
101 7957cbd9 2021-01-27 op 	else if (codepoint <= 0x7FF)
102 7957cbd9 2021-01-27 op 		l = 2;
103 7957cbd9 2021-01-27 op 	else if (codepoint <= 0xFFFF)
104 7957cbd9 2021-01-27 op 		l = 3;
105 7957cbd9 2021-01-27 op 	else if (codepoint <= 0x10FFFF)
106 7957cbd9 2021-01-27 op 		l = 4;
107 7957cbd9 2021-01-27 op 	else
108 7957cbd9 2021-01-27 op 		return 0;
109 7957cbd9 2021-01-27 op 
110 7957cbd9 2021-01-27 op 	if ((t = utf8_nth(out, i)) == NULL)
111 7957cbd9 2021-01-27 op 		return 0;
112 7957cbd9 2021-01-27 op 	if (t + l >= out + len)
113 7957cbd9 2021-01-27 op 		return 0;
114 7957cbd9 2021-01-27 op 
115 7957cbd9 2021-01-27 op 	memmove(t + l, t, strlen(t));
116 7957cbd9 2021-01-27 op 
117 7957cbd9 2021-01-27 op 	switch (l) {
118 7957cbd9 2021-01-27 op 	case 2:
119 7957cbd9 2021-01-27 op 		t[1] = ( codepoint        & 0x3F) + 0x80;
120 7957cbd9 2021-01-27 op 		t[0] = ((codepoint >> 6)  & 0x1F) + 0xC0;
121 7957cbd9 2021-01-27 op 		break;
122 7957cbd9 2021-01-27 op 	case 3:
123 7957cbd9 2021-01-27 op 		t[2] = ( codepoint        & 0x3F) + 0x80;
124 7957cbd9 2021-01-27 op 		t[1] = ((codepoint >> 6)  & 0x3F) + 0x80;
125 7957cbd9 2021-01-27 op 		t[0] = ((codepoint >> 12) & 0x0F) + 0xE0;
126 7957cbd9 2021-01-27 op 		break;
127 7957cbd9 2021-01-27 op 	case 4:
128 7957cbd9 2021-01-27 op 		t[3] = ( codepoint        & 0x3F) + 0x80;
129 7957cbd9 2021-01-27 op 		t[2] = ((codepoint >> 6)  & 0x3F) + 0x80;
130 7957cbd9 2021-01-27 op 		t[1] = ((codepoint >> 12) & 0x3F) + 0x80;
131 7957cbd9 2021-01-27 op 		t[0] = ((codepoint >> 18) & 0x07) + 0xF0;
132 7957cbd9 2021-01-27 op 		break;
133 7957cbd9 2021-01-27 op 	}
134 7957cbd9 2021-01-27 op 	return 1;
135 7957cbd9 2021-01-27 op }
136 7957cbd9 2021-01-27 op 
137 7957cbd9 2021-01-27 op static int
138 7957cbd9 2021-01-27 op decode(const char *str, char *out, size_t len)
139 7957cbd9 2021-01-27 op {
140 7957cbd9 2021-01-27 op 	size_t i;
141 7957cbd9 2021-01-27 op 	uint32_t n;
142 7957cbd9 2021-01-27 op 	unsigned int oldi, bias, w, k, digit, t;
143 7957cbd9 2021-01-27 op 	unsigned int numpoints;
144 7957cbd9 2021-01-27 op 	const char *s;
145 7957cbd9 2021-01-27 op 
146 44ee1bac 2021-01-27 op 	if (!starts_with(str, "xn--")) {
147 7957cbd9 2021-01-27 op 		strncpy(out, str, len);
148 7957cbd9 2021-01-27 op 		return 1;
149 7957cbd9 2021-01-27 op 	}
150 7957cbd9 2021-01-27 op 
151 7957cbd9 2021-01-27 op 	/* skip the xn-- */
152 7957cbd9 2021-01-27 op 	str += 4;
153 7957cbd9 2021-01-27 op 
154 7957cbd9 2021-01-27 op 	if (strchr(str, '-') != NULL) {
155 7957cbd9 2021-01-27 op 		if ((s = copy_until_delimiter(str, out, len)) == NULL)
156 7957cbd9 2021-01-27 op 			return 0;
157 7957cbd9 2021-01-27 op 		if (*s == '-')
158 7957cbd9 2021-01-27 op 			s++;
159 7957cbd9 2021-01-27 op 	} else
160 7957cbd9 2021-01-27 op 		s = str;
161 7957cbd9 2021-01-27 op 
162 7957cbd9 2021-01-27 op 	numpoints = strlen(out);
163 7957cbd9 2021-01-27 op 
164 7957cbd9 2021-01-27 op 	n = IN;
165 7957cbd9 2021-01-27 op 	i = 0;
166 7957cbd9 2021-01-27 op 	bias = IBIAS;
167 7957cbd9 2021-01-27 op 
168 7957cbd9 2021-01-27 op 	while (*s != '\0') {
169 7957cbd9 2021-01-27 op 		oldi = i;
170 7957cbd9 2021-01-27 op 		w = 1;
171 7957cbd9 2021-01-27 op 
172 7957cbd9 2021-01-27 op 		for (k = BASE; ; k += BASE) {
173 7957cbd9 2021-01-27 op 			if (*s == '\0')
174 7957cbd9 2021-01-27 op 				return 0;
175 7957cbd9 2021-01-27 op 			/* fail eventually? */
176 7957cbd9 2021-01-27 op 			digit = digit_value(*s);
177 7957cbd9 2021-01-27 op 			s++;
178 7957cbd9 2021-01-27 op 
179 7957cbd9 2021-01-27 op 			/* fail on overflow */
180 7957cbd9 2021-01-27 op 			i += digit * w;
181 7957cbd9 2021-01-27 op 
182 7957cbd9 2021-01-27 op 			if (k <= bias)
183 7957cbd9 2021-01-27 op 				t = TMIN;
184 7957cbd9 2021-01-27 op 			else if (k >= bias + TMAX)
185 7957cbd9 2021-01-27 op 				t = TMAX;
186 7957cbd9 2021-01-27 op 			else
187 7957cbd9 2021-01-27 op 				t = k - bias;
188 7957cbd9 2021-01-27 op 
189 7957cbd9 2021-01-27 op 			if (digit < t)
190 7957cbd9 2021-01-27 op 				break;
191 7957cbd9 2021-01-27 op 			w *= (BASE - t);
192 7957cbd9 2021-01-27 op 		}
193 7957cbd9 2021-01-27 op 
194 7957cbd9 2021-01-27 op 		bias = adapt(i - oldi, numpoints+1, oldi == 0);
195 7957cbd9 2021-01-27 op 		n += i / (numpoints+1); /* fail on overflow */
196 7957cbd9 2021-01-27 op 		i = i % (numpoints+1);
197 7957cbd9 2021-01-27 op 
198 7957cbd9 2021-01-27 op 		if (!insert(out, len, n, i))
199 7957cbd9 2021-01-27 op 			return 0;
200 7957cbd9 2021-01-27 op 		numpoints++;
201 7957cbd9 2021-01-27 op 		++i;
202 7957cbd9 2021-01-27 op 	}
203 7957cbd9 2021-01-27 op 
204 7957cbd9 2021-01-27 op 	return 1;
205 7957cbd9 2021-01-27 op }
206 7957cbd9 2021-01-27 op 
207 0cd66754 2021-01-27 op static const char *
208 7957cbd9 2021-01-27 op end_of_component(const char *hostname)
209 7957cbd9 2021-01-27 op {
210 7957cbd9 2021-01-27 op 	for (; *hostname != '\0' && *hostname != '.'; ++hostname)
211 7957cbd9 2021-01-27 op 		;		/* nop */
212 7957cbd9 2021-01-27 op 	return hostname;
213 7957cbd9 2021-01-27 op }
214 7957cbd9 2021-01-27 op 
215 7957cbd9 2021-01-27 op int
216 7957cbd9 2021-01-27 op puny_decode(const char *hostname, char *out, size_t len)
217 7957cbd9 2021-01-27 op {
218 7957cbd9 2021-01-27 op 	char comp[DOMAIN_NAME_LEN];
219 7957cbd9 2021-01-27 op 	const char *s, *end;
220 7957cbd9 2021-01-27 op 	size_t l;
221 7957cbd9 2021-01-27 op 
222 7957cbd9 2021-01-27 op 	memset(out, 0, len);
223 44ee1bac 2021-01-27 op 	if (hostname == NULL)
224 44ee1bac 2021-01-27 op 		return 1;
225 7957cbd9 2021-01-27 op 
226 7957cbd9 2021-01-27 op 	s = hostname;
227 7957cbd9 2021-01-27 op 	for (;;) {
228 7957cbd9 2021-01-27 op 		end = end_of_component(s);
229 0cd66754 2021-01-27 op 		l = end - s;
230 0cd66754 2021-01-27 op 		if (l >= sizeof(comp))
231 7957cbd9 2021-01-27 op 			return 0;
232 7957cbd9 2021-01-27 op 
233 7957cbd9 2021-01-27 op 		memcpy(comp, s, end - s);
234 7957cbd9 2021-01-27 op 		comp[end - s] = '\0';
235 7957cbd9 2021-01-27 op 
236 7957cbd9 2021-01-27 op 		if (!decode(comp, out, len))
237 7957cbd9 2021-01-27 op 			return 0;
238 7957cbd9 2021-01-27 op 
239 7957cbd9 2021-01-27 op 		if (*end == '\0')
240 7957cbd9 2021-01-27 op 			return 1;
241 7957cbd9 2021-01-27 op 
242 7957cbd9 2021-01-27 op 		if (strlcat(out, ".", len) >= len)
243 7957cbd9 2021-01-27 op 			return 0;
244 7957cbd9 2021-01-27 op 
245 7957cbd9 2021-01-27 op 		l = strlen(out);
246 7957cbd9 2021-01-27 op 		if (l >= len)
247 7957cbd9 2021-01-27 op 			return 0;
248 7957cbd9 2021-01-27 op 		out += l;
249 7957cbd9 2021-01-27 op 		len -= l;
250 7957cbd9 2021-01-27 op 
251 7957cbd9 2021-01-27 op 		s = end+1;
252 7957cbd9 2021-01-27 op 	}
253 7957cbd9 2021-01-27 op }