xref: /openbsd/usr.sbin/unbound/sldns/parseutil.c (revision 0bdb4f62)
1fdfb4ba6Ssthen /*
2fdfb4ba6Ssthen  * parseutil.c - parse utilities for string and wire conversion
3fdfb4ba6Ssthen  *
4fdfb4ba6Ssthen  * (c) NLnet Labs, 2004-2006
5fdfb4ba6Ssthen  *
6fdfb4ba6Ssthen  * See the file LICENSE for the license
7fdfb4ba6Ssthen  */
8fdfb4ba6Ssthen /**
9fdfb4ba6Ssthen  * \file
10fdfb4ba6Ssthen  *
11fdfb4ba6Ssthen  * Utility functions for parsing, base32(DNS variant) and base64 encoding
12fdfb4ba6Ssthen  * and decoding, Hex, Time units, Escape codes.
13fdfb4ba6Ssthen  */
14fdfb4ba6Ssthen 
15fdfb4ba6Ssthen #include "config.h"
16fdfb4ba6Ssthen #include "sldns/parseutil.h"
17fdfb4ba6Ssthen #include <sys/time.h>
18fdfb4ba6Ssthen #include <time.h>
19fdfb4ba6Ssthen #include <ctype.h>
20fdfb4ba6Ssthen 
21fdfb4ba6Ssthen sldns_lookup_table *
sldns_lookup_by_name(sldns_lookup_table * table,const char * name)22fdfb4ba6Ssthen sldns_lookup_by_name(sldns_lookup_table *table, const char *name)
23fdfb4ba6Ssthen {
24fdfb4ba6Ssthen         while (table->name != NULL) {
25fdfb4ba6Ssthen                 if (strcasecmp(name, table->name) == 0)
26fdfb4ba6Ssthen                         return table;
27fdfb4ba6Ssthen                 table++;
28fdfb4ba6Ssthen         }
29fdfb4ba6Ssthen         return NULL;
30fdfb4ba6Ssthen }
31fdfb4ba6Ssthen 
32fdfb4ba6Ssthen sldns_lookup_table *
sldns_lookup_by_id(sldns_lookup_table * table,int id)33fdfb4ba6Ssthen sldns_lookup_by_id(sldns_lookup_table *table, int id)
34fdfb4ba6Ssthen {
35fdfb4ba6Ssthen         while (table->name != NULL) {
36fdfb4ba6Ssthen                 if (table->id == id)
37fdfb4ba6Ssthen                         return table;
38fdfb4ba6Ssthen                 table++;
39fdfb4ba6Ssthen         }
40fdfb4ba6Ssthen         return NULL;
41fdfb4ba6Ssthen }
42fdfb4ba6Ssthen 
43fdfb4ba6Ssthen /* Number of days per month (except for February in leap years). */
44fdfb4ba6Ssthen static const int mdays[] = {
45fdfb4ba6Ssthen 	31, 28, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31
46fdfb4ba6Ssthen };
47fdfb4ba6Ssthen 
48fdfb4ba6Ssthen #define LDNS_MOD(x,y) (((x) % (y) < 0) ? ((x) % (y) + (y)) : ((x) % (y)))
49fdfb4ba6Ssthen #define LDNS_DIV(x,y) (((x) % (y) < 0) ? ((x) / (y) -  1 ) : ((x) / (y)))
50fdfb4ba6Ssthen 
51fdfb4ba6Ssthen static int
is_leap_year(int year)52fdfb4ba6Ssthen is_leap_year(int year)
53fdfb4ba6Ssthen {
54fdfb4ba6Ssthen 	return LDNS_MOD(year,   4) == 0 && (LDNS_MOD(year, 100) != 0
55fdfb4ba6Ssthen 	    || LDNS_MOD(year, 400) == 0);
56fdfb4ba6Ssthen }
57fdfb4ba6Ssthen 
58fdfb4ba6Ssthen static int
leap_days(int y1,int y2)59fdfb4ba6Ssthen leap_days(int y1, int y2)
60fdfb4ba6Ssthen {
61fdfb4ba6Ssthen 	--y1;
62fdfb4ba6Ssthen 	--y2;
63fdfb4ba6Ssthen 	return (LDNS_DIV(y2,   4) - LDNS_DIV(y1,   4)) -
64fdfb4ba6Ssthen 	       (LDNS_DIV(y2, 100) - LDNS_DIV(y1, 100)) +
65fdfb4ba6Ssthen 	       (LDNS_DIV(y2, 400) - LDNS_DIV(y1, 400));
66fdfb4ba6Ssthen }
67fdfb4ba6Ssthen 
68fdfb4ba6Ssthen /*
69fdfb4ba6Ssthen  * Code adapted from Python 2.4.1 sources (Lib/calendar.py).
70fdfb4ba6Ssthen  */
71fdfb4ba6Ssthen time_t
sldns_mktime_from_utc(const struct tm * tm)72fdfb4ba6Ssthen sldns_mktime_from_utc(const struct tm *tm)
73fdfb4ba6Ssthen {
74fdfb4ba6Ssthen 	int year = 1900 + tm->tm_year;
75fdfb4ba6Ssthen 	time_t days = 365 * ((time_t) year - 1970) + leap_days(1970, year);
76fdfb4ba6Ssthen 	time_t hours;
77fdfb4ba6Ssthen 	time_t minutes;
78fdfb4ba6Ssthen 	time_t seconds;
79fdfb4ba6Ssthen 	int i;
80fdfb4ba6Ssthen 
81fdfb4ba6Ssthen 	for (i = 0; i < tm->tm_mon; ++i) {
82fdfb4ba6Ssthen 		days += mdays[i];
83fdfb4ba6Ssthen 	}
84fdfb4ba6Ssthen 	if (tm->tm_mon > 1 && is_leap_year(year)) {
85fdfb4ba6Ssthen 		++days;
86fdfb4ba6Ssthen 	}
87fdfb4ba6Ssthen 	days += tm->tm_mday - 1;
88fdfb4ba6Ssthen 
89fdfb4ba6Ssthen 	hours = days * 24 + tm->tm_hour;
90fdfb4ba6Ssthen 	minutes = hours * 60 + tm->tm_min;
91fdfb4ba6Ssthen 	seconds = minutes * 60 + tm->tm_sec;
92fdfb4ba6Ssthen 
93fdfb4ba6Ssthen 	return seconds;
94fdfb4ba6Ssthen }
95fdfb4ba6Ssthen 
96fdfb4ba6Ssthen #if SIZEOF_TIME_T <= 4
97fdfb4ba6Ssthen 
98fdfb4ba6Ssthen static void
sldns_year_and_yday_from_days_since_epoch(int64_t days,struct tm * result)99fdfb4ba6Ssthen sldns_year_and_yday_from_days_since_epoch(int64_t days, struct tm *result)
100fdfb4ba6Ssthen {
101fdfb4ba6Ssthen 	int year = 1970;
102fdfb4ba6Ssthen 	int new_year;
103fdfb4ba6Ssthen 
104fdfb4ba6Ssthen 	while (days < 0 || days >= (int64_t) (is_leap_year(year) ? 366 : 365)) {
105fdfb4ba6Ssthen 		new_year = year + (int) LDNS_DIV(days, 365);
106fdfb4ba6Ssthen 		days -= (new_year - year) * 365;
107fdfb4ba6Ssthen 		days -= leap_days(year, new_year);
108fdfb4ba6Ssthen 		year  = new_year;
109fdfb4ba6Ssthen 	}
110fdfb4ba6Ssthen 	result->tm_year = year;
111fdfb4ba6Ssthen 	result->tm_yday = (int) days;
112fdfb4ba6Ssthen }
113fdfb4ba6Ssthen 
114fdfb4ba6Ssthen /* Number of days per month in a leap year. */
115fdfb4ba6Ssthen static const int leap_year_mdays[] = {
116fdfb4ba6Ssthen 	31, 29, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31
117fdfb4ba6Ssthen };
118fdfb4ba6Ssthen 
119fdfb4ba6Ssthen static void
sldns_mon_and_mday_from_year_and_yday(struct tm * result)120fdfb4ba6Ssthen sldns_mon_and_mday_from_year_and_yday(struct tm *result)
121fdfb4ba6Ssthen {
122fdfb4ba6Ssthen 	int idays = result->tm_yday;
123fdfb4ba6Ssthen 	const int *mon_lengths = is_leap_year(result->tm_year) ?
124fdfb4ba6Ssthen 					leap_year_mdays : mdays;
125fdfb4ba6Ssthen 
126fdfb4ba6Ssthen 	result->tm_mon = 0;
127fdfb4ba6Ssthen 	while  (idays >= mon_lengths[result->tm_mon]) {
128fdfb4ba6Ssthen 		idays -= mon_lengths[result->tm_mon++];
129fdfb4ba6Ssthen 	}
130fdfb4ba6Ssthen 	result->tm_mday = idays + 1;
131fdfb4ba6Ssthen }
132fdfb4ba6Ssthen 
133fdfb4ba6Ssthen static void
sldns_wday_from_year_and_yday(struct tm * result)134fdfb4ba6Ssthen sldns_wday_from_year_and_yday(struct tm *result)
135fdfb4ba6Ssthen {
136fdfb4ba6Ssthen 	result->tm_wday = 4 /* 1-1-1970 was a thursday */
137fdfb4ba6Ssthen 			+ LDNS_MOD((result->tm_year - 1970), 7) * LDNS_MOD(365, 7)
138fdfb4ba6Ssthen 			+ leap_days(1970, result->tm_year)
139fdfb4ba6Ssthen 			+ result->tm_yday;
140fdfb4ba6Ssthen 	result->tm_wday = LDNS_MOD(result->tm_wday, 7);
141fdfb4ba6Ssthen 	if (result->tm_wday < 0) {
142fdfb4ba6Ssthen 		result->tm_wday += 7;
143fdfb4ba6Ssthen 	}
144fdfb4ba6Ssthen }
145fdfb4ba6Ssthen 
146fdfb4ba6Ssthen static struct tm *
sldns_gmtime64_r(int64_t clock,struct tm * result)147fdfb4ba6Ssthen sldns_gmtime64_r(int64_t clock, struct tm *result)
148fdfb4ba6Ssthen {
149fdfb4ba6Ssthen 	result->tm_isdst = 0;
150fdfb4ba6Ssthen 	result->tm_sec   = (int) LDNS_MOD(clock, 60);
151fdfb4ba6Ssthen 	clock            =       LDNS_DIV(clock, 60);
152fdfb4ba6Ssthen 	result->tm_min   = (int) LDNS_MOD(clock, 60);
153fdfb4ba6Ssthen 	clock            =       LDNS_DIV(clock, 60);
154fdfb4ba6Ssthen 	result->tm_hour  = (int) LDNS_MOD(clock, 24);
155fdfb4ba6Ssthen 	clock            =       LDNS_DIV(clock, 24);
156fdfb4ba6Ssthen 
157fdfb4ba6Ssthen 	sldns_year_and_yday_from_days_since_epoch(clock, result);
158fdfb4ba6Ssthen 	sldns_mon_and_mday_from_year_and_yday(result);
159fdfb4ba6Ssthen 	sldns_wday_from_year_and_yday(result);
160fdfb4ba6Ssthen 	result->tm_year -= 1900;
161fdfb4ba6Ssthen 
162fdfb4ba6Ssthen 	return result;
163fdfb4ba6Ssthen }
164fdfb4ba6Ssthen 
165fdfb4ba6Ssthen #endif /* SIZEOF_TIME_T <= 4 */
166fdfb4ba6Ssthen 
167fdfb4ba6Ssthen static int64_t
sldns_serial_arithmetics_time(int32_t time,time_t now)168bdfc4d55Sflorian sldns_serial_arithmetics_time(int32_t time, time_t now)
169fdfb4ba6Ssthen {
170a3167c07Ssthen 	int32_t offset = (int32_t)((uint32_t) time - (uint32_t) now);
171fdfb4ba6Ssthen 	return (int64_t) now + offset;
172fdfb4ba6Ssthen }
173fdfb4ba6Ssthen 
174fdfb4ba6Ssthen struct tm *
sldns_serial_arithmetics_gmtime_r(int32_t time,time_t now,struct tm * result)175bdfc4d55Sflorian sldns_serial_arithmetics_gmtime_r(int32_t time, time_t now, struct tm *result)
176fdfb4ba6Ssthen {
177fdfb4ba6Ssthen #if SIZEOF_TIME_T <= 4
178bdfc4d55Sflorian 	int64_t secs_since_epoch = sldns_serial_arithmetics_time(time, now);
179fdfb4ba6Ssthen 	return  sldns_gmtime64_r(secs_since_epoch, result);
180fdfb4ba6Ssthen #else
181bdfc4d55Sflorian 	time_t  secs_since_epoch = sldns_serial_arithmetics_time(time, now);
182fdfb4ba6Ssthen 	return  gmtime_r(&secs_since_epoch, result);
183fdfb4ba6Ssthen #endif
184fdfb4ba6Ssthen }
185fdfb4ba6Ssthen 
186fdfb4ba6Ssthen int
sldns_hexdigit_to_int(char ch)187fdfb4ba6Ssthen sldns_hexdigit_to_int(char ch)
188fdfb4ba6Ssthen {
189fdfb4ba6Ssthen 	switch (ch) {
190fdfb4ba6Ssthen 	case '0': return 0;
191fdfb4ba6Ssthen 	case '1': return 1;
192fdfb4ba6Ssthen 	case '2': return 2;
193fdfb4ba6Ssthen 	case '3': return 3;
194fdfb4ba6Ssthen 	case '4': return 4;
195fdfb4ba6Ssthen 	case '5': return 5;
196fdfb4ba6Ssthen 	case '6': return 6;
197fdfb4ba6Ssthen 	case '7': return 7;
198fdfb4ba6Ssthen 	case '8': return 8;
199fdfb4ba6Ssthen 	case '9': return 9;
200fdfb4ba6Ssthen 	case 'a': case 'A': return 10;
201fdfb4ba6Ssthen 	case 'b': case 'B': return 11;
202fdfb4ba6Ssthen 	case 'c': case 'C': return 12;
203fdfb4ba6Ssthen 	case 'd': case 'D': return 13;
204fdfb4ba6Ssthen 	case 'e': case 'E': return 14;
205fdfb4ba6Ssthen 	case 'f': case 'F': return 15;
206fdfb4ba6Ssthen 	default:
207fdfb4ba6Ssthen 		return -1;
208fdfb4ba6Ssthen 	}
209fdfb4ba6Ssthen }
210fdfb4ba6Ssthen 
211fdfb4ba6Ssthen uint32_t
sldns_str2period(const char * nptr,const char ** endptr,int * overflow)212*0bdb4f62Ssthen sldns_str2period(const char *nptr, const char **endptr, int* overflow)
213fdfb4ba6Ssthen {
214fdfb4ba6Ssthen 	int sign = 0;
215fdfb4ba6Ssthen 	uint32_t i = 0;
216fdfb4ba6Ssthen 	uint32_t seconds = 0;
217*0bdb4f62Ssthen 	const uint32_t maxint = 0xffffffff;
218*0bdb4f62Ssthen 	*overflow = 0;
219fdfb4ba6Ssthen 
220fdfb4ba6Ssthen 	for(*endptr = nptr; **endptr; (*endptr)++) {
221fdfb4ba6Ssthen 		switch (**endptr) {
222fdfb4ba6Ssthen 			case ' ':
223fdfb4ba6Ssthen 			case '\t':
224fdfb4ba6Ssthen 				break;
225fdfb4ba6Ssthen 			case '-':
226fdfb4ba6Ssthen 				if(sign == 0) {
227fdfb4ba6Ssthen 					sign = -1;
228fdfb4ba6Ssthen 				} else {
229fdfb4ba6Ssthen 					return seconds;
230fdfb4ba6Ssthen 				}
231fdfb4ba6Ssthen 				break;
232fdfb4ba6Ssthen 			case '+':
233fdfb4ba6Ssthen 				if(sign == 0) {
234fdfb4ba6Ssthen 					sign = 1;
235fdfb4ba6Ssthen 				} else {
236fdfb4ba6Ssthen 					return seconds;
237fdfb4ba6Ssthen 				}
238fdfb4ba6Ssthen 				break;
239fdfb4ba6Ssthen 			case 's':
240fdfb4ba6Ssthen 			case 'S':
241*0bdb4f62Ssthen 				if(seconds > maxint-i) {
242*0bdb4f62Ssthen 					*overflow = 1;
243*0bdb4f62Ssthen 					return 0;
244*0bdb4f62Ssthen 				}
245fdfb4ba6Ssthen 				seconds += i;
246fdfb4ba6Ssthen 				i = 0;
247fdfb4ba6Ssthen 				break;
248fdfb4ba6Ssthen 			case 'm':
249fdfb4ba6Ssthen 			case 'M':
250*0bdb4f62Ssthen 				if(i > maxint/60 || seconds > maxint-(i*60)) {
251*0bdb4f62Ssthen 					*overflow = 1;
252*0bdb4f62Ssthen 					return 0;
253*0bdb4f62Ssthen 				}
254fdfb4ba6Ssthen 				seconds += i * 60;
255fdfb4ba6Ssthen 				i = 0;
256fdfb4ba6Ssthen 				break;
257fdfb4ba6Ssthen 			case 'h':
258fdfb4ba6Ssthen 			case 'H':
259*0bdb4f62Ssthen 				if(i > maxint/(60*60) || seconds > maxint-(i*60*60)) {
260*0bdb4f62Ssthen 					*overflow = 1;
261*0bdb4f62Ssthen 					return 0;
262*0bdb4f62Ssthen 				}
263fdfb4ba6Ssthen 				seconds += i * 60 * 60;
264fdfb4ba6Ssthen 				i = 0;
265fdfb4ba6Ssthen 				break;
266fdfb4ba6Ssthen 			case 'd':
267fdfb4ba6Ssthen 			case 'D':
268*0bdb4f62Ssthen 				if(i > maxint/(60*60*24) || seconds > maxint-(i*60*60*24)) {
269*0bdb4f62Ssthen 					*overflow = 1;
270*0bdb4f62Ssthen 					return 0;
271*0bdb4f62Ssthen 				}
272fdfb4ba6Ssthen 				seconds += i * 60 * 60 * 24;
273fdfb4ba6Ssthen 				i = 0;
274fdfb4ba6Ssthen 				break;
275fdfb4ba6Ssthen 			case 'w':
276fdfb4ba6Ssthen 			case 'W':
277*0bdb4f62Ssthen 				if(i > maxint/(60*60*24*7) || seconds > maxint-(i*60*60*24*7)) {
278*0bdb4f62Ssthen 					*overflow = 1;
279*0bdb4f62Ssthen 					return 0;
280*0bdb4f62Ssthen 				}
281fdfb4ba6Ssthen 				seconds += i * 60 * 60 * 24 * 7;
282fdfb4ba6Ssthen 				i = 0;
283fdfb4ba6Ssthen 				break;
284fdfb4ba6Ssthen 			case '0':
285fdfb4ba6Ssthen 			case '1':
286fdfb4ba6Ssthen 			case '2':
287fdfb4ba6Ssthen 			case '3':
288fdfb4ba6Ssthen 			case '4':
289fdfb4ba6Ssthen 			case '5':
290fdfb4ba6Ssthen 			case '6':
291fdfb4ba6Ssthen 			case '7':
292fdfb4ba6Ssthen 			case '8':
293fdfb4ba6Ssthen 			case '9':
294*0bdb4f62Ssthen 				if(i > maxint/10 || i*10 > maxint - (**endptr - '0')) {
295*0bdb4f62Ssthen 					*overflow = 1;
296*0bdb4f62Ssthen 					return 0;
297*0bdb4f62Ssthen 				}
298fdfb4ba6Ssthen 				i *= 10;
299fdfb4ba6Ssthen 				i += (**endptr - '0');
300fdfb4ba6Ssthen 				break;
301fdfb4ba6Ssthen 			default:
302*0bdb4f62Ssthen 				if(seconds > maxint-i) {
303*0bdb4f62Ssthen 					*overflow = 1;
304*0bdb4f62Ssthen 					return 0;
305*0bdb4f62Ssthen 				}
306fdfb4ba6Ssthen 				seconds += i;
307fdfb4ba6Ssthen 				/* disregard signedness */
308fdfb4ba6Ssthen 				return seconds;
309fdfb4ba6Ssthen 		}
310fdfb4ba6Ssthen 	}
311*0bdb4f62Ssthen 	if(seconds > maxint-i) {
312*0bdb4f62Ssthen 		*overflow = 1;
313*0bdb4f62Ssthen 		return 0;
314*0bdb4f62Ssthen 	}
315fdfb4ba6Ssthen 	seconds += i;
316fdfb4ba6Ssthen 	/* disregard signedness */
317fdfb4ba6Ssthen 	return seconds;
318fdfb4ba6Ssthen }
319fdfb4ba6Ssthen 
320fdfb4ba6Ssthen int
sldns_parse_escape(uint8_t * ch_p,const char ** str_p)321fdfb4ba6Ssthen sldns_parse_escape(uint8_t *ch_p, const char** str_p)
322fdfb4ba6Ssthen {
323fdfb4ba6Ssthen 	uint16_t val;
324fdfb4ba6Ssthen 
325fdfb4ba6Ssthen 	if ((*str_p)[0] && isdigit((unsigned char)(*str_p)[0]) &&
326fdfb4ba6Ssthen 	    (*str_p)[1] && isdigit((unsigned char)(*str_p)[1]) &&
327fdfb4ba6Ssthen 	    (*str_p)[2] && isdigit((unsigned char)(*str_p)[2])) {
328fdfb4ba6Ssthen 
329fdfb4ba6Ssthen 		val = (uint16_t)(((*str_p)[0] - '0') * 100 +
330fdfb4ba6Ssthen 				 ((*str_p)[1] - '0') *  10 +
331fdfb4ba6Ssthen 				 ((*str_p)[2] - '0'));
332fdfb4ba6Ssthen 
333fdfb4ba6Ssthen 		if (val > 255) {
334fdfb4ba6Ssthen 			goto error;
335fdfb4ba6Ssthen 		}
336fdfb4ba6Ssthen 		*ch_p = (uint8_t)val;
337fdfb4ba6Ssthen 		*str_p += 3;
338fdfb4ba6Ssthen 		return 1;
339fdfb4ba6Ssthen 
340fdfb4ba6Ssthen 	} else if ((*str_p)[0] && !isdigit((unsigned char)(*str_p)[0])) {
341fdfb4ba6Ssthen 
342fdfb4ba6Ssthen 		*ch_p = (uint8_t)*(*str_p)++;
343fdfb4ba6Ssthen 		return 1;
344fdfb4ba6Ssthen 	}
345fdfb4ba6Ssthen error:
346fdfb4ba6Ssthen 	*str_p = NULL;
347fdfb4ba6Ssthen 	return 0; /* LDNS_WIREPARSE_ERR_SYNTAX_BAD_ESCAPE */
348fdfb4ba6Ssthen }
349fdfb4ba6Ssthen 
350fdfb4ba6Ssthen /** parse one character, with escape codes */
351fdfb4ba6Ssthen int
sldns_parse_char(uint8_t * ch_p,const char ** str_p)352fdfb4ba6Ssthen sldns_parse_char(uint8_t *ch_p, const char** str_p)
353fdfb4ba6Ssthen {
354fdfb4ba6Ssthen 	switch (**str_p) {
355fdfb4ba6Ssthen 
356fdfb4ba6Ssthen 	case '\0':	return 0;
357fdfb4ba6Ssthen 
358fdfb4ba6Ssthen 	case '\\':	*str_p += 1;
359fdfb4ba6Ssthen 			return sldns_parse_escape(ch_p, str_p);
360fdfb4ba6Ssthen 
361fdfb4ba6Ssthen 	default:	*ch_p = (uint8_t)*(*str_p)++;
362fdfb4ba6Ssthen 			return 1;
363fdfb4ba6Ssthen 	}
364fdfb4ba6Ssthen }
365fdfb4ba6Ssthen 
sldns_b32_ntop_calculate_size(size_t src_data_length)366fdfb4ba6Ssthen size_t sldns_b32_ntop_calculate_size(size_t src_data_length)
367fdfb4ba6Ssthen {
368fdfb4ba6Ssthen 	return src_data_length == 0 ? 0 : ((src_data_length - 1) / 5 + 1) * 8;
369fdfb4ba6Ssthen }
370fdfb4ba6Ssthen 
sldns_b32_ntop_calculate_size_no_padding(size_t src_data_length)371fdfb4ba6Ssthen size_t sldns_b32_ntop_calculate_size_no_padding(size_t src_data_length)
372fdfb4ba6Ssthen {
373fdfb4ba6Ssthen 	return ((src_data_length + 3) * 8 / 5) - 4;
374fdfb4ba6Ssthen }
375fdfb4ba6Ssthen 
376fdfb4ba6Ssthen static int
sldns_b32_ntop_base(const uint8_t * src,size_t src_sz,char * dst,size_t dst_sz,int extended_hex,int add_padding)377fdfb4ba6Ssthen sldns_b32_ntop_base(const uint8_t* src, size_t src_sz, char* dst, size_t dst_sz,
378fdfb4ba6Ssthen 	int extended_hex, int add_padding)
379fdfb4ba6Ssthen {
380fdfb4ba6Ssthen 	size_t ret_sz;
381fdfb4ba6Ssthen 	const char* b32 = extended_hex ?  "0123456789abcdefghijklmnopqrstuv"
382fdfb4ba6Ssthen 					: "abcdefghijklmnopqrstuvwxyz234567";
383fdfb4ba6Ssthen 
384fdfb4ba6Ssthen 	size_t c = 0; /* c is used to carry partial base32 character over
385fdfb4ba6Ssthen 		       * byte boundaries for sizes with a remainder.
386fdfb4ba6Ssthen 		       * (i.e. src_sz % 5 != 0)
387fdfb4ba6Ssthen 		       */
388fdfb4ba6Ssthen 
389fdfb4ba6Ssthen 	ret_sz = add_padding ? sldns_b32_ntop_calculate_size(src_sz)
390fdfb4ba6Ssthen 			     : sldns_b32_ntop_calculate_size_no_padding(src_sz);
391fdfb4ba6Ssthen 
392fdfb4ba6Ssthen 	/* Do we have enough space? */
393fdfb4ba6Ssthen 	if (dst_sz < ret_sz + 1)
394fdfb4ba6Ssthen 		return -1;
395fdfb4ba6Ssthen 
396fdfb4ba6Ssthen 	/* We know the size; terminate the string */
397fdfb4ba6Ssthen 	dst[ret_sz] = '\0';
398fdfb4ba6Ssthen 
399fdfb4ba6Ssthen 	/* First process all chunks of five */
400fdfb4ba6Ssthen 	while (src_sz >= 5) {
401fdfb4ba6Ssthen 		/* 00000... ........ ........ ........ ........ */
402fdfb4ba6Ssthen 		dst[0] = b32[(src[0]       ) >> 3];
403fdfb4ba6Ssthen 
404fdfb4ba6Ssthen 		/* .....111 11...... ........ ........ ........ */
405fdfb4ba6Ssthen 		dst[1] = b32[(src[0] & 0x07) << 2 | src[1] >> 6];
406fdfb4ba6Ssthen 
407fdfb4ba6Ssthen 		/* ........ ..22222. ........ ........ ........ */
408fdfb4ba6Ssthen 		dst[2] = b32[(src[1] & 0x3e) >> 1];
409fdfb4ba6Ssthen 
410fdfb4ba6Ssthen 		/* ........ .......3 3333.... ........ ........ */
411fdfb4ba6Ssthen 		dst[3] = b32[(src[1] & 0x01) << 4 | src[2] >> 4];
412fdfb4ba6Ssthen 
413fdfb4ba6Ssthen 		/* ........ ........ ....4444 4....... ........ */
414fdfb4ba6Ssthen 		dst[4] = b32[(src[2] & 0x0f) << 1 | src[3] >> 7];
415fdfb4ba6Ssthen 
416fdfb4ba6Ssthen 		/* ........ ........ ........ .55555.. ........ */
417fdfb4ba6Ssthen 		dst[5] = b32[(src[3] & 0x7c) >> 2];
418fdfb4ba6Ssthen 
419fdfb4ba6Ssthen 		/* ........ ........ ........ ......66 666..... */
420fdfb4ba6Ssthen 		dst[6] = b32[(src[3] & 0x03) << 3 | src[4] >> 5];
421fdfb4ba6Ssthen 
422fdfb4ba6Ssthen 		/* ........ ........ ........ ........ ...77777 */
423fdfb4ba6Ssthen 		dst[7] = b32[(src[4] & 0x1f)     ];
424fdfb4ba6Ssthen 
425fdfb4ba6Ssthen 		src_sz -= 5;
426fdfb4ba6Ssthen 		src    += 5;
427fdfb4ba6Ssthen 		dst    += 8;
428fdfb4ba6Ssthen 	}
429fdfb4ba6Ssthen 	/* Process what remains */
430fdfb4ba6Ssthen 	switch (src_sz) {
431fdfb4ba6Ssthen 	case 4: /* ........ ........ ........ ......66 666..... */
432fdfb4ba6Ssthen 		dst[6] = b32[(src[3] & 0x03) << 3];
433fdfb4ba6Ssthen 
434fdfb4ba6Ssthen 		/* ........ ........ ........ .55555.. ........ */
435fdfb4ba6Ssthen 		dst[5] = b32[(src[3] & 0x7c) >> 2];
436fdfb4ba6Ssthen 
437fdfb4ba6Ssthen 		/* ........ ........ ....4444 4....... ........ */
438fdfb4ba6Ssthen 			 c =  src[3]         >> 7 ;
4397191de28Ssthen 		/* fallthrough */
440fdfb4ba6Ssthen 	case 3: dst[4] = b32[(src[2] & 0x0f) << 1 | c];
441fdfb4ba6Ssthen 
442fdfb4ba6Ssthen 		/* ........ .......3 3333.... ........ ........ */
443fdfb4ba6Ssthen 			 c =  src[2]         >> 4 ;
4447191de28Ssthen 		/* fallthrough */
445fdfb4ba6Ssthen 	case 2:	dst[3] = b32[(src[1] & 0x01) << 4 | c];
446fdfb4ba6Ssthen 
447fdfb4ba6Ssthen 		/* ........ ..22222. ........ ........ ........ */
448fdfb4ba6Ssthen 		dst[2] = b32[(src[1] & 0x3e) >> 1];
449fdfb4ba6Ssthen 
450fdfb4ba6Ssthen 		/* .....111 11...... ........ ........ ........ */
451fdfb4ba6Ssthen 			 c =  src[1]         >> 6 ;
4527191de28Ssthen 		/* fallthrough */
453fdfb4ba6Ssthen 	case 1:	dst[1] = b32[(src[0] & 0x07) << 2 | c];
454fdfb4ba6Ssthen 
455fdfb4ba6Ssthen 		/* 00000... ........ ........ ........ ........ */
456fdfb4ba6Ssthen 		dst[0] = b32[ src[0]         >> 3];
457fdfb4ba6Ssthen 	}
458fdfb4ba6Ssthen 	/* Add padding */
459fdfb4ba6Ssthen 	if (add_padding) {
460fdfb4ba6Ssthen 		switch (src_sz) {
461fdfb4ba6Ssthen 			case 1: dst[2] = '=';
462fdfb4ba6Ssthen 				dst[3] = '=';
4637191de28Ssthen 				/* fallthrough */
464fdfb4ba6Ssthen 			case 2: dst[4] = '=';
4657191de28Ssthen 				/* fallthrough */
466fdfb4ba6Ssthen 			case 3: dst[5] = '=';
467fdfb4ba6Ssthen 				dst[6] = '=';
4687191de28Ssthen 				/* fallthrough */
469fdfb4ba6Ssthen 			case 4: dst[7] = '=';
470fdfb4ba6Ssthen 		}
471fdfb4ba6Ssthen 	}
472fdfb4ba6Ssthen 	return (int)ret_sz;
473fdfb4ba6Ssthen }
474fdfb4ba6Ssthen 
475fdfb4ba6Ssthen int
sldns_b32_ntop(const uint8_t * src,size_t src_sz,char * dst,size_t dst_sz)476fdfb4ba6Ssthen sldns_b32_ntop(const uint8_t* src, size_t src_sz, char* dst, size_t dst_sz)
477fdfb4ba6Ssthen {
478fdfb4ba6Ssthen 	return sldns_b32_ntop_base(src, src_sz, dst, dst_sz, 0, 1);
479fdfb4ba6Ssthen }
480fdfb4ba6Ssthen 
481fdfb4ba6Ssthen int
sldns_b32_ntop_extended_hex(const uint8_t * src,size_t src_sz,char * dst,size_t dst_sz)482fdfb4ba6Ssthen sldns_b32_ntop_extended_hex(const uint8_t* src, size_t src_sz,
483fdfb4ba6Ssthen 		char* dst, size_t dst_sz)
484fdfb4ba6Ssthen {
485fdfb4ba6Ssthen 	return sldns_b32_ntop_base(src, src_sz, dst, dst_sz, 1, 1);
486fdfb4ba6Ssthen }
487fdfb4ba6Ssthen 
sldns_b32_pton_calculate_size(size_t src_text_length)488fdfb4ba6Ssthen size_t sldns_b32_pton_calculate_size(size_t src_text_length)
489fdfb4ba6Ssthen {
490fdfb4ba6Ssthen 	return src_text_length * 5 / 8;
491fdfb4ba6Ssthen }
492fdfb4ba6Ssthen 
493fdfb4ba6Ssthen static int
sldns_b32_pton_base(const char * src,size_t src_sz,uint8_t * dst,size_t dst_sz,int extended_hex,int check_padding)494fdfb4ba6Ssthen sldns_b32_pton_base(const char* src, size_t src_sz, uint8_t* dst, size_t dst_sz,
495fdfb4ba6Ssthen 	int extended_hex, int check_padding)
496fdfb4ba6Ssthen {
497fdfb4ba6Ssthen 	size_t i = 0;
498fdfb4ba6Ssthen 	char ch = '\0';
499fdfb4ba6Ssthen 	uint8_t buf[8];
500fdfb4ba6Ssthen 	uint8_t* start = dst;
501fdfb4ba6Ssthen 
502fdfb4ba6Ssthen 	while (src_sz) {
503fdfb4ba6Ssthen 		/* Collect 8 characters in buf (if possible) */
504fdfb4ba6Ssthen 		for (i = 0; i < 8; i++) {
505fdfb4ba6Ssthen 
506fdfb4ba6Ssthen 			do {
507fdfb4ba6Ssthen 				ch = *src++;
508fdfb4ba6Ssthen 				--src_sz;
509fdfb4ba6Ssthen 
510fdfb4ba6Ssthen 			} while (isspace((unsigned char)ch) && src_sz > 0);
511fdfb4ba6Ssthen 
512fdfb4ba6Ssthen 			if (ch == '=' || ch == '\0')
513fdfb4ba6Ssthen 				break;
514fdfb4ba6Ssthen 
515fdfb4ba6Ssthen 			else if (extended_hex)
516fdfb4ba6Ssthen 
517fdfb4ba6Ssthen 				if (ch >= '0' && ch <= '9')
518fdfb4ba6Ssthen 					buf[i] = (uint8_t)ch - '0';
519fdfb4ba6Ssthen 				else if (ch >= 'a' && ch <= 'v')
520fdfb4ba6Ssthen 					buf[i] = (uint8_t)ch - 'a' + 10;
521fdfb4ba6Ssthen 				else if (ch >= 'A' && ch <= 'V')
522fdfb4ba6Ssthen 					buf[i] = (uint8_t)ch - 'A' + 10;
523fdfb4ba6Ssthen 				else
524fdfb4ba6Ssthen 					return -1;
525fdfb4ba6Ssthen 
526fdfb4ba6Ssthen 			else if (ch >= 'a' && ch <= 'z')
527fdfb4ba6Ssthen 				buf[i] = (uint8_t)ch - 'a';
528fdfb4ba6Ssthen 			else if (ch >= 'A' && ch <= 'Z')
529fdfb4ba6Ssthen 				buf[i] = (uint8_t)ch - 'A';
530fdfb4ba6Ssthen 			else if (ch >= '2' && ch <= '7')
531fdfb4ba6Ssthen 				buf[i] = (uint8_t)ch - '2' + 26;
532fdfb4ba6Ssthen 			else
533fdfb4ba6Ssthen 				return -1;
534fdfb4ba6Ssthen 		}
535fdfb4ba6Ssthen 		/* Less that 8 characters. We're done. */
536fdfb4ba6Ssthen 		if (i < 8)
537fdfb4ba6Ssthen 			break;
538fdfb4ba6Ssthen 
539fdfb4ba6Ssthen 		/* Enough space available at the destination? */
540fdfb4ba6Ssthen 		if (dst_sz < 5)
541fdfb4ba6Ssthen 			return -1;
542fdfb4ba6Ssthen 
543fdfb4ba6Ssthen 		/* 00000... ........ ........ ........ ........ */
544fdfb4ba6Ssthen 		/* .....111 11...... ........ ........ ........ */
545fdfb4ba6Ssthen 		dst[0] = buf[0] << 3 | buf[1] >> 2;
546fdfb4ba6Ssthen 
547fdfb4ba6Ssthen 		/* .....111 11...... ........ ........ ........ */
548fdfb4ba6Ssthen 		/* ........ ..22222. ........ ........ ........ */
549fdfb4ba6Ssthen 		/* ........ .......3 3333.... ........ ........ */
550fdfb4ba6Ssthen 		dst[1] = buf[1] << 6 | buf[2] << 1 | buf[3] >> 4;
551fdfb4ba6Ssthen 
552fdfb4ba6Ssthen 		/* ........ .......3 3333.... ........ ........ */
553fdfb4ba6Ssthen 		/* ........ ........ ....4444 4....... ........ */
554fdfb4ba6Ssthen 		dst[2] = buf[3] << 4 | buf[4] >> 1;
555fdfb4ba6Ssthen 
556fdfb4ba6Ssthen 		/* ........ ........ ....4444 4....... ........ */
557fdfb4ba6Ssthen 		/* ........ ........ ........ .55555.. ........ */
558fdfb4ba6Ssthen 		/* ........ ........ ........ ......66 666..... */
559fdfb4ba6Ssthen 		dst[3] = buf[4] << 7 | buf[5] << 2 | buf[6] >> 3;
560fdfb4ba6Ssthen 
561fdfb4ba6Ssthen 		/* ........ ........ ........ ......66 666..... */
562fdfb4ba6Ssthen 		/* ........ ........ ........ ........ ...77777 */
563fdfb4ba6Ssthen 		dst[4] = buf[6] << 5 | buf[7];
564fdfb4ba6Ssthen 
565fdfb4ba6Ssthen 		dst += 5;
566fdfb4ba6Ssthen 		dst_sz -= 5;
567fdfb4ba6Ssthen 	}
568fdfb4ba6Ssthen 	/* Not ending on a eight byte boundary? */
569fdfb4ba6Ssthen 	if (i > 0 && i < 8) {
570fdfb4ba6Ssthen 
571fdfb4ba6Ssthen 		/* Enough space available at the destination? */
572fdfb4ba6Ssthen 		if (dst_sz < (i + 1) / 2)
573fdfb4ba6Ssthen 			return -1;
574fdfb4ba6Ssthen 
575fdfb4ba6Ssthen 		switch (i) {
576fdfb4ba6Ssthen 		case 7: /* ........ ........ ........ ......66 666..... */
577fdfb4ba6Ssthen 			/* ........ ........ ........ .55555.. ........ */
578fdfb4ba6Ssthen 			/* ........ ........ ....4444 4....... ........ */
579fdfb4ba6Ssthen 			dst[3] = buf[4] << 7 | buf[5] << 2 | buf[6] >> 3;
5807191de28Ssthen 			/* fallthrough */
581fdfb4ba6Ssthen 
582fdfb4ba6Ssthen 		case 5: /* ........ ........ ....4444 4....... ........ */
583fdfb4ba6Ssthen 			/* ........ .......3 3333.... ........ ........ */
584fdfb4ba6Ssthen 			dst[2] = buf[3] << 4 | buf[4] >> 1;
5857191de28Ssthen 			/* fallthrough */
586fdfb4ba6Ssthen 
587fdfb4ba6Ssthen 		case 4: /* ........ .......3 3333.... ........ ........ */
588fdfb4ba6Ssthen 			/* ........ ..22222. ........ ........ ........ */
589fdfb4ba6Ssthen 			/* .....111 11...... ........ ........ ........ */
590fdfb4ba6Ssthen 			dst[1] = buf[1] << 6 | buf[2] << 1 | buf[3] >> 4;
5917191de28Ssthen 			/* fallthrough */
592fdfb4ba6Ssthen 
593fdfb4ba6Ssthen 		case 2: /* .....111 11...... ........ ........ ........ */
594fdfb4ba6Ssthen 			/* 00000... ........ ........ ........ ........ */
595fdfb4ba6Ssthen 			dst[0] = buf[0] << 3 | buf[1] >> 2;
596fdfb4ba6Ssthen 
597fdfb4ba6Ssthen 			break;
598fdfb4ba6Ssthen 
599fdfb4ba6Ssthen 		default:
600fdfb4ba6Ssthen 			return -1;
601fdfb4ba6Ssthen 		}
602fdfb4ba6Ssthen 		dst += (i + 1) / 2;
603fdfb4ba6Ssthen 
604fdfb4ba6Ssthen 		if (check_padding) {
605fdfb4ba6Ssthen 			/* Check remaining padding characters */
606fdfb4ba6Ssthen 			if (ch != '=')
607fdfb4ba6Ssthen 				return -1;
608fdfb4ba6Ssthen 
609fdfb4ba6Ssthen 			/* One down, 8 - i - 1 more to come... */
610fdfb4ba6Ssthen 			for (i = 8 - i - 1; i > 0; i--) {
611fdfb4ba6Ssthen 
612fdfb4ba6Ssthen 				do {
613fdfb4ba6Ssthen 					if (src_sz == 0)
614fdfb4ba6Ssthen 						return -1;
615fdfb4ba6Ssthen 					ch = *src++;
616fdfb4ba6Ssthen 					src_sz--;
617fdfb4ba6Ssthen 
618fdfb4ba6Ssthen 				} while (isspace((unsigned char)ch));
619fdfb4ba6Ssthen 
620fdfb4ba6Ssthen 				if (ch != '=')
621fdfb4ba6Ssthen 					return -1;
622fdfb4ba6Ssthen 			}
623fdfb4ba6Ssthen 		}
624fdfb4ba6Ssthen 	}
625fdfb4ba6Ssthen 	return dst - start;
626fdfb4ba6Ssthen }
627fdfb4ba6Ssthen 
628fdfb4ba6Ssthen int
sldns_b32_pton(const char * src,size_t src_sz,uint8_t * dst,size_t dst_sz)629fdfb4ba6Ssthen sldns_b32_pton(const char* src, size_t src_sz, uint8_t* dst, size_t dst_sz)
630fdfb4ba6Ssthen {
631fdfb4ba6Ssthen 	return sldns_b32_pton_base(src, src_sz, dst, dst_sz, 0, 1);
632fdfb4ba6Ssthen }
633fdfb4ba6Ssthen 
634fdfb4ba6Ssthen int
sldns_b32_pton_extended_hex(const char * src,size_t src_sz,uint8_t * dst,size_t dst_sz)635fdfb4ba6Ssthen sldns_b32_pton_extended_hex(const char* src, size_t src_sz,
636fdfb4ba6Ssthen 		uint8_t* dst, size_t dst_sz)
637fdfb4ba6Ssthen {
638fdfb4ba6Ssthen 	return sldns_b32_pton_base(src, src_sz, dst, dst_sz, 1, 1);
639fdfb4ba6Ssthen }
640fdfb4ba6Ssthen 
sldns_b64_ntop_calculate_size(size_t srcsize)641fdfb4ba6Ssthen size_t sldns_b64_ntop_calculate_size(size_t srcsize)
642fdfb4ba6Ssthen {
643fdfb4ba6Ssthen 	return ((((srcsize + 2) / 3) * 4) + 1);
644fdfb4ba6Ssthen }
645fdfb4ba6Ssthen 
646fdfb4ba6Ssthen /* RFC 1521, section 5.2.
647fdfb4ba6Ssthen  *
648fdfb4ba6Ssthen  * The encoding process represents 24-bit groups of input bits as output
649fdfb4ba6Ssthen  * strings of 4 encoded characters. Proceeding from left to right, a
650fdfb4ba6Ssthen  * 24-bit input group is formed by concatenating 3 8-bit input groups.
651fdfb4ba6Ssthen  * These 24 bits are then treated as 4 concatenated 6-bit groups, each
652fdfb4ba6Ssthen  * of which is translated into a single digit in the base64 alphabet.
653fdfb4ba6Ssthen  *
654fdfb4ba6Ssthen  * This routine does not insert spaces or linebreaks after 76 characters.
655fdfb4ba6Ssthen  */
sldns_b64_ntop_base(uint8_t const * src,size_t srclength,char * target,size_t targsize,int base64url,int padding)6562c144df0Ssthen static int sldns_b64_ntop_base(uint8_t const *src, size_t srclength,
6572c144df0Ssthen 	char *target, size_t targsize, int base64url, int padding)
658fdfb4ba6Ssthen {
6592c144df0Ssthen 	char* b64;
660fdfb4ba6Ssthen 	const char pad64 = '=';
661fdfb4ba6Ssthen 	size_t i = 0, o = 0;
6622c144df0Ssthen 	if(base64url)
6632c144df0Ssthen 		b64 = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123"
6642c144df0Ssthen 			"456789-_";
6652c144df0Ssthen 	else
6662c144df0Ssthen 		b64 = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123"
6672c144df0Ssthen 			"456789+/";
668fdfb4ba6Ssthen 	if(targsize < sldns_b64_ntop_calculate_size(srclength))
669fdfb4ba6Ssthen 		return -1;
670fdfb4ba6Ssthen 	/* whole chunks: xxxxxxyy yyyyzzzz zzwwwwww */
671fdfb4ba6Ssthen 	while(i+3 <= srclength) {
672fdfb4ba6Ssthen 		if(o+4 > targsize) return -1;
673fdfb4ba6Ssthen 		target[o] = b64[src[i] >> 2];
674fdfb4ba6Ssthen 		target[o+1] = b64[ ((src[i]&0x03)<<4) | (src[i+1]>>4) ];
675fdfb4ba6Ssthen 		target[o+2] = b64[ ((src[i+1]&0x0f)<<2) | (src[i+2]>>6) ];
676fdfb4ba6Ssthen 		target[o+3] = b64[ (src[i+2]&0x3f) ];
677fdfb4ba6Ssthen 		i += 3;
678fdfb4ba6Ssthen 		o += 4;
679fdfb4ba6Ssthen 	}
680fdfb4ba6Ssthen 	/* remainder */
681fdfb4ba6Ssthen 	switch(srclength - i) {
682fdfb4ba6Ssthen 	case 2:
683fdfb4ba6Ssthen 		/* two at end, converted into A B C = */
684fdfb4ba6Ssthen 		target[o] = b64[src[i] >> 2];
685fdfb4ba6Ssthen 		target[o+1] = b64[ ((src[i]&0x03)<<4) | (src[i+1]>>4) ];
686fdfb4ba6Ssthen 		target[o+2] = b64[ ((src[i+1]&0x0f)<<2) ];
6872c144df0Ssthen 		if(padding) {
688fdfb4ba6Ssthen 			target[o+3] = pad64;
6892ee382b6Ssthen 			/* i += 2; */
690fdfb4ba6Ssthen 			o += 4;
6912c144df0Ssthen 		} else {
6922c144df0Ssthen 			o += 3;
6932c144df0Ssthen 		}
694fdfb4ba6Ssthen 		break;
695fdfb4ba6Ssthen 	case 1:
696fdfb4ba6Ssthen 		/* one at end, converted into A B = = */
697fdfb4ba6Ssthen 		target[o] = b64[src[i] >> 2];
698fdfb4ba6Ssthen 		target[o+1] = b64[ ((src[i]&0x03)<<4) ];
6992c144df0Ssthen 		if(padding) {
700fdfb4ba6Ssthen 			target[o+2] = pad64;
701fdfb4ba6Ssthen 			target[o+3] = pad64;
7022ee382b6Ssthen 			/* i += 1; */
703fdfb4ba6Ssthen 			o += 4;
7042c144df0Ssthen 		} else {
7052c144df0Ssthen 			o += 2;
7062c144df0Ssthen 		}
707fdfb4ba6Ssthen 		break;
708fdfb4ba6Ssthen 	case 0:
709fdfb4ba6Ssthen 	default:
710fdfb4ba6Ssthen 		/* nothing */
711fdfb4ba6Ssthen 		break;
712fdfb4ba6Ssthen 	}
713fdfb4ba6Ssthen 	/* assert: i == srclength */
714fdfb4ba6Ssthen 	if(o+1 > targsize) return -1;
715fdfb4ba6Ssthen 	target[o] = 0;
716fdfb4ba6Ssthen 	return (int)o;
717fdfb4ba6Ssthen }
718fdfb4ba6Ssthen 
sldns_b64_ntop(uint8_t const * src,size_t srclength,char * target,size_t targsize)7192c144df0Ssthen int sldns_b64_ntop(uint8_t const *src, size_t srclength, char *target,
7202c144df0Ssthen 	size_t targsize)
7212c144df0Ssthen {
7222c144df0Ssthen 	return sldns_b64_ntop_base(src, srclength, target, targsize,
7232c144df0Ssthen 		0 /* no base64url */, 1 /* padding */);
7242c144df0Ssthen }
7252c144df0Ssthen 
sldns_b64url_ntop(uint8_t const * src,size_t srclength,char * target,size_t targsize)7262c144df0Ssthen int sldns_b64url_ntop(uint8_t const *src, size_t srclength, char *target,
7272c144df0Ssthen 	size_t targsize)
7282c144df0Ssthen {
7292c144df0Ssthen 	return sldns_b64_ntop_base(src, srclength, target, targsize,
7302c144df0Ssthen 		1 /* base64url */, 0 /* no padding */);
7312c144df0Ssthen }
7322c144df0Ssthen 
sldns_b64_pton_calculate_size(size_t srcsize)733fdfb4ba6Ssthen size_t sldns_b64_pton_calculate_size(size_t srcsize)
734fdfb4ba6Ssthen {
735fdfb4ba6Ssthen 	return (((((srcsize + 3) / 4) * 3)) + 1);
736fdfb4ba6Ssthen }
737fdfb4ba6Ssthen 
7382c144df0Ssthen /* padding not required if srcsize is set */
sldns_b64_pton_base(char const * src,size_t srcsize,uint8_t * target,size_t targsize,int base64url)7392c144df0Ssthen static int sldns_b64_pton_base(char const *src, size_t srcsize, uint8_t *target,
7402c144df0Ssthen 	size_t targsize, int base64url)
741fdfb4ba6Ssthen {
742fdfb4ba6Ssthen 	const uint8_t pad64 = 64; /* is 64th in the b64 array */
743fdfb4ba6Ssthen 	const char* s = src;
744fdfb4ba6Ssthen 	uint8_t in[4];
745fdfb4ba6Ssthen 	size_t o = 0, incount = 0;
7462c144df0Ssthen 	int check_padding = (srcsize) ? 0 : 1;
747fdfb4ba6Ssthen 
7482c144df0Ssthen 	while(*s && (check_padding || srcsize)) {
749fdfb4ba6Ssthen 		/* skip any character that is not base64 */
750fdfb4ba6Ssthen 		/* conceptually we do:
751fdfb4ba6Ssthen 		const char* b64 =      pad'=' is appended to array
752fdfb4ba6Ssthen 		"ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/=";
753fdfb4ba6Ssthen 		const char* d = strchr(b64, *s++);
754fdfb4ba6Ssthen 		and use d-b64;
755fdfb4ba6Ssthen 		*/
756fdfb4ba6Ssthen 		char d = *s++;
7572c144df0Ssthen 		srcsize--;
758fdfb4ba6Ssthen 		if(d <= 'Z' && d >= 'A')
759fdfb4ba6Ssthen 			d -= 'A';
760fdfb4ba6Ssthen 		else if(d <= 'z' && d >= 'a')
761fdfb4ba6Ssthen 			d = d - 'a' + 26;
762fdfb4ba6Ssthen 		else if(d <= '9' && d >= '0')
763fdfb4ba6Ssthen 			d = d - '0' + 52;
7642c144df0Ssthen 		else if(!base64url && d == '+')
765fdfb4ba6Ssthen 			d = 62;
7662c144df0Ssthen 		else if(base64url && d == '-')
7672c144df0Ssthen 			d = 62;
7682c144df0Ssthen 		else if(!base64url && d == '/')
769fdfb4ba6Ssthen 			d = 63;
7702c144df0Ssthen 		else if(base64url && d == '_')
7712c144df0Ssthen 			d = 63;
7722c144df0Ssthen 		else if(d == '=') {
7732c144df0Ssthen 			if(!check_padding)
774fdfb4ba6Ssthen 				continue;
7752c144df0Ssthen 			d = 64;
7762c144df0Ssthen 		} else	continue;
7772c144df0Ssthen 
7782c144df0Ssthen 		in[incount++] = (uint8_t)d;
7792c144df0Ssthen 		/* work on block of 4, unless padding is not used and there are
7802c144df0Ssthen 		 * less than 4 chars left */
7812c144df0Ssthen 		if(incount != 4 && (check_padding || srcsize))
7822c144df0Ssthen 			continue;
7832c144df0Ssthen 		assert(!check_padding || incount==4);
784fdfb4ba6Ssthen 		/* process whole block of 4 characters into 3 output bytes */
7852c144df0Ssthen 		if((incount == 2 ||
7862c144df0Ssthen 			(incount == 4 && in[3] == pad64 && in[2] == pad64))) { /* A B = = */
787fdfb4ba6Ssthen 			if(o+1 > targsize)
788fdfb4ba6Ssthen 				return -1;
789fdfb4ba6Ssthen 			target[o] = (in[0]<<2) | ((in[1]&0x30)>>4);
790fdfb4ba6Ssthen 			o += 1;
791fdfb4ba6Ssthen 			break; /* we are done */
7922c144df0Ssthen 		} else if(incount == 3 ||
7932c144df0Ssthen 			(incount == 4 && in[3] == pad64)) { /* A B C = */
794fdfb4ba6Ssthen 			if(o+2 > targsize)
795fdfb4ba6Ssthen 				return -1;
796fdfb4ba6Ssthen 			target[o] = (in[0]<<2) | ((in[1]&0x30)>>4);
797fdfb4ba6Ssthen 			target[o+1]= ((in[1]&0x0f)<<4) | ((in[2]&0x3c)>>2);
798fdfb4ba6Ssthen 			o += 2;
799fdfb4ba6Ssthen 			break; /* we are done */
800fdfb4ba6Ssthen 		} else {
8012c144df0Ssthen 			if(incount != 4 || o+3 > targsize)
802fdfb4ba6Ssthen 				return -1;
803fdfb4ba6Ssthen 			/* write xxxxxxyy yyyyzzzz zzwwwwww */
804fdfb4ba6Ssthen 			target[o] = (in[0]<<2) | ((in[1]&0x30)>>4);
805fdfb4ba6Ssthen 			target[o+1]= ((in[1]&0x0f)<<4) | ((in[2]&0x3c)>>2);
806fdfb4ba6Ssthen 			target[o+2]= ((in[2]&0x03)<<6) | in[3];
807fdfb4ba6Ssthen 			o += 3;
808fdfb4ba6Ssthen 		}
809fdfb4ba6Ssthen 		incount = 0;
810fdfb4ba6Ssthen 	}
811fdfb4ba6Ssthen 	return (int)o;
812fdfb4ba6Ssthen }
8132c144df0Ssthen 
sldns_b64_pton(char const * src,uint8_t * target,size_t targsize)8142c144df0Ssthen int sldns_b64_pton(char const *src, uint8_t *target, size_t targsize)
8152c144df0Ssthen {
8162c144df0Ssthen 	return sldns_b64_pton_base(src, 0, target, targsize, 0);
8172c144df0Ssthen }
8182c144df0Ssthen 
sldns_b64url_pton(char const * src,size_t srcsize,uint8_t * target,size_t targsize)8192c144df0Ssthen int sldns_b64url_pton(char const *src, size_t srcsize, uint8_t *target,
8202c144df0Ssthen 	size_t targsize)
8212c144df0Ssthen {
8222c144df0Ssthen 	if(!srcsize) {
8232c144df0Ssthen 		return 0;
8242c144df0Ssthen 	}
8252c144df0Ssthen 	return sldns_b64_pton_base(src, srcsize, target, targsize, 1);
8262c144df0Ssthen }
827191f22c6Ssthen 
sldns_b64_contains_nonurl(char const * src,size_t srcsize)828191f22c6Ssthen int sldns_b64_contains_nonurl(char const *src, size_t srcsize)
829191f22c6Ssthen {
830191f22c6Ssthen 	const char* s = src;
831191f22c6Ssthen 	while(*s && srcsize) {
832191f22c6Ssthen 		char d = *s++;
833191f22c6Ssthen 		srcsize--;
834191f22c6Ssthen 		/* the '+' and the '/' and padding '=' is not allowed in b64
835191f22c6Ssthen 		 * url encoding */
836191f22c6Ssthen 		if(d == '+' || d == '/' || d == '=') {
837191f22c6Ssthen 			return 1;
838191f22c6Ssthen 		}
839191f22c6Ssthen 	}
840191f22c6Ssthen 	return 0;
841191f22c6Ssthen }
842