// SPDX-License-Identifier: GPL-2.0/* * Copyright (C) Qualcomm Technologies, Inc. and/or its subsidiaries */ #include <kunit/fwnode.h>#include <kunit/test.h> #include <linux/fwnode.h>#include <linux/property.h> KUNIT_DEFINE_ACTION_WRAPPER(fwnode_remove_software_node_wrapper,			    fwnode_remove_software_node,			    struct fwnode_handle *); /** * kunit_fwnode_create_software_node() - Create a kunit-managed software node * @test: Test context * @properties: Properties to use to create the new software node * @parent: Parent of this software node * * Create a test-managed software node and return its firmware node handle. * The software node is removed after the test case completes. * * Returns: * Firmware node handle of the newly created software node or IS_ERR() on * failure. */struct fwnode_handle *kunit_fwnode_create_software_node(struct kunit *test,				  const struct property_entry *properties,				  const struct fwnode_handle *parent){	struct fwnode_handle *fwnode;	int ret; 	fwnode = fwnode_create_software_node(properties, parent);	if (IS_ERR(fwnode))		return fwnode; 	ret = kunit_add_action_or_reset(test, fwnode_remove_software_node_wrapper,					fwnode);	if (ret)		return ERR_PTR(ret); 	return fwnode;}EXPORT_SYMBOL_GPL(kunit_fwnode_create_software_node); KUNIT_DEFINE_ACTION_WRAPPER(software_node_unregister_wrapper,			    software_node_unregister,			    const struct software_node *); /** * kunit_software_node_register() - Register a kunit-managed software node * @test: Test context * @swnode: Software node to register * * Register a test-managed software node and return its firmware node handle. * The software node is unregistered after the test case completes. * * Returns: * Firmware node handle of the registered software node or IS_ERR() on failure. */struct fwnode_handle *kunit_software_node_register(struct kunit *test,			     const struct software_node *swnode){	struct fwnode_handle *fwnode;	int ret; 	ret = software_node_register(swnode);	if (ret)		return ERR_PTR(ret); 	fwnode = software_node_fwnode(swnode);	if (WARN_ON(!fwnode))		return ERR_PTR(-ENOENT); 	ret = kunit_add_action_or_reset(test, software_node_unregister_wrapper,					(void *)swnode);	if (ret)		return ERR_PTR(ret); 	return fwnode;}EXPORT_SYMBOL_GPL(kunit_software_node_register); KUNIT_DEFINE_ACTION_WRAPPER(software_node_unregister_node_group_wrapper,			    software_node_unregister_node_group,			    const struct software_node *const *); /** * kunit_software_node_register_node_group() - Register a kunit-managed software node group * @test: Test context * @nodes: Software node group to register * * Register a test-managed software node group. The nodes are unregistered * after the test case completes. * * Returns: * 0 on success, negative error number on failure. */int kunit_software_node_register_node_group(struct kunit *test,					    const struct software_node *const *nodes){	int ret; 	ret = software_node_register_node_group(nodes);	if (ret)		return ret; 	return kunit_add_action_or_reset(test, software_node_unregister_node_group_wrapper,					 (void *)nodes);}EXPORT_SYMBOL_GPL(kunit_software_node_register_node_group); KUNIT_DEFINE_ACTION_WRAPPER(device_remove_software_node_wrapper,			    device_remove_software_node,			    struct device *); /** * kunit_device_add_software_node() - Assign a kunit-managed software node to a device * @test: Test context * @dev: Device to assign the software node for * @node: The software node to assign * * Make @node the secondary firmware node of @dev. If @dev has no primary * firmware node, @node will become the primary node. The software node will * be automatically removed from @dev when the test case completes. * * Returns: * 0 on success, negative error number on failure. */int kunit_device_add_software_node(struct kunit *test, struct device *dev,				   const struct software_node *node){	int ret; 	ret = device_add_software_node(dev, node);	if (ret)		return ret; 	return kunit_add_action_or_reset(test, device_remove_software_node_wrapper, dev);}EXPORT_SYMBOL_GPL(kunit_device_add_software_node);