client.cpp
3.5 KB
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
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
/** @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>
//! Ruta a la llave privada
/*! Ruta relativa o absoluta a la llave privada RSA, si no existe o es inválida, se creará una nueva en esa localización. */
#define PRIVATEKEY "bin/private_client.key"
//! Ruta a la llave pública
/*! Ruta relativa o absoluta a la llave pública RSA, si no existe o es inválida, se creará una nueva en esa localización. */
#define PUBLICKEY "bin/public_client.key"
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 para intercambiar llaves de encriptación
/*! Este método sirve para intercambiar las llaves de encriptación con el servidor al otro lado del socket. */
void exchangeKeys(Socket &s/*!<Socket cliente que intercambia las llaves con el servidor*/)
{
if(!s.receivePublicKey() || !s.sendPublicKey())
{
cout << "Error exchanging keys" << endl << "Exiting" << endl;
s.Close();
cout << "Socket closed" << endl;
exit(-1);
}
s.recvAES();
s.sendAES();
}
//! 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;
s.loadKeys(PUBLICKEY,PRIVATEKEY);
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;
exchangeKeys(s);
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;
}
}
}