1 /*
2 * Copyright 2016 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 * IptablesBaseTest.cpp - utility class for tests that use iptables
17 */
18
19 #include <deque>
20 #include <string>
21 #include <vector>
22
23 #include <gtest/gtest.h>
24
25 #include <android-base/stringprintf.h>
26
27 #include "IptablesBaseTest.h"
28 #include "NetdConstants.h"
29
30 #define LOG_TAG "IptablesBaseTest"
31 #include <cutils/log.h>
32
33 using android::base::StringPrintf;
34
IptablesBaseTest()35 IptablesBaseTest::IptablesBaseTest() {
36 sCmds.clear();
37 sRestoreCmds.clear();
38 sReturnValues.clear();
39 }
40
fake_android_fork_exec(int argc,char * argv[],int * status,bool,bool)41 int IptablesBaseTest::fake_android_fork_exec(int argc, char* argv[], int *status, bool, bool) {
42 std::string cmd = argv[0];
43 for (int i = 1; i < argc; i++) {
44 if (argv[i] == NULL) break; // NatController likes to pass in invalid argc values.
45 cmd += " ";
46 cmd += argv[i];
47 }
48 sCmds.push_back(cmd);
49
50 int ret;
51 if (sReturnValues.size()) {
52 ret = sReturnValues.front();
53 sReturnValues.pop_front();
54 } else {
55 ret = 0;
56 }
57
58 if (status) {
59 *status = ret;
60 }
61 return ret;
62 }
63
fake_popen(const char *,const char * type)64 FILE *IptablesBaseTest::fake_popen(const char * /* cmd */, const char *type) {
65 if (sPopenContents.empty() || strcmp(type, "r") != 0) {
66 return NULL;
67 }
68
69 std::string realCmd = StringPrintf("echo '%s'", sPopenContents.front().c_str());
70 sPopenContents.pop_front();
71 return popen(realCmd.c_str(), "r");
72 }
73
fakeExecIptablesRestoreWithOutput(IptablesTarget target,const std::string & commands,std::string * output)74 int IptablesBaseTest::fakeExecIptablesRestoreWithOutput(IptablesTarget target,
75 const std::string& commands,
76 std::string *output) {
77 sRestoreCmds.push_back({ target, commands });
78 if (output != nullptr) {
79 *output = sIptablesRestoreOutput.size() ? sIptablesRestoreOutput.front().c_str() : "";
80 }
81 if (sIptablesRestoreOutput.size()) {
82 sIptablesRestoreOutput.pop_front();
83 }
84 return 0;
85 }
86
fakeExecIptablesRestore(IptablesTarget target,const std::string & commands)87 int IptablesBaseTest::fakeExecIptablesRestore(IptablesTarget target, const std::string& commands) {
88 return fakeExecIptablesRestoreWithOutput(target, commands, nullptr);
89 }
90
fakeExecIptablesRestoreCommand(IptablesTarget target,const std::string & table,const std::string & command,std::string * output)91 int IptablesBaseTest::fakeExecIptablesRestoreCommand(IptablesTarget target,
92 const std::string& table,
93 const std::string& command,
94 std::string *output) {
95 std::string fullCmd = StringPrintf("-t %s %s", table.c_str(), command.c_str());
96 return fakeExecIptablesRestoreWithOutput(target, fullCmd, output);
97 }
98
expectIptablesRestoreCommands(const std::vector<std::string> & expectedCmds)99 void IptablesBaseTest::expectIptablesRestoreCommands(const std::vector<std::string>& expectedCmds) {
100 ExpectedIptablesCommands expected;
101 for (const auto& cmd : expectedCmds) {
102 expected.push_back({ V4V6, cmd });
103 }
104 expectIptablesRestoreCommands(expected);
105 }
106
expectIptablesRestoreCommands(const ExpectedIptablesCommands & expectedCmds)107 void IptablesBaseTest::expectIptablesRestoreCommands(const ExpectedIptablesCommands& expectedCmds) {
108 EXPECT_EQ(expectedCmds.size(), sRestoreCmds.size());
109 for (size_t i = 0; i < expectedCmds.size(); i++) {
110 EXPECT_EQ(expectedCmds[i], sRestoreCmds[i]) <<
111 "iptables-restore command " << i << " differs";
112 }
113 sRestoreCmds.clear();
114 }
115
setReturnValues(const std::deque<int> & returnValues)116 void IptablesBaseTest::setReturnValues(const std::deque<int>& returnValues) {
117 sReturnValues = returnValues;
118 }
119
120 std::vector<std::string> IptablesBaseTest::sCmds = {};
121 IptablesBaseTest::ExpectedIptablesCommands IptablesBaseTest::sRestoreCmds = {};
122 std::deque<std::string> IptablesBaseTest::sPopenContents = {};
123 std::deque<std::string> IptablesBaseTest::sIptablesRestoreOutput = {};
124 std::deque<int> IptablesBaseTest::sReturnValues = {};
125