1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
|
// SPDX-License-Identifier: GPL-2.0
/*
* T-HEAD TH1520 AON Firmware Reboot Driver
*
* Copyright (c) 2025 Icenowy Zheng <uwu@icenowy.me>
*/
#include <linux/auxiliary_bus.h>
#include <linux/firmware/thead/thead,th1520-aon.h>
#include <linux/module.h>
#include <linux/notifier.h>
#include <linux/of.h>
#include <linux/reboot.h>
#include <linux/slab.h>
#define TH1520_AON_REBOOT_PRIORITY 200
struct th1520_aon_msg_empty_body {
struct th1520_aon_rpc_msg_hdr hdr;
u16 reserved[12];
} __packed __aligned(1);
static int th1520_aon_pwroff_handler(struct sys_off_data *data)
{
struct th1520_aon_chan *aon_chan = data->cb_data;
struct th1520_aon_msg_empty_body msg = {};
msg.hdr.svc = TH1520_AON_RPC_SVC_WDG;
msg.hdr.func = TH1520_AON_WDG_FUNC_POWER_OFF;
msg.hdr.size = TH1520_AON_RPC_MSG_NUM;
th1520_aon_call_rpc(aon_chan, &msg);
return NOTIFY_DONE;
}
static int th1520_aon_restart_handler(struct sys_off_data *data)
{
struct th1520_aon_chan *aon_chan = data->cb_data;
struct th1520_aon_msg_empty_body msg = {};
msg.hdr.svc = TH1520_AON_RPC_SVC_WDG;
msg.hdr.func = TH1520_AON_WDG_FUNC_RESTART;
msg.hdr.size = TH1520_AON_RPC_MSG_NUM;
th1520_aon_call_rpc(aon_chan, &msg);
return NOTIFY_DONE;
}
static int th1520_aon_reboot_probe(struct auxiliary_device *adev,
const struct auxiliary_device_id *id)
{
struct device *dev = &adev->dev;
int ret;
/* Expect struct th1520_aon_chan to be passed via platform_data */
ret = devm_register_sys_off_handler(dev, SYS_OFF_MODE_POWER_OFF,
TH1520_AON_REBOOT_PRIORITY,
th1520_aon_pwroff_handler,
adev->dev.platform_data);
if (ret) {
dev_err(dev, "Failed to register power off handler\n");
return ret;
}
ret = devm_register_sys_off_handler(dev, SYS_OFF_MODE_RESTART,
TH1520_AON_REBOOT_PRIORITY,
th1520_aon_restart_handler,
adev->dev.platform_data);
if (ret) {
dev_err(dev, "Failed to register restart handler\n");
return ret;
}
return 0;
}
static const struct auxiliary_device_id th1520_aon_reboot_id_table[] = {
{ .name = "th1520_pm_domains.reboot" },
{},
};
MODULE_DEVICE_TABLE(auxiliary, th1520_aon_reboot_id_table);
static struct auxiliary_driver th1520_aon_reboot_driver = {
.driver = {
.name = "th1520-aon-reboot",
},
.probe = th1520_aon_reboot_probe,
.id_table = th1520_aon_reboot_id_table,
};
module_auxiliary_driver(th1520_aon_reboot_driver);
MODULE_AUTHOR("Icenowy Zheng <uwu@icenowy.me>");
MODULE_DESCRIPTION("T-HEAD TH1520 AON-firmware-based reboot driver");
MODULE_LICENSE("GPL");
|