1 /*
2 * Copyright (c) 2024 Huawei Device Co., Ltd.
3 * Licensed under the Apache License, Version 2.0 (the "License");
4 * you may not use this file except in compliance with the License.
5 * You may obtain a copy of the License at
6 *
7 * http://www.apache.org/licenses/LICENSE-2.0
8 *
9 * Unless required by applicable law or agreed to in writing, software
10 * distributed under the License is distributed on an "AS IS" BASIS,
11 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12 * See the License for the specific language governing permissions and
13 * limitations under the License.
14 */
15
16 #include <arpa/inet.h>
17 #include <gtest/gtest.h>
18
19 #include "bpf_netfirewall.h"
20
21 namespace {
22 using namespace testing::ext;
23 using namespace OHOS::NetManagerStandard;
24 constexpr int32_t USER_ID1 = 100;
25 }
26
27 class NetsysBpfNetFirewallTest : public testing::Test {
28 public:
29 static void SetUpTestCase();
30 static void TearDownTestCase();
31 void SetUp();
32 void TearDown();
33 };
34
SetUpTestCase()35 void NetsysBpfNetFirewallTest::SetUpTestCase() {}
36
TearDownTestCase()37 void NetsysBpfNetFirewallTest::TearDownTestCase() {}
38
SetUp()39 void NetsysBpfNetFirewallTest::SetUp() {}
40
TearDown()41 void NetsysBpfNetFirewallTest::TearDown() {}
42
43 HWTEST_F(NetsysBpfNetFirewallTest, AddDomainCache001, TestSize.Level0)
44 {
45 std::shared_ptr<NetsysBpfNetFirewall> bpfNet = NetsysBpfNetFirewall::GetInstance();
46 NetAddrInfo netInfo;
47 netInfo.aiFamily = AF_INET;
48 inet_pton(AF_INET, "192.168.8.116", &netInfo.aiAddr.sin);
49 bpfNet->AddDomainCache(netInfo);
50 EXPECT_EQ(netInfo.aiFamily, AF_INET);
51 netInfo.aiFamily = AF_INET6;
52 inet_pton(AF_INET6, "fe80::6bec:e9b9:a1df:f69d", &netInfo.aiAddr.sin6);
53 bpfNet->AddDomainCache(netInfo);
54 bpfNet->ClearDomainCache();
55 EXPECT_EQ(netInfo.aiFamily, AF_INET6);
56 }
57
58 HWTEST_F(NetsysBpfNetFirewallTest, ClearFirewallDefaultAction001, TestSize.Level0)
59 {
60 std::shared_ptr<NetsysBpfNetFirewall> bpfNet = NetsysBpfNetFirewall::GetInstance();
61 bpfNet->SetBpfLoaded(true);
62 int ret = bpfNet->SetFirewallDefaultAction(USER_ID1, FirewallRuleAction::RULE_ALLOW,
63 FirewallRuleAction::RULE_ALLOW);
64 bpfNet->ClearFirewallDefaultAction();
65 EXPECT_EQ(ret, FIREWALL_SUCCESS);
66 bpfNet->SetBpfLoaded(false);
67 ret = bpfNet->SetFirewallDefaultAction(USER_ID1, FirewallRuleAction::RULE_ALLOW,
68 FirewallRuleAction::RULE_ALLOW);
69 EXPECT_EQ(ret, NETFIREWALL_ERR);
70 }
71
72 HWTEST_F(NetsysBpfNetFirewallTest, ClearFirewallRules001, TestSize.Level0)
73 {
74 std::shared_ptr<NetsysBpfNetFirewall> bpfNet = NetsysBpfNetFirewall::GetInstance();
75 int ret = bpfNet->ClearFirewallRules(NetFirewallRuleType::RULE_ALL);
76 EXPECT_EQ(ret, FIREWALL_SUCCESS);
77 ret = FIREWALL_ERR_INTERNAL;
78 ret = bpfNet->ClearFirewallRules(NetFirewallRuleType::RULE_IP);
79 EXPECT_EQ(ret, FIREWALL_SUCCESS);
80 ret = FIREWALL_ERR_INTERNAL;
81 ret = bpfNet->ClearFirewallRules(NetFirewallRuleType::RULE_DOMAIN);
82 EXPECT_EQ(ret, FIREWALL_SUCCESS);
83 ret = FIREWALL_ERR_INTERNAL;
84 ret = bpfNet->ClearFirewallRules(NetFirewallRuleType::RULE_DEFAULT_ACTION);
85 EXPECT_EQ(ret, FIREWALL_SUCCESS);
86 }