/*
 * Copyright (C) 2021-2022 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 "dhcp_client_stub_lite.h"
#include "dhcp_client_callback_proxy.h"
#include "dhcp_manager_service_ipc_interface_code.h"
#include "dhcp_sdk_define.h"
#ifndef  OHOS_EUPDATER
#include "ipc_skeleton.h"
#include "rpc_errno.h"
#endif
#include "dhcp_client_state_machine.h"
#include "dhcp_logger.h"
#include "dhcp_errcode.h"

#ifndef OHOS_EUPDATER
DEFINE_DHCPLOG_DHCP_LABEL("DhcpClientStub");
#endif
namespace OHOS {
namespace DHCP {
DhcpClientStub::DhcpClientStub()
{
#ifndef OHOS_EUPDATER
    InitHandleMap();
#endif
}

DhcpClientStub::~DhcpClientStub()
{}

#ifndef OHOS_EUPDATER
void DhcpClientStub::InitHandleMap()
{
    handleFuncMap[static_cast<uint32_t>(DhcpClientInterfaceCode::DHCP_CLIENT_SVR_CMD_REG_CALL_BACK)] = &DhcpClientStub::OnRegisterCallBack;
    handleFuncMap[static_cast<uint32_t>(DhcpClientInterfaceCode::DHCP_CLIENT_SVR_CMD_START_DHCP_CLIENT)] = &DhcpClientStub::OnStartDhcpClient;
    handleFuncMap[static_cast<uint32_t>(DhcpClientInterfaceCode::DHCP_CLIENT_SVR_CMD_STOP_DHCP_CLIENT)] = &DhcpClientStub::OnStopDhcpClient;
    handleFuncMap[static_cast<uint32_t>(DhcpClientInterfaceCode::DHCP_CLIENT_SVR_CMD_SET_CONFIG)] =
        &DhcpClientStub::OnSetConfiguration;
}

int DhcpClientStub::OnRemoteRequest(uint32_t code, IpcIo *req, IpcIo *reply)
{
    DHCP_LOGI("run: %{public}s code: %{public}u L1", __func__, code);
    if (req == nullptr || reply == nullptr) {
        DHCP_LOGD("req:%{public}d, reply:%{public}d", req == nullptr, reply == nullptr);
        return DHCP_OPT_FAILED;
    }

    DHCP_LOGI("run ReadInterfaceToken L1 code %{public}u", code);
    size_t length;
    uint16_t* interfaceRead = nullptr;
    interfaceRead = ReadInterfaceToken(req, &length);
    for (size_t i = 0; i < length; i++) {
        if (i >= DECLARE_INTERFACE_DESCRIPTOR_L1_LENGTH || interfaceRead[i] != DECLARE_INTERFACE_DESCRIPTOR_L1[i]) {
            DHCP_LOGE("Sta stub token verification error: %{public}d", code);
            return DHCP_OPT_FAILED;
        }
    }

    int exception = 0;
    (void)ReadInt32(req, &exception);
    if (exception) {
        (void)WriteInt32(reply, 0);
        (void)WriteInt32(reply, DHCP_OPT_NOT_SUPPORTED);
        return DHCP_OPT_FAILED;
    }

    HandleFuncMap::iterator iter = handleFuncMap.find(code);
    if (iter == handleFuncMap.end()) {
        DHCP_LOGE("not find function to deal, code %{public}u", code);
        (void)WriteInt32(reply, 0);
        (void)WriteInt32(reply, DHCP_OPT_NOT_SUPPORTED);
    } else {
        (this->*(iter->second))(code, req, reply);
    }
    return DHCP_OPT_SUCCESS;
}

int DhcpClientStub::OnRegisterCallBack(uint32_t code, IpcIo *req, IpcIo *reply)
{
    DHCP_LOGI("run %{public}s code %{public}u", __func__, code);
    ErrCode ret = DHCP_E_FAILED;
    SvcIdentity sid;
    bool readSid = ReadRemoteObject(req, &sid);
    if (!readSid) {
        DHCP_LOGE("read SvcIdentity failed");
        (void)WriteInt32(reply, 0);
        (void)WriteInt32(reply, ret);
        return DHCP_OPT_FAILED;
    }

    std::shared_ptr<IDhcpClientCallBack> callback_ = std::make_shared<DhcpClientCallbackProxy>(&sid);
    DHCP_LOGI("create new DhcpClientCallbackProxy!");

    size_t readLen;
    std::string ifname = (char *)ReadString(req, &readLen);
    DHCP_LOGI("ifname:%{public}s", ifname.c_str());

    ret = RegisterDhcpClientCallBack(ifname, callback_);
    (void)WriteInt32(reply, 0);
    (void)WriteInt32(reply, ret);
    return DHCP_OPT_SUCCESS;
}

int DhcpClientStub::OnStartDhcpClient(uint32_t code, IpcIo *req, IpcIo *reply)
{
    DHCP_LOGI("run %{public}s code %{public}u", __func__, code);
    ErrCode ret = DHCP_E_FAILED;
    SvcIdentity sid;
    bool readSid = ReadRemoteObject(req, &sid);
    if (!readSid) {
        DHCP_LOGE("read SvcIdentity failed");
        (void)WriteInt32(reply, 0);
        (void)WriteInt32(reply, ret);
        return DHCP_OPT_FAILED;
    }

    size_t readLen;
    bool bIpv6;
    std::string ifname = (char *)ReadString(req, &readLen);
    (void)ReadBool(req, &bIpv6);
    DHCP_LOGI("ifname:%{public}s bIpv6:%{public}d", ifname.c_str(), bIpv6);

    ret = StartDhcpClient(ifname, bIpv6);
    (void)WriteInt32(reply, 0);
    (void)WriteInt32(reply, ret);
    return DHCP_OPT_SUCCESS;
}

int DhcpClientStub::OnSetConfiguration(uint32_t code, IpcIo *req, IpcIo *reply)
{
    DHCP_LOGI("run %{public}s code %{public}u", __func__, code);
    ErrCode ret = DHCP_E_FAILED;
    SvcIdentity sid;
    bool readSid = ReadRemoteObject(req, &sid);
    if (!readSid) {
        DHCP_LOGE("read SvcIdentity failed");
        (void)WriteInt32(reply, 0);
        (void)WriteInt32(reply, ret);
        return DHCP_OPT_FAILED;
    }

    size_t readLen;
    std::string ifname = (char *)ReadString(req, &readLen);
    RouterConfig config;
    config.bssid = (char *)ReadString(req, &readLen);
    DHCP_LOGI("ifname:%{public}s", ifname.c_str());

    ret = SetConfiguration(ifname, config);
    (void)WriteInt32(reply, 0);
    (void)WriteInt32(reply, ret);
    return DHCP_OPT_SUCCESS;
}
int DhcpClientStub::OnStopDhcpClient(uint32_t code, IpcIo *req, IpcIo *reply)
{
    DHCP_LOGI("run %{public}s code %{public}u", __func__, code);
    ErrCode ret = DHCP_E_FAILED;
    SvcIdentity sid;
    bool readSid = ReadRemoteObject(req, &sid);
    if (!readSid) {
        DHCP_LOGE("read SvcIdentity failed");
        (void)WriteInt32(reply, 0);
        (void)WriteInt32(reply, ret);
        return DHCP_OPT_FAILED;
    }

    size_t readLen;
    bool bIpv6;
    std::string ifname = (char *)ReadString(req, &readLen);
    (void)ReadBool(req, &bIpv6);
    DHCP_LOGI("ifname:%{public}s bIpv6:%{public}d", ifname.c_str(), bIpv6);

    ret = StopDhcpClient(ifname, bIpv6);
    (void)WriteInt32(reply, 0);
    (void)WriteInt32(reply, ret);
    return 0;
}
#endif
}
}