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 "discovery_callback_stub.h" 21 #include "resolve_callback_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 int32_t HandleStartDiscover(const MDnsServiceInfo &serviceInfo, int32_t retCode) override 36 { 37 return NETMANAGER_EXT_SUCCESS; 38 } 39 int32_t HandleStopDiscover(const MDnsServiceInfo &serviceInfo, int32_t retCode) override; 40 int32_t HandleServiceLost(const MDnsServiceInfo &serviceInfo, int32_t retCode) override; 41 int32_t HandleServiceFound(const MDnsServiceInfo &info, int32_t retCode) override; OnRemoteRequest(uint32_t code,MessageParcel & data,MessageParcel & reply,MessageOption & option)42 int32_t OnRemoteRequest(uint32_t code, MessageParcel &data, MessageParcel &reply, MessageOption &option) override 43 { 44 SCAN_HILOGD("GetScannerList HandleSerieFound OnRemoteRequest"); 45 return DiscoveryCallbackStub::OnRemoteRequest(code, data, reply, option); 46 } 47 public: 48 MDnsServiceInfo expected_; 49 }; 50 51 class ScanMDnsResolveObserver : public ResolveCallbackStub { 52 public: ScanMDnsResolveObserver(const MDnsServiceInfo & info)53 explicit ScanMDnsResolveObserver(const MDnsServiceInfo &info):_serviceInfo(info) {} 54 virtual ~ScanMDnsResolveObserver() = default; OnRemoteRequest(uint32_t code,MessageParcel & data,MessageParcel & reply,MessageOption & option)55 int32_t OnRemoteRequest(uint32_t code, MessageParcel &data, MessageParcel &reply, MessageOption& option) override 56 { 57 SCAN_HILOGD("GetScannerList ScanMDnsResolveObserver OnRemoteRequest"); 58 return ResolveCallbackStub::OnRemoteRequest(code, data, reply, option); 59 } 60 int32_t HandleResolveResult(const MDnsServiceInfo& info, int32_t retCode) override; 61 public: 62 MDnsServiceInfo _serviceInfo; 63 }; 64 65 class ScanMDnsLossResolveObserver : public ResolveCallbackStub { 66 public: ScanMDnsLossResolveObserver(const MDnsServiceInfo & info)67 explicit ScanMDnsLossResolveObserver(const MDnsServiceInfo &info):_serviceInfo(info) {} 68 virtual ~ScanMDnsLossResolveObserver() = default; OnRemoteRequest(uint32_t code,MessageParcel & data,MessageParcel & reply,MessageOption & option)69 int32_t OnRemoteRequest(uint32_t code, MessageParcel &data, MessageParcel &reply, MessageOption& option) override 70 { 71 SCAN_HILOGD("GetScannerList ScanMDnsLossResolveObserver OnRemoteRequest"); 72 return ResolveCallbackStub::OnRemoteRequest(code, data, reply, option); 73 } 74 int32_t HandleResolveResult(const MDnsServiceInfo& info, int32_t retCode) override; 75 public: 76 MDnsServiceInfo _serviceInfo; 77 }; 78 79 class ScanMdnsService { 80 public: 81 static void InsertIpToScannerInfo(const std::string& ip, ScanDeviceInfoTCP& scanDeviceInfoTCP); 82 static bool OnStartDiscoverService(); 83 static bool OnStopDiscoverService(); 84 static bool FindNetScannerInfoByIp(const std::string& ip, ScanDeviceInfoTCP& netScannerInfo); 85 private: 86 static void UpdateScannerIdThread(); 87 static std::map<std::string, sptr<ScanMDnsDiscoveryObserver>> discoveryCallBackPtrs_; 88 static std::mutex discoveryCallBackPtrsLock_; 89 static bool isListening_; 90 }; 91 } 92 #endif // !SCAN_MDNS_SERVICE_H 93