// Copyright 2019 The Dawn & Tint Authors
//
// Redistribution and use in source and binary forms, with or without
// modification, are permitted provided that the following conditions are met:
//
// 1. Redistributions of source code must retain the above copyright notice, this
//    list of conditions and the following disclaimer.
//
// 2. Redistributions in binary form must reproduce the above copyright notice,
//    this list of conditions and the following disclaimer in the documentation
//    and/or other materials provided with the distribution.
//
// 3. Neither the name of the copyright holder nor the names of its
//    contributors may be used to endorse or promote products derived from
//    this software without specific prior written permission.
//
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
// AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
// IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
// DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE
// FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
// DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
// SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
// CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
// OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

#include <memory>
#include <new>
#include <utility>

#include "src/dawn/wire/server/Server.h"
#include "src/utils/assert.h"
#include "src/utils/heap_array.h"

namespace dawn::wire::server {

void Server::OnQueueWorkDone(QueueWorkDoneUserdata* data,
                             wgpu::QueueWorkDoneStatus status,
                             StringView message) {
    ReturnQueueWorkDoneCallbackCmd cmd;
    cmd.instanceId = data->instanceId;
    cmd.future = data->future;
    cmd.status = status;
    cmd.message = message;

    SerializeCommand(std::move(cmd));
}

WireResult Server::DoQueueOnSubmittedWorkDone(Known<WGPUQueue> queue,
                                              Known<WGPUInstance> instance,
                                              Future future) {
    auto userdata = MakeUserdata<QueueWorkDoneUserdata>();
    userdata->queue = queue.AsHandle();
    userdata->instanceId = instance.id;
    userdata->future = future;

    mProcs->queueOnSubmittedWorkDone(
        queue->handle, MakeCallbackInfo<WGPUQueueWorkDoneCallbackInfo, &Server::OnQueueWorkDone>(
                           userdata.release()));
    return WireResult::Success;
}

WireResult Server::DoQueueWriteBuffer(Known<WGPUQueue> queue,
                                      Known<WGPUBuffer> buffer,
                                      uint64_t bufferOffset,
                                      Span<const volatile std::byte> data) {
    // Note we cast away the volatile here because the data is user provided and "safe" from TOCTOU
    // attacks, i.e. the data is/was already visible to the user.
    mProcs->queueWriteBuffer(queue->handle, buffer->handle, bufferOffset,
                             const_cast<const std::byte*>(data.data()), data.size());
    return WireResult::Success;
}

WireResult Server::DoQueueWriteBufferXl(Known<WGPUQueue> queue,
                                        Known<WGPUBuffer> buffer,
                                        uint64_t bufferOffset,
                                        size_t size,
                                        Span<const std::byte> memoryHandleCreateInfo,
                                        Span<const std::byte> memoryDataUpdateInfo) {
    // Deserialize metadata produced from the client to create a companion server handle.
    auto memoryHandle = mMemoryTransferService->DeserializeMemoryHandle(memoryHandleCreateInfo);
    if (memoryHandle == nullptr) {
        return WireResult::FatalError;
    }

    // Try first to use GetSourceData if the memory transfer service implements
    // it. If so, we can avoid a copy.
    Span<std::byte> source = memoryHandle->GetSource();
    if (!source.empty()) {
        if (source.size() < size) {
            return WireResult::FatalError;
        }

        mProcs->queueWriteBuffer(queue->handle, buffer->handle, bufferOffset, source.data(), size);
        return WireResult::Success;
    }

    // Otherwise, fall back to DeserializeDataUpdate.
    auto backingData = HeapArray<std::byte>(size, std::nothrow);
    if (!backingData) {
        return WireResult::FatalError;
    }

    // Deserialize the flush info and flush updated data from the handle into the target of the
    // handle that's just a temporary allocation from above right now.
    if (!memoryHandle->DeserializeDataUpdate(memoryDataUpdateInfo, 0u, size, backingData)) {
        return WireResult::FatalError;
    }

    mProcs->queueWriteBuffer(queue->handle, buffer->handle, bufferOffset, backingData.data(), size);
    return WireResult::Success;
}

WireResult Server::DoQueueWriteTexture(Known<WGPUQueue> queue,
                                       const TexelCopyTextureInfo* destination,
                                       Span<const volatile std::byte> data,
                                       const TexelCopyBufferLayout* dataLayout,
                                       const Extent3D* writeSize) {
    // Note we cast away the volatile here because the data is user provided and "safe" from TOCTOU
    // attacks, i.e. the data is/was already visible to the user.
    mProcs->queueWriteTexture(queue->handle, ToAPI(destination),
                              const_cast<const std::byte*>(data.data()), data.size(),
                              ToAPI(dataLayout), ToAPI(writeSize));
    return WireResult::Success;
}

WireResult Server::DoQueueWriteTextureXl(Known<WGPUQueue> queue,
                                         const TexelCopyTextureInfo* destination,
                                         size_t dataSize,
                                         const TexelCopyBufferLayout* dataLayout,
                                         const Extent3D* writeSize,
                                         Span<const std::byte> memoryHandleCreateInfo,
                                         Span<const std::byte> memoryDataUpdateInfo) {
    // Deserialize metadata produced from the client to create a companion server handle.
    auto memoryHandle = mMemoryTransferService->DeserializeMemoryHandle(memoryHandleCreateInfo);
    if (memoryHandle == nullptr) {
        return WireResult::FatalError;
    }

    // Try first to use GetSourceData if the memory transfer service implements
    // it. If so, we can avoid a copy.
    Span<std::byte> source = memoryHandle->GetSource();
    if (!source.empty()) {
        if (source.size() < dataSize) {
            return WireResult::FatalError;
        }

        mProcs->queueWriteTexture(queue->handle, ToAPI(destination), source.data(), dataSize,
                                  ToAPI(dataLayout), ToAPI(writeSize));
        return WireResult::Success;
    }

    // Otherwise, fall back to DeserializeDataUpdate.
    auto backingData = HeapArray<std::byte>(dataSize, std::nothrow);
    if (!backingData) {
        return WireResult::FatalError;
    }

    // Deserialize the flush info and flush updated data from the handle into the target of the
    // handle that's just a temporary allocation from above right now.
    if (!memoryHandle->DeserializeDataUpdate(memoryDataUpdateInfo, 0u, dataSize, backingData)) {
        return WireResult::FatalError;
    }

    mProcs->queueWriteTexture(queue->handle, ToAPI(destination), backingData.data(), dataSize,
                              ToAPI(dataLayout), ToAPI(writeSize));
    return WireResult::Success;
}

}  // namespace dawn::wire::server