/*
 * ZeroTier One - Network Virtualization Everywhere
 * Copyright (C) 2011-2015  ZeroTier, Inc.
 *
 * This program 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.
 *
 * This program 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 this program.  If not, see <http://www.gnu.org/licenses/>.
 *
 * --
 *
 * ZeroTier may be used and distributed under the terms of the GPLv3, which
 * are available at: http://www.gnu.org/licenses/gpl-3.0.html
 *
 * If you would like to embed ZeroTier into a commercial application or
 * redistribute it in a modified binary form, please contact ZeroTier Networks
 * LLC. Start here: http://www.zerotier.com/
 */

#ifdef USE_GNU_SOURCE
    #define _GNU_SOURCE
#endif

#include <unistd.h>
#include <stdio.h>
#include <stdint.h>
#include <stdlib.h>
#include <dlfcn.h>
#include <strings.h>
#include <pwd.h>
#include <errno.h>
#include <stdarg.h>
#include <netdb.h>
#include <string.h>
#include <fcntl.h>
#include <pthread.h>
#include <sys/time.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <sys/poll.h>
#include <sys/un.h>
#include <sys/resource.h>
#include <arpa/inet.h>
#include <netinet/in.h>

#if defined(__linux__)
    #include <linux/errno.h>
    #include <sys/syscall.h>
    #include <linux/net.h>
#endif

#ifdef __cplusplus
    extern "C" {
#endif
    
#if defined(__linux__)
    #define SOCK_MAX (SOCK_PACKET + 1)
#endif
#define SOCK_TYPE_MASK 0xf

#include "SDK.h"
#include "SDK_Signatures.h"
#include "SDK_Debug.h"
#include "SDK_RPC.h"
#include "node/Constants.hpp" // For Tap's MTU
    
void print_addr(struct sockaddr *addr);
void dwr(int level, const char *fmt, ... );

static char *api_netpath = (char *)0;

// TODO: Remove before production
void set_netpath(char * path) {
    dwr(MSG_DEBUG,"set_netpath(%s)", path);
    api_netpath = path;
    rpc_mutex_init(); // TODO: double-check this
}

const char *get_netpath() {
    return api_netpath;
}
        
    // ------------------------------------------------------------------------------
    // ---------------------------------- zt_init_rpc -------------------------------
    // ------------------------------------------------------------------------------
    
    void zt_init_rpc(char *nwid)
    {
    #if defined(__UNITY_3D__)
        //char *nw = "565799d8f6e1c11a";
        //char *path = "/Users/Joseph/utest2/nc_";
        //char *fullpath = malloc(strlen(path)+strlen(nw)+1);
        //if(fullpath) {
        //    strcpy(fullpath, path);
        //    strcat(fullpath, nw);
        //    api_netpath = fullpath;
        //}
        //api_netpath = nwid;
        api_netpath = "/Users/Joseph/utest2/nc_565799d8f6e1c11a";
        
    #endif
        
    #if defined(__IOS__)
        api_netpath = "ZeroTier/One/nc_" + nwid;
        
        void *spec = pthread_getspecific(thr_id_key);
        int thr_id = spec != NULL ? *((int*)spec) : -1;
        // dwr(MSG_DEBUG_EXTRA, "set_up_intercept(thr_id=%d)\n", thr_id);
        if(thr_id == INTERCEPT_ENABLED) {
            if (!api_netpath) {
                api_netpath = "ZeroTier/One/nc_e5cd7a9e1c3511dd"; // Path allowed on iOS devices
                
            }
            return 1;
        }
        return 0;
        
    #elif defined(__ANDROID__)
        api_netpath = "ZeroTier/One/nc_" + nwid;
        return 1;
    #else
        if (!api_netpath) {
            api_netpath = getenv("ZT_NC_NETWORK");
            set_netpath(api_netpath);
            if(!api_netpath) {
                // return 0;
            }
        }
    #endif
    }
        
    // ------------------------------------------------------------------------------
    // ------------------------------------ sendto() --------------------------------
    // ------------------------------------------------------------------------------
    // int sockfd, const void *buf, size_t len, int flags, 
    // const struct sockaddr *addr, socklen_t addr_len

#if !defined(__ANDROID__)
    ssize_t zt_sendto(SENDTO_SIG)
    {
        dwr(MSG_DEBUG, "zt_sendto()\n");
        if(len > ZT_UDP_DEFAULT_PAYLOAD_MTU) {
            errno = EMSGSIZE; // Msg is too large
            return -1;
        }
        int socktype = 0;
        socklen_t socktype_len;
        getsockopt(sockfd,SOL_SOCKET, SO_TYPE, (void*)&socktype, &socktype_len);

        if((socktype & SOCK_STREAM) || (socktype & SOCK_SEQPACKET)) {
            if(addr == NULL || flags != 0) {
                errno = EISCONN;
                return -1;
            }
        }
        
        // the socket isn't connected
        //int err = rpc_send_command(api_netpath, RPC_IS_CONNECTED, -1, &fd, sizeof(struct fd));
        //if(err == -1) {
        //    errno = ENOTCONN;
        //    return -1;
        //}

        // EMSGSIZE should be returned if the message is too long to be passed atomically through
        // the underlying protocol, in our case MTU?
        // TODO: More efficient solution
        // This connect call is used to get the address info to the stack for sending the packet
        int err;
        if((err = connect(sockfd, addr, addr_len)) < 0) {
            dwr(MSG_DEBUG, "sendto(): unknown problem passing address info to stack\n");
            errno = EISCONN; // double-check this is correct
            return -1;
        }
        return send(sockfd, buf, len, flags);
    }
#endif

    // ------------------------------------------------------------------------------
    // ----------------------------------- sendmsg() --------------------------------
    // ------------------------------------------------------------------------------
    // int socket, const struct msghdr *message, int flags

#if !defined(__ANDROID__)
    ssize_t zt_sendmsg(SENDMSG_SIG)
    {
        dwr(MSG_DEBUG, "zt_sendmsg()\n");
        char * p, * buf;
        size_t tot_len = 0;
        size_t err;
        struct iovec * iov = message->msg_iov;
        for(int i=0; i<message->msg_iovlen; ++i)
            tot_len += iov[i].iov_len;
        if(tot_len > ZT_UDP_DEFAULT_PAYLOAD_MTU) {
            errno = EMSGSIZE; // Message too large to send atomically via underlying protocol, don't send
            return -1;
        }
        buf = malloc(tot_len);
        if(tot_len != 0 && buf == NULL) {
            errno = ENOMEM; // Unable to allocate space for message
            return -1;
        }
        p = buf;
        for(int i=0; i < message->msg_iovlen; ++i) {
            memcpy(p, iov[i].iov_base, iov[i].iov_len);
            p += iov[i].iov_len;
        }
        err = sendto(socket, buf, tot_len, flags, message->msg_name, message->msg_namelen);
        free(buf);
        return err;
    }
#endif
    
    // ------------------------------------------------------------------------------
    // ---------------------------------- recvfrom() --------------------------------
    // ------------------------------------------------------------------------------
    // int socket, void *restrict buffer, size_t length, int flags, struct sockaddr
    // *restrict address, socklen_t *restrict address_len

#if !defined(__ANDROID__)
    ssize_t zt_recvfrom(RECVFROM_SIG)
    {
        dwr(MSG_DEBUG,"zt_recvfrom(%d)\n", socket);
        // TODO: Remove for production
        
        ssize_t err;
        //int sock_type;
        //socklen_t type_len;
        //realgetsockopt(socket, SOL_SOCKET, SO_TYPE, (void *) &sock_type, &type_len);
        unsigned int addr;
        unsigned short port;
        char addr_buf[sizeof(addr) + sizeof(port)];
        // Since this can be called for connection-oriented sockets,
        // we need to check the type before we try to read the address info
        //if(sock_type == SOCK_DGRAM && address != NULL && address_len != NULL) {
        err = read(socket, &addr_buf, sizeof(addr_buf)); // Read prepended address info
        memcpy(&addr, addr_buf, sizeof(addr));
        memcpy(&port, addr_buf+sizeof(addr), sizeof(port));
        *address_len=sizeof(addr_buf);
        //}
        err = read(socket, buffer, length); // Read what was placed on buffer from service
        if(err < 0)
            perror("read:\n");
        
        port = htons(port);
        memcpy(address->sa_data, &port, sizeof(port));
        memcpy(address->sa_data+2, &addr, sizeof(addr));
        return err;
    }
#endif

    // ------------------------------------------------------------------------------
    // ----------------------------------- recvmsg() --------------------------------
    // ------------------------------------------------------------------------------
    // int socket, struct msghdr *message, int flags

#if !defined(__ANDROID__)
    ssize_t zt_recvmsg(RECVMSG_SIG)
    {
        dwr(MSG_DEBUG, "zt_recvmsg(%d)\n", socket);
        ssize_t err, n, tot_len = 0;
        char *buf, *p;
        struct iovec *iov = message->msg_iov;
        
        for(int i = 0; i < message->msg_iovlen; ++i)
            tot_len += iov[i].iov_len;
        buf = malloc(tot_len);
        if(tot_len != 0 && buf == NULL) {
            errno = ENOMEM;
            return -1;
        }
        n = err = recvfrom(socket, buf, tot_len, flags, message->msg_name, &message->msg_namelen);
        p = buf;
        
        // According to: http://pubs.opengroup.org/onlinepubs/009695399/functions/recvmsg.html
        if(err > message->msg_controllen && !( message->msg_flags & MSG_PEEK)) {
            // excess data should be disgarded
            message->msg_flags |= MSG_TRUNC; // Indicate that the buffer has been truncated
        }
        
        while (n > 0) {
            ssize_t count = n < iov->iov_len ? n : iov->iov_len;
            memcpy (iov->iov_base, p, count);
            p += count;
            n -= count;
            ++iov;
        }
        free(buf);
        return err;
    }
#endif

    // ------------------------------------------------------------------------------
    // ----------------------- Exposed RX/TX API for UNITY 3D -----------------------
    // ------------------------------------------------------------------------------
        
#if defined(__UNITY_3D__)
       
        // Just expose some basic calls for configuring and RX/TXing through ZT sockets
        
        ssize_t zt_send(int fd, void *buf, int len)
        {
            return write(fd, buf, len);
        }
        
        ssize_t zt_recv(int fd, void *buf, int len)
        {
            return read(fd, buf, len);
        }
        
        int zt_set_nonblock(int fd)
        {
            return 	fcntl(fd, F_SETFL, O_NONBLOCK);
        }
#endif
        
    // ------------------------------------------------------------------------------
    // --------------------------------- setsockopt() -------------------------------
    // ------------------------------------------------------------------------------
    // int socket, int level, int option_name, const void *option_value, 
    // socklen_t option_len

    int zt_setsockopt(SETSOCKOPT_SIG)
    {
        dwr(MSG_DEBUG, "zt_setsockopt()\n");
        return 0;
    }
    
    // ------------------------------------------------------------------------------
    // --------------------------------- getsockopt() -------------------------------
    // ------------------------------------------------------------------------------
    // int sockfd, int level, int optname, void *optval, 
    // socklen_t *optlen 

    int zt_getsockopt(GETSOCKOPT_SIG)
    {
        dwr(MSG_DEBUG,"zt_getsockopt(%d)\n", sockfd);
        if(optname == SO_TYPE) {
            int* val = (int*)optval;
            *val = 2;
            optval = (void*)val;
        }
        return 0;
    }
    
    // ------------------------------------------------------------------------------
    // ----------------------------------- socket() ---------------------------------
    // ------------------------------------------------------------------------------
    // int socket_family, int socket_type, int protocol

    int zt_socket(SOCKET_SIG) {
        zt_init_rpc("");
        dwr(MSG_DEBUG, "zt_socket()\n");
        /* Check that type makes sense */
#if defined(__linux__)
        int flags = socket_type & ~SOCK_TYPE_MASK;
    #if !defined(__ANDROID__)
        if (flags & ~(SOCK_CLOEXEC | SOCK_NONBLOCK)) {
            errno = EINVAL;
            return -1;
        }
    #endif
#endif
        socket_type &= SOCK_TYPE_MASK;
        /* Check protocol is in range */
#if defined(__linux__)
        if (socket_family < 0 || socket_family >= NPROTO){
            errno = EAFNOSUPPORT;
            return -1;
        }
        if (socket_type < 0 || socket_type >= SOCK_MAX) {
            errno = EINVAL;
            return -1;
        }
#endif
        /* Assemble and send RPC */
        struct socket_st rpc_st;
        rpc_st.socket_family = socket_family;
        rpc_st.socket_type = socket_type;
        rpc_st.protocol = protocol;
#if defined(__linux__)
    #if !defined(__ANDROID__)
        rpc_st.__tid = 5; //syscall(SYS_gettid);
    #else
        rpc_st.__tid = gettid(); // dummy value
    #endif
#endif
        /* -1 is passed since we we're generating the new socket in this call */
        int err = rpc_send_command(api_netpath, RPC_SOCKET, -1, &rpc_st, sizeof(struct socket_st));
        dwr(MSG_DEBUG," socket() = %d\n", err);
        return err;
    }

    // ------------------------------------------------------------------------------
    // ---------------------------------- connect() ---------------------------------
    // ------------------------------------------------------------------------------
    // int __fd, const struct sockaddr * __addr, socklen_t __len

    int zt_connect(CONNECT_SIG)
    {
        dwr(MSG_DEBUG,"zt_connect(%d)\n", __fd);
        struct connect_st rpc_st;
#if defined(__linux__)
    #if !defined(__ANDROID__)
        rpc_st.__tid = syscall(SYS_gettid);
    #else
        rpc_st.__tid = gettid(); // dummy value
    #endif
#endif
        rpc_st.__fd = __fd;
        memcpy(&rpc_st.__addr, __addr, sizeof(struct sockaddr_storage));
        memcpy(&rpc_st.__len, &__len, sizeof(socklen_t));
        return rpc_send_command(api_netpath, RPC_CONNECT, __fd, &rpc_st, sizeof(struct connect_st));
    }
    
    // ------------------------------------------------------------------------------
    // ------------------------------------ bind() ----------------------------------
    // ------------------------------------------------------------------------------
    // int sockfd, const struct sockaddr *addr, socklen_t addrlen

#if !defined(__ANDROID__)
        int zt_bind(BIND_SIG)
        {
            dwr(MSG_DEBUG,"zt_bind(%d)\n", sockfd);
            struct bind_st rpc_st;
            rpc_st.sockfd = sockfd;
    #if defined(__linux__)
        #if !defined(__ANDROID__)
            rpc_st.__tid = 5;//syscall(SYS_gettid);
        #else
            rpc_st.__tid = gettid(); // dummy value
        #endif
    #endif
            memcpy(&rpc_st.addr, addr, sizeof(struct sockaddr_storage));
            memcpy(&rpc_st.addrlen, &addrlen, sizeof(socklen_t));
            return rpc_send_command(api_netpath, RPC_BIND, sockfd, &rpc_st, sizeof(struct bind_st));
        }
#endif

    // ------------------------------------------------------------------------------
    // ----------------------------------- accept4() --------------------------------
    // ------------------------------------------------------------------------------
    // int sockfd, struct sockaddr *addr, socklen_t *addrlen, int flags

#if defined(__linux)
        int zt_accept4(ACCEPT4_SIG)
        {
            dwr(MSG_DEBUG,"zt_accept4(%d):\n", sockfd);
        #if !defined(__ANDROID__)
            if ((flags & SOCK_CLOEXEC))
                fcntl(sockfd, F_SETFL, FD_CLOEXEC);
            if ((flags & SOCK_NONBLOCK))
               fcntl(sockfd, F_SETFL, O_NONBLOCK);
        #endif
            return accept(sockfd, addr, addrlen);
        }
#endif
    
    // ------------------------------------------------------------------------------
    // ----------------------------------- accept() ---------------------------------
    // ------------------------------------------------------------------------------
    // int sockfd struct sockaddr *addr, socklen_t *addrlen

    int zt_accept(ACCEPT_SIG)
    {
        //dwr(MSG_DEBUG,"zt_accept(%d):\n", sockfd);
        if(addr)
            addr->sa_family = AF_INET;
        
        int new_fd = get_new_fd(sockfd);
        if(new_fd > 0) {
            errno = ERR_OK;
            return new_fd;
        }
        errno = EAGAIN;
        return -EAGAIN;
    }
    
    // ------------------------------------------------------------------------------
    // ------------------------------------- listen()--------------------------------
    // ------------------------------------------------------------------------------
    // int sockfd, int backlog

    int zt_listen(LISTEN_SIG)
    {
        dwr(MSG_DEBUG,"zt_listen(%d):\n", sockfd);
        struct listen_st rpc_st;
        rpc_st.sockfd = sockfd;
        rpc_st.backlog = backlog;
#if defined(__linux__)
    #if !defined(__ANDROID__)
        rpc_st.__tid = syscall(SYS_gettid);
    #else
        rpc_st.__tid = gettid(); // dummy value
    #endif
#endif
        return rpc_send_command(api_netpath, RPC_LISTEN, sockfd, &rpc_st, sizeof(struct listen_st));
    }
    
    // ------------------------------------------------------------------------------
    // ------------------------------------- close() --------------------------------
    // ------------------------------------------------------------------------------
    // int fd

    int zt_close(CLOSE_SIG) {
        dwr(MSG_DEBUG, "zt_close(%d)", fd);
        return close(fd);
    }
    
    // ------------------------------------------------------------------------------
    // -------------------------------- getsockname() -------------------------------
    // ------------------------------------------------------------------------------
    // int sockfd, struct sockaddr *addr, socklen_t *addrlen
    
    int zt_getsockname(GETSOCKNAME_SIG)
    {
        dwr(MSG_DEBUG,"zt_getsockname(%d):\n", sockfd);
        /* TODO: This is kind of a hack as it stands -- assumes sockaddr is sockaddr_in
         * and is an IPv4 address. */
        struct getsockname_st rpc_st;
        rpc_st.sockfd = sockfd;
        memcpy(&rpc_st.addrlen, &addrlen, sizeof(socklen_t));
        int rpcfd = rpc_send_command(api_netpath, RPC_GETSOCKNAME, sockfd, &rpc_st, sizeof(struct getsockname_st));
        /* read address info from service */
        char addrbuf[sizeof(struct sockaddr_storage)];
        memset(&addrbuf, 0, sizeof(struct sockaddr_storage));
        
        if(rpcfd > -1)
            if(read(rpcfd, &addrbuf, sizeof(struct sockaddr_storage)) > 0)
                close(rpcfd);
        
        struct sockaddr_storage sock_storage;
        memcpy(&sock_storage, addrbuf, sizeof(struct sockaddr_storage));
        *addrlen = sizeof(struct sockaddr_in);
        memcpy(addr, &sock_storage, (*addrlen > sizeof(sock_storage)) ? sizeof(sock_storage) : *addrlen);
        addr->sa_family = AF_INET;
        return 0;
    }
        
#ifdef __cplusplus
}
#endif