Add a list of possible CRTCs to the plane configuration and helpers to
attach, detach and get the primary and cursor planes attached to a CRTC.

Now that the default configuration has its planes and CRTC correctly
attached, configure the output following the configuration.

Signed-off-by: Louis Chauvet <[email protected]>
Signed-off-by: José Expósito <[email protected]>
---
 drivers/gpu/drm/vkms/tests/vkms_config_test.c | 185 ++++++++++++++++++
 drivers/gpu/drm/vkms/vkms_config.c            | 166 ++++++++++++++--
 drivers/gpu/drm/vkms/vkms_config.h            |  55 +++++-
 drivers/gpu/drm/vkms/vkms_output.c            |  77 ++++++--
 4 files changed, 446 insertions(+), 37 deletions(-)

diff --git a/drivers/gpu/drm/vkms/tests/vkms_config_test.c 
b/drivers/gpu/drm/vkms/tests/vkms_config_test.c
index b5907b060a5c..5e698616491a 100644
--- a/drivers/gpu/drm/vkms/tests/vkms_config_test.c
+++ b/drivers/gpu/drm/vkms/tests/vkms_config_test.c
@@ -87,6 +87,18 @@ static void vkms_config_test_default_config(struct kunit 
*test)
        KUNIT_EXPECT_EQ(test, vkms_config_crtc_get_writeback(crtc_cfg),
                        params->enable_writeback);
 
+       list_for_each_entry(plane_cfg, &config->planes, link) {
+               struct vkms_config_crtc *possible_crtc;
+               int n_possible_crtcs = 0;
+               unsigned long idx = 0;
+
+               xa_for_each(&plane_cfg->possible_crtcs, idx, possible_crtc) {
+                       KUNIT_EXPECT_PTR_EQ(test, crtc_cfg, possible_crtc);
+                       n_possible_crtcs++;
+               }
+               KUNIT_EXPECT_EQ(test, n_possible_crtcs, 1);
+       }
+
        KUNIT_EXPECT_TRUE(test, vkms_config_is_valid(config));
 
        vkms_config_destroy(config);
@@ -191,6 +203,8 @@ static void vkms_config_test_valid_plane_type(struct kunit 
*test)
 {
        struct vkms_config *config;
        struct vkms_config_plane *plane_cfg;
+       struct vkms_config_crtc *crtc_cfg;
+       int err;
 
        config = vkms_config_default_create(false, false, false);
        KUNIT_ASSERT_NOT_ERR_OR_NULL(test, config);
@@ -198,16 +212,26 @@ static void vkms_config_test_valid_plane_type(struct 
kunit *test)
        plane_cfg = list_first_entry(&config->planes, typeof(*plane_cfg), link);
        vkms_config_destroy_plane(plane_cfg);
 
+       crtc_cfg = list_first_entry(&config->crtcs, typeof(*crtc_cfg), link);
+
        /* Invalid: No primary plane */
        plane_cfg = vkms_config_add_plane(config);
        vkms_config_plane_set_type(plane_cfg, DRM_PLANE_TYPE_OVERLAY);
+       err = vkms_config_plane_attach_crtc(plane_cfg, crtc_cfg);
+       KUNIT_EXPECT_EQ(test, err, 0);
        KUNIT_EXPECT_FALSE(test, vkms_config_is_valid(config));
 
        /* Invalid: Multiple primary planes */
        plane_cfg = vkms_config_add_plane(config);
        vkms_config_plane_set_type(plane_cfg, DRM_PLANE_TYPE_PRIMARY);
+       err = vkms_config_plane_attach_crtc(plane_cfg, crtc_cfg);
+       KUNIT_EXPECT_EQ(test, err, 0);
+
        plane_cfg = vkms_config_add_plane(config);
        vkms_config_plane_set_type(plane_cfg, DRM_PLANE_TYPE_PRIMARY);
+       err = vkms_config_plane_attach_crtc(plane_cfg, crtc_cfg);
+       KUNIT_EXPECT_EQ(test, err, 0);
+
        KUNIT_EXPECT_FALSE(test, vkms_config_is_valid(config));
 
        /* Valid: One primary plane */
@@ -217,14 +241,50 @@ static void vkms_config_test_valid_plane_type(struct 
kunit *test)
        /* Invalid: Multiple cursor planes */
        plane_cfg = vkms_config_add_plane(config);
        vkms_config_plane_set_type(plane_cfg, DRM_PLANE_TYPE_CURSOR);
+       err = vkms_config_plane_attach_crtc(plane_cfg, crtc_cfg);
+       KUNIT_EXPECT_EQ(test, err, 0);
+
        plane_cfg = vkms_config_add_plane(config);
        vkms_config_plane_set_type(plane_cfg, DRM_PLANE_TYPE_CURSOR);
+       err = vkms_config_plane_attach_crtc(plane_cfg, crtc_cfg);
+       KUNIT_EXPECT_EQ(test, err, 0);
+
        KUNIT_EXPECT_FALSE(test, vkms_config_is_valid(config));
 
        /* Valid: One primary and one cursor plane */
        vkms_config_destroy_plane(plane_cfg);
        KUNIT_EXPECT_TRUE(test, vkms_config_is_valid(config));
 
+       /* Invalid: Second CRTC without primary plane */
+       crtc_cfg = vkms_config_add_crtc(config);
+       KUNIT_EXPECT_FALSE(test, vkms_config_is_valid(config));
+
+       /* Valid: Second CRTC with a primary plane */
+       plane_cfg = vkms_config_add_plane(config);
+       vkms_config_plane_set_type(plane_cfg, DRM_PLANE_TYPE_PRIMARY);
+       err = vkms_config_plane_attach_crtc(plane_cfg, crtc_cfg);
+       KUNIT_EXPECT_EQ(test, err, 0);
+       KUNIT_EXPECT_TRUE(test, vkms_config_is_valid(config));
+
+       vkms_config_destroy(config);
+}
+
+static void vkms_config_test_valid_plane_possible_crtcs(struct kunit *test)
+{
+       struct vkms_config *config;
+       struct vkms_config_plane *plane_cfg;
+       struct vkms_config_crtc *crtc_cfg;
+
+       config = vkms_config_default_create(false, false, false);
+       KUNIT_ASSERT_NOT_ERR_OR_NULL(test, config);
+
+       plane_cfg = list_first_entry(&config->planes, typeof(*plane_cfg), link);
+       crtc_cfg = list_first_entry(&config->crtcs, typeof(*crtc_cfg), link);
+
+       /* Invalid: Primary plane without a possible CRTC */
+       vkms_config_plane_detach_crtc(plane_cfg, crtc_cfg);
+       KUNIT_EXPECT_FALSE(test, vkms_config_is_valid(config));
+
        vkms_config_destroy(config);
 }
 
@@ -251,6 +311,128 @@ static void vkms_config_test_valid_crtc_number(struct 
kunit *test)
        vkms_config_destroy(config);
 }
 
+static void vkms_config_test_plane_attach_crtc(struct kunit *test)
+{
+       struct vkms_config *config;
+       struct vkms_config_plane *overlay_cfg;
+       struct vkms_config_plane *primary_cfg;
+       struct vkms_config_plane *cursor_cfg;
+       struct vkms_config_crtc *crtc_cfg;
+       int err;
+
+       config = vkms_config_create("test");
+       KUNIT_ASSERT_NOT_ERR_OR_NULL(test, config);
+
+       overlay_cfg = vkms_config_add_plane(config);
+       vkms_config_plane_set_type(overlay_cfg, DRM_PLANE_TYPE_OVERLAY);
+       primary_cfg = vkms_config_add_plane(config);
+       vkms_config_plane_set_type(primary_cfg, DRM_PLANE_TYPE_PRIMARY);
+       cursor_cfg = vkms_config_add_plane(config);
+       vkms_config_plane_set_type(cursor_cfg, DRM_PLANE_TYPE_CURSOR);
+
+       crtc_cfg = vkms_config_add_crtc(config);
+
+       /* No primary or cursor planes */
+       KUNIT_EXPECT_NULL(test, vkms_config_crtc_primary_plane(config, 
crtc_cfg));
+       KUNIT_EXPECT_NULL(test, vkms_config_crtc_cursor_plane(config, 
crtc_cfg));
+
+       /* Overlay plane, but no primary or cursor planes */
+       err = vkms_config_plane_attach_crtc(overlay_cfg, crtc_cfg);
+       KUNIT_EXPECT_EQ(test, err, 0);
+       KUNIT_EXPECT_NULL(test, vkms_config_crtc_primary_plane(config, 
crtc_cfg));
+       KUNIT_EXPECT_NULL(test, vkms_config_crtc_cursor_plane(config, 
crtc_cfg));
+
+       /* Primary plane, attaching it twice must fail */
+       err = vkms_config_plane_attach_crtc(primary_cfg, crtc_cfg);
+       KUNIT_EXPECT_EQ(test, err, 0);
+       err = vkms_config_plane_attach_crtc(primary_cfg, crtc_cfg);
+       KUNIT_EXPECT_NE(test, err, 0);
+       KUNIT_EXPECT_PTR_EQ(test,
+                           vkms_config_crtc_primary_plane(config, crtc_cfg),
+                           primary_cfg);
+       KUNIT_EXPECT_NULL(test, vkms_config_crtc_cursor_plane(config, 
crtc_cfg));
+
+       /* Primary and cursor planes */
+       err = vkms_config_plane_attach_crtc(cursor_cfg, crtc_cfg);
+       KUNIT_EXPECT_EQ(test, err, 0);
+       KUNIT_EXPECT_PTR_EQ(test,
+                           vkms_config_crtc_primary_plane(config, crtc_cfg),
+                           primary_cfg);
+       KUNIT_EXPECT_PTR_EQ(test,
+                           vkms_config_crtc_cursor_plane(config, crtc_cfg),
+                           cursor_cfg);
+
+       /* Detach primary and destroy cursor plane */
+       vkms_config_plane_detach_crtc(overlay_cfg, crtc_cfg);
+       vkms_config_plane_detach_crtc(primary_cfg, crtc_cfg);
+       vkms_config_destroy_plane(cursor_cfg);
+       KUNIT_EXPECT_NULL(test, vkms_config_crtc_primary_plane(config, 
crtc_cfg));
+       KUNIT_EXPECT_NULL(test, vkms_config_crtc_cursor_plane(config, 
crtc_cfg));
+
+       vkms_config_destroy(config);
+}
+
+static void vkms_config_test_plane_get_possible_crtcs(struct kunit *test)
+{
+       struct vkms_config *config;
+       struct vkms_config_plane *plane_cfg1, *plane_cfg2;
+       struct vkms_config_crtc *crtc_cfg1, *crtc_cfg2;
+       struct vkms_config_crtc **array;
+       size_t length;
+       int err;
+
+       config = vkms_config_create("test");
+       KUNIT_ASSERT_NOT_ERR_OR_NULL(test, config);
+
+       plane_cfg1 = vkms_config_add_plane(config);
+       plane_cfg2 = vkms_config_add_plane(config);
+       crtc_cfg1 = vkms_config_add_crtc(config);
+       crtc_cfg2 = vkms_config_add_crtc(config);
+
+       /* No possible CRTCs */
+       array = vkms_config_plane_get_possible_crtcs(plane_cfg1, &length);
+       KUNIT_ASSERT_EQ(test, length, 0);
+       KUNIT_ASSERT_NULL(test, array);
+
+       array = vkms_config_plane_get_possible_crtcs(plane_cfg2, &length);
+       KUNIT_ASSERT_EQ(test, length, 0);
+       KUNIT_ASSERT_NULL(test, array);
+
+       /* Plane 1 attached to CRTC 1 and 2 */
+       err = vkms_config_plane_attach_crtc(plane_cfg1, crtc_cfg1);
+       KUNIT_EXPECT_EQ(test, err, 0);
+       err = vkms_config_plane_attach_crtc(plane_cfg1, crtc_cfg2);
+       KUNIT_EXPECT_EQ(test, err, 0);
+
+       array = vkms_config_plane_get_possible_crtcs(plane_cfg1, &length);
+       KUNIT_ASSERT_EQ(test, length, 2);
+       KUNIT_ASSERT_PTR_EQ(test, array[0], crtc_cfg1);
+       KUNIT_ASSERT_PTR_EQ(test, array[1], crtc_cfg2);
+       kfree(array);
+
+       array = vkms_config_plane_get_possible_crtcs(plane_cfg2, &length);
+       KUNIT_ASSERT_EQ(test, length, 0);
+       KUNIT_ASSERT_NULL(test, array);
+
+       /* Plane 1 attached to CRTC 1 and plane 2 to CRTC 2 */
+       vkms_config_plane_detach_crtc(plane_cfg1, crtc_cfg2);
+
+       array = vkms_config_plane_get_possible_crtcs(plane_cfg1, &length);
+       KUNIT_ASSERT_EQ(test, length, 1);
+       KUNIT_ASSERT_PTR_EQ(test, array[0], crtc_cfg1);
+       kfree(array);
+
+       err = vkms_config_plane_attach_crtc(plane_cfg2, crtc_cfg2);
+       KUNIT_EXPECT_EQ(test, err, 0);
+
+       array = vkms_config_plane_get_possible_crtcs(plane_cfg2, &length);
+       KUNIT_ASSERT_EQ(test, length, 1);
+       KUNIT_ASSERT_PTR_EQ(test, array[0], crtc_cfg2);
+       kfree(array);
+
+       vkms_config_destroy(config);
+}
+
 static struct kunit_case vkms_config_test_cases[] = {
        KUNIT_CASE(vkms_config_test_empty_config),
        KUNIT_CASE_PARAM(vkms_config_test_default_config,
@@ -259,7 +441,10 @@ static struct kunit_case vkms_config_test_cases[] = {
        KUNIT_CASE(vkms_config_test_get_crtcs),
        KUNIT_CASE(vkms_config_test_valid_plane_number),
        KUNIT_CASE(vkms_config_test_valid_plane_type),
+       KUNIT_CASE(vkms_config_test_valid_plane_possible_crtcs),
        KUNIT_CASE(vkms_config_test_valid_crtc_number),
+       KUNIT_CASE(vkms_config_test_plane_attach_crtc),
+       KUNIT_CASE(vkms_config_test_plane_get_possible_crtcs),
        {}
 };
 
diff --git a/drivers/gpu/drm/vkms/vkms_config.c 
b/drivers/gpu/drm/vkms/vkms_config.c
index 4128892836d7..a2ce4905589b 100644
--- a/drivers/gpu/drm/vkms/vkms_config.c
+++ b/drivers/gpu/drm/vkms/vkms_config.c
@@ -50,13 +50,20 @@ struct vkms_config *vkms_config_default_create(bool 
enable_cursor,
                goto err_alloc;
        vkms_config_crtc_set_writeback(crtc_cfg, enable_writeback);
 
+       if (vkms_config_plane_attach_crtc(plane_cfg, crtc_cfg))
+               goto err_alloc;
+
        if (enable_overlay) {
                for (n = 0; n < NUM_OVERLAY_PLANES; n++) {
                        plane_cfg = vkms_config_add_plane(config);
                        if (IS_ERR(plane_cfg))
                                goto err_alloc;
+
                        vkms_config_plane_set_type(plane_cfg,
                                                   DRM_PLANE_TYPE_OVERLAY);
+
+                       if (vkms_config_plane_attach_crtc(plane_cfg, crtc_cfg))
+                               goto err_alloc;
                }
        }
 
@@ -64,7 +71,11 @@ struct vkms_config *vkms_config_default_create(bool 
enable_cursor,
                plane_cfg = vkms_config_add_plane(config);
                if (IS_ERR(plane_cfg))
                        goto err_alloc;
+
                vkms_config_plane_set_type(plane_cfg, DRM_PLANE_TYPE_CURSOR);
+
+               if (vkms_config_plane_attach_crtc(plane_cfg, crtc_cfg))
+                       goto err_alloc;
        }
 
        return config;
@@ -156,31 +167,39 @@ static bool valid_plane_number(struct vkms_config *config)
        return true;
 }
 
-static bool valid_plane_type(struct vkms_config *config)
+static bool valid_plane_type(struct vkms_config *config,
+                            struct vkms_config_crtc *crtc_cfg)
 {
        struct vkms_config_plane *plane_cfg;
        bool has_primary_plane = false;
        bool has_cursor_plane = false;
 
        list_for_each_entry(plane_cfg, &config->planes, link) {
+               struct vkms_config_crtc *possible_crtc;
+               unsigned long idx = 0;
                enum drm_plane_type type;
 
                type = vkms_config_plane_get_type(plane_cfg);
 
-               if (type == DRM_PLANE_TYPE_PRIMARY) {
-                       if (has_primary_plane) {
-                               pr_err("Multiple primary planes\n");
-                               return false;
-                       }
+               xa_for_each(&plane_cfg->possible_crtcs, idx, possible_crtc) {
+                       if (possible_crtc != crtc_cfg)
+                               continue;
 
-                       has_primary_plane = true;
-               } else if (type == DRM_PLANE_TYPE_CURSOR) {
-                       if (has_cursor_plane) {
-                               pr_err("Multiple cursor planes\n");
-                               return false;
-                       }
+                       if (type == DRM_PLANE_TYPE_PRIMARY) {
+                               if (has_primary_plane) {
+                                       pr_err("Multiple primary planes\n");
+                                       return false;
+                               }
 
-                       has_cursor_plane = true;
+                               has_primary_plane = true;
+                       } else if (type == DRM_PLANE_TYPE_CURSOR) {
+                               if (has_cursor_plane) {
+                                       pr_err("Multiple cursor planes\n");
+                                       return false;
+                               }
+
+                               has_cursor_plane = true;
+                       }
                }
        }
 
@@ -192,6 +211,20 @@ static bool valid_plane_type(struct vkms_config *config)
        return true;
 }
 
+static bool valid_plane_possible_crtcs(struct vkms_config *config)
+{
+       struct vkms_config_plane *plane_cfg;
+
+       list_for_each_entry(plane_cfg, &config->planes, link) {
+               if (xa_empty(&plane_cfg->possible_crtcs)) {
+                       pr_err("All planes must have at least one possible 
CRTC\n");
+                       return false;
+               }
+       }
+
+       return true;
+}
+
 static bool valid_crtc_number(struct vkms_config *config)
 {
        size_t n_crtcs;
@@ -207,15 +240,22 @@ static bool valid_crtc_number(struct vkms_config *config)
 
 bool vkms_config_is_valid(struct vkms_config *config)
 {
+       struct vkms_config_crtc *crtc_cfg;
+
        if (!valid_plane_number(config))
                return false;
 
        if (!valid_crtc_number(config))
                return false;
 
-       if (!valid_plane_type(config))
+       if (!valid_plane_possible_crtcs(config))
                return false;
 
+       list_for_each_entry(crtc_cfg, &config->crtcs, link) {
+               if (!valid_plane_type(config, crtc_cfg))
+                       return false;
+       }
+
        return true;
 }
 
@@ -265,6 +305,7 @@ struct vkms_config_plane *vkms_config_add_plane(struct 
vkms_config *config)
                return ERR_PTR(-ENOMEM);
 
        vkms_config_plane_set_type(plane_cfg, DRM_PLANE_TYPE_OVERLAY);
+       xa_init_flags(&plane_cfg->possible_crtcs, XA_FLAGS_ALLOC);
 
        list_add_tail(&plane_cfg->link, &config->planes);
 
@@ -273,10 +314,69 @@ struct vkms_config_plane *vkms_config_add_plane(struct 
vkms_config *config)
 
 void vkms_config_destroy_plane(struct vkms_config_plane *plane_cfg)
 {
+       xa_destroy(&plane_cfg->possible_crtcs);
        list_del(&plane_cfg->link);
        kfree(plane_cfg);
 }
 
+int __must_check vkms_config_plane_attach_crtc(struct vkms_config_plane 
*plane_cfg,
+                                              struct vkms_config_crtc 
*crtc_cfg)
+{
+       struct vkms_config_crtc *possible_crtc;
+       unsigned long idx = 0;
+       u32 crtc_idx = 0;
+
+       xa_for_each(&plane_cfg->possible_crtcs, idx, possible_crtc) {
+               if (possible_crtc == crtc_cfg)
+                       return -EINVAL;
+       }
+
+       return xa_alloc(&plane_cfg->possible_crtcs, &crtc_idx, crtc_cfg,
+                       xa_limit_32b, GFP_KERNEL);
+}
+
+void vkms_config_plane_detach_crtc(struct vkms_config_plane *plane_cfg,
+                                  struct vkms_config_crtc *crtc_cfg)
+{
+       struct vkms_config_crtc *possible_crtc;
+       unsigned long idx = 0;
+
+       xa_for_each(&plane_cfg->possible_crtcs, idx, possible_crtc) {
+               if (possible_crtc == crtc_cfg)
+                       xa_erase(&plane_cfg->possible_crtcs, idx);
+       }
+}
+
+struct vkms_config_crtc **vkms_config_plane_get_possible_crtcs(struct 
vkms_config_plane *plane_cfg,
+                                                              size_t 
*out_length)
+{
+       struct vkms_config_crtc **array;
+       struct vkms_config_crtc *possible_crtc;
+       unsigned long idx;
+       size_t length = 0;
+       int n = 0;
+
+       xa_for_each(&plane_cfg->possible_crtcs, idx, possible_crtc)
+               length++;
+
+       if (length == 0) {
+               *out_length = length;
+               return NULL;
+       }
+
+       array = kmalloc_array(length, sizeof(*array), GFP_KERNEL);
+       if (!array)
+               return ERR_PTR(-ENOMEM);
+
+       xa_for_each(&plane_cfg->possible_crtcs, idx, possible_crtc) {
+               array[n] = possible_crtc;
+               n++;
+       }
+
+       *out_length = length;
+       return array;
+}
+
 struct vkms_config_crtc *vkms_config_add_crtc(struct vkms_config *config)
 {
        struct vkms_config_crtc *crtc_cfg;
@@ -295,6 +395,44 @@ struct vkms_config_crtc *vkms_config_add_crtc(struct 
vkms_config *config)
 void vkms_config_destroy_crtc(struct vkms_config *config,
                              struct vkms_config_crtc *crtc_cfg)
 {
+       struct vkms_config_plane *plane_cfg;
+
+       list_for_each_entry(plane_cfg, &config->planes, link)
+               vkms_config_plane_detach_crtc(plane_cfg, crtc_cfg);
+
        list_del(&crtc_cfg->link);
        kfree(crtc_cfg);
 }
+
+static struct vkms_config_plane *vkms_config_crtc_get_plane(const struct 
vkms_config *config,
+                                                           struct 
vkms_config_crtc *crtc_cfg,
+                                                           enum drm_plane_type 
type)
+{
+       struct vkms_config_plane *plane_cfg;
+       struct vkms_config_crtc *possible_crtc;
+       enum drm_plane_type current_type;
+       unsigned long idx;
+
+       list_for_each_entry(plane_cfg, &config->planes, link) {
+               current_type = vkms_config_plane_get_type(plane_cfg);
+
+               xa_for_each(&plane_cfg->possible_crtcs, idx, possible_crtc) {
+                       if (possible_crtc == crtc_cfg && current_type == type)
+                               return plane_cfg;
+               }
+       }
+
+       return NULL;
+}
+
+struct vkms_config_plane *vkms_config_crtc_primary_plane(const struct 
vkms_config *config,
+                                                        struct 
vkms_config_crtc *crtc_cfg)
+{
+       return vkms_config_crtc_get_plane(config, crtc_cfg, 
DRM_PLANE_TYPE_PRIMARY);
+}
+
+struct vkms_config_plane *vkms_config_crtc_cursor_plane(const struct 
vkms_config *config,
+                                                       struct vkms_config_crtc 
*crtc_cfg)
+{
+       return vkms_config_crtc_get_plane(config, crtc_cfg, 
DRM_PLANE_TYPE_CURSOR);
+}
diff --git a/drivers/gpu/drm/vkms/vkms_config.h 
b/drivers/gpu/drm/vkms/vkms_config.h
index a7828fe0c4b2..1f2e6c485d08 100644
--- a/drivers/gpu/drm/vkms/vkms_config.h
+++ b/drivers/gpu/drm/vkms/vkms_config.h
@@ -12,14 +12,12 @@
  * struct vkms_config - General configuration for VKMS driver
  *
  * @dev_name: Name of the device
- * @writeback: If true, a writeback buffer can be attached to the CRTC
  * @planes: List of planes configured for the device
  * @crtcs: List of CRTCs configured for the device
  * @dev: Used to store the current VKMS device. Only set when the device is 
instantiated.
  */
 struct vkms_config {
        const char *dev_name;
-       bool writeback;
        struct list_head planes;
        struct list_head crtcs;
        struct vkms_device *dev;
@@ -31,6 +29,7 @@ struct vkms_config {
  * @link: Link to the others planes in vkms_config
  * @type: Type of the plane. The creator of configuration needs to ensures that
  *        at least one primary plane is present.
+ * @possible_crtcs: Array of CRTCs that can be used with this plane
  * @plane: Internal usage. This pointer should never be considered as valid.
  *         It can be used to store a temporary reference to a VKMS plane during
  *         device creation. This pointer is not managed by the configuration 
and
@@ -40,6 +39,7 @@ struct vkms_config_plane {
        struct list_head link;
 
        enum drm_plane_type type;
+       struct xarray possible_crtcs;
 
        /* Internal usage */
        struct vkms_plane *plane;
@@ -194,6 +194,35 @@ vkms_config_plane_set_type(struct vkms_config_plane 
*plane_cfg,
        plane_cfg->type = type;
 }
 
+/**
+ * vkms_config_plane_attach_crtc - Attach a plane to a CRTC
+ * @plane_cfg: Plane to attach
+ * @crtc_cfg: CRTC to attach @plane_cfg to
+ */
+int __must_check vkms_config_plane_attach_crtc(struct vkms_config_plane 
*plane_cfg,
+                                              struct vkms_config_crtc 
*crtc_cfg);
+
+/**
+ * vkms_config_plane_attach_crtc - Detach a plane from a CRTC
+ * @plane_cfg: Plane to detach
+ * @crtc_cfg: CRTC to detach @plane_cfg from
+ */
+void vkms_config_plane_detach_crtc(struct vkms_config_plane *plane_cfg,
+                                  struct vkms_config_crtc *crtc_cfg);
+
+/**
+ * vkms_config_plane_get_possible_crtcs() - Return the array of possible CRTCs
+ * @plane_cfg: Plane to get the possible CRTCs from
+ * @out_length: Length of the returned array
+ *
+ * Returns:
+ * A list of pointers to the configurations. On success, the caller is
+ * responsible to free the returned array, but not its contents. On error,
+ * it returns an error and @out_length is invalid.
+ */
+struct vkms_config_crtc **vkms_config_plane_get_possible_crtcs(struct 
vkms_config_plane *plane_cfg,
+                                                              size_t 
*out_length);
+
 /**
  * vkms_config_add_crtc() - Add a new CRTC configuration
  * @config: Configuration to add the CRTC to
@@ -234,4 +263,26 @@ vkms_config_crtc_set_writeback(struct vkms_config_crtc 
*crtc_cfg,
        crtc_cfg->writeback = writeback;
 }
 
+/**
+ * vkms_config_crtc_primary_plane() - Return the primary plane for a CRTC
+ * @config: Configuration containing the CRTC
+ * @crtc_config: Target CRTC
+ *
+ * Returns:
+ * The primary plane or NULL if none is assigned yet.
+ */
+struct vkms_config_plane *vkms_config_crtc_primary_plane(const struct 
vkms_config *config,
+                                                        struct 
vkms_config_crtc *crtc_cfg);
+
+/**
+ * vkms_config_crtc_cursor_plane() - Return the cursor plane for a CRTC
+ * @config: Configuration containing the CRTC
+ * @crtc_config: Target CRTC
+ *
+ * Returns:
+ * The cursor plane or NULL if none is assigned yet.
+ */
+struct vkms_config_plane *vkms_config_crtc_cursor_plane(const struct 
vkms_config *config,
+                                                       struct vkms_config_crtc 
*crtc_cfg);
+
 #endif /* _VKMS_CONFIG_H_ */
diff --git a/drivers/gpu/drm/vkms/vkms_output.c 
b/drivers/gpu/drm/vkms/vkms_output.c
index b2ae269e5827..427d0aad8901 100644
--- a/drivers/gpu/drm/vkms/vkms_output.c
+++ b/drivers/gpu/drm/vkms/vkms_output.c
@@ -10,18 +10,24 @@ int vkms_output_init(struct vkms_device *vkmsdev)
        struct drm_device *dev = &vkmsdev->drm;
        struct vkms_connector *connector;
        struct drm_encoder *encoder;
-       struct vkms_output *output;
-       struct vkms_plane *primary = NULL, *cursor = NULL;
        struct vkms_config_plane **plane_cfgs = NULL;
        size_t n_planes;
+       struct vkms_config_crtc **crtc_cfgs = NULL;
+       size_t n_crtcs;
        int ret = 0;
        int writeback;
-       unsigned int n;
+       unsigned int n, i;
 
        plane_cfgs = vkms_config_get_planes(vkmsdev->config, &n_planes);
        if (IS_ERR(plane_cfgs))
                return PTR_ERR(plane_cfgs);
 
+       crtc_cfgs = vkms_config_get_crtcs(vkmsdev->config, &n_crtcs);
+       if (IS_ERR(crtc_cfgs)) {
+               ret = PTR_ERR(crtc_cfgs);
+               goto err_free;
+       }
+
        for (n = 0; n < n_planes; n++) {
                struct vkms_config_plane *plane_cfg;
                enum drm_plane_type type;
@@ -35,19 +41,54 @@ int vkms_output_init(struct vkms_device *vkmsdev)
                        ret = PTR_ERR(plane_cfg->plane);
                        goto err_free;
                }
+       }
+
+       for (n = 0; n < n_crtcs; n++) {
+               struct vkms_config_crtc *crtc_cfg;
+               struct vkms_config_plane *primary, *cursor;
+
+               crtc_cfg = crtc_cfgs[n];
+               primary = vkms_config_crtc_primary_plane(vkmsdev->config, 
crtc_cfg);
+               cursor = vkms_config_crtc_cursor_plane(vkmsdev->config, 
crtc_cfg);
+
+               crtc_cfg->crtc = vkms_crtc_init(dev, &primary->plane->base,
+                                               cursor ? &cursor->plane->base : 
NULL);
+               if (IS_ERR(crtc_cfg->crtc)) {
+                       DRM_ERROR("Failed to allocate CRTC\n");
+                       ret = PTR_ERR(crtc_cfg->crtc);
+                       goto err_free;
+               }
 
-               if (type == DRM_PLANE_TYPE_PRIMARY)
-                       primary = plane_cfg->plane;
-               else if (type == DRM_PLANE_TYPE_CURSOR)
-                       cursor = plane_cfg->plane;
+               /* Initialize the writeback component */
+               if (vkms_config_crtc_get_writeback(crtc_cfg)) {
+                       writeback = vkms_enable_writeback_connector(vkmsdev, 
crtc_cfg->crtc);
+                       if (writeback)
+                               DRM_ERROR("Failed to init writeback 
connector\n");
+               }
        }
 
-       output = vkms_crtc_init(dev, &primary->base,
-                               cursor ? &cursor->base : NULL);
-       if (IS_ERR(output)) {
-               DRM_ERROR("Failed to allocate CRTC\n");
-               ret = PTR_ERR(output);
-               goto err_free;
+       for (n = 0; n < n_planes; n++) {
+               struct vkms_config_plane *plane_cfg;
+               struct vkms_config_crtc **possible_crtcs;
+               size_t n_possible_crtcs;
+
+               plane_cfg = plane_cfgs[n];
+               possible_crtcs = vkms_config_plane_get_possible_crtcs(plane_cfg,
+                                                                     
&n_possible_crtcs);
+               if (IS_ERR(possible_crtcs)) {
+                       ret = PTR_ERR(possible_crtcs);
+                       goto err_free;
+               }
+
+               for (i = 0; i < n_possible_crtcs; i++) {
+                       struct vkms_config_crtc *possible_crtc;
+
+                       possible_crtc = possible_crtcs[i];
+                       plane_cfg->plane->base.possible_crtcs |=
+                               drm_crtc_mask(&possible_crtc->crtc->crtc);
+               }
+
+               kfree(possible_crtcs);
        }
 
        connector = vkms_connector_init(vkmsdev);
@@ -69,7 +110,7 @@ int vkms_output_init(struct vkms_device *vkmsdev)
                DRM_ERROR("Failed to init encoder\n");
                goto err_free;
        }
-       encoder->possible_crtcs = drm_crtc_mask(&output->crtc);
+       encoder->possible_crtcs = drm_crtc_mask(&crtc_cfgs[0]->crtc->crtc);
 
        /* Attach the encoder and the connector */
        ret = drm_connector_attach_encoder(&connector->base, encoder);
@@ -78,17 +119,11 @@ int vkms_output_init(struct vkms_device *vkmsdev)
                goto err_free;
        }
 
-       /* Initialize the writeback component */
-       if (vkmsdev->config->writeback) {
-               writeback = vkms_enable_writeback_connector(vkmsdev, output);
-               if (writeback)
-                       DRM_ERROR("Failed to init writeback connector\n");
-       }
-
        drm_mode_config_reset(dev);
 
 err_free:
        kfree(plane_cfgs);
+       kfree(crtc_cfgs);
 
        return ret;
 }
-- 
2.48.1

Reply via email to