toxcore/testing/DHT_test.c

/* DHT test
 * A file with a main that runs our DHT for testing.
 * 
 * Compile with: gcc -Wall -o test ../core/DHT.c DHT_test.c
 * 
 * Command line arguments are the ip and port of a node
 * EX: ./test 127.0.0.1 33445
 */

#include "../core/DHT.h"

//Sleep function (x = milliseconds)
#ifdef WIN32

#define c_sleep(x) Sleep(1*x)

#else
#include <unistd.h>
#define c_sleep(x) usleep(1000*x)

#endif

#define PORT 33445



int main(int argc, char *argv[])
{
    #ifdef WIN32
    WSADATA wsaData;
    if(WSAStartup(MAKEWORD(2,2), &wsaData) != NO_ERROR)
    {
        return -1;
    }
    #endif
    
    if (argc < 3) {
        printf("usage %s ip port\n", argv[0]);
        exit(0);
    }
    
    //initialize our socket
    sock = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP); 
    //Set socket nonblocking
    #ifdef WIN32
    //I think this works for windows
    u_long mode = 1;
    //ioctl(sock, FIONBIO, &mode);
    ioctlsocket(sock, FIONBIO, &mode); 
    #else
    fcntl(sock, F_SETFL, O_NONBLOCK, 1);
    #endif
    
    //Bind our socket to port PORT and address 0.0.0.0
    ADDR addr = {.family = AF_INET, .ip.i = 0, .port = htons(PORT)}; 
    bind(sock, (struct sockaddr*)&addr, sizeof(addr));
    
    IP_Port bootstrap_ip_port;
    bootstrap_ip_port.port = htons(atoi(argv[2]));
    //bootstrap_ip_port.ip.c[0] = 127;
    //bootstrap_ip_port.ip.c[1] = 0;
    //bootstrap_ip_port.ip.c[2] = 0;
    //bootstrap_ip_port.ip.c[3] = 1;
    bootstrap_ip_port.ip.i = inet_addr(argv[1]);
    bootstrap(bootstrap_ip_port);
    
    IP_Port ip_port;
    char data[MAX_UDP_PACKET_SIZE];
    uint32_t length;
    
    uint32_t i;
    
    while(1)
    {
            
        doDHT();
        
        if(recievepacket(&ip_port, data, &length) != -1)
        {
            if(DHT_recvpacket(data, length, ip_port))
            {
                printf("UNHANDLED PACKET RECEIVED\nLENGTH:%u\nCONTENTS:\n", length);
                printf("--------------------BEGIN-----------------------------\n");
                for(i = 0; i < length; i++)
                    printf("%c",data[i]);
                printf("\n--------------------END-----------------------------\n\n\n");
            }
            else
            {
                printf("Received handled packet with length: %u\n", length);
            }
        }
        c_sleep(100);
    }
    
    #ifdef WIN32
    WSACleanup();
    #endif
    return 0;   
}