#include <stdio.h>
#include <stdlib.h>
#include <network.h>

static int set_sock_opts(int sock)
{
	struct timeval tv;
	int d = 1;

	memset(&tv, 0, sizeof(tv));
#if 0
	tv.tv_sec = 2;
	net_setsockopt(sock, SOL_SOCKET, SO_RCVTIMEO,
			&tv, sizeof(struct timeval));
	net_setsockopt(sock, SOL_SOCKET, SO_SNDTIMEO,
			&tv, sizeof(struct timeval));
#endif
	return net_setsockopt(sock,SOL_SOCKET,SO_REUSEADDR, &d, sizeof(int));
}

bool Network::InitSockaddr (struct sockaddr_in *name,
		const char *hostname, uint16_t port)
{
	struct hostent *hostinfo = net_gethostbyname ((char*)hostname);

	if (hostinfo == NULL)
	{
		fprintf (stderr, "Unknown host %s.\n", hostname);
		return false;
	}

	memset(name, 0, sizeof(name));
	name->sin_family = AF_INET;
	name->sin_port = htons (port);
	name->sin_len = 8; // sizeof(struct sockaddr_in);
	name->sin_addr.s_addr = *(uint32*)hostinfo->h_addr_list[0];

	return true;
}

bool Network::InitSocket(const char *remote_host, int port)
{
	/* Create the socket. */
	this->sock = net_socket (PF_INET, SOCK_DGRAM, 0);
	if (this->sock < 0)
	{
		fprintf (stderr, "Could not init socket. Failed with %d\n", this->sock);
		sleep(5);
		return false;
	}
	int _true = false;
	if (net_ioctl (this->sock, FIONBIO, (char *)(&_true)) < 0)
		fprintf(stderr, "Could not set FIONBIO\n");

	set_sock_opts(this->sock);

	/* Connect to the server. */
	this->InitSockaddr(&this->connection_addr, remote_host, port);
	fprintf(stderr, "Bajning: %d\n",
			net_bind(this->sock,(struct sockaddr *)&this->connection_addr,sizeof(struct sockaddr)));
	return true;
}

bool Network::ReceiveData(void *dst, int sock, size_t sz)
{
	size_t received_sz = 0;

	while (received_sz < sz)
	{
		int v = net_read(sock, dst, sz);

		if (v < 0)
			return false;
		received_sz += v; 
	}
	this->traffic += received_sz;

	return sz > 0;
}

ssize_t Network::ReceiveFrom(void *dst, int sock, size_t sz,
		struct sockaddr_in *from)
{
	return net_recv(sock, dst, sz, 0);
}

ssize_t Network::SendTo(void *src, int sock, size_t sz, struct sockaddr_in *to)
{
	socklen_t to_sz = to->sin_len;

	assert(to);
	return net_sendto(sock, src, sz, 0, (struct sockaddr*)to, to_sz);
}

bool Network::SendData(void *src, int sock, size_t sz)
{
	size_t sent_sz = 0;

	while (sent_sz < sz)
	{
		int v = net_write(sock, (void*)src, sz);

		if (v < 0)
			return false;
		sent_sz += v;
	}

	return true;
}

bool Network::Select(int sock, struct timeval *tv)
{
	struct pollsd sds;
	int v;

	sds.socket = sock;
	sds.events = POLLIN;
	sds.revents = 0;

	v = net_poll(&sds, 1, tv->tv_sec * 1000 + tv->tv_usec / 1000);

	return v > 0; 
}

void Network::CloseSocket()
{
#define SHUT_RDWR 2
	net_shutdown(this->sock, 2);
	net_close(this->sock);
}

void Network::InitNetwork()
{
        char myIP[16];

        /* From Snes9x-gx */
        while (net_init() == -EAGAIN);

        if (if_config(myIP, NULL, NULL, true) < 0)
        {
        	fprintf(stderr, "\n\n\nError getting IP address via DHCP.\n");
        	sleep(2);
		exit(1);
        }
}

void Network::ShutdownNetwork()
{
}