|
22 | 22 | #include "absl/log/log.h" |
23 | 23 | #include "absl/strings/str_format.h" |
24 | 24 |
|
| 25 | +#include "cuttlefish/result/result.h" |
| 26 | + |
25 | 27 | namespace cuttlefish { |
26 | 28 |
|
27 | 29 | extern int RunExternalCommand(const std::string& name); |
28 | 30 |
|
29 | | -bool AddTapIface(std::string_view name) { |
| 31 | +Result<void> AddTapIface(std::string_view name) { |
30 | 32 | std::stringstream ss; |
31 | 33 | ss << "ip tuntap add dev " << name << " mode tap group cvdnetwork vnet_hdr"; |
32 | 34 | auto add_command = ss.str(); |
33 | 35 | LOG(INFO) << "Create tap interface: " << add_command; |
34 | 36 | int status = RunExternalCommand(add_command); |
35 | | - return status == 0; |
| 37 | + CF_EXPECT_EQ(status, 0); |
| 38 | + return {}; |
36 | 39 | } |
37 | 40 |
|
38 | | -bool ShutdownIface(std::string_view name) { |
| 41 | +Result<void> ShutdownIface(std::string_view name) { |
39 | 42 | std::stringstream ss; |
40 | 43 | ss << "ip link set dev " << name << " down"; |
41 | 44 | auto link_command = ss.str(); |
42 | 45 | LOG(INFO) << "Shutdown tap interface: " << link_command; |
43 | 46 | int status = RunExternalCommand(link_command); |
44 | | - |
45 | | - return status == 0; |
| 47 | + CF_EXPECT_EQ(status, 0); |
| 48 | + return {}; |
46 | 49 | } |
47 | 50 |
|
48 | | -bool BringUpIface(std::string_view name) { |
| 51 | +Result<void> BringUpIface(std::string_view name) { |
49 | 52 | std::stringstream ss; |
50 | 53 | ss << "ip link set dev " << name << " up"; |
51 | 54 | auto link_command = ss.str(); |
52 | 55 | LOG(INFO) << "Bring up tap interface: " << link_command; |
53 | 56 | int status = RunExternalCommand(link_command); |
54 | | - |
55 | | - return status == 0; |
| 57 | + CF_EXPECT_EQ(status, 0); |
| 58 | + return {}; |
56 | 59 | } |
57 | 60 |
|
58 | | -bool AddGateway(std::string_view name, std::string_view gateway, |
59 | | - std::string_view netmask) { |
| 61 | +Result<void> AddGateway(std::string_view name, std::string_view gateway, |
| 62 | + std::string_view netmask) { |
60 | 63 | std::stringstream ss; |
61 | 64 | ss << "ip addr add " << gateway << netmask << " broadcast + dev " << name; |
62 | 65 | auto command = ss.str(); |
63 | 66 | LOG(INFO) << "setup gateway: " << command; |
64 | 67 | int status = RunExternalCommand(command); |
65 | | - |
66 | | - return status == 0; |
| 68 | + CF_EXPECT_EQ(status, 0); |
| 69 | + return {}; |
67 | 70 | } |
68 | 71 |
|
69 | | -bool DestroyGateway(std::string_view name, std::string_view gateway, |
70 | | - std::string_view netmask) { |
| 72 | +Result<void> DestroyGateway(std::string_view name, std::string_view gateway, |
| 73 | + std::string_view netmask) { |
71 | 74 | std::stringstream ss; |
72 | 75 | ss << "ip addr del " << gateway << netmask << " broadcast + dev " << name; |
73 | 76 | auto command = ss.str(); |
74 | 77 | LOG(INFO) << "removing gateway: " << command; |
75 | 78 | int status = RunExternalCommand(command); |
76 | | - |
77 | | - return status == 0; |
| 79 | + CF_EXPECT_EQ(status, 0); |
| 80 | + return {}; |
78 | 81 | } |
79 | 82 |
|
80 | | -bool LinkTapToBridge(std::string_view tap_name, |
81 | | - std::string_view bridge_name) { |
| 83 | +Result<void> LinkTapToBridge(std::string_view tap_name, |
| 84 | + std::string_view bridge_name) { |
82 | 85 | std::stringstream ss; |
83 | 86 | ss << "ip link set dev " << tap_name << " master " << bridge_name; |
84 | 87 | auto command = ss.str(); |
85 | 88 | int status = RunExternalCommand(command); |
86 | | - |
87 | | - return status == 0; |
| 89 | + CF_EXPECT_EQ(status, 0); |
| 90 | + return {}; |
88 | 91 | } |
89 | 92 |
|
90 | | -bool DeleteIface(std::string_view name) { |
| 93 | +Result<void> DeleteIface(std::string_view name) { |
91 | 94 | std::stringstream ss; |
92 | 95 | ss << "ip link delete " << name; |
93 | 96 | auto link_command = ss.str(); |
94 | 97 | LOG(INFO) << "Delete tap interface: " << link_command; |
95 | 98 | int status = RunExternalCommand(link_command); |
96 | | - |
97 | | - return status == 0; |
| 99 | + CF_EXPECT_EQ(status, 0); |
| 100 | + return {}; |
98 | 101 | } |
99 | 102 |
|
100 | | -bool BridgeExists(std::string_view name) { |
| 103 | +Result<bool> BridgeExists(std::string_view name) { |
101 | 104 | std::stringstream ss; |
102 | 105 | ss << "ip link show " << name << " >/dev/null"; |
103 | 106 |
|
104 | 107 | auto command = ss.str(); |
105 | 108 | LOG(INFO) << "bridge exists: " << command; |
106 | 109 | int status = RunExternalCommand(command); |
107 | | - |
108 | 110 | return status == 0; |
109 | 111 | } |
110 | 112 |
|
111 | | -bool CreateBridge(std::string_view name) { |
| 113 | +Result<void> CreateBridge(std::string_view name) { |
112 | 114 | std::stringstream ss; |
113 | 115 | ss << "ip link add name " << name |
114 | 116 | << " type bridge forward_delay 0 stp_state 0"; |
115 | 117 |
|
116 | 118 | auto command = ss.str(); |
117 | 119 | LOG(INFO) << "create bridge: " << command; |
118 | 120 | int status = RunExternalCommand(command); |
119 | | - |
120 | | - if (status != 0) { |
121 | | - return false; |
122 | | - } |
123 | | - |
124 | | - return BringUpIface(name); |
| 121 | + CF_EXPECT_EQ(status, 0); |
| 122 | + CF_EXPECT(BringUpIface(name)); |
| 123 | + return {}; |
125 | 124 | } |
126 | 125 |
|
127 | | -bool IptableConfig(std::string_view network, bool add) { |
| 126 | +Result<void> IptableConfig(std::string_view network, bool add) { |
128 | 127 | std::stringstream ss; |
129 | 128 | ss << "iptables -t nat " << (add ? "-A" : "-D") << " POSTROUTING -s " |
130 | 129 | << network << " -j MASQUERADE"; |
131 | 130 |
|
132 | 131 | auto command = ss.str(); |
133 | 132 | LOG(INFO) << "iptable_config: " << command; |
134 | 133 | int status = RunExternalCommand(command); |
135 | | - |
136 | | - return status == 0; |
| 134 | + CF_EXPECT_EQ(status, 0); |
| 135 | + return {}; |
137 | 136 | } |
138 | 137 |
|
139 | 138 | } // namespace cuttlefish |
0 commit comments