1 files changed, 12 insertions, 0 deletions

diff --git a/drivers/gpu/drm/virtio/virtgpu_display.c b/drivers/gpu/drm/virtio/virtgpu_display.c
index 8f8fed471e34..b5580b11a063 100644
--- a/drivers/gpu/drm/virtio/virtgpu_display.c
+++ b/drivers/gpu/drm/virtio/virtgpu_display.c
@@ -169,6 +169,12 @@ static int virtio_gpu_conn_get_modes(struct drm_connector *connector)
         struct drm_display_mode *mode = NULL;
         int count, width, height;
 
+        if (output->edid) {
+                count = drm_add_edid_modes(connector, output->edid);
+                if (count)
+                        return count;
+        }
+
         width  = le32_to_cpu(output->info.r.width);
         height = le32_to_cpu(output->info.r.height);
         count = drm_add_modes_noedid(connector, XRES_MAX, YRES_MAX);
@@ -287,6 +293,8 @@ static int vgdev_output_init(struct virtio_gpu_device *vgdev, int index)
         drm_connector_init(dev, connector, &virtio_gpu_connector_funcs,
                            DRM_MODE_CONNECTOR_VIRTUAL);
         drm_connector_helper_add(connector, &virtio_gpu_conn_helper_funcs);
+        if (vgdev->has_edid)
+                drm_connector_attach_edid_property(connector);
 
         drm_encoder_init(dev, encoder, &virtio_gpu_enc_funcs,
                          DRM_MODE_ENCODER_VIRTUAL, NULL);
@@ -378,6 +386,10 @@ int virtio_gpu_modeset_init(struct virtio_gpu_device *vgdev)
 
 void virtio_gpu_modeset_fini(struct virtio_gpu_device *vgdev)
 {
+        int i;
+
+        for (i = 0 ; i < vgdev->num_scanouts; ++i)
+                kfree(vgdev->outputs[i].edid);
         virtio_gpu_fbdev_fini(vgdev);
         drm_mode_config_cleanup(vgdev->ddev);
 }