toxcore/testing/Lossless_UDP_testclient.c

232 lines
6.8 KiB
C
Raw Normal View History

2013-06-27 22:39:37 +08:00
/* Lossless_UDP testclient
* A program that connects and sends a file using our lossless UDP algorithm.
* NOTE: this program simulates a 33% packet loss.
2013-08-17 01:11:09 +08:00
*
2013-06-27 22:39:37 +08:00
* Best used in combination with Lossless_UDP_testserver
2013-08-17 01:11:09 +08:00
*
* Compile with: gcc -O2 -Wall -lsodium -o testclient ../core/network.c ../core/Lossless_UDP.c Lossless_UDP_testclient.c
2013-08-17 01:11:09 +08:00
*
2013-06-30 23:34:35 +08:00
* Command line arguments are the ip and port to connect and send the file to.
* EX: ./testclient 127.0.0.1 33445 filename.txt
*
* Copyright (C) 2013 Tox project All Rights Reserved.
*
* This file is part of Tox.
*
* Tox is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* Tox is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with Tox. If not, see <http://www.gnu.org/licenses/>.
2013-08-17 01:11:09 +08:00
*
2013-06-27 22:39:37 +08:00
*/
#include "../core/network.h"
#include "../core/Lossless_UDP.h"
#ifdef WIN32
#define c_sleep(x) Sleep(1*x)
#else
#include <unistd.h>
#include <arpa/inet.h>
#define c_sleep(x) usleep(1000*x)
#endif
#define PORT 33446
2013-07-27 11:07:25 +08:00
void printpacket(uint8_t *data, uint32_t length, IP_Port ip_port)
2013-06-27 22:39:37 +08:00
{
uint32_t i;
printf("UNHANDLED PACKET RECEIVED\nLENGTH:%u\nCONTENTS:\n", length);
printf("--------------------BEGIN-----------------------------\n");
2013-08-17 01:11:09 +08:00
2013-07-27 11:07:25 +08:00
for (i = 0; i < length; i++) {
if (data[i] < 16)
2013-06-27 22:39:37 +08:00
printf("0");
2013-08-17 01:11:09 +08:00
printf("%hhX", data[i]);
2013-06-27 22:39:37 +08:00
}
2013-08-17 01:11:09 +08:00
2013-06-27 22:39:37 +08:00
printf("\n--------------------END-----------------------------\n\n\n");
}
void printip(IP_Port ip_port)
{
2013-08-17 01:11:09 +08:00
printf("\nIP: %u.%u.%u.%u Port: %u", ip_port.ip.c[0], ip_port.ip.c[1], ip_port.ip.c[2], ip_port.ip.c[3],
ntohs(ip_port.port));
}
/*
void printpackets(Data test)
{
int i;
if(test.size == 0)
return;
printf("SIZE: %u\n", test.size);
for(i =0; i < test.size; i++)
{
printf("%hhX", test.data[i]);
}
printf("\n");
}
void printconnection(int connection_id)
{
printf("--------------------BEGIN---------------------\n");
IP_Port ip_port = connections[connection_id].ip_port;
printf("IP: %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));
2013-08-17 01:11:09 +08:00
printf("status: %u, inbound: %u, SYNC_rate: %u\n", connections[connection_id].status,
connections[connection_id].inbound, connections[connection_id].SYNC_rate);
2013-08-17 01:11:09 +08:00
printf("data rate: %u, last sync: %llu, last sent: %llu, last recv: %llu \n", connections[connection_id].data_rate,
connections[connection_id].last_SYNC, connections[connection_id].last_sent, connections[connection_id].last_recv);
int i;
for(i =0; i < MAX_QUEUE_NUM; i++)
{
printf(" %u ",i);
printpackets(connections[connection_id].sendbuffer[i]);
}
for(i =0; i < MAX_QUEUE_NUM; i++)
{
printf(" %u ",i);
printpackets(connections[connection_id].recvbuffer[i]);
}
Data sendbuffer[MAX_QUEUE_NUM];
Data recvbuffer[MAX_QUEUE_NUM];
2013-08-17 01:11:09 +08:00
printf("recv_num: %u, orecv_num: %u, sent_packetnum %u, osent_packetnum: %u, successful_sent: %u, successful_read: %u\n",
connections[connection_id].recv_packetnum,
connections[connection_id].orecv_packetnum, connections[connection_id].sent_packetnum, connections[connection_id].osent_packetnum,
connections[connection_id].successful_sent,
connections[connection_id].successful_read);
2013-08-17 01:11:09 +08:00
printf("req packets: \n");
for(i = 0; i < BUFFER_PACKET_NUM; i++)
{
printf(" %u ", connections[connection_id].req_packets[i]);
}
printf("\nNumber: %u recv_counter: %u, send_counter: %u\n", connections[connection_id].num_req_paquets,
connections[connection_id].recv_counter, connections[connection_id].send_counter);
printf("--------------------END---------------------\n");
2013-08-17 01:11:09 +08:00
}
*/
/*( receive packets and send them to the packethandler */
/*run doLossless_UDP(); */
2013-08-21 07:20:42 +08:00
//void Lossless_UDP()
//{
/* IP_Port ip_port;
uint8_t data[MAX_UDP_PACKET_SIZE];
uint32_t length;
while (receivepacket(&ip_port, data, &length) != -1) {
printf("packet with length: %u\n", length); */
/* if(rand() % 3 != 1)//add packet loss
{ */
/*
if (LosslessUDP_handlepacket(data, length, ip_port))
printpacket(data, length, ip_port);
else
printf("Received handled packet with length: %u\n", length); //printconnection(0); */
/* } */
/* }*/
//networking_poll();
//doLossless_UDP();
2013-08-17 01:11:09 +08:00
2013-08-21 07:20:42 +08:00
//}
2013-06-28 21:15:19 +08:00
2013-06-27 22:39:37 +08:00
int main(int argc, char *argv[])
{
if (argc < 4) {
2013-06-27 22:39:37 +08:00
printf("usage: %s ip port filename\n", argv[0]);
exit(0);
}
2013-08-17 01:11:09 +08:00
2013-07-16 22:22:32 +08:00
uint8_t buffer[512];
2013-06-27 22:39:37 +08:00
int read;
2013-08-17 01:11:09 +08:00
2013-06-27 22:39:37 +08:00
FILE *file = fopen(argv[3], "rb");
2013-08-17 01:11:09 +08:00
2013-07-27 11:07:25 +08:00
if (file == NULL)
2013-08-17 01:11:09 +08:00
return 1;
/* initialize networking */
/* bind to ip 0.0.0.0:PORT */
2013-06-27 22:39:37 +08:00
IP ip;
ip.i = 0;
Lossless_UDP *ludp = new_lossless_udp(new_networking(ip, PORT));
2013-06-27 22:39:37 +08:00
perror("Initialization");
IP_Port serverip;
serverip.ip.i = inet_addr(argv[1]);
serverip.port = htons(atoi(argv[2]));
printip(serverip);
2013-08-21 07:20:42 +08:00
int connection = new_connection(ludp, serverip);
2013-06-27 22:39:37 +08:00
uint64_t timer = current_time();
2013-08-17 01:11:09 +08:00
2013-07-27 11:07:25 +08:00
while (1) {
2013-08-17 01:11:09 +08:00
/* printconnection(connection); */
2013-08-21 07:20:42 +08:00
networking_poll(ludp->net);
do_lossless_udp(ludp);
2013-08-17 01:11:09 +08:00
2013-08-21 07:20:42 +08:00
if (is_connected(ludp, connection) == 3) {
printf("Connecting took: %llu us\n", (unsigned long long)(current_time() - timer));
break;
}
2013-08-17 01:11:09 +08:00
2013-08-21 07:20:42 +08:00
if (is_connected(ludp, connection) == 0) {
printf("Connection timeout after: %llu us\n", (unsigned long long)(current_time() - timer));
return 1;
2013-06-27 22:39:37 +08:00
}
2013-08-17 01:11:09 +08:00
c_sleep(1);
2013-06-27 22:39:37 +08:00
}
2013-08-17 01:11:09 +08:00
2013-06-27 22:39:37 +08:00
timer = current_time();
2013-08-17 01:11:09 +08:00
/*read first part of file */
2013-07-16 22:22:32 +08:00
read = fread(buffer, 1, 512, file);
2013-08-17 01:11:09 +08:00
2013-07-27 11:07:25 +08:00
while (1) {
/* printconnection(connection); */
2013-08-21 07:20:42 +08:00
networking_poll(ludp->net);
do_lossless_udp(ludp);
2013-08-21 07:20:42 +08:00
if (is_connected(ludp, connection) == 3) {
2013-08-17 01:11:09 +08:00
2013-08-21 07:20:42 +08:00
if (write_packet(ludp, connection, buffer, read)) {
2013-08-17 01:11:09 +08:00
/* printf("Wrote data.\n"); */
2013-07-16 22:22:32 +08:00
read = fread(buffer, 1, 512, file);
2013-06-27 22:39:37 +08:00
}
2013-08-17 01:11:09 +08:00
/* printf("%u\n", sendqueue(connection)); */
2013-08-21 07:20:42 +08:00
if (sendqueue(ludp, connection) == 0) {
2013-07-27 11:07:25 +08:00
if (read == 0) {
printf("Sent file successfully in: %llu us\n", (unsigned long long)(current_time() - timer));
2013-06-27 22:39:37 +08:00
break;
}
}
2013-08-17 01:11:09 +08:00
} else {
printf("Connecting Lost after: %llu us\n", (unsigned long long)(current_time() - timer));
return 0;
2013-06-27 22:39:37 +08:00
}
2013-08-17 01:11:09 +08:00
/* c_sleep(1); */
2013-06-27 22:39:37 +08:00
}
2013-08-17 01:11:09 +08:00
2013-06-27 22:39:37 +08:00
return 0;
}