blob: ae36a70d6499064fb708bcadd921003d10233fb9 [file] [log] [blame]
Garrick Evansf0ab7132019-06-18 14:50:42 +09001// Copyright 2019 The Chromium OS Authors. All rights reserved.
2// Use of this source code is governed by a BSD-style license that can be
3// found in the LICENSE file.
4
Garrick Evans3388a032020-03-24 11:25:55 +09005#include "patchpanel/datapath.h"
Garrick Evansf0ab7132019-06-18 14:50:42 +09006
Garrick Evans3d97a392020-02-21 15:24:37 +09007#include <arpa/inet.h>
Garrick Evansc7ae82c2019-09-04 16:25:10 +09008#include <fcntl.h>
9#include <linux/if_tun.h>
10#include <linux/sockios.h>
11#include <net/if.h>
12#include <net/if_arp.h>
13#include <netinet/in.h>
14#include <string.h>
15#include <sys/ioctl.h>
16#include <sys/socket.h>
17
Hugo Benichid82d8832020-08-14 10:05:03 +090018#include <vector>
19
Garrick Evansc7ae82c2019-09-04 16:25:10 +090020#include <base/files/scoped_file.h>
21#include <base/logging.h>
Taoyu Li79871c92020-07-02 16:09:39 +090022#include <base/posix/eintr_wrapper.h>
Garrick Evans54861622019-07-19 09:05:09 +090023#include <base/strings/string_number_conversions.h>
Garrick Evans4f9f5572019-11-26 10:25:16 +090024#include <brillo/userdb_utils.h>
Garrick Evans54861622019-07-19 09:05:09 +090025
Jason Jeremy Imana7273a32020-08-04 11:25:31 +090026#include "patchpanel/adb_proxy.h"
Garrick Evans3388a032020-03-24 11:25:55 +090027#include "patchpanel/net_util.h"
28#include "patchpanel/scoped_ns.h"
Garrick Evansc7ae82c2019-09-04 16:25:10 +090029
Garrick Evans3388a032020-03-24 11:25:55 +090030namespace patchpanel {
Garrick Evans54861622019-07-19 09:05:09 +090031
Garrick Evansc7ae82c2019-09-04 16:25:10 +090032namespace {
Hugo Benichi76675592020-04-08 14:29:57 +090033// TODO(hugobenichi) Consolidate this constant definition in a single place.
34constexpr pid_t kTestPID = -2;
Garrick Evansc7ae82c2019-09-04 16:25:10 +090035constexpr char kDefaultIfname[] = "vmtap%d";
36constexpr char kTunDev[] = "/dev/net/tun";
Jason Jeremy Imana7273a32020-08-04 11:25:31 +090037constexpr char kArcAddr[] = "100.115.92.2";
38constexpr char kLocalhostAddr[] = "127.0.0.1";
39constexpr uint16_t kAdbServerPort = 5555;
Hugo Benichie8758b52020-04-03 14:49:01 +090040
Garrick Evans8a067562020-05-11 12:47:30 +090041std::string PrefixIfname(const std::string& prefix, const std::string& ifname) {
42 std::string n = prefix + ifname;
Garrick Evans2f581a02020-05-11 10:43:35 +090043 if (n.length() < IFNAMSIZ)
44 return n;
Garrick Evans54861622019-07-19 09:05:09 +090045
Garrick Evans2f581a02020-05-11 10:43:35 +090046 // Best effort attempt to preserve the interface number, assuming it's the
47 // last char in the name.
48 auto c = ifname[ifname.length() - 1];
49 n.resize(IFNAMSIZ - 1);
50 n[n.length() - 1] = c;
51 return n;
Garrick Evans54861622019-07-19 09:05:09 +090052}
Garrick Evansf0ab7132019-06-18 14:50:42 +090053
Garrick Evans8a067562020-05-11 12:47:30 +090054} // namespace
55
56std::string ArcVethHostName(const std::string& ifname) {
57 return PrefixIfname("veth", ifname);
58}
59
60std::string ArcBridgeName(const std::string& ifname) {
61 return PrefixIfname("arc_", ifname);
62}
63
Jason Jeremy Imana7273a32020-08-04 11:25:31 +090064Datapath::Datapath(MinijailedProcessRunner* process_runner, Firewall* firewall)
65 : Datapath(process_runner, firewall, ioctl) {}
Garrick Evansc7ae82c2019-09-04 16:25:10 +090066
Jason Jeremy Imana7273a32020-08-04 11:25:31 +090067Datapath::Datapath(MinijailedProcessRunner* process_runner,
68 Firewall* firewall,
69 ioctl_t ioctl_hook)
70 : process_runner_(process_runner), firewall_(firewall), ioctl_(ioctl_hook) {
Garrick Evansf0ab7132019-06-18 14:50:42 +090071 CHECK(process_runner_);
72}
73
Garrick Evans260ff302019-07-25 11:22:50 +090074MinijailedProcessRunner& Datapath::runner() const {
75 return *process_runner_;
76}
77
Hugo Benichi33860d72020-07-09 16:34:01 +090078bool Datapath::NetnsAttachName(const std::string& netns_name, pid_t netns_pid) {
79 // Try first to delete any netns with name |netns_name| in case patchpanel
80 // did not exit cleanly.
81 if (process_runner_->ip_netns_delete(netns_name, false /*log_failures*/) == 0)
82 LOG(INFO) << "Deleted left over network namespace name " << netns_name;
83 return process_runner_->ip_netns_attach(netns_name, netns_pid) == 0;
84}
85
86bool Datapath::NetnsDeleteName(const std::string& netns_name) {
87 return process_runner_->ip_netns_delete(netns_name) == 0;
88}
89
Garrick Evans8a949dc2019-07-18 16:17:53 +090090bool Datapath::AddBridge(const std::string& ifname,
Garrick Evans7a1a9ee2020-01-28 11:03:57 +090091 uint32_t ipv4_addr,
92 uint32_t ipv4_prefix_len) {
Garrick Evans8a949dc2019-07-18 16:17:53 +090093 // Configure the persistent Chrome OS bridge interface with static IP.
Garrick Evans8e8e3472020-01-23 14:03:50 +090094 if (process_runner_->brctl("addbr", {ifname}) != 0) {
Garrick Evans8a949dc2019-07-18 16:17:53 +090095 return false;
96 }
97
Garrick Evans6f4fa3a2020-02-10 16:15:09 +090098 if (process_runner_->ip(
99 "addr", "add",
100 {IPv4AddressToCidrString(ipv4_addr, ipv4_prefix_len), "brd",
101 IPv4AddressToString(Ipv4BroadcastAddr(ipv4_addr, ipv4_prefix_len)),
102 "dev", ifname}) != 0) {
Garrick Evans7a1a9ee2020-01-28 11:03:57 +0900103 RemoveBridge(ifname);
104 return false;
105 }
106
107 if (process_runner_->ip("link", "set", {ifname, "up"}) != 0) {
Garrick Evans8a949dc2019-07-18 16:17:53 +0900108 RemoveBridge(ifname);
109 return false;
110 }
111
112 // See nat.conf in chromeos-nat-init for the rest of the NAT setup rules.
Hugo Benichie8758b52020-04-03 14:49:01 +0900113 if (!AddOutboundIPv4SNATMark(ifname)) {
Garrick Evans8a949dc2019-07-18 16:17:53 +0900114 RemoveBridge(ifname);
115 return false;
116 }
117
118 return true;
119}
120
121void Datapath::RemoveBridge(const std::string& ifname) {
Hugo Benichie8758b52020-04-03 14:49:01 +0900122 RemoveOutboundIPv4SNATMark(ifname);
Garrick Evans7a1a9ee2020-01-28 11:03:57 +0900123 process_runner_->ip("link", "set", {ifname, "down"});
Garrick Evans8e8e3472020-01-23 14:03:50 +0900124 process_runner_->brctl("delbr", {ifname});
Garrick Evans8a949dc2019-07-18 16:17:53 +0900125}
126
Garrick Evans621ed262019-11-13 12:28:43 +0900127bool Datapath::AddToBridge(const std::string& br_ifname,
128 const std::string& ifname) {
Garrick Evans8e8e3472020-01-23 14:03:50 +0900129 return (process_runner_->brctl("addif", {br_ifname, ifname}) == 0);
Garrick Evans621ed262019-11-13 12:28:43 +0900130}
131
Garrick Evansc7ae82c2019-09-04 16:25:10 +0900132std::string Datapath::AddTAP(const std::string& name,
Garrick Evans621ed262019-11-13 12:28:43 +0900133 const MacAddress* mac_addr,
134 const SubnetAddress* ipv4_addr,
Garrick Evans4f9f5572019-11-26 10:25:16 +0900135 const std::string& user) {
Garrick Evansc7ae82c2019-09-04 16:25:10 +0900136 base::ScopedFD dev(open(kTunDev, O_RDWR | O_NONBLOCK));
137 if (!dev.is_valid()) {
138 PLOG(ERROR) << "Failed to open " << kTunDev;
139 return "";
140 }
141
142 struct ifreq ifr;
143 memset(&ifr, 0, sizeof(ifr));
144 strncpy(ifr.ifr_name, name.empty() ? kDefaultIfname : name.c_str(),
145 sizeof(ifr.ifr_name));
146 ifr.ifr_flags = IFF_TAP | IFF_NO_PI;
147
148 // If a template was given as the name, ifr_name will be updated with the
149 // actual interface name.
150 if ((*ioctl_)(dev.get(), TUNSETIFF, &ifr) != 0) {
Garrick Evans621ed262019-11-13 12:28:43 +0900151 PLOG(ERROR) << "Failed to create tap interface " << name;
Garrick Evansc7ae82c2019-09-04 16:25:10 +0900152 return "";
153 }
154 const char* ifname = ifr.ifr_name;
155
156 if ((*ioctl_)(dev.get(), TUNSETPERSIST, 1) != 0) {
Garrick Evans621ed262019-11-13 12:28:43 +0900157 PLOG(ERROR) << "Failed to persist the interface " << ifname;
Garrick Evansc7ae82c2019-09-04 16:25:10 +0900158 return "";
159 }
160
Garrick Evans4f9f5572019-11-26 10:25:16 +0900161 if (!user.empty()) {
162 uid_t uid = -1;
163 if (!brillo::userdb::GetUserInfo(user, &uid, nullptr)) {
164 PLOG(ERROR) << "Unable to look up UID for " << user;
165 RemoveTAP(ifname);
166 return "";
167 }
168 if ((*ioctl_)(dev.get(), TUNSETOWNER, uid) != 0) {
169 PLOG(ERROR) << "Failed to set owner " << uid << " of tap interface "
170 << ifname;
171 RemoveTAP(ifname);
172 return "";
173 }
Garrick Evansc7ae82c2019-09-04 16:25:10 +0900174 }
175
Hugo Benichib9b93fe2019-10-25 23:36:01 +0900176 // Create control socket for configuring the interface.
Garrick Evansc7ae82c2019-09-04 16:25:10 +0900177 base::ScopedFD sock(socket(AF_INET, SOCK_DGRAM | SOCK_CLOEXEC, 0));
178 if (!sock.is_valid()) {
179 PLOG(ERROR) << "Failed to create control socket for tap interface "
Garrick Evans621ed262019-11-13 12:28:43 +0900180 << ifname;
Garrick Evansc7ae82c2019-09-04 16:25:10 +0900181 RemoveTAP(ifname);
182 return "";
183 }
184
Garrick Evans621ed262019-11-13 12:28:43 +0900185 if (ipv4_addr) {
186 struct sockaddr_in* addr =
187 reinterpret_cast<struct sockaddr_in*>(&ifr.ifr_addr);
188 addr->sin_family = AF_INET;
189 addr->sin_addr.s_addr = static_cast<in_addr_t>(ipv4_addr->Address());
190 if ((*ioctl_)(sock.get(), SIOCSIFADDR, &ifr) != 0) {
191 PLOG(ERROR) << "Failed to set ip address for vmtap interface " << ifname
192 << " {" << ipv4_addr->ToCidrString() << "}";
193 RemoveTAP(ifname);
194 return "";
195 }
196
197 struct sockaddr_in* netmask =
198 reinterpret_cast<struct sockaddr_in*>(&ifr.ifr_netmask);
199 netmask->sin_family = AF_INET;
200 netmask->sin_addr.s_addr = static_cast<in_addr_t>(ipv4_addr->Netmask());
201 if ((*ioctl_)(sock.get(), SIOCSIFNETMASK, &ifr) != 0) {
202 PLOG(ERROR) << "Failed to set netmask for vmtap interface " << ifname
203 << " {" << ipv4_addr->ToCidrString() << "}";
204 RemoveTAP(ifname);
205 return "";
206 }
Garrick Evansc7ae82c2019-09-04 16:25:10 +0900207 }
208
Garrick Evans621ed262019-11-13 12:28:43 +0900209 if (mac_addr) {
210 struct sockaddr* hwaddr = &ifr.ifr_hwaddr;
211 hwaddr->sa_family = ARPHRD_ETHER;
212 memcpy(&hwaddr->sa_data, mac_addr, sizeof(*mac_addr));
213 if ((*ioctl_)(sock.get(), SIOCSIFHWADDR, &ifr) != 0) {
214 PLOG(ERROR) << "Failed to set mac address for vmtap interface " << ifname
215 << " {" << MacAddressToString(*mac_addr) << "}";
216 RemoveTAP(ifname);
217 return "";
218 }
Garrick Evansc7ae82c2019-09-04 16:25:10 +0900219 }
220
221 if ((*ioctl_)(sock.get(), SIOCGIFFLAGS, &ifr) != 0) {
Garrick Evans621ed262019-11-13 12:28:43 +0900222 PLOG(ERROR) << "Failed to get flags for tap interface " << ifname;
Garrick Evansc7ae82c2019-09-04 16:25:10 +0900223 RemoveTAP(ifname);
224 return "";
225 }
226
227 ifr.ifr_flags |= (IFF_UP | IFF_RUNNING);
228 if ((*ioctl_)(sock.get(), SIOCSIFFLAGS, &ifr) != 0) {
Garrick Evans621ed262019-11-13 12:28:43 +0900229 PLOG(ERROR) << "Failed to enable tap interface " << ifname;
Garrick Evansc7ae82c2019-09-04 16:25:10 +0900230 RemoveTAP(ifname);
231 return "";
232 }
233
234 return ifname;
235}
236
237void Datapath::RemoveTAP(const std::string& ifname) {
Garrick Evans8e8e3472020-01-23 14:03:50 +0900238 process_runner_->ip("tuntap", "del", {ifname, "mode", "tap"});
Garrick Evansc7ae82c2019-09-04 16:25:10 +0900239}
240
Hugo Benichi33860d72020-07-09 16:34:01 +0900241bool Datapath::ConnectVethPair(pid_t netns_pid,
242 const std::string& netns_name,
Hugo Benichi76675592020-04-08 14:29:57 +0900243 const std::string& veth_ifname,
244 const std::string& peer_ifname,
245 const MacAddress& remote_mac_addr,
246 uint32_t remote_ipv4_addr,
247 uint32_t remote_ipv4_prefix_len,
248 bool remote_multicast_flag) {
Hugo Benichi33860d72020-07-09 16:34:01 +0900249 // Set up the virtual pair across the current namespace and |netns_name|.
250 if (!AddVirtualInterfacePair(netns_name, veth_ifname, peer_ifname)) {
251 LOG(ERROR) << "Failed to create veth pair " << veth_ifname << ","
252 << peer_ifname;
253 return false;
254 }
255
256 // Configure the remote veth in namespace |netns_name|.
Hugo Benichi76675592020-04-08 14:29:57 +0900257 {
Hugo Benichi33860d72020-07-09 16:34:01 +0900258 ScopedNS ns(netns_pid);
259 if (!ns.IsValid() && netns_pid != kTestPID) {
Hugo Benichi76675592020-04-08 14:29:57 +0900260 LOG(ERROR)
261 << "Cannot create virtual link -- invalid container namespace?";
262 return false;
263 }
264
Hugo Benichi76675592020-04-08 14:29:57 +0900265 if (!ConfigureInterface(peer_ifname, remote_mac_addr, remote_ipv4_addr,
266 remote_ipv4_prefix_len, true /* link up */,
267 remote_multicast_flag)) {
268 LOG(ERROR) << "Failed to configure interface " << peer_ifname;
269 RemoveInterface(peer_ifname);
270 return false;
271 }
272 }
273
Hugo Benichi76675592020-04-08 14:29:57 +0900274 if (!ToggleInterface(veth_ifname, true /*up*/)) {
275 LOG(ERROR) << "Failed to bring up interface " << veth_ifname;
276 RemoveInterface(veth_ifname);
277 return false;
278 }
Hugo Benichi33860d72020-07-09 16:34:01 +0900279
Hugo Benichi76675592020-04-08 14:29:57 +0900280 return true;
281}
282
Hugo Benichi33860d72020-07-09 16:34:01 +0900283bool Datapath::AddVirtualInterfacePair(const std::string& netns_name,
284 const std::string& veth_ifname,
Garrick Evans2470caa2020-03-04 14:15:41 +0900285 const std::string& peer_ifname) {
Hugo Benichi33860d72020-07-09 16:34:01 +0900286 return process_runner_->ip("link", "add",
287 {veth_ifname, "type", "veth", "peer", "name",
288 peer_ifname, "netns", netns_name}) == 0;
Garrick Evans2470caa2020-03-04 14:15:41 +0900289}
Garrick Evans54861622019-07-19 09:05:09 +0900290
Garrick Evans2470caa2020-03-04 14:15:41 +0900291bool Datapath::ToggleInterface(const std::string& ifname, bool up) {
292 const std::string link = up ? "up" : "down";
293 return process_runner_->ip("link", "set", {ifname, link}) == 0;
294}
Garrick Evans54861622019-07-19 09:05:09 +0900295
Garrick Evans2470caa2020-03-04 14:15:41 +0900296bool Datapath::ConfigureInterface(const std::string& ifname,
297 const MacAddress& mac_addr,
298 uint32_t ipv4_addr,
299 uint32_t ipv4_prefix_len,
300 bool up,
301 bool enable_multicast) {
302 const std::string link = up ? "up" : "down";
303 const std::string multicast = enable_multicast ? "on" : "off";
304 return (process_runner_->ip(
305 "addr", "add",
306 {IPv4AddressToCidrString(ipv4_addr, ipv4_prefix_len), "brd",
307 IPv4AddressToString(
308 Ipv4BroadcastAddr(ipv4_addr, ipv4_prefix_len)),
309 "dev", ifname}) == 0) &&
310 (process_runner_->ip("link", "set",
311 {
312 "dev",
313 ifname,
314 link,
315 "addr",
316 MacAddressToString(mac_addr),
317 "multicast",
318 multicast,
319 }) == 0);
Garrick Evans54861622019-07-19 09:05:09 +0900320}
321
322void Datapath::RemoveInterface(const std::string& ifname) {
Garrick Evans8e8e3472020-01-23 14:03:50 +0900323 process_runner_->ip("link", "delete", {ifname}, false /*log_failures*/);
Garrick Evans54861622019-07-19 09:05:09 +0900324}
325
Hugo Benichi8d622b52020-08-13 15:24:12 +0900326void Datapath::StartRoutingDevice(const std::string& ext_ifname,
327 const std::string& int_ifname,
328 uint32_t int_ipv4_addr,
329 TrafficSource source) {
330 if (!ext_ifname.empty() &&
331 !AddInboundIPv4DNAT(ext_ifname, IPv4AddressToString(int_ipv4_addr)))
332 LOG(ERROR) << "Failed to configure ingress traffic rules for " << ext_ifname
333 << "->" << int_ifname;
334
Hugo Benichic6ae67c2020-08-14 15:02:13 +0900335 if (StartIpForwarding(IpFamily::IPv4, ext_ifname, int_ifname))
336 LOG(ERROR) << "Failed to enable IP forwarding for " << ext_ifname << "->"
337 << int_ifname;
Hugo Benichi8d622b52020-08-13 15:24:12 +0900338
Hugo Benichic6ae67c2020-08-14 15:02:13 +0900339 if (StartIpForwarding(IpFamily::IPv4, int_ifname, ext_ifname))
340 LOG(ERROR) << "Failed to enable IP forwarding for " << ext_ifname << "<-"
341 << int_ifname;
Hugo Benichi8d622b52020-08-13 15:24:12 +0900342
343 // TODO(b/161507671) If ext_ifname is not null, mark egress traffic with the
344 // fwmark routing tag corresponding to |ext_ifname|, and set up strong routing
345 // in ip rule.
346
347 // TODO(b/161507671) If ext_ifname is null, set up connection tracking for the
348 // current default interface.
349
Hugo Benichi9be19b12020-08-14 15:33:40 +0900350 if (!ModifyFwmarkSourceTag("-A", int_ifname, source))
351 LOG(ERROR) << "Failed to add PREROUTING fwmark tagging rule for source "
352 << source << " for " << int_ifname;
Hugo Benichi8d622b52020-08-13 15:24:12 +0900353}
354
355void Datapath::StopRoutingDevice(const std::string& ext_ifname,
356 const std::string& int_ifname,
357 uint32_t int_ipv4_addr,
358 TrafficSource source) {
359 if (!ext_ifname.empty())
360 RemoveInboundIPv4DNAT(ext_ifname, IPv4AddressToString(int_ipv4_addr));
Hugo Benichic6ae67c2020-08-14 15:02:13 +0900361 StopIpForwarding(IpFamily::IPv4, ext_ifname, int_ifname);
362 StopIpForwarding(IpFamily::IPv4, int_ifname, ext_ifname);
Hugo Benichi9be19b12020-08-14 15:33:40 +0900363 ModifyFwmarkSourceTag("-D", int_ifname, source);
Hugo Benichi8d622b52020-08-13 15:24:12 +0900364 // TODO(b/161507671) Remove routing tag marking for egress traffic.
365}
366
Garrick Evansf0ab7132019-06-18 14:50:42 +0900367bool Datapath::AddInboundIPv4DNAT(const std::string& ifname,
368 const std::string& ipv4_addr) {
369 // Direct ingress IP traffic to existing sockets.
Garrick Evans8e8e3472020-01-23 14:03:50 +0900370 if (process_runner_->iptables(
371 "nat", {"-A", "PREROUTING", "-i", ifname, "-m", "socket",
372 "--nowildcard", "-j", "ACCEPT", "-w"}) != 0)
Garrick Evansf0ab7132019-06-18 14:50:42 +0900373 return false;
374
375 // Direct ingress TCP & UDP traffic to ARC interface for new connections.
Garrick Evans8e8e3472020-01-23 14:03:50 +0900376 if (process_runner_->iptables(
377 "nat", {"-A", "PREROUTING", "-i", ifname, "-p", "tcp", "-j", "DNAT",
378 "--to-destination", ipv4_addr, "-w"}) != 0) {
Garrick Evansf0ab7132019-06-18 14:50:42 +0900379 RemoveInboundIPv4DNAT(ifname, ipv4_addr);
380 return false;
381 }
Garrick Evans8e8e3472020-01-23 14:03:50 +0900382 if (process_runner_->iptables(
383 "nat", {"-A", "PREROUTING", "-i", ifname, "-p", "udp", "-j", "DNAT",
384 "--to-destination", ipv4_addr, "-w"}) != 0) {
Garrick Evansf0ab7132019-06-18 14:50:42 +0900385 RemoveInboundIPv4DNAT(ifname, ipv4_addr);
386 return false;
387 }
388
389 return true;
390}
391
392void Datapath::RemoveInboundIPv4DNAT(const std::string& ifname,
393 const std::string& ipv4_addr) {
Garrick Evans8e8e3472020-01-23 14:03:50 +0900394 process_runner_->iptables(
395 "nat", {"-D", "PREROUTING", "-i", ifname, "-p", "udp", "-j", "DNAT",
396 "--to-destination", ipv4_addr, "-w"});
397 process_runner_->iptables(
398 "nat", {"-D", "PREROUTING", "-i", ifname, "-p", "tcp", "-j", "DNAT",
399 "--to-destination", ipv4_addr, "-w"});
400 process_runner_->iptables(
401 "nat", {"-D", "PREROUTING", "-i", ifname, "-m", "socket", "--nowildcard",
402 "-j", "ACCEPT", "-w"});
Garrick Evansf0ab7132019-06-18 14:50:42 +0900403}
404
Hugo Benichi8d622b52020-08-13 15:24:12 +0900405// TODO(hugobenichi) The name incorrectly refers to egress traffic, but this
406// FORWARD rule actually enables forwarding for ingress traffic. Fix the name.
Garrick Evansf0ab7132019-06-18 14:50:42 +0900407bool Datapath::AddOutboundIPv4(const std::string& ifname) {
Hugo Benichid82d8832020-08-14 10:05:03 +0900408 return StartIpForwarding(IpFamily::IPv4, "", ifname);
Garrick Evansf0ab7132019-06-18 14:50:42 +0900409}
410
411void Datapath::RemoveOutboundIPv4(const std::string& ifname) {
Hugo Benichid82d8832020-08-14 10:05:03 +0900412 StopIpForwarding(IpFamily::IPv4, "", ifname);
Garrick Evansf0ab7132019-06-18 14:50:42 +0900413}
414
Hugo Benichic6ae67c2020-08-14 15:02:13 +0900415// TODO(b/161507671) Stop relying on the traffic fwmark 1/1 once forwarded
416// egress traffic is routed through the fwmark routing tag.
Garrick Evansd291af62020-05-25 10:39:06 +0900417bool Datapath::AddSNATMarkRules() {
Taoyu Li79871c92020-07-02 16:09:39 +0900418 // chromium:1050579: INVALID packets cannot be tracked by conntrack therefore
419 // need to be explicitly dropped.
420 if (process_runner_->iptables(
Hugo Benichi6c445322020-08-12 16:46:19 +0900421 "filter", {"-A", "FORWARD", "-m", "mark", "--mark", "1/1", "-m",
Taoyu Li79871c92020-07-02 16:09:39 +0900422 "state", "--state", "INVALID", "-j", "DROP", "-w"}) != 0) {
423 return false;
424 }
Garrick Evansd291af62020-05-25 10:39:06 +0900425 if (process_runner_->iptables(
Hugo Benichi6c445322020-08-12 16:46:19 +0900426 "filter", {"-A", "FORWARD", "-m", "mark", "--mark", "1/1", "-j",
Garrick Evansd291af62020-05-25 10:39:06 +0900427 "ACCEPT", "-w"}) != 0) {
428 return false;
429 }
430 if (process_runner_->iptables(
Hugo Benichi6c445322020-08-12 16:46:19 +0900431 "nat", {"-A", "POSTROUTING", "-m", "mark", "--mark", "1/1", "-j",
Garrick Evansd291af62020-05-25 10:39:06 +0900432 "MASQUERADE", "-w"}) != 0) {
433 RemoveSNATMarkRules();
434 return false;
435 }
436 return true;
437}
438
439void Datapath::RemoveSNATMarkRules() {
440 process_runner_->iptables("nat", {"-D", "POSTROUTING", "-m", "mark", "--mark",
Hugo Benichi6c445322020-08-12 16:46:19 +0900441 "1/1", "-j", "MASQUERADE", "-w"});
Garrick Evansd291af62020-05-25 10:39:06 +0900442 process_runner_->iptables("filter", {"-D", "FORWARD", "-m", "mark", "--mark",
Hugo Benichi6c445322020-08-12 16:46:19 +0900443 "1/1", "-j", "ACCEPT", "-w"});
Taoyu Li79871c92020-07-02 16:09:39 +0900444 process_runner_->iptables(
Hugo Benichi6c445322020-08-12 16:46:19 +0900445 "filter", {"-D", "FORWARD", "-m", "mark", "--mark", "1/1", "-m", "state",
Taoyu Li79871c92020-07-02 16:09:39 +0900446 "--state", "INVALID", "-j", "DROP", "-w"});
Garrick Evansd291af62020-05-25 10:39:06 +0900447}
448
Garrick Evansff6e37f2020-05-25 10:54:47 +0900449bool Datapath::AddInterfaceSNAT(const std::string& ifname) {
450 return process_runner_->iptables("nat", {"-A", "POSTROUTING", "-o", ifname,
451 "-j", "MASQUERADE", "-w"}) == 0;
452}
453
454void Datapath::RemoveInterfaceSNAT(const std::string& ifname) {
455 process_runner_->iptables(
456 "nat", {"-D", "POSTROUTING", "-o", ifname, "-j", "MASQUERADE", "-w"});
457}
458
Hugo Benichie8758b52020-04-03 14:49:01 +0900459bool Datapath::AddOutboundIPv4SNATMark(const std::string& ifname) {
460 return process_runner_->iptables(
461 "mangle", {"-A", "PREROUTING", "-i", ifname, "-j", "MARK",
Hugo Benichi6c445322020-08-12 16:46:19 +0900462 "--set-mark", "1/1", "-w"}) == 0;
Hugo Benichie8758b52020-04-03 14:49:01 +0900463}
464
465void Datapath::RemoveOutboundIPv4SNATMark(const std::string& ifname) {
466 process_runner_->iptables("mangle", {"-D", "PREROUTING", "-i", ifname, "-j",
Hugo Benichi6c445322020-08-12 16:46:19 +0900467 "MARK", "--set-mark", "1/1", "-w"});
Hugo Benichie8758b52020-04-03 14:49:01 +0900468}
469
Garrick Evansd291af62020-05-25 10:39:06 +0900470bool Datapath::AddForwardEstablishedRule() {
471 return process_runner_->iptables(
472 "filter", {"-A", "FORWARD", "-m", "state", "--state",
473 "ESTABLISHED,RELATED", "-j", "ACCEPT", "-w"}) == 0;
474}
475
476void Datapath::RemoveForwardEstablishedRule() {
477 process_runner_->iptables("filter",
478 {"-D", "FORWARD", "-m", "state", "--state",
479 "ESTABLISHED,RELATED", "-j", "ACCEPT", "-w"});
480}
481
Garrick Evans664a82f2019-12-17 12:18:05 +0900482bool Datapath::MaskInterfaceFlags(const std::string& ifname,
483 uint16_t on,
484 uint16_t off) {
Taoyu Li90c13912019-11-26 17:56:54 +0900485 base::ScopedFD sock(socket(AF_INET, SOCK_DGRAM | SOCK_CLOEXEC, 0));
486 if (!sock.is_valid()) {
487 PLOG(ERROR) << "Failed to create control socket";
488 return false;
489 }
490 ifreq ifr;
491 snprintf(ifr.ifr_name, IFNAMSIZ, "%s", ifname.c_str());
492 if ((*ioctl_)(sock.get(), SIOCGIFFLAGS, &ifr) < 0) {
493 PLOG(WARNING) << "ioctl() failed to get interface flag on " << ifname;
494 return false;
495 }
Garrick Evans664a82f2019-12-17 12:18:05 +0900496 ifr.ifr_flags |= on;
497 ifr.ifr_flags &= ~off;
Taoyu Li90c13912019-11-26 17:56:54 +0900498 if ((*ioctl_)(sock.get(), SIOCSIFFLAGS, &ifr) < 0) {
Garrick Evans664a82f2019-12-17 12:18:05 +0900499 PLOG(WARNING) << "ioctl() failed to set flag 0x" << std::hex << on
500 << " unset flag 0x" << std::hex << off << " on " << ifname;
Taoyu Li90c13912019-11-26 17:56:54 +0900501 return false;
502 }
503 return true;
504}
505
Garrick Evans260ff302019-07-25 11:22:50 +0900506bool Datapath::AddIPv6HostRoute(const std::string& ifname,
507 const std::string& ipv6_addr,
508 int ipv6_prefix_len) {
509 std::string ipv6_addr_cidr =
510 ipv6_addr + "/" + std::to_string(ipv6_prefix_len);
511
Garrick Evans8e8e3472020-01-23 14:03:50 +0900512 return process_runner_->ip6("route", "replace",
513 {ipv6_addr_cidr, "dev", ifname}) == 0;
Garrick Evans260ff302019-07-25 11:22:50 +0900514}
515
516void Datapath::RemoveIPv6HostRoute(const std::string& ifname,
517 const std::string& ipv6_addr,
518 int ipv6_prefix_len) {
519 std::string ipv6_addr_cidr =
520 ipv6_addr + "/" + std::to_string(ipv6_prefix_len);
521
Garrick Evans8e8e3472020-01-23 14:03:50 +0900522 process_runner_->ip6("route", "del", {ipv6_addr_cidr, "dev", ifname});
Garrick Evans260ff302019-07-25 11:22:50 +0900523}
524
525bool Datapath::AddIPv6Neighbor(const std::string& ifname,
526 const std::string& ipv6_addr) {
Garrick Evans8e8e3472020-01-23 14:03:50 +0900527 return process_runner_->ip6("neigh", "add",
528 {"proxy", ipv6_addr, "dev", ifname}) == 0;
Garrick Evans260ff302019-07-25 11:22:50 +0900529}
530
531void Datapath::RemoveIPv6Neighbor(const std::string& ifname,
532 const std::string& ipv6_addr) {
Garrick Evans8e8e3472020-01-23 14:03:50 +0900533 process_runner_->ip6("neigh", "del", {"proxy", ipv6_addr, "dev", ifname});
Garrick Evans260ff302019-07-25 11:22:50 +0900534}
535
Hugo Benichi9be19b12020-08-14 15:33:40 +0900536bool Datapath::ModifyFwmarkSourceTag(const std::string& op,
537 const std::string& iif,
538 TrafficSource source) {
539 return ModifyFwmarkPrerouting(IpFamily::Dual, op, iif,
540 Fwmark::FromSource(source),
541 kFwmarkAllSourcesMask);
542}
543
544bool Datapath::ModifyFwmarkPrerouting(IpFamily family,
545 const std::string& op,
546 const std::string& iif,
547 Fwmark mark,
548 Fwmark mask,
549 bool log_failures) {
550 if (iif.empty()) {
551 LOG(ERROR)
552 << "Cannot change PREROUTING set-fwmark with no interface specified";
553 return false;
554 }
555
556 switch (family) {
557 case IPv4:
558 case IPv6:
559 case Dual:
560 break;
561 default:
562 LOG(ERROR) << "Cannot change PREROUTING set-fwmark for " << iif
563 << ": incorrect IP family " << family;
564 return false;
565 }
566
567 std::vector<std::string> args = {
568 op, "PREROUTING", "-i", iif,
569 "-j", "MARK", "--set-mark", mark.ToString() + "/" + mask.ToString(),
570 "-w"};
571
572 bool success = true;
573 if (family & AF_INET)
574 success &= process_runner_->iptables("mangle", args, log_failures) == 0;
575 if (family & AF_INET6)
576 success &= process_runner_->ip6tables("mangle", args, log_failures) == 0;
577 return false;
578}
579
Hugo Benichid82d8832020-08-14 10:05:03 +0900580bool Datapath::ModifyIpForwarding(IpFamily family,
581 const std::string& op,
582 const std::string& iif,
583 const std::string& oif,
584 bool log_failures) {
585 if (iif.empty() && oif.empty()) {
586 LOG(ERROR) << "Cannot change IP forwarding with no input or output "
587 "interface specified";
Garrick Evans260ff302019-07-25 11:22:50 +0900588 return false;
589 }
590
Hugo Benichid82d8832020-08-14 10:05:03 +0900591 switch (family) {
592 case IPv4:
593 case IPv6:
594 case Dual:
595 break;
596 default:
597
598 LOG(ERROR) << "Cannot change IP forwarding from \"" << iif << "\" to \""
599 << oif << "\": incorrect IP family " << family;
600 return false;
601 }
602
603 std::vector<std::string> args = {op, "FORWARD"};
604 if (!iif.empty()) {
605 args.push_back("-i");
606 args.push_back(iif);
607 }
608 if (!oif.empty()) {
609 args.push_back("-o");
610 args.push_back(oif);
611 }
612 args.push_back("-j");
613 args.push_back("ACCEPT");
614 args.push_back("-w");
615
616 bool success = true;
617 if (family & IpFamily::IPv4)
618 success &= process_runner_->iptables("filter", args, log_failures) == 0;
619 if (family & IpFamily::IPv6)
620 success &= process_runner_->ip6tables("filter", args, log_failures) == 0;
621 return success;
622}
623
624bool Datapath::StartIpForwarding(IpFamily family,
625 const std::string& iif,
626 const std::string& oif) {
627 return ModifyIpForwarding(family, "-A", iif, oif);
628}
629
630bool Datapath::StopIpForwarding(IpFamily family,
631 const std::string& iif,
632 const std::string& oif) {
633 return ModifyIpForwarding(family, "-D", iif, oif);
634}
635
636bool Datapath::AddIPv6Forwarding(const std::string& ifname1,
637 const std::string& ifname2) {
638 // Only start Ipv6 forwarding if -C returns false and it had not been
639 // started yet.
640 if (!ModifyIpForwarding(IpFamily::IPv6, "-C", ifname1, ifname2,
641 false /*log_failures*/) &&
642 !StartIpForwarding(IpFamily::IPv6, ifname1, ifname2)) {
643 return false;
644 }
645
646 if (!ModifyIpForwarding(IpFamily::IPv6, "-C", ifname2, ifname1,
647 false /*log_failures*/) &&
648 !StartIpForwarding(IpFamily::IPv6, ifname2, ifname1)) {
Garrick Evans260ff302019-07-25 11:22:50 +0900649 RemoveIPv6Forwarding(ifname1, ifname2);
650 return false;
651 }
652
653 return true;
654}
655
656void Datapath::RemoveIPv6Forwarding(const std::string& ifname1,
657 const std::string& ifname2) {
Hugo Benichid82d8832020-08-14 10:05:03 +0900658 StopIpForwarding(IpFamily::IPv6, ifname1, ifname2);
659 StopIpForwarding(IpFamily::IPv6, ifname2, ifname1);
Garrick Evans260ff302019-07-25 11:22:50 +0900660}
661
Garrick Evans3d97a392020-02-21 15:24:37 +0900662bool Datapath::AddIPv4Route(uint32_t gateway_addr,
663 uint32_t addr,
664 uint32_t netmask) {
665 struct rtentry route;
666 memset(&route, 0, sizeof(route));
Hugo Benichie8758b52020-04-03 14:49:01 +0900667 SetSockaddrIn(&route.rt_gateway, gateway_addr);
668 SetSockaddrIn(&route.rt_dst, addr & netmask);
669 SetSockaddrIn(&route.rt_genmask, netmask);
Garrick Evans3d97a392020-02-21 15:24:37 +0900670 route.rt_flags = RTF_UP | RTF_GATEWAY;
Hugo Benichie8758b52020-04-03 14:49:01 +0900671 return ModifyRtentry(SIOCADDRT, &route);
672}
Garrick Evans3d97a392020-02-21 15:24:37 +0900673
Hugo Benichie8758b52020-04-03 14:49:01 +0900674bool Datapath::DeleteIPv4Route(uint32_t gateway_addr,
675 uint32_t addr,
676 uint32_t netmask) {
677 struct rtentry route;
678 memset(&route, 0, sizeof(route));
679 SetSockaddrIn(&route.rt_gateway, gateway_addr);
680 SetSockaddrIn(&route.rt_dst, addr & netmask);
681 SetSockaddrIn(&route.rt_genmask, netmask);
682 route.rt_flags = RTF_UP | RTF_GATEWAY;
683 return ModifyRtentry(SIOCDELRT, &route);
684}
685
686bool Datapath::AddIPv4Route(const std::string& ifname,
687 uint32_t addr,
688 uint32_t netmask) {
689 struct rtentry route;
690 memset(&route, 0, sizeof(route));
691 SetSockaddrIn(&route.rt_dst, addr & netmask);
692 SetSockaddrIn(&route.rt_genmask, netmask);
693 char rt_dev[IFNAMSIZ];
694 strncpy(rt_dev, ifname.c_str(), IFNAMSIZ);
695 rt_dev[IFNAMSIZ - 1] = '\0';
696 route.rt_dev = rt_dev;
697 route.rt_flags = RTF_UP | RTF_GATEWAY;
698 return ModifyRtentry(SIOCADDRT, &route);
699}
700
701bool Datapath::DeleteIPv4Route(const std::string& ifname,
702 uint32_t addr,
703 uint32_t netmask) {
704 struct rtentry route;
705 memset(&route, 0, sizeof(route));
706 SetSockaddrIn(&route.rt_dst, addr & netmask);
707 SetSockaddrIn(&route.rt_genmask, netmask);
708 char rt_dev[IFNAMSIZ];
709 strncpy(rt_dev, ifname.c_str(), IFNAMSIZ);
710 rt_dev[IFNAMSIZ - 1] = '\0';
711 route.rt_dev = rt_dev;
712 route.rt_flags = RTF_UP | RTF_GATEWAY;
713 return ModifyRtentry(SIOCDELRT, &route);
714}
715
716bool Datapath::ModifyRtentry(unsigned long op, struct rtentry* route) {
717 DCHECK(route);
718 if (op != SIOCADDRT && op != SIOCDELRT) {
Andreea Costinas34aa7a92020-08-04 10:36:10 +0200719 LOG(ERROR) << "Invalid operation " << op << " for rtentry " << *route;
Garrick Evans3d97a392020-02-21 15:24:37 +0900720 return false;
721 }
Hugo Benichie8758b52020-04-03 14:49:01 +0900722 base::ScopedFD fd(socket(AF_INET, SOCK_DGRAM | SOCK_CLOEXEC, 0));
723 if (!fd.is_valid()) {
Andreea Costinas34aa7a92020-08-04 10:36:10 +0200724 PLOG(ERROR) << "Failed to create socket for adding rtentry " << *route;
Hugo Benichie8758b52020-04-03 14:49:01 +0900725 return false;
726 }
727 if (HANDLE_EINTR(ioctl_(fd.get(), op, route)) != 0) {
728 std::string opname = op == SIOCADDRT ? "add" : "delete";
Andreea Costinas34aa7a92020-08-04 10:36:10 +0200729 PLOG(ERROR) << "Failed to " << opname << " rtentry " << *route;
Garrick Evans3d97a392020-02-21 15:24:37 +0900730 return false;
731 }
732 return true;
733}
734
Jason Jeremy Imana7273a32020-08-04 11:25:31 +0900735bool Datapath::AddAdbPortForwardRule(const std::string& ifname) {
736 return firewall_->AddIpv4ForwardRule(patchpanel::ModifyPortRuleRequest::TCP,
737 kArcAddr, kAdbServerPort, ifname,
738 kLocalhostAddr, kAdbProxyTcpListenPort);
739}
740
741void Datapath::DeleteAdbPortForwardRule(const std::string& ifname) {
742 firewall_->DeleteIpv4ForwardRule(patchpanel::ModifyPortRuleRequest::TCP,
743 kArcAddr, kAdbServerPort, ifname,
744 kLocalhostAddr, kAdbProxyTcpListenPort);
745}
746
747bool Datapath::AddAdbPortAccessRule(const std::string& ifname) {
748 return firewall_->AddAcceptRules(patchpanel::ModifyPortRuleRequest::TCP,
749 kAdbProxyTcpListenPort, ifname);
750}
751
752void Datapath::DeleteAdbPortAccessRule(const std::string& ifname) {
753 firewall_->DeleteAcceptRules(patchpanel::ModifyPortRuleRequest::TCP,
754 kAdbProxyTcpListenPort, ifname);
755}
756
Garrick Evans3388a032020-03-24 11:25:55 +0900757} // namespace patchpanel