request_gettable.c

/*
 * Part of `snmp-query-engine`.
 *
 * Copyright 2012-2013, Anton Berezin <tobez@tobez.org>
 * Modified BSD license.
 * (See LICENSE file in the distribution.)
 *
 */
#include "sqe.h"

/*
 * gettable request:
 * [ 0, $cid, $ip, $port, $oid, optional $max_repetitions ]
 *
 */

int
handle_gettable_request(struct socket_info *si, unsigned cid, msgpack_object *o)
{
	unsigned port = 65536;
	struct in_addr ip;
	struct client_requests_info *cri;
	struct cid_info *ci;
	struct oid_info *oi;
	int max_repetitions;

	if (o->via.array.size != 5 && o->via.array.size != 6)
		return error_reply(si, RT_GETTABLE|RT_ERROR, cid, "bad request length");

	if (o->via.array.ptr[RI_GETTABLE_PORT].type == MSGPACK_OBJECT_POSITIVE_INTEGER)
		port = o->via.array.ptr[RI_GETTABLE_PORT].via.u64;
	if (port > 65535)
		return error_reply(si, RT_GETTABLE|RT_ERROR, cid, "bad port number");

	if (!object2ip(&o->via.array.ptr[RI_GETTABLE_IP], &ip))
		return error_reply(si, RT_GETTABLE|RT_ERROR, cid, "bad IP");

	cri = get_client_requests_info(&ip, port, si);
	max_repetitions = cri->dest->max_repetitions;

	if (o->via.array.size == 6) {
		max_repetitions = -1;
		if (o->via.array.ptr[RI_GETTABLE_MREP].type == MSGPACK_OBJECT_POSITIVE_INTEGER)
			max_repetitions = o->via.array.ptr[RI_GETTABLE_MREP].via.u64;
		if (max_repetitions < 1 || max_repetitions > 255)
			return error_reply(si, RT_GETTABLE|RT_ERROR, cid, "bad max repetitions");
	}

	ci = get_cid_info(cri, cid);
	if (ci->n_oids != 0)
		return error_reply(si, RT_GETTABLE|RT_ERROR, cid, "duplicate request id");

	if (!( oi = allocate_oid_info(&o->via.array.ptr[RI_GETTABLE_OID], ci)))
		return error_reply(si, RT_GETTABLE|RT_ERROR, cid, "bad oid");
	oi->last_known_table_entry = oi;
	oi->max_repetitions = max_repetitions;
	ci->n_oids++;

	TAILQ_INSERT_TAIL(&cri->oids_to_query, oi, oid_list);

	PS.gettable_requests++;
	si->PS.gettable_requests++;
	maybe_query_destination(cri->dest);
	return 0;
}