//    SERVER TCP PROGRAM
// revised and tidied up by
// J.W. Atwood
// 1999 June 30
// There is still some leftover trash in this code.

/* send and receive codes between client and server */
/* This is your basic WINSOCK shell */
#pragma comment( linker, "/defaultlib:ws2_32.lib" )
#include <winsock2.h>
#include <ws2tcpip.h>

#include <winsock.h>
#include <iostream>
#include <windows.h>

#include <fstream>

using namespace std;

//port data types

#define REQUEST_PORT 0x5001

int port=REQUEST_PORT;

//socket data types
SOCKET s;

SOCKET s1;
SOCKADDR_IN sa;      // filled by bind
SOCKADDR_IN sa1;     // fill with server info, IP, port
SOCKADDR_IN sa_in;
union {struct sockaddr generic;
struct sockaddr_in ca_in;}ca;

int calen=sizeof(ca); 

//buffer data types

char szbuffer[1024];

char *buffer;
int ibufferlen;
int ibytesrecv;

int ibytessent;

//host data types

char localhost[11];

HOSTENT *hp;

//wait variables

int nsa1;

int r,infds=1, outfds=0;

#define UTIMER 300000
#define STIMER 0 
struct timeval timeout;

const struct timeval *tp=&timeout;

fd_set readfds;

//others

HANDLE test;

DWORD dwtest;

//packet

struct Packet {
	char header[8];
	char sequence[1];
	char message[1016];
};

//reference for used structures

/*  * Host structure

struct  hostent {
char    FAR * h_name;             official name of host *
char    FAR * FAR * h_aliases;    alias list *
short   h_addrtype;               host address type *
short   h_length;                 length of address *
char    FAR * FAR * h_addr_list;  list of addresses *
#define h_addr  h_addr_list[0]            address, for backward compat *
};

* Socket address structure

struct sockaddr_in {
short   sin_family;
u_short sin_port;
struct  in_addr sin_addr;
char    sin_zero[8];
}; */

int main(void){

	WSADATA wsadata;

	timeout.tv_sec=STIMER;
	timeout.tv_usec=UTIMER;

	//Blank packet
	Packet packet;
	memset(&packet,0,1024*sizeof(char));

	try{        		 
		if (WSAStartup(0x0202,&wsadata)!=0){  
			cout<<"Error in starting WSAStartup()\n";
		}else{
			buffer="WSAStartup was suuccessful\n";   
			WriteFile(test,buffer,sizeof(buffer),&dwtest,NULL); 

			/* display the wsadata structure */
			cout<< endl
				<< "wsadata.wVersion "       << wsadata.wVersion       << endl
				<< "wsadata.wHighVersion "   << wsadata.wHighVersion   << endl
				<< "wsadata.szDescription "  << wsadata.szDescription  << endl
				<< "wsadata.szSystemStatus " << wsadata.szSystemStatus << endl
				<< "wsadata.iMaxSockets "    << wsadata.iMaxSockets    << endl
				<< "wsadata.iMaxUdpDg "      << wsadata.iMaxUdpDg      << endl;
		}  

		//Display info of local host

		gethostname(localhost,10);
		cout<<"hostname: "<<localhost<< endl;

		if((hp=gethostbyname(localhost)) == NULL) {
			cout << "gethostbyname() cannot get local host info?"
				<< WSAGetLastError() << endl; 
			exit(1);
		}

		//Create the server socket
		if((s = socket(AF_INET,SOCK_DGRAM,0))==INVALID_SOCKET) 
			throw "can't initialize socket";
		// For UDP protocol replace SOCK_STREAM with SOCK_DGRAM 


		//Fill-in Server Port and Address info.
		memset(&sa,0,sizeof(sa));
		sa.sin_family = AF_INET;
		sa.sin_port = htons(port);
		sa.sin_addr.s_addr = htonl(INADDR_ANY);

		//Bind the server port

		if (bind(s,(LPSOCKADDR)&sa,sizeof(sa)) == SOCKET_ERROR)
			throw "can't bind the socket";
		cout << "Bind was successful" << endl;

		//Successfull bind, now listen for client requests.

		/*if(listen(s,10) == SOCKET_ERROR)
			throw "couldn't  set up listen on socket";
		else cout << "Listen was successful" << endl;*/

		//wait loop

		while(1)

		{
			FD_ZERO(&readfds);
			FD_SET(s,&readfds);  //always check the listener

			if(!(outfds=select(1,&readfds,NULL,NULL,tp))) {} //timed out, return

			int size = sizeof(sa_in);

			if (outfds == 1) {//receive frame
				if(ibytesrecv = recvfrom(s, (char*)& packet, sizeof(packet),0, (struct sockaddr*)&sa_in, &size) == SOCKET_ERROR)
					throw "Receive error in server program\n";
				cout << "yay?" << endl;
			}
			if (ibytesrecv != 0)
				cout << ibytesrecv << endl;

			/*else if (outfds == SOCKET_ERROR) throw "failure in Select";

			else if (FD_ISSET(s,&readfds))  cout << "got a connection request" << endl; 

			//Found a connection request, try to accept. 

			if((s1=accept(s,&ca.generic,&calen))==INVALID_SOCKET)
				throw "Couldn't accept connection\n";

			//Connection request accepted.
			cout<<"accepted connection from "<<inet_ntoa(ca.ca_in.sin_addr)<<":"
				<<hex<<htons(ca.ca_in.sin_port)<<endl;

			//Fill in packet from accepted request.
			if((ibytesrecv = recv(s1,szbuffer,1024,0)) == SOCKET_ERROR)
				throw "Receive error in server program\n";*/

			//Get file name
			/*memset(szbuffer,0,1024*sizeof(char));
			if((ibytesrecv = recv(s1,szbuffer,1024,0)) == SOCKET_ERROR)
				throw "Receive error in server program\n";;
			memset(&packet,0,1024*sizeof(char));
			memcpy(packet.header,szbuffer,8*sizeof(char));
			memcpy(packet.message,szbuffer+8*sizeof(char),1016*sizeof(char));
			cout << packet.header << "\n";
			cout << packet.message << "\n";

			string header(packet.header);
			char location[1500];

			if(header.compare("RETR") == 0) {
				sprintf_s(location, packet.message);
				cout << "RETR: " << location << "\n";
				ifstream file (location, ios::in|ios::binary|ios::ate);
				ifstream::pos_type size;
				if(file.is_open()) {
					size = file.tellg();
					memset(packet.message,0,1016*sizeof(char));
					memset(szbuffer+8*sizeof(char),0,1016*sizeof(char));
					itoa((int)size,packet.message,10);
					memcpy(szbuffer,packet.header,8*sizeof(char));
					memcpy(szbuffer+8*sizeof(char),packet.message,1016*sizeof(char));
					//Send size
					if((ibytessent = send(s1,szbuffer,1024,0))==SOCKET_ERROR)
						throw "error in send in server program\n";
					cout << "Size: " << atoi(packet.message) << "bytes\n";
					file.seekg(0, ios::beg);
					memset(packet.message,0,1016*sizeof(char));
					memset(szbuffer+8*sizeof(char),0,1016*sizeof(char));
					while(file.read(packet.message,1016*sizeof(char))) {
						memcpy(szbuffer,packet.header,8*sizeof(char));
						memcpy(szbuffer+8*sizeof(char),packet.message,1016*sizeof(char));
						//Send packet
						if((ibytessent = send(s1,szbuffer,1024,0))==SOCKET_ERROR)
							throw "error in send in server program\n";
						memset(packet.message,0,1016*sizeof(char));
						memset(szbuffer+8*sizeof(char),0,1016*sizeof(char));
						size -= 1016;
					}
					memcpy(szbuffer,packet.header,8*sizeof(char));
					memcpy(szbuffer+8*sizeof(char),packet.message,size*sizeof(char));
					//Send last packet                             
					if((ibytessent = send(s1,szbuffer,1024,0))==SOCKET_ERROR)
						throw "error in send in server program\n";
					memset(packet.message,0,1016*sizeof(char));
					memset(szbuffer+8*sizeof(char),0,1016*sizeof(char));
					file.close();
					cout << "Transfer Complete\n";
				}
				else {
					memset(packet.message,0,1016*sizeof(char));
					memset(szbuffer+8*sizeof(char),0,1016*sizeof(char));
					//Send packet
					memcpy(szbuffer,packet.header,8*sizeof(char));
					memcpy(szbuffer+8*sizeof(char),packet.message,size*sizeof(char));
					if((ibytessent = send(s1,szbuffer,1024,0))==SOCKET_ERROR)
						throw "error in send in server program\n";
					cout << "No such file.\n";
					
				}
			} else if (header.compare("STOR") == 0) {
				sprintf_s(location, packet.message);
				ofstream file (location, ios::out);
				file.close();
				file.open(location, ios::out|ios::binary|ios::trunc);
				memset(packet.message,0,1016*sizeof(char));
				memset(szbuffer+8*sizeof(char),0,1016*sizeof(char));
				//Get file size
				if((ibytesrecv = recv(s1,szbuffer,1024,0)) == SOCKET_ERROR)
					throw "Receive error in server program\n";
				memcpy(packet.message, szbuffer+8*sizeof(char), 1016*sizeof(char));
				ofstream::pos_type size;
				cout << "Size: " << atoi(packet.message) << "bytes\n";;
				for (size = atoi(packet.message); size > 1016*sizeof(char); size -= 1016) {
					memset(packet.message,0,1016*sizeof(char));
					memset(szbuffer+8*sizeof(char),0,1016*sizeof(char));
					//Get packet
					if((ibytesrecv = recv(s1,szbuffer,1024,0)) == SOCKET_ERROR)
						throw "Receive error in server program\n";
					memcpy(packet.message, szbuffer+8*sizeof(char), 1016*sizeof(char));
					file.write(packet.message, 1016*sizeof(char));
				}
				memset(packet.message,0,1016*sizeof(char));
				memset(szbuffer+8*sizeof(char),0,1016*sizeof(char));
				//Get last packet
				if((ibytesrecv = recv(s1,szbuffer,1024,0)) == SOCKET_ERROR)
					throw "Receive error in server program\n";
				memcpy(packet.message, szbuffer+8*sizeof(char), size*sizeof(char));
				file.write(packet.message, size*sizeof(char));
				file.close();
				cout << "Transfer Complete\n";
			} else {

			}*/
		}//wait loop

	} //try loop

	//Display needed error message.

	catch(char* str) { cerr<<str<<WSAGetLastError()<<endl;}

	//close Client socket
	closesocket(s1);		

	//close server socket
	closesocket(s);

	/* When done uninstall winsock.dll (WSACleanup()) and exit */ 
	WSACleanup();
	system("pause");
	return 0;
}