// SPDX-License-Identifier: GPL-2.0 #include <drm/drm_drv.h>#include <drm/drm_kunit_helpers.h>#include <drm/drm_managed.h> #include <kunit/resource.h> #include <linux/device.h> /* Ought to be enough for anybody */#define TEST_TIMEOUT_MS	100 struct managed_test_priv {	struct drm_device *drm;	bool action_done;	wait_queue_head_t action_wq;}; static void drm_action(struct drm_device *drm, void *ptr){	struct managed_test_priv *priv = ptr; 	priv->action_done = true;	wake_up_interruptible(&priv->action_wq);} /* * The test verifies that the release action is called when * drmm_release_action is called. */static void drm_test_managed_release_action(struct kunit *test){	struct managed_test_priv *priv = test->priv;	int ret; 	ret = drmm_add_action_or_reset(priv->drm, drm_action, priv);	KUNIT_EXPECT_EQ(test, ret, 0); 	ret = drm_dev_register(priv->drm, 0);	KUNIT_ASSERT_EQ(test, ret, 0); 	drmm_release_action(priv->drm, drm_action, priv);	ret = wait_event_interruptible_timeout(priv->action_wq, priv->action_done,					       msecs_to_jiffies(TEST_TIMEOUT_MS));	KUNIT_EXPECT_GT(test, ret, 0); 	drm_dev_unregister(priv->drm);	drm_kunit_helper_free_device(test, priv->drm->dev);} /* * The test verifies that the release action is called automatically when the * device is released. */static void drm_test_managed_run_action(struct kunit *test){	struct managed_test_priv *priv = test->priv;	int ret; 	ret = drmm_add_action_or_reset(priv->drm, drm_action, priv);	KUNIT_EXPECT_EQ(test, ret, 0); 	ret = drm_dev_register(priv->drm, 0);	KUNIT_ASSERT_EQ(test, ret, 0); 	drm_dev_unregister(priv->drm);	drm_kunit_helper_free_device(test, priv->drm->dev); 	ret = wait_event_interruptible_timeout(priv->action_wq, priv->action_done,					       msecs_to_jiffies(TEST_TIMEOUT_MS));	KUNIT_EXPECT_GT(test, ret, 0);} static int drm_managed_test_init(struct kunit *test){	struct managed_test_priv *priv;	struct device *dev; 	priv = kunit_kzalloc(test, sizeof(*priv), GFP_KERNEL);	KUNIT_ASSERT_NOT_ERR_OR_NULL(test, priv);	init_waitqueue_head(&priv->action_wq); 	dev = drm_kunit_helper_alloc_device(test);	KUNIT_ASSERT_NOT_ERR_OR_NULL(test, dev); 	/*	 * DRM device can't be embedded in priv, since priv->action_done needs	 * to remain allocated beyond both parent device and drm_device	 * lifetime.	 */	priv->drm = __drm_kunit_helper_alloc_drm_device(test, dev, sizeof(*priv->drm), 0,							DRIVER_MODESET);	KUNIT_ASSERT_NOT_ERR_OR_NULL(test, priv->drm); 	test->priv = priv; 	return 0;} static struct kunit_case drm_managed_tests[] = {	KUNIT_CASE(drm_test_managed_release_action),	KUNIT_CASE(drm_test_managed_run_action),	{}}; static struct kunit_suite drm_managed_test_suite = {	.name = "drm_managed",	.init = drm_managed_test_init,	.test_cases = drm_managed_tests}; kunit_test_suite(drm_managed_test_suite); MODULE_AUTHOR("Maxime Ripard <maxime@cerno.tech>");MODULE_DESCRIPTION("KUnit DRM managed test suite");MODULE_LICENSE("GPL");