/*
 * Copyright (c) 2023-2025 Ian Marco Moffett and the Osmora Team.
 * 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.
 * 3. Neither the name of Hyra nor the names of its
 *    contributors may be used to endorse or promote products derived from
 *    this software without specific prior written permission.
 *
 * 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 OWNER 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 <sys/errno.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <linux/if_packet.h>
#include <net/ethernet.h>
#include <net/if.h>
#include <stdlib.h>
#include <string.h>
#include <poll.h>
#include "if_ether.h"
#include "dgram.h"
#include "crc.h"

/*
 * Represents datagram parameters to use for
 * dgram_do_send()
 *
 * @dst: Destination MAC address
 * @buf: Buffer to use
 * @len: Length to transmit
 * @type: Packet type to use
 */
struct dgram_params {
    mac_addr_t dst;
    void *buf;
    uint16_t len;
    uint8_t type;
};

/*
 * Send data through a link using specific parameters
 *
 * @link: Link to transmit through
 * @params: Parameters to use
 */
static tx_len_t
dgram_do_send(struct onet_link *link, struct dgram_params *params)
{
    struct sockaddr_ll saddr;
    struct ether_hdr *eth;
    struct onet_dgram *dgram;
    size_t dgram_len;
    char *p, *data;

    dgram_len = DGRAM_LEN(params->len);
    p = malloc(dgram_len);
    if (p == NULL) {
        return -1;
    }

    eth = (struct ether_hdr *)p;
    data = DGRAM_DATA(p);
    dgram = DGRAM_HDR(p);

    /* Copy data to send buffer */
    memset(data, 0, params->len);
    memcpy(data, params->buf, params->len);

    /* Hardware address needs to be big endian */
    params->dst = mac_swap((void *)&params->dst);

    /*
     * Set up link layer sockaddr, load up the frame, datagram
     * and send it off.
     */
    saddr.sll_ifindex = link->iface_idx;
    saddr.sll_halen = HW_ADDR_LEN;
    ether_load_route(link->hwaddr, params->dst, eth);
    dgram_load(params->len, 50, params->type, dgram);
    sendto(
        link->sockfd, p, dgram_len, 0,
        (struct sockaddr *)&saddr, sizeof(struct sockaddr_ll)
    );

    free(p);
    return params->len;
}

/*
 * Squeak back at a machine that squeaks at
 * us.
 *
 * @link: Link to squeak through
 * @src: Who squeaked first?
 * @dst: Who was intended to be squeaked at?
 */
static int
squeak_back(struct onet_link *link, mac_addr_t src, mac_addr_t dst)
{
    if (link == NULL) {
        return -EINVAL;
    }

    /*
     * If one were to spoof their address as the broadcast
     * address, that could end up VERY badly as it would
     * result in a feedback loop.
     */
    if (src == MAC_BROADCAST) {
        return -1;
    }

    /*
     * If we were not intended and this was a not broadcast
     * squeak, it must have been directed to another node.
     */
    if (dst != link->hwaddr && dst != MAC_BROADCAST) {
        return -1;
    }

    return dgram_squeak(link, src);
}

tx_len_t
dgram_send(struct onet_link *link, mac_addr_t dst, void *buf, uint16_t len)
{
    struct dgram_params params;

    if (link == NULL || buf == NULL) {
        return -EINVAL;
    }

    params.dst = dst;
    params.buf = buf;
    params.len = len;
    params.type = OTYPE_DATA;
    return dgram_do_send(link, &params);
}

tx_len_t
dgram_squeak(struct onet_link *link, mac_addr_t dst)
{
    struct dgram_params params;
    uint8_t pad[SQUEAK_NPAD];

    if (link == NULL) {
        return -EINVAL;
    }

    memset(pad, 0, sizeof(pad));
    params.dst = dst;
    params.buf = pad;
    params.len = sizeof(pad);
    params.type = OTYPE_SQUEAK;
    return dgram_do_send(link, &params);
}

rx_len_t
dgram_recv(struct onet_link *link, void *buf, uint16_t len)
{
    const size_t MAX_MISS = 8;
    socklen_t addr_len;
    struct sockaddr_ll saddr;
    struct onet_dgram *o1p_hdr;
    struct ether_hdr *hdr;
    struct pollfd pfd;
    size_t dgram_len, miss_count = 0;
    ssize_t recv_len;
    uint32_t crc;
    uint16_t proto;
    mac_addr_t dest_mac, src_mac;
    char *p;

    if (link == NULL || buf == NULL) {
        return -1;
    }

    if (len == 0) {
        return -1;
    }

    /* Allocate an RX buffer */
    dgram_len = DGRAM_LEN(len);
    p = malloc(dgram_len);
    if (p == NULL) {
        return -1;
    }

    addr_len = sizeof(struct sockaddr_ll);
    saddr.sll_ifindex = link->iface_idx;
    saddr.sll_halen = HW_ADDR_LEN;

    /* Setup poll descriptor */
    pfd.fd = link->sockfd;
    pfd.events = POLLIN;

    /*
     * Wait until we get a packet for us with the right
     * protocol ID.
     */
    for (;;) {
        if (miss_count > MAX_MISS) {
            free(p);
            return -ETIMEDOUT;
        }

        if (poll(&pfd, 1, RECV_POLL_MS) < 0) {
            free(p);
            return -ETIMEDOUT;
        }

        recv_len = recvfrom(
            link->sockfd, p, dgram_len,
            0, (struct sockaddr *)&saddr,
            &addr_len
        );

        hdr = (void *)p;
        proto = ntohs(hdr->proto);
        dest_mac = mac_swap(hdr->dest);
        src_mac = mac_swap(hdr->source);

        if (proto != PROTO_ID) {
            ++miss_count;
            continue;
        }

        o1p_hdr = DGRAM_HDR(p);
        crc = crc32(o1p_hdr, sizeof(*o1p_hdr) - sizeof(crc));
        if (crc != o1p_hdr->crc32) {
            ++miss_count;
            continue;
        }

        link->last_recv = src_mac;

        /* Is this a squeak? */
        if (o1p_hdr->type == OTYPE_SQUEAK) {
            squeak_back(link, src_mac, dest_mac);
            break;
        }

        /* If this is for everyone, take it */
        if (dest_mac == MAC_BROADCAST) {
            break;
        }

        /* If this is for us, take it */
        if (dest_mac == link->hwaddr) {
            break;
        }
    }

    memcpy(buf, DGRAM_DATA(p), len);
    free(p);
    return dgram_len;
}