hvheartbeat.c revision 1.2.2.2 1 1.2.2.2 martin /* $NetBSD: hvheartbeat.c,v 1.2.2.2 2019/03/09 17:10:19 martin Exp $ */
2 1.2.2.2 martin
3 1.2.2.2 martin /*-
4 1.2.2.2 martin * Copyright (c) 2014,2016 Microsoft Corp.
5 1.2.2.2 martin * All rights reserved.
6 1.2.2.2 martin *
7 1.2.2.2 martin * Redistribution and use in source and binary forms, with or without
8 1.2.2.2 martin * modification, are permitted provided that the following conditions
9 1.2.2.2 martin * are met:
10 1.2.2.2 martin * 1. Redistributions of source code must retain the above copyright
11 1.2.2.2 martin * notice unmodified, this list of conditions, and the following
12 1.2.2.2 martin * disclaimer.
13 1.2.2.2 martin * 2. Redistributions in binary form must reproduce the above copyright
14 1.2.2.2 martin * notice, this list of conditions and the following disclaimer in the
15 1.2.2.2 martin * documentation and/or other materials provided with the distribution.
16 1.2.2.2 martin *
17 1.2.2.2 martin * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
18 1.2.2.2 martin * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
19 1.2.2.2 martin * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
20 1.2.2.2 martin * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
21 1.2.2.2 martin * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
22 1.2.2.2 martin * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
23 1.2.2.2 martin * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
24 1.2.2.2 martin * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
25 1.2.2.2 martin * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
26 1.2.2.2 martin * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
27 1.2.2.2 martin */
28 1.2.2.2 martin
29 1.2.2.2 martin #include <sys/cdefs.h>
30 1.2.2.2 martin #ifdef __KERNEL_RCSID
31 1.2.2.2 martin __KERNEL_RCSID(0, "$NetBSD: hvheartbeat.c,v 1.2.2.2 2019/03/09 17:10:19 martin Exp $");
32 1.2.2.2 martin #endif
33 1.2.2.2 martin #ifdef __FBSDID
34 1.2.2.2 martin __FBSDID("$FreeBSD: head/sys/dev/hyperv/utilities/vmbus_heartbeat.c 310324 2016-12-20 09:46:14Z sephe $");
35 1.2.2.2 martin #endif
36 1.2.2.2 martin
37 1.2.2.2 martin #include <sys/param.h>
38 1.2.2.2 martin #include <sys/systm.h>
39 1.2.2.2 martin #include <sys/device.h>
40 1.2.2.2 martin #include <sys/module.h>
41 1.2.2.2 martin #include <sys/pmf.h>
42 1.2.2.2 martin
43 1.2.2.2 martin #include <dev/hyperv/vmbusvar.h>
44 1.2.2.2 martin #include <dev/hyperv/vmbusicreg.h>
45 1.2.2.2 martin #include <dev/hyperv/vmbusicvar.h>
46 1.2.2.2 martin
47 1.2.2.2 martin #define VMBUS_HEARTBEAT_FWVER_MAJOR 3
48 1.2.2.2 martin #define VMBUS_HEARTBEAT_FWVER \
49 1.2.2.2 martin VMBUS_IC_VERSION(VMBUS_HEARTBEAT_FWVER_MAJOR, 0)
50 1.2.2.2 martin
51 1.2.2.2 martin #define VMBUS_HEARTBEAT_MSGVER_MAJOR 3
52 1.2.2.2 martin #define VMBUS_HEARTBEAT_MSGVER \
53 1.2.2.2 martin VMBUS_IC_VERSION(VMBUS_HEARTBEAT_MSGVER_MAJOR, 0)
54 1.2.2.2 martin
55 1.2.2.2 martin static int hvheartbeat_match(device_t, cfdata_t, void *);
56 1.2.2.2 martin static void hvheartbeat_attach(device_t, device_t, void *);
57 1.2.2.2 martin static int hvheartbeat_detach(device_t, int);
58 1.2.2.2 martin
59 1.2.2.2 martin static void hvheartbeat_channel_cb(void *);
60 1.2.2.2 martin
61 1.2.2.2 martin struct hvheartbeat_softc {
62 1.2.2.2 martin struct vmbusic_softc sc_vmbusic;
63 1.2.2.2 martin };
64 1.2.2.2 martin
65 1.2.2.2 martin CFATTACH_DECL_NEW(hvheartbeat, sizeof(struct hvheartbeat_softc),
66 1.2.2.2 martin hvheartbeat_match, hvheartbeat_attach, hvheartbeat_detach, NULL);
67 1.2.2.2 martin
68 1.2.2.2 martin static int
69 1.2.2.2 martin hvheartbeat_match(device_t parent, cfdata_t cf, void *aux)
70 1.2.2.2 martin {
71 1.2.2.2 martin struct vmbus_attach_args *aa = aux;
72 1.2.2.2 martin
73 1.2.2.2 martin return vmbusic_probe(aa, &hyperv_guid_heartbeat);
74 1.2.2.2 martin }
75 1.2.2.2 martin
76 1.2.2.2 martin static void
77 1.2.2.2 martin hvheartbeat_attach(device_t parent, device_t self, void *aux)
78 1.2.2.2 martin {
79 1.2.2.2 martin struct vmbus_attach_args *aa = aux;
80 1.2.2.2 martin int error;
81 1.2.2.2 martin
82 1.2.2.2 martin aprint_naive("\n");
83 1.2.2.2 martin aprint_normal(": Hyper-V Heartbeat Service\n");
84 1.2.2.2 martin
85 1.2.2.2 martin error = vmbusic_attach(self, aa, hvheartbeat_channel_cb);
86 1.2.2.2 martin if (error)
87 1.2.2.2 martin return;
88 1.2.2.2 martin
89 1.2.2.2 martin (void) pmf_device_register(self, NULL, NULL);
90 1.2.2.2 martin }
91 1.2.2.2 martin
92 1.2.2.2 martin static int
93 1.2.2.2 martin hvheartbeat_detach(device_t self, int flags)
94 1.2.2.2 martin {
95 1.2.2.2 martin int error;
96 1.2.2.2 martin
97 1.2.2.2 martin error = vmbusic_detach(self, flags);
98 1.2.2.2 martin if (error)
99 1.2.2.2 martin return error;
100 1.2.2.2 martin
101 1.2.2.2 martin pmf_device_deregister(self);
102 1.2.2.2 martin
103 1.2.2.2 martin return 0;
104 1.2.2.2 martin }
105 1.2.2.2 martin
106 1.2.2.2 martin static void
107 1.2.2.2 martin hvheartbeat_channel_cb(void *arg)
108 1.2.2.2 martin {
109 1.2.2.2 martin struct hvheartbeat_softc *sc = arg;
110 1.2.2.2 martin struct vmbusic_softc *vsc = &sc->sc_vmbusic;
111 1.2.2.2 martin struct vmbus_channel *ch = vsc->sc_chan;
112 1.2.2.2 martin struct vmbus_icmsg_hdr *hdr;
113 1.2.2.2 martin struct vmbus_icmsg_heartbeat *msg;
114 1.2.2.2 martin uint64_t rid;
115 1.2.2.2 martin uint32_t rlen;
116 1.2.2.2 martin int error;
117 1.2.2.2 martin
118 1.2.2.2 martin error = vmbus_channel_recv(ch, vsc->sc_buf, vsc->sc_buflen,
119 1.2.2.2 martin &rlen, &rid, 0);
120 1.2.2.2 martin if (error || rlen == 0) {
121 1.2.2.2 martin if (error != EAGAIN) {
122 1.2.2.2 martin DPRINTF("%s: heartbeat error=%d len=%u\n",
123 1.2.2.2 martin device_xname(vsc->sc_dev), error, rlen);
124 1.2.2.2 martin }
125 1.2.2.2 martin return;
126 1.2.2.2 martin }
127 1.2.2.2 martin if (rlen < sizeof(*hdr)) {
128 1.2.2.2 martin DPRINTF("%s: heartbeat short read len=%u\n",
129 1.2.2.2 martin device_xname(vsc->sc_dev), rlen);
130 1.2.2.2 martin return;
131 1.2.2.2 martin }
132 1.2.2.2 martin
133 1.2.2.2 martin hdr = (struct vmbus_icmsg_hdr *)vsc->sc_buf;
134 1.2.2.2 martin switch (hdr->ic_type) {
135 1.2.2.2 martin case VMBUS_ICMSG_TYPE_NEGOTIATE:
136 1.2.2.2 martin error = vmbusic_negotiate(vsc, hdr, &rlen,
137 1.2.2.2 martin VMBUS_HEARTBEAT_FWVER, VMBUS_HEARTBEAT_MSGVER);
138 1.2.2.2 martin if (error)
139 1.2.2.2 martin return;
140 1.2.2.2 martin break;
141 1.2.2.2 martin
142 1.2.2.2 martin case VMBUS_ICMSG_TYPE_HEARTBEAT:
143 1.2.2.2 martin if (rlen < VMBUS_ICMSG_HEARTBEAT_SIZE_MIN) {
144 1.2.2.2 martin DPRINTF("%s: invalid heartbeat len=%u\n",
145 1.2.2.2 martin device_xname(vsc->sc_dev), rlen);
146 1.2.2.2 martin return;
147 1.2.2.2 martin }
148 1.2.2.2 martin
149 1.2.2.2 martin msg = (struct vmbus_icmsg_heartbeat *)hdr;
150 1.2.2.2 martin msg->ic_seq++;
151 1.2.2.2 martin break;
152 1.2.2.2 martin
153 1.2.2.2 martin default:
154 1.2.2.2 martin device_printf(vsc->sc_dev,
155 1.2.2.2 martin "unhandled heartbeat message type %u\n", hdr->ic_type);
156 1.2.2.2 martin return;
157 1.2.2.2 martin }
158 1.2.2.2 martin
159 1.2.2.2 martin (void) vmbusic_sendresp(vsc, ch, vsc->sc_buf, rlen, rid);
160 1.2.2.2 martin }
161 1.2.2.2 martin
162 1.2.2.2 martin MODULE(MODULE_CLASS_DRIVER, hvheartbeat, "vmbus");
163 1.2.2.2 martin
164 1.2.2.2 martin #ifdef _MODULE
165 1.2.2.2 martin #include "ioconf.c"
166 1.2.2.2 martin #endif
167 1.2.2.2 martin
168 1.2.2.2 martin static int
169 1.2.2.2 martin hvheartbeat_modcmd(modcmd_t cmd, void *aux)
170 1.2.2.2 martin {
171 1.2.2.2 martin int error = 0;
172 1.2.2.2 martin
173 1.2.2.2 martin switch (cmd) {
174 1.2.2.2 martin case MODULE_CMD_INIT:
175 1.2.2.2 martin #ifdef _MODULE
176 1.2.2.2 martin error = config_init_component(cfdriver_ioconf_hvheartbeat,
177 1.2.2.2 martin cfattach_ioconf_hvheartbeat, cfdata_ioconf_hvheartbeat);
178 1.2.2.2 martin #endif
179 1.2.2.2 martin break;
180 1.2.2.2 martin
181 1.2.2.2 martin case MODULE_CMD_FINI:
182 1.2.2.2 martin #ifdef _MODULE
183 1.2.2.2 martin error = config_fini_component(cfdriver_ioconf_hvheartbeat,
184 1.2.2.2 martin cfattach_ioconf_hvheartbeat, cfdata_ioconf_hvheartbeat);
185 1.2.2.2 martin #endif
186 1.2.2.2 martin break;
187 1.2.2.2 martin
188 1.2.2.2 martin case MODULE_CMD_AUTOUNLOAD:
189 1.2.2.2 martin error = EBUSY;
190 1.2.2.2 martin break;
191 1.2.2.2 martin
192 1.2.2.2 martin default:
193 1.2.2.2 martin error = ENOTTY;
194 1.2.2.2 martin break;
195 1.2.2.2 martin }
196 1.2.2.2 martin
197 1.2.2.2 martin return error;
198 1.2.2.2 martin }
199