[mirror_iproute2.git] / ip / ipaddrlabel.c

/*
 * ipaddrlabel.c	"ip addrlabel"
 *
 * Copyright (C)2007 USAGI/WIDE Project
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, see <http://www.gnu.org/licenses>.
 *
 *
 * Based on iprule.c.
 *
 * Authors:	YOSHIFUJI Hideaki <yoshfuji@linux-ipv6.org>
 *
 */

#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <fcntl.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <netinet/ip.h>
#include <arpa/inet.h>
#include <string.h>
#include <linux/types.h>
#include <linux/if_addrlabel.h>

#include "rt_names.h"
#include "utils.h"
#include "ip_common.h"
#include "json_print.h"

#define IFAL_RTA(r)	((struct rtattr *)(((char *)(r)) + NLMSG_ALIGN(sizeof(struct ifaddrlblmsg))))
#define IFAL_PAYLOAD(n)	NLMSG_PAYLOAD(n, sizeof(struct ifaddrlblmsg))

extern struct rtnl_handle rth;

static void usage(void) __attribute__((noreturn));

static void usage(void)
{
	fprintf(stderr,
		"Usage: ip addrlabel { add | del } prefix PREFIX [ dev DEV ] [ label LABEL ]\n"
		"       ip addrlabel [ list | flush | help ]\n");
	exit(-1);
}

int print_addrlabel(struct nlmsghdr *n, void *arg)
{
	struct ifaddrlblmsg *ifal = NLMSG_DATA(n);
	int len = n->nlmsg_len;
	struct rtattr *tb[IFAL_MAX+1];

	if (n->nlmsg_type != RTM_NEWADDRLABEL && n->nlmsg_type != RTM_DELADDRLABEL)
		return 0;

	len -= NLMSG_LENGTH(sizeof(*ifal));
	if (len < 0)
		return -1;

	parse_rtattr(tb, IFAL_MAX, IFAL_RTA(ifal), len);

	open_json_object(NULL);
	if (n->nlmsg_type == RTM_DELADDRLABEL)
		print_bool(PRINT_ANY, "deleted", "Deleted ", true);

	if (tb[IFAL_ADDRESS]) {
		const char *host
			= format_host_rta(ifal->ifal_family,
					  tb[IFAL_ADDRESS]);

		print_string(PRINT_FP, NULL, "prefix ", NULL);
		print_color_string(PRINT_ANY,
				   ifa_family_color(ifal->ifal_family),
				   "address", "%s", host);

		print_uint(PRINT_ANY, "prefixlen", "/%u ",
			   ifal->ifal_prefixlen);
	}

	if (ifal->ifal_index) {
		print_string(PRINT_FP, NULL, "dev ", NULL);
		print_color_string(PRINT_ANY, COLOR_IFNAME,
				   "ifname", "%s ",
				   ll_index_to_name(ifal->ifal_index));
	}

	if (tb[IFAL_LABEL] && RTA_PAYLOAD(tb[IFAL_LABEL]) == sizeof(uint32_t)) {
		uint32_t label = rta_getattr_u32(tb[IFAL_LABEL]);

		print_uint(PRINT_ANY,
			   "label", "label %u ", label);
	}
	print_string(PRINT_FP, NULL, "\n", "");
	close_json_object();

	return 0;
}

static int ipaddrlabel_list(int argc, char **argv)
{
	int af = preferred_family;

	if (af == AF_UNSPEC)
		af = AF_INET6;

	if (argc > 0) {
		fprintf(stderr, "\"ip addrlabel show\" does not take any arguments.\n");
		return -1;
	}

	if (rtnl_addrlbldump_req(&rth, af) < 0) {
		perror("Cannot send dump request");
		return 1;
	}

	new_json_obj(json);
	if (rtnl_dump_filter(&rth, print_addrlabel, stdout) < 0) {
		fprintf(stderr, "Dump terminated\n");
		return 1;
	}
	delete_json_obj();

	return 0;
}


static int ipaddrlabel_modify(int cmd, int argc, char **argv)
{
	struct {
		struct nlmsghdr	n;
		struct ifaddrlblmsg	ifal;
		char			buf[1024];
	} req = {
		.n.nlmsg_type = cmd,
		.n.nlmsg_len = NLMSG_LENGTH(sizeof(struct ifaddrlblmsg)),
		.n.nlmsg_flags = NLM_F_REQUEST,
		.ifal.ifal_family = preferred_family,
	};

	inet_prefix prefix = {};
	uint32_t label = 0xffffffffUL;
	char *p = NULL;
	char *l = NULL;

	if (cmd == RTM_NEWADDRLABEL) {
		req.n.nlmsg_flags |= NLM_F_CREATE|NLM_F_EXCL;
	}

	while (argc > 0) {
		if (strcmp(*argv, "prefix") == 0) {
			NEXT_ARG();
			p = *argv;
			get_prefix(&prefix, *argv, preferred_family);
		} else if (strcmp(*argv, "dev") == 0) {
			NEXT_ARG();
			if ((req.ifal.ifal_index = ll_name_to_index(*argv)) == 0)
				invarg("dev is invalid\n", *argv);
		} else if (strcmp(*argv, "label") == 0) {
			NEXT_ARG();
			l = *argv;
			if (get_u32(&label, *argv, 0) || label == 0xffffffffUL)
				invarg("label is invalid\n", *argv);
		}
		argc--;
		argv++;
	}
	if (p == NULL) {
		fprintf(stderr, "Not enough information: \"prefix\" argument is required.\n");
		return -1;
	}
	if (l == NULL) {
		fprintf(stderr, "Not enough information: \"label\" argument is required.\n");
		return -1;
	}
	addattr32(&req.n, sizeof(req), IFAL_LABEL, label);
	addattr_l(&req.n, sizeof(req), IFAL_ADDRESS, &prefix.data, prefix.bytelen);
	req.ifal.ifal_prefixlen = prefix.bitlen;

	if (req.ifal.ifal_family == AF_UNSPEC)
		req.ifal.ifal_family = AF_INET6;

	if (rtnl_talk(&rth, &req.n, NULL) < 0)
		return -2;

	return 0;
}


static int flush_addrlabel(struct nlmsghdr *n, void *arg)
{
	struct rtnl_handle rth2;
	struct rtmsg *r = NLMSG_DATA(n);
	int len = n->nlmsg_len;
	struct rtattr *tb[IFAL_MAX+1];

	len -= NLMSG_LENGTH(sizeof(*r));
	if (len < 0)
		return -1;

	parse_rtattr(tb, IFAL_MAX, RTM_RTA(r), len);

	if (tb[IFAL_ADDRESS]) {
		n->nlmsg_type = RTM_DELADDRLABEL;
		n->nlmsg_flags = NLM_F_REQUEST;

		if (rtnl_open(&rth2, 0) < 0)
			return -1;

		if (rtnl_talk(&rth2, n, NULL) < 0)
			return -2;

		rtnl_close(&rth2);
	}

	return 0;
}

static int ipaddrlabel_flush(int argc, char **argv)
{
	int af = preferred_family;

	if (af == AF_UNSPEC)
		af = AF_INET6;

	if (argc > 0) {
		fprintf(stderr, "\"ip addrlabel flush\" does not allow extra arguments\n");
		return -1;
	}

	if (rtnl_addrlbldump_req(&rth, af) < 0) {
		perror("Cannot send dump request");
		return -1;
	}

	if (rtnl_dump_filter(&rth, flush_addrlabel, NULL) < 0) {
		fprintf(stderr, "Flush terminated\n");
		return -1;
	}

	return 0;
}

int do_ipaddrlabel(int argc, char **argv)
{
	if (argc < 1) {
		return ipaddrlabel_list(0, NULL);
	} else if (matches(argv[0], "list") == 0 ||
		   matches(argv[0], "lst") == 0 ||
		   matches(argv[0], "show") == 0) {
		return ipaddrlabel_list(argc-1, argv+1);
	} else if (matches(argv[0], "add") == 0) {
		return ipaddrlabel_modify(RTM_NEWADDRLABEL, argc-1, argv+1);
	} else if (matches(argv[0], "delete") == 0) {
		return ipaddrlabel_modify(RTM_DELADDRLABEL, argc-1, argv+1);
	} else if (matches(argv[0], "flush") == 0) {
		return ipaddrlabel_flush(argc-1, argv+1);
	} else if (matches(argv[0], "help") == 0)
		usage();

	fprintf(stderr, "Command \"%s\" is unknown, try \"ip addrlabel help\".\n", *argv);
	exit(-1);
}
Commit	Line	Data
4759758c YH	1	/*
	2	* ipaddrlabel.c "ip addrlabel"
	3	*
	4	* Copyright (C)2007 USAGI/WIDE Project
	5	*
	6	* This program is free software; you can redistribute it and/or modify
	7	* it under the terms of the GNU General Public License as published by
	8	* the Free Software Foundation; either version 2 of the License, or
	9	* (at your option) any later version.
	10	*
	11	* This program is distributed in the hope that it will be useful,
	12	* but WITHOUT ANY WARRANTY; without even the implied warranty of
	13	* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
	14	* GNU General Public License for more details.
	15	*
	16	* You should have received a copy of the GNU General Public License
4d98ab00	17	* along with this program; if not, see <http://www.gnu.org/licenses>.
4759758c YH	18	*
	19	*
	20	* Based on iprule.c.
	21	*
	22	* Authors: YOSHIFUJI Hideaki <yoshfuji@linux-ipv6.org>
	23	*
	24	*/
	25
	26	#include <stdio.h>
	27	#include <stdlib.h>
	28	#include <unistd.h>
4759758c YH	29	#include <fcntl.h>
	30	#include <sys/socket.h>
	31	#include <netinet/in.h>
	32	#include <netinet/ip.h>
	33	#include <arpa/inet.h>
	34	#include <string.h>
	35	#include <linux/types.h>
	36	#include <linux/if_addrlabel.h>
	37
	38	#include "rt_names.h"
	39	#include "utils.h"
	40	#include "ip_common.h"
a7ad1c8a	41	#include "json_print.h"
4759758c	42
56f5daac SH	43	#define IFAL_RTA(r) ((struct rtattr )(((char )(r)) + NLMSG_ALIGN(sizeof(struct ifaddrlblmsg))))
56f5daac SH	44	#define IFAL_PAYLOAD(n) NLMSG_PAYLOAD(n, sizeof(struct ifaddrlblmsg))
4759758c YH	45
	46	extern struct rtnl_handle rth;
	47
	48	static void usage(void) __attribute__((noreturn));
	49
	50	static void usage(void)
	51	{
8589eb4e MC	52	fprintf(stderr,
	53	"Usage: ip addrlabel { add \| del } prefix PREFIX [ dev DEV ] [ label LABEL ]\n"
	54	" ip addrlabel [ list \| flush \| help ]\n");
4759758c YH	55	exit(-1);
	56	}
	57
cd554f2c	58	int print_addrlabel(struct nlmsghdr n, void arg)
4759758c	59	{
4759758c YH	60	struct ifaddrlblmsg *ifal = NLMSG_DATA(n);
4759758c YH	61	int len = n->nlmsg_len;
4759758c	62	struct rtattr *tb[IFAL_MAX+1];
4759758c YH	63
	64	if (n->nlmsg_type != RTM_NEWADDRLABEL && n->nlmsg_type != RTM_DELADDRLABEL)
	65	return 0;
	66
	67	len -= NLMSG_LENGTH(sizeof(*ifal));
	68	if (len < 0)
	69	return -1;
	70
	71	parse_rtattr(tb, IFAL_MAX, IFAL_RTA(ifal), len);
	72
a7ad1c8a	73	open_json_object(NULL);
4759758c	74	if (n->nlmsg_type == RTM_DELADDRLABEL)
a7ad1c8a	75	print_bool(PRINT_ANY, "deleted", "Deleted ", true);
4759758c YH	76
4759758c YH	77	if (tb[IFAL_ADDRESS]) {
a7ad1c8a SH	78	const char *host
	79	= format_host_rta(ifal->ifal_family,
	80	tb[IFAL_ADDRESS]);
	81
	82	print_string(PRINT_FP, NULL, "prefix ", NULL);
	83	print_color_string(PRINT_ANY,
	84	ifa_family_color(ifal->ifal_family),
	85	"address", "%s", host);
	86
	87	print_uint(PRINT_ANY, "prefixlen", "/%u ",
	88	ifal->ifal_prefixlen);
4759758c YH	89	}
4759758c YH	90
a7ad1c8a SH	91	if (ifal->ifal_index) {
	92	print_string(PRINT_FP, NULL, "dev ", NULL);
	93	print_color_string(PRINT_ANY, COLOR_IFNAME,
	94	"ifname", "%s ",
	95	ll_index_to_name(ifal->ifal_index));
	96	}
4759758c	97
97870334	98	if (tb[IFAL_LABEL] && RTA_PAYLOAD(tb[IFAL_LABEL]) == sizeof(uint32_t)) {
0b9b0d08	99	uint32_t label = rta_getattr_u32(tb[IFAL_LABEL]);
56f5daac	100
a7ad1c8a SH	101	print_uint(PRINT_ANY,
a7ad1c8a SH	102	"label", "label %u ", label);
4759758c	103	}
a7ad1c8a SH	104	print_string(PRINT_FP, NULL, "\n", "");
a7ad1c8a SH	105	close_json_object();
4759758c	106
4759758c YH	107	return 0;
	108	}
	109
	110	static int ipaddrlabel_list(int argc, char **argv)
	111	{
	112	int af = preferred_family;
	113
	114	if (af == AF_UNSPEC)
	115	af = AF_INET6;
	116
	117	if (argc > 0) {
	118	fprintf(stderr, "\"ip addrlabel show\" does not take any arguments.\n");
	119	return -1;
	120	}
	121
39360023	122	if (rtnl_addrlbldump_req(&rth, af) < 0) {
4759758c YH	123	perror("Cannot send dump request");
	124	return 1;
	125	}
	126
a7ad1c8a	127	new_json_obj(json);
cd70f3f5	128	if (rtnl_dump_filter(&rth, print_addrlabel, stdout) < 0) {
4759758c YH	129	fprintf(stderr, "Dump terminated\n");
	130	return 1;
	131	}
a7ad1c8a	132	delete_json_obj();
4759758c YH	133
	134	return 0;
	135	}
	136
	137
	138	static int ipaddrlabel_modify(int cmd, int argc, char **argv)
	139	{
	140	struct {
4806867a	141	struct nlmsghdr n;
4759758c	142	struct ifaddrlblmsg ifal;
56f5daac	143	char buf[1024];
d17b136f PS	144	} req = {
	145	.n.nlmsg_type = cmd,
	146	.n.nlmsg_len = NLMSG_LENGTH(sizeof(struct ifaddrlblmsg)),
	147	.n.nlmsg_flags = NLM_F_REQUEST,
	148	.ifal.ifal_family = preferred_family,
	149	};
	150
	151	inet_prefix prefix = {};
4759758c	152	uint32_t label = 0xffffffffUL;
6579feea	153	char *p = NULL;
0612519e	154	char *l = NULL;
4759758c	155
4759758c YH	156	if (cmd == RTM_NEWADDRLABEL) {
	157	req.n.nlmsg_flags \|= NLM_F_CREATE\|NLM_F_EXCL;
	158	}
	159
	160	while (argc > 0) {
	161	if (strcmp(*argv, "prefix") == 0) {
	162	NEXT_ARG();
6579feea	163	p = *argv;
4759758c YH	164	get_prefix(&prefix, *argv, preferred_family);
	165	} else if (strcmp(*argv, "dev") == 0) {
	166	NEXT_ARG();
	167	if ((req.ifal.ifal_index = ll_name_to_index(*argv)) == 0)
	168	invarg("dev is invalid\n", *argv);
	169	} else if (strcmp(*argv, "label") == 0) {
	170	NEXT_ARG();
6579feea	171	l = *argv;
4759758c YH	172	if (get_u32(&label, *argv, 0) \|\| label == 0xffffffffUL)
	173	invarg("label is invalid\n", *argv);
	174	}
	175	argc--;
	176	argv++;
	177	}
6579feea VC	178	if (p == NULL) {
	179	fprintf(stderr, "Not enough information: \"prefix\" argument is required.\n");
	180	return -1;
	181	}
	182	if (l == NULL) {
	183	fprintf(stderr, "Not enough information: \"label\" argument is required.\n");
	184	return -1;
	185	}
4759758c YH	186	addattr32(&req.n, sizeof(req), IFAL_LABEL, label);
4759758c YH	187	addattr_l(&req.n, sizeof(req), IFAL_ADDRESS, &prefix.data, prefix.bytelen);
3490740b	188	req.ifal.ifal_prefixlen = prefix.bitlen;
4759758c YH	189
	190	if (req.ifal.ifal_family == AF_UNSPEC)
	191	req.ifal.ifal_family = AF_INET6;
	192
86bf43c7	193	if (rtnl_talk(&rth, &req.n, NULL) < 0)
906cafe3	194	return -2;
4759758c YH	195
	196	return 0;
	197	}
	198
	199
cd554f2c	200	static int flush_addrlabel(struct nlmsghdr n, void arg)
4759758c YH	201	{
	202	struct rtnl_handle rth2;
	203	struct rtmsg *r = NLMSG_DATA(n);
	204	int len = n->nlmsg_len;
56f5daac	205	struct rtattr *tb[IFAL_MAX+1];
4759758c YH	206
	207	len -= NLMSG_LENGTH(sizeof(*r));
	208	if (len < 0)
	209	return -1;
	210
	211	parse_rtattr(tb, IFAL_MAX, RTM_RTA(r), len);
	212
	213	if (tb[IFAL_ADDRESS]) {
	214	n->nlmsg_type = RTM_DELADDRLABEL;
	215	n->nlmsg_flags = NLM_F_REQUEST;
	216
	217	if (rtnl_open(&rth2, 0) < 0)
	218	return -1;
	219
86bf43c7	220	if (rtnl_talk(&rth2, n, NULL) < 0)
4759758c YH	221	return -2;
	222
	223	rtnl_close(&rth2);
	224	}
	225
	226	return 0;
	227	}
	228
	229	static int ipaddrlabel_flush(int argc, char **argv)
	230	{
	231	int af = preferred_family;
	232
	233	if (af == AF_UNSPEC)
	234	af = AF_INET6;
	235
	236	if (argc > 0) {
	237	fprintf(stderr, "\"ip addrlabel flush\" does not allow extra arguments\n");
	238	return -1;
	239	}
	240
39360023	241	if (rtnl_addrlbldump_req(&rth, af) < 0) {
4759758c	242	perror("Cannot send dump request");
906cafe3	243	return -1;
4759758c YH	244	}
4759758c YH	245
cd70f3f5	246	if (rtnl_dump_filter(&rth, flush_addrlabel, NULL) < 0) {
4759758c	247	fprintf(stderr, "Flush terminated\n");
906cafe3	248	return -1;
4759758c YH	249	}
	250
	251	return 0;
	252	}
	253
	254	int do_ipaddrlabel(int argc, char **argv)
	255	{
	256	if (argc < 1) {
	257	return ipaddrlabel_list(0, NULL);
	258	} else if (matches(argv[0], "list") == 0 \|\|
473544d9	259	matches(argv[0], "lst") == 0 \|\|
4759758c YH	260	matches(argv[0], "show") == 0) {
	261	return ipaddrlabel_list(argc-1, argv+1);
	262	} else if (matches(argv[0], "add") == 0) {
	263	return ipaddrlabel_modify(RTM_NEWADDRLABEL, argc-1, argv+1);
	264	} else if (matches(argv[0], "delete") == 0) {
	265	return ipaddrlabel_modify(RTM_DELADDRLABEL, argc-1, argv+1);
	266	} else if (matches(argv[0], "flush") == 0) {
	267	return ipaddrlabel_flush(argc-1, argv+1);
	268	} else if (matches(argv[0], "help") == 0)
	269	usage();
	270
	271	fprintf(stderr, "Command \"%s\" is unknown, try \"ip addrlabel help\".\n", *argv);
	272	exit(-1);
	273	}