1 /*
2  * Copyright (C) 2021-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 #ifndef INCLUDE_NET_MANAGER_NATIVE_H
17 #define INCLUDE_NET_MANAGER_NATIVE_H
18 
19 #include <memory>
20 #include <string>
21 #include <vector>
22 
23 #include "bandwidth_manager.h"
24 #include "conn_manager.h"
25 #include "dns_manager.h"
26 #include "firewall_manager.h"
27 #include "interface_manager.h"
28 #include "interface_type.h"
29 #include "route_manager.h"
30 #include "vnic_manager.h"
31 #include "route_type.h"
32 #include "sharing_manager.h"
33 #include "uid_range.h"
34 #include "net_all_capabilities.h"
35 
36 namespace OHOS {
37 namespace nmd {
38 using namespace OHOS::NetManagerStandard;
39 using namespace OHOS::NetsysNative;
40 class NetManagerNative {
41 public:
42     NetManagerNative();
43     ~NetManagerNative() = default;
44 
45     static void GetOriginInterfaceIndex();
46     static std::vector<uint32_t> GetCurrentInterfaceIndex();
47     static void UpdateInterfaceIndex(uint32_t infIndex);
48 
49     void Init();
50 
51     int32_t NetworkReinitRoute();
52     int32_t SetInternetPermission(uint32_t uid, uint8_t allow, uint8_t isBroker);
53     int32_t NetworkCreatePhysical(int32_t netId, int32_t permission);
54     int32_t NetworkCreateVirtual(int32_t netId, bool hasDns);
55     int32_t NetworkDestroy(int32_t netId);
56     int32_t CreateVnic(uint16_t mtu, const std::string &tunAddr, int32_t prefix, const std::set<int32_t> &uids);
57     int32_t DestroyVnic();
58     int32_t NetworkAddUids(int32_t netId, const std::vector<UidRange> &uidRanges);
59     int32_t NetworkDelUids(int32_t netId, const std::vector<UidRange> &uidRanges);
60     int32_t NetworkAddInterface(int32_t netId, std::string iface, NetBearType netBearerType);
61     int32_t NetworkRemoveInterface(int32_t netId, std::string iface);
62 
63     MarkMaskParcel GetFwmarkForNetwork(int32_t netId);
64     int32_t NetworkAddRoute(int32_t netId, std::string ifName, std::string destination, std::string nextHop);
65     int32_t NetworkRemoveRoute(int32_t netId, std::string ifName, std::string destination, std::string nextHop);
66     int32_t NetworkGetDefault();
67     int32_t NetworkSetDefault(int32_t netId);
68     int32_t NetworkClearDefault();
69     int32_t NetworkSetPermissionForNetwork(int32_t netId, NetworkPermission permission);
70     std::vector<std::string> InterfaceGetList();
71 
72     int32_t SetProcSysNet(int32_t family, int32_t which, const std::string ifname, const std::string parameter,
73                           const std::string value);
74     int32_t GetProcSysNet(int32_t family, int32_t which, const std::string ifname, const std::string parameter,
75                           std::string *value);
76 
77     nmd::InterfaceConfigurationParcel GetInterfaceConfig(std::string ifName);
78     void SetInterfaceConfig(InterfaceConfigurationParcel cfg);
79     void ClearInterfaceAddrs(const std::string ifName);
80     int32_t GetInterfaceMtu(std::string ifName);
81     int32_t SetInterfaceMtu(std::string ifName, int32_t mtuValue);
82     int32_t SetTcpBufferSizes(const std::string &tcpBufferSizes);
83     int32_t AddInterfaceAddress(std::string ifName, std::string addrString, int32_t prefixLength);
84     int32_t DelInterfaceAddress(std::string ifName, std::string addrString, int32_t prefixLength);
85     int32_t DelInterfaceAddress(std::string ifName, std::string addrString, int32_t prefixLength,
86                                 const std::string &netCapabilities);
87     int32_t InterfaceSetIpAddress(const std::string &ifaceName, const std::string &ipAddress);
88     int32_t InterfaceSetIffUp(std::string ifaceName);
89     int32_t NetworkAddRouteParcel(int32_t netId, RouteInfoParcel routeInfo);
90     int32_t NetworkRemoveRouteParcel(int32_t netId, RouteInfoParcel routeInfo);
91 
92     int64_t GetCellularRxBytes();
93     int64_t GetCellularTxBytes();
94     int64_t GetAllRxBytes();
95     int64_t GetAllTxBytes();
96     int64_t GetUidTxBytes(int32_t uid);
97     int64_t GetUidRxBytes(int32_t uid);
98     int64_t GetIfaceRxBytes(std::string interfaceName);
99     int64_t GetIfaceTxBytes(std::string interfaceName);
100     int32_t IpEnableForwarding(const std::string &requester);
101     int32_t IpDisableForwarding(const std::string &requester);
102     int32_t EnableNat(const std::string &downstreamIface, const std::string &upstreamIface);
103     int32_t DisableNat(const std::string &downstreamIface, const std::string &upsteramIface);
104     int32_t IpfwdAddInterfaceForward(const std::string &fromIface, const std::string &toIface);
105     int32_t IpfwdRemoveInterfaceForward(const std::string &fromIface, const std::string &toIface);
106 
107     int32_t DnsSetResolverConfig(uint16_t netId, uint16_t baseTimeoutMsec, uint8_t retryCount,
108                                  const std::vector<std::string> &servers, const std::vector<std::string> &domains);
109     int32_t DnsGetResolverConfig(uint16_t netId, std::vector<std::string> &servers, std::vector<std::string> &domains,
110                                  uint16_t &baseTimeoutMsec, uint8_t &retryCount);
111     int32_t DnsCreateNetworkCache(uint16_t netid);
112     int32_t DnsDestroyNetworkCache(uint16_t netId);
113     int32_t BandwidthEnableDataSaver(bool enable);
114     int32_t BandwidthSetIfaceQuota(const std::string &ifName, int64_t bytes);
115     int32_t BandwidthRemoveIfaceQuota(const std::string &ifName);
116     int32_t BandwidthAddDeniedList(uint32_t uid);
117     int32_t BandwidthRemoveDeniedList(uint32_t uid);
118     int32_t BandwidthAddAllowedList(uint32_t uid);
119     int32_t BandwidthRemoveAllowedList(uint32_t uid);
120 
121     int32_t FirewallSetUidsAllowedListChain(uint32_t chain, const std::vector<uint32_t> &uids);
122     int32_t FirewallSetUidsDeniedListChain(uint32_t chain, const std::vector<uint32_t> &uids);
123     int32_t FirewallEnableChain(uint32_t chain, bool enable);
124     int32_t FirewallSetUidRule(uint32_t chain, const std::vector<uint32_t> &uids, uint32_t firewallRule);
125     void ShareDnsSet(uint16_t netId);
126     void StartDnsProxyListen();
127     void StopDnsProxyListen();
128     void GetDumpInfo(std::string &infos);
129     int32_t DnsGetAddrInfo(const std::string &hostName, const std::string &serverName, const AddrInfo &hints,
130                            uint16_t netId, std::vector<AddrInfo> &res);
131     int32_t AddStaticArp(const std::string &ipAddr, const std::string &macAddr, const std::string &ifName);
132     int32_t DelStaticArp(const std::string &ipAddr, const std::string &macAddr, const std::string &ifName);
133     int32_t RegisterDnsResultCallback(const sptr<INetDnsResultCallback> &callback, uint32_t timeStep);
134     int32_t UnregisterDnsResultCallback(const sptr<INetDnsResultCallback> &callback);
135     int32_t RegisterDnsHealthCallback(const sptr<INetDnsHealthCallback> &callback);
136     int32_t UnregisterDnsHealthCallback(const sptr<INetDnsHealthCallback> &callback);
137     int32_t SetIpv6PrivacyExtensions(const std::string &interfaceName, const uint32_t on);
138     int32_t SetEnableIpv6(const std::string &interfaceName, const uint32_t on);
139 #ifdef FEATURE_NET_FIREWALL_ENABLE
140     int32_t SetFirewallDefaultAction(FirewallRuleAction inDefault, FirewallRuleAction outDefault);
141     int32_t SetFirewallCurrentUserId(int32_t userId);
142     int32_t SetFirewallRules(NetFirewallRuleType type, const std::vector<sptr<NetFirewallBaseRule>> &ruleList,
143                              bool isFinish);
144     int32_t ClearFirewallRules(NetFirewallRuleType type);
145     int32_t RegisterNetFirewallCallback(const sptr<NetsysNative::INetFirewallCallback> &callback);
146     int32_t UnRegisterNetFirewallCallback(const sptr<NetsysNative::INetFirewallCallback> &callback);
147 #endif
148     int32_t SetNetworkAccessPolicy(uint32_t uid, NetworkAccessPolicy policy, bool reconfirmFlag, bool isBroker);
149     int32_t DeleteNetworkAccessPolicy(uint32_t uid);
150     int32_t NotifyNetBearerTypeChange(std::set<NetBearType> bearerTypes);
151     int32_t ClearFirewallAllRules();
152 private:
153     std::shared_ptr<BandwidthManager> bandwidthManager_ = nullptr;
154     std::shared_ptr<ConnManager> connManager_ = nullptr;
155     std::shared_ptr<FirewallManager> firewallManager_ = nullptr;
156     std::shared_ptr<RouteManager> routeManager_ = nullptr;
157     std::shared_ptr<InterfaceManager> interfaceManager_ = nullptr;
158     std::shared_ptr<SharingManager> sharingManager_ = nullptr;
159     std::shared_ptr<DnsManager> dnsManager_ = nullptr;
160     static inline std::vector<uint32_t> interfaceIdex_;
161 };
162 } // namespace nmd
163 } // namespace OHOS
164 #endif // !INCLUDE_NET_MANAGER_NATIVE_H
165