#include <net-snmp/net-snmp-config.h>
#include <net-snmp/net-snmp-includes.h>
#include <net-snmp/agent/net-snmp-agent-includes.h>
#include "updates.h"

static int
handle_updates(netsnmp_mib_handler *handler,
               netsnmp_handler_registration *reginfo,
               netsnmp_agent_request_info *reqinfo,
               netsnmp_request_info *requests)
{
#ifndef NETSNMP_NO_WRITE_SUPPORT
    int *set = (int*)handler->myvoid;

    if (reqinfo->mode == MODE_SET_RESERVE1 && *set < 0)
        netsnmp_request_set_error(requests, SNMP_ERR_NOTWRITABLE);
    else if (reqinfo->mode == MODE_SET_COMMIT) {
        *set = 1;
        snmp_store_needed(reginfo->handlerName);
    }
#endif /* NETSNMP_NO_WRITE_SUPPORT */
    return SNMP_ERR_NOERROR;
}

netsnmp_handler_registration*
netsnmp_create_update_handler_registration(
    const char* name, const oid* id, size_t idlen, int mode, int* set)
{
    netsnmp_handler_registration *res;
    netsnmp_mib_handler *hnd;

    hnd = netsnmp_create_handler("update", handle_updates);
    if (hnd == NULL)
        return NULL;

    hnd->flags |= MIB_HANDLER_AUTO_NEXT;
    hnd->myvoid = set;

    res = netsnmp_handler_registration_create(name, hnd, id, idlen, mode);
    if (res == NULL) {
        netsnmp_handler_free(hnd);
        return NULL;
    }

    return res;
}