#include "datagram_raw.h"

#include <errno.h>
#include <stdint.h>
#include <stdlib.h>
#include <string.h>

#include "klog.h"
#include "compat.h"
#include "endpoint.h"

#if COMPAT_HAS_IPV4_RAW

static uint16_t checksum16(const void *data, size_t len)
{
	const uint8_t *p = data;
	uint32_t sum = 0;

	while (len >= 2) {
		sum += ((uint16_t)p[0] << 8) | p[1];
		p += 2;
		len -= 2;
	}

	if (len != 0) {
		sum += ((uint16_t)p[0] << 8);
	}

	while ((sum >> 16) != 0) {
		sum = (sum & 0xffffu) + (sum >> 16);
	}

	return (uint16_t)~sum;
}

int datagram_raw_open_ipv4(evutil_socket_t *out_fd)
{
	evutil_socket_t fd;
	int one = 1;

	if (!out_fd) {
		return -EINVAL;
	}

#ifdef _WIN32
	fd = socket(AF_INET, SOCK_RAW, IPPROTO_UDP);
#else
	fd = socket(AF_INET, SOCK_RAW, IPPROTO_RAW);
#endif
	if (fd < 0) {
		return -compat_socket_errno();
	}

	if (setsockopt(fd, IPPROTO_IP, IP_HDRINCL, compat_setsockopt_optval(&one), sizeof(one)) < 0) {
		int err = compat_socket_errno();
		evutil_closesocket(fd);
		return -err;
	}

	*out_fd = fd;
	return 0;
}

static int endpoint_to_sockaddr_in(
	const struct endpoint *ep,
	struct sockaddr_in *out)
{
	if (!ep || !out) {
		return -EINVAL;
	}

	if (ep->kind != ENDPOINT_INET) {
		return -EINVAL;
	}

	memset(out, 0, sizeof(*out));
	out->sin_family = AF_INET;
	out->sin_port = htons(ep->port);

	if (inet_pton(AF_INET, ep->host, &out->sin_addr) != 1) {
		return -EINVAL;
	}

	return 0;
}

int datagram_raw_send_udp_ipv4(
	evutil_socket_t raw_fd,
	const struct endpoint *src_ep,
	const struct sockaddr *dst_addr,
	socklen_t dst_addr_len,
	const unsigned char *data,
	size_t data_len)
{
	struct sockaddr_in src_addr;
	const struct sockaddr_in *dst;
	struct sockaddr_in send_dst;
	unsigned char *packet;
	struct raw_ipv4_hdr *ip;
	struct raw_udp_hdr *udp;
	size_t ip_len = sizeof(*ip);
	size_t udp_len = sizeof(*udp);
	size_t packet_len;
	ssize_t n;
	int rc;

	if (raw_fd < 0 || !src_ep || !dst_addr || !data) {
		return -EINVAL;
	}

	if (dst_addr_len < (socklen_t)sizeof(struct sockaddr_in)) {
		return -EINVAL;
	}

	if (dst_addr->sa_family != AF_INET) {
		return -EAFNOSUPPORT;
	}

	rc = endpoint_to_sockaddr_in(src_ep, &src_addr);
	if (rc != 0) {
		return rc;
	}

	dst = (const struct sockaddr_in *)dst_addr;

	if (data_len > UINT16_MAX - ip_len - udp_len) {
		return -EMSGSIZE;
	}

	packet_len = ip_len + udp_len + data_len;

	packet = calloc(1, packet_len);
	if (!packet) {
		return -ENOMEM;
	}

	ip = (struct raw_ipv4_hdr *)packet;
	udp = (struct raw_udp_hdr *)(packet + ip_len);

	ip->vhl = (uint8_t)((4u << 4) | 5u);
	ip->tos = 0;
	ip->len = compat_raw_ip_len((uint16_t)packet_len);
	ip->id = 0;
	ip->off = compat_raw_ip_off(0);
	ip->ttl = 64;
	ip->proto = IPPROTO_UDP;
	ip->sum = 0;
	ip->src = src_addr.sin_addr.s_addr;
	ip->dst = dst->sin_addr.s_addr;

	udp->sport = src_addr.sin_port;
	udp->dport = dst->sin_port;
	udp->len = htons((uint16_t)(udp_len + data_len));
	udp->sum = 0;

	memcpy(packet + ip_len + udp_len, data, data_len);

	ip->sum = checksum16(ip, ip_len);

	/*
	 * For IPv4, UDP checksum 0 is legal and means "no checksum".
	 * Keep this simple for now; it avoids checksum/offload weirdness while
	 * testing spoofed discovery replies.
	 */
	udp->sum = 0;

	memset(&send_dst, 0, sizeof(send_dst));
	send_dst.sin_family = AF_INET;
	send_dst.sin_addr = dst->sin_addr;
	send_dst.sin_port = 0;

#ifdef __APPLE__
	send_dst.sin_len = sizeof(send_dst);
#endif

	n = sendto(
		raw_fd,
		compat_send_buf(packet),
		compat_send_len(packet_len),
		0,
		(const struct sockaddr *)&send_dst,
		sizeof(send_dst)
	);

	free(packet);

	if (n < 0) {
#ifdef _WIN32
		LOG_ERROR("datagram raw send failed",
				"wsaerr", _LOGV(WSAGetLastError()),
				"packet_len", _LOGV(packet_len),
				"src_ip", _LOGV(src_addr.sin_addr.s_addr),
				"src_port", _LOGV(ntohs(src_addr.sin_port)),
				"dst_ip", _LOGV(dst->sin_addr.s_addr),
				"dst_port", _LOGV(ntohs(dst->sin_port))
				);
#else
		LOG_ERROR("datagram raw send failed",
				"err", _LOGV(strerror(errno)),
				"packet_len", _LOGV(packet_len),
				"src_port", _LOGV(ntohs(src_addr.sin_port)),
				"dst_port", _LOGV(ntohs(dst->sin_port))
				);
#endif
		return -compat_socket_errno();
	}

	if ((size_t)n != packet_len) {
		return -EIO;
	}

	return 0;
}

#else

int datagram_raw_open_ipv4(evutil_socket_t *out_fd)
{
	(void)out_fd;
	return -ENOTSUP;
}

int datagram_raw_send_udp_ipv4(
	evutil_socket_t raw_fd,
	const struct endpoint *src_ep,
	const struct sockaddr *dst_addr,
	socklen_t dst_addr_len,
	const unsigned char *data,
	size_t data_len)
{
	(void)raw_fd;
	(void)src_ep;
	(void)dst_addr;
	(void)dst_addr_len;
	(void)data;
	(void)data_len;
	LOG_ERROR("IPv4 raw is not supported on this platform");

	return -ENOTSUP;
}

#endif