src/network.c

/*
 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER.
 *
 * Copyright 2016 Mike Becker. All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions are met:
 *
 *   1. Redistributions of source code must retain the above copyright
 *      notice, this list of conditions and the following disclaimer.
 *
 *   2. Redistributions in binary form must reproduce the above copyright
 *      notice, this list of conditions and the following disclaimer in the
 *      documentation and/or other materials provided with the distribution.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
 * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE
 * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
 * POSSIBILITY OF SUCH DAMAGE.
 *
 */

#include <stdlib.h>
#include <string.h>
#include <stdio.h>
#include <unistd.h>
#include <sys/un.h>
#include <sys/stat.h>
#include <netdb.h>
#include "network.h"

int net_create_tcp(Server *server, short port) {
    memset(server, 0, sizeof(Server));
    server->fd = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
    if (server->fd > -1) {
        int one = 1;
        setsockopt(server->fd, SOL_SOCKET, SO_REUSEADDR, &one, sizeof(int));

        struct sockaddr_in addr;
        addr.sin_family = AF_INET;
        addr.sin_addr.s_addr = INADDR_ANY;
        addr.sin_port = htons(port);
        if (bind(server->fd, (struct sockaddr*)&addr, sizeof(addr))) {
            server->fd = -1;
            return 1;
        } else {
            return 0;
        }
    } else {
        return 1;
    }
}

int net_find_tcp(Server *server, char *host, short port) {
    memset(server, 0, sizeof(Server));
    server->fd = -1;

    struct addrinfo hints = {0};
    hints.ai_socktype = SOCK_STREAM;
    hints.ai_protocol = IPPROTO_TCP;
    hints.ai_family = AF_INET;
    char portstr[6];
    sprintf(portstr, "%hd", port);
    return getaddrinfo(host, portstr, &hints, &server->info);
}

int net_create_sock(Server *server, char * path) {
    memset(server, 0, sizeof(Server));
    server->fd = socket(AF_UNIX, SOCK_STREAM, 0);
    if (server->fd > -1) {
        struct sockaddr_un addr;
        addr.sun_family = AF_UNIX;
        addr.sun_family = AF_UNIX;
        strncpy(addr.sun_path, path, sizeof(addr.sun_path) - 1);
        if (bind(server->fd, (struct sockaddr*)&addr, sizeof(addr))) {
            server->fd = -1;
            return 1;
        } else {
            chmod(path, S_IRWXU | S_IRWXG | S_IRWXO);
            return 0;
        }
    } else {
        return 1;
    }
}

int net_find_sock(Server *server, char *path) {
    memset(server, 0, sizeof(Server));
    server->fd = -1;

    server->info = calloc(1, sizeof(struct addrinfo));
    server->info->ai_family = AF_UNIX;
    server->info->ai_socktype = SOCK_STREAM;
    server->info->ai_protocol = 0;
    struct sockaddr_un *addr = malloc(sizeof(struct sockaddr_un));
    addr->sun_family = AF_UNIX;
    strncpy(addr->sun_path, path, sizeof(addr->sun_path) - 1);
    server->info->ai_addr = (struct sockaddr*) addr;
    server->info->ai_addrlen = sizeof(struct sockaddr_un);
    return 0;
}

int net_listen(Server *server) {
    listen(server->fd, 1);
    Client* client = malloc(sizeof(Client));
    client->fd = -1;
    client->address_len = sizeof(client->address);
    server->client = client;
    
    client->fd = accept(server->fd,
        &(client->address), &(client->address_len));
    
    return client->fd == -1;
}

int net_connect(Server *server) {
    
    Client* client = calloc(1, sizeof(Client));
    client->fd = -1;
    server->fd = socket(
        server->info->ai_family,
        server->info->ai_socktype,
        server->info->ai_protocol
    );
    server->client = client;
    
    if (server->fd == -1) {
        return 1;
    }
    
    return connect(server->fd, server->info->ai_addr, server->info->ai_addrlen);
}

void net_destroy(Server *server) {
    if (server->info) {
        if (server->info->ai_family == AF_UNIX) {
            /* this was allocated by our custom function */
            free(server->info->ai_addr);
            free(server->info);
            server->info = NULL;
        } else {
            freeaddrinfo(server->info);
        }
    }
    if (server->client) {
        shutdown(server->client->fd, SHUT_RDWR);
        close(server->client->fd);
        free(server->client);
        server->client = NULL;
    }
    if (server->fd > -1) {
        shutdown(server->fd, SHUT_RDWR);
        close(server->fd);
        server->fd = -1;
    }
}

void net_send_code(int socket, uint8_t code) {
    send(socket, &code, sizeof(uint8_t), 0);
}

void net_send_data(int socket, uint8_t code, void *data, size_t len) {
    uint8_t pkg[len+1];
    pkg[0] = code;
    memcpy(pkg+1, data, len);
    send(socket, pkg, len+1, 0);
}

uint8_t net_recieve_code(int socket) {
    uint8_t code;
    if (recv(socket, &code, sizeof(code), 0) == sizeof(code)) {
        return code;
    } else {
        return NETCODE_CONNLOST;
    }
}

uint8_t net_recieve_code_async(int socket) {
    fd_set readfds;
    FD_ZERO(&readfds);
    FD_SET(socket, &readfds);
    struct timeval timeout;
    timeout.tv_sec = 0;
    timeout.tv_usec = 50000;
    int result = select(socket+1, &readfds, NULL, NULL, &timeout);
    if (result == 0) return NETCODE_AGAIN;
    if (result == -1) return NETCODE_ERROR;
    uint8_t code;
    if (recv(socket, &code, sizeof(code), 0) == sizeof(code)) {
        return code;
    } else {
        return NETCODE_CONNLOST;
    }
}

void net_recieve_data(int socket, void *data, size_t len) {
    recv(socket, data, len, MSG_WAITALL);
}