| /* |
| * Copyright (C) 2022 The Android Open Source Project |
| * |
| * Licensed under the Apache License, Version 2.0 (the "License"); |
| * you may not use this file except in compliance with the License. |
| * You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| * |
| */ |
| |
| #include <gtest/gtest.h> |
| #include <vintf/VintfObject.h> |
| |
| #include <fstream> |
| #include <string> |
| |
| namespace android { |
| namespace net { |
| |
| namespace { |
| |
| using ::android::vintf::RuntimeInfo; |
| using ::android::vintf::VintfObject; |
| |
| class KernelConfigVerifier final { |
| public: |
| KernelConfigVerifier() : mRuntimeInfo(VintfObject::GetRuntimeInfo()) {} |
| |
| bool hasOption(const std::string& option) const { |
| const auto& configMap = mRuntimeInfo->kernelConfigs(); |
| auto it = configMap.find(option); |
| if (it != configMap.cend()) { |
| return it->second == "y"; |
| } |
| return false; |
| } |
| |
| private: |
| std::shared_ptr<const RuntimeInfo> mRuntimeInfo; |
| }; |
| |
| bool isGsiImage() { |
| std::ifstream ifs("/system/system_ext/etc/init/init.gsi.rc"); |
| return ifs.good(); |
| } |
| |
| } // namespace |
| |
| /** |
| * If this test fails, enable the following kernel modules in your kernel config: |
| * CONFIG_NET_CLS_MATCHALL=y |
| * CONFIG_NET_ACT_POLICE=y |
| * CONFIG_NET_ACT_BPF=y |
| */ |
| TEST(KernelTest, TestRateLimitingSupport) { |
| if (isGsiImage()) { |
| // skip test on gsi images |
| GTEST_SKIP() << "GSI Image"; |
| } |
| KernelConfigVerifier configVerifier; |
| ASSERT_TRUE(configVerifier.hasOption("CONFIG_NET_CLS_MATCHALL")); |
| ASSERT_TRUE(configVerifier.hasOption("CONFIG_NET_ACT_POLICE")); |
| ASSERT_TRUE(configVerifier.hasOption("CONFIG_NET_ACT_BPF")); |
| } |
| |
| } // namespace net |
| } // namespace android |