src/intel/vulkan/anv_magma_connection.cc - third_party/mesa - Git at Google

 /*
  * Copyright © 2018 Google, LLC
  *
  * Permission is hereby granted, free of charge, to any person obtaining a
  * copy of this software and associated documentation files (the "Software"),
  * to deal in the Software without restriction, including without limitation
  * the rights to use, copy, modify, merge, publish, distribute, sublicense,
  * and/or sell copies of the Software, and to permit persons to whom the
  * Software is furnished to do so, subject to the following conditions:
  *
  * The above copyright notice and this permission notice (including the next
  * paragraph) shall be included in all copies or substantial portions of the
  * Software.
  *
  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL
  * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
  * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
  * IN THE SOFTWARE.
  */

 #include "anv_magma.h"
 #include "vk_magma_syncobj.h"
 #include "common/intel_gem.h"
 #include "util/magma/inflight_list.h"
 #include "util/log.h"
 #include <assert.h>
 #include <chrono>
 #include <map>
 #include <unistd.h>
 #include <vector>

 #if VK_USE_PLATFORM_FUCHSIA
 #include "magma/magma_sysmem.h"
 #include "os/fuchsia.h"
 #endif

 #define LOG_VERBOSE(...)                                                                           \
    do {                                                                                            \
       if (false)                                                                                   \
          mesa_logd(__VA_ARGS__);                                                                   \
    } while (0)

 static inline uint64_t page_size() { return sysconf(_SC_PAGESIZE); }

 static inline bool is_page_aligned(uint64_t val) { return (val & (page_size() - 1)) == 0; }

 // Note, alignment must be a power of 2
 template <class T> static inline T round_up(T val, uint32_t alignment)
 {
    return ((val - 1) | (alignment - 1)) + 1;
 }

 class Buffer : public anv_magma_buffer {
 public:
    Buffer(magma_buffer_t buffer, uint64_t id)
    {
       anv_magma_buffer::buffer = buffer;
       anv_magma_buffer::id = id;
    }

    ~Buffer() { assert(!get()); }

    magma_buffer_t get() { return anv_magma_buffer::buffer; }

    void release(magma_connection_t connection)
    {
       magma_connection_release_buffer(connection, get());
       anv_magma_buffer::buffer = 0;
    }

    void AddMapping(uint64_t offset, uint64_t length, uint64_t addr)
    {
       mappings_[addr] = {.offset = offset, .length = length};
    }

    struct Segment {
       uint64_t offset = 0;
       uint64_t length = 0;
    };

    bool HasMapping(uint64_t addr, Segment* segment_out) const
    {
       auto iter = mappings_.find(addr);
       if (iter != mappings_.end()) {
          if (segment_out) {
             *segment_out = iter->second;
          }
          return true;
       }
       return false;
    }

    void RemoveMapping(uint64_t addr) { mappings_.erase(addr); }

 private:
    std::map<uint64_t, Segment> mappings_;
 };

 static void notification_callback(void* context) {
    AnvMagmaConnectionServiceNotifications(
       get_anv_connection(reinterpret_cast<vk_magma_connection*>(context)));
 }

 class Connection : public anv_connection {
 public:
    Connection(magma_connection_t magma_connection, magma_handle_t notification_channel)
        : inflight_list_(InflightList_Create())
    {
       anv_connection::vk.connection = magma_connection;
       anv_connection::vk.notification_channel = notification_channel;
       anv_connection::vk.notification_callback = notification_callback;
    }

    ~Connection()
    {
 #if VK_USE_PLATFORM_FUCHSIA
       if (sysmem_connection_) {
          magma_sysmem_connection_release(sysmem_connection_);
       }
 #endif // VK_USE_PLATFORM_FUCHSIA
       magma_connection_release(magma_connection());
       InflightList_Destroy(inflight_list_);
    }

    magma_connection_t magma_connection() { return anv_connection::vk.connection; }

    InflightList* inflight_list() { return inflight_list_; }

 #if VK_USE_PLATFORM_FUCHSIA
    magma_status_t GetSysmemConnection(magma_sysmem_connection_t* sysmem_connection_out)
    {
       if (!sysmem_connection_) {
          zx_handle_t client_handle;
          if (!fuchsia_open("/svc/fuchsia.sysmem.Allocator", &client_handle))
             return ANV_MAGMA_DRET(MAGMA_STATUS_INTERNAL_ERROR);
          magma_status_t status = magma_sysmem_connection_import(client_handle, &sysmem_connection_);
          if (status != MAGMA_STATUS_OK)
             return ANV_MAGMA_DRET(status);
       }
       *sysmem_connection_out = sysmem_connection_;
       return MAGMA_STATUS_OK;
    }
 #endif // VK_USE_PLATFORM_FUCHSIA

    static Connection* cast(anv_connection* connection)
    {
       return static_cast<Connection*>(connection);
    }

 private:
 #if VK_USE_PLATFORM_FUCHSIA
    magma_sysmem_connection_t sysmem_connection_{};
 #endif // #if VK_USE_PLATFORM_FUCHSIA
    InflightList* inflight_list_;
 };

 anv_connection* AnvMagmaCreateConnection(magma_connection_t connection)
 {
    return new Connection(connection,
       magma_connection_get_notification_channel_handle(connection));
 }

 void AnvMagmaReleaseConnection(anv_connection* connection)
 {
    delete static_cast<Connection*>(connection);
 }

 #if VK_USE_PLATFORM_FUCHSIA
 magma_status_t AnvMagmaGetSysmemConnection(struct anv_connection* connection,
                                            magma_sysmem_connection_t* sysmem_connection_out)
 {
    return Connection::cast(connection)->GetSysmemConnection(sysmem_connection_out);
 }
 #endif // VK_USE_PLATFORM_FUCHSIA

 void AnvMagmaConnectionServiceNotifications(anv_connection* connection)
 {
    InflightList_TryUpdate(Connection::cast(connection)->inflight_list(),
                           Connection::cast(connection)->magma_connection());
 }

 magma_status_t AnvMagmaConnectionWait(anv_connection* connection, uint64_t buffer_id,
                                       uint64_t timeout_ns)
 {
    return InflightList_WaitForBuffer(Connection::cast(connection)->inflight_list(),
                                      Connection::cast(connection)->magma_connection(),
                                      connection->vk.notification_channel, buffer_id, timeout_ns);
 }

 int AnvMagmaConnectionExec(anv_connection* connection, uint32_t context_id,
                            struct drm_i915_gem_execbuffer2* execbuf,
                            struct anv_magma_buffer* buffers[])
 {
    if (execbuf->buffer_count == 0)
       return 0;

    auto exec_objects = reinterpret_cast<drm_i915_gem_exec_object2*>(execbuf->buffers_ptr);

    std::vector<magma_exec_resource> resources;
    resources.reserve(execbuf->buffer_count);

    for (uint32_t i = 0; i < execbuf->buffer_count; i++) {
       auto buffer = static_cast<Buffer*>(buffers[i]);

       uint64_t offset = 0;
       uint64_t length = exec_objects[i].rsvd2;

       resources.push_back({
           .buffer_id = buffer->id,
           .offset = offset,
           .length = length,
       });

       if (!is_page_aligned(offset))
          return ANV_MAGMA_DRET_MSG(-1, "offset (0x%" PRIx64 ") not page aligned", offset);

       uint64_t gpu_addr = intel_48b_address(exec_objects[i].offset);
       length = round_up(length, page_size());

       Buffer::Segment segment;
       bool has_mapping = buffer->HasMapping(gpu_addr, &segment);
       if (has_mapping) {
          assert(offset == segment.offset);
          if (length > segment.length) {
             // Growing an existing mapping.
             buffer->RemoveMapping(gpu_addr);
             has_mapping = false;
          } else {
             assert(length == segment.length);
          }
       }

       if (!has_mapping) {
          LOG_VERBOSE("mapping to gpu addr 0x%" PRIx64 ": id %" PRIu64 " offset %" PRIu64
                      " length %" PRIu64,
                      gpu_addr, buffer->id, offset, length);
          constexpr uint64_t kMapFlags =
              MAGMA_MAP_FLAG_READ | MAGMA_MAP_FLAG_WRITE | MAGMA_MAP_FLAG_EXECUTE;
          magma_connection_map_buffer(Connection::cast(connection)->magma_connection(),
             gpu_addr, buffer->get(), offset, length, kMapFlags);

          buffer->AddMapping(offset, length, gpu_addr);
       }
    }

    uint32_t syncobj_count = execbuf->num_cliprects;

    std::vector<uint64_t> semaphore_ids;
    semaphore_ids.reserve(syncobj_count);
    uint32_t wait_semaphore_count = 0;

    // Wait semaphores first, then signal
    for (uint32_t i = 0; i < syncobj_count; i++) {
       auto& syncobj = reinterpret_cast<drm_i915_gem_exec_fence*>(execbuf->cliprects_ptr)[i];
       if (syncobj.flags & I915_EXEC_FENCE_WAIT) {
          semaphore_ids.push_back(reinterpret_cast<vk_magma_syncobj*>(syncobj.handle)->id);
          wait_semaphore_count++;
       }
    }
    for (uint32_t i = 0; i < syncobj_count; i++) {
       auto& syncobj = reinterpret_cast<drm_i915_gem_exec_fence*>(execbuf->cliprects_ptr)[i];
       if (syncobj.flags & I915_EXEC_FENCE_SIGNAL) {
          semaphore_ids.push_back(reinterpret_cast<vk_magma_syncobj*>(syncobj.handle)->id);
       }
    }

    magma_exec_command_buffer command_buffer = {.resource_index =
                                                    execbuf->buffer_count - 1, // by drm convention
                                                .start_offset = execbuf->batch_start_offset};

    magma_command_descriptor descriptor = {.resource_count = execbuf->buffer_count,
                                           .command_buffer_count = 1,
                                           .wait_semaphore_count = wait_semaphore_count,
                                           .signal_semaphore_count =
                                               syncobj_count - wait_semaphore_count,
                                           .resources = resources.data(),
                                           .command_buffers = &command_buffer,
                                           .semaphore_ids = semaphore_ids.data(),
                                           .flags = 0};

    // Add to inflight list first to avoid race with any other thread reading completions from the
    // notification channel, in case this thread is preempted just after sending the command buffer
    // and the completion happens quickly.
    InflightList_AddAndUpdate(Connection::cast(connection)->inflight_list(),
                              Connection::cast(connection)->magma_connection(), resources.data(),
                              execbuf->buffer_count);

    magma_connection_execute_command(Connection::cast(connection)->magma_connection(),
       context_id, &descriptor);

    return 0;
 }

 anv_magma_buffer* AnvMagmaCreateBuffer(anv_connection* connection, magma_buffer_t buffer,
                                        magma_buffer_id_t buffer_id)
 {
    return new Buffer(buffer, buffer_id);
 }

 void AnvMagmaReleaseBuffer(anv_connection* connection, anv_magma_buffer* anv_buffer)
 {
    auto buffer = static_cast<Buffer*>(anv_buffer);
    // Hardware mappings are released when the buffer is released.
    buffer->release(Connection::cast(connection)->magma_connection());
    delete buffer;
 }
	/*
	* Copyright © 2018 Google, LLC
	*
	* Permission is hereby granted, free of charge, to any person obtaining a
	* copy of this software and associated documentation files (the "Software"),
	* to deal in the Software without restriction, including without limitation
	* the rights to use, copy, modify, merge, publish, distribute, sublicense,
	* and/or sell copies of the Software, and to permit persons to whom the
	* Software is furnished to do so, subject to the following conditions:
	*
	* The above copyright notice and this permission notice (including the next
	* paragraph) shall be included in all copies or substantial portions of the
	* Software.
	*
	* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
	* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
	* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
	* THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
	* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
	* FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
	* IN THE SOFTWARE.
	*/

	#include "anv_magma.h"
	#include "vk_magma_syncobj.h"
	#include "common/intel_gem.h"
	#include "util/magma/inflight_list.h"
	#include "util/log.h"
	#include <assert.h>
	#include <chrono>
	#include <map>
	#include <unistd.h>
	#include <vector>

	#if VK_USE_PLATFORM_FUCHSIA
	#include "magma/magma_sysmem.h"
	#include "os/fuchsia.h"
	#endif

	#define LOG_VERBOSE(...) \
	do { \
	if (false) \
	mesa_logd(__VA_ARGS__); \
	} while (0)

	static inline uint64_t page_size() { return sysconf(_SC_PAGESIZE); }

	static inline bool is_page_aligned(uint64_t val) { return (val & (page_size() - 1)) == 0; }

	// Note, alignment must be a power of 2
	template <class T> static inline T round_up(T val, uint32_t alignment)
	{
	return ((val - 1) \| (alignment - 1)) + 1;
	}

	class Buffer : public anv_magma_buffer {
	public:
	Buffer(magma_buffer_t buffer, uint64_t id)
	{
	anv_magma_buffer::buffer = buffer;
	anv_magma_buffer::id = id;
	}

	~Buffer() { assert(!get()); }

	magma_buffer_t get() { return anv_magma_buffer::buffer; }

	void release(magma_connection_t connection)
	{
	magma_connection_release_buffer(connection, get());
	anv_magma_buffer::buffer = 0;
	}

	void AddMapping(uint64_t offset, uint64_t length, uint64_t addr)
	{
	mappings_[addr] = {.offset = offset, .length = length};
	}

	struct Segment {
	uint64_t offset = 0;
	uint64_t length = 0;
	};

	bool HasMapping(uint64_t addr, Segment* segment_out) const
	{
	auto iter = mappings_.find(addr);
	if (iter != mappings_.end()) {
	if (segment_out) {
	*segment_out = iter->second;
	}
	return true;
	}
	return false;
	}

	void RemoveMapping(uint64_t addr) { mappings_.erase(addr); }

	private:
	std::map<uint64_t, Segment> mappings_;
	};

	static void notification_callback(void* context) {
	AnvMagmaConnectionServiceNotifications(
	get_anv_connection(reinterpret_cast<vk_magma_connection*>(context)));
	}

	class Connection : public anv_connection {
	public:
	Connection(magma_connection_t magma_connection, magma_handle_t notification_channel)
	: inflight_list_(InflightList_Create())
	{
	anv_connection::vk.connection = magma_connection;
	anv_connection::vk.notification_channel = notification_channel;
	anv_connection::vk.notification_callback = notification_callback;
	}

	~Connection()
	{
	#if VK_USE_PLATFORM_FUCHSIA
	if (sysmem_connection_) {
	magma_sysmem_connection_release(sysmem_connection_);
	}
	#endif // VK_USE_PLATFORM_FUCHSIA
	magma_connection_release(magma_connection());
	InflightList_Destroy(inflight_list_);
	}

	magma_connection_t magma_connection() { return anv_connection::vk.connection; }

	InflightList* inflight_list() { return inflight_list_; }

	#if VK_USE_PLATFORM_FUCHSIA
	magma_status_t GetSysmemConnection(magma_sysmem_connection_t* sysmem_connection_out)
	{
	if (!sysmem_connection_) {
	zx_handle_t client_handle;
	if (!fuchsia_open("/svc/fuchsia.sysmem.Allocator", &client_handle))
	return ANV_MAGMA_DRET(MAGMA_STATUS_INTERNAL_ERROR);
	magma_status_t status = magma_sysmem_connection_import(client_handle, &sysmem_connection_);
	if (status != MAGMA_STATUS_OK)
	return ANV_MAGMA_DRET(status);
	}
	*sysmem_connection_out = sysmem_connection_;
	return MAGMA_STATUS_OK;
	}
	#endif // VK_USE_PLATFORM_FUCHSIA

	static Connection* cast(anv_connection* connection)
	{
	return static_cast<Connection*>(connection);
	}

	private:
	#if VK_USE_PLATFORM_FUCHSIA
	magma_sysmem_connection_t sysmem_connection_{};
	#endif // #if VK_USE_PLATFORM_FUCHSIA
	InflightList* inflight_list_;
	};

	anv_connection* AnvMagmaCreateConnection(magma_connection_t connection)
	{
	return new Connection(connection,
	magma_connection_get_notification_channel_handle(connection));
	}

	void AnvMagmaReleaseConnection(anv_connection* connection)
	{
	delete static_cast<Connection*>(connection);
	}

	#if VK_USE_PLATFORM_FUCHSIA
	magma_status_t AnvMagmaGetSysmemConnection(struct anv_connection* connection,
	magma_sysmem_connection_t* sysmem_connection_out)
	{
	return Connection::cast(connection)->GetSysmemConnection(sysmem_connection_out);
	}
	#endif // VK_USE_PLATFORM_FUCHSIA

	void AnvMagmaConnectionServiceNotifications(anv_connection* connection)
	{
	InflightList_TryUpdate(Connection::cast(connection)->inflight_list(),
	Connection::cast(connection)->magma_connection());
	}

	magma_status_t AnvMagmaConnectionWait(anv_connection* connection, uint64_t buffer_id,
	uint64_t timeout_ns)
	{
	return InflightList_WaitForBuffer(Connection::cast(connection)->inflight_list(),
	Connection::cast(connection)->magma_connection(),
	connection->vk.notification_channel, buffer_id, timeout_ns);
	}

	int AnvMagmaConnectionExec(anv_connection* connection, uint32_t context_id,
	struct drm_i915_gem_execbuffer2* execbuf,
	struct anv_magma_buffer* buffers[])
	{
	if (execbuf->buffer_count == 0)
	return 0;

	auto exec_objects = reinterpret_cast<drm_i915_gem_exec_object2*>(execbuf->buffers_ptr);

	std::vector<magma_exec_resource> resources;
	resources.reserve(execbuf->buffer_count);

	for (uint32_t i = 0; i < execbuf->buffer_count; i++) {
	auto buffer = static_cast<Buffer*>(buffers[i]);

	uint64_t offset = 0;
	uint64_t length = exec_objects[i].rsvd2;

	resources.push_back({
	.buffer_id = buffer->id,
	.offset = offset,
	.length = length,
	});

	if (!is_page_aligned(offset))
	return ANV_MAGMA_DRET_MSG(-1, "offset (0x%" PRIx64 ") not page aligned", offset);

	uint64_t gpu_addr = intel_48b_address(exec_objects[i].offset);
	length = round_up(length, page_size());

	Buffer::Segment segment;
	bool has_mapping = buffer->HasMapping(gpu_addr, &segment);
	if (has_mapping) {
	assert(offset == segment.offset);
	if (length > segment.length) {
	// Growing an existing mapping.
	buffer->RemoveMapping(gpu_addr);
	has_mapping = false;
	} else {
	assert(length == segment.length);
	}
	}

	if (!has_mapping) {
	LOG_VERBOSE("mapping to gpu addr 0x%" PRIx64 ": id %" PRIu64 " offset %" PRIu64
	" length %" PRIu64,
	gpu_addr, buffer->id, offset, length);
	constexpr uint64_t kMapFlags =
	MAGMA_MAP_FLAG_READ \| MAGMA_MAP_FLAG_WRITE \| MAGMA_MAP_FLAG_EXECUTE;
	magma_connection_map_buffer(Connection::cast(connection)->magma_connection(),
	gpu_addr, buffer->get(), offset, length, kMapFlags);

	buffer->AddMapping(offset, length, gpu_addr);
	}
	}

	uint32_t syncobj_count = execbuf->num_cliprects;

	std::vector<uint64_t> semaphore_ids;
	semaphore_ids.reserve(syncobj_count);
	uint32_t wait_semaphore_count = 0;

	// Wait semaphores first, then signal
	for (uint32_t i = 0; i < syncobj_count; i++) {
	auto& syncobj = reinterpret_cast<drm_i915_gem_exec_fence*>(execbuf->cliprects_ptr)[i];
	if (syncobj.flags & I915_EXEC_FENCE_WAIT) {
	semaphore_ids.push_back(reinterpret_cast<vk_magma_syncobj*>(syncobj.handle)->id);
	wait_semaphore_count++;
	}
	}
	for (uint32_t i = 0; i < syncobj_count; i++) {
	auto& syncobj = reinterpret_cast<drm_i915_gem_exec_fence*>(execbuf->cliprects_ptr)[i];
	if (syncobj.flags & I915_EXEC_FENCE_SIGNAL) {
	semaphore_ids.push_back(reinterpret_cast<vk_magma_syncobj*>(syncobj.handle)->id);
	}
	}

	magma_exec_command_buffer command_buffer = {.resource_index =
	execbuf->buffer_count - 1, // by drm convention
	.start_offset = execbuf->batch_start_offset};

	magma_command_descriptor descriptor = {.resource_count = execbuf->buffer_count,
	.command_buffer_count = 1,
	.wait_semaphore_count = wait_semaphore_count,
	.signal_semaphore_count =
	syncobj_count - wait_semaphore_count,
	.resources = resources.data(),
	.command_buffers = &command_buffer,
	.semaphore_ids = semaphore_ids.data(),
	.flags = 0};

	// Add to inflight list first to avoid race with any other thread reading completions from the
	// notification channel, in case this thread is preempted just after sending the command buffer
	// and the completion happens quickly.
	InflightList_AddAndUpdate(Connection::cast(connection)->inflight_list(),
	Connection::cast(connection)->magma_connection(), resources.data(),
	execbuf->buffer_count);

	magma_connection_execute_command(Connection::cast(connection)->magma_connection(),
	context_id, &descriptor);

	return 0;
	}

	anv_magma_buffer* AnvMagmaCreateBuffer(anv_connection* connection, magma_buffer_t buffer,
	magma_buffer_id_t buffer_id)
	{
	return new Buffer(buffer, buffer_id);
	}

	void AnvMagmaReleaseBuffer(anv_connection* connection, anv_magma_buffer* anv_buffer)
	{
	auto buffer = static_cast<Buffer*>(anv_buffer);
	// Hardware mappings are released when the buffer is released.
	buffer->release(Connection::cast(connection)->magma_connection());
	delete buffer;
	}