src/utils/ipc_utils.cpp

/*
 * Copyright (c) 2023-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 "ipc_utils.h"
#include "dp_subscribe_info.h"
#include "macro_utils.h"

namespace OHOS {
namespace DistributedDeviceProfile {
namespace {
    const std::string TAG = "IpcUtils";
}
bool IpcUtils::Marshalling(MessageParcel& parcel, const std::vector<TrustDeviceProfile>& trustDeviceProfiles)
{
    uint32_t size = trustDeviceProfiles.size();
    WRITE_HELPER_RET(parcel, Uint32, size, false);
    if (trustDeviceProfiles.empty() || trustDeviceProfiles.size() > MAX_PROFILE_SIZE) {
        HILOGE("profile size is invalid!size : %{public}zu", trustDeviceProfiles.size());
        return false;
    }
    for (const auto& profile : trustDeviceProfiles) {
        if (!profile.Marshalling(parcel)) {
            HILOGE("profile Marshalling fail!");
            return false;
        }
    }
    return true;
}

bool IpcUtils::Marshalling(MessageParcel& parcel, const std::vector<AccessControlProfile>& aclProfiles)
{
    uint32_t size = aclProfiles.size();
    WRITE_HELPER_RET(parcel, Uint32, size, false);
    if (aclProfiles.empty() || aclProfiles.size() > MAX_PROFILE_SIZE) {
        HILOGE("profile size is invalid!size : %{public}zu", aclProfiles.size());
        return false;
    }
    for (const auto& profile : aclProfiles) {
        if (!profile.Marshalling(parcel)) {
            HILOGE("profile Marshalling fail!");
            return false;
        }
    }
    return true;
}

bool IpcUtils::Marshalling(MessageParcel& parcel, const std::vector<DeviceProfile>& deviceProfiles)
{
    uint32_t size = deviceProfiles.size();
    WRITE_HELPER_RET(parcel, Uint32, size, false);
    if (deviceProfiles.empty() || deviceProfiles.size() > MAX_PROFILE_SIZE) {
        HILOGE("profile size is invalid!size : %{public}zu", deviceProfiles.size());
        return false;
    }
    for (const auto& profile : deviceProfiles) {
        if (!profile.Marshalling(parcel)) {
            HILOGE("profile Marshalling fail!");
            return false;
        }
    }
    return true;
}

bool IpcUtils::Marshalling(MessageParcel& parcel, const std::vector<ProductInfo>& productInfos)
{
    uint32_t size = productInfos.size();
    WRITE_HELPER_RET(parcel, Uint32, size, false);
    if (productInfos.empty() || productInfos.size() > MAX_PROFILE_SIZE) {
        HILOGE("profile size is invalid!size : %{public}zu", productInfos.size());
        return false;
    }
    for (const auto& productInfo : productInfos) {
        if (!productInfo.Marshalling(parcel)) {
            HILOGE("productInfo Marshalling fail!");
            return false;
        }
    }
    return true;
}

bool IpcUtils::Marshalling(MessageParcel& parcel, const std::vector<ServiceProfile>& serviceProfiles)
{
    uint32_t size = serviceProfiles.size();
    WRITE_HELPER_RET(parcel, Uint32, size, false);
    if (serviceProfiles.empty() || serviceProfiles.size() > MAX_PROFILE_SIZE) {
        HILOGE("profile size is invalid!size : %{public}zu", serviceProfiles.size());
        return false;
    }
    for (const auto& profile : serviceProfiles) {
        if (!profile.Marshalling(parcel)) {
            HILOGE("profile Marshalling fail!");
            return false;
        }
    }
    return true;
}

bool IpcUtils::Marshalling(MessageParcel& parcel, const std::vector<CharacteristicProfile>& charProfiles)
{
    uint32_t size = charProfiles.size();
    WRITE_HELPER_RET(parcel, Uint32, size, false);
    if (charProfiles.empty() || charProfiles.size() > MAX_PROFILE_SIZE) {
        HILOGE("profile size is invalid!size : %{public}zu", charProfiles.size());
        return false;
    }
    for (const auto& profile : charProfiles) {
        profile.Marshalling(parcel);
    }
    return true;
}

bool IpcUtils::Marshalling(MessageParcel& parcel, const std::vector<std::string>& strings)
{
    size_t size = strings.size();
    WRITE_HELPER_RET(parcel, Uint32, size, false);
    if (strings.empty() || strings.size() > MAX_ID_SIZE) {
        HILOGE("string vector, strings size is invalid! size : %{public}zu", strings.size());
        return false;
    }
    for (const auto& item : strings) {
        WRITE_HELPER_RET(parcel, String, item, false);
    }
    return true;
}

bool IpcUtils::Marshalling(MessageParcel& parcel, const std::vector<int32_t>& params)
{
    size_t size = params.size();
    WRITE_HELPER_RET(parcel, Uint32, size, false);
    if (params.empty() || params.size() > MAX_ID_SIZE) {
        HILOGE("int32_t vector, params size is invalid! size : %{public}zu", params.size());
        return false;
    }
    for (const auto& item : params) {
        WRITE_HELPER_RET(parcel, Int32, item, false);
    }
    return true;
}

bool IpcUtils::Marshalling(MessageParcel& parcel, const std::vector<uint8_t>& params)
{
    int32_t length = static_cast<int32_t>(params.size());
    WRITE_HELPER_RET(parcel, Int32, length, false);
    if (params.empty() || params.size() > MAX_ICON_SIZE) {
        HILOGE("uint8_t vector, params size is invalid! size : %{public}zu", params.size());
        return false;
    }
    if (length > 0) {
        const unsigned char *buffer = reinterpret_cast<const unsigned char *>(params.data());
        parcel.WriteRawData(buffer, length);
    }
    return true;
}

bool IpcUtils::Marshalling(MessageParcel& parcel, const std::map<std::string, std::string>& params)
{
    uint32_t size = params.size();
    WRITE_HELPER_RET(parcel, Uint32, size, false);
    if (params.size() == 0 || params.size() > MAX_PARAM_SIZE) {
        HILOGE("Params size is invalid!size : %{public}zu", params.size());
        return false;
    }
    for (const auto& item : params) {
        WRITE_HELPER_RET(parcel, String, item.first + SEPARATOR + item.second, false);
    }
    return true;
}

bool IpcUtils::Marshalling(MessageParcel& parcel, const std::map<std::string,
    OHOS::DistributedDeviceProfile::SubscribeInfo>& listenerMap)
{
    uint32_t size = listenerMap.size();
    WRITE_HELPER_RET(parcel, Uint32, size, false);
    if (listenerMap.size() == 0 || listenerMap.size() > MAX_LISTENER_SIZE) {
        HILOGE("listenerMap size is invalid!size : %{public}zu", listenerMap.size());
        return false;
    }
    for (const auto& item : listenerMap) {
        OHOS::DistributedDeviceProfile::SubscribeInfo subscribeInfo = item.second;
        if (!subscribeInfo.Marshalling(parcel)) {
            HILOGE("subscribeInfo Marshalling fail!");
            return false;
        }
    }
    return true;
}

bool IpcUtils::Marshalling(MessageParcel& parcel, const std::unordered_set<ProfileChangeType>& changeTypes)
{
    uint32_t size = changeTypes.size();
    WRITE_HELPER_RET(parcel, Uint32, size, false);
    if (changeTypes.size() == 0 || changeTypes.size() > MAX_SUBSCRIBE_CHANGE_SIZE) {
        HILOGE("listenerMap size is invalid!size : %{public}zu", changeTypes.size());
        return false;
    }
    for (ProfileChangeType item : changeTypes) {
        WRITE_HELPER_RET(parcel, Int32, static_cast<int32_t>(item), false);
    }
    return true;
}

bool IpcUtils::Marshalling(MessageParcel& parcel, const std::vector<TrustedDeviceInfo>& deviceInfos)
{
    uint32_t size = deviceInfos.size();
    WRITE_HELPER_RET(parcel, Uint32, size, false);
    if (deviceInfos.empty() || deviceInfos.size() > MAX_PROFILE_SIZE) {
        HILOGE("deviceInfos size is invalid!size : %{public}zu", deviceInfos.size());
        return false;
    }
    for (const auto& item : deviceInfos) {
        item.Marshalling(parcel);
    }
    return true;
}

bool IpcUtils::Marshalling(MessageParcel& parcel, const std::vector<DeviceIconInfo>& deviceIconInfos)
{
    uint32_t size = deviceIconInfos.size();
    WRITE_HELPER_RET(parcel, Uint32, size, false);
    if (deviceIconInfos.size() > MAX_PROFILE_SIZE) {
        HILOGE("deviceInfos size is invalid!size : %{public}zu", deviceIconInfos.size());
        return false;
    }
    for (const auto& item : deviceIconInfos) {
        item.Marshalling(parcel);
    }
    return true;
}

bool IpcUtils::Marshalling(MessageParcel& parcel, const std::vector<ServiceInfoProfile>& serviceInfoProfiles)
{
    uint32_t size = serviceInfoProfiles.size();
    WRITE_HELPER_RET(parcel, Uint32, size, false);
    if (serviceInfoProfiles.empty() || serviceInfoProfiles.size() > MAX_PROFILE_SIZE) {
        HILOGE("deviceInfos size is invalid!size : %{public}zu", serviceInfoProfiles.size());
        return false;
    }
    for (const auto& item : serviceInfoProfiles) {
        item.Marshalling(parcel);
    }
    return true;
}

bool IpcUtils::UnMarshalling(MessageParcel& parcel, std::vector<ServiceInfoProfile>& serviceInfoProfiles)
{
    uint32_t size = parcel.ReadUint32();
    if (size == 0 || size > MAX_PROFILE_SIZE) {
        HILOGE("Profile size is invalid!size : %{public}u", size);
        return false;
    }
    for (uint32_t i = 0; i < size; i++) {
        ServiceInfoProfile serviceInfoProfile;
        if (!serviceInfoProfile.UnMarshalling(parcel)) {
            HILOGE("Profile UnMarshalling fail!");
            return false;
        }
        serviceInfoProfiles.emplace_back(serviceInfoProfile);
    }
    return true;
}

bool IpcUtils::UnMarshalling(MessageParcel& parcel, std::vector<DeviceIconInfo>& deviceIconInfos)
{
    uint32_t size = parcel.ReadUint32();
    if (size == 0 || size > MAX_PROFILE_SIZE) {
        HILOGE("Profile size is invalid!size : %{public}u", size);
        return false;
    }
    for (uint32_t i = 0; i < size; i++) {
        DeviceIconInfo deviceIconInfo;
        if (!deviceIconInfo.UnMarshalling(parcel)) {
            HILOGE("Profile UnMarshalling fail!");
            return false;
        }
        deviceIconInfos.emplace_back(deviceIconInfo);
    }
    return true;
}

bool IpcUtils::UnMarshalling(MessageParcel& parcel, std::vector<ProductInfo>& productInfos)
{
    uint32_t size = parcel.ReadUint32();
    if (size == 0 || size > MAX_PROFILE_SIZE) {
        HILOGE("Profile size is invalid!size : %{public}u", size);
        return false;
    }
    for (uint32_t i = 0; i < size; i++) {
        ProductInfo productInfo;
        if (!productInfo.UnMarshalling(parcel)) {
            HILOGE("Profile UnMarshalling fail!");
            return false;
        }
        productInfos.emplace_back(productInfo);
    }
    return true;
}

bool IpcUtils::UnMarshalling(MessageParcel& parcel, std::vector<TrustDeviceProfile>& trustDeviceProfiles)
{
    uint32_t size = parcel.ReadUint32();
    if (size == 0 || size > MAX_PROFILE_SIZE) {
        HILOGE("Profile size is invalid!size : %{public}u", size);
        return false;
    }
    for (uint32_t i = 0; i < size; i++) {
        TrustDeviceProfile trustDeviceProfile;
        if (!trustDeviceProfile.UnMarshalling(parcel)) {
            HILOGE("Profile UnMarshalling fail!");
            return false;
        }
        trustDeviceProfiles.emplace_back(trustDeviceProfile);
    }
    return true;
}

bool IpcUtils::UnMarshalling(MessageParcel& parcel, std::vector<AccessControlProfile>& aclProfiles)
{
    uint32_t size = parcel.ReadUint32();
    if (size == 0 || size > MAX_PROFILE_SIZE) {
        HILOGE("Profile size is invalid!size : %{public}u", size);
        return false;
    }
    for (uint32_t i = 0; i < size; i++) {
        AccessControlProfile aclProfile;
        if (!aclProfile.UnMarshalling(parcel)) {
            HILOGE("Profile UnMarshalling fail!");
            return false;
        }
        aclProfiles.emplace_back(aclProfile);
    }
    return true;
}

bool IpcUtils::UnMarshalling(MessageParcel& parcel, std::vector<DeviceProfile>& deviceProfiles)
{
    uint32_t size = parcel.ReadUint32();
    if (size == 0 || size > MAX_PROFILE_SIZE) {
        HILOGE("Profile size is invalid!size : %{public}u", size);
        return false;
    }
    for (uint32_t i = 0; i < size; i++) {
        DeviceProfile deviceProfile;
        if (!deviceProfile.UnMarshalling(parcel)) {
            HILOGE("Profile UnMarshalling fail!");
            return false;
        }
        deviceProfiles.emplace_back(deviceProfile);
    }
    return true;
}

bool IpcUtils::UnMarshalling(MessageParcel& parcel, std::vector<ServiceProfile>& serviceProfiles)
{
    uint32_t size = parcel.ReadUint32();
    if (size == 0 || size > MAX_PROFILE_SIZE) {
        HILOGE("Profile size is invalid!size : %{public}u", size);
        return false;
    }
    for (uint32_t i = 0; i < size; i++) {
        ServiceProfile serviceProfile;
        if (!serviceProfile.UnMarshalling(parcel)) {
            HILOGE("Profile UnMarshalling fail!");
            return false;
        }
        serviceProfiles.emplace_back(serviceProfile);
    }
    return true;
}

bool IpcUtils::UnMarshalling(MessageParcel& parcel, std::vector<CharacteristicProfile>& charProfiles)
{
    uint32_t size = parcel.ReadUint32();
    if (size == 0 || size > MAX_PROFILE_SIZE) {
        HILOGE("Profile size is invalid!size : %{public}u", size);
        return false;
    }
    for (uint32_t i = 0; i < size; i++) {
        CharacteristicProfile charProfile;
        if (!charProfile.UnMarshalling(parcel)) {
            HILOGE("Profile UnMarshalling fail!");
            return false;
        }
        charProfiles.emplace_back(charProfile);
    }
    return true;
}

bool IpcUtils::UnMarshalling(MessageParcel& parcel, std::vector<std::string>& strings)
{
    size_t size = parcel.ReadUint32();
    if (size == 0 || size > MAX_ID_SIZE) {
        HILOGE("string vector, strings size is invalid!size : %{public}zu", size);
        return false;
    }
    for (uint32_t i = 0; i < size; i++) {
        std::string item = "";
        READ_HELPER_RET(parcel, String, item, false);
        strings.emplace_back(item);
    }
    return true;
}

bool IpcUtils::UnMarshalling(MessageParcel& parcel, std::vector<int32_t>& params)
{
    size_t size = parcel.ReadUint32();
    if (size == 0 || size > MAX_ID_SIZE) {
        HILOGE("int32_t vector, params size is invalid! size : %{public}zu", size);
        return false;
    }
    for (uint32_t i = 0; i < size; i++) {
        int32_t item = 0;
        READ_HELPER_RET(parcel, Int32, item, false);
        params.emplace_back(item);
    }
    return true;
}

bool IpcUtils::UnMarshalling(MessageParcel& parcel, std::vector<uint8_t>& params)
{
    int32_t length = parcel.ReadInt32();
    if (length <= 0 || length > MAX_ICON_SIZE) {
        HILOGE("uint8_t vector, params size is invalid! size : %{public}d", length);
        return false;
    }
    const void *temp = parcel.ReadRawData((size_t)length);
    if (temp == nullptr) {
        HILOGE("read raw data failed, length = %{public}d", length);
        return false;
    }
    const unsigned char *buffer = reinterpret_cast<const unsigned char *>(temp);
    if (buffer == nullptr) {
        HILOGE("Unexpected null after reinterpret_cast");
        return false;
    }
    std::vector<uint8_t> icon(buffer, buffer + length);
    params = icon;
    return true;
}

bool IpcUtils::UnMarshalling(MessageParcel& parcel, std::map<std::string, std::string>& params)
{
    uint32_t size = parcel.ReadUint32();
    if (size == 0 || size > MAX_PARAM_SIZE) {
        HILOGE("Params size is invalid!size : %{public}u", size);
        return false;
    }
    for (uint32_t i = 0; i < size; i++) {
        std::string item = "";
        READ_HELPER_RET(parcel, String, item, false);
        std::string::size_type position = item.find(SEPARATOR);
        if (position == item.npos) {
            HILOGE("Not found the separator!");
            continue;
        }
        std::string key = item.substr(0, position);
        std::string value = item.substr(position + 1);
        params[key] = value;
    }
    return true;
}

bool IpcUtils::UnMarshalling(MessageParcel& parcel, std::map<std::string,
    OHOS::DistributedDeviceProfile::SubscribeInfo>& listenerMap)
{
    uint32_t size = parcel.ReadUint32();
    if (size == 0 || size > MAX_LISTENER_SIZE) {
        HILOGE("Params size is invalid!size : %{public}u", size);
        return false;
    }
    for (uint32_t i = 0; i < size; i++) {
        OHOS::DistributedDeviceProfile::SubscribeInfo subscribeInfo;
        if (!subscribeInfo.UnMarshalling(parcel)) {
            HILOGE("subscribeInfo UnMarshalling fail!");
            return false;
        }
        listenerMap[subscribeInfo.GetSubscribeKey() + SEPARATOR + std::to_string(subscribeInfo.GetSaId())] =
            subscribeInfo;
    }
    return true;
}

bool IpcUtils::UnMarshalling(MessageParcel& parcel, std::unordered_set<ProfileChangeType>& changeTypes)
{
    uint32_t size = parcel.ReadUint32();
    if (size == 0 || size > MAX_SUBSCRIBE_CHANGE_SIZE) {
        HILOGE("Params size is invalid!size : %{public}u", size);
        return false;
    }
    for (uint32_t i = 0; i < size; i++) {
        uint32_t num = parcel.ReadUint32();
        if (num <= ProfileChangeType::PROFILE_CHANGE_TYPE_MIN || num >= ProfileChangeType::PROFILE_CHANGE_TYPE_MAX) {
            HILOGE("The value is invalid!");
            return false;
        }
        changeTypes.emplace(static_cast<ProfileChangeType>(num));
    }
    return true;
}

bool IpcUtils::UnMarshalling(MessageParcel& parcel, std::vector<TrustedDeviceInfo>& deviceInfos)
{
    uint32_t size = parcel.ReadUint32();
    if (size == 0 || size > MAX_PROFILE_SIZE) {
        HILOGE("deviceInfos size is invalid!size : %{public}u", size);
        return false;
    }
    for (uint32_t i = 0; i < size; i++) {
        TrustedDeviceInfo item;
        if (!item.UnMarshalling(parcel)) {
            HILOGE("Profile UnMarshalling fail!");
            return false;
        }
        deviceInfos.emplace_back(item);
    }
    return true;
}
} // namespace DistributedDeviceProfile
} // namespace OHOS