/*
 * 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 "browser_proxy.h"

#include "edm_constants.h"
#include "edm_log.h"
#include "func_code.h"
#include "securec.h"

namespace OHOS {
namespace EDM {
std::shared_ptr<BrowserProxy> BrowserProxy::instance_ = nullptr;
std::once_flag BrowserProxy::flag_;
const std::u16string DESCRIPTOR = u"ohos.edm.IEnterpriseDeviceMgr";
constexpr int32_t MAX_POLICY_FILE_SIZE = 134217728; // 128 * 1024 * 1024

std::shared_ptr<BrowserProxy> BrowserProxy::GetBrowserProxy()
{
    std::call_once(flag_, []() {
        if (instance_ == nullptr) {
            instance_ = std::make_shared<BrowserProxy>();
        }
    });
    return instance_;
}

int32_t BrowserProxy::GetPolicies(std::string &policies)
{
    EDMLOGD("BrowserProxy::GetPolicies inner api");
    if (!EnterpriseDeviceMgrProxy::GetInstance()->IsEdmEnabled()) {
        EDMLOGD("BrowserProxy::GetPolicies edm service not start.");
        policies = "";
        return ERR_OK;
    }
    return GetPolicies(nullptr, "", policies);
}

int32_t BrowserProxy::GetPolicies(AppExecFwk::ElementName &admin, const std::string &appId, std::string &policies)
{
    EDMLOGD("BrowserProxy::GetPolicies");
    if (appId.empty()) {
        EDMLOGE("BrowserProxy::GetPolicies appId is empty.");
        return EdmReturnErrCode::PARAM_ERROR;
    }
    return GetPolicies(&admin, appId, policies);
}

int32_t BrowserProxy::GetPolicies(AppExecFwk::ElementName *admin, const std::string &appId, std::string &policies)
{
    MessageParcel data;
    MessageParcel reply;
    data.WriteInterfaceToken(DESCRIPTOR);
    data.WriteInt32(WITHOUT_USERID);
    data.WriteString(WITHOUT_PERMISSION_TAG);
    if (admin == nullptr) {
        data.WriteInt32(WITHOUT_ADMIN);
    } else {
        data.WriteInt32(HAS_ADMIN);
        data.WriteParcelable(admin);
    }
    data.WriteString(appId);
    EnterpriseDeviceMgrProxy::GetInstance()->GetPolicy(EdmInterfaceCode::SET_BROWSER_POLICIES, data, reply);
    int32_t ret = ERR_INVALID_VALUE;
    bool blRes = reply.ReadInt32(ret) && (ret == ERR_OK);
    if (!blRes) {
        EDMLOGW("EnterpriseDeviceMgrProxy:GetPolicy fail. %{public}d", ret);
        return ret;
    }
    reply.ReadString(policies);
    return ERR_OK;
}

int32_t BrowserProxy::SetPolicy(const AppExecFwk::ElementName &admin, const std::string &appId,
    const std::string &policyName, const std::string &policyValue)
{
    EDMLOGI("BrowserProxy::SetPolicy");
    if (appId.empty()) {
        EDMLOGE("BrowserProxy::SetPolicy appId is empty");
        return EdmReturnErrCode::PARAM_ERROR;
    }
    MessageParcel data;
    std::uint32_t funcCode =
        POLICY_FUNC_CODE((std::uint32_t)FuncOperateType::SET, EdmInterfaceCode::SET_BROWSER_POLICIES);
    data.WriteInterfaceToken(DESCRIPTOR);
    data.WriteInt32(WITHOUT_USERID);
    data.WriteParcelable(&admin);
    data.WriteString(WITHOUT_PERMISSION_TAG);
    std::vector<std::string> params{appId, policyName, policyValue};
    data.WriteStringVector(params);
    return EnterpriseDeviceMgrProxy::GetInstance()->HandleDevicePolicy(funcCode, data);
}

int32_t BrowserProxy::SetManagedBrowserPolicy(MessageParcel &data)
{
    EDMLOGI("BrowserProxy::SetManagedBrowserPolicy");
    std::uint32_t funcCode =
        POLICY_FUNC_CODE((std::uint32_t)FuncOperateType::SET, EdmInterfaceCode::MANAGED_BROWSER_POLICY);
    return EnterpriseDeviceMgrProxy::GetInstance()->HandleDevicePolicy(funcCode, data);
}

int32_t BrowserProxy::GetManagedBrowserPolicy(MessageParcel &data, void** rawData, int32_t &size)
{
    MessageParcel reply;
    EnterpriseDeviceMgrProxy::GetInstance()->GetPolicy(EdmInterfaceCode::MANAGED_BROWSER_POLICY, data, reply);
    int32_t ret = ERR_INVALID_VALUE;
    bool blRes = reply.ReadInt32(ret) && (ret == ERR_OK);
    if (!blRes) {
        EDMLOGW("EnterpriseDeviceMgrProxy:GetManagedBrowserPolicy fail. %{public}d", ret);
        return ret;
    }
    return GetRawData(reply, rawData, size);
}

int32_t BrowserProxy::GetSelfManagedBrowserPolicyVersion(int32_t &version)
{
    MessageParcel data;
    MessageParcel reply;
    data.WriteInterfaceToken(DESCRIPTOR);
    data.WriteInt32(WITHOUT_USERID);
    data.WriteString(WITHOUT_PERMISSION_TAG);
    data.WriteInt32(WITHOUT_ADMIN);
    data.WriteString(EdmConstants::Browser::GET_MANAGED_BROWSER_VERSION);
    EnterpriseDeviceMgrProxy::GetInstance()->GetPolicy(EdmInterfaceCode::MANAGED_BROWSER_POLICY, data, reply);
    int32_t ret = ERR_INVALID_VALUE;
    bool blRes = reply.ReadInt32(ret) && (ret == ERR_OK);
    if (!blRes) {
        EDMLOGW("EnterpriseDeviceMgrProxy:GetSelfManagedBrowserPolicyVersion fail. %{public}d", ret);
        return ret;
    }
    reply.ReadInt32(version);
    return ERR_OK;
}

int32_t BrowserProxy::GetSelfManagedBrowserPolicy(void** rawData, int32_t &size)
{
    MessageParcel data;
    MessageParcel reply;
    data.WriteInterfaceToken(DESCRIPTOR);
    data.WriteInt32(WITHOUT_USERID);
    data.WriteString(WITHOUT_PERMISSION_TAG);
    data.WriteInt32(WITHOUT_ADMIN);
    data.WriteString(EdmConstants::Browser::GET_SELF_MANAGED_BROWSER_FILE_DATA);
    EnterpriseDeviceMgrProxy::GetInstance()->GetPolicy(EdmInterfaceCode::MANAGED_BROWSER_POLICY, data, reply);
    int32_t ret = ERR_INVALID_VALUE;
    bool blRes = reply.ReadInt32(ret) && (ret == ERR_OK);
    if (!blRes) {
        EDMLOGW("EnterpriseDeviceMgrProxy:GetSelfManagedBrowserPolicy fail. %{public}d", ret);
        return ret;
    }
    return GetRawData(reply, rawData, size);
}

int32_t BrowserProxy::GetRawData(MessageParcel &reply, void** rawData, int32_t &size)
{
    reply.ReadInt32(size);
    if (size < 0 || size >= MAX_POLICY_FILE_SIZE) {
        EDMLOGE("BrowserProxy::GetRawData fileData.size error.size: %{public}d",
            size);
        return EdmReturnErrCode::SYSTEM_ABNORMALLY;
    }
    void* tempData = const_cast<void*>(reply.ReadRawData(size));
    if (tempData == nullptr) {
        EDMLOGW("EnterpriseDeviceMgrProxy:GetRawData ReadRawData fail.");
        return EdmReturnErrCode::SYSTEM_ABNORMALLY;
    }
    *rawData = malloc(size);
    if (*rawData == nullptr) {
        EDMLOGW("EnterpriseDeviceMgrProxy:GetRawData malloc fail.");
        return EdmReturnErrCode::SYSTEM_ABNORMALLY;
    }
    errno_t cpRet = memcpy_s(*rawData, size, tempData, size);
    if (cpRet != EOK) {
        EDMLOGW("EnterpriseDeviceMgrProxy:GetRawData memcpy fail.ret:%{public}d", cpRet);
        free(*rawData);
        *rawData = nullptr;
        return EdmReturnErrCode::SYSTEM_ABNORMALLY;
    }
    return ERR_OK;
}
} // namespace EDM
} // namespace OHOS