Blame view

src/client.cpp 2.59 KB
Imanol-Mikel Barba Sabariego authored
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
/** @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 "Socket.h"
#include <iostream>
#include "SocketException.h"
#include <sstream>
#include <signal.h>

#include <sys/time.h>

using namespace std;

//! 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*/)
{
	cout << "Server connection terminated unexpectedly" << endl << "Exiting" << endl;
	exit(-1);
}

//! 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;
	string send, recv, host;
	int port;
	s.Create();
	cout << "Created socket" << endl;
	cout << "Hostname: ";
	cin >> host;
	cout << "Port: ";
	cin >> port;
	cin.ignore();
	try
	{
		s.Connect(host,port);
	}
	catch(SocketException& e)
	{
		cout << e.description() << endl;
		return -1;
	}
	cout << "Connected" << endl;
	while(true)
	{
		cout << "> ";
		getline(cin,send);
		if(cin.eof())
		{
			send = "exit";
		}
		try
		{
			s << send;
			if(send == "exit")
			{
				s >> recv;
				if(recv == "OK")
				{
					cout << "Exiting" << endl;
					s.Close();
	           	 	cout << "Socket closed" << endl;
	            	return 0;
				}
			}
			else if(send == "test")
			{
				string data = "", answer;
				double size;
				double start, duration;
				cout << "Size in MB: ";
				cin >> size;
				cin.ignore();
				for(long int i = 0; i < size*1e6; i++ )
				{
					data += (char)( 65 + i % 26);
				}
				cout << "Data generated, commencing transfer" << endl;
				struct timeval  st, ed;
				gettimeofday(&st, NULL);
				s << data;
				cout << "Data sent" << endl;
				s >> answer;
				gettimeofday(&ed, NULL);
				start = (st.tv_sec) + (st.tv_usec) / 1e6;
				duration = ((ed.tv_sec) + (ed.tv_usec) / 1e6) - start;
				if(answer == "ACK")
				{
					cout << "Transferred " << size << " MB  in " << duration << " seconds" << endl;
					cout << "Data rate: " << size/duration << " MB/s" <<  endl;
				}
			}
			else
			{
				s >> recv;
				cout << "Received: " << recv << endl;
			}
		}
		catch(SocketException& e)
		{
			cout << e.description() << endl;
			cout << "Exiting" << endl;
			s.Close();
       	 	cout << "Socket closed" << endl;
        	return -1;
		}
	}
}