// Copyright 2021 yuzu emulator team
// Licensed under GPLv2 or any later version
// Refer to the license.txt file included.

#pragma once

#include <memory>

#include "core/hle/kernel/slab_helpers.h"
#include "core/hle/kernel/svc_types.h"
#include "core/hle/result.h"

union ResultCode;

namespace Core::Memory {
class Memory;
}

namespace Kernel {

class KernelCore;
class Process;

class KTransferMemory final
    : public KAutoObjectWithSlabHeapAndContainer<KTransferMemory, KAutoObjectWithList> {
    KERNEL_AUTOOBJECT_TRAITS(KTransferMemory, KAutoObject);

public:
    explicit KTransferMemory(KernelCore& kernel);
    ~KTransferMemory() override;

    static constexpr HandleType HANDLE_TYPE = HandleType::TransferMemory;

    ResultCode Initialize(VAddr address_, std::size_t size_, Svc::MemoryPermission owner_perm_);

    virtual void Finalize() override;

    virtual bool IsInitialized() const override {
        return is_initialized;
    }

    virtual uintptr_t GetPostDestroyArgument() const override {
        return reinterpret_cast<uintptr_t>(owner);
    }

    static void PostDestroy(uintptr_t arg);

    Process* GetOwner() const {
        return owner;
    }

    VAddr GetSourceAddress() const {
        return address;
    }

    size_t GetSize() const {
        return is_initialized ? size * PageSize : 0;
    }

    // DEPRECATED

    std::string GetTypeName() const override {
        return "TransferMemory";
    }

    std::string GetName() const override {
        return GetTypeName();
    }

    HandleType GetHandleType() const override {
        return HANDLE_TYPE;
    }

private:
    Process* owner{};
    VAddr address{};
    Svc::MemoryPermission owner_perm{};
    size_t size{};
    bool is_initialized{};
};

} // namespace Kernel