1 /*
2  * Copyright (c) 2021-2023 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 #ifndef NETSYS_NATIVE_SERVICE_STUB_H
16 #define NETSYS_NATIVE_SERVICE_STUB_H
17 
18 #include <map>
19 
20 #include "i_netsys_service.h"
21 #include "iremote_stub.h"
22 
23 namespace OHOS {
24 namespace NetsysNative {
25 enum {
26     UID_ROOT = 0,
27     UID_SHELL = 2000,
28     UID_NET_MANAGER = 1099,
29     UID_WIFI = 1010,
30     UID_RADIO = 1001,
31     UID_HIDUMPER_SERVICE = 1212,
32     UID_SAMGR = 5555,
33     UID_PARAM_WATCHER = 1101,
34     UID_EDM = 3057,
35     UID_SECURITY_COLLECTOR = 3521,
36 };
37 
38 class NetsysNativeServiceStub : public IRemoteStub<INetsysService> {
39 public:
40     NetsysNativeServiceStub();
41     ~NetsysNativeServiceStub() = default;
42     int32_t OnRemoteRequest(uint32_t code, MessageParcel &data, MessageParcel &reply, MessageOption &option) override;
43     int32_t NetsysFreeAddrinfo(struct addrinfo *aihead);
44 
45 private:
46     using ServiceInterface = int32_t (NetsysNativeServiceStub::*)(MessageParcel &data, MessageParcel &reply);
47     std::map<uint32_t, ServiceInterface> opToInterfaceMap_;
48     void InitNetInfoOpToInterfaceMap();
49     void InitNetInfoOpToInterfaceMapPart2();
50     void InitBandwidthOpToInterfaceMap();
51     void InitFirewallOpToInterfaceMap();
52     void InitOpToInterfaceMapExt();
53     void InitNetDiagOpToInterfaceMap();
54     void InitNetDnsDiagOpToInterfaceMap();
55     void InitNetVnicInterfaceMap();
56     void InitStaticArpToInterfaceMap();
57     int32_t CmdSetResolverConfig(MessageParcel &data, MessageParcel &reply);
58     int32_t CmdGetResolverConfig(MessageParcel &data, MessageParcel &reply);
59     int32_t CmdCreateNetworkCache(MessageParcel &data, MessageParcel &reply);
60     int32_t CmdDestroyNetworkCache(MessageParcel &data, MessageParcel &reply);
61     int32_t CmdGetAddrInfo(MessageParcel &data, MessageParcel &reply);
62     int32_t CmdSetInterfaceMtu(MessageParcel &data, MessageParcel &reply);
63     int32_t CmdGetInterfaceMtu(MessageParcel &data, MessageParcel &reply);
64     int32_t CmdSetTcpBufferSizes(MessageParcel &data, MessageParcel &reply);
65 
66     int32_t CmdRegisterNotifyCallback(MessageParcel &data, MessageParcel &reply);
67     int32_t CmdUnRegisterNotifyCallback(MessageParcel &data, MessageParcel &reply);
68 
69     int32_t CmdNetworkAddRoute(MessageParcel &data, MessageParcel &reply);
70     int32_t CmdNetworkRemoveRoute(MessageParcel &data, MessageParcel &reply);
71     int32_t CmdNetworkAddRouteParcel(MessageParcel &data, MessageParcel &reply);
72     int32_t CmdNetworkRemoveRouteParcel(MessageParcel &data, MessageParcel &reply);
73     int32_t CmdNetworkSetDefault(MessageParcel &data, MessageParcel &reply);
74     int32_t CmdNetworkGetDefault(MessageParcel &data, MessageParcel &reply);
75     int32_t CmdNetworkClearDefault(MessageParcel &data, MessageParcel &reply);
76     int32_t CmdGetProcSysNet(MessageParcel &data, MessageParcel &reply);
77     int32_t CmdSetProcSysNet(MessageParcel &data, MessageParcel &reply);
78     int32_t CmdSetInternetPermission(MessageParcel &data, MessageParcel &reply);
79     int32_t CmdNetworkCreatePhysical(MessageParcel &data, MessageParcel &reply);
80     int32_t CmdNetworkCreateVirtual(MessageParcel &data, MessageParcel &reply);
81     int32_t CmdNetworkAddUids(MessageParcel &data, MessageParcel &reply);
82     int32_t CmdNetworkDelUids(MessageParcel &data, MessageParcel &reply);
83     int32_t CmdAddInterfaceAddress(MessageParcel &data, MessageParcel &reply);
84     int32_t CmdDelInterfaceAddress(MessageParcel &data, MessageParcel &reply);
85     int32_t CmdInterfaceSetIpAddress(MessageParcel &data, MessageParcel &reply);
86     int32_t CmdInterfaceSetIffUp(MessageParcel &data, MessageParcel &reply);
87     int32_t CmdNetworkAddInterface(MessageParcel &data, MessageParcel &reply);
88     int32_t CmdNetworkRemoveInterface(MessageParcel &data, MessageParcel &reply);
89     int32_t CmdNetworkDestroy(MessageParcel &data, MessageParcel &reply);
90     int32_t CmdCreateVnic(MessageParcel &data, MessageParcel &reply);
91     int32_t CmdDestroyVnic(MessageParcel &data, MessageParcel &reply);
92     int32_t CmdGetFwmarkForNetwork(MessageParcel &data, MessageParcel &reply);
93     int32_t CmdSetInterfaceConfig(MessageParcel &data, MessageParcel &reply);
94     int32_t CmdGetInterfaceConfig(MessageParcel &data, MessageParcel &reply);
95     int32_t CmdInterfaceGetList(MessageParcel &data, MessageParcel &reply);
96     int32_t CmdStartDhcpClient(MessageParcel &data, MessageParcel &reply);
97     int32_t CmdStopDhcpClient(MessageParcel &data, MessageParcel &reply);
98     int32_t CmdStartDhcpService(MessageParcel &data, MessageParcel &reply);
99     int32_t CmdStopDhcpService(MessageParcel &data, MessageParcel &reply);
100     int32_t CmdIpEnableForwarding(MessageParcel &data, MessageParcel &reply);
101     int32_t CmdIpDisableForwarding(MessageParcel &data, MessageParcel &reply);
102     int32_t CmdEnableNat(MessageParcel &data, MessageParcel &reply);
103     int32_t CmdDisableNat(MessageParcel &data, MessageParcel &reply);
104     int32_t CmdIpfwdAddInterfaceForward(MessageParcel &data, MessageParcel &reply);
105     int32_t CmdIpfwdRemoveInterfaceForward(MessageParcel &data, MessageParcel &reply);
106     int32_t CmdBandwidthEnableDataSaver(MessageParcel &data, MessageParcel &reply);
107     int32_t CmdBandwidthSetIfaceQuota(MessageParcel &data, MessageParcel &reply);
108     int32_t CmdBandwidthRemoveIfaceQuota(MessageParcel &data, MessageParcel &reply);
109     int32_t CmdBandwidthAddDeniedList(MessageParcel &data, MessageParcel &reply);
110     int32_t CmdBandwidthRemoveDeniedList(MessageParcel &data, MessageParcel &reply);
111     int32_t CmdBandwidthAddAllowedList(MessageParcel &data, MessageParcel &reply);
112     int32_t CmdBandwidthRemoveAllowedList(MessageParcel &data, MessageParcel &reply);
113     int32_t CmdFirewallSetUidsAllowedListChain(MessageParcel &data, MessageParcel &reply);
114     int32_t CmdFirewallSetUidsDeniedListChain(MessageParcel &data, MessageParcel &reply);
115     int32_t CmdFirewallEnableChain(MessageParcel &data, MessageParcel &reply);
116     int32_t CmdFirewallSetUidRule(MessageParcel &data, MessageParcel &reply);
117     int32_t CmdShareDnsSet(MessageParcel &data, MessageParcel &reply);
118     int32_t CmdStartDnsProxyListen(MessageParcel &data, MessageParcel &reply);
119     int32_t CmdStopDnsProxyListen(MessageParcel &data, MessageParcel &reply);
120     int32_t CmdGetNetworkSharingTraffic(MessageParcel &data, MessageParcel &reply);
121     int32_t CmdGetTotalStats(MessageParcel &data, MessageParcel &reply);
122     int32_t CmdGetUidStats(MessageParcel &data, MessageParcel &reply);
123     int32_t CmdGetIfaceStats(MessageParcel &data, MessageParcel &reply);
124     int32_t CmdGetAllSimStatsInfo(MessageParcel &data, MessageParcel &reply);
125     int32_t CmdDeleteSimStatsInfo(MessageParcel &data, MessageParcel &reply);
126     int32_t CmdGetAllStatsInfo(MessageParcel &data, MessageParcel &reply);
127     int32_t CmdDeleteStatsInfo(MessageParcel &data, MessageParcel &reply);
128     int32_t CmdSetIptablesCommandForRes(MessageParcel &data, MessageParcel &reply);
129     int32_t CmdNetDiagPingHost(MessageParcel &data, MessageParcel &reply);
130     int32_t CmdNetDiagGetRouteTable(MessageParcel &data, MessageParcel &reply);
131     int32_t CmdNetDiagGetSocketsInfo(MessageParcel &data, MessageParcel &reply);
132     int32_t CmdNetDiagGetInterfaceConfig(MessageParcel &data, MessageParcel &reply);
133     int32_t CmdNetDiagUpdateInterfaceConfig(MessageParcel &data, MessageParcel &reply);
134     int32_t CmdNetDiagSetInterfaceActiveState(MessageParcel &data, MessageParcel &reply);
135     int32_t CmdAddStaticArp(MessageParcel &data, MessageParcel &reply);
136     int32_t CmdDelStaticArp(MessageParcel &data, MessageParcel &reply);
137     int32_t CmdRegisterDnsResultListener(MessageParcel &data, MessageParcel &reply);
138     int32_t CmdUnregisterDnsResultListener(MessageParcel &data, MessageParcel &reply);
139     int32_t CmdRegisterDnsHealthListener(MessageParcel &data, MessageParcel &reply);
140     int32_t CmdUnregisterDnsHealthListener(MessageParcel &data, MessageParcel &reply);
141     int32_t CmdGetCookieStats(MessageParcel &data, MessageParcel &reply);
142     int32_t CmdGetNetworkSharingType(MessageParcel &data, MessageParcel &reply);
143     int32_t CmdUpdateNetworkSharingType(MessageParcel &data, MessageParcel &reply);
144 #ifdef FEATURE_NET_FIREWALL_ENABLE
145     int32_t CmdSetFirewallRules(MessageParcel &data, MessageParcel &reply);
146     int32_t CmdSetFirewallDefaultAction(MessageParcel &data, MessageParcel &reply);
147     int32_t CmdSetFirewallCurrentUserId(MessageParcel &data, MessageParcel &reply);
148     int32_t CmdClearFirewallRules(MessageParcel &data, MessageParcel &reply);
149     int32_t CmdRegisterNetFirewallCallback(MessageParcel &data, MessageParcel &reply);
150     int32_t CmdUnRegisterNetFirewallCallback(MessageParcel &data, MessageParcel &reply);
151 #endif
152     int32_t CmdSetIpv6PrivacyExtensions(MessageParcel &data, MessageParcel &reply);
153     int32_t CmdSetIpv6Enable(MessageParcel &data, MessageParcel &reply);
154     int32_t CmdSetNetworkAccessPolicy(MessageParcel &data, MessageParcel &reply);
155     int32_t CmdDelNetworkAccessPolicy(MessageParcel &data, MessageParcel &reply);
156     int32_t CmdNotifyNetBearerTypeChange(MessageParcel &data, MessageParcel &reply);
157     int32_t CmdStartClat(MessageParcel &data, MessageParcel &reply);
158     int32_t CmdStopClat(MessageParcel &data, MessageParcel &reply);
159     int32_t CmdClearFirewallAllRules(MessageParcel &data, MessageParcel &reply);
160     int32_t CmdSetNicTrafficAllowed(MessageParcel &data, MessageParcel &reply);
161 private:
162     std::vector<int32_t> uids_;
163 };
164 } // namespace NetsysNative
165 } // namespace OHOS
166 #endif // NETSYS_NATIVE_SERVICE_STUB_H
167