blob: d65525908958807e6f9c893f0cdfbc2b6ca26b30 [file] [log] [blame]
Tomasz Wasilczyk87329672019-07-12 11:43:00 -07001/*
2 * Copyright (C) 2019 The Android Open Source Project
3 *
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
7 *
8 * http://www.apache.org/licenses/LICENSE-2.0
9 *
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
15 */
16
17#include <libnetdevice/libnetdevice.h>
18
Tomasz Wasilczyk87329672019-07-12 11:43:00 -070019#include "common.h"
Tomasz Wasilczyka5c83a52020-06-16 15:58:19 -070020#include "ifreqs.h"
Tomasz Wasilczyk87329672019-07-12 11:43:00 -070021
22#include <android-base/logging.h>
Tomasz Wasilczyk66fc9392020-08-03 10:22:52 -070023#include <libnl++/MessageFactory.h>
24#include <libnl++/Socket.h>
Tomasz Wasilczyk87329672019-07-12 11:43:00 -070025
26#include <linux/can.h>
Tomasz Wasilczyk390c7112020-07-30 11:08:29 -070027#include <linux/rtnetlink.h>
Tomasz Wasilczyk87329672019-07-12 11:43:00 -070028#include <net/if.h>
Tomasz Wasilczyk426708b2024-10-10 11:34:36 -070029#include <sys/ioctl.h>
Tomasz Wasilczyk87329672019-07-12 11:43:00 -070030
Tomasz Wasilczykb87cdae2024-01-09 11:56:43 -080031#include <algorithm>
32#include <iterator>
Tomasz Wasilczykb759eca2020-08-06 16:12:40 -070033#include <sstream>
34
Tomasz Wasilczyk55f21932019-12-20 09:20:24 -080035namespace android::netdevice {
Tomasz Wasilczyk87329672019-07-12 11:43:00 -070036
Tomasz Wasilczyk2aa1a122020-07-20 14:03:05 -070037void useSocketDomain(int domain) {
38 ifreqs::socketDomain = domain;
Tomasz Wasilczykad107412020-05-29 16:32:43 -070039}
40
Tomasz Wasilczykf2dcb642024-10-10 12:13:29 -070041bool exists(std::string_view ifname) {
Tomasz Wasilczyk87329672019-07-12 11:43:00 -070042 return nametoindex(ifname) != 0;
43}
44
Tomasz Wasilczykf2dcb642024-10-10 12:13:29 -070045bool up(std::string_view ifname) {
Tomasz Wasilczyka5c83a52020-06-16 15:58:19 -070046 auto ifr = ifreqs::fromName(ifname);
47 if (!ifreqs::send(SIOCGIFFLAGS, ifr)) return false;
Tomasz Wasilczyk87329672019-07-12 11:43:00 -070048 ifr.ifr_flags |= IFF_UP;
Tomasz Wasilczyka5c83a52020-06-16 15:58:19 -070049 return ifreqs::send(SIOCSIFFLAGS, ifr);
Tomasz Wasilczyk87329672019-07-12 11:43:00 -070050}
51
Tomasz Wasilczykf2dcb642024-10-10 12:13:29 -070052bool down(std::string_view ifname) {
Tomasz Wasilczyka5c83a52020-06-16 15:58:19 -070053 auto ifr = ifreqs::fromName(ifname);
54 if (!ifreqs::send(SIOCGIFFLAGS, ifr)) return false;
Tomasz Wasilczyk87329672019-07-12 11:43:00 -070055 ifr.ifr_flags &= ~IFF_UP;
Tomasz Wasilczyka5c83a52020-06-16 15:58:19 -070056 return ifreqs::send(SIOCSIFFLAGS, ifr);
Tomasz Wasilczyk87329672019-07-12 11:43:00 -070057}
58
Tomasz Wasilczykf2dcb642024-10-10 12:13:29 -070059bool add(std::string_view dev, std::string_view type) {
Tomasz Wasilczyk3ab105b2020-08-05 14:07:40 -070060 nl::MessageFactory<ifinfomsg> req(RTM_NEWLINK,
61 NLM_F_REQUEST | NLM_F_CREATE | NLM_F_EXCL | NLM_F_ACK);
62 req.add(IFLA_IFNAME, dev);
Tomasz Wasilczyk87329672019-07-12 11:43:00 -070063
64 {
Tomasz Wasilczyk3ab105b2020-08-05 14:07:40 -070065 auto linkinfo = req.addNested(IFLA_LINKINFO);
Keith Mok6355aa92022-07-12 21:03:18 +000066 req.addBuffer(IFLA_INFO_KIND, type);
Tomasz Wasilczyk87329672019-07-12 11:43:00 -070067 }
68
Tomasz Wasilczyk66fc9392020-08-03 10:22:52 -070069 nl::Socket sock(NETLINK_ROUTE);
Tomasz Wasilczyk71396052020-08-04 16:21:39 -070070 return sock.send(req) && sock.receiveAck(req);
Tomasz Wasilczyk87329672019-07-12 11:43:00 -070071}
72
Tomasz Wasilczykf2dcb642024-10-10 12:13:29 -070073bool del(std::string_view dev) {
Tomasz Wasilczyk3ab105b2020-08-05 14:07:40 -070074 nl::MessageFactory<ifinfomsg> req(RTM_DELLINK, NLM_F_REQUEST | NLM_F_ACK);
75 req.add(IFLA_IFNAME, dev);
Tomasz Wasilczyk87329672019-07-12 11:43:00 -070076
Tomasz Wasilczyk66fc9392020-08-03 10:22:52 -070077 nl::Socket sock(NETLINK_ROUTE);
Tomasz Wasilczyk71396052020-08-04 16:21:39 -070078 return sock.send(req) && sock.receiveAck(req);
Tomasz Wasilczyk87329672019-07-12 11:43:00 -070079}
80
Tomasz Wasilczykf2dcb642024-10-10 12:13:29 -070081std::optional<hwaddr_t> getHwAddr(std::string_view ifname) {
Tomasz Wasilczyk91c3a042020-06-24 15:30:10 -070082 auto ifr = ifreqs::fromName(ifname);
83 if (!ifreqs::send(SIOCGIFHWADDR, ifr)) return std::nullopt;
84
85 hwaddr_t hwaddr;
86 memcpy(hwaddr.data(), ifr.ifr_hwaddr.sa_data, hwaddr.size());
87 return hwaddr;
88}
89
Tomasz Wasilczykf2dcb642024-10-10 12:13:29 -070090bool setHwAddr(std::string_view ifname, hwaddr_t hwaddr) {
Tomasz Wasilczykcaeae192020-06-25 13:36:48 -070091 auto ifr = ifreqs::fromName(ifname);
92
93 // fetch sa_family
94 if (!ifreqs::send(SIOCGIFHWADDR, ifr)) return false;
95
96 memcpy(ifr.ifr_hwaddr.sa_data, hwaddr.data(), hwaddr.size());
97 return ifreqs::send(SIOCSIFHWADDR, ifr);
98}
99
Tomasz Wasilczykf2dcb642024-10-10 12:13:29 -0700100std::optional<bool> isUp(std::string_view ifname) {
Tomasz Wasilczykb759eca2020-08-06 16:12:40 -0700101 auto ifr = ifreqs::fromName(ifname);
102 if (!ifreqs::send(SIOCGIFFLAGS, ifr)) return std::nullopt;
103 return ifr.ifr_flags & IFF_UP;
104}
105
Tomasz Wasilczykf2dcb642024-10-10 12:13:29 -0700106static bool hasIpv4(std::string_view ifname) {
Tomasz Wasilczyk2b162da2022-07-21 18:11:21 +0000107 auto ifr = ifreqs::fromName(ifname);
Yi Kong810d41a2023-12-06 14:03:58 +0900108 switch (ifreqs::trySend(SIOCGIFADDR, ifr)) {
Tomasz Wasilczyk2b162da2022-07-21 18:11:21 +0000109 case 0:
110 return true;
111 case EADDRNOTAVAIL:
112 case ENODEV:
113 return false;
114 default:
115 PLOG(WARNING) << "Failed checking IPv4 address";
116 return false;
117 }
118}
119
Tomasz Wasilczykb759eca2020-08-06 16:12:40 -0700120struct WaitState {
121 bool present;
122 bool up;
Tomasz Wasilczyk2b162da2022-07-21 18:11:21 +0000123 bool hasIpv4Addr;
Tomasz Wasilczykb759eca2020-08-06 16:12:40 -0700124
125 bool satisfied(WaitCondition cnd) const {
126 switch (cnd) {
127 case WaitCondition::PRESENT:
Tomasz Wasilczyk2b162da2022-07-21 18:11:21 +0000128 return present;
Tomasz Wasilczykb759eca2020-08-06 16:12:40 -0700129 case WaitCondition::PRESENT_AND_UP:
Tomasz Wasilczyk2b162da2022-07-21 18:11:21 +0000130 return present && up;
131 case WaitCondition::PRESENT_AND_IPV4:
132 return present && up && hasIpv4Addr;
Tomasz Wasilczykb759eca2020-08-06 16:12:40 -0700133 case WaitCondition::DOWN_OR_GONE:
Tomasz Wasilczyk2b162da2022-07-21 18:11:21 +0000134 return !present || !up;
Tomasz Wasilczykb759eca2020-08-06 16:12:40 -0700135 }
Tomasz Wasilczykb759eca2020-08-06 16:12:40 -0700136 }
137};
138
139static std::string toString(WaitCondition cnd) {
140 switch (cnd) {
141 case WaitCondition::PRESENT:
142 return "become present";
143 case WaitCondition::PRESENT_AND_UP:
144 return "come up";
Tomasz Wasilczyk2b162da2022-07-21 18:11:21 +0000145 case WaitCondition::PRESENT_AND_IPV4:
146 return "get IPv4 address";
Tomasz Wasilczykb759eca2020-08-06 16:12:40 -0700147 case WaitCondition::DOWN_OR_GONE:
148 return "go down";
149 }
150}
151
Tomasz Wasilczyk2b162da2022-07-21 18:11:21 +0000152static std::string toString(Quantifier quant) {
153 switch (quant) {
154 case Quantifier::ALL_OF:
155 return "all of";
156 case Quantifier::ANY_OF:
157 return "any of";
158 }
159}
160
Tomasz Wasilczykb759eca2020-08-06 16:12:40 -0700161static std::string toString(const std::set<std::string>& ifnames) {
162 std::stringstream ss;
163 std::copy(ifnames.begin(), ifnames.end(), std::ostream_iterator<std::string>(ss, ","));
164 auto str = ss.str();
165 str.pop_back();
166 return str;
167}
168
Tomasz Wasilczyk2b162da2022-07-21 18:11:21 +0000169std::optional<std::string> waitFor(std::set<std::string> ifnames, WaitCondition cnd,
170 Quantifier quant) {
171 nl::Socket sock(NETLINK_ROUTE, 0, RTMGRP_LINK | RTMGRP_IPV4_IFADDR);
Tomasz Wasilczykb759eca2020-08-06 16:12:40 -0700172
173 using StatesMap = std::map<std::string, WaitState>;
174 StatesMap states = {};
175 for (const auto ifname : ifnames) {
176 const auto present = exists(ifname);
177 const auto up = present && isUp(ifname).value_or(false);
Tomasz Wasilczyk2b162da2022-07-21 18:11:21 +0000178 const auto hasIpv4Addr = present && hasIpv4(ifname);
179 states[ifname] = {present, up, hasIpv4Addr};
Tomasz Wasilczykb759eca2020-08-06 16:12:40 -0700180 }
181
182 const auto mapConditionChecker = [cnd](const StatesMap::iterator::value_type& it) {
183 return it.second.satisfied(cnd);
184 };
Tomasz Wasilczyk2b162da2022-07-21 18:11:21 +0000185 const auto isFullySatisfied = [&states, quant,
186 mapConditionChecker]() -> std::optional<std::string> {
187 if (quant == Quantifier::ALL_OF) {
188 if (!std::all_of(states.begin(), states.end(), mapConditionChecker)) return {};
189 return states.begin()->first;
190 } else { // Quantifier::ANY_OF
191 const auto it = std::find_if(states.begin(), states.end(), mapConditionChecker);
192 if (it == states.end()) return {};
193 return it->first;
Tomasz Wasilczykb759eca2020-08-06 16:12:40 -0700194 }
195 };
196
Tomasz Wasilczyk2b162da2022-07-21 18:11:21 +0000197 if (const auto iface = isFullySatisfied()) return iface;
Tomasz Wasilczykb759eca2020-08-06 16:12:40 -0700198
Tomasz Wasilczyk2b162da2022-07-21 18:11:21 +0000199 LOG(DEBUG) << "Waiting for " << toString(quant) << " " << toString(ifnames) << " to "
Tomasz Wasilczykb759eca2020-08-06 16:12:40 -0700200 << toString(cnd);
Tomasz Wasilczyk2424b712020-08-07 11:04:26 -0700201 for (const auto rawMsg : sock) {
Tomasz Wasilczyk2b162da2022-07-21 18:11:21 +0000202 if (const auto msg = nl::Message<ifinfomsg>::parse(rawMsg, {RTM_NEWLINK, RTM_DELLINK});
203 msg.has_value()) {
204 // Interface added / removed
205 const auto ifname = msg->attributes.get<std::string>(IFLA_IFNAME);
206 if (ifnames.count(ifname) == 0) continue;
Tomasz Wasilczykb759eca2020-08-06 16:12:40 -0700207
Tomasz Wasilczyk2b162da2022-07-21 18:11:21 +0000208 auto& state = states[ifname];
209 state.present = (msg->header.nlmsg_type != RTM_DELLINK);
210 state.up = state.present && (msg->data.ifi_flags & IFF_UP) != 0;
211 if (!state.present) state.hasIpv4Addr = false;
Tomasz Wasilczykb759eca2020-08-06 16:12:40 -0700212
Tomasz Wasilczyk2b162da2022-07-21 18:11:21 +0000213 } else if (const auto msg =
214 nl::Message<ifaddrmsg>::parse(rawMsg, {RTM_NEWADDR, RTM_DELADDR});
215 msg.has_value()) {
216 // Address added / removed
217 const auto ifname = msg->attributes.get<std::string>(IFLA_IFNAME);
218 if (ifnames.count(ifname) == 0) continue;
Tomasz Wasilczykb759eca2020-08-06 16:12:40 -0700219
Tomasz Wasilczyk2b162da2022-07-21 18:11:21 +0000220 if (msg->header.nlmsg_type == RTM_NEWADDR) {
221 states[ifname].hasIpv4Addr = true;
222 } else {
223 // instead of tracking which one got deleted, let's just ask
224 states[ifname].hasIpv4Addr = hasIpv4(ifname);
225 }
226 }
227
228 if (const auto iface = isFullySatisfied()) {
229 LOG(DEBUG) << "Finished waiting for " << toString(quant) << " " << toString(ifnames)
Tomasz Wasilczyk2424b712020-08-07 11:04:26 -0700230 << " to " << toString(cnd);
Tomasz Wasilczyk2b162da2022-07-21 18:11:21 +0000231 return iface;
Tomasz Wasilczykb759eca2020-08-06 16:12:40 -0700232 }
233 }
Tomasz Wasilczyk2424b712020-08-07 11:04:26 -0700234 LOG(FATAL) << "Can't read Netlink socket";
Tomasz Wasilczyk2b162da2022-07-21 18:11:21 +0000235 return {};
Tomasz Wasilczykb759eca2020-08-06 16:12:40 -0700236}
237
Tomasz Wasilczyk55f21932019-12-20 09:20:24 -0800238} // namespace android::netdevice
Tomasz Wasilczyk91c3a042020-06-24 15:30:10 -0700239
240bool operator==(const android::netdevice::hwaddr_t lhs, const unsigned char rhs[ETH_ALEN]) {
241 static_assert(lhs.size() == ETH_ALEN);
242 return 0 == memcmp(lhs.data(), rhs, lhs.size());
243}