273 lines
6.6 KiB
C
Executable File
273 lines
6.6 KiB
C
Executable File
/*
|
|
* 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;
|
|
int host_len = -1;
|
|
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 (ifal->ifal_family == AF_INET)
|
|
host_len = 32;
|
|
else if (ifal->ifal_family == AF_INET6)
|
|
host_len = 128;
|
|
|
|
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(int32_t)) {
|
|
int32_t label;
|
|
memcpy(&label, RTA_DATA(tb[IFAL_LABEL]), sizeof(label));
|
|
fprintf(fp, "label %d ", 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, NULL, NULL) < 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, NULL, 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, NULL, 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, NULL, 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);
|
|
}
|
|
|