1 /*
2  * Copyright (c) 2021-2022 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_CALLBACK_STUB_H
17 #define NET_POLICY_CALLBACK_STUB_H
18 
19 #include <map>
20 
21 #include "iremote_stub.h"
22 
23 #include "i_net_policy_callback.h"
24 
25 namespace OHOS {
26 namespace NetManagerStandard {
27 class NetPolicyCallbackStub : public IRemoteStub<INetPolicyCallback> {
28 public:
29     NetPolicyCallbackStub();
30     virtual ~NetPolicyCallbackStub();
31 
32     int32_t OnRemoteRequest(uint32_t code, MessageParcel &data, MessageParcel &reply, MessageOption &option) override;
33 
34 public:
35     int32_t NetUidPolicyChange(uint32_t uid, uint32_t policy) override;
36     int32_t NetUidRuleChange(uint32_t uid, uint32_t rule) override;
37     int32_t NetQuotaPolicyChange(const std::vector<NetQuotaPolicy> &quotaPolicies) override;
38     int32_t NetStrategySwitch(const std::string &simId, bool enable) override;
39     int32_t NetMeteredIfacesChange(std::vector<std::string> &ifaces) override;
40     int32_t NetBackgroundPolicyChange(bool isBackgroundPolicyAllow) override;
41 
42 private:
43     using NetPolicyCallbackFunc = int32_t (NetPolicyCallbackStub::*)(MessageParcel &, MessageParcel &);
44 
45 private:
46     int32_t OnNetUidPolicyChange(MessageParcel &data, MessageParcel &reply);
47     int32_t OnNetUidRuleChange(MessageParcel &data, MessageParcel &reply);
48     int32_t OnNetQuotaPolicyChange(MessageParcel &data, MessageParcel &reply);
49     int32_t OnNetStrategySwitch(MessageParcel &data, MessageParcel &reply);
50     int32_t OnNetMeteredIfacesChange(MessageParcel &data, MessageParcel &reply);
51     int32_t OnNetBackgroundPolicyChange(MessageParcel &data, MessageParcel &reply);
52 
53 private:
54     std::map<uint32_t, NetPolicyCallbackFunc> memberFuncMap_;
55 };
56 } // namespace NetManagerStandard
57 } // namespace OHOS
58 #endif // NET_POLICY_CALLBACK_STUB_H