aboutsummaryrefslogtreecommitdiffstats
path: root/strtouh.c
blob: fb2f037e66574d84083c1c2425aa7b981714e3ba (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
/* See LICENSE file for copyright and license details. */
#include "common.h"
#ifndef TEST


unsigned short int
libsimple_strtouh(const char *restrict nptr, char **restrict end, int base)
{
	unsigned long int r = strtoul(nptr, end, base);
	if (r > USHRT_MAX) {
		r = USHRT_MAX;
		errno = ERANGE;
	}
	return (unsigned short int)r;
}


#else
#include "test.h"

static void
add_one(char *buf)
{
	char *p = strchr(buf, '\0');
	while (*--p == '9')
		*p = '0';
	*p += 1;
}

int
main(void)
{
	char str[128];
	char *e;
	sprintf(str, "0x%x", (unsigned int)USHRT_MAX);
	errno = 0;
	assert(strtouh(str, NULL, 0) == USHRT_MAX && !errno);
	assert(strtouh(str, NULL, 16) == USHRT_MAX && !errno);
	assert(strtouh(&str[2], NULL, 16) == USHRT_MAX && !errno);
	assert(strtouh(str, NULL, 10) == 0 && !errno);
	assert(strtouh(str, &e, 0) == USHRT_MAX && !*e && !errno);
	assert(strtouh(str, &e, 10) == 0 && *e == 'x' && !errno);
	sprintf(str, "0x%x ", (unsigned int)USHRT_MAX);
	assert(strtouh(str, &e, 16) == USHRT_MAX && *e == ' ' && !errno);
	sprintf(str, "%u", (unsigned int)USHRT_MAX);
	assert(strtouh(str, &e, 10) == USHRT_MAX && !*e && !errno);
	assert(strtouh("1234", &e, 10) == 1234 && !*e && !errno);
	assert(strtouh("1234", &e, 8) == 01234 && !*e && !errno);
	assert(strtouh("01234", &e, 0) == 01234 && !*e && !errno);
	sprintf(str, "%u", (unsigned int)USHRT_MAX);
	add_one(str);
	assert(strtouh(str, &e, 10) == USHRT_MAX && !*e && errno == ERANGE);
	errno = 0;
	assert(!strtouh("1", &e, -10000) && errno == EINVAL);
	errno = 0;
	return 0;
}

#endif