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 SCAN_MDNS_SERVICE_H 17 #define SCAN_MDNS_SERVICE_H 18 19 #include <unordered_set> 20 #include "i_mdns_event.h" 21 #include "mdns_event_stub.h" 22 #include "mdns_client.h" 23 #include "mdns_common.h" 24 #include "scanner_info.h" 25 #include "scan_log.h" 26 #include "scan_service_ability.h" 27 28 namespace OHOS::Scan { 29 using namespace OHOS::NetManagerStandard; 30 31 class ScanMDnsDiscoveryObserver : public DiscoveryCallbackStub { 32 public: ScanMDnsDiscoveryObserver(const MDnsServiceInfo & info)33 explicit ScanMDnsDiscoveryObserver(const MDnsServiceInfo &info) : expected_(info) {} 34 virtual ~ScanMDnsDiscoveryObserver() = default; HandleStartDiscover(const MDnsServiceInfo & serviceInfo,int32_t retCode)35 void HandleStartDiscover(const MDnsServiceInfo &serviceInfo, int32_t retCode) override{} 36 void HandleStopDiscover(const MDnsServiceInfo &serviceInfo, int32_t retCode) override; 37 void HandleServiceLost(const MDnsServiceInfo &serviceInfo, int32_t retCode) override; 38 void HandleServiceFound(const MDnsServiceInfo &info, int32_t retCode) override; OnRemoteRequest(uint32_t code,MessageParcel & data,MessageParcel & reply,MessageOption & option)39 int32_t OnRemoteRequest(uint32_t code, MessageParcel &data, MessageParcel &reply, MessageOption &option) override 40 { 41 SCAN_HILOGD("GetScannerList HandleSerieFound OnRemoteRequest"); 42 return DiscoveryCallbackStub::OnRemoteRequest(code, data, reply, option); 43 } 44 public: 45 MDnsServiceInfo expected_; 46 }; 47 48 class ScanMDnsResolveObserver : public ResolveCallbackStub { 49 public: ScanMDnsResolveObserver(const MDnsServiceInfo & info)50 explicit ScanMDnsResolveObserver(const MDnsServiceInfo &info):_serviceInfo(info) {} 51 virtual ~ScanMDnsResolveObserver() = default; OnRemoteRequest(uint32_t code,MessageParcel & data,MessageParcel & reply,MessageOption & option)52 int32_t OnRemoteRequest(uint32_t code, MessageParcel &data, MessageParcel &reply, MessageOption& option) override 53 { 54 SCAN_HILOGD("GetScannerList ScanMDnsResolveObserver OnRemoteRequest"); 55 return ResolveCallbackStub::OnRemoteRequest(code, data, reply, option); 56 } 57 void HandleResolveResult(const MDnsServiceInfo& info, int32_t retCode) override; 58 public: 59 MDnsServiceInfo _serviceInfo; 60 }; 61 62 class ScanMDnsLossResolveObserver : public ResolveCallbackStub { 63 public: ScanMDnsLossResolveObserver(const MDnsServiceInfo & info)64 explicit ScanMDnsLossResolveObserver(const MDnsServiceInfo &info):_serviceInfo(info) {} 65 virtual ~ScanMDnsLossResolveObserver() = default; OnRemoteRequest(uint32_t code,MessageParcel & data,MessageParcel & reply,MessageOption & option)66 int32_t OnRemoteRequest(uint32_t code, MessageParcel &data, MessageParcel &reply, MessageOption& option) override 67 { 68 SCAN_HILOGD("GetScannerList ScanMDnsLossResolveObserver OnRemoteRequest"); 69 return ResolveCallbackStub::OnRemoteRequest(code, data, reply, option); 70 } 71 void HandleResolveResult(const MDnsServiceInfo& info, int32_t retCode) override; 72 public: 73 MDnsServiceInfo _serviceInfo; 74 }; 75 76 class ScanMdnsService { 77 public: 78 static void InsertIpToScannerInfo(const std::string& ip, ScanDeviceInfoTCP& scanDeviceInfoTCP); 79 static bool OnStartDiscoverService(); 80 static bool OnStopDiscoverService(); 81 static bool FindNetScannerInfoByIp(const std::string& ip, ScanDeviceInfoTCP& netScannerInfo); 82 private: 83 static void UpdateScannerIdThread(); 84 static std::map<std::string, sptr<ScanMDnsDiscoveryObserver>> discoveryCallBackPtrs_; 85 static std::mutex discoveryCallBackPtrsLock_; 86 static bool isListening_; 87 }; 88 } 89 #endif // !SCAN_MDNS_SERVICE_H 90