1 /* 2 * Copyright (c) 2021 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 #ifndef NET_POLICY_SERVICE_H 17 #define NET_POLICY_SERVICE_H 18 19 #include <mutex> 20 21 #include "singleton.h" 22 #include "system_ability.h" 23 #include "system_ability_definition.h" 24 25 #include "net_policy_callback.h" 26 #include "net_policy_traffic.h" 27 #include "net_policy_firewall.h" 28 29 #include "net_policy_service_stub.h" 30 #include "net_policy_service_common.h" 31 32 namespace OHOS { 33 namespace NetManagerStandard { 34 class NetPolicyService : public SystemAbility, 35 public NetPolicyServiceStub, 36 public std::enable_shared_from_this<NetPolicyService> { 37 DECLARE_DELAYED_SINGLETON(NetPolicyService) 38 DECLARE_SYSTEM_ABILITY(NetPolicyService) 39 40 public: 41 void OnStart() override; 42 void OnStop() override; 43 44 NetPolicyResultCode SetPolicyByUid(uint32_t uid, NetUidPolicy policy) override; 45 NetUidPolicy GetPolicyByUid(uint32_t uid) override; 46 std::vector<uint32_t> GetUidsByPolicy(NetUidPolicy policy) override; 47 bool IsUidNetAccess(uint32_t uid, bool metered) override; 48 bool IsUidNetAccess(uint32_t uid, const std::string &ifaceName) override; 49 int32_t RegisterNetPolicyCallback(const sptr<INetPolicyCallback> &callback) override; 50 int32_t UnregisterNetPolicyCallback(const sptr<INetPolicyCallback> &callback) override; 51 NetPolicyResultCode SetNetQuotaPolicies(const std::vector<NetPolicyQuotaPolicy> "aPolicies) override; 52 NetPolicyResultCode GetNetQuotaPolicies(std::vector<NetPolicyQuotaPolicy> "aPolicies) override; 53 NetPolicyResultCode SetCellularPolicies(const std::vector<NetPolicyCellularPolicy> &cellularPolicies) override; 54 NetPolicyResultCode GetCellularPolicies(std::vector<NetPolicyCellularPolicy> &cellularPolicies) override; 55 NetPolicyResultCode SetFactoryPolicy(const std::string &simId) override; 56 NetPolicyResultCode SetBackgroundPolicy(bool backgroundPolicy) override; 57 bool GetBackgroundPolicy() override; 58 bool GetBackgroundPolicyByUid(uint32_t uid) override; 59 NetBackgroundPolicy GetCurrentBackgroundPolicy() override; 60 NetPolicyResultCode SetSnoozePolicy(int8_t netType, const std::string &simId) override; 61 NetPolicyResultCode SetIdleTrustlist(uint32_t uid, bool isTrustlist) override; 62 NetPolicyResultCode GetIdleTrustlist(std::vector<uint32_t> &uids) override; 63 void CheckNetStatsOverLimit(const std::vector<NetPolicyQuotaPolicy> "aPolicies); 64 void CheckNetStatsOverLimit(const std::vector<NetPolicyCellularPolicy> &cellularPolicies); 65 int32_t GetPeriodEndTime(); 66 int64_t GetCurrentTime(); 67 68 private: 69 bool Init(); 70 71 private: 72 enum ServiceRunningState { 73 STATE_STOPPED = 0, 74 STATE_RUNNING, 75 }; 76 77 sptr<NetPolicyTraffic> netPolicyTraffic_; 78 sptr<NetPolicyFirewall> netPolicyFirewall_; 79 sptr<NetPolicyFile> netPolicyFile_; 80 bool registerToService_; 81 ServiceRunningState state_; 82 sptr<NetPolicyCallback> netPolicyCallback_; 83 sptr<NetPolicyServiceCommon> serviceComm_ = nullptr; 84 std::mutex mutex_; 85 std::vector<uint16_t> monthDay_; 86 }; 87 } // namespace NetManagerStandard 88 } // namespace OHOS 89 #endif // NET_POLICY_SERVICE_H 90