summaryrefslogtreecommitdiff
path: root/testing
diff options
context:
space:
mode:
Diffstat (limited to 'testing')
-rw-r--r--testing/DHT_sendfiletest.c178
-rw-r--r--testing/Lossless_UDP_testserver.c1
2 files changed, 178 insertions, 1 deletions
diff --git a/testing/DHT_sendfiletest.c b/testing/DHT_sendfiletest.c
new file mode 100644
index 00000000..fd4fd29f
--- /dev/null
+++ b/testing/DHT_sendfiletest.c
@@ -0,0 +1,178 @@
1/* DHT sendfiletest
2 *
3 * Sends the data from a file to another client.
4 * Receives the file data that that client sends us.
5 *
6 * NOTE: this program simulates 33% packet loss.
7 *
8 * Compile with: gcc -O2 -Wall -o test ../core/DHT.c ../core/network.c ../core/Lossless_UDP.c DHT_sendfiletest.c
9 *
10 * Command line arguments are the ip and port of a node (for bootstrapping), the
11 * client_id (32 bytes) of the friend you want to send the data in filename to and
12 * the client_id this node will take.
13 *
14 * Saves all received data to: received.txt
15 *
16 * EX: ./test 127.0.0.1 33445 ABCDEFGHIJKLMNOPQRSTUVWXYZabcdef filename.txt ABCDEFGHIJKLMNOPQRSTUVWXYZabcdeg
17 */
18#include "../core/network.h"
19#include "../core/DHT.h"
20#include "../core/Lossless_UDP.h"
21
22#include <string.h>
23
24//Sleep function (x = milliseconds)
25#ifdef WIN32
26
27#define c_sleep(x) Sleep(1*x)
28
29#else
30#include <unistd.h>
31#include <arpa/inet.h>
32#define c_sleep(x) usleep(1000*x)
33
34#endif
35
36#define PORT 33445
37
38void printip(IP_Port ip_port)
39{
40 printf("\nIP: %u.%u.%u.%u Port: %u\n",ip_port.ip.c[0],ip_port.ip.c[1],ip_port.ip.c[2],ip_port.ip.c[3],ntohs(ip_port.port));
41}
42
43int main(int argc, char *argv[])
44{
45 //memcpy(self_client_id, "qqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqq", 32);
46
47 if (argc < 6) {
48 printf("usage %s ip port client_id(of friend to find ip_port of) filename(of file to send) client_id(ours)\n", argv[0]);
49 exit(0);
50 }
51 addfriend(argv[3]);
52 IP_Port friend_ip;
53 int connection = -1;
54 int inconnection = -1;
55
56 //initialize networking
57 //bind to ip 0.0.0.0:PORT
58 IP ip;
59 ip.i = 0;
60 init_networking(ip, PORT);
61
62 memcpy(self_client_id, argv[5], 32);
63
64
65 perror("Initialization");
66 IP_Port bootstrap_ip_port;
67 bootstrap_ip_port.port = htons(atoi(argv[2]));
68 bootstrap_ip_port.ip.i = inet_addr(argv[1]);
69 bootstrap(bootstrap_ip_port);
70
71 IP_Port ip_port;
72 char data[MAX_UDP_PACKET_SIZE];
73 uint32_t length;
74
75 char buffer1[128];
76 int read1 = 0;
77 char buffer2[128];
78 int read2 = 0;
79 FILE *file1 = fopen(argv[4], "rb");
80 if ( file1==NULL ){printf("Error opening file.\n");return 1;}
81 FILE *file2 = fopen("received.txt", "wb");
82 if ( file2==NULL ){return 1;}
83 read1 = fread(buffer1, 1, 128, file1);
84
85 while(1)
86 {
87
88 while(recievepacket(&ip_port, data, &length) != -1)
89 {
90 if(rand() % 3 != 1)//simulate packet loss
91 {
92 if(DHT_handlepacket(data, length, ip_port) && LosslessUDP_handlepacket(data, length, ip_port))
93 {
94 //if packet is not recognized
95 printf("Received unhandled packet with length: %u\n", length);
96 }
97 else
98 {
99 printf("Received handled packet with length: %u\n", length);
100 }
101 }
102 }
103 friend_ip = getfriendip(argv[3]);
104 if(friend_ip.ip.i != 0)
105 {
106 if(connection == -1)
107 {
108 printf("Started connecting to friend:");
109 printip(friend_ip);
110 connection = new_connection(friend_ip);
111 }
112 }
113 if(inconnection == -1)
114 {
115 inconnection = incoming_connection();
116 if(inconnection != -1)
117 {
118 printf("Someone connected to us:");
119 printip(connection_ip(inconnection));
120 }
121 }
122 //if someone connected to us write what he sends to a file
123 //also send him our file.
124 if(inconnection != -1)
125 {
126 if(write_packet(inconnection, buffer1, read1))
127 {
128 printf("Wrote data.\n");
129 read1 = fread(buffer1, 1, 128, file1);
130 }
131 read2 = read_packet(inconnection, buffer2);
132 if(read2 != 0)
133 {
134 printf("Received data.\n");
135 if(!fwrite(buffer2, read2, 1, file2))
136 {
137 printf("file write error\n");
138 }
139 if(read2 < 128)
140 {
141 fclose(file2);
142 }
143 }
144 }
145 //if we are connected to a friend send him data from the file.
146 //also put what he sends us in a file.
147 if(is_connected(connection) == 3)
148 {
149 if(write_packet(0, buffer1, read1))
150 {
151 printf("Wrote data.\n");
152 read1 = fread(buffer1, 1, 128, file1);
153 }
154 read2 = read_packet(0, buffer2);
155 if(read2 != 0)
156 {
157 printf("Received data.\n");
158 if(!fwrite(buffer2, read2, 1, file2))
159 {
160 printf("file write error\n");
161 }
162 if(read2 < 128)
163 {
164 fclose(file2);
165 }
166 }
167 }
168 doDHT();
169 doLossless_UDP();
170 //print_clientlist();
171 //print_friendlist();
172 //c_sleep(300);
173 c_sleep(1);
174 }
175
176 shutdown_networking();
177 return 0;
178} \ No newline at end of file
diff --git a/testing/Lossless_UDP_testserver.c b/testing/Lossless_UDP_testserver.c
index 00e537ac..8357c2ef 100644
--- a/testing/Lossless_UDP_testserver.c
+++ b/testing/Lossless_UDP_testserver.c
@@ -174,7 +174,6 @@ int main(int argc, char *argv[])
174 if(is_connected(connection) >= 2) 174 if(is_connected(connection) >= 2)
175 { 175 {
176 read = read_packet(connection, buffer); 176 read = read_packet(connection, buffer);
177
178 if(read != 0) 177 if(read != 0)
179 { 178 {
180 // printf("Recieved data.\n"); 179 // printf("Recieved data.\n");