/*
 * Copyright 2008, Oliver Tappe, zooey@hirschkaefer.de.
 * Distributed under the terms of the MIT license.
 */


#include <NetAddress.h>

#include <errno.h>
#include <netinet/in.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>


void
checkNetAddr(const BNetAddress& netAddr, uint32 nwAddr, uint16 nwPort)
{
	int32 status;
	in_addr addr;
	unsigned short port;
	if ((status = netAddr.GetAddr(addr, &port)) != B_OK) {
		fprintf(stderr, 
			"failed to get in_addr/port from localhost address: %s\n",
			strerror(status));
		exit(1);
	}
	if (addr.s_addr != nwAddr) {
		fprintf(stderr, "expected in_addr %lx - got %x\n", nwAddr, addr.s_addr);
		exit(1);
	}
	uint16 hostPort = ntohs(nwPort);
	if (port != hostPort) {
		fprintf(stderr, "expected port %u - got %u\n", hostPort, port);
		exit(1);
	}

	sockaddr_in sockaddr;
	if ((status = netAddr.GetAddr(sockaddr)) != B_OK) {
		fprintf(stderr, 
			"failed to get sockaddr_in from netAddr: %s\n",
			strerror(status));
		exit(1);
	}
	if (sockaddr.sin_family != AF_INET) {
		fprintf(stderr, "expected sockaddr-family %u - got %u\n", AF_INET,
			sockaddr.sin_family);
		exit(1);
	}
	if (sockaddr.sin_port != nwPort) {
		fprintf(stderr, "expected sockaddr-port %x - got %x\n", nwPort,
			sockaddr.sin_port);
		exit(1);
	}
	if (sockaddr.sin_addr.s_addr != nwAddr) {
		fprintf(stderr, "expected sockaddr-address %lx - got %x\n", nwAddr,
			sockaddr.sin_addr.s_addr);
		exit(1);
	}
}


int
main(int argc, const char* const* argv)
{
	BNetAddress netAddr;
	if (sizeof(netAddr) != 52) {
		fprintf(stderr, "expected sizeof(netAddr) to be 52 - is %ld\n",
			sizeof(netAddr));
		exit(1);
	}

	netAddr.SetTo("localhost", 123);
	checkNetAddr(netAddr, htonl(0x7F000001), htons(123));

	netAddr.SetTo(htonl(0x7F000001), 123);
	checkNetAddr(netAddr, htonl(0x7F000001), htons(123));

	in_addr addr;
	addr.s_addr = htonl(0x7F000001);
	netAddr.SetTo(addr, 54321);
	checkNetAddr(netAddr, htonl(0x7F000001), htons(54321));

	sockaddr_in sockaddr;
	sockaddr.sin_family = AF_INET;
	sockaddr.sin_addr.s_addr = htonl(0x7F000001);
	sockaddr.sin_port = htons(54321);
	netAddr.SetTo(sockaddr);
	checkNetAddr(netAddr, htonl(0x7F000001), htons(54321));

	printf("Everything went fine.\n");
	
	return 0;
}