/*
 * Copyright (c) 2001,2002 Florian Schulze.
 * All rights reserved.
 * 
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the distribution.
 * 3. Neither the name of the authors nor the names of the contributors
 *    may be used to endorse or promote products derived from this software
 *    without specific prior written permission.
 * 
 * THIS SOFTWARE IS PROVIDED BY THE AUTHORS AND CONTRIBUTORS ``AS IS'' AND
 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
 * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHORS OR CONTRIBUTORS BE LIABLE
 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
 * SUCH DAMAGE.
 *
 * test.c - This file is part of lwIPtest
 *
 */

#include <stdio.h>
#include <stdarg.h>
#include <time.h>
#include <string.h>
#include <conio.h>

#include "lwip/debug.h"
#include "lwip/mem.h"
#include "lwip/memp.h"
#include "lwip/sys.h"

#include "lwip/stats.h"

#include "lwip/tcpip.h"

#include "netif/loopif.h"

#include "arch/perf.h"

//#include "httpd.h"
//#include "ftpd.h"
//#include "fs.h"

void ethernetif_init(struct netif *netif);
int init_adapter(int adapter_num);
void shutdown_adapter(void);
void update_adapter(void);

int dbg_printf(const char *fmt, ...)
{
	va_list v;
	int r;

	va_start(v, fmt);
	r = vfprintf(stderr,fmt, v);
	va_end(v);
	return r;
}

static err_t netio_recv(void *arg, struct tcp_pcb *pcb, struct pbuf *p, err_t err)
{
	printf("TCP packet received. Length = %d\n", p->tot_len);
	if (err == ERR_OK && p != NULL)
	{
		printf("TCP packet received. Length = %d\n", p->tot_len);
		tcp_recved(pcb, p->tot_len);
		pbuf_free(p);
	}
	else
		pbuf_free(p);

	if (err == ERR_OK && p == NULL)
	{
		tcp_arg(pcb, NULL);
		tcp_sent(pcb, NULL);
		tcp_recv(pcb, NULL);
		tcp_close(pcb);
	}

	return ERR_OK;
}

void tcp_accept (struct tcp_pcb *pcb, err_t (* accept)(void *arg, struct tcp_pcb *newpcb, err_t err));
err_t tcp_connect (struct tcp_pcb *pcb, struct ip_addr *ipaddr, u16_t port, err_t (* connected)(void *arg, struct tcp_pcb *tpcb, err_t err));

static err_t netio_accept(void *arg, struct tcp_pcb *pcb, err_t err)
{
	tcp_arg(pcb, NULL);
	tcp_sent(pcb, NULL);
	tcp_recv(pcb, netio_recv);
	return ERR_OK;
}

static err_t netio_connected(void *arg, struct tcp_pcb *pcb, err_t err)
{
	tcp_arg(pcb, NULL);
	tcp_sent(pcb, NULL);
	tcp_recv(pcb, netio_recv);
	printf("Connection Successfully Established.\n");
	return ERR_OK;
}


void netio_init(void)
{
	struct tcp_pcb *pcb;

	pcb = tcp_new();
	tcp_bind(pcb, IP_ADDR_ANY, 18767);

	pcb = tcp_listen(pcb);
	tcp_accept(pcb, netio_accept);
}

struct tcp_pcb *pcb;

err_t netio_connect(struct ip_addr* remote_ipaddr, int remote_port)
{
	err_t err = ERR_OK;

	pcb = tcp_new();
	tcp_bind(pcb, IP_ADDR_ANY, 18767);
	err = tcp_connect(pcb, remote_ipaddr, remote_port, netio_connected);

	return err;
}

struct netif netif;

void main_loop()
{
	struct ip_addr ipaddr, netmask, gw;
	struct ip_addr remote_ipaddr;
	int remote_port;
	int last_time;
	int timer1;
	int timer2;
	int done;
	int size = 0;
	char data[1500] = {0};
	FILE* finput = NULL;
	FILE* foutput = NULL;
	
	finput = fopen("D://story.dat", "rb");
	if(finput == NULL)
	{
		printf("Failed to open input file.\n");
		return;
	}

	foutput = fopen("D://loc_out.dat", "wb");
	if(foutput == NULL)
	{
		printf("Failed to open output file.\n");
		return;
	}
	
	IP4_ADDR(&gw, 192,168,5,228);
	IP4_ADDR(&ipaddr, 192,168,5,228);
	IP4_ADDR(&netmask, 255,255,255,0);
	IP4_ADDR(&remote_ipaddr, 192,168,5,186);
	remote_port = 4343;

	if (init_adapter(2) != 0)
		return;

	netif_set_default(netif_add(&netif, &ipaddr, &netmask, &gw, NULL, ethernetif_init,
		ip_input));
	netif_set_up(&netif);

	/*
	IP4_ADDR(&gw, 127,0,0,1);
	IP4_ADDR(&ipaddr, 127,0,0,1);
	IP4_ADDR(&netmask, 255,0,0,0);
	
	netif_add(&ipaddr, &netmask, &gw, loopif_init,
		ip_input);
	*/

	tcp_init();
	udp_init();
	ip_init();

	//httpd_init();
	//netio_init();
	if(netio_connect(&remote_ipaddr, remote_port) != ERR_OK)
	{
		printf("cannot connect to server.\n");
		return;
	}
	//ftpd_init();

	last_time=clock();
	timer1=0;
	timer2=0;
	done=0;
	
	printf("TCP_SND_BUF = %d\n", TCP_SND_BUF);
	
	while (!done)
	{
		int cur_time;
		int time_diff;

		cur_time=clock();
		time_diff=cur_time-last_time;
		if (time_diff>0)
		{
			last_time=cur_time;
			timer1+=time_diff;
			timer2+=time_diff;
		}

		if (timer1>10)
		{
			tcp_fasttmr();
			timer1=0;
		}

		if (timer2>45)
		{
			tcp_slowtmr();
			timer2=0;
			done=_kbhit();
		}

		update_adapter();

		size = fread(data, sizeof(char), 1400, finput);
		if(ferror(finput))
		{
			printf("Error Occured while reading input file.\n");
			fclose(finput);
			return;
		}
		if(feof(finput))
		{
			printf("EOF reached.\n");
			continue;
		}
		tcp_write(pcb, data, size, 1);
		tcp_output(pcb);
		fwrite(data, sizeof(char), size, foutput);
		memset(data, 0, 1500);
	}

	shutdown_adapter();
}

void bcopy(const void *src, void *dest, int len)
{
  memcpy(dest,src,len);
}

void bzero(void *data, int n)
{
  memset(data,0,n);
}

int main(void)
{
	setvbuf(stdout,NULL,_IONBF,0);
#ifdef PERF
	perf_init("/tmp/lwip.perf");
#endif /* PERF */
#ifdef STATS
	stats_init();
#endif /* STATS */
	sys_init();
	mem_init();
	memp_init();
	pbuf_init();

	//tcpdump_init();

	printf("System initialized.\n");

	main_loop();

	return 0;
}