/*
	CPP Final Projekt
	JMVermillion

	chat.cpp

	If they can do it, so can I. 24 hours!

   All of the WinSock routines were blatantly butchered from
   Open-Source code. Thanks to the authors. :)
*/

#include <iostream>
using namespace std;

#include "chat.h"

CComm::CComm()
{
	ListenSocket = INVALID_SOCKET;	// Start with INVALID socket. Ease of use.

	/*
		Check to see if it's a Windows machine. If not, don't worry.
	*/
	#ifdef _WIN32 // Why does Windows *need* WinSock? Who knows...
		WORD VersionRequested = MAKEWORD(1,1);
		WSADATA wsaData;
		WSAStartup(VersionRequested, &wsaData);	// starts the Winsock service
		if ( wsaData.wVersion != VersionRequested )
		{
			cout << "Error Code \"MICROSOFT\": Wrong version or WinSock not loaded!\n";
		}
	#endif
}

CComm::~CComm()
{
	if ( ListenSocket != INVALID_SOCKET )
		closesocket( ListenSocket );	// Was there a socket? Close!

	#ifdef _WIN32	  // Windows is the only crappy OS that uses WinSock!
		WSACleanup(); // WinSock cleanup routine.
	#endif
}

bool CComm::SendMsg( char *Msg, int Len, char *host, short port )
{
	signed int Sent;
	hostent *hostdata;
	if ( atoi(host) )	// Is it an IP, or not?
	{
		u_long ip = inet_addr( host );
		hostdata = gethostbyaddr( (char *)&ip, sizeof(ip), PF_INET );
	}
	else	// Don't we love DNS!? This also does WINS resolution!
	{
		hostdata = gethostbyname( host );
	}

	if ( !hostdata )
	{
		cout << "Error Code 1227: We couldn't find your host. Sorry. :)\n";
		return false;
	}

	sockaddr_in dest;	// What's our destination?
	dest.sin_family = PF_INET;
	dest.sin_addr = *(in_addr *)(hostdata->h_addr_list[0]);
	dest.sin_port = htons( port );
	cout << "Message sent to: " << inet_ntoa(dest.sin_addr) << ":"
		  << ntohs(dest.sin_port) << "\n";
	Sent = sendto(ListenSocket, Msg, Len, 0, (sockaddr *)&dest, sizeof(sockaddr_in));

	if ( Sent != Len )
	{
		cout << "Error Code 666: We couldn't send that there UDP packet on ListenPort. :-P\n";
		return false;
	}

	return true;
}

void *CComm::ListenThread( void *data )
{
	char buf[4096];
	CComm *Comm = (CComm *)data;
	int len = sizeof(Comm->client);

	while(1)	// Recursion? I think so!
	{
		int result = recvfrom( Comm->ListenSocket, buf, sizeof(buf)-1, 0, (sockaddr *)&Comm->client, (socklen_t *)&len);
		if ( result > 0 )
		{
			buf[result] = 0;
			cout << "Message rcvd from: " << inet_ntoa(Comm->client.sin_addr)
				  << ":" << ntohs(Comm->client.sin_port) << "\n";
			cout << "]--[\n " << buf << "]--[\n";
		}
	}	// end infinite loop
}

bool CComm::Listen( int PortNum )
{
	ListenSocket = socket(PF_INET, SOCK_DGRAM, 0);
	if ( ListenSocket == INVALID_SOCKET )
	{
		cout << "Error Code 123: We failed to initialize a listen socket.\n";
		return false;
	}

	srv.sin_family = PF_INET;
	srv.sin_addr.s_addr = htonl( INADDR_ANY );	// any address
	srv.sin_port = htons( PortNum );

	if ( bind( ListenSocket, (struct sockaddr *)&srv, sizeof(srv)) != 0 )
	{
		cout << "Error Code 2001: Sorry Dave, we couldn't bind the listen socket.\n";
		closesocket( ListenSocket );
		return false;
	}

	int ThreadID;	// the listening thread's handle

	#ifdef _WIN32
   /*
   	Always extra routines for Windows. Why!?
   	Well, the 'net said I should do this to make it work
      properly, and the 'net is never wrong.
   */
		DWORD thread;
		ThreadID = (int)CreateThread(NULL, 0, (LPTHREAD_START_ROUTINE)(CComm::ListenThread), (void *)this, 0, &thread);
		ThreadID = ThreadID ? 0 : 1;	// reverse the value for Windows

	/*
		Are we NOT running Windows? Who cares!?
	*/
	#else
		pthread_t thread;
		ThreadID = pthread_create(&thread, 0, CComm::ListenThread, (void *)this);
	#endif

	if(ThreadID)	// Did we fail to create a listen thread?
	{
		cout << "Error Code 911: Did we fail? Of course we did! No listen thread for you!\n";
		return false;
	}
	else
		return true;
}

/*
	CPP Final Projekt
	JMVermillion

	chat.h

	Well? Who's laughing now!?

    Again, WinSock crap butchered from Open-Source code. Thanks! :)

    Also, OS-Specific routines either taken from John Gray(UNIX) or random
    internet tutorials(WINDOWS). Others noted in comments.
*/

#ifndef _CHAT_H_
#define _CHAT_H_

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

#ifndef _WIN32	// Are we UNIX? Huzzah! Use these includes.
	#include <sys/types.h>
	#include <sys/socket.h>
	#include <netinet/in.h>
	#include <unistd.h>
	#include <netdb.h>
	#include <arpa/inet.h>
	#define SOCKET			int
	#define INVALID_SOCKET	-1
	#define closesocket		close
	#include <pthread.h>
#else	// Are we using Windows? WHY!?!?!?
	#include <winsock.h>
	#define socklen_t	int
	/*
		Library stuff that the internet said I needed.
		Always trust the 'net.[tm]
	*/
	#if defined(_MSC_VER)
		#pragma comment(lib, "wsock32.lib")
	#elif defined(__BORLANDC__)
		#pragma(lib, "mswsock.lib")
	#endif
#endif

class CComm
{
private:
	static void *ListenThread(void *data);
	SOCKET ListenSocket;	// Listen Socket?
	sockaddr_in srv;		// Address for listen socket?
	sockaddr_in client;	    // Where did we get our message?

public:
	/*
		CComm default constructor
	*/
	CComm();

	/*
		CComm destructor
	*/
	~CComm();

	/*
		Send messages!
	*/
	bool SendMsg(char *Msg, int Len, char *host, short port);

    /*
   	    Receive messages!
	*/
	bool Listen(int PortNum);

};

#endif

int main(int argc, char *argv[]) {
    return 0;
}