1 /*
2  * Copyright (c) 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 LNN_NET_LEDGER_MOCK_H
17 #define LNN_NET_LEDGER_MOCK_H
18 
19 #include <gmock/gmock.h>
20 #include <mutex>
21 
22 #include "auth_hichain_adapter.h"
23 #include "bus_center_manager.h"
24 #include "lnn_node_info.h"
25 #include "lnn_distributed_net_ledger.h"
26 #include "bus_center_info_key.h"
27 #include "lnn_device_info.h"
28 #include "softbus_bus_center.h"
29 #include "softbus_adapter_crypto.h"
30 #include "bus_center_event.h"
31 
32 namespace OHOS {
33 class LnnNetLedgerInterface {
34 public:
LnnNetLedgerInterface()35     LnnNetLedgerInterface() {};
~LnnNetLedgerInterface()36     virtual ~LnnNetLedgerInterface() {};
37     virtual int32_t LnnGetLocalStrInfo(InfoKey key, char *info, uint32_t len) = 0;
38     virtual const NodeInfo *LnnGetLocalNodeInfo(void) = 0;
39     virtual int32_t LnnGetAuthPort(const NodeInfo *info) = 0;
40     virtual int32_t LnnGetSessionPort(const NodeInfo *info) = 0;
41     virtual int32_t LnnGetProxyPort(const NodeInfo *info) = 0;
42     virtual const char *LnnGetBtMac(const NodeInfo *info) = 0;
43     virtual const char *LnnGetDeviceName(const DeviceBasicInfo *info) = 0;
44     virtual char *LnnConvertIdToDeviceType(uint16_t typeId) = 0;
45     virtual int32_t LnnGetP2pRole(const NodeInfo *info) = 0;
46     virtual const char *LnnGetP2pMac(const NodeInfo *info) = 0;
47     virtual uint64_t LnnGetSupportedProtocols(const NodeInfo *info) = 0;
48     virtual int32_t LnnConvertDeviceTypeToId(const char *deviceType, uint16_t *typeId) = 0;
49     virtual int32_t LnnGetLocalNumInfo(InfoKey key, int32_t *info) = 0;
50     virtual int32_t LnnGetLocalNumU32Info(InfoKey key, uint32_t *info) = 0;
51     virtual bool LnnHasCapability(uint32_t capability, NetCapability type) = 0;
52     virtual int32_t LnnSetNetCapability(uint32_t *capability, NetCapability type) = 0;
53     virtual int32_t LnnClearNetCapability(uint32_t *capability, NetCapability type) = 0;
54     virtual int32_t LnnSetLocalNumInfo(InfoKey key, int32_t info) = 0;
55     virtual int32_t LnnSetLocalStrInfo(InfoKey key, const char *info) = 0;
56     virtual int32_t LnnConvertDlId(const char *srcId, IdCategory srcIdType, IdCategory dstIdType,
57         char *dstIdBuf, uint32_t dstIdBufLen) = 0;
58     virtual bool LnnSetDLDeviceInfoName(const char *udid, const char *name) = 0;
59     virtual bool LnnSetDLP2pInfo(const char *networkId, const P2pInfo *info) = 0;
60     virtual int32_t LnnSetP2pRole(NodeInfo *info, int32_t role) = 0;
61     virtual int32_t LnnSetP2pMac(NodeInfo *info, const char *p2pMac) = 0;
62     virtual int32_t LnnSetP2pGoMac(NodeInfo *info, const char *goMac) = 0;
63     virtual int32_t LnnGetAllOnlineAndMetaNodeInfo(NodeBasicInfo **info, int32_t *infoNum) = 0;
64     virtual int32_t LnnGetAllOnlineNodeInfo(NodeBasicInfo **info, int32_t *infoNum) = 0;
65     virtual bool LnnIsLSANode(const NodeBasicInfo *info) = 0;
66     virtual NodeInfo *LnnGetNodeInfoById(const char *id, IdCategory type) = 0;
67     virtual int32_t LnnGetLnnRelation(const char *id, IdCategory type, uint8_t *relation, uint32_t len) = 0;
68     virtual int32_t LnnSetDLConnCapability(const char *networkId, uint64_t connCapability) = 0;
69     virtual bool LnnHasDiscoveryType(const NodeInfo *info, DiscoveryType type) = 0;
70     virtual bool LnnIsNodeOnline(const NodeInfo *info) = 0;
71     virtual short LnnGetCnnCode(const char *uuid, DiscoveryType type) = 0;
72     virtual ReportCategory LnnAddOnlineNode(NodeInfo *info) = 0;
73     virtual int32_t LnnGetBasicInfoByUdid(const char *udid, NodeBasicInfo *basicInfo) = 0;
74     virtual int32_t LnnInsertSpecificTrustedDevInfo(const char *udid) = 0;
75     virtual int32_t LnnSetSupportDiscoveryType(char *info, const char *type) = 0;
76     virtual bool LnnHasSupportDiscoveryType(const char *destType, const char *type) = 0;
77     virtual bool LnnPeerHasExchangeDiscoveryType(const NodeInfo *info, DiscoveryType type) = 0;
78     virtual const char *LnnGetDeviceUdid(const NodeInfo *info) = 0;
79     virtual ReportCategory LnnSetNodeOffline(const char *udid, ConnectionAddrType type, int32_t authId) = 0;
80     virtual void LnnRemoveNode(const char *udid) = 0;
81     virtual int32_t LnnGetNetworkIdByBtMac(const char *btMac, char *buf, uint32_t len) = 0;
82     virtual int32_t LnnSetLocalNum64Info(InfoKey key, int64_t info) = 0;
83     virtual int32_t LnnGetNodeKeyInfo(const char *networkId, int key, uint8_t *info, uint32_t infoLen) = 0;
84     virtual int32_t LnnGetRemoteNumInfo(const char *netWorkId, InfoKey key, int32_t *info) = 0;
85     virtual int32_t LnnGetLocalDeviceInfo(NodeBasicInfo *info) = 0;
86     virtual int32_t LnnGetLocalByteInfo(InfoKey key, uint8_t *info, uint32_t len) = 0;
87     virtual bool LnnIsDefaultOhosAccount(void) = 0;
88     virtual bool IsPotentialTrustedDevice(TrustedRelationIdType idType, const char *deviceId,
89         bool isPrecise, bool isPointToPoint) = 0;
90     virtual bool LnnIsPotentialHomeGroup(const char *udid) = 0;
91     virtual int32_t LnnGetRemoteNodeInfoById(const char *id, IdCategory type, NodeInfo *info) = 0;
92     virtual int32_t LnnRegisterBleLpDeviceMediumMgr(void) = 0;
93     virtual int32_t LnnGetLocalNumU64Info(InfoKey key, uint64_t *info) = 0;
94     virtual int32_t LnnSetWifiDirectAddr(NodeInfo *info, const char *wifiDirectAddr);
95     virtual const char *LnnGetWifiDirectAddr(const NodeInfo *info) = 0;
96     virtual bool LnnSetDLWifiDirectAddr(const char *networkId, const char *addr) = 0;
97     virtual bool IsActiveOsAccountUnlocked(void) = 0;
98     virtual int32_t GetActiveOsAccountIds(void) = 0;
99     virtual int32_t AuthDeviceGetLatestIdByUuid(const char *uuid, AuthLinkType type, AuthHandle *authHandle) = 0;
100     virtual int32_t AuthGetLatestAuthSeqListByType(const char *udid, int64_t *seqList, uint64_t *authVerifyTime,
101         DiscoveryType type) = 0;
102     virtual int32_t LnnSetDLUnifiedDeviceName(const char *udid, const char *name) = 0;
103     virtual int32_t LnnSetDLUnifiedDefaultDeviceName(const char *udid, const char *name) = 0;
104     virtual int32_t LnnSetDLDeviceNickNameByUdid(const char *udid, const char *name) = 0;
105     virtual int32_t LnnSetDLDeviceStateVersion(const char *udid, int32_t stateVersion) = 0;
106     virtual int32_t LnnUpdateDistributedNodeInfo(NodeInfo *newInfo, const char *udid) = 0;
107     virtual int32_t LnnSetDLDeviceBroadcastCipherKey(const char *udid, const void *cipherKey) = 0;
108     virtual int32_t LnnSetDLDeviceBroadcastCipherIv(const char *udid, const void *cipherIv) = 0;
109     virtual int32_t LnnSetDLBssTransInfo(const char *networkId, const BssTransInfo *info) = 0;
110     virtual int32_t LnnSetDLBatteryInfo(const char *networkId, const BatteryInfo *info) = 0;
111     virtual int32_t LnnGetOsTypeByNetworkId(const char *networkId, int32_t *osType) = 0;
112     virtual bool LnnSetDLDeviceNickName(const char *networkId, const char *name);
113     virtual int32_t LnnUpdateLocalScreenStatus(bool isScreenOn) = 0;
114 };
115 class LnnNetLedgertInterfaceMock : public LnnNetLedgerInterface {
116 public:
117     LnnNetLedgertInterfaceMock();
118     ~LnnNetLedgertInterfaceMock() override;
119     MOCK_METHOD3(LnnGetLocalStrInfo, int32_t (InfoKey, char *, uint32_t));
120     MOCK_METHOD0(LnnGetLocalNodeInfo, const NodeInfo *());
121     MOCK_METHOD1(LnnGetAuthPort, int32_t (const NodeInfo *));
122     MOCK_METHOD1(LnnGetSessionPort, int32_t (const NodeInfo *));
123     MOCK_METHOD1(LnnGetProxyPort, int32_t (const NodeInfo *));
124     MOCK_METHOD1(LnnGetBtMac, const char *(const NodeInfo *));
125     MOCK_METHOD1(LnnGetDeviceName, const char *(const DeviceBasicInfo *));
126     MOCK_METHOD1(LnnConvertIdToDeviceType, char *(uint16_t));
127     MOCK_METHOD1(LnnGetP2pRole, int32_t (const NodeInfo *));
128     MOCK_METHOD1(LnnGetP2pMac, const char *(const NodeInfo *));
129     MOCK_METHOD1(LnnGetWifiDirectAddr,  const char *(const NodeInfo *));
130     MOCK_METHOD1(LnnGetSupportedProtocols, uint64_t (const NodeInfo *));
131     MOCK_METHOD2(LnnConvertDeviceTypeToId, int32_t (const char *, uint16_t *));
132     MOCK_METHOD2(LnnGetLocalNumInfo, int32_t (InfoKey, int32_t *));
133     MOCK_METHOD2(LnnGetLocalNumU32Info, int32_t (InfoKey, uint32_t *));
134     MOCK_METHOD2(LnnHasCapability, bool (uint32_t, NetCapability));
135     MOCK_METHOD2(LnnSetNetCapability, int32_t (uint32_t *, NetCapability));
136     MOCK_METHOD2(LnnClearNetCapability, int32_t (uint32_t *, NetCapability));
137     MOCK_METHOD2(LnnSetLocalNumInfo, int32_t (InfoKey, int32_t));
138     MOCK_METHOD2(LnnSetLocalStrInfo, int32_t (InfoKey, const char *));
139     MOCK_METHOD5(LnnConvertDlId, int32_t (const char *, IdCategory, IdCategory, char *, uint32_t));
140     MOCK_METHOD2(LnnSetDLDeviceInfoName, bool (const char *, const char *));
141     MOCK_METHOD2(LnnSetDLP2pInfo, bool (const char *, const P2pInfo *));
142     MOCK_METHOD2(LnnSetDLWifiDirectAddr, bool (const char *, const char *addr));
143     MOCK_METHOD2(LnnSetP2pRole, int32_t (NodeInfo *, int32_t));
144     MOCK_METHOD2(LnnSetP2pMac, int32_t (NodeInfo *, const char *));
145     MOCK_METHOD2(LnnSetP2pGoMac, int32_t (NodeInfo *, const char *));
146     MOCK_METHOD2(LnnSetWifiDirectAddr, int32_t (NodeInfo *, const char *));
147     MOCK_METHOD2(LnnGetAllOnlineAndMetaNodeInfo, int32_t (NodeBasicInfo **, int32_t *));
148     MOCK_METHOD2(LnnGetAllOnlineNodeInfo, int32_t (NodeBasicInfo **, int32_t *));
149     MOCK_METHOD1(LnnIsLSANode, bool(const NodeBasicInfo *));
150     MOCK_METHOD2(LnnGetNodeInfoById, NodeInfo *(const char *, IdCategory));
151     MOCK_METHOD4(LnnGetLnnRelation, int32_t(const char *, IdCategory, uint8_t *, uint32_t));
152     MOCK_METHOD2(LnnSetDLConnCapability, int32_t (const char *, uint64_t));
153     MOCK_METHOD2(LnnHasDiscoveryType, bool (const NodeInfo *, DiscoveryType));
154     MOCK_METHOD1(LnnIsNodeOnline, bool (const NodeInfo *));
155     MOCK_METHOD2(LnnGetCnnCode, short (const char *, DiscoveryType));
156     MOCK_METHOD1(LnnAddOnlineNode, ReportCategory (NodeInfo *));
157     MOCK_METHOD2(LnnGetBasicInfoByUdid, int32_t (const char *, NodeBasicInfo *));
158     MOCK_METHOD1(LnnInsertSpecificTrustedDevInfo, int32_t (const char *));
159     MOCK_METHOD1(LnnInsertSpecificTrustedDevInfo, char * (const NodeInfo *));
160     MOCK_METHOD3(LnnSetNodeOffline, ReportCategory (const char *, ConnectionAddrType, int32_t));
161     MOCK_METHOD1(LnnRemoveNode, void (const char *));
162     MOCK_METHOD2(LnnSetSupportDiscoveryType, int32_t(char *, const char *));
163     MOCK_METHOD2(LnnHasSupportDiscoveryType, bool(const char *, const char *));
164     MOCK_METHOD2(LnnPeerHasExchangeDiscoveryType, bool(const NodeInfo *, DiscoveryType));
165     MOCK_METHOD1(LnnGetDeviceUdid, const char *(const NodeInfo *));
166     MOCK_METHOD3(LnnGetNetworkIdByBtMac, int32_t (const char *, char *, uint32_t));
167     MOCK_METHOD2(LnnSetLocalNum64Info, int32_t (InfoKey, int64_t));
168     MOCK_METHOD4(LnnGetNodeKeyInfo, int32_t(const char *, int, uint8_t *, uint32_t));
169     MOCK_METHOD3(LnnGetRemoteNumInfo, int32_t (const char*, InfoKey, int32_t*));
170     MOCK_METHOD1(LnnGetLocalDeviceInfo, int32_t(NodeBasicInfo *));
171     MOCK_METHOD3(LnnGetLocalByteInfo, int32_t(InfoKey key, uint8_t *info, uint32_t len));
172     MOCK_METHOD0(LnnIsDefaultOhosAccount, bool (void));
173     MOCK_METHOD4(IsPotentialTrustedDevice, bool (TrustedRelationIdType idType, const char *deviceId,
174         bool isPrecise, bool isPointToPoint));
175     MOCK_METHOD1(LnnIsPotentialHomeGroup, bool (const char *udid));
176     MOCK_METHOD3(LnnGetRemoteNodeInfoById, int32_t (const char *id, IdCategory type, NodeInfo *info));
177     MOCK_METHOD0(LnnRegisterBleLpDeviceMediumMgr, int32_t (void));
178     MOCK_METHOD2(LnnGetLocalNumU64Info, int32_t(InfoKey, uint64_t *));
179     MOCK_METHOD0(IsActiveOsAccountUnlocked, bool (void));
180     MOCK_METHOD0(GetActiveOsAccountIds, int32_t (void));
181     MOCK_METHOD3(AuthDeviceGetLatestIdByUuid, int32_t (const char *, AuthLinkType, AuthHandle *));
182     MOCK_METHOD4(AuthGetLatestAuthSeqListByType, int32_t (const char *, int64_t *, uint64_t *, DiscoveryType));
183     MOCK_METHOD2(LnnSetDLUnifiedDeviceName, int32_t (const char *, const char *));
184     MOCK_METHOD2(LnnSetDLUnifiedDefaultDeviceName, int32_t (const char *, const char *));
185     MOCK_METHOD2(LnnSetDLDeviceNickNameByUdid, int32_t (const char *, const char *));
186     MOCK_METHOD2(LnnSetDLDeviceStateVersion, int32_t (const char *, int32_t));
187     MOCK_METHOD2(LnnUpdateDistributedNodeInfo, int32_t (NodeInfo *, const char *));
188     MOCK_METHOD2(LnnSetDLDeviceBroadcastCipherKey, int32_t (const char *, const void *));
189     MOCK_METHOD2(LnnSetDLDeviceBroadcastCipherIv, int32_t (const char *, const void *));
190     MOCK_METHOD2(LnnSetDLBssTransInfo, int32_t (const char *, const BssTransInfo *));
191     MOCK_METHOD2(LnnSetDLBatteryInfo, int32_t (const char *, const BatteryInfo *));
192     MOCK_METHOD2(LnnGetOsTypeByNetworkId, int32_t (const char *, int32_t *));
193     MOCK_METHOD2(LnnSetDLDeviceNickName, bool (const char *, const char *));
194     MOCK_METHOD1(LnnUpdateLocalScreenStatus, int32_t(bool));
195     static int32_t ActionOfLnnGetAllOnline(NodeBasicInfo **info, int32_t *infoNum);
196     static int32_t ActionOfLnnConvertDlId(const char *srcId, IdCategory srcIdType, IdCategory dstIdType,
197         char *dstIdBuf, uint32_t dstIdBufLen);
198     static int32_t ActionOfLnnConvertDlId1(const char *srcId, IdCategory srcIdType, IdCategory dstIdType,
199         char *dstIdBuf, uint32_t dstIdBufLen);
200     static inline std::map<LnnEventType, LnnEventHandler> g_lnnevent_handlers;
201     static int32_t ActionOfLnnGetAllOnlineNodeInfo(NodeBasicInfo **info, int32_t *infoNum);
202     static int32_t ActionOfLnnGetAllOnlineNodeInfo1(NodeBasicInfo **info, int32_t *infoNum);
203     static int32_t ActionOfLnnGetLnnRelation(const char *id, IdCategory type,
204         uint8_t *relation, uint32_t len);
205     static int32_t ActionOfLnnGetLnnRelation1(const char *id, IdCategory type,
206         uint8_t *relation, uint32_t len);
207     static int32_t ActionOfLnnGetLocalStrInfo(InfoKey key, char *info, uint32_t len);
208     static int32_t ActionOfLnnGetLocalStrInfo1(InfoKey key, char *info, uint32_t len);
209     static int32_t ActionOfLnnGetLocalStrInfo2(InfoKey key, char *info, uint32_t len);
210     static int32_t ActionOfLnnGetAuthHandle(const char *uuid, AuthLinkType type, AuthHandle *authHandle);
211     static int32_t ActionOfLnnGetAuthSeqList(const char *udid, int64_t *seqList, uint64_t *authVerifyTime,
212         DiscoveryType type);
213 
214     static inline std::string localId = "6c38a1a0a8476679d0da6b24ec1e99eac67857244586e781bb8a07c77f636a3f";
215     static inline std::string peerId = "54fec99a89886bef18373f0742661809b83b2d5dac9afbaf1728b940b9d51cb5";
216 };
217 } // namespace OHOS
218 #endif // LNN_NET_LEDGER_MOCK_H