Imanol-Mikel Barba Sabariego / PAD · Files

/** @file 
* \brief Fichero de implementación de un cliente
* \author Imanol Barba Sabariego
* \date 13/06/2013
*
* En este fichero se implementa un cliente para poder usar con el servidor creado, usando la clase Socket.
*/
#include "client.h"

using namespace std;

bool connected, finished;

//! Método para terminar el cliente
/*! Este método se usa para terminar el cliente immediatamente en el caso que el servidor cierre la conexión de forma inesperada, capturando
el signal SIGPIPE. */
void exitClient(int signal/*!<Parámetro que captura el signal recibido*/)
{
	if(signal == SIGPIPE)
	{
		cout << "Server connection terminated unexpectedly" << endl << "Exiting" << endl;
		exit(-1);
	}
}

void killThread(thread_args *t_arg)
{
	if(t_arg != 0)
	{
		delete t_arg;
		t_arg = 0;
	}
	pthread_exit(NULL);
}

bool connect(Socket& s)
{
	string host, nick, response;
	int port;
	/*cout << "Hostname: ";
	getline(cin,host);
	cout << "Port: ";
	cin >> port;*/
	cout << "Nickname: ";
	getline(cin,nick);
	host = "localhost";
	port = 3001;

	try
	{
		s.Create();
		s.Connect(host,port);
		cout << "Connected" << endl;
		connected = true;
		s << nick;
		s >> response;
		while(response == "CHATNICKINVALID")
		{
			cout << "Spaces not allowed in nicknames, please enter another nickname: ";
			getline(cin,nick);
			s << nick;
			s >> response;
		}
		while(response == "CHATNICKEXIST")
		{
			cout << "Nickname in use, please enter another nickname: ";
			getline(cin,nick);
			s << nick;
			s >> response;
		}
		while(response == "CHATFULL")
		{
			cout << "Chatroom is full, please wait..." << endl;
			s >> response;
		}
		if(response != "CHATOK")
		{
			cout << "Error: " << response << endl;
			return false;
		}
	}
	catch(SocketException& e)
	{
		cout << e.description() << endl;
		exit(-1);
	}
	return true;
}

void* sendThread(void* args)
{
	string send;
	struct thread_args *t_arg = (struct thread_args*)args;
	while(connected)
	{
		cout << "> ";
		getline(cin,send);
		if(cin.eof())
		{
			send = "/exit";
		}
		try
		{
			*(t_arg->s) << send;
			if(send == "/disconnect" || send == "/exit")
			{
				break;
			}
		}
		catch(SocketException& e)
		{
			cout << e.description() << endl;
		}
	}
	killThread(t_arg);
}

void* recvThread(void* args)
{
	string recv;
	struct thread_args *t_arg = (struct thread_args*)args;
	while(true)
	{
		try
		{
			*(t_arg->s) >> recv;
		}
		catch(SocketException &e)
		{
			connected = false;
			cout << e.description() << endl;
			(t_arg->s)->Close();
			break;
		}
		if(recv == "DISC_OK")
		{
			cout << "Disconnecting" << endl;
	    	connected = false;
	    	pthread_cond_signal(t_arg->condition);
	    	break;
		}
		else if(recv == "EXIT_OK")
		{
			cout << "Exiting" << endl;
			connected = false;
			finished = true;
			pthread_cond_signal(t_arg->condition);
			break;
		}
		else
		{
			cout << recv << endl;
		}
	}
	killThread(t_arg);
}

//! Método principal del cliente
/*! Este método inicializa el Socket, establece la conexión y realiza las acciones que se le hayan programado para comunicarse con el servidor.*/
int main()
{
	signal(SIGPIPE, exitClient);
	Socket s;
	connected = finished = false;
	pthread_mutex_t mutex;
	pthread_mutex_init(&mutex,0);
	pthread_cond_t condition;
	pthread_cond_init(&condition,0);
	thread_args *sArgs = new thread_args;
	thread_args *rArgs = new thread_args;
	pthread_t recv, send;
	pthread_attr_t attr;
	pthread_attr_init(&attr);
    pthread_attr_setdetachstate(&attr, PTHREAD_CREATE_JOINABLE);
	while (!finished)
	{
		if(!connect(s))
		{
			exit(-1);
		}
		pthread_mutex_lock(&mutex);
		sArgs->mutex = &mutex;
		sArgs->condition = &condition;
		sArgs->s = &s;

		rArgs->mutex = &mutex;
		rArgs->condition = &condition;
		rArgs->s = &s;

		pthread_create(&send,&attr,sendThread,(void *)sArgs);
		pthread_create(&recv,&attr,recvThread,(void *)rArgs);

		while(connected)
		{
			pthread_cond_wait(&condition,&mutex);
		}
		pthread_mutex_unlock(&mutex);
		pthread_join(recv,NULL);
		pthread_join(send,NULL);
		s.Close();
	}
	pthread_cond_destroy(&condition);
	pthread_mutex_destroy(&mutex);
}

/* TODO
 *
 * CTRL+C
 */