/*
 * libretroshare/src/test/pqi net_test.cc
 *
 * 3P/PQI network interface for RetroShare.
 *
 * Copyright 2007-2010 by Robert Fernie.
 *
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Library General Public
 * License Version 2 as published by the Free Software Foundation.
 *
 * This library is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Library General Public License for more details.
 *
 * You should have received a copy of the GNU Library General Public
 * License along with this library; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
 * USA.
 *
 * Please report all bugs and problems to "retroshare@lunamutt.com".
 *
 */

/******
 * NETWORKING Test to check Big/Little Endian behaviour
 * as well as socket behaviour
 *
 */

#include "util/utest.h"

#include "pqi/pqinetwork.h"
#include "util/rsnet.h"
#include <iostream>
#include <sstream>


bool test_byte_manipulation();
bool test_local_address_manipulation();
bool test_address_listen();


INITTEST();

int main(int argc, char **argv)
{
	std::cerr << "libretroshare:pqi " << argv[0] << std::endl;

	test_byte_manipulation();
	test_local_address_manipulation();
	test_address_listen();

	FINALREPORT("libretroshare Net Basics Tests");
	return TESTRESULT();
}

	/* test 1: byte manipulation */
bool test_byte_manipulation()
{
	uint64_t num1 = 0x0000000000000000ffULL; /* 255 */
	uint64_t num2 = 0x00000000000000ff00ULL; /*  */

	uint64_t n_num1 = htonll(num1);
	uint64_t n_num2 = htonll(num2);

	uint64_t h_num1 = ntohll(n_num1);
	uint64_t h_num2 = ntohll(n_num2);

	std::ostringstream out;
	out << std::hex;
	out << "num1: " << num1 << " netOrder: " << n_num1 << " hostOrder: " << h_num1 << std::endl;
	out << "num2: " << num2 << " netOrder: " << n_num2 << " hostOrder: " << h_num2 << std::endl;
	
	std::cerr << out.str();

        CHECK( num1 == h_num1 );
        CHECK( num2 == h_num2 );

        REPORT("Test Byte Manipulation");

	return true;
}

const char * loopback_addrstr = "127.0.0.1";
const char * localnet1_addrstr = "192.168.0.1";
const char * localnet2_addrstr = "10.0.0.1";
const char * localnet3_addrstr = "10.5.63.78";
const char * localnet4_addrstr = "192.168.74.91";

	/* test 2: address manipulation */
bool test_local_address_manipulation()
{
	struct sockaddr_in loopback_addr;
	struct sockaddr_in localnet1_addr;
	struct sockaddr_in localnet2_addr;
	struct sockaddr_in localnet3_addr;
	struct sockaddr_in localnet4_addr;

	std::cerr << "test_local_address_manipulation() NB: This function demonstrates ";
	std::cerr << "the strange behaviour of inet_netof() and inet_network()";
	std::cerr << std::endl;
	std::cerr << "FAILures are not counted in the test!";
	std::cerr << std::endl;

	/* setup some addresses */
	inet_aton(loopback_addrstr, &(loopback_addr.sin_addr));
	inet_aton(localnet1_addrstr, &(localnet1_addr.sin_addr));
	inet_aton(localnet2_addrstr, &(localnet2_addr.sin_addr));
	inet_aton(localnet3_addrstr, &(localnet3_addr.sin_addr));
	inet_aton(localnet4_addrstr, &(localnet4_addr.sin_addr));


	std::cerr << "Loopback  Addr " << rs_inet_ntoa(loopback_addr.sin_addr);
	std::cerr << std::endl;

	std::cerr << "Localnet1 Addr " << rs_inet_ntoa(localnet1_addr.sin_addr);
	std::cerr << std::endl;
	std::cerr << "Localnet2 Addr " << rs_inet_ntoa(localnet2_addr.sin_addr);
	std::cerr << std::endl;
	std::cerr << "Localnet3 Addr " << rs_inet_ntoa(localnet3_addr.sin_addr);
	std::cerr << std::endl;
	std::cerr << "Localnet4 Addr " << rs_inet_ntoa(localnet4_addr.sin_addr);
	std::cerr << std::endl;
	std::cerr << std::endl;

	std::cerr << "Test 1a - networks";
	std::cerr << std::endl;

	struct sockaddr_in addr_ans, addr1, addr2;
	std::string addr_ans_str;

	addr_ans_str = "127.0.0.0";
	inet_aton(addr_ans_str.c_str(), &(addr_ans.sin_addr));
	addr1.sin_addr.s_addr = htonl(inet_netof(loopback_addr.sin_addr));
	addr2.sin_addr.s_addr = htonl(inet_network(loopback_addrstr));

	std::cerr << "Loopback Net(expected): " << addr_ans_str;
	std::cerr << " -> " << rs_inet_ntoa(addr_ans.sin_addr);
	std::cerr << " Net(1):" << rs_inet_ntoa(addr1.sin_addr);
	std::cerr << " Net(2):" << rs_inet_ntoa(addr2.sin_addr);
	std::cerr << std::endl;

        //CHECK( 0 == strcmp(addr_ans_str.c_str(), rs_inet_ntoa(addr1.sin_addr)));
        //CHECK( 0 == strcmp(addr_ans_str.c_str(), rs_inet_ntoa(addr2.sin_addr)));


	addr_ans_str = "192.168.0.0";
	inet_aton(addr_ans_str.c_str(), &(addr_ans.sin_addr));
	addr1.sin_addr.s_addr = htonl(inet_netof(localnet1_addr.sin_addr));
	addr2.sin_addr.s_addr = htonl(inet_network(localnet1_addrstr));

	std::cerr << "Localnet1 Net(expected): " << addr_ans_str;
	std::cerr << " -> " << rs_inet_ntoa(addr_ans.sin_addr);
	std::cerr << " Net(1):" << rs_inet_ntoa(addr1.sin_addr);
	std::cerr << " Net(2):" << rs_inet_ntoa(addr2.sin_addr);
	std::cerr << std::endl;

        //CHECK( 0 == strcmp(addr_ans_str.c_str(), rs_inet_ntoa(addr1.sin_addr)));
        //CHECK( 0 == strcmp(addr_ans_str.c_str(), rs_inet_ntoa(addr2.sin_addr)));

	addr_ans_str = "10.0.0.0";
	inet_aton(addr_ans_str.c_str(), &(addr_ans.sin_addr));
	addr1.sin_addr.s_addr = htonl(inet_netof(localnet2_addr.sin_addr));
	addr2.sin_addr.s_addr = htonl(inet_network(localnet2_addrstr));

	std::cerr << "Localnet2 Net(expected): " << addr_ans_str;
	std::cerr << " -> " << rs_inet_ntoa(addr_ans.sin_addr);
	std::cerr << " Net(1):" << rs_inet_ntoa(addr1.sin_addr);
	std::cerr << " Net(2):" << rs_inet_ntoa(addr2.sin_addr);
	std::cerr << std::endl;


        //CHECK( 0 == strcmp(addr_ans_str.c_str(), rs_inet_ntoa(addr1.sin_addr)));
        //CHECK( 0 == strcmp(addr_ans_str.c_str(), rs_inet_ntoa(addr2.sin_addr)));


	addr_ans_str = "10.0.0.0";
	inet_aton(addr_ans_str.c_str(), &(addr_ans.sin_addr));
	addr1.sin_addr.s_addr = htonl(inet_netof(localnet3_addr.sin_addr));
	addr2.sin_addr.s_addr = htonl(inet_network(localnet3_addrstr));

	std::cerr << "Localnet3 Net(expected): " << addr_ans_str;
	std::cerr << " -> " << rs_inet_ntoa(addr_ans.sin_addr);
	std::cerr << " Net(1):" << rs_inet_ntoa(addr1.sin_addr);
	std::cerr << " Net(2):" << rs_inet_ntoa(addr2.sin_addr);
	std::cerr << std::endl;


        //CHECK( 0 == strcmp(addr_ans_str.c_str(), rs_inet_ntoa(addr1.sin_addr)));
        //CHECK( 0 == strcmp(addr_ans_str.c_str(), rs_inet_ntoa(addr2.sin_addr)));


	addr_ans_str = "192.168.0.0";
	inet_aton(addr_ans_str.c_str(), &(addr_ans.sin_addr));
	addr1.sin_addr.s_addr = htonl(inet_netof(localnet4_addr.sin_addr));
	addr2.sin_addr.s_addr = htonl(inet_network(localnet4_addrstr));

	std::cerr << "Localnet4 Net(expected): " << addr_ans_str;
	std::cerr << " -> " << rs_inet_ntoa(addr_ans.sin_addr);
	std::cerr << " Net(1):" << rs_inet_ntoa(addr1.sin_addr);
	std::cerr << " Net(2):" << rs_inet_ntoa(addr2.sin_addr);
	std::cerr << std::endl;


        //CHECK( 0 == strcmp(addr_ans_str.c_str(), rs_inet_ntoa(addr1.sin_addr)));
        //CHECK( 0 == strcmp(addr_ans_str.c_str(), rs_inet_ntoa(addr2.sin_addr)));

        REPORT("Test Local Address Manipulation");
	return true;
}



#if 0

std::ostream &showSocketError(std::ostream &out);

std::string socket_errorType(int err);
int sockaddr_cmp(struct sockaddr_in &addr1, struct sockaddr_in &addr2 );
int inaddr_cmp(struct sockaddr_in addr1, struct sockaddr_in addr1 );
int inaddr_cmp(struct sockaddr_in addr1, unsigned long);

std::list<std::string> getLocalInterfaces(); // returns all possible addrs.
bool    isExternalNet(struct in_addr *addr); // if Valid & is not Private or Loopback.
bool	isPrivateNet(struct in_addr *addr); // if inside 10.0.0.0 or 
						// other then firewalled.
bool	isLoopbackNet(struct in_addr *addr); 
bool	sameNet(struct in_addr *addr, struct in_addr *addr2); 
bool	isValidNet(struct in_addr *addr);

	// checks (addr1 & 255.255.255.0) == (addr2 & 255.255.255.0)
bool    isSameSubnet(struct in_addr *addr1, struct in_addr *addr2);
					

struct in_addr getPreferredInterface(); // returns best addr.

in_addr_t pqi_inet_netof(struct in_addr addr); // our implementation.

bool LookupDNSAddr(std::string name, struct sockaddr_in &addr);

/* universal socket interface */

int unix_close(int sockfd);
int unix_socket(int domain, int type, int protocol);
int unix_fcntl_nonblock(int sockfd);
int unix_connect(int sockfd, const struct sockaddr *serv_addr, socklen_t addrlen);
int unix_getsockopt_error(int sockfd, int *err);

#endif


bool test_bind_addr(struct sockaddr_in addr);

bool test_address_listen()
{
	struct sockaddr_in addr1, addr2, addr3;

	sockaddr_clear(&addr1);
	addr1.sin_family = AF_INET;
	inet_aton(loopback_addrstr, &(addr1.sin_addr));
	addr1.sin_port = htons(12345);

	sockaddr_clear(&addr2);
	addr2.sin_family = AF_INET;
	getPreferredInterface(addr2.sin_addr); // returns best addr.
	addr2.sin_port = htons(13245);

	sockaddr_clear(&addr3);
	addr3.sin_family = AF_INET;
	getPreferredInterface(addr3.sin_addr); // returns best addr.
	addr3.sin_port = htons(23451);

	/* test bind to loopback, and preferred interfaces */
	test_bind_addr(addr1);
	test_bind_addr(addr2);
	test_bind_addr(addr3);



        REPORT("Test Address Listen");
	return true;
}

bool test_bind_addr(struct sockaddr_in addr)
{

	int err;

        std::cerr << "test_bind_addr()";
        std::cerr << std::endl;

        std::cerr << "\tAddress Family: " << (int) addr.sin_family;
        std::cerr << std::endl;
        std::cerr << "\tAddress: " << rs_inet_ntoa(addr.sin_addr);
        std::cerr << std::endl;
        std::cerr << "\tPort: " << ntohs(addr.sin_port);
        std::cerr << std::endl;

	int sockfd = unix_socket(PF_INET, SOCK_STREAM, 0);
        CHECK( 0 < sockfd );
	
        if (0 != (err = bind(sockfd, (struct sockaddr *) &addr, sizeof(addr))))
        {
                std::cerr <<  " Failed to Bind to Local Address!" << std::endl;
                showSocketError(std::cerr);

        	FAILED("Couldn't Bind to socket");

		return false;
        }

        std::cerr <<  " Successfully Bound Socket to Address" << std::endl;
        unix_close(sockfd);
	
	return true;
}