111 lines
2.7 KiB
C
111 lines
2.7 KiB
C
|
// 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>
|
||
|
#include <linux/platform_device.h>
|
||
|
|
||
|
#define KUNIT_DEVICE_NAME "drm-kunit-mock-device"
|
||
|
|
||
|
static const struct drm_mode_config_funcs drm_mode_config_funcs = {
|
||
|
};
|
||
|
|
||
|
static int fake_probe(struct platform_device *pdev)
|
||
|
{
|
||
|
return 0;
|
||
|
}
|
||
|
|
||
|
static int fake_remove(struct platform_device *pdev)
|
||
|
{
|
||
|
return 0;
|
||
|
}
|
||
|
|
||
|
static struct platform_driver fake_platform_driver = {
|
||
|
.probe = fake_probe,
|
||
|
.remove = fake_remove,
|
||
|
.driver = {
|
||
|
.name = KUNIT_DEVICE_NAME,
|
||
|
},
|
||
|
};
|
||
|
|
||
|
/**
|
||
|
* drm_kunit_helper_alloc_device - Allocate a mock device for a KUnit test
|
||
|
* @test: The test context object
|
||
|
*
|
||
|
* This allocates a fake struct &device to create a mock for a KUnit
|
||
|
* test. The device will also be bound to a fake driver. It will thus be
|
||
|
* able to leverage the usual infrastructure and most notably the
|
||
|
* device-managed resources just like a "real" device.
|
||
|
*
|
||
|
* Callers need to make sure drm_kunit_helper_free_device() on the
|
||
|
* device when done.
|
||
|
*
|
||
|
* Returns:
|
||
|
* A pointer to the new device, or an ERR_PTR() otherwise.
|
||
|
*/
|
||
|
struct device *drm_kunit_helper_alloc_device(struct kunit *test)
|
||
|
{
|
||
|
struct platform_device *pdev;
|
||
|
int ret;
|
||
|
|
||
|
ret = platform_driver_register(&fake_platform_driver);
|
||
|
KUNIT_ASSERT_EQ(test, ret, 0);
|
||
|
|
||
|
pdev = platform_device_alloc(KUNIT_DEVICE_NAME, PLATFORM_DEVID_NONE);
|
||
|
KUNIT_ASSERT_NOT_ERR_OR_NULL(test, pdev);
|
||
|
|
||
|
ret = platform_device_add(pdev);
|
||
|
KUNIT_ASSERT_EQ(test, ret, 0);
|
||
|
|
||
|
return &pdev->dev;
|
||
|
}
|
||
|
EXPORT_SYMBOL_GPL(drm_kunit_helper_alloc_device);
|
||
|
|
||
|
/**
|
||
|
* drm_kunit_helper_free_device - Frees a mock device
|
||
|
* @test: The test context object
|
||
|
* @dev: The device to free
|
||
|
*
|
||
|
* Frees a device allocated with drm_kunit_helper_alloc_device().
|
||
|
*/
|
||
|
void drm_kunit_helper_free_device(struct kunit *test, struct device *dev)
|
||
|
{
|
||
|
struct platform_device *pdev = to_platform_device(dev);
|
||
|
|
||
|
platform_device_unregister(pdev);
|
||
|
platform_driver_unregister(&fake_platform_driver);
|
||
|
}
|
||
|
EXPORT_SYMBOL_GPL(drm_kunit_helper_free_device);
|
||
|
|
||
|
struct drm_device *
|
||
|
__drm_kunit_helper_alloc_drm_device_with_driver(struct kunit *test,
|
||
|
struct device *dev,
|
||
|
size_t size, size_t offset,
|
||
|
const struct drm_driver *driver)
|
||
|
{
|
||
|
struct drm_device *drm;
|
||
|
void *container;
|
||
|
int ret;
|
||
|
|
||
|
container = __devm_drm_dev_alloc(dev, driver, size, offset);
|
||
|
if (IS_ERR(container))
|
||
|
return ERR_CAST(container);
|
||
|
|
||
|
drm = container + offset;
|
||
|
drm->mode_config.funcs = &drm_mode_config_funcs;
|
||
|
|
||
|
ret = drmm_mode_config_init(drm);
|
||
|
if (ret)
|
||
|
return ERR_PTR(ret);
|
||
|
|
||
|
return drm;
|
||
|
}
|
||
|
EXPORT_SYMBOL_GPL(__drm_kunit_helper_alloc_drm_device_with_driver);
|
||
|
|
||
|
MODULE_AUTHOR("Maxime Ripard <maxime@cerno.tech>");
|
||
|
MODULE_LICENSE("GPL");
|