#include <stdio.h>
#include <unistd.h>
#include <stdlib.h>
#include <string.h>
#include <errno.h>
#include <ncurses.h>
#include <pthread.h>
#include <locale.h>

typedef struct
{
	unsigned long long int up_bytes;
	unsigned long long int dn_bytes;
	unsigned long long int up_packs;
	unsigned long long int dn_packs;
}SP;

int get_traffic(SP *sp,char *interface)
{
	char *buf=NULL;
	size_t len;
	FILE *fp;

	if((fp=fopen("/proc/net/dev","rb")) == NULL)
	{
		perror("Read /proc/net/dev Error!");
		return -1;
	}

	while(!feof(fp))
	{
		getline(&buf,&len,fp);
		if(buf == NULL)
			return -1;

		if(strstr(buf,interface))
			break;
		
		free(buf);
		buf=NULL;
	}

	if(feof(fp))
	{
		printf("Can Not Find %s Interface!\n",interface);
		fclose(fp);

		return -1;
	}

	sscanf(buf,"%*[^:]: %lld %lld %*s %*s %*s %*s %*s %*s %lld %lld",
			&sp->dn_bytes,&sp->dn_packs,
			&sp->up_bytes,&sp->up_packs);

	free(buf);
	fclose(fp);
	return 0;
}

void _quit(WINDOW *win)
{
	while(getchar() != 'q')
		continue;

	delwin(win);
	echo();
	endwin();

	exit(0);
}

void help(void)
{
	printf("Use:\n");
	printf("netsp [interface]\n");

	exit(-1);
}

int main(int argc,char **argv)
{
	unsigned long up_times=0;
	unsigned long dn_times=0;
	double up_bytes;
	double dn_bytes;
	double up_sp;
	double dn_sp;
	SP start,end,temp;
	WINDOW *win;
	int x,y;
	pthread_t thread;

	if(argc != 2)
		help();

	setlocale(LC_ALL,"");
	if(get_traffic(&start,argv[1]) == -1)
		return -1;

	initscr();
	curs_set(0);
	noecho();
	if(has_colors())
	{
		start_color();
		use_default_colors();
		init_pair(1,COLOR_CYAN,-1);
		init_pair(2,COLOR_GREEN,-1);
	}

	getmaxyx(stdscr,y,x);
	win=newwin(12,30,y/2-10/2,x/2-30/2);
	refresh();
	wattron(win,COLOR_PAIR(1));
	box(win,0,0);
	wattroff(win,COLOR_PAIR(1));
	wrefresh(win);
	pthread_create(&thread,NULL,(void *)_quit,win);

	while(1)
	{
		get_traffic(&temp,argv[1]);
		usleep(500000);
		//sleep(1);
		++up_times;
		++dn_times;
		get_traffic(&end,argv[1]);

		if(up_times % 10 == 0)
		{
			up_times=1;
			start.up_bytes=temp.up_bytes;
		}

		if(dn_times % 10 == 0)
		{
			dn_times=1;
			start.dn_bytes=temp.dn_bytes;
		}

		if(end.up_bytes-temp.up_bytes == 0)
		{
			up_times=1;
			temp.dn_bytes=start.dn_bytes;
			get_traffic(&start,argv[1]);
			start.dn_bytes=temp.dn_bytes;
		}

		if(end.dn_bytes-temp.dn_bytes == 0)
		{
			dn_times=1;
			temp.up_bytes=start.up_bytes;
			get_traffic(&start,argv[1]);
			start.up_bytes=temp.up_bytes;
		}

		wattron(win,COLOR_PAIR(2));
		up_bytes=(double)end.up_bytes/1024/1024;
		dn_bytes=(double)end.dn_bytes/1024/1024;

		if(up_bytes < 1024)
			mvwprintw(win,2,6,"已上传 %.2f MB",up_bytes);
		else
			mvwprintw(win,2,6,"已上传 %.2f GB",up_bytes/1024);
		if(dn_bytes < 1024)
			mvwprintw(win,3,6,"已下载 %.2f MB",dn_bytes);
		else
			mvwprintw(win,3,6,"已下载 %.2f GB",dn_bytes/1024);

		mvwprintw(win,5,6,"已上传包数量 %ld",end.up_packs);
		mvwprintw(win,6,6,"已下载包数量 %ld",end.dn_packs);

		up_sp=(double)(end.up_bytes-start.up_bytes)/up_times/1024*2;
		dn_sp=(double)(end.dn_bytes-start.dn_bytes)/up_times/1024*2;

		if(up_sp < 1024)
			mvwprintw(win,8,6,"上传速度 %.2f kb/s",up_sp);
		else
			mvwprintw(win,8,6,"上传速度 %.2f mb/s",up_sp/1024);
		if(dn_sp < 1024)
			mvwprintw(win,9,6,"下载速度 %.2f kb/s",dn_sp);
		else
			mvwprintw(win,9,6,"下载速度 %.2f mb/s",dn_sp/1024);

		wattroff(win,COLOR_PAIR(2));
		wrefresh(win);
		wclear(win);
		wattron(win,COLOR_PAIR(1));
		box(win,0,0);
		wattroff(win,COLOR_PAIR(1));
	}

	return 0;
}