src/discovery/discovery_manager.cpp

/*
 * Copyright (c) 2022-2024 Huawei Device Co., Ltd.
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#include "discovery_manager.h"

#include <dlfcn.h>
#include "softbus_common.h"

#include "dm_anonymous.h"
#include "dm_constants.h"
#include "dm_random.h"
#include "parameter.h"
#if !(defined(__LITEOS_M__) || defined(LITE_DEVICE))
#include "multiple_user_connector.h"
#endif

namespace OHOS {
namespace DistributedHardware {
const int32_t DISCOVERY_TIMEOUT = 120;
const uint16_t DM_INVALID_FLAG_ID = 0;
constexpr const char* LNN_DISC_CAPABILITY = "capability";
const std::string TYPE_MINE = "findDeviceMode";
const int32_t DECIMALISM = 10;

#if !(defined(__LITEOS_M__) || defined(LITE_DEVICE))
static std::mutex comDependencyLoadLock;
constexpr const char* LIB_DM_COMDENPENDENCY_NAME = "libdevicemanagerdependency.z.so";
bool DiscoveryManager::isSoLoaded_ = false;
IDeviceProfileConnector* DiscoveryManager::dpConnector_ = nullptr;
void* DiscoveryManager::dpConnectorHandle_ = nullptr;
#endif

DiscoveryManager::DiscoveryManager(std::shared_ptr<SoftbusListener> softbusListener,
    std::shared_ptr<IDeviceManagerServiceListener> listener) : softbusListener_(softbusListener), listener_(listener)
{
    LOGI("DiscoveryManager constructor.");
}

DiscoveryManager::~DiscoveryManager()
{
#if !(defined(__LITEOS_M__) || defined(LITE_DEVICE))
    CloseCommonDependencyObj();
#endif
    LOGI("DiscoveryManager destructor.");
}

int32_t DiscoveryManager::EnableDiscoveryListener(const std::string &pkgName,
    const std::map<std::string, std::string> &discoverParam, const std::map<std::string, std::string> &filterOptions)
{
    if (pkgName.empty()) {
        LOGE("Invalid parameter, pkgName is empty.");
        return ERR_DM_INPUT_PARA_INVALID;
    }
    LOGI("begin for pkgName = %{public}s.", pkgName.c_str());
    std::string pkgNameTemp = AddMultiUserIdentify(pkgName);
    DmSubscribeInfo dmSubInfo;
    dmSubInfo.subscribeId = DM_INVALID_FLAG_ID;
    dmSubInfo.mode = DmDiscoverMode::DM_DISCOVER_MODE_PASSIVE;
    dmSubInfo.freq = DmExchangeFreq::DM_LOW;
    dmSubInfo.isSameAccount = false;
    dmSubInfo.isWakeRemote = false;
    if (strcpy_s(dmSubInfo.capability, DM_MAX_DEVICE_CAPABILITY_LEN, DM_CAPABILITY_APPROACH) != EOK) {
        LOGE("capability copy err.");
        return ERR_DM_ENABLE_DISCOVERY_LISTENER_FAILED;
    }
    UpdateInfoFreq(discoverParam, dmSubInfo);
    if (discoverParam.find(PARAM_KEY_META_TYPE) != discoverParam.end()) {
        std::string metaType = discoverParam.find(PARAM_KEY_META_TYPE)->second;
        LOGI("input MetaType = %{public}s.", metaType.c_str());
    }
    if (discoverParam.find(PARAM_KEY_SUBSCRIBE_ID) != discoverParam.end() &&
        IsNumberString((discoverParam.find(PARAM_KEY_SUBSCRIBE_ID)->second))) {
            uint16_t externalSubId =
                static_cast<uint16_t>(std::atoi((discoverParam.find(PARAM_KEY_SUBSCRIBE_ID)->second).c_str()));
            dmSubInfo.subscribeId = GenInnerSubId(pkgNameTemp, externalSubId);
    }
    if (discoverParam.find(PARAM_KEY_DISC_CAPABILITY) != discoverParam.end()) {
        std::string capability = discoverParam.find(PARAM_KEY_DISC_CAPABILITY)->second;
        if (strcpy_s(dmSubInfo.capability, DM_MAX_DEVICE_CAPABILITY_LEN, capability.c_str()) != EOK) {
            LOGE("failed, capability copy err.");
            return ERR_DM_ENABLE_DISCOVERY_LISTENER_FAILED;
        }
    }
    LOGI("capability = %{public}s,", std::string(dmSubInfo.capability).c_str());
    {
        std::lock_guard<std::mutex> capLock(capabilityMapLocks_);
        capabilityMap_[pkgNameTemp] = std::string(dmSubInfo.capability);
    }
    UpdateInfoMedium(discoverParam, dmSubInfo);
    int32_t ret = softbusListener_->RefreshSoftbusLNN(DM_PKG_NAME, dmSubInfo, LNN_DISC_CAPABILITY);
    if (ret != DM_OK) {
        LOGE("failed, softbus refresh lnn ret: %{public}d.", ret);
        return ret;
    }
    softbusListener_->RegisterSoftbusLnnOpsCbk(pkgNameTemp, shared_from_this());
    return DM_OK;
}

int32_t DiscoveryManager::DisableDiscoveryListener(const std::string &pkgName,
    const std::map<std::string, std::string> &extraParam)
{
    if (pkgName.empty()) {
        LOGE("Invalid parameter, pkgName is empty.");
        return ERR_DM_INPUT_PARA_INVALID;
    }
    LOGI("begin for pkgName = %{public}s.", pkgName.c_str());
    std::string pkgNameTemp = RemoveMultiUserIdentify(pkgName);
    if (extraParam.find(PARAM_KEY_META_TYPE) != extraParam.end()) {
        LOGI("DisableDiscoveryListener, input MetaType = %{public}s",
            (extraParam.find(PARAM_KEY_META_TYPE)->second).c_str());
    }
    uint16_t innerSubId = DM_INVALID_FLAG_ID;
    if (extraParam.find(PARAM_KEY_SUBSCRIBE_ID) != extraParam.end() &&
        IsNumberString(extraParam.find(PARAM_KEY_SUBSCRIBE_ID)->second)) {
        uint16_t externalSubId =
            static_cast<uint16_t>(std::atoi((extraParam.find(PARAM_KEY_SUBSCRIBE_ID)->second).c_str()));
        innerSubId = GetAndRemoveInnerSubId(pkgNameTemp, externalSubId);
    }
    {
        std::lock_guard<std::mutex> capLock(capabilityMapLocks_);
        if (capabilityMap_.find(pkgNameTemp) != capabilityMap_.end()) {
            capabilityMap_.erase(pkgNameTemp);
        }
    }
    softbusListener_->UnRegisterSoftbusLnnOpsCbk(pkgNameTemp);
    if (innerSubId == DM_INVALID_FLAG_ID) {
        LOGE("Invalid parameter, cannot find subscribeId in cache map.");
        return ERR_DM_INPUT_PARA_INVALID;
    }
    return softbusListener_->StopRefreshSoftbusLNN(innerSubId);
}

int32_t DiscoveryManager::StartDiscovering(const std::string &pkgName,
    const std::map<std::string, std::string> &discoverParam, const std::map<std::string, std::string> &filterOptions)
{
    if (pkgName.empty()) {
        LOGE("Invalid parameter, pkgName is empty.");
        return ERR_DM_INPUT_PARA_INVALID;
    }
    LOGI("begin for pkgName = %{public}s.", pkgName.c_str());
    std::string pkgNameTemp = AddMultiUserIdentify(pkgName);
    DmSubscribeInfo dmSubInfo;
    ConfigDiscParam(discoverParam, &dmSubInfo);
    if (HandleDiscoveryQueue(pkgNameTemp, dmSubInfo.subscribeId, filterOptions) != DM_OK) {
        return ERR_DM_DISCOVERY_REPEATED;
    }
    dmSubInfo.subscribeId = GenInnerSubId(pkgNameTemp, dmSubInfo.subscribeId);

    bool isStandardMetaNode = true;
    if (discoverParam.find(PARAM_KEY_META_TYPE) != discoverParam.end()) {
        MetaNodeType metaType = (MetaNodeType)(std::atoi((discoverParam.find(PARAM_KEY_META_TYPE)->second).c_str()));
        isStandardMetaNode = (metaType == MetaNodeType::PROXY_TRANSMISION);
    }

    softbusListener_->RegisterSoftbusLnnOpsCbk(pkgNameTemp, shared_from_this());
    StartDiscoveryTimer(pkgNameTemp);

    auto it = filterOptions.find(PARAM_KEY_FILTER_OPTIONS);
    if (it != filterOptions.end()) {
        JsonObject jsonObject(it->second);
        if (!jsonObject.IsDiscarded() && jsonObject.Contains(TYPE_MINE)) {
            return StartDiscovering4MineLibary(pkgNameTemp, dmSubInfo, it->second);
        }
    }

    int32_t ret = isStandardMetaNode ? StartDiscoveringNoMetaType(pkgNameTemp, dmSubInfo, discoverParam) :
            StartDiscovering4MetaType(pkgNameTemp, dmSubInfo, discoverParam);
    if (ret != DM_OK) {
        LOGE("StartDiscovering for meta node process failed, ret = %{public}d", ret);
        return ret;
    }
    return ret;
}

void DiscoveryManager::ConfigDiscParam(const std::map<std::string, std::string> &discoverParam,
    DmSubscribeInfo *dmSubInfo)
{
    if (dmSubInfo == nullptr) {
        LOGE("ConfigDiscParam failed, dmSubInfo is nullptr.");
        return;
    }
    dmSubInfo->subscribeId = DM_INVALID_FLAG_ID;
    dmSubInfo->mode = DmDiscoverMode::DM_DISCOVER_MODE_ACTIVE;
    dmSubInfo->medium = DmExchangeMedium::DM_AUTO;
    dmSubInfo->freq = DmExchangeFreq::DM_LOW;
    dmSubInfo->isSameAccount = false;
    dmSubInfo->isWakeRemote = false;
    if (discoverParam.find(PARAM_KEY_SUBSCRIBE_ID) != discoverParam.end()) {
        dmSubInfo->subscribeId = std::atoi((discoverParam.find(PARAM_KEY_SUBSCRIBE_ID)->second).c_str());
    }
    if (discoverParam.find(PARAM_KEY_DISC_MEDIUM) != discoverParam.end()) {
        int32_t medium = std::atoi((discoverParam.find(PARAM_KEY_DISC_MEDIUM)->second).c_str());
        dmSubInfo->medium = static_cast<DmExchangeMedium>(medium);
    }
    if (discoverParam.find(PARAM_KEY_DISC_FREQ) != discoverParam.end()) {
        int32_t freq = std::atoi((discoverParam.find(PARAM_KEY_DISC_FREQ)->second).c_str());
        dmSubInfo->freq = static_cast<DmExchangeFreq>(freq);
    }
    if (discoverParam.find(PARAM_KEY_DISC_MODE) != discoverParam.end()) {
        dmSubInfo->mode =
            static_cast<DmDiscoverMode>(std::atoi((discoverParam.find(PARAM_KEY_DISC_MODE)->second).c_str()));
    }
}

int32_t DiscoveryManager::StartDiscovering4MineLibary(const std::string &pkgName, DmSubscribeInfo &dmSubInfo,
    const std::string &searchJson)
{
    if (strcpy_s(dmSubInfo.capability, DM_MAX_DEVICE_CAPABILITY_LEN, DM_CAPABILITY_OSD) != EOK) {
        LOGE("capability copy err.");
        return ERR_DM_START_DISCOVERING_FAILED;
    }
    LOGI("mine meta node process, pkgName = %{public}s, capability = %{public}s",
        pkgName.c_str(), std::string(dmSubInfo.capability).c_str());
    {
        std::lock_guard<std::mutex> capLock(capabilityMapLocks_);
        CHECK_SIZE_RETURN(capabilityMap_, ERR_DM_START_DISCOVERING_FAILED);
        capabilityMap_[pkgName] = std::string(dmSubInfo.capability);
    }
    int32_t ret = mineSoftbusListener_->RefreshSoftbusLNN(pkgName, searchJson, dmSubInfo);
    if (ret != DM_OK) {
        LOGE("StartDiscovering for meta node process failed, ret = %{public}d", ret);
        return ret;
    }
    return ret;
}

int32_t DiscoveryManager::StartDiscoveringNoMetaType(const std::string &pkgName, DmSubscribeInfo &dmSubInfo,
    const std::map<std::string, std::string> &param)
{
    if (param.find(PARAM_KEY_DISC_CAPABILITY) != param.end() &&
        !param.find(PARAM_KEY_DISC_CAPABILITY)->second.empty()) {
        if (strcpy_s(dmSubInfo.capability, DM_MAX_DEVICE_CAPABILITY_LEN,
            param.find(PARAM_KEY_DISC_CAPABILITY)->second.c_str()) != EOK) {
            LOGE("capability copy err.");
            return ERR_DM_START_DISCOVERING_FAILED;
        }
    } else if (strcpy_s(dmSubInfo.capability, DM_MAX_DEVICE_CAPABILITY_LEN, DM_CAPABILITY_OSD) != EOK) {
        LOGE("capability copy err.");
        return ERR_DM_START_DISCOVERING_FAILED;
    }
    LOGI("standard meta node process, pkgName = %{public}s, capability = %{public}s",
        pkgName.c_str(), std::string(dmSubInfo.capability).c_str());

    {
        std::lock_guard<std::mutex> capLock(capabilityMapLocks_);
        CHECK_SIZE_RETURN(capabilityMap_, ERR_DM_START_DISCOVERING_FAILED);
        capabilityMap_[pkgName] = std::string(dmSubInfo.capability);
    }
    std::string customData = LNN_DISC_CAPABILITY;
    if (param.find(PARAM_KEY_CUSTOM_DATA) != param.end() && !param.find(PARAM_KEY_CUSTOM_DATA)->second.empty()) {
        customData = param.find(PARAM_KEY_CUSTOM_DATA)->second;
    }
    int32_t ret = softbusListener_->RefreshSoftbusLNN(DM_PKG_NAME, dmSubInfo, customData);
    if (ret != DM_OK) {
        LOGE("StartDiscoveringNoMetaType failed, softbus refresh lnn ret: %{public}d.", ret);
    }
    return ret;
}

int32_t DiscoveryManager::StartDiscovering4MetaType(const std::string &pkgName, DmSubscribeInfo &dmSubInfo,
    const std::map<std::string, std::string> &param)
{
    MetaNodeType metaType = MetaNodeType::CUSTOM_UNKNOWN;
    if (param.find(PARAM_KEY_META_TYPE) != param.end()) {
        LOGI("meta node process, input metaType = %{public}s, pkgName = %{public}s",
            (param.find(PARAM_KEY_META_TYPE)->second).c_str(), pkgName.c_str());
        metaType = (MetaNodeType)(std::atoi((param.find(PARAM_KEY_META_TYPE)->second).c_str()));
    }
    switch (metaType) {
        case MetaNodeType::PROXY_SHARE:
            if (strcpy_s(dmSubInfo.capability, DM_MAX_DEVICE_CAPABILITY_LEN, DM_CAPABILITY_SHARE) != EOK) {
                LOGE("capability copy error.");
                return ERR_DM_FAILED;
            }
            break;
        case MetaNodeType::PROXY_WEAR:
            if (strcpy_s(dmSubInfo.capability, DM_MAX_DEVICE_CAPABILITY_LEN, DM_CAPABILITY_WEAR) != EOK) {
                LOGE("capability copy error.");
                return ERR_DM_FAILED;
            }
            break;
        case MetaNodeType::PROXY_CASTPLUS:
            if (strcpy_s(dmSubInfo.capability, DM_MAX_DEVICE_CAPABILITY_LEN, DM_CAPABILITY_CASTPLUS) != EOK) {
                LOGE("capability copy error.");
                return ERR_DM_FAILED;
            }
            break;
        default:
            return ERR_DM_UNSUPPORTED_METHOD;
    }

    std::string customData = "";
    if (param.find(PARAM_KEY_CUSTOM_DATA) != param.end()) {
        customData = param.find(PARAM_KEY_CUSTOM_DATA)->second;
    }
    LOGI("capability = %{public}s,", std::string(dmSubInfo.capability).c_str());
    {
        std::lock_guard<std::mutex> capLock(capabilityMapLocks_);
        CHECK_SIZE_RETURN(capabilityMap_, ERR_DM_START_DISCOVERING_FAILED);
        capabilityMap_[pkgName] =std::string(dmSubInfo.capability);
    }

    int32_t ret = softbusListener_->RefreshSoftbusLNN(DM_PKG_NAME, dmSubInfo, customData);
    if (ret != DM_OK) {
        LOGE("StartDiscovering4MetaType failed, softbus refresh lnn ret: %{public}d.", ret);
    }
    return ret;
}

int32_t DiscoveryManager::StopDiscovering(const std::string &pkgName, uint16_t subscribeId)
{
    if (pkgName.empty()) {
        LOGE("Invalid parameter, pkgName is empty.");
        return ERR_DM_INPUT_PARA_INVALID;
    }
    LOGI("begin for pkgName = %{public}s.", pkgName.c_str());
    std::string pkgNameTemp = RemoveMultiUserIdentify(pkgName);
    return StopDiscoveringByInnerSubId(pkgNameTemp, subscribeId);
}

int32_t DiscoveryManager::StopDiscoveringByInnerSubId(const std::string &pkgName, uint16_t subscribeId)
{
    if (pkgName.empty()) {
        LOGE("Invalid parameter, pkgName is empty.");
        return ERR_DM_INPUT_PARA_INVALID;
    }
    LOGI("begin for pkgName = %{public}s.", pkgName.c_str());
    uint16_t innerSubId = static_cast<uint16_t>(GetAndRemoveInnerSubId(pkgName, subscribeId));
    if (innerSubId == DM_INVALID_FLAG_ID) {
        LOGE("Invalid parameter, cannot find subscribeId in cache map.");
        return ERR_DM_INPUT_PARA_INVALID;
    }
    {
        std::lock_guard<std::mutex> autoLock(locks_);
        if (pkgNameSet_.find(pkgName) != pkgNameSet_.end()) {
            pkgNameSet_.erase(pkgName);
        }

        if (discoveryContextMap_.find(pkgName) != discoveryContextMap_.end()) {
            discoveryContextMap_.erase(pkgName);
            if (timer_ != nullptr) {
                timer_->DeleteTimer(pkgName);
            }
        }
    }
    {
        std::lock_guard<std::mutex> capLock(capabilityMapLocks_);
        if (capabilityMap_.find(pkgName) != capabilityMap_.end()) {
            capabilityMap_.erase(pkgName);
        }
    }
    softbusListener_->UnRegisterSoftbusLnnOpsCbk(pkgName);
#if (defined(MINE_HARMONY))
    return mineSoftbusListener_->StopRefreshSoftbusLNN(innerSubId);
#else
    return softbusListener_->StopRefreshSoftbusLNN(innerSubId);
#endif
}

void DiscoveryManager::OnDeviceFound(const std::string &pkgName, const DmDeviceInfo &info, bool isOnline)
{
    int32_t userId = -1;
    std::string callerPkgName = "";
    GetPkgNameAndUserId(pkgName, callerPkgName, userId);
    DeviceFilterPara filterPara;
    filterPara.isOnline = false;
    filterPara.range = info.range;
    filterPara.deviceType = info.deviceTypeId;
    std::string deviceIdHash = static_cast<std::string>(info.deviceId);
    if (isOnline && GetDeviceAclParam(callerPkgName, userId, deviceIdHash, filterPara.isOnline,
        filterPara.authForm) != DM_OK) {
        LOGE("The found device get online param failed.");
    }
    JsonObject jsonObject(info.extraData);
    if (jsonObject.IsDiscarded()) {
        LOGE("OnDeviceFound jsonStr error");
        return;
    }
    if (!IsUint32(jsonObject, PARAM_KEY_DISC_CAPABILITY)) {
        LOGE("err json string: %{public}s", PARAM_KEY_DISC_CAPABILITY);
        return;
    }
    uint32_t capabilityType = jsonObject[PARAM_KEY_DISC_CAPABILITY].Get<uint32_t>();
    OnDeviceFound(pkgName, capabilityType, info, filterPara);
}

void DiscoveryManager::OnDeviceFound(const std::string &pkgName, const uint32_t capabilityType,
    const DmDeviceInfo &info, const DeviceFilterPara &filterPara)
{
    int32_t userId = -1;
    std::string callerPkgName = "";
    GetPkgNameAndUserId(pkgName, callerPkgName, userId);
    ProcessInfo processInfo;
    processInfo.userId = userId;
    processInfo.pkgName = callerPkgName;
    bool isIndiscoveryContextMap = false;
    DiscoveryContext discoveryContext;
    {
        std::lock_guard<std::mutex> autoLock(locks_);
        auto iter = discoveryContextMap_.find(pkgName);
        isIndiscoveryContextMap = (iter != discoveryContextMap_.end());
        if (isIndiscoveryContextMap) {
            discoveryContext = iter->second;
        }
    }
    uint16_t externalSubId = DM_INVALID_FLAG_ID;
    {
        std::lock_guard<std::mutex> autoLock(subIdMapLocks_);
        for (auto iter : pkgName2SubIdMap_[pkgName]) {
            externalSubId = iter.first;
            break;
        }
    }
    if (!isIndiscoveryContextMap) {
        {
            std::lock_guard<std::mutex> capLock(capabilityMapLocks_);
            if (capabilityMap_.find(pkgName) == capabilityMap_.end() ||
                !CompareCapability(capabilityType, capabilityMap_[pkgName])) {
                return;
            }
        }
        LOGD("OnDeviceFound, pkgName = %{public}s, cabability = %{public}d", pkgName.c_str(), capabilityType);
        listener_->OnDeviceFound(processInfo, externalSubId, info);
        return;
    }
    DiscoveryFilter filter;
    if (filter.IsValidDevice(discoveryContext.filterOp, discoveryContext.filters, filterPara)) {
        {
            std::lock_guard<std::mutex> capLock(capabilityMapLocks_);
            if (capabilityMap_.find(pkgName) == capabilityMap_.end() ||
                !CompareCapability(capabilityType, capabilityMap_[pkgName])) {
                return;
            }
        }
        LOGD("OnDeviceFound, pkgName = %{public}s, cabability = %{public}d", pkgName.c_str(), capabilityType);
        listener_->OnDeviceFound(processInfo, externalSubId, info);
    }
}

bool DiscoveryManager::CompareCapability(uint32_t capabilityType, const std::string &capabilityStr)
{
    for (uint32_t i = 0; i < sizeof(g_capabilityMap) / sizeof(g_capabilityMap[0]); i++) {
        if (strcmp(capabilityStr.c_str(), g_capabilityMap[i].capability) == 0) {
            LOGD("capabilityType: %{public}d, capabilityStr: %{public}s", capabilityType, capabilityStr.c_str());
            return ((capabilityType >> static_cast<uint32_t>(g_capabilityMap[i].bitmap)) & 1);
        }
    }
    return false;
}

void DiscoveryManager::OnDiscoveringResult(const std::string &pkgName, int32_t subscribeId, int32_t result)
{
    LOGI("subscribeId = %{public}d, result = %{public}d.", subscribeId, result);
    int32_t userId = -1;
    std::string callerPkgName = "";
    GetPkgNameAndUserId(pkgName, callerPkgName, userId);
    ProcessInfo processInfo;
    processInfo.userId = userId;
    processInfo.pkgName = callerPkgName;
    if (pkgName.empty() || (listener_ == nullptr)) {
        LOGE("DiscoveryManager::OnDiscoveringResult failed, IDeviceManagerServiceListener is null.");
        return;
    }
    uint16_t externalSubId = DM_INVALID_FLAG_ID;
    {
        std::lock_guard<std::mutex> autoLock(subIdMapLocks_);
        for (auto iter : pkgName2SubIdMap_[pkgName]) {
            if (iter.second == subscribeId) {
                externalSubId = iter.first;
                break;
            }
        }
    }
    if (result == 0) {
        std::lock_guard<std::mutex> autoLock(locks_);
        discoveryContextMap_[pkgName].subscribeId = (uint32_t)externalSubId;
        listener_->OnDiscoverySuccess(processInfo, externalSubId);
        return;
    }
    {
        std::lock_guard<std::mutex> autoLock(locks_);
        if (pkgNameSet_.find(pkgName) != pkgNameSet_.end()) {
            pkgNameSet_.erase(pkgName);
        }
        if (discoveryContextMap_.find(pkgName) != discoveryContextMap_.end()) {
            discoveryContextMap_.erase(pkgName);
            if (timer_ != nullptr) {
                timer_->DeleteTimer(pkgName);
            }
        }
    }
    {
        std::lock_guard<std::mutex> capLock(capabilityMapLocks_);
        if (capabilityMap_.find(pkgName) != capabilityMap_.end()) {
            capabilityMap_.erase(pkgName);
        }
    }
    listener_->OnDiscoveryFailed(processInfo, (uint32_t)externalSubId, result);
    softbusListener_->StopRefreshSoftbusLNN(subscribeId);
}

void DiscoveryManager::StartDiscoveryTimer(const std::string &pkgName)
{
    if (timer_ == nullptr) {
        timer_ = std::make_shared<DmTimer>();
    }
    timer_->StartTimer(pkgName, DISCOVERY_TIMEOUT,
        [this] (std::string name) {
            DiscoveryManager::HandleDiscoveryTimeout(name);
        });
}

int32_t DiscoveryManager::HandleDiscoveryQueue(const std::string &pkgName, uint16_t subscribeId,
    const std::map<std::string, std::string> &filterOps)
{
    std::string filterData = "";
    if (filterOps.find(PARAM_KEY_FILTER_OPTIONS) != filterOps.end()) {
        filterData = filterOps.find(PARAM_KEY_FILTER_OPTIONS)->second;
    }
    DeviceFilterOption dmFilter;
    if ((dmFilter.TransformToFilter(filterData) != DM_OK) && (dmFilter.TransformFilterOption(filterData) != DM_OK)) {
        return ERR_DM_INPUT_PARA_INVALID;
    }
    {
        std::lock_guard<std::mutex> autoLock(locks_);
        CHECK_SIZE_RETURN(pkgNameSet_, ERR_DM_DISCOVERY_REPEATED);
        CHECK_SIZE_RETURN(discoveryContextMap_, ERR_DM_DISCOVERY_REPEATED);
        if (pkgNameSet_.find(pkgName) == pkgNameSet_.end()) {
            pkgNameSet_.emplace(pkgName);
            DiscoveryContext context = {pkgName, filterData, subscribeId, dmFilter.filterOp_, dmFilter.filters_};
            discoveryContextMap_.emplace(pkgName, context);
            return DM_OK;
        } else {
            LOGE("DiscoveryManager::HandleDiscoveryQueue repeated, pkgName : %{public}s.", pkgName.c_str());
            return ERR_DM_DISCOVERY_REPEATED;
        }
    }
}

void DiscoveryManager::HandleDiscoveryTimeout(const std::string &pkgName)
{
    LOGI("pkgName: %{public}s.", pkgName.c_str());
    uint16_t subscribeId = 0;
    {
        std::lock_guard<std::mutex> autoLock(locks_);
        if (pkgNameSet_.find(pkgName) == pkgNameSet_.end()) {
            LOGE("HandleDiscoveryTimeout: pkgName: %{public}s is not exist.", pkgName.c_str());
            return;
        }
        auto iter = discoveryContextMap_.find(pkgName);
        if (iter == discoveryContextMap_.end()) {
            LOGE("HandleDiscoveryTimeout: subscribeId not found by pkgName %{public}s.",
                GetAnonyString(pkgName).c_str());
            return;
        }
        subscribeId = discoveryContextMap_[pkgName].subscribeId;
    }
    StopDiscoveringByInnerSubId(pkgName, subscribeId);
    {
        std::lock_guard<std::mutex> autoLock(multiUserDiscLocks_);
        multiUserDiscMap_.erase(pkgName);
    }
}

void DiscoveryManager::UpdateInfoFreq(
    const std::map<std::string, std::string> &discoverParam, DmSubscribeInfo &dmSubInfo)
{
    if (auto it = discoverParam.find(PARAM_KEY_DISC_FREQ); it != discoverParam.end()) {
        int32_t freq = StringToInt(it->second, DECIMALISM);
        if (freq < DmExchangeFreq::DM_LOW || freq > DmExchangeFreq::DM_FREQ_BUTT) {
            LOGE("Invalid freq value.");
            return;
        }
        dmSubInfo.freq = static_cast<DmExchangeFreq>(freq);
    }
}

void DiscoveryManager::UpdateInfoMedium(
    const std::map<std::string, std::string> &discoverParam, DmSubscribeInfo &dmSubInfo)
{
    dmSubInfo.medium = DmExchangeMedium::DM_BLE;
    if (discoverParam.find(PARAM_KEY_DISC_MEDIUM) != discoverParam.end()) {
        int32_t medium = std::atoi((discoverParam.find(PARAM_KEY_DISC_MEDIUM)->second).c_str());
        dmSubInfo.medium = static_cast<DmExchangeMedium>(medium);
    }
}

int32_t DiscoveryManager::GetDeviceAclParam(const std::string &pkgName, int32_t userId, std::string deviceId,
    bool &isOnline, int32_t &authForm)
{
#if !(defined(__LITEOS_M__) || defined(LITE_DEVICE))
    char localDeviceId[DEVICE_UUID_LENGTH];
    GetDevUdid(localDeviceId, DEVICE_UUID_LENGTH);
    std::string requestDeviceId = static_cast<std::string>(localDeviceId);
    DmDiscoveryInfo discoveryInfo;
    discoveryInfo.pkgname = pkgName;
    discoveryInfo.localDeviceId = requestDeviceId;
    discoveryInfo.remoteDeviceIdHash = deviceId;
    discoveryInfo.userId = userId;
    if (DiscoveryManager::IsCommonDependencyReady() && DiscoveryManager::GetCommonDependencyObj() != nullptr) {
        if (DiscoveryManager::GetCommonDependencyObj()->GetDeviceAclParam(discoveryInfo, isOnline, authForm) != DM_OK) {
            LOGE("GetDeviceAclParam failed.");
            return ERR_DM_FAILED;
        }
    }
#endif
    return DM_OK;
}

#if !(defined(__LITEOS_M__) || defined(LITE_DEVICE))
IDeviceProfileConnector* DiscoveryManager::GetCommonDependencyObj()
{
    return dpConnector_;
}

bool DiscoveryManager::IsCommonDependencyReady()
{
    std::lock_guard<std::mutex> lock(comDependencyLoadLock);
    if (isSoLoaded_ && dpConnector_ != nullptr && dpConnectorHandle_ != nullptr) {
        return true;
    }
    dpConnectorHandle_ = dlopen(LIB_DM_COMDENPENDENCY_NAME, RTLD_NOW | RTLD_NODELETE | RTLD_NOLOAD);
    if (dpConnectorHandle_ == nullptr) {
        dpConnectorHandle_ = dlopen(LIB_DM_COMDENPENDENCY_NAME, RTLD_NOW | RTLD_NODELETE);
    }
    if (dpConnectorHandle_ == nullptr) {
        LOGE("load libdevicemanagerdependency so failed, errMsg: %{public}s.", dlerror());
        return false;
    }
    dlerror();
    auto func = (CreateDpConnectorFuncPtr)dlsym(dpConnectorHandle_, "CreateDpConnectorInstance");
    if (dlerror() != nullptr || func == nullptr) {
        dlclose(dpConnectorHandle_);
        LOGE("Create object function is not exist.");
        return false;
    }
    dpConnector_ = func();
    isSoLoaded_ = true;
    return true;
}

bool DiscoveryManager::CloseCommonDependencyObj()
{
    LOGI("start.");
    std::lock_guard<std::mutex> lock(comDependencyLoadLock);
    if (!isSoLoaded_ && (dpConnector_ == nullptr) && (dpConnectorHandle_ == nullptr)) {
        return true;
    }

    int32_t ret = dlclose(dpConnectorHandle_);
    if (ret != 0) {
        LOGE("close libdevicemanagerdependency failed ret = %{public}d.", ret);
        return false;
    }
    isSoLoaded_ = false;
    dpConnector_ = nullptr;
    dpConnectorHandle_ = nullptr;
    return true;
}
#endif

void DiscoveryManager::ClearDiscoveryCache(const ProcessInfo &processInfo)
{
    LOGI("PkgName: %{public}s, userId: %{public}d", processInfo.pkgName.c_str(), processInfo.userId);
    std::string pkgName = processInfo.pkgName + "#";
    std::set<uint16_t> subscribeIdSet = ClearDiscoveryPkgName(pkgName);

    CHECK_NULL_VOID(softbusListener_);
    for (auto it : subscribeIdSet) {
        std::string pkgNameTemp = (ComposeStr(ComposeStr(processInfo.pkgName, it), processInfo.userId));
        softbusListener_->UnRegisterSoftbusLnnOpsCbk(pkgNameTemp);
        uint16_t innerSubId = DM_INVALID_FLAG_ID;
        {
            std::lock_guard<std::mutex> autoLock(subIdMapLocks_);
            innerSubId = static_cast<uint16_t>(pkgName2SubIdMap_[pkgNameTemp][it]);
            randSubIdSet_.erase(innerSubId);
            pkgName2SubIdMap_.erase(pkgNameTemp);
        }
        softbusListener_->StopRefreshSoftbusLNN(innerSubId);
    }

    CHECK_NULL_VOID(timer_);
    for (auto it : subscribeIdSet) {
        std::string pkgNameTemp = (ComposeStr(ComposeStr(processInfo.pkgName, it), processInfo.userId));
        timer_->DeleteTimer(pkgNameTemp);
    }
}

std::set<uint16_t> DiscoveryManager::ClearDiscoveryPkgName(const std::string &pkgName)
{
    std::set<uint16_t> subscribeIdSet;
    {
        std::lock_guard<std::mutex> autoLock(locks_);
        for (auto it = pkgNameSet_.begin(); it != pkgNameSet_.end();) {
            if ((*it).find(pkgName) != std::string::npos) {
                LOGI("Erase pkgname %{public}s from pkgNameSet.", (*it).c_str());
                it = pkgNameSet_.erase(it);
            } else {
                ++it;
            }
        }
        for (auto it = discoveryContextMap_.begin(); it != discoveryContextMap_.end();) {
            if (it->first.find(pkgName) != std::string::npos) {
                LOGI("Erase pkgname %{public}s from discoveryContextMap_.", it->first.c_str());
                subscribeIdSet.insert(it->second.subscribeId);
                it = discoveryContextMap_.erase(it);
            } else {
                ++it;
            }
        }
    }
    {
        std::lock_guard<std::mutex> capLock(capabilityMapLocks_);
        for (auto it = capabilityMap_.begin(); it != capabilityMap_.end();) {
            if (it->first.find(pkgName) != std::string::npos) {
                LOGI("Erase pkgname %{public}s from capabilityMap_.", it->first.c_str());
                it = capabilityMap_.erase(it);
            } else {
                ++it;
            }
        }
    }
    {
        std::lock_guard<std::mutex> autoLock(subIdMapLocks_);
        for (auto it = pkgName2SubIdMap_.begin(); it != pkgName2SubIdMap_.end(); ++it) {
            if (it->first.find(pkgName) != std::string::npos) {
                LOGI("Erase pkgname %{public}s from pkgName2SubIdMap_.", it->first.c_str());
                for (auto iter : it->second) {
                    subscribeIdSet.insert(iter.first);
                }
            }
        }
    }
    return subscribeIdSet;
}

std::string DiscoveryManager::AddMultiUserIdentify(const std::string &pkgName)
{
    int32_t userId = -1;
#if !(defined(__LITEOS_M__) || defined(LITE_DEVICE))
    MultipleUserConnector::GetCallerUserId(userId);
#endif
    if (userId == -1) {
        LOGE("Get caller userId failed.");
        return pkgName;
    }
    MultiUserDiscovery multiUserDisc;
    multiUserDisc.pkgName = pkgName;
    multiUserDisc.userId = userId;
    std::string pkgNameTemp = ComposeStr(pkgName, userId);
    {
        std::lock_guard<std::mutex> autoLock(multiUserDiscLocks_);
        CHECK_SIZE_RETURN(multiUserDiscMap_, pkgNameTemp);
        multiUserDiscMap_[pkgNameTemp] = multiUserDisc;
    }
    return pkgNameTemp;
}

std::string DiscoveryManager::RemoveMultiUserIdentify(const std::string &pkgName)
{
    int32_t userId = -1;
#if !(defined(__LITEOS_M__) || defined(LITE_DEVICE))
    MultipleUserConnector::GetCallerUserId(userId);
#endif
    if (userId == -1) {
        LOGE("Get caller userId failed.");
        return pkgName;
    }
    std::string pkgNameTemp = ComposeStr(pkgName, userId);
    {
        std::lock_guard<std::mutex> autoLock(multiUserDiscLocks_);
        if (multiUserDiscMap_.find(pkgNameTemp) != multiUserDiscMap_.end()) {
            multiUserDiscMap_.erase(pkgNameTemp);
        }
    }
    return pkgNameTemp;
}

void DiscoveryManager::GetPkgNameAndUserId(const std::string &pkgName, std::string &callerPkgName,
    int32_t &userId)
{
    {
        std::lock_guard<std::mutex> autoLock(multiUserDiscLocks_);
        if (multiUserDiscMap_.find(pkgName) != multiUserDiscMap_.end()) {
            callerPkgName = GetCallerPkgName(multiUserDiscMap_[pkgName].pkgName);
            userId = multiUserDiscMap_[pkgName].userId;
            return;
        }
    }
    LOGE("find failed PkgName %{public}s.", pkgName.c_str());
}

int32_t DiscoveryManager::GenInnerSubId(const std::string &pkgName, uint16_t subId)
{
    uint16_t tempSubId = DM_INVALID_FLAG_ID;
    {
        std::lock_guard<std::mutex> autoLock(subIdMapLocks_);
        CHECK_SIZE_RETURN(pkgName2SubIdMap_, tempSubId);
        CHECK_SIZE_RETURN(pkgName2SubIdMap_[pkgName], tempSubId);
        if (pkgName2SubIdMap_[pkgName].find(subId) != pkgName2SubIdMap_[pkgName].end()) {
            return pkgName2SubIdMap_[pkgName][subId];
        }
        if (pkgName2SubIdMap_.find(pkgName) == pkgName2SubIdMap_.end()) {
            pkgName2SubIdMap_[pkgName] = std::map<uint16_t, uint16_t>();
        }
        tempSubId = GenUniqueRandUint(randSubIdSet_);
        pkgName2SubIdMap_[pkgName][subId] = tempSubId;
    }
    return tempSubId;
}

int32_t DiscoveryManager::GetAndRemoveInnerSubId(const std::string &pkgName, uint16_t subId)
{
    uint16_t tempSubId = DM_INVALID_FLAG_ID;
    {
        std::lock_guard<std::mutex> autoLock(subIdMapLocks_);
        if (pkgName2SubIdMap_[pkgName].find(subId) != pkgName2SubIdMap_[pkgName].end()) {
            tempSubId = pkgName2SubIdMap_[pkgName][subId];
            pkgName2SubIdMap_[pkgName].erase(subId);
            randSubIdSet_.erase(tempSubId);
        }
        if (pkgName2SubIdMap_[pkgName].empty()) {
            pkgName2SubIdMap_.erase(pkgName);
        }
    }
    return tempSubId;
}
} // namespace DistributedHardware
} // namespace OHOS