2 * Copyright 2016 The Android Open Source Project
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
8 * http://www.apache.org/licenses/LICENSE-2.0
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.
16 * IptablesBaseTest.cpp - utility class for tests that use iptables
22 #include <gtest/gtest.h>
24 #include "IptablesBaseTest.h"
25 #include "NetdConstants.h"
27 IptablesBaseTest::IptablesBaseTest() {
32 int IptablesBaseTest::fake_android_fork_exec(int argc, char* argv[], int *status, bool, bool) {
33 std::string cmd = argv[0];
34 for (int i = 1; i < argc; i++) {
43 int IptablesBaseTest::fakeExecIptables(IptablesTarget target, ...) {
44 std::string cmd = " -w";
46 va_start(args, target);
49 arg = va_arg(args, const char *);
56 if (target == V4 || target == V4V6) {
57 sCmds.push_back(IPTABLES_PATH + cmd);
59 if (target == V6 || target == V4V6) {
60 sCmds.push_back(IP6TABLES_PATH + cmd);
66 int IptablesBaseTest::fakeExecIptablesRestore(IptablesTarget target, const std::string& commands) {
67 EXPECT_EQ(V4V6, target);
68 sRestoreCmds.push_back(commands);
72 int IptablesBaseTest::expectIptablesCommand(IptablesTarget target, int pos,
73 const std::string& cmd) {
74 if (target == V4 || target == V4V6) {
75 EXPECT_EQ("/system/bin/iptables -w " + cmd, sCmds[pos++]);
77 if (target == V6 || target == V4V6) {
78 EXPECT_EQ("/system/bin/ip6tables -w " + cmd, sCmds[pos++]);
80 return target == V4V6 ? 2 : 1;
83 void IptablesBaseTest::expectIptablesCommands(const std::vector<std::string>& expectedCmds) {
84 ExpectedIptablesCommands expected;
85 for (auto cmd : expectedCmds) {
86 expected.push_back({ V4V6, cmd });
88 expectIptablesCommands(expected);
91 void IptablesBaseTest::expectIptablesCommands(const ExpectedIptablesCommands& expectedCmds) {
93 for (size_t i = 0; i < expectedCmds.size(); i ++) {
94 auto target = expectedCmds[i].first;
95 auto cmd = expectedCmds[i].second;
96 pos += expectIptablesCommand(target, pos, cmd);
99 EXPECT_EQ(pos, sCmds.size());
103 void IptablesBaseTest::expectIptablesRestoreCommands(const std::vector<std::string>& expectedCmds) {
104 EXPECT_EQ(expectedCmds.size(), sRestoreCmds.size());
105 EXPECT_EQ(expectedCmds, sRestoreCmds);
106 sRestoreCmds.clear();
109 std::vector<std::string> IptablesBaseTest::sCmds = {};
110 std::vector<std::string> IptablesBaseTest::sRestoreCmds = {};