/*
 * Copyright 2021 Max Planck Institute for Software Systems, and
 * National University of Singapore
 *
 * Permission is hereby granted, free of charge, to any person obtaining
 * a copy of this software and associated documentation files (the
 * "Software"), to deal in the Software without restriction, including
 * without limitation the rights to use, copy, modify, merge, publish,
 * distribute, sublicense, and/or sell copies of the Software, and to
 * permit persons to whom the Software is furnished to do so, subject to
 * the following conditions:
 *
 * The above copyright notice and this permission notice shall be
 * included in all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
 * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
 * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.
 * IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY
 * CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
 * TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
 * SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
 */

#include <fcntl.h>
#include <pthread.h>
#include <stdlib.h>
#include <stdio.h>
#include <string.h>
#include <sys/ioctl.h>
#include <sys/mman.h>
#include <unistd.h>
#include <linux/if.h>
#include <linux/if_tun.h>

#include <netsim.h>

//#define DEBUG_PKTMETA

static struct netsim_interface nsif;
static int tap_fd;

static int tap_open(const char *name)
{
    struct ifreq ifr;
    int fd;

    if ((fd = open("/dev/net/tun", O_RDWR)) < 0) {
        perror("tap_open: open failed");
        return -1;
    }

    memset(&ifr, 0, sizeof(ifr));
    ifr.ifr_flags = IFF_TAP | IFF_NO_PI;
    strncpy(ifr.ifr_name, name, IFNAMSIZ);

    if (ioctl(fd, TUNSETIFF, &ifr) != 0) {
        perror("tap_open: ioctl failed");
        close(fd);
        return -1;
    }

    return fd;
}

static void d2n_send(volatile struct cosim_eth_proto_d2n_send *s)
{
#ifdef DEBUG_PKTMETA
    printf("sent packet: len=%u\n", s->len);
#endif

    if (write(tap_fd, (void *) s->data, s->len) != (ssize_t) s->len) {
        perror("d2n_send: send failed");
    }
}

static void poll_d2n(void)
{
    volatile union cosim_eth_proto_d2n *msg = netsim_d2n_poll(&nsif, 0);
    uint8_t type;

    /* message not ready */
    if (msg == NULL)
        return;

    type = msg->dummy.own_type & COSIM_ETH_PROTO_D2N_MSG_MASK;
    switch (type) {
        case COSIM_ETH_PROTO_D2N_MSG_SEND:
            d2n_send(&msg->send);
            break;

        default:
            fprintf(stderr, "poll_d2n: unsupported type=%u\n", type);
    }

    netsim_d2n_done(&nsif, msg);
}

static void *rx_handler(void *arg)
{
    volatile union cosim_eth_proto_n2d *msg;
    volatile struct cosim_eth_proto_n2d_recv *rx;
    ssize_t len;

    while (1) {
        msg = netsim_n2d_alloc(&nsif, 0, 0);
        if (msg == NULL) {
            fprintf(stderr, "coudl not allocate message for rx\n");
            abort();
        }
        rx = &msg->recv;

        len = read(tap_fd, (void *) rx->data, nsif.n2d_elen - sizeof(*msg));
        if (len <= 0) {
            perror("rx handler: read failed");
        }
        rx->len = len;
        rx->port = 0;
#ifdef DEBUG_PKTMETA
        printf("received packet: len=%u\n", rx->len);
#endif

        // WMB();
        rx->own_type = COSIM_ETH_PROTO_N2D_MSG_RECV |
            COSIM_ETH_PROTO_N2D_OWN_DEV;
    }
}

int main(int argc, char *argv[])
{
    int sync;

    if (argc != 3) {
        fprintf(stderr, "Usage: net_tap TAP_DEVICE_NAME SOCKET\n");
        return EXIT_FAILURE;
    }

    if ((tap_fd = tap_open(argv[1])) < 0) {
        return -1;
    }

    sync = 0;
    if (netsim_init(&nsif, argv[2], &sync) != 0) {
        close(tap_fd);
        return -1;
    }

    pthread_t worker;
    if (pthread_create(&worker, NULL, rx_handler, NULL) != 0) {
        return EXIT_FAILURE;
    }

    printf("start polling\n");
    while (1) {
        poll_d2n();
    }
    return 0;
}
