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

#include <cstddef>
#include <cstdint>
#include <cstdlib>
#include <unistd.h>
#include <errno.h>
#include <securec.h>
#include <fcntl.h>
#include <sys/poll.h>
#include <linux/sync_file.h>
#include <sys/ioctl.h>
#include "hilog/log.h"

namespace OHOS {
using namespace OHOS::HiviewDFX;

namespace {
#undef LOG_DOMAIN
#define LOG_DOMAIN 0xD001400
#undef LOG_TAG
#define LOG_TAG "SyncFence"

#define B_CPRINTF(func, fmt, ...) \
    func(LOG_CORE, "<%{public}d>%{public}s: " fmt, \
        __LINE__, __func__, ##__VA_ARGS__)
        
#define UTILS_LOGD(fmt, ...) B_CPRINTF(HILOG_DEBUG, fmt, ##__VA_ARGS__)
#define UTILS_LOGI(fmt, ...) B_CPRINTF(HILOG_INFO, fmt, ##__VA_ARGS__)
#define UTILS_LOGW(fmt, ...) B_CPRINTF(HILOG_WARN, fmt, ##__VA_ARGS__)
#define UTILS_LOGE(fmt, ...) B_CPRINTF(HILOG_ERROR, fmt, ##__VA_ARGS__)

constexpr int32_t INVALID_FD = -1;
}  // namespace

const sptr<SyncFence> SyncFence::INVALID_FENCE = sptr<SyncFence>(new SyncFence(INVALID_FD));
const ns_sec_t SyncFence::INVALID_TIMESTAMP = -1;
const ns_sec_t SyncFence::FENCE_PENDING_TIMESTAMP = INT64_MAX;

SyncFence::SyncFence(int32_t fenceFd) : fenceFd_(fenceFd)
{
}

SyncFence::~SyncFence()
{
}

int32_t SyncFence::Wait(uint32_t timeout)
{
    int retCode = -1;
    if (fenceFd_ < 0) {
        return retCode;
    }

    struct pollfd pollfds = {0};
    pollfds.fd = fenceFd_;
    pollfds.events = POLLIN;

    do {
        retCode = poll(&pollfds, 1, timeout);
    } while (retCode == -1 && (errno == EINTR || errno == EAGAIN));

    if (retCode == 0) {
        retCode = -1;
        errno = ETIME;
    } else if (retCode > 0) {
        retCode = 0;
        if (pollfds.revents & (POLLERR | POLLNVAL)) {
            retCode = -1;
            errno = EINVAL;
        }
    }

    return retCode < 0 ? -errno : 0;
}

int32_t SyncFence::SyncMerge(const char *name, int32_t fd1, int32_t fd2, int32_t &newFenceFd)
{
    struct sync_merge_data syncMergeData = {};
    syncMergeData.fd2 = fd2;
    if (strcpy_s(syncMergeData.name, sizeof(syncMergeData.name), name)) {
        UTILS_LOGE("SyncMerge ctrcpy fence name failed.");
        return -1;
    }

    int32_t retCode = ioctl(fd1, SYNC_IOC_MERGE, &syncMergeData);
    if (retCode < 0) {
        errno = EINVAL;
        UTILS_LOGE("Fence merge failed, errno: %{public}d, ret: %{public}d.", errno, retCode);
        return -1;
    }

    newFenceFd = syncMergeData.fence;
    return 0;
}

sptr<SyncFence> SyncFence::MergeFence(const std::string &name,
                                      const sptr<SyncFence>& fence1, const sptr<SyncFence>& fence2)
{
    int32_t newFenceFd = INVALID_FD;
    int32_t fenceFd1 = fence1->fenceFd_;
    int32_t fenceFd2 = fence2->fenceFd_;

    if (fenceFd1 >= 0 && fenceFd2 >= 0) {
        (void)SyncFence::SyncMerge(name.c_str(), fenceFd1, fenceFd2, newFenceFd);
    } else if (fenceFd1 >= 0) {
        (void)SyncFence::SyncMerge(name.c_str(), fenceFd1, fenceFd1, newFenceFd);
    } else if (fenceFd2 >= 0) {
        (void)SyncFence::SyncMerge(name.c_str(), fenceFd2, fenceFd2, newFenceFd);
    } else {
        return INVALID_FENCE;
    }

    if (newFenceFd == INVALID_FD) {
        UTILS_LOGE("sync_merge(%{public}s) failed, error: %{public}s (%{public}d)",
                     name.c_str(), strerror(errno), errno);
        return INVALID_FENCE;
    }

    return sptr<SyncFence>(new SyncFence(newFenceFd));
}

ns_sec_t SyncFence::SyncFileReadTimestamp()
{
    std::vector<SyncPointInfo> ptInfos = GetFenceInfo();
    if (ptInfos.empty()) {
        return FENCE_PENDING_TIMESTAMP;
    }
    size_t signalFenceCount = 0;
    for (const auto &info : ptInfos) {
        if (info.status == SIGNALED) {
            signalFenceCount++;
        }
    }
    if (signalFenceCount == ptInfos.size()) {
        // fence signaled
        uint64_t timestamp = 0;
        for (const auto &ptInfo : ptInfos) {
            if (ptInfo.timestampNs > timestamp) {
                timestamp = ptInfo.timestampNs;
            }
        }
        return static_cast<ns_sec_t>(timestamp);
    } else {
        // fence still active
        return FENCE_PENDING_TIMESTAMP;
    }
}

std::vector<SyncPointInfo> SyncFence::GetFenceInfo()
{
    struct sync_file_info arg;
    struct sync_file_info *infoPtr = nullptr;

    errno_t retCode = memset_s(&arg, sizeof(struct sync_file_info), 0, sizeof(struct sync_file_info));
    if (retCode != 0) {
        UTILS_LOGE("memset_s error, retCode = %{public}d", retCode);
        return {};
    }
    int32_t ret = ioctl(fenceFd_, SYNC_IOC_FILE_INFO, &arg);
    if (ret < 0) {
        UTILS_LOGD("GetFenceInfo SYNC_IOC_FILE_INFO ioctl failed, ret: %{public}d", ret);
        return {};
    }

    if (arg.num_fences <= 0) {
        UTILS_LOGD("GetFenceInfo arg.num_fences failed, num_fences: %{public}d", arg.num_fences);
        return {};
    }
    // to malloc sync_file_info and the number of 'sync_fence_info' memory
    size_t syncFileInfoMemSize = sizeof(struct sync_file_info) + sizeof(struct sync_fence_info) * arg.num_fences;
    infoPtr = (struct sync_file_info *)malloc(syncFileInfoMemSize);
    if (infoPtr == nullptr) {
        UTILS_LOGD("GetFenceInfo malloc failed oom");
        return {};
    }
    retCode = memset_s(infoPtr, syncFileInfoMemSize, 0, syncFileInfoMemSize);
    if (retCode != 0) {
        UTILS_LOGE("memset_s error, retCode = %{public}d", retCode);
        free(infoPtr);
        return {};
    }
    infoPtr->num_fences = arg.num_fences;
    infoPtr->sync_fence_info = static_cast<uint64_t>(uintptr_t(infoPtr + 1));

    ret = ioctl(fenceFd_, SYNC_IOC_FILE_INFO, infoPtr);
    if (ret < 0) {
        UTILS_LOGE("GetTotalFenceInfo SYNC_IOC_FILE_INFO ioctl failed, ret: %{public}d", ret);
        free(infoPtr);
        return {};
    }
    std::vector<SyncPointInfo> infos;
    const auto fenceInfos = (struct sync_fence_info *)(uintptr_t)(infoPtr->sync_fence_info);
    for (uint32_t i = 0; i < infoPtr->num_fences; i++) {
        infos.push_back(SyncPointInfo { fenceInfos[i].timestamp_ns,
            static_cast<FenceStatus>(fenceInfos[i].status) } );
    }

    free(infoPtr);
    return infos;
}

int32_t SyncFence::Dup() const
{
    return ::dup(fenceFd_);
}

FenceStatus SyncFence::GetStatus()
{
    if (fenceFd_ < 0) {
        return ERROR;
    }
    std::vector<SyncPointInfo> ptInfos = GetFenceInfo();
    if (ptInfos.empty()) {
        return ERROR;
    }
    size_t signalFenceCount = 0;
    for (const auto &info : ptInfos) {
        if (info.status == SIGNALED) {
            signalFenceCount++;
        }
    }
    if (signalFenceCount == ptInfos.size()) {
        return SIGNALED;
    } else {
        return ACTIVE;
    }
}

int32_t SyncFence::Get() const
{
    return fenceFd_;
}

bool SyncFence::IsValid() const
{
    return fenceFd_ != -1;
}

sptr<SyncFence> SyncFence::ReadFromMessageParcel(MessageParcel &parcel)
{
    int32_t fence = parcel.ReadInt32();
    if (fence < 0) {
        return INVALID_FENCE;
    }

    fence = parcel.ReadFileDescriptor();

    return sptr<SyncFence>(new SyncFence(fence));
}

sptr<SyncFence> SyncFence::InvalidFence()
{
    return sptr<SyncFence>(new SyncFence(-1));
}

void SyncFence::WriteToMessageParcel(MessageParcel &parcel)
{
    int32_t fence = fenceFd_;
    if (fence >= 0 && fcntl(fence, F_GETFL) == -1 && errno == EBADF) {
        fence = -1;
    }

    parcel.WriteInt32(fence);

    if (fence < 0) {
        UTILS_LOGD("WriteToMessageParcel fence is invalid : %{public}d", fence);
        return;
    }

    parcel.WriteFileDescriptor(fence);
}

} // namespace OHOS