blob: b60e18c6ef0626da17bfa64625c816c5e2a0d2b7 [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
18#include <base/files/scoped_file.h>
19#include <base/logging.h>
Taoyu Li79871c92020-07-02 16:09:39 +090020#include <base/posix/eintr_wrapper.h>
Garrick Evans54861622019-07-19 09:05:09 +090021#include <base/strings/string_number_conversions.h>
Garrick Evans4f9f5572019-11-26 10:25:16 +090022#include <brillo/userdb_utils.h>
Garrick Evans54861622019-07-19 09:05:09 +090023
Jason Jeremy Imana7273a32020-08-04 11:25:31 +090024#include "patchpanel/adb_proxy.h"
Garrick Evans3388a032020-03-24 11:25:55 +090025#include "patchpanel/net_util.h"
26#include "patchpanel/scoped_ns.h"
Garrick Evansc7ae82c2019-09-04 16:25:10 +090027
Garrick Evans3388a032020-03-24 11:25:55 +090028namespace patchpanel {
Garrick Evans54861622019-07-19 09:05:09 +090029
Garrick Evansc7ae82c2019-09-04 16:25:10 +090030namespace {
Hugo Benichi76675592020-04-08 14:29:57 +090031// TODO(hugobenichi) Consolidate this constant definition in a single place.
32constexpr pid_t kTestPID = -2;
Garrick Evansc7ae82c2019-09-04 16:25:10 +090033constexpr char kDefaultIfname[] = "vmtap%d";
34constexpr char kTunDev[] = "/dev/net/tun";
Jason Jeremy Imana7273a32020-08-04 11:25:31 +090035constexpr char kArcAddr[] = "100.115.92.2";
36constexpr char kLocalhostAddr[] = "127.0.0.1";
37constexpr uint16_t kAdbServerPort = 5555;
Hugo Benichie8758b52020-04-03 14:49:01 +090038
Garrick Evans8a067562020-05-11 12:47:30 +090039std::string PrefixIfname(const std::string& prefix, const std::string& ifname) {
40 std::string n = prefix + ifname;
Garrick Evans2f581a02020-05-11 10:43:35 +090041 if (n.length() < IFNAMSIZ)
42 return n;
Garrick Evans54861622019-07-19 09:05:09 +090043
Garrick Evans2f581a02020-05-11 10:43:35 +090044 // Best effort attempt to preserve the interface number, assuming it's the
45 // last char in the name.
46 auto c = ifname[ifname.length() - 1];
47 n.resize(IFNAMSIZ - 1);
48 n[n.length() - 1] = c;
49 return n;
Garrick Evans54861622019-07-19 09:05:09 +090050}
Garrick Evansf0ab7132019-06-18 14:50:42 +090051
Garrick Evans8a067562020-05-11 12:47:30 +090052} // namespace
53
54std::string ArcVethHostName(const std::string& ifname) {
55 return PrefixIfname("veth", ifname);
56}
57
58std::string ArcBridgeName(const std::string& ifname) {
59 return PrefixIfname("arc_", ifname);
60}
61
Jason Jeremy Imana7273a32020-08-04 11:25:31 +090062Datapath::Datapath(MinijailedProcessRunner* process_runner, Firewall* firewall)
63 : Datapath(process_runner, firewall, ioctl) {}
Garrick Evansc7ae82c2019-09-04 16:25:10 +090064
Jason Jeremy Imana7273a32020-08-04 11:25:31 +090065Datapath::Datapath(MinijailedProcessRunner* process_runner,
66 Firewall* firewall,
67 ioctl_t ioctl_hook)
68 : process_runner_(process_runner), firewall_(firewall), ioctl_(ioctl_hook) {
Garrick Evansf0ab7132019-06-18 14:50:42 +090069 CHECK(process_runner_);
70}
71
Garrick Evans260ff302019-07-25 11:22:50 +090072MinijailedProcessRunner& Datapath::runner() const {
73 return *process_runner_;
74}
75
Hugo Benichi33860d72020-07-09 16:34:01 +090076bool Datapath::NetnsAttachName(const std::string& netns_name, pid_t netns_pid) {
77 // Try first to delete any netns with name |netns_name| in case patchpanel
78 // did not exit cleanly.
79 if (process_runner_->ip_netns_delete(netns_name, false /*log_failures*/) == 0)
80 LOG(INFO) << "Deleted left over network namespace name " << netns_name;
81 return process_runner_->ip_netns_attach(netns_name, netns_pid) == 0;
82}
83
84bool Datapath::NetnsDeleteName(const std::string& netns_name) {
85 return process_runner_->ip_netns_delete(netns_name) == 0;
86}
87
Garrick Evans8a949dc2019-07-18 16:17:53 +090088bool Datapath::AddBridge(const std::string& ifname,
Garrick Evans7a1a9ee2020-01-28 11:03:57 +090089 uint32_t ipv4_addr,
90 uint32_t ipv4_prefix_len) {
Garrick Evans8a949dc2019-07-18 16:17:53 +090091 // Configure the persistent Chrome OS bridge interface with static IP.
Garrick Evans8e8e3472020-01-23 14:03:50 +090092 if (process_runner_->brctl("addbr", {ifname}) != 0) {
Garrick Evans8a949dc2019-07-18 16:17:53 +090093 return false;
94 }
95
Garrick Evans6f4fa3a2020-02-10 16:15:09 +090096 if (process_runner_->ip(
97 "addr", "add",
98 {IPv4AddressToCidrString(ipv4_addr, ipv4_prefix_len), "brd",
99 IPv4AddressToString(Ipv4BroadcastAddr(ipv4_addr, ipv4_prefix_len)),
100 "dev", ifname}) != 0) {
Garrick Evans7a1a9ee2020-01-28 11:03:57 +0900101 RemoveBridge(ifname);
102 return false;
103 }
104
105 if (process_runner_->ip("link", "set", {ifname, "up"}) != 0) {
Garrick Evans8a949dc2019-07-18 16:17:53 +0900106 RemoveBridge(ifname);
107 return false;
108 }
109
110 // See nat.conf in chromeos-nat-init for the rest of the NAT setup rules.
Hugo Benichie8758b52020-04-03 14:49:01 +0900111 if (!AddOutboundIPv4SNATMark(ifname)) {
Garrick Evans8a949dc2019-07-18 16:17:53 +0900112 RemoveBridge(ifname);
113 return false;
114 }
115
116 return true;
117}
118
119void Datapath::RemoveBridge(const std::string& ifname) {
Hugo Benichie8758b52020-04-03 14:49:01 +0900120 RemoveOutboundIPv4SNATMark(ifname);
Garrick Evans7a1a9ee2020-01-28 11:03:57 +0900121 process_runner_->ip("link", "set", {ifname, "down"});
Garrick Evans8e8e3472020-01-23 14:03:50 +0900122 process_runner_->brctl("delbr", {ifname});
Garrick Evans8a949dc2019-07-18 16:17:53 +0900123}
124
Garrick Evans621ed262019-11-13 12:28:43 +0900125bool Datapath::AddToBridge(const std::string& br_ifname,
126 const std::string& ifname) {
Garrick Evans8e8e3472020-01-23 14:03:50 +0900127 return (process_runner_->brctl("addif", {br_ifname, ifname}) == 0);
Garrick Evans621ed262019-11-13 12:28:43 +0900128}
129
Garrick Evansc7ae82c2019-09-04 16:25:10 +0900130std::string Datapath::AddTAP(const std::string& name,
Garrick Evans621ed262019-11-13 12:28:43 +0900131 const MacAddress* mac_addr,
132 const SubnetAddress* ipv4_addr,
Garrick Evans4f9f5572019-11-26 10:25:16 +0900133 const std::string& user) {
Garrick Evansc7ae82c2019-09-04 16:25:10 +0900134 base::ScopedFD dev(open(kTunDev, O_RDWR | O_NONBLOCK));
135 if (!dev.is_valid()) {
136 PLOG(ERROR) << "Failed to open " << kTunDev;
137 return "";
138 }
139
140 struct ifreq ifr;
141 memset(&ifr, 0, sizeof(ifr));
142 strncpy(ifr.ifr_name, name.empty() ? kDefaultIfname : name.c_str(),
143 sizeof(ifr.ifr_name));
144 ifr.ifr_flags = IFF_TAP | IFF_NO_PI;
145
146 // If a template was given as the name, ifr_name will be updated with the
147 // actual interface name.
148 if ((*ioctl_)(dev.get(), TUNSETIFF, &ifr) != 0) {
Garrick Evans621ed262019-11-13 12:28:43 +0900149 PLOG(ERROR) << "Failed to create tap interface " << name;
Garrick Evansc7ae82c2019-09-04 16:25:10 +0900150 return "";
151 }
152 const char* ifname = ifr.ifr_name;
153
154 if ((*ioctl_)(dev.get(), TUNSETPERSIST, 1) != 0) {
Garrick Evans621ed262019-11-13 12:28:43 +0900155 PLOG(ERROR) << "Failed to persist the interface " << ifname;
Garrick Evansc7ae82c2019-09-04 16:25:10 +0900156 return "";
157 }
158
Garrick Evans4f9f5572019-11-26 10:25:16 +0900159 if (!user.empty()) {
160 uid_t uid = -1;
161 if (!brillo::userdb::GetUserInfo(user, &uid, nullptr)) {
162 PLOG(ERROR) << "Unable to look up UID for " << user;
163 RemoveTAP(ifname);
164 return "";
165 }
166 if ((*ioctl_)(dev.get(), TUNSETOWNER, uid) != 0) {
167 PLOG(ERROR) << "Failed to set owner " << uid << " of tap interface "
168 << ifname;
169 RemoveTAP(ifname);
170 return "";
171 }
Garrick Evansc7ae82c2019-09-04 16:25:10 +0900172 }
173
Hugo Benichib9b93fe2019-10-25 23:36:01 +0900174 // Create control socket for configuring the interface.
Garrick Evansc7ae82c2019-09-04 16:25:10 +0900175 base::ScopedFD sock(socket(AF_INET, SOCK_DGRAM | SOCK_CLOEXEC, 0));
176 if (!sock.is_valid()) {
177 PLOG(ERROR) << "Failed to create control socket for tap interface "
Garrick Evans621ed262019-11-13 12:28:43 +0900178 << ifname;
Garrick Evansc7ae82c2019-09-04 16:25:10 +0900179 RemoveTAP(ifname);
180 return "";
181 }
182
Garrick Evans621ed262019-11-13 12:28:43 +0900183 if (ipv4_addr) {
184 struct sockaddr_in* addr =
185 reinterpret_cast<struct sockaddr_in*>(&ifr.ifr_addr);
186 addr->sin_family = AF_INET;
187 addr->sin_addr.s_addr = static_cast<in_addr_t>(ipv4_addr->Address());
188 if ((*ioctl_)(sock.get(), SIOCSIFADDR, &ifr) != 0) {
189 PLOG(ERROR) << "Failed to set ip address for vmtap interface " << ifname
190 << " {" << ipv4_addr->ToCidrString() << "}";
191 RemoveTAP(ifname);
192 return "";
193 }
194
195 struct sockaddr_in* netmask =
196 reinterpret_cast<struct sockaddr_in*>(&ifr.ifr_netmask);
197 netmask->sin_family = AF_INET;
198 netmask->sin_addr.s_addr = static_cast<in_addr_t>(ipv4_addr->Netmask());
199 if ((*ioctl_)(sock.get(), SIOCSIFNETMASK, &ifr) != 0) {
200 PLOG(ERROR) << "Failed to set netmask for vmtap interface " << ifname
201 << " {" << ipv4_addr->ToCidrString() << "}";
202 RemoveTAP(ifname);
203 return "";
204 }
Garrick Evansc7ae82c2019-09-04 16:25:10 +0900205 }
206
Garrick Evans621ed262019-11-13 12:28:43 +0900207 if (mac_addr) {
208 struct sockaddr* hwaddr = &ifr.ifr_hwaddr;
209 hwaddr->sa_family = ARPHRD_ETHER;
210 memcpy(&hwaddr->sa_data, mac_addr, sizeof(*mac_addr));
211 if ((*ioctl_)(sock.get(), SIOCSIFHWADDR, &ifr) != 0) {
212 PLOG(ERROR) << "Failed to set mac address for vmtap interface " << ifname
213 << " {" << MacAddressToString(*mac_addr) << "}";
214 RemoveTAP(ifname);
215 return "";
216 }
Garrick Evansc7ae82c2019-09-04 16:25:10 +0900217 }
218
219 if ((*ioctl_)(sock.get(), SIOCGIFFLAGS, &ifr) != 0) {
Garrick Evans621ed262019-11-13 12:28:43 +0900220 PLOG(ERROR) << "Failed to get flags for tap interface " << ifname;
Garrick Evansc7ae82c2019-09-04 16:25:10 +0900221 RemoveTAP(ifname);
222 return "";
223 }
224
225 ifr.ifr_flags |= (IFF_UP | IFF_RUNNING);
226 if ((*ioctl_)(sock.get(), SIOCSIFFLAGS, &ifr) != 0) {
Garrick Evans621ed262019-11-13 12:28:43 +0900227 PLOG(ERROR) << "Failed to enable tap interface " << ifname;
Garrick Evansc7ae82c2019-09-04 16:25:10 +0900228 RemoveTAP(ifname);
229 return "";
230 }
231
232 return ifname;
233}
234
235void Datapath::RemoveTAP(const std::string& ifname) {
Garrick Evans8e8e3472020-01-23 14:03:50 +0900236 process_runner_->ip("tuntap", "del", {ifname, "mode", "tap"});
Garrick Evansc7ae82c2019-09-04 16:25:10 +0900237}
238
Hugo Benichi33860d72020-07-09 16:34:01 +0900239bool Datapath::ConnectVethPair(pid_t netns_pid,
240 const std::string& netns_name,
Hugo Benichi76675592020-04-08 14:29:57 +0900241 const std::string& veth_ifname,
242 const std::string& peer_ifname,
243 const MacAddress& remote_mac_addr,
244 uint32_t remote_ipv4_addr,
245 uint32_t remote_ipv4_prefix_len,
246 bool remote_multicast_flag) {
Hugo Benichi33860d72020-07-09 16:34:01 +0900247 // Set up the virtual pair across the current namespace and |netns_name|.
248 if (!AddVirtualInterfacePair(netns_name, veth_ifname, peer_ifname)) {
249 LOG(ERROR) << "Failed to create veth pair " << veth_ifname << ","
250 << peer_ifname;
251 return false;
252 }
253
254 // Configure the remote veth in namespace |netns_name|.
Hugo Benichi76675592020-04-08 14:29:57 +0900255 {
Hugo Benichi33860d72020-07-09 16:34:01 +0900256 ScopedNS ns(netns_pid);
257 if (!ns.IsValid() && netns_pid != kTestPID) {
Hugo Benichi76675592020-04-08 14:29:57 +0900258 LOG(ERROR)
259 << "Cannot create virtual link -- invalid container namespace?";
260 return false;
261 }
262
Hugo Benichi76675592020-04-08 14:29:57 +0900263 if (!ConfigureInterface(peer_ifname, remote_mac_addr, remote_ipv4_addr,
264 remote_ipv4_prefix_len, true /* link up */,
265 remote_multicast_flag)) {
266 LOG(ERROR) << "Failed to configure interface " << peer_ifname;
267 RemoveInterface(peer_ifname);
268 return false;
269 }
270 }
271
Hugo Benichi76675592020-04-08 14:29:57 +0900272 if (!ToggleInterface(veth_ifname, true /*up*/)) {
273 LOG(ERROR) << "Failed to bring up interface " << veth_ifname;
274 RemoveInterface(veth_ifname);
275 return false;
276 }
Hugo Benichi33860d72020-07-09 16:34:01 +0900277
Hugo Benichi76675592020-04-08 14:29:57 +0900278 return true;
279}
280
Hugo Benichi33860d72020-07-09 16:34:01 +0900281bool Datapath::AddVirtualInterfacePair(const std::string& netns_name,
282 const std::string& veth_ifname,
Garrick Evans2470caa2020-03-04 14:15:41 +0900283 const std::string& peer_ifname) {
Hugo Benichi33860d72020-07-09 16:34:01 +0900284 return process_runner_->ip("link", "add",
285 {veth_ifname, "type", "veth", "peer", "name",
286 peer_ifname, "netns", netns_name}) == 0;
Garrick Evans2470caa2020-03-04 14:15:41 +0900287}
Garrick Evans54861622019-07-19 09:05:09 +0900288
Garrick Evans2470caa2020-03-04 14:15:41 +0900289bool Datapath::ToggleInterface(const std::string& ifname, bool up) {
290 const std::string link = up ? "up" : "down";
291 return process_runner_->ip("link", "set", {ifname, link}) == 0;
292}
Garrick Evans54861622019-07-19 09:05:09 +0900293
Garrick Evans2470caa2020-03-04 14:15:41 +0900294bool Datapath::ConfigureInterface(const std::string& ifname,
295 const MacAddress& mac_addr,
296 uint32_t ipv4_addr,
297 uint32_t ipv4_prefix_len,
298 bool up,
299 bool enable_multicast) {
300 const std::string link = up ? "up" : "down";
301 const std::string multicast = enable_multicast ? "on" : "off";
302 return (process_runner_->ip(
303 "addr", "add",
304 {IPv4AddressToCidrString(ipv4_addr, ipv4_prefix_len), "brd",
305 IPv4AddressToString(
306 Ipv4BroadcastAddr(ipv4_addr, ipv4_prefix_len)),
307 "dev", ifname}) == 0) &&
308 (process_runner_->ip("link", "set",
309 {
310 "dev",
311 ifname,
312 link,
313 "addr",
314 MacAddressToString(mac_addr),
315 "multicast",
316 multicast,
317 }) == 0);
Garrick Evans54861622019-07-19 09:05:09 +0900318}
319
320void Datapath::RemoveInterface(const std::string& ifname) {
Garrick Evans8e8e3472020-01-23 14:03:50 +0900321 process_runner_->ip("link", "delete", {ifname}, false /*log_failures*/);
Garrick Evans54861622019-07-19 09:05:09 +0900322}
323
Hugo Benichi8d622b52020-08-13 15:24:12 +0900324void Datapath::StartRoutingDevice(const std::string& ext_ifname,
325 const std::string& int_ifname,
326 uint32_t int_ipv4_addr,
327 TrafficSource source) {
328 if (!ext_ifname.empty() &&
329 !AddInboundIPv4DNAT(ext_ifname, IPv4AddressToString(int_ipv4_addr)))
330 LOG(ERROR) << "Failed to configure ingress traffic rules for " << ext_ifname
331 << "->" << int_ifname;
332
333 // TODO(b/161508179) Explicitly enable IPv4 forwarding for this pair of
334 // interfaces for the ingress direction.
335 if (!AddOutboundIPv4(int_ifname))
336 LOG(ERROR) << "Failed to configure egress traffic rules for " << ext_ifname
337 << "<-" << int_ifname;
338
339 // TODO(b/161508179) Explicitly enable IPv4 forwarding for this pair of
340 // interfaces for the egress direction and stop relying on the traffic fwmark
341 // matching 1/1 once forwarded egress traffic is routed through the fwmark
342 // routing tag.
343
344 // TODO(b/161507671) If ext_ifname is not null, mark egress traffic with the
345 // fwmark routing tag corresponding to |ext_ifname|, and set up strong routing
346 // in ip rule.
347
348 // TODO(b/161507671) If ext_ifname is null, set up connection tracking for the
349 // current default interface.
350
351 // TODO(b/161508179) Start marking egress traffic with the corresponding
352 // source fwmark in mangle PREROUTING.
353}
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));
361 RemoveOutboundIPv4(int_ifname);
362
363 // TODO(b/161508179) Stops IPv4 forwarding for this pair of interfaces.
364 // TODO(b/161508179) Remove source marking for egress traffic.
365 // TODO(b/161507671) Remove routing tag marking for egress traffic.
366}
367
Garrick Evansf0ab7132019-06-18 14:50:42 +0900368bool Datapath::AddInboundIPv4DNAT(const std::string& ifname,
369 const std::string& ipv4_addr) {
370 // Direct ingress IP traffic to existing sockets.
Garrick Evans8e8e3472020-01-23 14:03:50 +0900371 if (process_runner_->iptables(
372 "nat", {"-A", "PREROUTING", "-i", ifname, "-m", "socket",
373 "--nowildcard", "-j", "ACCEPT", "-w"}) != 0)
Garrick Evansf0ab7132019-06-18 14:50:42 +0900374 return false;
375
376 // Direct ingress TCP & UDP traffic to ARC interface for new connections.
Garrick Evans8e8e3472020-01-23 14:03:50 +0900377 if (process_runner_->iptables(
378 "nat", {"-A", "PREROUTING", "-i", ifname, "-p", "tcp", "-j", "DNAT",
379 "--to-destination", ipv4_addr, "-w"}) != 0) {
Garrick Evansf0ab7132019-06-18 14:50:42 +0900380 RemoveInboundIPv4DNAT(ifname, ipv4_addr);
381 return false;
382 }
Garrick Evans8e8e3472020-01-23 14:03:50 +0900383 if (process_runner_->iptables(
384 "nat", {"-A", "PREROUTING", "-i", ifname, "-p", "udp", "-j", "DNAT",
385 "--to-destination", ipv4_addr, "-w"}) != 0) {
Garrick Evansf0ab7132019-06-18 14:50:42 +0900386 RemoveInboundIPv4DNAT(ifname, ipv4_addr);
387 return false;
388 }
389
390 return true;
391}
392
393void Datapath::RemoveInboundIPv4DNAT(const std::string& ifname,
394 const std::string& ipv4_addr) {
Garrick Evans8e8e3472020-01-23 14:03:50 +0900395 process_runner_->iptables(
396 "nat", {"-D", "PREROUTING", "-i", ifname, "-p", "udp", "-j", "DNAT",
397 "--to-destination", ipv4_addr, "-w"});
398 process_runner_->iptables(
399 "nat", {"-D", "PREROUTING", "-i", ifname, "-p", "tcp", "-j", "DNAT",
400 "--to-destination", ipv4_addr, "-w"});
401 process_runner_->iptables(
402 "nat", {"-D", "PREROUTING", "-i", ifname, "-m", "socket", "--nowildcard",
403 "-j", "ACCEPT", "-w"});
Garrick Evansf0ab7132019-06-18 14:50:42 +0900404}
405
Hugo Benichi8d622b52020-08-13 15:24:12 +0900406// TODO(hugobenichi) The name incorrectly refers to egress traffic, but this
407// FORWARD rule actually enables forwarding for ingress traffic. Fix the name.
Garrick Evansf0ab7132019-06-18 14:50:42 +0900408bool Datapath::AddOutboundIPv4(const std::string& ifname) {
Garrick Evans8e8e3472020-01-23 14:03:50 +0900409 return process_runner_->iptables("filter", {"-A", "FORWARD", "-o", ifname,
410 "-j", "ACCEPT", "-w"}) == 0;
Garrick Evansf0ab7132019-06-18 14:50:42 +0900411}
412
413void Datapath::RemoveOutboundIPv4(const std::string& ifname) {
Garrick Evans8e8e3472020-01-23 14:03:50 +0900414 process_runner_->iptables(
415 "filter", {"-D", "FORWARD", "-o", ifname, "-j", "ACCEPT", "-w"});
Garrick Evansf0ab7132019-06-18 14:50:42 +0900416}
417
Garrick Evansd291af62020-05-25 10:39:06 +0900418bool Datapath::AddSNATMarkRules() {
Taoyu Li79871c92020-07-02 16:09:39 +0900419 // chromium:1050579: INVALID packets cannot be tracked by conntrack therefore
420 // need to be explicitly dropped.
421 if (process_runner_->iptables(
Hugo Benichi6c445322020-08-12 16:46:19 +0900422 "filter", {"-A", "FORWARD", "-m", "mark", "--mark", "1/1", "-m",
Taoyu Li79871c92020-07-02 16:09:39 +0900423 "state", "--state", "INVALID", "-j", "DROP", "-w"}) != 0) {
424 return false;
425 }
Garrick Evansd291af62020-05-25 10:39:06 +0900426 if (process_runner_->iptables(
Hugo Benichi6c445322020-08-12 16:46:19 +0900427 "filter", {"-A", "FORWARD", "-m", "mark", "--mark", "1/1", "-j",
Garrick Evansd291af62020-05-25 10:39:06 +0900428 "ACCEPT", "-w"}) != 0) {
429 return false;
430 }
431 if (process_runner_->iptables(
Hugo Benichi6c445322020-08-12 16:46:19 +0900432 "nat", {"-A", "POSTROUTING", "-m", "mark", "--mark", "1/1", "-j",
Garrick Evansd291af62020-05-25 10:39:06 +0900433 "MASQUERADE", "-w"}) != 0) {
434 RemoveSNATMarkRules();
435 return false;
436 }
437 return true;
438}
439
440void Datapath::RemoveSNATMarkRules() {
441 process_runner_->iptables("nat", {"-D", "POSTROUTING", "-m", "mark", "--mark",
Hugo Benichi6c445322020-08-12 16:46:19 +0900442 "1/1", "-j", "MASQUERADE", "-w"});
Garrick Evansd291af62020-05-25 10:39:06 +0900443 process_runner_->iptables("filter", {"-D", "FORWARD", "-m", "mark", "--mark",
Hugo Benichi6c445322020-08-12 16:46:19 +0900444 "1/1", "-j", "ACCEPT", "-w"});
Taoyu Li79871c92020-07-02 16:09:39 +0900445 process_runner_->iptables(
Hugo Benichi6c445322020-08-12 16:46:19 +0900446 "filter", {"-D", "FORWARD", "-m", "mark", "--mark", "1/1", "-m", "state",
Taoyu Li79871c92020-07-02 16:09:39 +0900447 "--state", "INVALID", "-j", "DROP", "-w"});
Garrick Evansd291af62020-05-25 10:39:06 +0900448}
449
Garrick Evansff6e37f2020-05-25 10:54:47 +0900450bool Datapath::AddInterfaceSNAT(const std::string& ifname) {
451 return process_runner_->iptables("nat", {"-A", "POSTROUTING", "-o", ifname,
452 "-j", "MASQUERADE", "-w"}) == 0;
453}
454
455void Datapath::RemoveInterfaceSNAT(const std::string& ifname) {
456 process_runner_->iptables(
457 "nat", {"-D", "POSTROUTING", "-o", ifname, "-j", "MASQUERADE", "-w"});
458}
459
Hugo Benichie8758b52020-04-03 14:49:01 +0900460bool Datapath::AddOutboundIPv4SNATMark(const std::string& ifname) {
461 return process_runner_->iptables(
462 "mangle", {"-A", "PREROUTING", "-i", ifname, "-j", "MARK",
Hugo Benichi6c445322020-08-12 16:46:19 +0900463 "--set-mark", "1/1", "-w"}) == 0;
Hugo Benichie8758b52020-04-03 14:49:01 +0900464}
465
466void Datapath::RemoveOutboundIPv4SNATMark(const std::string& ifname) {
467 process_runner_->iptables("mangle", {"-D", "PREROUTING", "-i", ifname, "-j",
Hugo Benichi6c445322020-08-12 16:46:19 +0900468 "MARK", "--set-mark", "1/1", "-w"});
Hugo Benichie8758b52020-04-03 14:49:01 +0900469}
470
Garrick Evansd291af62020-05-25 10:39:06 +0900471bool Datapath::AddForwardEstablishedRule() {
472 return process_runner_->iptables(
473 "filter", {"-A", "FORWARD", "-m", "state", "--state",
474 "ESTABLISHED,RELATED", "-j", "ACCEPT", "-w"}) == 0;
475}
476
477void Datapath::RemoveForwardEstablishedRule() {
478 process_runner_->iptables("filter",
479 {"-D", "FORWARD", "-m", "state", "--state",
480 "ESTABLISHED,RELATED", "-j", "ACCEPT", "-w"});
481}
482
Garrick Evans664a82f2019-12-17 12:18:05 +0900483bool Datapath::MaskInterfaceFlags(const std::string& ifname,
484 uint16_t on,
485 uint16_t off) {
Taoyu Li90c13912019-11-26 17:56:54 +0900486 base::ScopedFD sock(socket(AF_INET, SOCK_DGRAM | SOCK_CLOEXEC, 0));
487 if (!sock.is_valid()) {
488 PLOG(ERROR) << "Failed to create control socket";
489 return false;
490 }
491 ifreq ifr;
492 snprintf(ifr.ifr_name, IFNAMSIZ, "%s", ifname.c_str());
493 if ((*ioctl_)(sock.get(), SIOCGIFFLAGS, &ifr) < 0) {
494 PLOG(WARNING) << "ioctl() failed to get interface flag on " << ifname;
495 return false;
496 }
Garrick Evans664a82f2019-12-17 12:18:05 +0900497 ifr.ifr_flags |= on;
498 ifr.ifr_flags &= ~off;
Taoyu Li90c13912019-11-26 17:56:54 +0900499 if ((*ioctl_)(sock.get(), SIOCSIFFLAGS, &ifr) < 0) {
Garrick Evans664a82f2019-12-17 12:18:05 +0900500 PLOG(WARNING) << "ioctl() failed to set flag 0x" << std::hex << on
501 << " unset flag 0x" << std::hex << off << " on " << ifname;
Taoyu Li90c13912019-11-26 17:56:54 +0900502 return false;
503 }
504 return true;
505}
506
Garrick Evans260ff302019-07-25 11:22:50 +0900507bool Datapath::AddIPv6HostRoute(const std::string& ifname,
508 const std::string& ipv6_addr,
509 int ipv6_prefix_len) {
510 std::string ipv6_addr_cidr =
511 ipv6_addr + "/" + std::to_string(ipv6_prefix_len);
512
Garrick Evans8e8e3472020-01-23 14:03:50 +0900513 return process_runner_->ip6("route", "replace",
514 {ipv6_addr_cidr, "dev", ifname}) == 0;
Garrick Evans260ff302019-07-25 11:22:50 +0900515}
516
517void Datapath::RemoveIPv6HostRoute(const std::string& ifname,
518 const std::string& ipv6_addr,
519 int ipv6_prefix_len) {
520 std::string ipv6_addr_cidr =
521 ipv6_addr + "/" + std::to_string(ipv6_prefix_len);
522
Garrick Evans8e8e3472020-01-23 14:03:50 +0900523 process_runner_->ip6("route", "del", {ipv6_addr_cidr, "dev", ifname});
Garrick Evans260ff302019-07-25 11:22:50 +0900524}
525
526bool Datapath::AddIPv6Neighbor(const std::string& ifname,
527 const std::string& ipv6_addr) {
Garrick Evans8e8e3472020-01-23 14:03:50 +0900528 return process_runner_->ip6("neigh", "add",
529 {"proxy", ipv6_addr, "dev", ifname}) == 0;
Garrick Evans260ff302019-07-25 11:22:50 +0900530}
531
532void Datapath::RemoveIPv6Neighbor(const std::string& ifname,
533 const std::string& ipv6_addr) {
Garrick Evans8e8e3472020-01-23 14:03:50 +0900534 process_runner_->ip6("neigh", "del", {"proxy", ipv6_addr, "dev", ifname});
Garrick Evans260ff302019-07-25 11:22:50 +0900535}
536
537bool Datapath::AddIPv6Forwarding(const std::string& ifname1,
538 const std::string& ifname2) {
Garrick Evans8e8e3472020-01-23 14:03:50 +0900539 if (process_runner_->ip6tables(
540 "filter",
541 {"-C", "FORWARD", "-i", ifname1, "-o", ifname2, "-j", "ACCEPT", "-w"},
542 false /*log_failures*/) != 0 &&
543 process_runner_->ip6tables(
544 "filter", {"-A", "FORWARD", "-i", ifname1, "-o", ifname2, "-j",
545 "ACCEPT", "-w"}) != 0) {
Garrick Evans260ff302019-07-25 11:22:50 +0900546 return false;
547 }
548
Garrick Evans8e8e3472020-01-23 14:03:50 +0900549 if (process_runner_->ip6tables(
550 "filter",
551 {"-C", "FORWARD", "-i", ifname2, "-o", ifname1, "-j", "ACCEPT", "-w"},
552 false /*log_failures*/) != 0 &&
553 process_runner_->ip6tables(
554 "filter", {"-A", "FORWARD", "-i", ifname2, "-o", ifname1, "-j",
555 "ACCEPT", "-w"}) != 0) {
Garrick Evans260ff302019-07-25 11:22:50 +0900556 RemoveIPv6Forwarding(ifname1, ifname2);
557 return false;
558 }
559
560 return true;
561}
562
563void Datapath::RemoveIPv6Forwarding(const std::string& ifname1,
564 const std::string& ifname2) {
Garrick Evans8e8e3472020-01-23 14:03:50 +0900565 process_runner_->ip6tables("filter", {"-D", "FORWARD", "-i", ifname1, "-o",
566 ifname2, "-j", "ACCEPT", "-w"});
Garrick Evans260ff302019-07-25 11:22:50 +0900567
Garrick Evans8e8e3472020-01-23 14:03:50 +0900568 process_runner_->ip6tables("filter", {"-D", "FORWARD", "-i", ifname2, "-o",
569 ifname1, "-j", "ACCEPT", "-w"});
Garrick Evans260ff302019-07-25 11:22:50 +0900570}
571
Garrick Evans3d97a392020-02-21 15:24:37 +0900572bool Datapath::AddIPv4Route(uint32_t gateway_addr,
573 uint32_t addr,
574 uint32_t netmask) {
575 struct rtentry route;
576 memset(&route, 0, sizeof(route));
Hugo Benichie8758b52020-04-03 14:49:01 +0900577 SetSockaddrIn(&route.rt_gateway, gateway_addr);
578 SetSockaddrIn(&route.rt_dst, addr & netmask);
579 SetSockaddrIn(&route.rt_genmask, netmask);
Garrick Evans3d97a392020-02-21 15:24:37 +0900580 route.rt_flags = RTF_UP | RTF_GATEWAY;
Hugo Benichie8758b52020-04-03 14:49:01 +0900581 return ModifyRtentry(SIOCADDRT, &route);
582}
Garrick Evans3d97a392020-02-21 15:24:37 +0900583
Hugo Benichie8758b52020-04-03 14:49:01 +0900584bool Datapath::DeleteIPv4Route(uint32_t gateway_addr,
585 uint32_t addr,
586 uint32_t netmask) {
587 struct rtentry route;
588 memset(&route, 0, sizeof(route));
589 SetSockaddrIn(&route.rt_gateway, gateway_addr);
590 SetSockaddrIn(&route.rt_dst, addr & netmask);
591 SetSockaddrIn(&route.rt_genmask, netmask);
592 route.rt_flags = RTF_UP | RTF_GATEWAY;
593 return ModifyRtentry(SIOCDELRT, &route);
594}
595
596bool Datapath::AddIPv4Route(const std::string& ifname,
597 uint32_t addr,
598 uint32_t netmask) {
599 struct rtentry route;
600 memset(&route, 0, sizeof(route));
601 SetSockaddrIn(&route.rt_dst, addr & netmask);
602 SetSockaddrIn(&route.rt_genmask, netmask);
603 char rt_dev[IFNAMSIZ];
604 strncpy(rt_dev, ifname.c_str(), IFNAMSIZ);
605 rt_dev[IFNAMSIZ - 1] = '\0';
606 route.rt_dev = rt_dev;
607 route.rt_flags = RTF_UP | RTF_GATEWAY;
608 return ModifyRtentry(SIOCADDRT, &route);
609}
610
611bool Datapath::DeleteIPv4Route(const std::string& ifname,
612 uint32_t addr,
613 uint32_t netmask) {
614 struct rtentry route;
615 memset(&route, 0, sizeof(route));
616 SetSockaddrIn(&route.rt_dst, addr & netmask);
617 SetSockaddrIn(&route.rt_genmask, netmask);
618 char rt_dev[IFNAMSIZ];
619 strncpy(rt_dev, ifname.c_str(), IFNAMSIZ);
620 rt_dev[IFNAMSIZ - 1] = '\0';
621 route.rt_dev = rt_dev;
622 route.rt_flags = RTF_UP | RTF_GATEWAY;
623 return ModifyRtentry(SIOCDELRT, &route);
624}
625
626bool Datapath::ModifyRtentry(unsigned long op, struct rtentry* route) {
627 DCHECK(route);
628 if (op != SIOCADDRT && op != SIOCDELRT) {
Andreea Costinas34aa7a92020-08-04 10:36:10 +0200629 LOG(ERROR) << "Invalid operation " << op << " for rtentry " << *route;
Garrick Evans3d97a392020-02-21 15:24:37 +0900630 return false;
631 }
Hugo Benichie8758b52020-04-03 14:49:01 +0900632 base::ScopedFD fd(socket(AF_INET, SOCK_DGRAM | SOCK_CLOEXEC, 0));
633 if (!fd.is_valid()) {
Andreea Costinas34aa7a92020-08-04 10:36:10 +0200634 PLOG(ERROR) << "Failed to create socket for adding rtentry " << *route;
Hugo Benichie8758b52020-04-03 14:49:01 +0900635 return false;
636 }
637 if (HANDLE_EINTR(ioctl_(fd.get(), op, route)) != 0) {
638 std::string opname = op == SIOCADDRT ? "add" : "delete";
Andreea Costinas34aa7a92020-08-04 10:36:10 +0200639 PLOG(ERROR) << "Failed to " << opname << " rtentry " << *route;
Garrick Evans3d97a392020-02-21 15:24:37 +0900640 return false;
641 }
642 return true;
643}
644
Jason Jeremy Imana7273a32020-08-04 11:25:31 +0900645bool Datapath::AddAdbPortForwardRule(const std::string& ifname) {
646 return firewall_->AddIpv4ForwardRule(patchpanel::ModifyPortRuleRequest::TCP,
647 kArcAddr, kAdbServerPort, ifname,
648 kLocalhostAddr, kAdbProxyTcpListenPort);
649}
650
651void Datapath::DeleteAdbPortForwardRule(const std::string& ifname) {
652 firewall_->DeleteIpv4ForwardRule(patchpanel::ModifyPortRuleRequest::TCP,
653 kArcAddr, kAdbServerPort, ifname,
654 kLocalhostAddr, kAdbProxyTcpListenPort);
655}
656
657bool Datapath::AddAdbPortAccessRule(const std::string& ifname) {
658 return firewall_->AddAcceptRules(patchpanel::ModifyPortRuleRequest::TCP,
659 kAdbProxyTcpListenPort, ifname);
660}
661
662void Datapath::DeleteAdbPortAccessRule(const std::string& ifname) {
663 firewall_->DeleteAcceptRules(patchpanel::ModifyPortRuleRequest::TCP,
664 kAdbProxyTcpListenPort, ifname);
665}
666
Garrick Evans3388a032020-03-24 11:25:55 +0900667} // namespace patchpanel