/*-
 * SPDX-License-Identifier: BSD-2-Clause
 *
 * Copyright (c) 2009 Hiroki Sato.  All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the distribution.
 *
 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
 * IN NO EVENT SHALL THE AUTHOR OR HIS RELATIVES BE LIABLE FOR ANY DIRECT,
 * INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
 * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
 * SERVICES; LOSS OF MIND, USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
 * STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING
 * IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
 * THE POSSIBILITY OF SUCH DAMAGE.
 */

#include <sys/param.h>
#include <sys/ioctl.h>
#include <sys/socket.h>
#include <sys/sockio.h>

#include <stdlib.h>
#include <unistd.h>

#include <net/ethernet.h>
#include <net/if.h>
#include <net/if_gif.h>
#ifdef WITHOUT_NETLINK
#include <net/route.h>
#endif

#include <ctype.h>
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <unistd.h>
#include <err.h>
#include <errno.h>

#include "ifconfig.h"

#ifndef WITHOUT_NETLINK
#include "ifconfig_netlink.h"
#endif

static const char *GIFBITS[] = {
	[0] = "NOCLAMP",
	[1] = "IGNORE_SOURCE",
};

#ifndef WITHOUT_NETLINK
struct nl_parsed_gif {
	uint32_t	ifla_flags;
};

struct nla_gif_info {
	const char		*kind;
	struct nl_parsed_gif	data;
};

struct nla_gif_link {
	uint32_t		ifi_index;
	struct nla_gif_info	linkinfo;
};

static inline void
gif_nl_init(if_ctx *ctx, struct snl_writer *nw, uint32_t flags)
{
	struct nlmsghdr *hdr;

	snl_init_writer(ctx->io_ss, nw);
	hdr = snl_create_msg_request(nw, NL_RTM_NEWLINK);
	hdr->nlmsg_flags |= flags;
	snl_reserve_msg_object(nw, struct ifinfomsg);
	snl_add_msg_attr_string(nw, IFLA_IFNAME, ctx->ifname);
}

static inline void
gif_nl_fini(if_ctx *ctx, struct snl_writer *nw)
{
	struct nlmsghdr *hdr;
	struct snl_errmsg_data errmsg = {};

	hdr = snl_finalize_msg(nw);
	if (hdr == NULL || !snl_send_message(ctx->io_ss, hdr))
		err(1, "unable to send netlink message");

	if (!snl_read_reply_code(ctx->io_ss, hdr->nlmsg_seq, &errmsg))
		errx(errmsg.error, "%s", errmsg.error_str);
}

#define _OUT(_field)   offsetof(struct nl_parsed_gif, _field)
static const struct snl_attr_parser nla_p_gif[] = {
	{ .type = IFLA_IPTUN_FLAGS, .off = _OUT(ifla_flags), .cb = snl_attr_get_uint32 },
};
#undef _OUT
SNL_DECLARE_ATTR_PARSER(gif_linkinfo_data_parser, nla_p_gif);

#define _OUT(_field)   offsetof(struct nla_gif_info, _field)
static const struct snl_attr_parser ap_gif_linkinfo[] = {
	{ .type = IFLA_INFO_KIND, .off = _OUT(kind), .cb = snl_attr_get_string },
	{ .type = IFLA_INFO_DATA, .off = _OUT(data),
		.arg = &gif_linkinfo_data_parser, .cb = snl_attr_get_nested },
};
#undef _OUT
SNL_DECLARE_ATTR_PARSER(gif_linkinfo_parser, ap_gif_linkinfo);

#define _IN(_field)    offsetof(struct ifinfomsg, _field)
#define _OUT(_field)   offsetof(struct nla_gif_link, _field)
static const struct snl_attr_parser ap_gif_link[] = {
	{ .type = IFLA_LINKINFO, .off = _OUT(linkinfo),
		.arg = &gif_linkinfo_parser, .cb = snl_attr_get_nested },
};

static const struct snl_field_parser fp_geneve_link[] = {
	{ .off_in = _IN(ifi_index), .off_out = _OUT(ifi_index),
		.cb = snl_field_get_uint32 },
};
#undef _IN
#undef _OUT
SNL_DECLARE_PARSER(gif_parser, struct ifinfomsg, fp_geneve_link, ap_gif_link);

static const struct snl_hdr_parser *all_parsers[] = {
	&gif_linkinfo_data_parser,
	&gif_linkinfo_parser,
	&gif_parser,
};

static void
gif_status_nl(if_ctx *ctx)
{
	struct snl_writer nw;
	struct nlmsghdr *hdr;
	struct snl_errmsg_data errmsg = {};
	struct nla_gif_link gif_link;

	if (strncmp(ctx->ifname, "gif", sizeof("gif") - 1) != 0)
		 return;

	snl_init_writer(ctx->io_ss, &nw);
	hdr = snl_create_msg_request(&nw, NL_RTM_GETLINK);
	hdr->nlmsg_flags |= NLM_F_DUMP;
	snl_reserve_msg_object(&nw, struct ifinfomsg);
	snl_add_msg_attr_string(&nw, IFLA_IFNAME, ctx->ifname);

	hdr = snl_finalize_msg(&nw);
	if (hdr == NULL || !snl_send_message(ctx->io_ss, hdr))
		 err(1, "unable to send netlink message");

	hdr = snl_read_reply(ctx->io_ss, hdr->nlmsg_seq);
	if (hdr->nlmsg_type != NL_RTM_NEWLINK) {
		if (!snl_parse_errmsg(ctx->io_ss, hdr, &errmsg))
			errx(EINVAL, "(NETLINK)");
		if (errmsg.error_str != NULL)
			errx(errmsg.error, "(NETLINK) %s", errmsg.error_str);
	}

	if (!snl_parse_nlmsg(ctx->io_ss, hdr, &gif_parser, &gif_link))
		 return;

	struct nla_gif_info gif_info = gif_link.linkinfo;
	struct nl_parsed_gif gif_data = gif_info.data;

	if (gif_data.ifla_flags == 0)
		 return;

	printf("\toptions=%x", gif_data.ifla_flags);
	print_bits("options", &gif_data.ifla_flags, 1, GIFBITS, nitems(GIFBITS));
	putchar('\n');
}

static void
setgifopts_nl(if_ctx *ctx, const char *val __unused, int d)
{
	struct snl_writer nw = {};
	struct nlmsghdr *hdr;
	struct snl_errmsg_data errmsg = {};
	struct nla_gif_link gif_link = {};
	int off, off2;

	snl_init_writer(ctx->io_ss, &nw);
	hdr = snl_create_msg_request(&nw, NL_RTM_GETLINK);
	hdr->nlmsg_flags |= NLM_F_DUMP;
	snl_reserve_msg_object(&nw, struct ifinfomsg);
	snl_add_msg_attr_string(&nw, IFLA_IFNAME, ctx->ifname);

	hdr = snl_finalize_msg(&nw);
	if (hdr == NULL || (!snl_send_message(ctx->io_ss, hdr)))
		err(1, "unable to send netlink message");

	hdr = snl_read_reply(ctx->io_ss, hdr->nlmsg_seq);
	if (hdr->nlmsg_type != NL_RTM_GETLINK) {
		if (!snl_parse_errmsg(ctx->io_ss, hdr, &errmsg))
			errx(EINVAL, "(NETLINK)");
		if (errmsg.error_str != NULL)
			errx(errmsg.error, "(NETLINK) %s", errmsg.error_str);
	}

	if (!snl_parse_nlmsg(ctx->io_ss, hdr, &gif_parser, &gif_link))
		return;

	struct nla_gif_info gif_info = gif_link.linkinfo;
	struct nl_parsed_gif gif_data = gif_info.data;

	if (d < 0)
		gif_data.ifla_flags &= ~(-d);
	else
		gif_data.ifla_flags |= d;

	gif_nl_init(ctx, &nw, 0);
	off = snl_add_msg_attr_nested(&nw, IFLA_LINKINFO);
	snl_add_msg_attr_string(&nw, IFLA_INFO_KIND, "gif");

	off2 = snl_add_msg_attr_nested(&nw, IFLA_INFO_DATA);
	snl_add_msg_attr_u32(&nw, IFLA_IPTUN_FLAGS, gif_data.ifla_flags);

	snl_end_attr_nested(&nw, off2);
	snl_end_attr_nested(&nw, off);

	gif_nl_fini(ctx, &nw);
}

static struct cmd gif_cmds[] = {
	DEF_CMD("noclamp",		GIF_NOCLAMP,		setgifopts_nl),
	DEF_CMD("-noclamp",		-GIF_NOCLAMP,		setgifopts_nl),
	DEF_CMD("ignore_source",	GIF_IGNORE_SOURCE,	setgifopts_nl),
	DEF_CMD("-ignore_source",	-GIF_IGNORE_SOURCE,	setgifopts_nl),
};

#else
static void
gif_status(if_ctx *ctx)
{
	int opts;
	struct ifreq ifr = { .ifr_data = (caddr_t)&opts };

	if (ioctl_ctx_ifr(ctx, GIFGOPTS, &ifr) == -1)
		return;
	if (opts == 0)
		return;
	printf("\toptions=%x", opts);
	print_bits("options", &opts, 1, GIFBITS, nitems(GIFBITS));
	putchar('\n');
}

static void
setgifopts(if_ctx *ctx, const char *val __unused, int d)
{
	int opts;
	struct ifreq ifr = { .ifr_data = (caddr_t)&opts };

	if (ioctl_ctx_ifr(ctx, GIFGOPTS, &ifr) == -1) {
		warn("ioctl(GIFGOPTS)");
		return;
	}

	if (d < 0)
		opts &= ~(-d);
	else
		opts |= d;

	if (ioctl_ctx(ctx, GIFSOPTS, &ifr) == -1) {
		warn("ioctl(GIFSOPTS)");
		return;
	}
}

static struct cmd gif_cmds[] = {
	DEF_CMD("noclamp",		GIF_NOCLAMP,		setgifopts),
	DEF_CMD("-noclamp",		-GIF_NOCLAMP,		setgifopts),
	DEF_CMD("ignore_source",	GIF_IGNORE_SOURCE,	setgifopts),
	DEF_CMD("-ignore_source",	-GIF_IGNORE_SOURCE,	setgifopts),
};
#endif /* !WITHOUT_NETLINK */

static struct afswtch af_gif = {
	.af_name	= "af_gif",
	.af_af		= AF_UNSPEC,
#ifndef WITHOUT_NETLINK
	.af_other_status = gif_status_nl,
#else
	.af_other_status = gif_status,
#endif
};

static __constructor void
gif_ctor(void)
{
	size_t i;

	for (i = 0; i < nitems(gif_cmds); i++)
		cmd_register(&gif_cmds[i]);
	af_register(&af_gif);
#ifndef WITHOUT_NETLINK
	SNL_VERIFY_PARSERS(all_parsers);
#endif
}