| /* |
| * Copyright 2017, The Android Open Source Project |
| * |
| * Licensed 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. |
| */ |
| |
| #include "interface.h" |
| |
| #include "netlink.h" |
| |
| #include <errno.h> |
| #include <linux/if.h> |
| #include <linux/if_ether.h> |
| #include <linux/route.h> |
| #include <linux/rtnetlink.h> |
| #include <string.h> |
| #include <unistd.h> |
| |
| in_addr_t broadcastFromNetmask(in_addr_t address, in_addr_t netmask) { |
| // The broadcast address is the address with the bits excluded in the |
| // netmask set to 1. For example if address = 10.0.2.15 and netmask is |
| // 255.255.255.0 then the broadcast is 10.0.2.255. If instead netmask was |
| // 255.0.0.0.0 then the broadcast would be 10.255.255.255 |
| // |
| // Simply set all the lower bits to 1 and that should do it. |
| return address | (~netmask); |
| } |
| |
| Interface::Interface() : mSocketFd(-1) { |
| } |
| |
| Interface::~Interface() { |
| if (mSocketFd != -1) { |
| close(mSocketFd); |
| mSocketFd = -1; |
| } |
| } |
| |
| Result Interface::init(const char* interfaceName) { |
| mInterfaceName = interfaceName; |
| |
| if (mSocketFd != -1) { |
| return Result::error("Interface initialized more than once"); |
| } |
| |
| mSocketFd = ::socket(AF_NETLINK, SOCK_RAW | SOCK_CLOEXEC, NETLINK_ROUTE); |
| if (mSocketFd == -1) { |
| return Result::error("Failed to create interface socket for '%s': %s", |
| interfaceName, strerror(errno)); |
| } |
| |
| Result res = populateIndex(); |
| if (!res) { |
| return res; |
| } |
| |
| res = populateMacAddress(); |
| if (!res) { |
| return res; |
| } |
| |
| res = bringUp(); |
| if (!res) { |
| return res; |
| } |
| |
| res = setAddress(0, 0); |
| if (!res) { |
| return res; |
| } |
| |
| return Result::success(); |
| } |
| |
| Result Interface::bringUp() { |
| return setInterfaceUp(true); |
| } |
| |
| Result Interface::bringDown() { |
| return setInterfaceUp(false); |
| } |
| |
| Result Interface::setMtu(uint16_t mtu) { |
| struct ifreq request = createRequest(); |
| |
| strncpy(request.ifr_name, mInterfaceName.c_str(), sizeof(request.ifr_name)); |
| request.ifr_mtu = mtu; |
| int status = ::ioctl(mSocketFd, SIOCSIFMTU, &request); |
| if (status != 0) { |
| return Result::error("Failed to set interface MTU %u for '%s': %s", |
| static_cast<unsigned int>(mtu), |
| mInterfaceName.c_str(), |
| strerror(errno)); |
| } |
| |
| return Result::success(); |
| } |
| |
| Result Interface::setAddress(in_addr_t address, in_addr_t subnetMask) { |
| struct Request { |
| struct nlmsghdr hdr; |
| struct ifaddrmsg msg; |
| char buf[256]; |
| } request; |
| |
| memset(&request, 0, sizeof(request)); |
| |
| request.hdr.nlmsg_len = NLMSG_LENGTH(sizeof(request.msg)); |
| request.hdr.nlmsg_type = RTM_NEWADDR; |
| request.hdr.nlmsg_flags = NLM_F_REQUEST | |
| NLM_F_ACK | |
| NLM_F_CREATE | |
| NLM_F_REPLACE; |
| |
| request.msg.ifa_family = AF_INET; |
| // Count the number of bits in the subnet mask, this is the length. |
| request.msg.ifa_prefixlen = __builtin_popcount(subnetMask); |
| request.msg.ifa_index = mIndex; |
| |
| addRouterAttribute(request, IFA_ADDRESS, &address, sizeof(address)); |
| addRouterAttribute(request, IFA_LOCAL, &address, sizeof(address)); |
| in_addr_t broadcast = broadcastFromNetmask(address, subnetMask); |
| addRouterAttribute(request, IFA_BROADCAST, &broadcast, sizeof(broadcast)); |
| |
| struct sockaddr_nl nlAddr; |
| memset(&nlAddr, 0, sizeof(nlAddr)); |
| nlAddr.nl_family = AF_NETLINK; |
| |
| int status = ::sendto(mSocketFd, &request, request.hdr.nlmsg_len, 0, |
| reinterpret_cast<sockaddr*>(&nlAddr), |
| sizeof(nlAddr)); |
| if (status == -1) { |
| return Result::error("Unable to set interface address: %s", |
| strerror(errno)); |
| } |
| char buffer[8192]; |
| status = ::recv(mSocketFd, buffer, sizeof(buffer), 0); |
| if (status < 0) { |
| return Result::error("Unable to read netlink response: %s", |
| strerror(errno)); |
| } |
| size_t responseSize = static_cast<size_t>(status); |
| if (responseSize < sizeof(nlmsghdr)) { |
| return Result::error("Received incomplete response from netlink"); |
| } |
| auto response = reinterpret_cast<const nlmsghdr*>(buffer); |
| if (response->nlmsg_type == NLMSG_ERROR) { |
| if (responseSize < NLMSG_HDRLEN + sizeof(nlmsgerr)) { |
| return Result::error("Recieved an error from netlink but the " |
| "response was incomplete"); |
| } |
| auto err = reinterpret_cast<const nlmsgerr*>(NLMSG_DATA(response)); |
| if (err->error) { |
| return Result::error("Could not set interface address: %s", |
| strerror(-err->error)); |
| } |
| } |
| return Result::success(); |
| } |
| |
| struct ifreq Interface::createRequest() const { |
| struct ifreq request; |
| memset(&request, 0, sizeof(request)); |
| strncpy(request.ifr_name, mInterfaceName.c_str(), sizeof(request.ifr_name)); |
| request.ifr_name[sizeof(request.ifr_name) - 1] = '\0'; |
| |
| return request; |
| } |
| |
| Result Interface::populateIndex() { |
| struct ifreq request = createRequest(); |
| |
| int status = ::ioctl(mSocketFd, SIOCGIFINDEX, &request); |
| if (status != 0) { |
| return Result::error("Failed to get interface index for '%s': %s", |
| mInterfaceName.c_str(), strerror(errno)); |
| } |
| mIndex = request.ifr_ifindex; |
| return Result::success(); |
| } |
| |
| Result Interface::populateMacAddress() { |
| struct ifreq request = createRequest(); |
| |
| int status = ::ioctl(mSocketFd, SIOCGIFHWADDR, &request); |
| if (status != 0) { |
| return Result::error("Failed to get MAC address for '%s': %s", |
| mInterfaceName.c_str(), strerror(errno)); |
| } |
| memcpy(mMacAddress, &request.ifr_hwaddr.sa_data, ETH_ALEN); |
| return Result::success(); |
| } |
| |
| Result Interface::setInterfaceUp(bool shouldBeUp) { |
| struct ifreq request = createRequest(); |
| |
| int status = ::ioctl(mSocketFd, SIOCGIFFLAGS, &request); |
| if (status != 0) { |
| return Result::error("Failed to get interface flags for '%s': %s", |
| mInterfaceName.c_str(), strerror(errno)); |
| } |
| |
| bool isUp = (request.ifr_flags & IFF_UP) != 0; |
| if (isUp != shouldBeUp) { |
| // Toggle the up flag |
| request.ifr_flags ^= IFF_UP; |
| } else { |
| // Interface is already in desired state, do nothing |
| return Result::success(); |
| } |
| |
| status = ::ioctl(mSocketFd, SIOCSIFFLAGS, &request); |
| if (status != 0) { |
| return Result::error("Failed to set interface flags for '%s': %s", |
| mInterfaceName.c_str(), strerror(errno)); |
| } |
| |
| return Result::success(); |
| } |
| |