#include <sys/socket.h>

#include <ber.h>
#include <err.h>
#include <stdint.h>
#include <stdio.h>
#include <string.h>
#include <unistd.h>

#include "regress.h"

#define MIB_SUBAGENT_TRANSPORT_TCP MIB_SUBAGENT_TRANSPORT, 1
#define MIB_TRANSPORT_TCP MIB_TRANSPORT, 1

void
transport_tcp_get(void)
{
        struct sockaddr_storage ss;
        struct sockaddr *sa = (struct sockaddr *)&ss;
        socklen_t salen;
        int snmp_s, ax_s;
        uint32_t sessionid;
        struct varbind varbind = {
                .type = TYPE_NULL,
                .name = OID_STRUCT(MIB_TRANSPORT_TCP, 1, 0),
                .data.int32 = 1
        };
        int32_t requestid;
        char buf[1024];
        size_t n;

        ax_s = agentx_connect(axsocket);
        sessionid = agentx_open(ax_s, 0, 0,
            OID_ARG(MIB_SUBAGENT_TRANSPORT_TCP, 1), __func__);
        agentx_register(ax_s, sessionid, 0, 0, 127, 0,
            OID_ARG(MIB_TRANSPORT_TCP, 1), 0);

        salen = snmp_resolve(SOCK_STREAM, hostname, servname, sa);
        snmp_s = snmp_connect(SOCK_STREAM, sa, salen);
        requestid = snmpv2_get(snmp_s, community, 0, &varbind, 1);

        n = agentx_read(ax_s, buf, sizeof(buf), 1000);
        agentx_get_handle(__func__, buf, n, 0, sessionid, &varbind, 1);

        varbind.type = TYPE_INTEGER;
        agentx_response(ax_s, buf, NOERROR, 0, &varbind, 1);

        snmpv2_response_validate(snmp_s, 1000, community, requestid, 0, 0,
            &varbind, 1);
}

void
transport_tcp_disconnect(void)
{
        struct sockaddr_storage ss;
        struct sockaddr *sa = (struct sockaddr *)&ss;
        socklen_t salen;
        int snmp_s, ax_s;
        uint32_t sessionid;
        struct varbind varbind = {
                .type = TYPE_NULL,
                .name = OID_STRUCT(MIB_TRANSPORT_TCP, 2, 0),
                .data.int32 = 1
        };
        int32_t requestid;
        char buf[1024];
        size_t n;

        ax_s = agentx_connect(axsocket);
        sessionid = agentx_open(ax_s, 0, 0,
            OID_ARG(MIB_SUBAGENT_TRANSPORT_TCP, 2), __func__);
        agentx_register(ax_s, sessionid, 0, 0, 127, 0,
            OID_ARG(MIB_TRANSPORT_TCP, 2), 0);

        salen = snmp_resolve(SOCK_STREAM, hostname, servname, sa);
        snmp_s = snmp_connect(SOCK_STREAM, sa, salen);
        requestid = snmpv2_get(snmp_s, community, 0, &varbind, 1);

        close(snmp_s);

        n = agentx_read(ax_s, buf, sizeof(buf), 1000);
        agentx_get_handle(__func__, buf, n, 0, sessionid, &varbind, 1);

        varbind.type = TYPE_INTEGER;
        agentx_response(ax_s, buf, NOERROR, 0, &varbind, 1);
}

void
transport_tcp_double_get_disconnect(void)
{
        struct sockaddr_storage ss;
        struct sockaddr *sa = (struct sockaddr *)&ss;
        socklen_t salen;
        int snmp_s, ax_s;
        uint32_t sessionid;
        struct varbind varbind = {
                .type = TYPE_NULL,
                .name = OID_STRUCT(MIB_TRANSPORT_TCP, 2, 0),
                .data.int32 = 1
        };
        struct ber_element *message;
        struct ber ber = {};
        char buf[1024];
        ssize_t buflen, writelen;
        void *ptr;
        size_t n;

        ax_s = agentx_connect(axsocket);
        sessionid = agentx_open(ax_s, 0, 0,
            OID_ARG(MIB_SUBAGENT_TRANSPORT_TCP, 2), __func__);
        agentx_register(ax_s, sessionid, 0, 0, 127, 0,
            OID_ARG(MIB_TRANSPORT_TCP, 2), 0);

        salen = snmp_resolve(SOCK_STREAM, hostname, servname, sa);
        snmp_s = snmp_connect(SOCK_STREAM, sa, salen);

        message = snmpv2_build(community, REQUEST_GET, 0, 0, 0, &varbind, 1);

        if (ober_write_elements(&ber, message) == -1)
                err(1, "ober_write_elements");

        buflen = ober_get_writebuf(&ber, &ptr);
        /* send two copies in a single window */
        memcpy(buf, ptr, buflen);
        memcpy(buf + buflen, ptr, buflen);

        if ((writelen = write(snmp_s, buf, buflen * 2)) == -1)
                err(1, "write");
        if (writelen != buflen * 2)
                errx(1, "write: short write");

        if (verbose) {
                printf("SNMP send(%d):\n", snmp_s);
                smi_debug_elements(message);
        }
        ober_free_elements(message);
        ober_free(&ber);
        
        close(snmp_s);

        n = agentx_read(ax_s, buf, sizeof(buf), 1000);
        agentx_get_handle(__func__, buf, n, 0, sessionid, &varbind, 1);

        varbind.type = TYPE_INTEGER;
        agentx_response(ax_s, buf, NOERROR, 0, &varbind, 1);

        n = agentx_read(ax_s, buf, sizeof(buf), 1000);
        agentx_get_handle(__func__, buf, n, 0, sessionid, &varbind, 1);
        agentx_response(ax_s, buf, NOERROR, 0, &varbind, 1);
}