/*
 * Copyright (c) 2014 Brian Swetland
 *
 * Use of this source code is governed by a MIT-style
 * license that can be found in the LICENSE file or at
 * https://opensource.org/licenses/MIT
 */

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>

#include "network.h"

in_addr_t lookup_hostname(const char *hostname) {
    int err;
    struct addrinfo *info, *temp;
    in_addr_t addr = 0;

    struct addrinfo hints;
    memset(&hints, 0, sizeof(hints));
    hints.ai_family = AF_INET;
    hints.ai_socktype = SOCK_STREAM;

    err = getaddrinfo(hostname, NULL, &hints, &info);
    if (err < 0) {
        printf("getaddrinfo() returns %d, error '%s'\n", err, gai_strerror(err));
        return 0;
    }

    for (temp = info; temp; temp = temp->ai_next) {
//      printf("flags 0x%x, family %d, socktype %d, protocol %d, addrlen %d\n",
//          temp->ai_flags, temp->ai_family, temp->ai_socktype, temp->ai_protocol, temp->ai_addrlen);

        if (temp->ai_family == AF_INET && temp->ai_protocol == IPPROTO_TCP) {
            struct sockaddr_in *sa = (struct sockaddr_in *)temp->ai_addr;
//          printf("port %d, addr 0x%x\n", sa->sin_port, sa->sin_addr.s_addr);
            addr = sa->sin_addr.s_addr;
        }
    }

    freeaddrinfo(info);

    return addr;
}

static int inet_listen(in_addr_t addr, int type, unsigned port, int shared) {
    struct sockaddr_in sa;
    int s;
    if ((s = socket(AF_INET, type, 0)) < 0) {
        return -1;
    }
    if (shared) {
        int opt = 1;
        setsockopt(s, SOL_SOCKET, SO_REUSEADDR, &opt, sizeof(opt));
        //setsockopt(s, SOL_SOCKET, SO_REUSEPORT, &opt, sizeof(opt));
    }
    memset(&sa, 0, sizeof(sa));
    sa.sin_family = AF_INET;
    sa.sin_port = htons(port);
    sa.sin_addr.s_addr = addr;
    if (bind(s, (struct sockaddr *) &sa, sizeof(sa)) < 0) {
        close(s);
        return -1;
    }
    return s;
}

static int inet_connect(in_addr_t addr, int type, unsigned port) {
    struct sockaddr_in sa;
    int s;
    if ((s = socket(AF_INET, type, 0)) < 0) {
        return -1;
    }
    if (addr == 0xFFFFFFFF) {
        int opt = 1;
        setsockopt(s, SOL_SOCKET, SO_BROADCAST, &opt, sizeof(opt));
    }
    memset(&sa, 0, sizeof(sa));
    sa.sin_family = AF_INET;
    sa.sin_port = htons(port);
    sa.sin_addr.s_addr = addr;
    if (connect(s, (struct sockaddr *) &sa, sizeof(sa)) < 0) {
        close(s);
        return -1;
    }
    return s;
}

int udp_listen(in_addr_t addr, unsigned port, int shared) {
    return inet_listen(addr, SOCK_DGRAM, port, shared);
}

int udp_connect(in_addr_t addr, unsigned port) {
    return inet_connect(addr, SOCK_DGRAM, port);
}

int tcp_listen(in_addr_t addr, unsigned port) {
    return inet_listen(addr, SOCK_STREAM, port, 0);
}

int tcp_connect(in_addr_t addr, unsigned port) {
    return inet_connect(addr, SOCK_STREAM, port);
}