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 AUTH_NET_LEDGER_MOCK_H 17 #define AUTH_NET_LEDGER_MOCK_H 18 19 #include <gmock/gmock.h> 20 #include <mutex> 21 22 #include "auth_common.h" 23 #include "auth_session_fsm.h" 24 #include "bus_center_manager.h" 25 #include "cJSON.h" 26 #include "lnn_distributed_net_ledger.h" 27 #include "lnn_hichain_mock.h" 28 #include "lnn_local_net_ledger.h" 29 #include "lnn_node_info.h" 30 #include "softbus_conn_manager.h" 31 #include "softbus_json_utils.h" 32 33 namespace OHOS { 34 class AuthNetLedgerInterface { 35 public: AuthNetLedgerInterface()36 AuthNetLedgerInterface() {}; ~AuthNetLedgerInterface()37 virtual ~AuthNetLedgerInterface() {}; 38 39 virtual int32_t LnnGetLocalStrInfo(InfoKey key, char *info, uint32_t len) = 0; 40 virtual int32_t LnnDeleteSpecificTrustedDevInfo(const char *udid, int32_t localUserId) = 0; 41 virtual const NodeInfo *LnnGetLocalNodeInfo(void) = 0; 42 virtual int32_t LnnGetAuthPort(const NodeInfo *info, int32_t ifnameIdx) = 0; 43 virtual int32_t LnnGetSessionPort(const NodeInfo *info, int32_t ifnameIdx) = 0; 44 virtual int32_t LnnGetProxyPort(const NodeInfo *info, int32_t ifnameIdx) = 0; 45 virtual const char *LnnGetBtMac(const NodeInfo *info) = 0; 46 virtual const char *LnnGetDeviceName(const DeviceBasicInfo *info) = 0; 47 virtual char *LnnConvertIdToDeviceType(uint16_t typeId) = 0; 48 virtual const char *LnnGetDeviceUdid(const NodeInfo *info) = 0; 49 virtual int32_t LnnGetP2pRole(const NodeInfo *info) = 0; 50 virtual const char *LnnGetP2pMac(const NodeInfo *info) = 0; 51 virtual uint64_t LnnGetSupportedProtocols(const NodeInfo *info) = 0; 52 virtual int32_t LnnConvertDeviceTypeToId(const char *deviceType, uint16_t *typeId) = 0; 53 virtual int32_t LnnGetLocalNumInfo(InfoKey key, int32_t *info) = 0; 54 virtual NodeInfo *LnnGetNodeInfoById(const char *id, IdCategory type) = 0; 55 virtual bool LnnHasDiscoveryType(const NodeInfo *info, DiscoveryType type) = 0; 56 virtual int32_t LnnGetNetworkIdByUdid(const char *udid, char *buf, uint32_t len) = 0; 57 virtual int32_t LnnGetRemoteNumInfo(const char *netWorkId, InfoKey key, int32_t *info) = 0; 58 virtual int32_t LnnSetSupportDiscoveryType(char *info, const char *type) = 0; 59 virtual bool LnnHasSupportDiscoveryType(const char *destType, const char *type) = 0; 60 virtual bool LnnPeerHasExchangeDiscoveryType(const NodeInfo *info, DiscoveryType type) = 0; 61 virtual void RouteBuildClientAuthManager(int32_t cfd) = 0; 62 virtual void RouteClearAuthChannelId(int32_t cfd) = 0; 63 virtual bool GetJsonObjectStringItem( 64 const cJSON *json, const char * const string, char *target, uint32_t targetLen) = 0; 65 virtual int32_t LnnGetRemoteNodeInfoById(const char *id, IdCategory type, NodeInfo *info) = 0; 66 virtual bool LnnSetDlPtk(const char *networkId, const char *remotePtk) = 0; 67 virtual void LnnDumpRemotePtk(const char *oldPtk, const char *newPtk, const char *log) = 0; 68 virtual bool LnnGetOnlineStateById(const char *id, IdCategory type) = 0; 69 virtual int32_t LnnGetLocalNodeInfoSafe(NodeInfo *info) = 0; 70 virtual int32_t LnnGetLocalByteInfo(InfoKey key, uint8_t *info, uint32_t len) = 0; 71 virtual void LnnAnonymizePtk(const char *ptk, uint32_t len, char **anonymizedStr) = 0; 72 }; 73 class AuthNetLedgertInterfaceMock : public AuthNetLedgerInterface { 74 public: 75 AuthNetLedgertInterfaceMock(); 76 ~AuthNetLedgertInterfaceMock() override; 77 MOCK_METHOD3(LnnGetLocalStrInfo, int32_t(InfoKey, char *, uint32_t)); 78 MOCK_METHOD2(LnnDeleteSpecificTrustedDevInfo, int32_t(const char *, int32_t)); 79 MOCK_METHOD0(LnnGetLocalNodeInfo, const NodeInfo *()); 80 MOCK_METHOD2(LnnGetAuthPort, int32_t(const NodeInfo *, int32_t)); 81 MOCK_METHOD2(LnnGetSessionPort, int32_t(const NodeInfo *, int32_t)); 82 MOCK_METHOD2(LnnGetProxyPort, int32_t(const NodeInfo *, int32_t)); 83 MOCK_METHOD1(LnnGetBtMac, const char *(const NodeInfo *)); 84 MOCK_METHOD1(LnnGetDeviceName, const char *(const DeviceBasicInfo *)); 85 MOCK_METHOD1(LnnConvertIdToDeviceType, char *(uint16_t)); 86 MOCK_METHOD1(LnnGetDeviceUdid, const char *(const NodeInfo *)); 87 MOCK_METHOD1(LnnGetP2pRole, int32_t(const NodeInfo *)); 88 MOCK_METHOD1(LnnGetP2pMac, const char *(const NodeInfo *)); 89 MOCK_METHOD1(LnnGetSupportedProtocols, uint64_t(const NodeInfo *)); 90 MOCK_METHOD2(LnnConvertDeviceTypeToId, int32_t(const char *, uint16_t *)); 91 MOCK_METHOD2(LnnGetLocalNumInfo, int32_t(InfoKey, int32_t *)); 92 MOCK_METHOD2(LnnGetNodeInfoById, NodeInfo *(const char *, IdCategory)); 93 MOCK_METHOD2(LnnHasDiscoveryType, bool(const NodeInfo *, DiscoveryType)); 94 MOCK_METHOD3(LnnGetNetworkIdByUdid, int32_t(const char *, char *, uint32_t)); 95 MOCK_METHOD3(LnnGetRemoteNumInfo, int32_t(const char *, InfoKey, int32_t *)); 96 MOCK_METHOD2(LnnSetSupportDiscoveryType, int32_t(char *, const char *)); 97 MOCK_METHOD2(LnnHasSupportDiscoveryType, bool(const char *, const char *)); 98 MOCK_METHOD2(LnnPeerHasExchangeDiscoveryType, bool(const NodeInfo *, DiscoveryType)); 99 MOCK_METHOD1(RouteBuildClientAuthManager, void(int32_t)); 100 MOCK_METHOD1(RouteClearAuthChannelId, void(int32_t)); 101 MOCK_METHOD(bool, GetJsonObjectStringItem, 102 (const cJSON *json, const char * const string, char *target, uint32_t targetLen), (override)); 103 MOCK_METHOD3(LnnGetRemoteNodeInfoById, int32_t(const char *, IdCategory, NodeInfo *)); 104 MOCK_METHOD2(LnnSetDlPtk, bool(const char *, const char *)); 105 MOCK_METHOD3(LnnDumpRemotePtk, void(const char *, const char *, const char *)); 106 MOCK_METHOD2(LnnGetOnlineStateById, bool(const char *, IdCategory)); 107 MOCK_METHOD1(LnnGetLocalNodeInfoSafe, int32_t(NodeInfo *)); 108 MOCK_METHOD3(LnnGetLocalByteInfo, int32_t (InfoKey, uint8_t *, uint32_t)); 109 MOCK_METHOD3(LnnAnonymizePtk, void (const char *ptk, uint32_t len, char **anonymizedStr)); 110 111 static inline bool isRuned; 112 static inline SoftBusMutex mutex; 113 static char *Pack(int64_t authSeq, const AuthSessionInfo *info, AuthDataHead &head); 114 static void OnDeviceVerifyPass(AuthHandle authHandle, const NodeInfo *info); 115 static void OnDeviceNotTrusted(const char *peerUdid); 116 static void OnDeviceDisconnect(AuthHandle authHandle); 117 }; 118 } // namespace OHOS 119 #endif // AUTH_NET_LEDGER_MOCK_H