/*	$OpenBSD: query.c,v 1.5 2016/09/02 16:46:29 renato Exp $ */

/*
 * Copyright (c) 2015 Renato Westphal <renato@openbsd.org>
 *
 * Permission to use, copy, modify, and distribute this software for any
 * purpose with or without fee is hereby granted, provided that the above
 * copyright notice and this permission notice appear in all copies.
 *
 * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
 * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
 * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
 * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
 * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
 * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
 * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
 */

#include <sys/types.h>
#include <netinet/in.h>
#include <netinet/ip.h>
#include <netinet/ip6.h>

#include <stdlib.h>

#include "eigrpd.h"
#include "eigrpe.h"
#include "log.h"

/* query packet handling */

void
send_query(struct eigrp_iface *ei, struct nbr *nbr,
    struct rinfo_head *rinfo_list, int siaquery)
{
	struct eigrp		*eigrp = ei->eigrp;
	struct ibuf		*buf;
	uint16_t		 opcode;
	struct rinfo_entry	*re;
	int			 size;
	int			 route_len;

	if (rinfo_list == NULL || TAILQ_EMPTY(rinfo_list))
		return;

	do {
		if ((buf = ibuf_dynamic(PKG_DEF_SIZE,
		    IP_MAXPACKET - sizeof(struct ip))) == NULL)
			fatal("send_query");

		if (!siaquery)
			 opcode = EIGRP_OPC_QUERY;
		else
			 opcode = EIGRP_OPC_SIAQUERY;

		/* EIGRP header */
		if (gen_eigrp_hdr(buf, opcode, 0, eigrp->seq_num, eigrp->as))
			goto fail;

		switch (eigrp->af) {
		case AF_INET:
			size = sizeof(struct ip);
			break;
		case AF_INET6:
			size = sizeof(struct ip6_hdr);
			break;
		default:
			fatalx("send_query: unknown af");
		}
		size += sizeof(struct eigrp_hdr);

		while ((re = TAILQ_FIRST(rinfo_list)) != NULL) {
			route_len = len_route_tlv(&re->rinfo);
			/* don't exceed the MTU to avoid IP fragmentation */
			if (size + route_len > ei->iface->mtu) {
				rtp_send(ei, nbr, buf);
				break;
			}
			size += route_len;

			if (gen_route_tlv(buf, &re->rinfo))
				goto fail;
			TAILQ_REMOVE(rinfo_list, re, entry);
			free(re);
		}
	} while (!TAILQ_EMPTY(rinfo_list));

	rtp_send(ei, nbr, buf);
	return;
fail:
	log_warnx("%s: failed to send message", __func__);
	if (rinfo_list)
		message_list_clr(rinfo_list);
	ibuf_free(buf);
	return;
}

void
recv_query(struct nbr *nbr, struct rinfo_head *rinfo_list, int siaquery)
{
	int			 type;
	struct rinfo_entry	*re;

	rtp_ack_start_timer(nbr);

	if (!siaquery)
		type = IMSG_RECV_QUERY;
	else
		type = IMSG_RECV_SIAQUERY;

	TAILQ_FOREACH(re, rinfo_list, entry)
		eigrpe_imsg_compose_rde(type, nbr->peerid, 0, &re->rinfo,
		    sizeof(re->rinfo));
}