shaba/systemd-stable
1
1
Fork 0
mirror of https://github.com/systemd/systemd-stable.git synced 2024-12-24 21:34:08 +03:00
Code

rtnl: add support for routes

Browse Source
This commit is contained in:
Tom Gundersen 2013-11-04 02:42:23 +01:00
parent be32eb9b7f
commit 03d7e632c0
3 changed files with 133 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

89
src/libsystemd-rtnl/rtnl-message.c
View File

@ -67,6 +67,42 @@ static int message_new(sd_rtnl_message **ret, size_t initial_size) {
return 0;
}
int sd_rtnl_message_route_new(uint16_t nlmsg_type, unsigned char rtm_family,
unsigned char rtm_dst_len, unsigned char rtm_src_len,
unsigned char rtm_tos, unsigned char rtm_table,
unsigned char rtm_scope, unsigned char rtm_protocol,
unsigned char rtm_type, unsigned rtm_flags, sd_rtnl_message **ret) {
struct rtmsg *rtm;
int r;
assert_return(nlmsg_type == RTM_NEWROUTE || nlmsg_type == RTM_DELROUTE ||
nlmsg_type == RTM_GETROUTE, -EINVAL);
assert_return(ret, -EINVAL);
r = message_new(ret, NLMSG_SPACE(sizeof(struct rtmsg)));
if (r < 0)
return r;
(*ret)->hdr->nlmsg_len = NLMSG_LENGTH(sizeof(struct rtmsg));
(*ret)->hdr->nlmsg_type = nlmsg_type;
if (nlmsg_type == RTM_NEWROUTE)
(*ret)->hdr->nlmsg_flags |= NLM_F_CREATE | NLM_F_EXCL;
rtm = NLMSG_DATA((*ret)->hdr);
rtm->rtm_family = rtm_family;
rtm->rtm_dst_len = rtm_dst_len;
rtm->rtm_src_len = rtm_src_len;
rtm->rtm_tos = rtm_tos;
rtm->rtm_table = rtm_table;
rtm->rtm_protocol = rtm_protocol;
rtm->rtm_scope = rtm_scope;
rtm->rtm_type = rtm_type;
rtm->rtm_flags = rtm_flags;
return 0;
}
int sd_rtnl_message_link_new(uint16_t nlmsg_type, int index, unsigned int type, unsigned int flags, sd_rtnl_message **ret) {
struct ifinfomsg *ifi;
int r;
@ -189,6 +225,7 @@ static int add_rtattr(sd_rtnl_message *m, unsigned short type, const void *data,
int sd_rtnl_message_append(sd_rtnl_message *m, unsigned short type, const void *data) {
uint16_t rtm_type;
struct ifaddrmsg *ifa;
struct rtmsg *rtm;
assert_return(m, -EINVAL);
assert_return(data, -EINVAL);
@ -204,9 +241,9 @@ int sd_rtnl_message_append(sd_rtnl_message *m, unsigned short type, const void *
case IFLA_QDISC:
return add_rtattr(m, type, data, strlen(data) + 1);
case IFLA_MTU:
return add_rtattr(m, type, data, sizeof(unsigned int));
return add_rtattr(m, type, data, sizeof(uint32_t));
case IFLA_LINK:
return add_rtattr(m, type, data, sizeof(int));
return add_rtattr(m, type, data, sizeof(uint32_t));
case IFLA_STATS:
return add_rtattr(m, type, data, sizeof(struct rtnl_link_stats));
case IFLA_ADDRESS:
@ -237,6 +274,30 @@ int sd_rtnl_message_append(sd_rtnl_message *m, unsigned short type, const void *
default:
return -ENOTSUP;
}
case RTM_NEWROUTE:
case RTM_DELROUTE:
case RTM_GETROUTE:
switch (type) {
case RTA_DST:
case RTA_SRC:
case RTA_GATEWAY:
rtm = NLMSG_DATA(m->hdr);
switch (rtm->rtm_family) {
case AF_INET:
return add_rtattr(m, type, data, sizeof(struct in_addr));
case AF_INET6:
return add_rtattr(m, type, data, sizeof(struct in6_addr));
default:
return -EINVAL;
}
case RTA_TABLE:
case RTA_PRIORITY:
case RTA_IIF:
case RTA_OIF:
return add_rtattr(m, type, data, sizeof(uint32_t));
default:
return -ENOTSUP;
}
default:
return -ENOTSUP;
}
@ -275,7 +336,7 @@ int sd_rtnl_message_read(sd_rtnl_message *m, unsigned short *type, void **data)
m->next_rta = IFLA_RTA(ifi);
m->remaining_size = IFLA_PAYLOAD(m->hdr);
}
break;;
break;
case RTM_NEWADDR:
case RTM_DELADDR:
case RTM_GETADDR:
@ -283,9 +344,19 @@ int sd_rtnl_message_read(sd_rtnl_message *m, unsigned short *type, void **data)
struct ifaddrmsg *ifa = NLMSG_DATA(m->hdr);
m->next_rta = IFA_RTA(ifa);
m->remaining_size = IFLA_PAYLOAD(m->hdr);
m->remaining_size = IFA_PAYLOAD(m->hdr);
}
break;;
break;
case RTM_NEWROUTE:
case RTM_DELROUTE:
case RTM_GETROUTE:
if (!m->next_rta) {
struct rtmesg *rtm = NLMSG_DATA(m->hdr);
m->next_rta = RTM_RTA(rtm);
m->remaining_size = RTM_PAYLOAD(m->hdr);
}
break;
default:
return -ENOTSUP;
}
@ -415,22 +486,22 @@ int socket_read_message(sd_rtnl *nl, sd_rtnl_message **ret) {
case NLMSG_ERROR:
if (m->hdr->nlmsg_len < NLMSG_LENGTH(sizeof(struct nlmsgerr)))
k = -EIO;
break;;
break;
case RTM_NEWLINK:
case RTM_DELLINK:
case RTM_GETLINK:
if (m->hdr->nlmsg_len < NLMSG_LENGTH(sizeof(struct ifinfomsg)))
k = -EIO;
break;;
break;
case RTM_NEWADDR:
case RTM_DELADDR:
case RTM_GETADDR:
if (m->hdr->nlmsg_len < NLMSG_LENGTH(sizeof(struct ifaddrmsg)))
k = -EIO;
break;;
break;
case NLMSG_NOOP:
k = 0;
break;;
break;
default:
k = 0; /* ignoring message of unknown type */
}

47
src/libsystemd-rtnl/test-rtnl.c
View File

@ -54,6 +54,43 @@ static void test_link_configure(sd_rtnl *rtnl, int ifindex) {
assert(sd_rtnl_send_with_reply_and_block(rtnl, message, 0, NULL) == 0);
}
static void test_route(void) {
_cleanup_sd_rtnl_message_unref_ sd_rtnl_message *req;
uint32_t addr = htonl(INADDR_LOOPBACK);
uint32_t index = 2;
uint16_t type;
void *data;
int r;
r = sd_rtnl_message_route_new(RTM_NEWROUTE, AF_INET, 0, 0, 0,
RT_TABLE_MAIN, RT_SCOPE_UNIVERSE, RTPROT_BOOT,
RTN_UNICAST, 0, &req);
if (r < 0) {
log_error("Could not create RTM_NEWROUTE message: %s", strerror(-r));
return;
}
r = sd_rtnl_message_append(req, RTA_GATEWAY, &addr);
if (r < 0) {
log_error("Could not append RTA_GATEWAY attribute: %s", strerror(-r));
return;
}
r = sd_rtnl_message_append(req, RTA_OIF, &index);
if (r < 0) {
log_error("Could not append RTA_OIF attribute: %s", strerror(-r));
return;
}
assert(sd_rtnl_message_read(req, &type, &data) > 0);
assert(type == RTA_GATEWAY);
assert(*(uint32_t *) data == addr);
assert(sd_rtnl_message_read(req, &type, &data) > 0);
assert(type == RTA_OIF);
assert(*(uint32_t *) data == index);
}
static void test_multiple(void) {
sd_rtnl *rtnl1, *rtnl2;
@ -76,6 +113,8 @@ int main(void) {
test_multiple();
test_route();
assert(sd_rtnl_open(0, &rtnl) >= 0);
assert(rtnl);
@ -115,17 +154,17 @@ int main(void) {
assert(sd_rtnl_message_read(m, &type, data) == 0);
assert(sd_rtnl_send_with_reply_and_block(rtnl, m, -1, &r) >= 0);
while (sd_rtnl_message_read(r, &type, &data)) {
while (sd_rtnl_message_read(r, &type, &data) > 0) {
switch (type) {
// case IFLA_MTU:
// assert(*(unsigned int *) data == 65536);
// break;;
// break;
// case IFLA_QDISC:
// assert(streq((char *) data, "noqueue"));
// break;;
// break;
case IFLA_IFNAME:
assert(streq((char *) data, "lo"));
break;;
break;
}
}

12
src/systemd/sd-rtnl.h
View File

@ -38,8 +38,16 @@ sd_rtnl *sd_rtnl_unref(sd_rtnl *nl);
int sd_rtnl_send_with_reply_and_block(sd_rtnl *nl, sd_rtnl_message *message, uint64_t timeout, sd_rtnl_message **reply);
/* messages */
int sd_rtnl_message_link_new(uint16_t msg_type, int index, unsigned int type, unsigned int flags, sd_rtnl_message **ret);
int sd_rtnl_message_addr_new(uint16_t msg_type, int index, unsigned char family, unsigned char prefixlen, unsigned char flags, unsigned char scope, sd_rtnl_message **ret);
int sd_rtnl_message_link_new(uint16_t msg_type, int index, unsigned int type,
unsigned int flags, sd_rtnl_message **ret);
int sd_rtnl_message_addr_new(uint16_t msg_type, int index, unsigned char family,
unsigned char prefixlen, unsigned char flags,
unsigned char scope, sd_rtnl_message **ret);
int sd_rtnl_message_route_new(uint16_t nlmsg_type, unsigned char rtm_family,
unsigned char rtm_dst_len, unsigned char rtm_src_len,
unsigned char rtm_tos, unsigned char rtm_table,
unsigned char rtm_scope, unsigned char rtm_protocol,
unsigned char rtm_type, unsigned flags, sd_rtnl_message **ret);
sd_rtnl_message *sd_rtnl_message_ref(sd_rtnl_message *m);
sd_rtnl_message *sd_rtnl_message_unref(sd_rtnl_message *m);