[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, write to the Free Software
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 *
 *
 * Based on iprule.c.
 *
 * Authors:	YOSHIFUJI Hideaki <yoshfuji@linux-ipv6.org>
 *
 */

#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <syslog.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"

#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 [ list | add | del | flush ] prefix PREFIX [ dev DEV ] [ label LABEL ]\n");
	exit(-1);
}

int print_addrlabel(const struct sockaddr_nl *who, struct nlmsghdr *n, void *arg)
{
	FILE *fp = (FILE*)arg;
	struct ifaddrlblmsg *ifal = NLMSG_DATA(n);
	int len = n->nlmsg_len;
	struct rtattr *tb[IFAL_MAX+1];
	char abuf[256];

	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);

	if (n->nlmsg_type == RTM_DELADDRLABEL)
		fprintf(fp, "Deleted ");

	if (tb[IFAL_ADDRESS]) {
		fprintf(fp, "prefix %s/%u ",
			format_host(ifal->ifal_family,
				    RTA_PAYLOAD(tb[IFAL_ADDRESS]),
				    RTA_DATA(tb[IFAL_ADDRESS]),
				    abuf, sizeof(abuf)),
			ifal->ifal_prefixlen);
	}

	if (ifal->ifal_index)
		fprintf(fp, "dev %s ", ll_index_to_name(ifal->ifal_index));

	if (tb[IFAL_LABEL] && RTA_PAYLOAD(tb[IFAL_LABEL]) == sizeof(uint32_t)) {
		uint32_t label;
		memcpy(&label, RTA_DATA(tb[IFAL_LABEL]), sizeof(label));
		fprintf(fp, "label %u ", label);
	}

	fprintf(fp, "\n");
	fflush(fp);
	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_wilddump_request(&rth, af, RTM_GETADDRLABEL) < 0) {
		perror("Cannot send dump request");
		return 1;
	}

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

	return 0;
}


static int ipaddrlabel_modify(int cmd, int argc, char **argv)
{
	struct {
		struct nlmsghdr 	n;
		struct ifaddrlblmsg	ifal;
		char   			buf[1024];
	} req;

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

	memset(&req, 0, sizeof(req));
	memset(&prefix, 0, sizeof(prefix));

	req.n.nlmsg_type = cmd;
	req.n.nlmsg_len = NLMSG_LENGTH(sizeof(struct ifaddrlblmsg));
	req.n.nlmsg_flags = NLM_F_REQUEST;
	req.ifal.ifal_family = preferred_family;
	req.ifal.ifal_prefixlen = 0;
	req.ifal.ifal_index = 0;

	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, 0, 0, NULL) < 0)
		return 2;

	return 0;
}


static int flush_addrlabel(const struct sockaddr_nl *who, 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, 0, 0, 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_wilddump_request(&rth, af, RTM_GETADDRLABEL) < 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], "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
	17	* along with this program; if not, write to the Free Software
	18	* Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
	19	*
	20	*
	21	* Based on iprule.c.
	22	*
	23	* Authors: YOSHIFUJI Hideaki <yoshfuji@linux-ipv6.org>
	24	*
	25	*/
	26
	27	#include <stdio.h>
	28	#include <stdlib.h>
	29	#include <unistd.h>
	30	#include <syslog.h>
	31	#include <fcntl.h>
	32	#include <sys/socket.h>
	33	#include <netinet/in.h>
	34	#include <netinet/ip.h>
	35	#include <arpa/inet.h>
	36	#include <string.h>
	37	#include <linux/types.h>
	38	#include <linux/if_addrlabel.h>
	39
	40	#include "rt_names.h"
	41	#include "utils.h"
	42	#include "ip_common.h"
	43
	44	#define IFAL_RTA(r) ((struct rtattr)(((char)(r)) + NLMSG_ALIGN(sizeof(struct ifaddrlblmsg))))
	45	#define IFAL_PAYLOAD(n) NLMSG_PAYLOAD(n,sizeof(struct ifaddrlblmsg))
	46
	47	extern struct rtnl_handle rth;
	48
	49	static void usage(void) __attribute__((noreturn));
	50
	51	static void usage(void)
	52	{
	53	fprintf(stderr, "Usage: ip addrlabel [ list \| add \| del \| flush ] prefix PREFIX [ dev DEV ] [ label LABEL ]\n");
	54	exit(-1);
	55	}
	56
	57	int print_addrlabel(const struct sockaddr_nl who, struct nlmsghdr n, void *arg)
	58	{
	59	FILE fp = (FILE)arg;
	60	struct ifaddrlblmsg *ifal = NLMSG_DATA(n);
	61	int len = n->nlmsg_len;
4759758c YH	62	struct rtattr *tb[IFAL_MAX+1];
	63	char abuf[256];
	64
	65	if (n->nlmsg_type != RTM_NEWADDRLABEL && n->nlmsg_type != RTM_DELADDRLABEL)
	66	return 0;
	67
	68	len -= NLMSG_LENGTH(sizeof(*ifal));
	69	if (len < 0)
	70	return -1;
	71
	72	parse_rtattr(tb, IFAL_MAX, IFAL_RTA(ifal), len);
	73
4759758c YH	74	if (n->nlmsg_type == RTM_DELADDRLABEL)
	75	fprintf(fp, "Deleted ");
	76
	77	if (tb[IFAL_ADDRESS]) {
	78	fprintf(fp, "prefix %s/%u ",
	79	format_host(ifal->ifal_family,
	80	RTA_PAYLOAD(tb[IFAL_ADDRESS]),
	81	RTA_DATA(tb[IFAL_ADDRESS]),
	82	abuf, sizeof(abuf)),
	83	ifal->ifal_prefixlen);
	84	}
	85
	86	if (ifal->ifal_index)
	87	fprintf(fp, "dev %s ", ll_index_to_name(ifal->ifal_index));
	88
97870334 HL	89	if (tb[IFAL_LABEL] && RTA_PAYLOAD(tb[IFAL_LABEL]) == sizeof(uint32_t)) {
97870334 HL	90	uint32_t label;
4759758c	91	memcpy(&label, RTA_DATA(tb[IFAL_LABEL]), sizeof(label));
97870334	92	fprintf(fp, "label %u ", label);
4759758c YH	93	}
	94
	95	fprintf(fp, "\n");
	96	fflush(fp);
	97	return 0;
	98	}
	99
	100	static int ipaddrlabel_list(int argc, char **argv)
	101	{
	102	int af = preferred_family;
	103
	104	if (af == AF_UNSPEC)
	105	af = AF_INET6;
	106
	107	if (argc > 0) {
	108	fprintf(stderr, "\"ip addrlabel show\" does not take any arguments.\n");
	109	return -1;
	110	}
	111
	112	if (rtnl_wilddump_request(&rth, af, RTM_GETADDRLABEL) < 0) {
	113	perror("Cannot send dump request");
	114	return 1;
	115	}
	116
cd70f3f5	117	if (rtnl_dump_filter(&rth, print_addrlabel, stdout) < 0) {
4759758c YH	118	fprintf(stderr, "Dump terminated\n");
	119	return 1;
	120	}
	121
	122	return 0;
	123	}
	124
	125
	126	static int ipaddrlabel_modify(int cmd, int argc, char **argv)
	127	{
	128	struct {
	129	struct nlmsghdr n;
	130	struct ifaddrlblmsg ifal;
	131	char buf[1024];
	132	} req;
	133
	134	inet_prefix prefix;
	135	uint32_t label = 0xffffffffUL;
6579feea VC	136	char *p = NULL;
6579feea VC	137	char *l = NULL;
4759758c YH	138
	139	memset(&req, 0, sizeof(req));
	140	memset(&prefix, 0, sizeof(prefix));
	141
	142	req.n.nlmsg_type = cmd;
	143	req.n.nlmsg_len = NLMSG_LENGTH(sizeof(struct ifaddrlblmsg));
	144	req.n.nlmsg_flags = NLM_F_REQUEST;
	145	req.ifal.ifal_family = preferred_family;
	146	req.ifal.ifal_prefixlen = 0;
	147	req.ifal.ifal_index = 0;
	148
	149	if (cmd == RTM_NEWADDRLABEL) {
	150	req.n.nlmsg_flags \|= NLM_F_CREATE\|NLM_F_EXCL;
	151	}
	152
	153	while (argc > 0) {
	154	if (strcmp(*argv, "prefix") == 0) {
	155	NEXT_ARG();
6579feea	156	p = *argv;
4759758c YH	157	get_prefix(&prefix, *argv, preferred_family);
	158	} else if (strcmp(*argv, "dev") == 0) {
	159	NEXT_ARG();
	160	if ((req.ifal.ifal_index = ll_name_to_index(*argv)) == 0)
	161	invarg("dev is invalid\n", *argv);
	162	} else if (strcmp(*argv, "label") == 0) {
	163	NEXT_ARG();
6579feea	164	l = *argv;
4759758c YH	165	if (get_u32(&label, *argv, 0) \|\| label == 0xffffffffUL)
	166	invarg("label is invalid\n", *argv);
	167	}
	168	argc--;
	169	argv++;
	170	}
6579feea VC	171	if (p == NULL) {
	172	fprintf(stderr, "Not enough information: \"prefix\" argument is required.\n");
	173	return -1;
	174	}
	175	if (l == NULL) {
	176	fprintf(stderr, "Not enough information: \"label\" argument is required.\n");
	177	return -1;
	178	}
4759758c YH	179	addattr32(&req.n, sizeof(req), IFAL_LABEL, label);
4759758c YH	180	addattr_l(&req.n, sizeof(req), IFAL_ADDRESS, &prefix.data, prefix.bytelen);
3490740b	181	req.ifal.ifal_prefixlen = prefix.bitlen;
4759758c YH	182
	183	if (req.ifal.ifal_family == AF_UNSPEC)
	184	req.ifal.ifal_family = AF_INET6;
	185
cd70f3f5	186	if (rtnl_talk(&rth, &req.n, 0, 0, NULL) < 0)
4759758c YH	187	return 2;
	188
	189	return 0;
	190	}
	191
	192
	193	static int flush_addrlabel(const struct sockaddr_nl who, struct nlmsghdr n, void *arg)
	194	{
	195	struct rtnl_handle rth2;
	196	struct rtmsg *r = NLMSG_DATA(n);
	197	int len = n->nlmsg_len;
	198	struct rtattr * tb[IFAL_MAX+1];
	199
	200	len -= NLMSG_LENGTH(sizeof(*r));
	201	if (len < 0)
	202	return -1;
	203
	204	parse_rtattr(tb, IFAL_MAX, RTM_RTA(r), len);
	205
	206	if (tb[IFAL_ADDRESS]) {
	207	n->nlmsg_type = RTM_DELADDRLABEL;
	208	n->nlmsg_flags = NLM_F_REQUEST;
	209
	210	if (rtnl_open(&rth2, 0) < 0)
	211	return -1;
	212
cd70f3f5	213	if (rtnl_talk(&rth2, n, 0, 0, NULL) < 0)
4759758c YH	214	return -2;
	215
	216	rtnl_close(&rth2);
	217	}
	218
	219	return 0;
	220	}
	221
	222	static int ipaddrlabel_flush(int argc, char **argv)
	223	{
	224	int af = preferred_family;
	225
	226	if (af == AF_UNSPEC)
	227	af = AF_INET6;
	228
	229	if (argc > 0) {
	230	fprintf(stderr, "\"ip addrlabel flush\" does not allow extra arguments\n");
	231	return -1;
	232	}
	233
	234	if (rtnl_wilddump_request(&rth, af, RTM_GETADDRLABEL) < 0) {
	235	perror("Cannot send dump request");
	236	return 1;
	237	}
	238
cd70f3f5	239	if (rtnl_dump_filter(&rth, flush_addrlabel, NULL) < 0) {
4759758c YH	240	fprintf(stderr, "Flush terminated\n");
	241	return 1;
	242	}
	243
	244	return 0;
	245	}
	246
	247	int do_ipaddrlabel(int argc, char **argv)
	248	{
	249	if (argc < 1) {
	250	return ipaddrlabel_list(0, NULL);
	251	} else if (matches(argv[0], "list") == 0 \|\|
	252	matches(argv[0], "show") == 0) {
	253	return ipaddrlabel_list(argc-1, argv+1);
	254	} else if (matches(argv[0], "add") == 0) {
	255	return ipaddrlabel_modify(RTM_NEWADDRLABEL, argc-1, argv+1);
	256	} else if (matches(argv[0], "delete") == 0) {
	257	return ipaddrlabel_modify(RTM_DELADDRLABEL, argc-1, argv+1);
	258	} else if (matches(argv[0], "flush") == 0) {
	259	return ipaddrlabel_flush(argc-1, argv+1);
	260	} else if (matches(argv[0], "help") == 0)
	261	usage();
	262
	263	fprintf(stderr, "Command \"%s\" is unknown, try \"ip addrlabel help\".\n", *argv);
	264	exit(-1);
	265	}
	266