// inet_addr, inet_ntoa, inet_pton and inet_ntop tests with roundtrip check

#include <errno.h>

#include <string.h>

#include <stdio.h>

#include <arpa/inet.h>

#include "test.h"

static int digit(int c)

{

	c-='0';

	if (c>9) c-='a'-'0'-10;

	return c;

}

static void tobin(void *d, char *s)

{

	int i;

	unsigned char *p = d;

	for (i=0; s[2*i]; i++) p[i] = digit(s[2*i])*16+digit(s[2*i+1]);

}

static void tohex(char *d, void *s, int n)

{

	int i;

	unsigned char *p = s;

	for (i=0; i

}

#define V6(src,ret,hex) do{\

	int r; \

	char binaddr[16]={0}; \

	char hexaddr[40]={0}; \

	char txtaddr[60]={0}; \

	\

	r=inet_pton(AF_INET6,src,binaddr); \

	if (r!=ret) \

		t_error("inet_pton(AF_INET6, "#src", addr) returned %d, want %d\n", r, ret); \

	if (ret!=1) break; \

	tohex(hexaddr,binaddr,16); \

	if (strcmp(hexaddr,hex)) \

		t_error("inet_pton(AF_INET6, "#src", addr) got addr %s, want %s\n", hexaddr, hex); \

	\

	tobin(binaddr,hex); \

	if (inet_ntop(AF_INET6,binaddr,txtaddr,sizeof txtaddr)!=txtaddr) \

		t_error("inet_ntop(AF_INET6, <"#hex">, buf, size) did not return buf\n"); \

	if (inet_pton(AF_INET6,txtaddr,binaddr)!=1) \

		t_error("inet_ntop(AF_INET6, <"#hex">, buf, size) got %s, it is rejected by inet_pton\n", txtaddr); \

	tohex(hexaddr,binaddr,16); \

	if (strcmp(hexaddr,hex)) \

		t_error("inet_ntop(AF_INET6, <"#hex">, buf, size) got %s that is %s, want %s\n", txtaddr, hexaddr, hex); \

	if (strncmp(hex,"00000000000000000000ffff",24)==0 && !strchr(txtaddr,'.')) \

		t_error("inet_ntop(AF_INET6, <"#hex">, buf, size) got %s, should be ipv4 mapped\n", txtaddr); \

}while(0);

// ret and hex are the results of inet_pton and inet_addr respectively

#define V4(src,ret,hex) do{\

	int r; \

	uint32_t a; \

	struct in_addr in; \

	char buf[20]={0}; \

	char *p; \

	\

	a=inet_addr(src); \

	tohex(buf,&a,4); \

	if (strcmp(buf,hex)) \

		t_error("inet_addr("#src") returned %s, want %s\n", buf, hex); \

	\

	r=inet_pton(AF_INET,src,&a); \

	if (r!=ret) \

		t_error("inet_pton(AF_INET, "#src", addr) returned %d, want %d\n", r, ret); \

	\

	if (ret!=1) break; \

	\

	tohex(buf,&a,4); \

	if (strcmp(buf,hex)) \

		t_error("inet_pton(AF_INET, "#src", addr) got addr %s, want %s\n", buf, hex); \

	\

	tobin(&a,hex); \

	if (inet_ntop(AF_INET,&a,buf,sizeof buf)!=buf) \

		t_error("inet_ntop(AF_INET, <"#hex">, buf, size) did not return buf\n"); \

	if (strcmp(buf,src)) \

		t_error("inet_ntop(AF_INET, <"#hex">, buf, size) got %s, want %s\n", buf, src); \

	\

	in.s_addr = a; \

	p=inet_ntoa(in); \

	if (strcmp(p,src)) \

		t_error("inet_ntoa(<"#hex">) returned %s, want %s\n", p, src); \

}while(0);

int main(void)

{

// errors

if (inet_pton(12345, "", 0) != -1 || errno != EAFNOSUPPORT)

	t_error("inet_pton(12345,,) should fail with EAFNOSUPPORT, got %s\n", strerror(errno));

errno=0;

if (inet_ntop(AF_INET,"xxxx","",0) != 0 || errno != ENOSPC)

	t_error("inet_ntop(,,0,0) should fail with ENOSPC, got %s\n", strerror(errno));

errno=0;

// dotted-decimal notation

V4("0.0.0.0", 1, "00000000")

V4("127.0.0.1", 1, "7f000001")

V4("10.0.128.31", 1, "0a00801f")

V4("255.255.255.255", 1, "ffffffff")

// numbers-and-dots notation, but not dotted-decimal

V4("1.2.03.4", 0, "01020304")

V4("1.2.0x33.4", 0, "01023304")

V4("1.2.0XAB.4", 0, "0102ab04")

V4("1.2.0xabcd", 0, "0102abcd")

V4("1.0xabcdef", 0, "01abcdef")

V4("00377.0x0ff.65534", 0, "fffffffe")

// invalid

V4(".1.2.3", 0, "ffffffff")

V4("1..2.3", 0, "ffffffff")

V4("1.2.3.", 0, "ffffffff")

V4("1.2.3.4.5", 0, "ffffffff")

V4("1.2.3.a", 0, "ffffffff")

V4("1.256.2.3", 0, "ffffffff")

V4("1.2.4294967296.3", 0, "ffffffff")

V4("1.2.-4294967295.3", 0, "ffffffff")

V4("1.2. 3.4", 0, "ffffffff")

// ipv6

V6(":", 0, "")

V6("::", 1, "00000000000000000000000000000000")

V6("::1", 1, "00000000000000000000000000000001")

V6(":::", 0, "")

V6("192.168.1.1", 0, "")

V6(":192.168.1.1", 0, "")

V6("::192.168.1.1", 1, "000000000000000000000000c0a80101")

V6("0:0:0:0:0:0:192.168.1.1", 1, "000000000000000000000000c0a80101")

V6("0:0::0:0:0:192.168.1.1", 1, "000000000000000000000000c0a80101")

V6("::012.34.56.78", 0, "")

V6(":ffff:192.168.1.1", 0, "")

V6("::ffff:192.168.1.1", 1, "00000000000000000000ffffc0a80101")

V6(".192.168.1.1", 0, "")

V6(":.192.168.1.1", 0, "")

V6("a:0b:00c:000d:E:F::", 1, "000a000b000c000d000e000f00000000")

V6("a:0b:00c:000d:0000e:f::", 0, "")

V6("1:2:3:4:5:6::", 1, "00010002000300040005000600000000")

V6("1:2:3:4:5:6:7::", 1, "00010002000300040005000600070000")

V6("1:2:3:4:5:6:7:8::", 0, "")

V6("1:2:3:4:5:6:7::9", 0, "")

V6("::1:2:3:4:5:6", 1, "00000000000100020003000400050006")

V6("::1:2:3:4:5:6:7", 1, "00000001000200030004000500060007")

V6("::1:2:3:4:5:6:7:8", 0, "")

V6("a:b::c:d:e:f", 1, "000a000b00000000000c000d000e000f")

V6("ffff:c0a8:5e4", 0, "")

V6(":ffff:c0a8:5e4", 0, "")

V6("0:0:0:0:0:ffff:c0a8:5e4", 1, "00000000000000000000ffffc0a805e4")

V6("0:0:0:0:ffff:c0a8:5e4", 0, "")

V6("0::ffff:c0a8:5e4", 1, "00000000000000000000ffffc0a805e4")

V6("::0::ffff:c0a8:5e4", 0, "")

V6("c0a8", 0, "")

return t_status;

}