/*
 * Copyright (c) 2026 The FreeBSD Foundation
 *
 * This software was developed by Mark Johnston under sponsorship from
 * the FreeBSD Foundation.
 *
 * SPDX-License-Identifier: BSD-2-Clause
 */

/*
 * A minimal regression test for a buffer overflow in alias_rtsp_out().
 */

#include <sys/types.h>
#include <sys/sbuf.h>

#include <arpa/inet.h>
#include <netinet/in.h>
#include <netinet/ip.h>
#include <netinet/tcp_var.h>

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

#include <alias.h>

int
main(void)
{
        uint8_t *packet;
        struct ip ip;
        struct tcphdr tcp;
        struct sbuf sb;
        struct libalias *la;

        sbuf_new(&sb, NULL, 0, SBUF_AUTOEXTEND);
        sbuf_printf(&sb, "SETUP rtsp://example.com/media.mp4 RTSP/1.0\r\n");
        sbuf_printf(&sb, "CSeq: 1\r\n");
        sbuf_printf(&sb, "Transport: RTP/AVP;unicast;");
        for (int i = 0; i < 200; i++)
                sbuf_printf(&sb, "client_port=%d-%d;", 2 * i, 2 * i + 1);
        sbuf_printf(&sb, "\r\n\r\n");
        sbuf_finish(&sb);

        memset(&tcp, 0, sizeof(tcp));
        tcp.th_sport = htons(1234);
        tcp.th_dport = htons(554);
        tcp.th_off = 5;

        memset(&ip, 0, sizeof(ip));
        ip.ip_v = IPVERSION;
        ip.ip_hl = sizeof(ip) / 4;
        ip.ip_len = htons(sizeof(ip) + sizeof(tcp) + sbuf_len(&sb));
        ip.ip_id = htons(1);
        ip.ip_ttl = 64;
        ip.ip_p = IPPROTO_TCP;
        ip.ip_src.s_addr = inet_addr("127.0.0.1");
        ip.ip_dst.s_addr = inet_addr("127.0.0.2");

        packet = malloc(sizeof(ip) + sizeof(tcp) + sbuf_len(&sb));
        memcpy(packet, &ip, sizeof(ip));
        memcpy(packet + sizeof(ip), &tcp, sizeof(tcp));
        memcpy(packet + sizeof(ip) + sizeof(tcp), sbuf_data(&sb),
            sbuf_len(&sb));

        la = LibAliasInit(NULL);
        LibAliasSetAddress(la,
            (struct in_addr){.s_addr = inet_addr("127.0.0.1")});
        if (LibAliasOut(la, packet, sizeof(ip) + sizeof(tcp) + sbuf_len(&sb)) !=
            PKT_ALIAS_OK)
                return (1);
        LibAliasUninit(la);
        return (0);
}