/* network.h
* 
* Functions for the core networking.
* 
 
    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/>.
    
*/

#include "network.h"


/* returns current UNIX time in microseconds (us). */
uint64_t current_time()
{
    uint64_t time;
    #ifdef WIN32
    /* This probably works fine */
    FILETIME ft;
    GetSystemTimeAsFileTime(&ft);
    time = ft.dwHighDateTime;
    time <<=32;
    time |= ft.dwLowDateTime;
    time -= 116444736000000000UL;
    return time/10;
    #else
    struct timeval a;
    gettimeofday(&a, NULL);
    time = 1000000UL*a.tv_sec + a.tv_usec;
    return time;
    #endif
    
    
}

/* return a random number
   NOTE: this function should probably not be used where cryptographic randomness is absolutely necessary */
uint32_t random_int()
{
    #ifndef VANILLA_NACL
    //NOTE: this function comes from libsodium
    return randombytes_random();
    #else
    return random();
    #endif
}

/* our UDP socket, a global variable. */
static int sock;

/* Basic network functions:
   Function to send packet(data) of length length to ip_port */
int sendpacket(IP_Port ip_port, uint8_t * data, uint32_t length)
{
    ADDR addr = {AF_INET, ip_port.port, ip_port.ip}; 
    return sendto(sock,(char *) data, length, 0, (struct sockaddr *)&addr, sizeof(addr));
    
}

/* Function to receive data, ip and port of sender is put into ip_port
   the packet data into data
   the packet length into length.
   dump all empty packets. */
int receivepacket(IP_Port * ip_port, uint8_t * data, uint32_t * length)
{
    ADDR addr;
    #ifdef WIN32
    int addrlen = sizeof(addr);
    #else
    uint32_t addrlen = sizeof(addr);
    #endif    
    (*(int32_t *)length) = recvfrom(sock,(char *) data, MAX_UDP_PACKET_SIZE, 0, (struct sockaddr *)&addr, &addrlen);
    if(*(int32_t *)length <= 0)
    {
        /* nothing received
           or empty packet */
        return -1;
    }
    ip_port->ip = addr.ip;
    ip_port->port = addr.port;
    return 0;
    
}

/* initialize networking
   bind to ip and port
   ip must be in network order EX: 127.0.0.1 = (7F000001)
   port is in host byte order (this means don't worry about it)
   returns 0 if no problems 
   returns -1 if there are problems */
int init_networking(IP ip ,uint16_t port)
{
    #ifdef WIN32
    WSADATA wsaData;
    if(WSAStartup(MAKEWORD(2,2), &wsaData) != NO_ERROR)
    {
        return -1;
    }
    
    #else
    srandom((uint32_t)current_time());
    #endif
    srand((uint32_t)current_time());
    
    /* initialize our socket */
    sock = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP); 

	/* Check for socket error */
	#ifdef WIN32
	if (sock == INVALID_SOCKET) //MSDN recommends this
		return -1;
	#else
	if (sock < 0)
		return -1;
	#endif


    /* Functions to increase the size of the send and receive UDP buffers
       NOTE: uncomment if necessary */
    /*
    int n = 1024 * 1024 * 2;
    if(setsockopt(sock, SOL_SOCKET, SO_RCVBUF, (char*)&n, sizeof(n)) == -1)
    {
        return -1;
    }

    if(setsockopt(sock, SOL_SOCKET, SO_SNDBUF, (char*)&n, sizeof(n)) == -1)
    {
        return -1;
    }*/
    
    /* 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 = {AF_INET, htons(port), ip}; 
    bind(sock, (struct sockaddr*)&addr, sizeof(addr));

    return 0;

}

/* function to cleanup networking stuff */
void shutdown_networking()
{
    #ifdef WIN32
    closesocket(sock);
    WSACleanup();
    #else
    close(sock);
    #endif
    return;
}

/* resolves provided address to a binary data in network byte order
  address is ASCII null terminated string
  address should represent IPv4, IPv6 or a hostname
  on success returns a data in network byte order that can be used to set IP.i or IP_Port.ip.i
  on failure returns -1 */
int resolve_addr(char *address)
{
    struct addrinfo hints;
    memset(&hints, 0, sizeof(hints));
    hints.ai_family = AF_UNSPEC; //support both IPv4 and IPv6
    hints.ai_socktype = SOCK_DGRAM; //type of socket Tox uses

    struct addrinfo *server = NULL;

    int success = getaddrinfo(address, "7", &hints, &server);
    if(success != 0)
    {
        return -1;
    }

    int resolved = ((struct sockaddr_in*)server->ai_addr)->sin_addr.s_addr;
    freeaddrinfo(server);
    return resolved;
}