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; 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 116 std::vector<std::string> IptablesBaseTest::sCmds = {}; 117 IptablesBaseTest::ExpectedIptablesCommands IptablesBaseTest::sRestoreCmds = {}; 118 std::deque<std::string> IptablesBaseTest::sPopenContents = {}; 119 std::deque<std::string> IptablesBaseTest::sIptablesRestoreOutput = {}; 120 std::deque<int> IptablesBaseTest::sReturnValues = {}; 121