1 // SPDX-License-Identifier: GPL-2.0-or-later
11 #include <linux/reboot.h>
12 #include <linux/reboot-mode.h>
14 #define PREFIX "mode-"
17 	const char *mode;  member
22 static unsigned int get_reboot_mode_magic(struct reboot_mode_driver *reboot,  in get_reboot_mode_magic()  argument
32 	list_for_each_entry(info, &reboot->head, list)  in get_reboot_mode_magic()
33 		if (!strcmp(info->mode, cmd))  in get_reboot_mode_magic()
34 			return info->magic;  in get_reboot_mode_magic()
37 	if (strscpy(cmd_, cmd, sizeof(cmd_)) == -E2BIG)  in get_reboot_mode_magic()
40 	strreplace(cmd_, ' ', '-');  in get_reboot_mode_magic()
41 	strreplace(cmd_, ',', '-');  in get_reboot_mode_magic()
42 	strreplace(cmd_, '/', '-');  in get_reboot_mode_magic()
44 	list_for_each_entry(info, &reboot->head, list)  in get_reboot_mode_magic()
45 		if (!strcmp(info->mode, cmd_))  in get_reboot_mode_magic()
46 			return info->magic;  in get_reboot_mode_magic()
52 			      unsigned long mode, void *cmd)  in reboot_mode_notify()  argument
54 	struct reboot_mode_driver *reboot;  in reboot_mode_notify()  local
57 	reboot = container_of(this, struct reboot_mode_driver, reboot_notifier);  in reboot_mode_notify()
58 	magic = get_reboot_mode_magic(reboot, cmd);  in reboot_mode_notify()
60 		reboot->write(reboot, magic);  in reboot_mode_notify()
66  * reboot_mode_register - register a reboot mode driver
67  * @reboot: reboot mode driver
71 int reboot_mode_register(struct reboot_mode_driver *reboot)  in reboot_mode_register()  argument
75 	struct device_node *np = reboot->dev->of_node;  in reboot_mode_register()
79 	INIT_LIST_HEAD(&reboot->head);  in reboot_mode_register()
82 		if (strncmp(prop->name, PREFIX, len))  in reboot_mode_register()
85 		info = devm_kzalloc(reboot->dev, sizeof(*info), GFP_KERNEL);  in reboot_mode_register()
87 			ret = -ENOMEM;  in reboot_mode_register()
91 		if (of_property_read_u32(np, prop->name, &info->magic)) {  in reboot_mode_register()
92 			dev_err(reboot->dev, "reboot mode %s without magic number\n",  in reboot_mode_register()
93 				info->mode);  in reboot_mode_register()
94 			devm_kfree(reboot->dev, info);  in reboot_mode_register()
98 		info->mode = kstrdup_const(prop->name + len, GFP_KERNEL);  in reboot_mode_register()
99 		if (!info->mode) {  in reboot_mode_register()
100 			ret =  -ENOMEM;  in reboot_mode_register()
102 		} else if (info->mode[0] == '\0') {  in reboot_mode_register()
103 			kfree_const(info->mode);  in reboot_mode_register()
104 			ret = -EINVAL;  in reboot_mode_register()
105 			dev_err(reboot->dev, "invalid mode name(%s): too short!\n",  in reboot_mode_register()
106 				prop->name);  in reboot_mode_register()
110 		list_add_tail(&info->list, &reboot->head);  in reboot_mode_register()
113 	reboot->reboot_notifier.notifier_call = reboot_mode_notify;  in reboot_mode_register()
114 	register_reboot_notifier(&reboot->reboot_notifier);  in reboot_mode_register()
119 	list_for_each_entry(info, &reboot->head, list)  in reboot_mode_register()
120 		kfree_const(info->mode);  in reboot_mode_register()
127  * reboot_mode_unregister - unregister a reboot mode driver
128  * @reboot: reboot mode driver
130 int reboot_mode_unregister(struct reboot_mode_driver *reboot)  in reboot_mode_unregister()  argument
134 	unregister_reboot_notifier(&reboot->reboot_notifier);  in reboot_mode_unregister()
136 	list_for_each_entry(info, &reboot->head, list)  in reboot_mode_unregister()
137 		kfree_const(info->mode);  in reboot_mode_unregister()
149  * devm_reboot_mode_register() - resource managed reboot_mode_register()
151  * @reboot: reboot mode driver
156 			      struct reboot_mode_driver *reboot)  in devm_reboot_mode_register()  argument
163 		return -ENOMEM;  in devm_reboot_mode_register()
165 	rc = reboot_mode_register(reboot);  in devm_reboot_mode_register()
171 	*dr = reboot;  in devm_reboot_mode_register()
189  * devm_reboot_mode_unregister() - resource managed reboot_mode_unregister()
191  * @reboot: reboot mode driver
194 				 struct reboot_mode_driver *reboot)  in devm_reboot_mode_unregister()  argument
198 			       devm_reboot_mode_match, reboot));  in devm_reboot_mode_unregister()
202 MODULE_AUTHOR("Andy Yan <andy.yan@rock-chips.com>");
203 MODULE_DESCRIPTION("System reboot mode core library");