Defer proxy deletion until after driver vk*Destroy

Author: solidpixel

generator/vk_layer/source/device.cpp

@@ -79,10 +79,15 @@ Device* Device::retrieve(
 }
 
 /* See header for documentation. */
-void Device::destroy(
-    Device* device
+std::unique_ptr<Device> Device::destroy(
+    VkDevice handle
 ) {
-    g_devices.erase(getDispatchKey(device));
+    void* key = getDispatchKey(handle);
+    assert(isInMap(key, g_devices));
+
+    auto device = std::move(g_devices.at(key));
+    g_devices.erase(key);
+    return device;
 }
 
 /* See header for documentation. */

generator/vk_layer/source/device.hpp

@@ -115,12 +115,17 @@ class Device
         VkCommandBuffer handle);
 
     /**
-     * @brief Drop a device from the global store of dispatchable devices.
+     * \brief Drop a device from the global store of dispatchable devices.
      *
-     * @param device   The device to drop.
+     * This must be called before the driver VkDevice has been destroyed, as
+     * we deference the native device handle to get the dispatch key.
+     *
+     * \param handle   The dispatchable device handle to use as an indirect lookup.
+     *
+     * \return Returns the ownership of the Device object to the caller.
      */
-    static void destroy(
-        Device* device);
+    static std::unique_ptr<Device> destroy(
+        VkDevice handle);
 
     /**
      * @brief Create a new layer device object.

generator/vk_layer/source/instance.cpp

@@ -70,10 +70,15 @@ Instance* Instance::retrieve(
 }
 
 /* See header for documentation. */
-void Instance::destroy(
-    Instance* instance
+std::unique_ptr<Instance> Instance::destroy(
+    VkInstance handle
 ) {
-    g_instances.erase(getDispatchKey(instance->instance));
+    void* key = getDispatchKey(handle);
+    assert(isInMap(key, g_instances));
+
+    auto instance = std::move(g_instances.at(key));
+    g_instances.erase(key);
+    return instance;
 }
 
 /* See header for documentation. */

generator/vk_layer/source/instance.hpp

@@ -99,12 +99,17 @@ class Instance
         VkPhysicalDevice handle);
 
     /**
-     * @brief Drop an instance from the global store of dispatchable instances.
+     * \brief Drop an instance from the global store of dispatchable instances.
      *
-     * @param instance   The instance to drop.
+     * This must be called before the driver VkInstance has been destroyed, as
+     * we deference the native instance handle to get the dispatch key.
+     *
+     * \param handle   The dispatchable instance handle to use as an indirect lookup.
+     *
+     * \return Returns the ownership of the Instance object to the caller.
      */
-    static void destroy(
-        Instance* instance);
+    static std::unique_ptr<Instance> destroy(
+        VkInstance handle);
 
     /**
      * @brief Create a new layer instance object.

layer_example/source/device.cpp

@@ -75,9 +75,15 @@ Device* Device::retrieve(VkCommandBuffer handle)
 }
 
 /* See header for documentation. */
-void Device::destroy(Device* device)
-{
-    g_devices.erase(getDispatchKey(device));
+std::unique_ptr<Device> Device::destroy(
+    VkDevice handle
+) {
+    void* key = getDispatchKey(handle);
+    assert(isInMap(key, g_devices));
+
+    auto device = std::move(g_devices.at(key));
+    g_devices.erase(key);
+    return device;
 }
 
 /* See header for documentation. */

layer_example/source/device.hpp

@@ -110,11 +110,17 @@ class Device
     static Device* retrieve(VkCommandBuffer handle);
 
     /**
-     * @brief Drop a device from the global store of dispatchable devices.
+     * \brief Drop a device from the global store of dispatchable devices.
      *
-     * @param device   The device to drop.
+     * This must be called before the driver VkDevice has been destroyed, as
+     * we deference the native device handle to get the dispatch key.
+     *
+     * \param handle   The dispatchable device handle to use as an indirect lookup.
+     *
+     * \return Returns the ownership of the Device object to the caller.
      */
-    static void destroy(Device* device);
+    static std::unique_ptr<Device> destroy(
+        VkDevice handle);
 
     /**
      * @brief Create a new layer device object.

layer_example/source/instance.cpp

@@ -64,9 +64,15 @@ Instance* Instance::retrieve(VkPhysicalDevice handle)
 }
 
 /* See header for documentation. */
-void Instance::destroy(Instance* instance)
-{
-    g_instances.erase(getDispatchKey(instance->instance));
+std::unique_ptr<Instance> Instance::destroy(
+    VkInstance handle
+) {
+    void* key = getDispatchKey(handle);
+    assert(isInMap(key, g_instances));
+
+    auto instance = std::move(g_instances.at(key));
+    g_instances.erase(key);
+    return instance;
 }
 
 /* See header for documentation. */

layer_example/source/instance.hpp

@@ -95,11 +95,17 @@ class Instance
     static Instance* retrieve(VkPhysicalDevice handle);
 
     /**
-     * @brief Drop an instance from the global store of dispatchable instances.
+     * \brief Drop an instance from the global store of dispatchable instances.
      *
-     * @param instance   The instance to drop.
+     * This must be called before the driver VkInstance has been destroyed, as
+     * we deference the native instance handle to get the dispatch key.
+     *
+     * \param handle   The dispatchable instance handle to use as an indirect lookup.
+     *
+     * \return Returns the ownership of the Instance object to the caller.
      */
-    static void destroy(Instance* instance);
+    static std::unique_ptr<Instance> destroy(
+        VkInstance handle);
 
     /**
      * @brief Create a new layer instance object.

layer_gpu_profile/source/device.cpp

@@ -82,9 +82,15 @@ Device* Device::retrieve(VkCommandBuffer handle)
 }
 
 /* See header for documentation. */
-void Device::destroy(Device* device)
-{
-    g_devices.erase(getDispatchKey(device));
+std::unique_ptr<Device> Device::destroy(
+    VkDevice handle
+) {
+    void* key = getDispatchKey(handle);
+    assert(isInMap(key, g_devices));
+
+    auto device = std::move(g_devices.at(key));
+    g_devices.erase(key);
+    return device;
 }
 
 /* See header for documentation. */

layer_gpu_profile/source/device.hpp

@@ -118,11 +118,17 @@ class Device
     static Device* retrieve(VkCommandBuffer handle);
 
     /**
-     * @brief Drop a device from the global store of dispatchable devices.
+     * \brief Drop a device from the global store of dispatchable devices.
      *
-     * @param device   The device to drop.
+     * This must be called before the driver VkDevice has been destroyed, as
+     * we deference the native device handle to get the dispatch key.
+     *
+     * \param handle   The dispatchable device handle to use as an indirect lookup.
+     *
+     * \return Returns the ownership of the Device object to the caller.
      */
-    static void destroy(Device* device);
+    static std::unique_ptr<Device> destroy(
+        VkDevice handle);
 
     /**
      * @brief Create a new layer device object.