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.h - utility class for tests that use iptables
17  */
18 
19 #include <deque>
20 
21 #include "NetdConstants.h"
22 
23 class IptablesBaseTest : public ::testing::Test {
24 public:
25     IptablesBaseTest();
26 
27     typedef std::vector<std::pair<IptablesTarget, std::string>> ExpectedIptablesCommands;
28 
29     static int fake_android_fork_exec(int argc, char* argv[], int *status, bool, bool);
30     static int fake_android_fork_execvp(int argc, char* argv[], int *status, bool, bool);
31     static int fakeExecIptablesRestore(IptablesTarget target, const std::string& commands);
32     static int fakeExecIptablesRestoreWithOutput(IptablesTarget target, const std::string& commands,
33                                                  std::string *output);
34     static int fakeExecIptablesRestoreCommand(IptablesTarget target, const std::string& table,
35                                               const std::string& commands, std::string *output);
36     static FILE *fake_popen(const char *cmd, const char *type);
37     void expectIptablesRestoreCommands(const std::vector<std::string>& expectedCmds);
38     void expectIptablesRestoreCommands(const ExpectedIptablesCommands& expectedCmds);
39 
setReturnValues(const std::deque<int> & returnValues)40     static void setReturnValues(const std::deque<int>& returnValues) {
41         sReturnValues = returnValues;
42     }
43 
addIptablesRestoreOutput(std::string contents)44     static void addIptablesRestoreOutput(std::string contents) {
45         sIptablesRestoreOutput.push_back(contents);
46     }
47 
addIptablesRestoreOutput(std::string contents1,std::string contents2)48     static void addIptablesRestoreOutput(std::string contents1, std::string contents2) {
49         addIptablesRestoreOutput(contents1);
50         addIptablesRestoreOutput(contents2);
51     }
52 
clearIptablesRestoreOutput()53     static void clearIptablesRestoreOutput() {
54         sIptablesRestoreOutput.clear();
55     }
56 
57 protected:
58     static std::vector<std::string> sCmds;
59     static ExpectedIptablesCommands sRestoreCmds;
60     static std::deque<int> sReturnValues;
61     static std::deque<std::string> sPopenContents;
62     static std::deque<std::string> sIptablesRestoreOutput;
63 };
64