1 /* 2 * Copyright (c) 2022-2025 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_event.h" 24 #include "bus_center_info_key.h" 25 #include "bus_center_manager.h" 26 #include "lnn_device_info.h" 27 #include "lnn_distributed_net_ledger.h" 28 #include "lnn_node_info.h" 29 #include "softbus_adapter_crypto.h" 30 #include "softbus_bus_center.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, int32_t ifnameIdx) = 0; 40 virtual int32_t LnnGetSessionPort(const NodeInfo *info, int32_t ifnameIdx) = 0; 41 virtual int32_t LnnGetProxyPort(const NodeInfo *info, int32_t ifnameIdx) = 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( 57 const char *srcId, IdCategory srcIdType, IdCategory dstIdType, 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, int32_t key, uint8_t *info, uint32_t infoLen) = 0; 84 virtual int32_t LnnGetLocalDeviceInfo(NodeBasicInfo *info) = 0; 85 virtual int32_t LnnGetLocalByteInfo(InfoKey key, uint8_t *info, uint32_t len) = 0; 86 virtual bool LnnIsDefaultOhosAccount(void) = 0; 87 virtual bool IsPotentialTrustedDevice( 88 TrustedRelationIdType idType, const char *deviceId, bool isPrecise, bool isPointToPoint) = 0; 89 virtual bool LnnIsPotentialHomeGroup(const char *udid) = 0; 90 virtual int32_t LnnGetRemoteNodeInfoById(const char *id, IdCategory type, NodeInfo *info) = 0; 91 virtual int32_t LnnRegisterBleLpDeviceMediumMgr(void) = 0; 92 virtual int32_t LnnGetLocalNumU64Info(InfoKey key, uint64_t *info) = 0; 93 virtual int32_t LnnSetWifiDirectAddr(NodeInfo *info, const char *wifiDirectAddr); 94 virtual const char *LnnGetWifiDirectAddr(const NodeInfo *info) = 0; 95 virtual bool LnnSetDLWifiDirectAddr(const char *networkId, const char *addr) = 0; 96 virtual bool IsActiveOsAccountUnlocked(void) = 0; 97 virtual int32_t GetActiveOsAccountIds(void) = 0; 98 virtual int32_t AuthDeviceGetLatestIdByUuid(const char *uuid, AuthLinkType type, AuthHandle *authHandle) = 0; 99 virtual int32_t AuthGetLatestAuthSeqListByType( 100 const char *udid, int64_t *seqList, uint64_t *authVerifyTime, DiscoveryType type) = 0; 101 virtual int32_t LnnSetDLUnifiedDeviceName(const char *udid, const char *name) = 0; 102 virtual int32_t LnnSetDLUnifiedDefaultDeviceName(const char *udid, const char *name) = 0; 103 virtual int32_t LnnSetDLDeviceNickNameByUdid(const char *udid, const char *name) = 0; 104 virtual int32_t LnnSetDLDeviceStateVersion(const char *udid, int32_t stateVersion) = 0; 105 virtual int32_t LnnUpdateDistributedNodeInfo(NodeInfo *newInfo, const char *udid) = 0; 106 virtual int32_t LnnSetDLDeviceBroadcastCipherKey(const char *udid, const void *cipherKey) = 0; 107 virtual int32_t LnnSetDLDeviceBroadcastCipherIv(const char *udid, const void *cipherIv) = 0; 108 virtual int32_t LnnSetDLBssTransInfo(const char *networkId, const BssTransInfo *info) = 0; 109 virtual int32_t LnnSetDLBatteryInfo(const char *networkId, const BatteryInfo *info) = 0; 110 virtual int32_t LnnGetOsTypeByNetworkId(const char *networkId, int32_t *osType) = 0; 111 virtual bool LnnSetDLDeviceNickName(const char *networkId, const char *name); 112 virtual int32_t LnnUpdateLocalScreenStatus(bool isScreenOn) = 0; 113 virtual int32_t LnnClearStaticNetCap(uint32_t *capability, StaticNetCapability type) = 0; 114 virtual int32_t LnnSetStaticNetCap(uint32_t *capability, StaticNetCapability type) = 0; 115 virtual int32_t LnnSetLocalNumU32Info(InfoKey key, uint32_t info) = 0; 116 virtual int32_t LnnGetLocalStrInfoByIfnameIdx(InfoKey key, char *info, uint32_t len, int32_t ifIdx) = 0; 117 virtual int32_t LnnGetLocalNumInfoByIfnameIdx(InfoKey key, int32_t *info, int32_t ifIdx) = 0; 118 virtual int32_t LnnSetLocalStrInfoByIfnameIdx(InfoKey key, const char *info, int32_t ifIdx) = 0; 119 virtual int32_t LnnSetLocalNumInfoByIfnameIdx(InfoKey key, int32_t info, int32_t ifIdx) = 0; 120 }; 121 class LnnNetLedgertInterfaceMock : public LnnNetLedgerInterface { 122 public: 123 LnnNetLedgertInterfaceMock(); 124 ~LnnNetLedgertInterfaceMock() override; 125 MOCK_METHOD3(LnnGetLocalStrInfo, int32_t(InfoKey, char *, uint32_t)); 126 MOCK_METHOD0(LnnGetLocalNodeInfo, const NodeInfo *()); 127 MOCK_METHOD2(LnnGetAuthPort, int32_t(const NodeInfo *, int32_t)); 128 MOCK_METHOD2(LnnGetSessionPort, int32_t(const NodeInfo *, int32_t)); 129 MOCK_METHOD2(LnnGetProxyPort, int32_t(const NodeInfo *, int32_t)); 130 MOCK_METHOD1(LnnGetBtMac, const char *(const NodeInfo *)); 131 MOCK_METHOD1(LnnGetDeviceName, const char *(const DeviceBasicInfo *)); 132 MOCK_METHOD1(LnnConvertIdToDeviceType, char *(uint16_t)); 133 MOCK_METHOD1(LnnGetP2pRole, int32_t(const NodeInfo *)); 134 MOCK_METHOD1(LnnGetP2pMac, const char *(const NodeInfo *)); 135 MOCK_METHOD1(LnnGetWifiDirectAddr, const char *(const NodeInfo *)); 136 MOCK_METHOD1(LnnGetSupportedProtocols, uint64_t(const NodeInfo *)); 137 MOCK_METHOD2(LnnConvertDeviceTypeToId, int32_t(const char *, uint16_t *)); 138 MOCK_METHOD2(LnnGetLocalNumInfo, int32_t(InfoKey, int32_t *)); 139 MOCK_METHOD2(LnnGetLocalNumU32Info, int32_t(InfoKey, uint32_t *)); 140 MOCK_METHOD2(LnnHasCapability, bool(uint32_t, NetCapability)); 141 MOCK_METHOD2(LnnSetNetCapability, int32_t(uint32_t *, NetCapability)); 142 MOCK_METHOD2(LnnClearNetCapability, int32_t(uint32_t *, NetCapability)); 143 MOCK_METHOD2(LnnSetLocalNumInfo, int32_t(InfoKey, int32_t)); 144 MOCK_METHOD2(LnnSetLocalStrInfo, int32_t(InfoKey, const char *)); 145 MOCK_METHOD5(LnnConvertDlId, int32_t(const char *, IdCategory, IdCategory, char *, uint32_t)); 146 MOCK_METHOD2(LnnSetDLDeviceInfoName, bool(const char *, const char *)); 147 MOCK_METHOD2(LnnSetDLP2pInfo, bool(const char *, const P2pInfo *)); 148 MOCK_METHOD2(LnnSetDLWifiDirectAddr, bool(const char *, const char *addr)); 149 MOCK_METHOD2(LnnSetP2pRole, int32_t(NodeInfo *, int32_t)); 150 MOCK_METHOD2(LnnSetP2pMac, int32_t(NodeInfo *, const char *)); 151 MOCK_METHOD2(LnnSetP2pGoMac, int32_t(NodeInfo *, const char *)); 152 MOCK_METHOD2(LnnSetWifiDirectAddr, int32_t(NodeInfo *, const char *)); 153 MOCK_METHOD2(LnnGetAllOnlineAndMetaNodeInfo, int32_t(NodeBasicInfo **, int32_t *)); 154 MOCK_METHOD2(LnnGetAllOnlineNodeInfo, int32_t(NodeBasicInfo **, int32_t *)); 155 MOCK_METHOD1(LnnIsLSANode, bool(const NodeBasicInfo *)); 156 MOCK_METHOD2(LnnGetNodeInfoById, NodeInfo *(const char *, IdCategory)); 157 MOCK_METHOD4(LnnGetLnnRelation, int32_t(const char *, IdCategory, uint8_t *, uint32_t)); 158 MOCK_METHOD2(LnnSetDLConnCapability, int32_t(const char *, uint64_t)); 159 MOCK_METHOD2(LnnHasDiscoveryType, bool(const NodeInfo *, DiscoveryType)); 160 MOCK_METHOD1(LnnIsNodeOnline, bool(const NodeInfo *)); 161 MOCK_METHOD2(LnnGetCnnCode, short(const char *, DiscoveryType)); 162 MOCK_METHOD1(LnnAddOnlineNode, ReportCategory(NodeInfo *)); 163 MOCK_METHOD2(LnnGetBasicInfoByUdid, int32_t(const char *, NodeBasicInfo *)); 164 MOCK_METHOD1(LnnInsertSpecificTrustedDevInfo, int32_t(const char *)); 165 MOCK_METHOD1(LnnInsertSpecificTrustedDevInfo, char *(const NodeInfo *)); 166 MOCK_METHOD3(LnnSetNodeOffline, ReportCategory(const char *, ConnectionAddrType, int32_t)); 167 MOCK_METHOD1(LnnRemoveNode, void(const char *)); 168 MOCK_METHOD2(LnnSetSupportDiscoveryType, int32_t(char *, const char *)); 169 MOCK_METHOD2(LnnHasSupportDiscoveryType, bool(const char *, const char *)); 170 MOCK_METHOD2(LnnPeerHasExchangeDiscoveryType, bool(const NodeInfo *, DiscoveryType)); 171 MOCK_METHOD1(LnnGetDeviceUdid, const char *(const NodeInfo *)); 172 MOCK_METHOD3(LnnGetNetworkIdByBtMac, int32_t(const char *, char *, uint32_t)); 173 MOCK_METHOD2(LnnSetLocalNum64Info, int32_t(InfoKey, int64_t)); 174 MOCK_METHOD4(LnnGetNodeKeyInfo, int32_t(const char *, int, uint8_t *, uint32_t)); 175 MOCK_METHOD1(LnnGetLocalDeviceInfo, int32_t(NodeBasicInfo *)); 176 MOCK_METHOD3(LnnGetLocalByteInfo, int32_t(InfoKey key, uint8_t *info, uint32_t len)); 177 MOCK_METHOD0(LnnIsDefaultOhosAccount, bool(void)); 178 MOCK_METHOD4(IsPotentialTrustedDevice, 179 bool(TrustedRelationIdType idType, const char *deviceId, bool isPrecise, bool isPointToPoint)); 180 MOCK_METHOD1(LnnIsPotentialHomeGroup, bool(const char *udid)); 181 MOCK_METHOD3(LnnGetRemoteNodeInfoById, int32_t(const char *id, IdCategory type, NodeInfo *info)); 182 MOCK_METHOD0(LnnRegisterBleLpDeviceMediumMgr, int32_t(void)); 183 MOCK_METHOD2(LnnGetLocalNumU64Info, int32_t(InfoKey, uint64_t *)); 184 MOCK_METHOD0(IsActiveOsAccountUnlocked, bool(void)); 185 MOCK_METHOD0(GetActiveOsAccountIds, int32_t(void)); 186 MOCK_METHOD3(AuthDeviceGetLatestIdByUuid, int32_t(const char *, AuthLinkType, AuthHandle *)); 187 MOCK_METHOD4(AuthGetLatestAuthSeqListByType, int32_t(const char *, int64_t *, uint64_t *, DiscoveryType)); 188 MOCK_METHOD2(LnnSetDLUnifiedDeviceName, int32_t(const char *, const char *)); 189 MOCK_METHOD2(LnnSetDLUnifiedDefaultDeviceName, int32_t(const char *, const char *)); 190 MOCK_METHOD2(LnnSetDLDeviceNickNameByUdid, int32_t(const char *, const char *)); 191 MOCK_METHOD2(LnnSetDLDeviceStateVersion, int32_t(const char *, int32_t)); 192 MOCK_METHOD2(LnnUpdateDistributedNodeInfo, int32_t(NodeInfo *, const char *)); 193 MOCK_METHOD2(LnnSetDLDeviceBroadcastCipherKey, int32_t(const char *, const void *)); 194 MOCK_METHOD2(LnnSetDLDeviceBroadcastCipherIv, int32_t(const char *, const void *)); 195 MOCK_METHOD2(LnnSetDLBssTransInfo, int32_t(const char *, const BssTransInfo *)); 196 MOCK_METHOD2(LnnSetDLBatteryInfo, int32_t(const char *, const BatteryInfo *)); 197 MOCK_METHOD2(LnnGetOsTypeByNetworkId, int32_t(const char *, int32_t *)); 198 MOCK_METHOD2(LnnSetDLDeviceNickName, bool(const char *, const char *)); 199 MOCK_METHOD1(LnnUpdateLocalScreenStatus, int32_t(bool)); 200 MOCK_METHOD2(LnnClearStaticNetCap, int32_t(uint32_t *, StaticNetCapability)); 201 MOCK_METHOD2(LnnSetStaticNetCap, int32_t(uint32_t *, StaticNetCapability)); 202 MOCK_METHOD2(LnnSetLocalNumU32Info, int32_t(InfoKey key, uint32_t info)); 203 MOCK_METHOD4(LnnGetLocalStrInfoByIfnameIdx, int32_t(InfoKey, char *, uint32_t, int32_t)); 204 MOCK_METHOD3(LnnGetLocalNumInfoByIfnameIdx, int32_t(InfoKey, int32_t *, int32_t)); 205 MOCK_METHOD3(LnnSetLocalStrInfoByIfnameIdx, int32_t(InfoKey, const char *, int32_t)); 206 MOCK_METHOD3(LnnSetLocalNumInfoByIfnameIdx, int32_t(InfoKey, int32_t, int32_t)); 207 static int32_t ActionOfLnnGetAllOnline(NodeBasicInfo **info, int32_t *infoNum); 208 static int32_t ActionOfLnnConvertDlId( 209 const char *srcId, IdCategory srcIdType, IdCategory dstIdType, char *dstIdBuf, uint32_t dstIdBufLen); 210 static int32_t ActionOfLnnConvertDlId1( 211 const char *srcId, IdCategory srcIdType, IdCategory dstIdType, char *dstIdBuf, uint32_t dstIdBufLen); 212 static inline std::map<LnnEventType, LnnEventHandler> g_lnnevent_handlers; 213 static int32_t ActionOfLnnGetAllOnlineNodeInfo(NodeBasicInfo **info, int32_t *infoNum); 214 static int32_t ActionOfLnnGetAllOnlineNodeInfo1(NodeBasicInfo **info, int32_t *infoNum); 215 static int32_t ActionOfLnnGetLnnRelation(const char *id, IdCategory type, uint8_t *relation, uint32_t len); 216 static int32_t ActionOfLnnGetLnnRelation1(const char *id, IdCategory type, uint8_t *relation, uint32_t len); 217 static int32_t ActionOfLnnGetLocalStrInfo(InfoKey key, char *info, uint32_t len); 218 static int32_t ActionOfLnnGetLocalStrInfo1(InfoKey key, char *info, uint32_t len); 219 static int32_t ActionOfLnnGetLocalStrInfo2(InfoKey key, char *info, uint32_t len); 220 static int32_t ActionOfLnnGetAuthHandle(const char *uuid, AuthLinkType type, AuthHandle *authHandle); 221 static int32_t ActionOfLnnGetAuthSeqList( 222 const char *udid, int64_t *seqList, uint64_t *authVerifyTime, DiscoveryType type); 223 static int32_t ActionOfLnnGetLocalStrInfoByIfnameIdx(InfoKey key, char *info, uint32_t len, int32_t ifIdx); 224 225 static inline std::string localId = "6c38a1a0a8476679d0da6b24ec1e99eac67857244586e781bb8a07c77f636a3f"; 226 static inline std::string peerId = "54fec99a89886bef18373f0742661809b83b2d5dac9afbaf1728b940b9d51cb5"; 227 }; 228 } // namespace OHOS 229 #endif // LNN_NET_LEDGER_MOCK_H