OpenHarmony-v5.0.1-Release/s

/*
 * Copyright (C) 2023 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 "napi_bluetooth_gatt_server.h"
#include "bluetooth_gatt_service.h"
#include "bluetooth_host.h"
#include "bluetooth_log.h"
#include "bluetooth_utils.h"
#include "napi_bluetooth_ble.h"
#include "napi_bluetooth_ble_utils.h"
#include "napi_bluetooth_error.h"
#include "napi_bluetooth_utils.h"
#include "napi_event_subscribe_module.h"
#include "../parser/napi_parser_utils.h"

namespace OHOS {
namespace Bluetooth {
using namespace std;

std::vector<std::string> NapiGattServer::deviceList_;
std::mutex NapiGattServer::deviceListMutex_;
thread_local napi_ref NapiGattServer::consRef_ = nullptr;

napi_value NapiGattServer::CreateGattServer(napi_env env, napi_callback_info info)
{
    HILOGI("enter");
    napi_value result;
    napi_value constructor = nullptr;
    napi_get_reference_value(env, consRef_, &constructor);
    napi_new_instance(env, constructor, 0, nullptr, &result);

    return result;
}

void NapiGattServer::DefineGattServerJSClass(napi_env env)
{
    napi_property_descriptor gattserverDesc[] = {
#ifdef BLUETOOTH_API_SINCE_10
        DECLARE_NAPI_FUNCTION("notifyCharacteristicChanged", NotifyCharacteristicChangedEx),
#else
        DECLARE_NAPI_FUNCTION("startAdvertising", StartAdvertising),
        DECLARE_NAPI_FUNCTION("stopAdvertising", StopAdvertising),
        DECLARE_NAPI_FUNCTION("notifyCharacteristicChanged", NotifyCharacteristicChanged),
#endif
        DECLARE_NAPI_FUNCTION("addService", AddService),
        DECLARE_NAPI_FUNCTION("removeService", RemoveGattService),
        DECLARE_NAPI_FUNCTION("close", Close),
        DECLARE_NAPI_FUNCTION("sendResponse", SendResponse),
        DECLARE_NAPI_FUNCTION("on", On),
        DECLARE_NAPI_FUNCTION("off", Off),
    };

    napi_value constructor = nullptr;
    napi_define_class(env, "GattServer", NAPI_AUTO_LENGTH, GattServerConstructor, nullptr,
        sizeof(gattserverDesc) / sizeof(gattserverDesc[0]), gattserverDesc, &constructor);
    napi_create_reference(env, constructor, 1, &consRef_);
}

napi_value NapiGattServer::GattServerConstructor(napi_env env, napi_callback_info info)
{
    napi_value thisVar = nullptr;
    napi_get_cb_info(env, info, nullptr, nullptr, &thisVar, nullptr);
    NapiGattServer* gattServer = new NapiGattServer();

    napi_wrap(
        env, thisVar, gattServer,
        [](napi_env env, void* data, void* hint) {
            NapiGattServer* server = static_cast<NapiGattServer*>(data);
            if (server) {
                delete server;
                server = nullptr;
            }
        },
        nullptr,
        nullptr);

    return thisVar;
}

static NapiGattServer *NapiGetGattServer(napi_env env, napi_value thisVar)
{
    NapiGattServer *gattServer = nullptr;
    auto status = napi_unwrap(env, thisVar, (void **)&gattServer);
    if (status != napi_ok) {
        return nullptr;
    }
    return gattServer;
}

static NapiGattServer *NapiGetGattServer(napi_env env, napi_callback_info info)
{
    size_t argc = 0;
    napi_value thisVar = nullptr;
    if (napi_get_cb_info(env, info, &argc, nullptr, &thisVar, nullptr) != napi_ok) {
        return nullptr;
    }
    return NapiGetGattServer(env, thisVar);
}

napi_value NapiGattServer::On(napi_env env, napi_callback_info info)
{
    NapiGattServer *napiGattServer = NapiGetGattServer(env, info);
    if (napiGattServer && napiGattServer->GetCallback()) {
        auto status = napiGattServer->GetCallback()->eventSubscribe_.Register(env, info);
        NAPI_BT_ASSERT_RETURN_UNDEF(env, status == napi_ok, BT_ERR_INVALID_PARAM);
    }
    return NapiGetUndefinedRet(env);
}

napi_value NapiGattServer::Off(napi_env env, napi_callback_info info)
{
    NapiGattServer *napiGattServer = NapiGetGattServer(env, info);
    if (napiGattServer && napiGattServer->GetCallback()) {
        auto status = napiGattServer->GetCallback()->eventSubscribe_.Deregister(env, info);
        NAPI_BT_ASSERT_RETURN_UNDEF(env, status == napi_ok, BT_ERR_INVALID_PARAM);
    }
    return NapiGetUndefinedRet(env);
}

static napi_status CheckGattsAddService(napi_env env, napi_callback_info info, std::shared_ptr<GattServer> &outServer,
    std::unique_ptr<GattService> &outService)
{
    size_t argc = ARGS_SIZE_ONE;
    napi_value argv[ARGS_SIZE_ONE] = {nullptr};
    napi_value thisVar = nullptr;
    NAPI_BT_CALL_RETURN(napi_get_cb_info(env, info, &argc, argv, &thisVar, NULL));
    NAPI_BT_RETURN_IF(argc != ARGS_SIZE_ONE, "Requires 1 arguments.", napi_invalid_arg);

    // std::unique_ptr<GattService> service {nullptr};
    NapiGattService napiGattService;
    NAPI_BT_CALL_RETURN(NapiParseGattService(env, argv[PARAM0], napiGattService));

    NapiGattServer *gattServer = NapiGetGattServer(env, thisVar);
    NAPI_BT_RETURN_IF(gattServer == nullptr, "gattServer is nullptr.", napi_invalid_arg);
    outServer = gattServer->GetServer();

    GattServiceType type = napiGattService.isPrimary ? GattServiceType::PRIMARY : GattServiceType::SECONDARY;
    outService = std::make_unique<GattService>(napiGattService.serviceUuid, type);
    for (const auto &napiCharacter : napiGattService.characteristics) {
        int charPermissions = napiCharacter.permissions;
        int charProperties = napiCharacter.properties;
        GattCharacteristic character(napiCharacter.characteristicUuid, charPermissions, charProperties);
        character.SetValue(napiCharacter.characteristicValue.data(), napiCharacter.characteristicValue.size());

        for (const auto &napiDescriptor : napiCharacter.descriptors) {
            GattDescriptor descriptor(napiDescriptor.descriptorUuid, napiDescriptor.permissions);
            descriptor.SetValue(napiDescriptor.descriptorValue.data(), napiDescriptor.descriptorValue.size());
            character.AddDescriptor(descriptor);
        }
        outService->AddCharacteristic(character);
    }

    return napi_ok;
}

napi_value NapiGattServer::AddService(napi_env env, napi_callback_info info)
{
    HILOGI("enter");
    std::shared_ptr<GattServer> server {nullptr};
    std::unique_ptr<GattService> gattService {nullptr};
    auto status = CheckGattsAddService(env, info, server, gattService);
    NAPI_BT_ASSERT_RETURN_FALSE(env, (status == napi_ok && server != nullptr), BT_ERR_INVALID_PARAM);

    int ret = server->AddService(*gattService);
    NAPI_BT_ASSERT_RETURN_FALSE(env, ret == BT_NO_ERROR, ret);
    return NapiGetBooleanTrue(env);
}

static napi_status CheckGattsClose(napi_env env, napi_callback_info info, std::shared_ptr<GattServer> &outServer)
{
    size_t argc = 0;
    napi_value thisVar = nullptr;
    NAPI_BT_CALL_RETURN(napi_get_cb_info(env, info, &argc, nullptr, &thisVar, NULL));
    NAPI_BT_RETURN_IF(argc > 0, "no needed arguments.", napi_invalid_arg);
    NapiGattServer *gattServer = NapiGetGattServer(env, thisVar);
    NAPI_BT_RETURN_IF(gattServer == nullptr, "gattServer is nullptr.", napi_invalid_arg);

    outServer = gattServer->GetServer();
    return napi_ok;
}

napi_value NapiGattServer::Close(napi_env env, napi_callback_info info)
{
    HILOGI("enter");
    std::shared_ptr<GattServer> server {nullptr};
    auto status = CheckGattsClose(env, info, server);
    NAPI_BT_ASSERT_RETURN_UNDEF(env, (status == napi_ok && server != nullptr), BT_ERR_INVALID_PARAM);

    int ret = server->Close();
    NAPI_BT_ASSERT_RETURN_UNDEF(env, ret == BT_NO_ERROR, ret);
    return NapiGetUndefinedRet(env);
}

static napi_status CheckGattsRemoveService(napi_env env, napi_callback_info info,
    std::shared_ptr<GattServer> &outServer, UUID &outUuid)
{
    size_t argc = 1;
    napi_value argv[1] = {nullptr};
    napi_value thisVar = nullptr;
    NAPI_BT_CALL_RETURN(napi_get_cb_info(env, info, &argc, argv, &thisVar, NULL));
    NAPI_BT_RETURN_IF(argc != 1, "Requires 1 arguments.", napi_invalid_arg);

    std::string uuid {};
    NAPI_BT_CALL_RETURN(NapiParseUuid(env, argv[0], uuid));

    NapiGattServer *gattServer = NapiGetGattServer(env, thisVar);
    NAPI_BT_RETURN_IF(gattServer == nullptr, "gattServer is nullptr.", napi_invalid_arg);
    outServer = gattServer->GetServer();
    outUuid = UUID::FromString(uuid);
    return napi_ok;
}

napi_value NapiGattServer::RemoveGattService(napi_env env, napi_callback_info info)
{
    HILOGI("enter");
    std::shared_ptr<GattServer> server {nullptr};
    UUID serviceUuid;
    auto status = CheckGattsRemoveService(env, info, server, serviceUuid);
    NAPI_BT_ASSERT_RETURN_FALSE(env, (status == napi_ok && server != nullptr), BT_ERR_INVALID_PARAM);

    int ret = BT_ERR_INTERNAL_ERROR;
    auto primaryService = server->GetService(serviceUuid, true);
    if (primaryService.has_value()) {
        ret = server->RemoveGattService(*primaryService);
        NAPI_BT_ASSERT_RETURN_FALSE(env, ret == BT_NO_ERROR, ret);
    }
    auto secondService = server->GetService(serviceUuid, false);
    if (secondService.has_value()) {
        ret = server->RemoveGattService(*secondService);
        NAPI_BT_ASSERT_RETURN_FALSE(env, ret == BT_NO_ERROR, ret);
    }
    NAPI_BT_ASSERT_RETURN_FALSE(env, (primaryService.has_value() || secondService.has_value()), BT_ERR_INVALID_PARAM);
    return NapiGetBooleanRet(env, ret == BT_NO_ERROR);
}

static napi_status CheckGattsSendRsp(napi_env env, napi_callback_info info, std::shared_ptr<GattServer> &outServer,
    NapiGattsServerResponse &outRsp)
{
    size_t argc = 1;
    napi_value argv[1] = {nullptr};
    napi_value thisVar = nullptr;
    NAPI_BT_CALL_RETURN(napi_get_cb_info(env, info, &argc, argv, &thisVar, NULL));
    NAPI_BT_RETURN_IF(argc != 1, "Requires 1 arguments.", napi_invalid_arg);

    NapiGattsServerResponse rsp;
    NAPI_BT_CALL_RETURN(NapiParseGattsServerResponse(env, argv[0], rsp));

    NapiGattServer *gattServer = NapiGetGattServer(env, thisVar);
    NAPI_BT_RETURN_IF(gattServer == nullptr, "gattServer is nullptr.", napi_invalid_arg);
    outServer = gattServer->GetServer();
    outRsp = std::move(rsp);
    return napi_ok;
}

napi_value NapiGattServer::SendResponse(napi_env env, napi_callback_info info)
{
    HILOGI("enter");
    std::shared_ptr<GattServer> server {nullptr};
    NapiGattsServerResponse rsp;
    auto status = CheckGattsSendRsp(env, info, server, rsp);
    NAPI_BT_ASSERT_RETURN_FALSE(env, (status == napi_ok && server != nullptr), BT_ERR_INVALID_PARAM);

    BluetoothRemoteDevice remoteDevice(rsp.deviceId, BTTransport::ADAPTER_BLE);
    HILOGI("Remote device address: %{public}s", GET_ENCRYPT_ADDR(remoteDevice));
    int ret = server->SendResponse(remoteDevice, rsp.transId, rsp.status, rsp.offset, rsp.value.data(),
        static_cast<int>(rsp.value.size()));
    NAPI_BT_ASSERT_RETURN_FALSE(env, ret == BT_NO_ERROR, ret);
    return NapiGetBooleanTrue(env);
}

static GattCharacteristic *GetGattCharacteristic(const std::shared_ptr<GattServer> &server, const UUID &serviceUuid,
    const UUID &characterUuid)
{
    auto service = server->GetService(serviceUuid, true);
    if (!service.has_value()) {
        service = server->GetService(serviceUuid, false);
    }
    if (!service.has_value()) {
        HILOGE("not found service uuid: %{public}s", serviceUuid.ToString().c_str());
        return nullptr;
    }
    GattCharacteristic *character = service.value().get().GetCharacteristic(characterUuid);
    return character;
}

#ifdef BLUETOOTH_API_SINCE_10
static napi_status CheckNotifyCharacteristicChangedEx(napi_env env, napi_callback_info info,
    NapiGattServer **outServer, std::string &outDeviceId, NapiNotifyCharacteristic &outCharacter)
{
    size_t argc = ARGS_SIZE_THREE;
    napi_value argv[ARGS_SIZE_THREE] = {0};
    napi_value thisVar = nullptr;
    NAPI_BT_CALL_RETURN(napi_get_cb_info(env, info, &argc, argv, &thisVar, nullptr));
    NAPI_BT_RETURN_IF(argc != ARGS_SIZE_TWO && argc != ARGS_SIZE_THREE, "Requires 2 or 3 arguments.", napi_invalid_arg);

    std::string deviceId {};
    NapiNotifyCharacteristic character;
    NAPI_BT_CALL_RETURN(NapiParseBdAddr(env, argv[PARAM0], deviceId));
    NAPI_BT_CALL_RETURN(NapiParseNotifyCharacteristic(env, argv[PARAM1], character));

    NapiGattServer *gattServer = NapiGetGattServer(env, thisVar);
    NAPI_BT_RETURN_IF(gattServer == nullptr || outServer ==nullptr, "gattServer is nullptr.", napi_invalid_arg);
    *outServer = gattServer;
    outDeviceId = std::move(deviceId);
    outCharacter = std::move(character);
    return napi_ok;
}

napi_value NapiGattServer::NotifyCharacteristicChangedEx(napi_env env, napi_callback_info info)
{
    HILOGI("enter");
    NapiGattServer* napiServer = nullptr;
    std::string deviceId {};
    NapiNotifyCharacteristic notifyCharacter;
    auto status = CheckNotifyCharacteristicChangedEx(env, info, &napiServer, deviceId, notifyCharacter);
    NAPI_BT_ASSERT_RETURN_FALSE(env, (status == napi_ok && napiServer && napiServer->GetServer()),
        BT_ERR_INVALID_PARAM);

    auto func = [server = napiServer->GetServer(), notifyCharacter, deviceId]() {
        int ret = BT_ERR_INTERNAL_ERROR;
        auto character = GetGattCharacteristic(server, notifyCharacter.serviceUuid, notifyCharacter.characterUuid);
        if (character == nullptr) {
            HILOGI("character is null!");
            return NapiAsyncWorkRet(ret);
        }
        character->SetValue(notifyCharacter.characterValue.data(), notifyCharacter.characterValue.size());
        BluetoothRemoteDevice remoteDevice(deviceId, BTTransport::ADAPTER_BLE);
        ret = server->NotifyCharacteristicChanged(remoteDevice, *character, notifyCharacter.confirm);
        return NapiAsyncWorkRet(ret);
    };
    auto asyncWork = NapiAsyncWorkFactory::CreateAsyncWork(env, info, func, ASYNC_WORK_NEED_CALLBACK);
    NAPI_BT_ASSERT_RETURN_UNDEF(env, asyncWork, BT_ERR_INTERNAL_ERROR);

    bool success = napiServer->GetCallback()->asyncWorkMap_.TryPush(NapiAsyncType::GATT_SERVER_NOTIFY_CHARACTERISTIC,
        asyncWork);
    NAPI_BT_ASSERT_RETURN_UNDEF(env, success, BT_ERR_INTERNAL_ERROR);

    asyncWork->Run();
    return asyncWork->GetRet();
}
#else
static napi_status CheckGattsNotify(napi_env env, napi_callback_info info, std::shared_ptr<GattServer> &outServer,
    std::string &outDeviceId, NapiNotifyCharacteristic &outCharacter)
{
    size_t argc = ARGS_SIZE_TWO;
    napi_value argv[ARGS_SIZE_TWO] = {nullptr};
    napi_value thisVar = nullptr;
    NAPI_BT_CALL_RETURN(napi_get_cb_info(env, info, &argc, argv, &thisVar, NULL));
    NAPI_BT_RETURN_IF(argc != ARGS_SIZE_TWO, "Requires 2 arguments.", napi_invalid_arg);

    std::string deviceId {};
    NapiNotifyCharacteristic character;
    NAPI_BT_CALL_RETURN(NapiParseBdAddr(env, argv[PARAM0], deviceId));
    NAPI_BT_CALL_RETURN(NapiParseNotifyCharacteristic(env, argv[PARAM1], character));

    NapiGattServer *gattServer = NapiGetGattServer(env, thisVar);
    NAPI_BT_RETURN_IF(gattServer == nullptr, "gattServer is nullptr.", napi_invalid_arg);
    outServer = gattServer->GetServer();
    outDeviceId = std::move(deviceId);
    outCharacter = std::move(character);
    return napi_ok;
}

napi_value NapiGattServer::NotifyCharacteristicChanged(napi_env env, napi_callback_info info)
{
    HILOGI("enter");
    std::shared_ptr<GattServer> server {nullptr};
    std::string deviceId {};
    NapiNotifyCharacteristic notifyCharacter;
    auto status = CheckGattsNotify(env, info, server, deviceId, notifyCharacter);
    NAPI_BT_ASSERT_RETURN_FALSE(env, (status == napi_ok && server != nullptr), BT_ERR_INVALID_PARAM);

    auto character = GetGattCharacteristic(server, notifyCharacter.serviceUuid, notifyCharacter.characterUuid);
    NAPI_BT_ASSERT_RETURN_FALSE(env, character != nullptr, BT_ERR_INVALID_PARAM);
    character->SetValue(notifyCharacter.characterValue.data(), notifyCharacter.characterValue.size());

    BluetoothRemoteDevice remoteDevice(deviceId, BTTransport::ADAPTER_BLE);
    int ret = server->NotifyCharacteristicChanged(remoteDevice, *character, notifyCharacter.confirm);
    NAPI_BT_ASSERT_RETURN_FALSE(env, ret == BT_NO_ERROR, ret);
    return NapiGetBooleanTrue(env);
}
#endif
} // namespace Bluetooth
} // namespace OHOS