/*	$OpenBSD: mrt.c,v 1.14 2004/01/01 21:18:13 henning Exp $ */

/*
 * Copyright (c) 2003 Claudio Jeker <claudio@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 <sys/queue.h>

#include <errno.h>
#include <fcntl.h>
#include <stdlib.h>
#include <string.h>
#include <time.h>
#include <unistd.h>

#include "bgpd.h"
#include "rde.h"
#include "session.h"	/* needed for MSGSIZE_HEADER et al. */

#include "mrt.h"

/*
 * XXX These functions break the imsg encapsulation.
 * XXX The imsg API is way to basic, we need something like
 * XXX imsg_create(), imsg_add(), imsg_close() ...
 */

static int	mrt_dump_entry(struct mrt *, struct prefix *, u_int16_t,
		    struct peer_config *);
static void	mrt_dump_header(struct buf *, u_int16_t, u_int16_t, u_int32_t);
static int	mrt_open(struct mrtdump_config *);

#define DUMP_BYTE(x, b)							\
	do {								\
		u_char		t = (b);				\
		if (buf_add((x), &t, sizeof(t)) == -1)			\
			fatalx("buf_add error");			\
	} while (0)

#define DUMP_SHORT(x, s)						\
	do {								\
		u_int16_t	t;					\
		t = htons((s));						\
		if (buf_add((x), &t, sizeof(t)) == -1)			\
			fatalx("buf_add error");			\
	} while (0)

#define DUMP_LONG(x, l)							\
	do {								\
		u_int32_t	t;					\
		t = htonl((l));						\
		if (buf_add((x), &t, sizeof(t)) == -1)			\
			fatalx("buf_add error");			\
	} while (0)

#define DUMP_NLONG(x, l)						\
	do {								\
		u_int32_t	t = (l);				\
		if (buf_add((x), &t, sizeof(t)) == -1)			\
			fatalx("buf_add error");			\
	} while (0)

int
mrt_dump_bgp_msg(struct mrt *mrt, void *pkg, u_int16_t pkglen, int type,
    struct peer_config *peer, struct bgpd_config *bgp)
{
	struct buf	*buf;
	struct imsg_hdr	 hdr;
	int		 i, n;
	u_int16_t	 len;

	len = pkglen + MRT_BGP4MP_HEADER_SIZE + type > 0 ? MSGSIZE_HEADER : 0;

	hdr.len = len + IMSG_HEADER_SIZE + MRT_HEADER_SIZE;
	hdr.type = IMSG_MRT_MSG;
	hdr.peerid = mrt->id;
	buf = buf_open(hdr.len);
	if (buf == NULL)
		fatal("mrt_dump_bgp_msg");
	if (buf_add(buf, &hdr, sizeof(hdr)) == -1)
		fatalx("buf_add error");

	mrt_dump_header(buf, MSG_PROTOCOL_BGP4MP, BGP4MP_MESSAGE, len);

	DUMP_SHORT(buf, bgp->as);
	DUMP_SHORT(buf, peer->remote_as);
	DUMP_SHORT(buf, /* ifindex */ 0);
	DUMP_SHORT(buf, 4);
	DUMP_NLONG(buf, peer->local_addr.sin_addr.s_addr);
	DUMP_NLONG(buf, peer->remote_addr.sin_addr.s_addr);

	/* bgp header was chopped off so glue a new one together. */
	if (type > 0) {
		for (i = 0; i < MSGSIZE_HEADER_MARKER; i++)
			DUMP_BYTE(buf, 0xff);
		DUMP_SHORT(buf, pkglen + MSGSIZE_HEADER);
		DUMP_BYTE(buf, type);
	}

	if (buf_add(buf, pkg, pkglen) == -1)
		fatalx("buf_add error");

	if ((n = buf_close(mrt->msgbuf, buf)) < 0)
		fatalx("buf_close error");

	return (n);
}

static int
mrt_dump_entry(struct mrt *mrt, struct prefix *p, u_int16_t snum,
    struct peer_config *peer)
{
	struct buf	*buf;
	void		*bptr;
	struct imsg_hdr	 hdr;
	u_int16_t	 len, attr_len;
	int		 n;

	attr_len = attr_length(&p->aspath->flags);
	len = MRT_DUMP_HEADER_SIZE + attr_len;

	hdr.len = len + IMSG_HEADER_SIZE + MRT_HEADER_SIZE;
	hdr.type = IMSG_MRT_MSG;
	hdr.peerid = mrt->id;
	buf = buf_open(hdr.len);
	if (buf == NULL)
		fatal("mrt_dump_entry");
	if (buf_add(buf, &hdr, sizeof(hdr)) == -1)
		fatalx("buf_add error");

	mrt_dump_header(buf, MSG_TABLE_DUMP, AFI_IPv4, len);

	DUMP_SHORT(buf, 0);
	DUMP_SHORT(buf, snum);
	DUMP_NLONG(buf, p->prefix->prefix.s_addr);
	DUMP_BYTE(buf, p->prefix->prefixlen);
	DUMP_BYTE(buf, 1);
	DUMP_LONG(buf, p->lastchange);	/* originated */
	DUMP_NLONG(buf, peer->remote_addr.sin_addr.s_addr);
	DUMP_SHORT(buf, peer->remote_as);
	DUMP_SHORT(buf, attr_len);

	if ((bptr = buf_reserve(buf, attr_len)) == NULL)
		fatalx("buf_reserve error");

	if (attr_dump(bptr, attr_len, &p->aspath->flags) == -1)
		fatalx("attr_dump error");

	if ((n = buf_close(mrt->msgbuf, buf)) < 0)
		fatalx("buf_close error");

	return (n);
}

static u_int16_t sequencenum = 0;

void
mrt_clear_seq(void)
{
	sequencenum = 0;
}

void
mrt_dump_upcall(struct pt_entry *pt, void *ptr)
{
	struct mrt	*mrtbuf = ptr;
	struct prefix	*p;

	/*
	 * dump all prefixes even the inactive ones. That is the way zebra
	 * dumps the table so we do the same. If only the active route should
	 * be dumped p should be set to p = pt->active.
	 */
	LIST_FOREACH(p, &pt->prefix_h, prefix_l)
		mrt_dump_entry(mrtbuf, p, sequencenum++, &p->peer->conf);
}

static void
mrt_dump_header(struct buf *buf, u_int16_t type, u_int16_t subtype,
    u_int32_t len)
{
	struct mrt_header	mrt;
	time_t			now;

	now = time(NULL);

	mrt.timestamp = htonl(now);
	mrt.type = ntohs(type);
	mrt.subtype = ntohs(subtype);
	mrt.length = htonl(len);

	if (buf_add(buf, &mrt, sizeof(mrt)) == -1)
		fatalx("buf_add error");
}

static int
mrt_open(struct mrtdump_config *mconf)
{
	time_t	now;

	now = time(NULL);
	if (strftime(mconf->file, sizeof(mconf->file), mconf->name,
		    localtime(&now)) == 0) {
		logit(LOG_CRIT, "mrt_open strftime failed");
		mconf->msgbuf.sock = -1;
		return -1;
	}

	mconf->msgbuf.sock = open(mconf->file,
	    O_WRONLY|O_NONBLOCK|O_CREAT|O_TRUNC, 0644);
	if (mconf->msgbuf.sock == -1)
		logit(LOG_CRIT, "mrt_open %s: %s",
		    mconf->file, strerror(errno));

	return mconf->msgbuf.sock;
}

int
mrt_state(struct mrtdump_config *m, enum imsg_type type,
    struct imsgbuf *buf)
{
	switch (m->state) {
	case MRT_STATE_DONE:
		/* no dump expected */
		return (0);
	case MRT_STATE_CLOSE:
		switch (type) {
		case IMSG_NONE:
			if (m->type == MRT_TABLE_DUMP)
				imsg_compose(buf, IMSG_MRT_END, m->id, NULL, 0);
			return (0);
		case IMSG_MRT_END:
			/* dump no longer valid */
			close(m->msgbuf.sock);
			LIST_REMOVE(m, list);
			free(m);
			return (0);
		default:
			break;
		}
		break;
	case MRT_STATE_REOPEN:
		switch (type) {
		case IMSG_NONE:
			if (m->type == MRT_TABLE_DUMP)
				imsg_compose(buf, IMSG_MRT_END, m->id, NULL, 0);
			return (0);
		case IMSG_MRT_END:
			if (m->msgbuf.sock != -1)
				close(m->msgbuf.sock);
			m->state = MRT_STATE_OPEN;
			if (m->type == MRT_TABLE_DUMP)
				imsg_compose(buf, IMSG_MRT_REQ, m->id, NULL, 0);
			return (0);
		default:
			break;
		}
		break;
	case MRT_STATE_OPEN:
		switch (type) {
		case IMSG_NONE:
			if (m->type == MRT_TABLE_DUMP)
				imsg_compose(buf, IMSG_MRT_REQ, m->id, NULL, 0);
			return (0);
		case IMSG_MRT_MSG:
			mrt_open(m);
			m->state = MRT_STATE_RUNNING;
			break;
		default:
			return (0);
		}
		break;
	case MRT_STATE_RUNNING:
		if (type == IMSG_MRT_END) {
			if (m->msgbuf.sock != -1)
				close(m->msgbuf.sock);
			m->msgbuf.sock = -1;
			m->state = MRT_STATE_DONE;
			return (0);
		}
		break;
	}
	return (1);
}

int
mrt_usr1(struct mrt_config *mconf, struct imsgbuf *buf)
{
	struct mrtdump_config	*m;
	time_t			 now;
	int			 interval = INT_MAX;

	now = time(NULL);
	LIST_FOREACH(m, mconf, list) {
		if (m->type == MRT_TABLE_DUMP) {
			switch (m->state) {
			case MRT_STATE_REOPEN:
			case MRT_STATE_CLOSE:
				break;
			case MRT_STATE_DONE:
				m->state = MRT_STATE_OPEN;
				imsg_compose(buf, IMSG_MRT_REQ, m->id, NULL, 0);
				break;
			default:
				m->state = MRT_STATE_REOPEN;
				imsg_compose(buf, IMSG_MRT_END, m->id, NULL, 0);
				break;
			}

			m->ReopenTimer = now + m->ReopenTimerInterval;
		}
		if (m->ReopenTimer - now < interval)
			interval = m->ReopenTimer - now;
	}

	if (interval != INT_MAX)
		alarm(interval);

	return (0);
}

int
mrt_alrm(struct mrt_config *mconf, struct imsgbuf *buf)
{
	struct mrtdump_config	*m;
	time_t			 now;
	int			 interval = INT_MAX;

	now = time(NULL);
	LIST_FOREACH(m, mconf, list) {
		if (m->ReopenTimerInterval == 0)
			continue;
		if (m->ReopenTimer <= now) {
			switch (m->state) {
			case MRT_STATE_REOPEN:
			case MRT_STATE_CLOSE:
				break;
			case MRT_STATE_DONE:
				m->state = MRT_STATE_OPEN;
				if (m->type == MRT_TABLE_DUMP)
					imsg_compose(buf, IMSG_MRT_REQ, m->id,
					    NULL, 0);
				break;
			default:
				m->state = MRT_STATE_REOPEN;
				if (m->type == MRT_TABLE_DUMP)
					imsg_compose(buf, IMSG_MRT_END, m->id,
					    NULL, 0);
				break;
			}

			m->ReopenTimer = now + m->ReopenTimerInterval;
		}
		if (m->ReopenTimer - now < interval)
			interval = m->ReopenTimer - now;
	}

	if (interval != INT_MAX)
		alarm(interval);

	return (0);
}

static u_int32_t	 max_id = 1;

static struct mrtdump_config *
getconf(struct mrt_config *c, struct mrtdump_config *m)
{
	struct mrtdump_config	*t;
	LIST_FOREACH(t, c, list)
		if (t->type == m->type)
			return t;
	return (NULL);
}

int
mrt_mergeconfig(struct mrt_config *xconf, struct mrt_config *nconf)
{
	struct mrtdump_config	*m, *xm;
	time_t			 now;
	int			 interval = INT_MAX;

	now = time(NULL);
	LIST_FOREACH(m, nconf, list)
		if ((xm = getconf(xconf, m)) == NULL) {
			/* NEW */
			if ((xm = calloc(1, sizeof(struct mrtdump_config))) ==
			    NULL)
				fatal("mrt_mergeconfig");
			memcpy(xm, m, sizeof(struct mrtdump_config));
			xm->id = max_id++;
			if (xm->ReopenTimerInterval != 0) {
				xm->ReopenTimer = now + xm->ReopenTimerInterval;
				interval = xm->ReopenTimerInterval < interval ?
				    xm->ReopenTimerInterval : interval;
			}
			xm->state=MRT_STATE_OPEN;
			LIST_INSERT_HEAD(xconf, xm, list);
		} else {
			/* MERGE */
			if (strlcpy(xm->name, m->name, sizeof(xm->name)) >
			    sizeof(xm->name))
				fatalx("mrt_mergeconfig: strlcpy");
			xm->ReopenTimerInterval = m->ReopenTimerInterval;
			if (xm->ReopenTimerInterval != 0) {
				xm->ReopenTimer = now + xm->ReopenTimerInterval;
				interval = xm->ReopenTimerInterval < interval ?
				    xm->ReopenTimerInterval : interval;
			}
			xm->state=MRT_STATE_REOPEN;
		}
	LIST_FOREACH(xm, xconf, list)
		if (getconf(nconf, xm) == NULL)
			/* REMOVE */
			xm->state = MRT_STATE_CLOSE;

	/* free config */
	for (m = LIST_FIRST(nconf); m != LIST_END(nconf); m = xm) {
		xm = LIST_NEXT(m, list);
		free(m);
	}

	if (interval != INT_MAX)
		alarm(interval);

	return (0);
}