1 // SPDX-License-Identifier: GPL-2.0+
2 /*
3  * Copyright (C) 2017 Álvaro Fernández Rojas <noltari@gmail.com>
4  */
5 
6 #include <common.h>
7 #include <dm.h>
8 #include <dm/device-internal.h>
9 #include <errno.h>
10 #include <malloc.h>
11 #include <sysreset.h>
12 #include <wdt.h>
13 
14 struct wdt_reboot_plat {
15 	struct udevice *wdt;
16 };
17 
wdt_reboot_request(struct udevice * dev,enum sysreset_t type)18 static int wdt_reboot_request(struct udevice *dev, enum sysreset_t type)
19 {
20 	struct wdt_reboot_plat *plat = dev_get_plat(dev);
21 	int ret;
22 
23 	switch (type) {
24 	case SYSRESET_COLD:
25 	case SYSRESET_WARM:
26 		ret = wdt_expire_now(plat->wdt, 0);
27 		if (ret)
28 			return ret;
29 		break;
30 	default:
31 		return -ENOSYS;
32 	}
33 
34 	return -EINPROGRESS;
35 }
36 
37 static struct sysreset_ops wdt_reboot_ops = {
38 	.request = wdt_reboot_request,
39 };
40 
wdt_reboot_of_to_plat(struct udevice * dev)41 static int wdt_reboot_of_to_plat(struct udevice *dev)
42 {
43 	struct wdt_reboot_plat *plat = dev_get_plat(dev);
44 	int err;
45 
46 	err = uclass_get_device_by_phandle(UCLASS_WDT, dev,
47 					   "wdt", &plat->wdt);
48 	if (err) {
49 		pr_err("unable to find wdt device\n");
50 		return err;
51 	}
52 
53 	return 0;
54 }
55 
56 static const struct udevice_id wdt_reboot_ids[] = {
57 	{ .compatible = "wdt-reboot" },
58 	{ /* sentinel */ }
59 };
60 
61 U_BOOT_DRIVER(wdt_reboot) = {
62 	.name = "wdt_reboot",
63 	.id = UCLASS_SYSRESET,
64 	.of_match = wdt_reboot_ids,
65 	.of_to_plat	= wdt_reboot_of_to_plat,
66 	.plat_auto	= sizeof(struct wdt_reboot_plat),
67 	.ops = &wdt_reboot_ops,
68 };
69 
70 #if IS_ENABLED(CONFIG_SYSRESET_WATCHDOG_AUTO)
sysreset_register_wdt(struct udevice * dev)71 int sysreset_register_wdt(struct udevice *dev)
72 {
73 	struct wdt_reboot_plat *plat = malloc(sizeof(*plat));
74 	int ret;
75 
76 	if (!plat)
77 		return -ENOMEM;
78 
79 	plat->wdt = dev;
80 
81 	ret = device_bind(dev, DM_DRIVER_GET(wdt_reboot),
82 			  dev->name, plat, ofnode_null(), NULL);
83 	if (ret) {
84 		free(plat);
85 		return ret;
86 	}
87 
88 	return 0;
89 }
90 #endif
91