/**************************************************************************** * apps/netutils/netlib/netlib_getroute.c * * Licensed to the Apache Software Foundation (ASF) under one or more * contributor license agreements. See the NOTICE file distributed with * this work for additional information regarding copyright ownership. The * ASF licenses this file to you under the Apache License, Version 2.0 (the * "License"); you may not use this file except in compliance with the * License. You may obtain a copy of the License at * * http://www.apache.org/licenses/LICENSE-2.0 * * Unless required by applicable law or agreed to in writing, software * distributed under the License is distributed on an "AS IS" BASIS, WITHOUT * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the * License for the specific language governing permissions and limitations * under the License. * ****************************************************************************/ /**************************************************************************** * Included Files ****************************************************************************/ #include #include #include #include #include #include #include #include #include #include #include "netutils/netlib.h" #if defined(CONFIG_NETLINK_ROUTE) && defined(CONFIG_NET_ROUTE) /**************************************************************************** * Pre-processor definitions ****************************************************************************/ #define RXBUFFER_SIZE 64 /**************************************************************************** * Private Types ****************************************************************************/ struct netlib_sendto_request_s { struct nlmsghdr hdr; struct rtgenmsg gen; }; struct netlib_recvfrom_response_s { struct nlmsghdr hdr; struct rtmsg rte; struct rtattr attr; /* Attribute data and additional attributes follow */ }; /**************************************************************************** * Private Functions ****************************************************************************/ static int copy_address(FAR struct sockaddr_storage *dest, FAR void *src, unsigned int addrlen, unsigned int maxaddr, sa_family_t family) { DEBUGASSERT(addrlen <= maxaddr); if (addrlen > maxaddr) { fprintf(stderr, "ERROR: Bad address length: %u\n", addrlen); return -EIO; } dest->ss_family = family; memcpy(dest->ss_data, src, addrlen); return OK; } /**************************************************************************** * Public Functions ****************************************************************************/ /**************************************************************************** * Name: netlib_get_route * * Description: * Return a snapshot of the routing table for the selected address family. * * Parameters: * rtelist - The location to store the list of devices. * nentries - The size of the provided 'rtelist' in number of entries. * The size of one entry is given by sizeof(struct rtentry); * family - Address family. See AF_* definitions in sys/socket.h. * * Return: * The number of routing table entries read is returned on success; a * negated errno value is returned on failure. * ****************************************************************************/ ssize_t netlib_get_route(FAR struct rtentry *rtelist, unsigned int nentries, sa_family_t family) { struct netlib_sendto_request_s req; struct sockaddr_nl addr; static unsigned int seqno = 0; unsigned int thiseq; unsigned int maxaddr; ssize_t nsent; ssize_t nrecvd; size_t ncopied = 0; pid_t pid; bool enddump; int fd; int ret; union { char rxbuffer[RXBUFFER_SIZE]; struct netlib_recvfrom_response_s resp; } u; /* Create a NetLink socket with NETLINK_ROUTE protocol */ fd = socket(PF_NETLINK, SOCK_DGRAM, NETLINK_ROUTE); if (fd < 0) { int errcode = errno; fprintf(stderr, "ERROR: socket() failed: %d\n", errcode); return -errcode; } /* Bind the socket so that we can use send() and receive() */ pid = gettid(); addr.nl_family = AF_NETLINK; addr.nl_pad = 0; addr.nl_pid = pid; addr.nl_groups = 0; ret = bind(fd, (FAR const struct sockaddr *)&addr, sizeof(struct sockaddr_nl)); if (ret < 0) { int errcode = errno; fprintf(stderr, "ERROR: bind() failed: %d\n", errcode); ret = -errcode; goto errout_with_socket; } /* Initialize the request */ thiseq = ++seqno; memset(&req, 0, sizeof(req)); req.hdr.nlmsg_len = NLMSG_LENGTH(sizeof(struct rtgenmsg)); req.hdr.nlmsg_flags = NLM_F_REQUEST | NLM_F_DUMP; req.hdr.nlmsg_seq = thiseq; req.hdr.nlmsg_type = RTM_GETROUTE; req.hdr.nlmsg_pid = pid; req.gen.rtgen_family = family; nsent = send(fd, &req, req.hdr.nlmsg_len, 0); if (nsent < 0) { int errcode = errno; fprintf(stderr, "ERROR: send() failed: %d\n", errcode); ret = -errcode; goto errout_with_socket; } /* Read the response(s) */ maxaddr = sizeof(struct sockaddr_storage) - sizeof(sa_family_t); for (enddump = false; !enddump; ) { /* Receive the next device response */ nrecvd = recv(fd, &u.rxbuffer, RXBUFFER_SIZE, 0); if (nrecvd < 0) { int errcode = errno; fprintf(stderr, "ERROR: recv() failed: %d\n", errcode); ret = -errcode; goto errout_with_socket; } /* Verify the data and transfer the device list to the caller */ if (u.resp.hdr.nlmsg_len < sizeof(struct nlmsghdr) || u.resp.hdr.nlmsg_len != nrecvd) { fprintf(stderr, "ERROR: Bad message\n"); ret = -EIO; goto errout_with_socket; } /* The sequence number in the response should match the sequence * number in the request (since we created the socket, this should * always be true). */ if (u.resp.hdr.nlmsg_seq != thiseq) { fprintf(stderr, "ERROR: Bad sequence number in response\n"); ret = -EIO; goto errout_with_socket; } /* This should be a routing table response */ if (u.resp.rte.rtm_table != RT_TABLE_MAIN) { fprintf(stderr, "ERROR: Not a routing table response\n"); ret = -EIO; goto errout_with_socket; } /* Copy the routing table entry to the caller's buffer */ switch (u.resp.hdr.nlmsg_type) { case NLMSG_DONE: enddump = true; break; case RTM_NEWROUTE: { FAR struct rtentry *dest; FAR struct rtattr *attr; unsigned int paylen; /* Verify the expected message length */ if (nrecvd <= sizeof(struct netlib_recvfrom_response_s) || u.resp.rte.rtm_family != family) { fprintf(stderr, "ERROR: Bad massage size: %ld\n", (long)nrecvd); goto errout_with_socket; } /* Decode the response */ dest = &rtelist[ncopied]; memset(dest, 0, sizeof(struct rtentry)); attr = &u.resp.attr; paylen = u.resp.hdr.nlmsg_len - sizeof(struct nlmsghdr) - sizeof(struct rtmsg); for (; RTA_OK(attr, paylen); attr = RTA_NEXT(attr, paylen)) { unsigned int attrlen = RTA_PAYLOAD(attr); switch (attr->rta_type) { case RTA_DST: /* The destination network */ ret = copy_address(&dest->rt_dst, RTA_DATA(attr), attrlen, maxaddr, family); if (ret < 0) { goto errout_with_socket; } break; case RTA_SRC: /* The source address mask */ break; case RTA_GENMASK: /* The network address mask */ ret = copy_address(&dest->rt_genmask, RTA_DATA(attr), attrlen, maxaddr, family); if (ret < 0) { goto errout_with_socket; } break; case RTA_GATEWAY: /* Route packets via this router */ ret = copy_address(&dest->rt_gateway, RTA_DATA(attr), attrlen, maxaddr, family); if (ret < 0) { goto errout_with_socket; } break; default: break; } } ncopied++; } break; default: fprintf(stderr, "ERROR: Message type %u, length %u\n", u.resp.hdr.nlmsg_type, (unsigned)u.resp.hdr.nlmsg_len); ret = -EIO; goto errout_with_socket; } } close(fd); return ncopied; errout_with_socket: close(fd); return ret; } #endif /* CONFIG_NETLINK_ROUTE && CONFIG_NET_ROUTE */