15c072c8eSSepherosa Ziehau /*-
25c072c8eSSepherosa Ziehau * Copyright (c) 2014,2016 Microsoft Corp.
35c072c8eSSepherosa Ziehau * All rights reserved.
45c072c8eSSepherosa Ziehau *
55c072c8eSSepherosa Ziehau * Redistribution and use in source and binary forms, with or without
65c072c8eSSepherosa Ziehau * modification, are permitted provided that the following conditions
75c072c8eSSepherosa Ziehau * are met:
85c072c8eSSepherosa Ziehau * 1. Redistributions of source code must retain the above copyright
95c072c8eSSepherosa Ziehau * notice unmodified, this list of conditions, and the following
105c072c8eSSepherosa Ziehau * disclaimer.
115c072c8eSSepherosa Ziehau * 2. Redistributions in binary form must reproduce the above copyright
125c072c8eSSepherosa Ziehau * notice, this list of conditions and the following disclaimer in the
135c072c8eSSepherosa Ziehau * documentation and/or other materials provided with the distribution.
145c072c8eSSepherosa Ziehau *
155c072c8eSSepherosa Ziehau * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
165c072c8eSSepherosa Ziehau * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
175c072c8eSSepherosa Ziehau * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
185c072c8eSSepherosa Ziehau * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
195c072c8eSSepherosa Ziehau * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
205c072c8eSSepherosa Ziehau * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
215c072c8eSSepherosa Ziehau * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
225c072c8eSSepherosa Ziehau * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
235c072c8eSSepherosa Ziehau * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
245c072c8eSSepherosa Ziehau * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
255c072c8eSSepherosa Ziehau */
265c072c8eSSepherosa Ziehau
275c072c8eSSepherosa Ziehau #include <sys/param.h>
285c072c8eSSepherosa Ziehau #include <sys/bus.h>
295c072c8eSSepherosa Ziehau #include <sys/kernel.h>
305c072c8eSSepherosa Ziehau #include <sys/module.h>
315c072c8eSSepherosa Ziehau #include <sys/reboot.h>
325c072c8eSSepherosa Ziehau #include <sys/systm.h>
335c072c8eSSepherosa Ziehau
345c072c8eSSepherosa Ziehau #include <dev/hyperv/include/hyperv.h>
355c072c8eSSepherosa Ziehau #include <dev/hyperv/include/vmbus.h>
365c072c8eSSepherosa Ziehau #include <dev/hyperv/utilities/vmbus_icreg.h>
375c072c8eSSepherosa Ziehau #include <dev/hyperv/utilities/vmbus_icvar.h>
385c072c8eSSepherosa Ziehau
395c072c8eSSepherosa Ziehau #define VMBUS_SHUTDOWN_FWVER_MAJOR 3
405c072c8eSSepherosa Ziehau #define VMBUS_SHUTDOWN_FWVER \
415c072c8eSSepherosa Ziehau VMBUS_IC_VERSION(VMBUS_SHUTDOWN_FWVER_MAJOR, 0)
425c072c8eSSepherosa Ziehau
435c072c8eSSepherosa Ziehau #define VMBUS_SHUTDOWN_MSGVER_MAJOR 3
445c072c8eSSepherosa Ziehau #define VMBUS_SHUTDOWN_MSGVER \
455c072c8eSSepherosa Ziehau VMBUS_IC_VERSION(VMBUS_SHUTDOWN_MSGVER_MAJOR, 0)
465c072c8eSSepherosa Ziehau
475c072c8eSSepherosa Ziehau static int vmbus_shutdown_probe(device_t);
485c072c8eSSepherosa Ziehau static int vmbus_shutdown_attach(device_t);
495c072c8eSSepherosa Ziehau
505c072c8eSSepherosa Ziehau static const struct vmbus_ic_desc vmbus_shutdown_descs[] = {
515c072c8eSSepherosa Ziehau {
525c072c8eSSepherosa Ziehau .ic_guid = { .hv_guid = {
535c072c8eSSepherosa Ziehau 0x31, 0x60, 0x0b, 0x0e, 0x13, 0x52, 0x34, 0x49,
545c072c8eSSepherosa Ziehau 0x81, 0x8b, 0x38, 0xd9, 0x0c, 0xed, 0x39, 0xdb } },
555c072c8eSSepherosa Ziehau .ic_desc = "Hyper-V Shutdown"
565c072c8eSSepherosa Ziehau },
575c072c8eSSepherosa Ziehau VMBUS_IC_DESC_END
585c072c8eSSepherosa Ziehau };
595c072c8eSSepherosa Ziehau
605c072c8eSSepherosa Ziehau static device_method_t vmbus_shutdown_methods[] = {
615c072c8eSSepherosa Ziehau /* Device interface */
625c072c8eSSepherosa Ziehau DEVMETHOD(device_probe, vmbus_shutdown_probe),
635c072c8eSSepherosa Ziehau DEVMETHOD(device_attach, vmbus_shutdown_attach),
645c072c8eSSepherosa Ziehau DEVMETHOD(device_detach, vmbus_ic_detach),
655c072c8eSSepherosa Ziehau DEVMETHOD_END
665c072c8eSSepherosa Ziehau };
675c072c8eSSepherosa Ziehau
685c072c8eSSepherosa Ziehau static driver_t vmbus_shutdown_driver = {
695c072c8eSSepherosa Ziehau "hvshutdown",
705c072c8eSSepherosa Ziehau vmbus_shutdown_methods,
715c072c8eSSepherosa Ziehau sizeof(struct vmbus_ic_softc)
725c072c8eSSepherosa Ziehau };
735c072c8eSSepherosa Ziehau
74c1cef544SJohn Baldwin DRIVER_MODULE(hv_shutdown, vmbus, vmbus_shutdown_driver, NULL, NULL);
755c072c8eSSepherosa Ziehau MODULE_VERSION(hv_shutdown, 1);
765c072c8eSSepherosa Ziehau MODULE_DEPEND(hv_shutdown, vmbus, 1, 1, 1);
775c072c8eSSepherosa Ziehau
785c072c8eSSepherosa Ziehau static void
vmbus_shutdown_cb(struct vmbus_channel * chan,void * xsc)795c072c8eSSepherosa Ziehau vmbus_shutdown_cb(struct vmbus_channel *chan, void *xsc)
805c072c8eSSepherosa Ziehau {
815c072c8eSSepherosa Ziehau struct vmbus_ic_softc *sc = xsc;
825c072c8eSSepherosa Ziehau struct vmbus_icmsg_hdr *hdr;
835c072c8eSSepherosa Ziehau struct vmbus_icmsg_shutdown *msg;
845c072c8eSSepherosa Ziehau int dlen, error, do_shutdown = 0;
855c072c8eSSepherosa Ziehau uint64_t xactid;
865c072c8eSSepherosa Ziehau void *data;
875c072c8eSSepherosa Ziehau
885c072c8eSSepherosa Ziehau /*
895c072c8eSSepherosa Ziehau * Receive request.
905c072c8eSSepherosa Ziehau */
915c072c8eSSepherosa Ziehau data = sc->ic_buf;
925c072c8eSSepherosa Ziehau dlen = sc->ic_buflen;
935c072c8eSSepherosa Ziehau error = vmbus_chan_recv(chan, data, &dlen, &xactid);
945c072c8eSSepherosa Ziehau KASSERT(error != ENOBUFS, ("icbuf is not large enough"));
955c072c8eSSepherosa Ziehau if (error)
965c072c8eSSepherosa Ziehau return;
975c072c8eSSepherosa Ziehau
985c072c8eSSepherosa Ziehau if (dlen < sizeof(*hdr)) {
995c072c8eSSepherosa Ziehau device_printf(sc->ic_dev, "invalid data len %d\n", dlen);
1005c072c8eSSepherosa Ziehau return;
1015c072c8eSSepherosa Ziehau }
1025c072c8eSSepherosa Ziehau hdr = data;
1035c072c8eSSepherosa Ziehau
1045c072c8eSSepherosa Ziehau /*
1055c072c8eSSepherosa Ziehau * Update request, which will be echoed back as response.
1065c072c8eSSepherosa Ziehau */
1075c072c8eSSepherosa Ziehau switch (hdr->ic_type) {
1085c072c8eSSepherosa Ziehau case VMBUS_ICMSG_TYPE_NEGOTIATE:
1095c072c8eSSepherosa Ziehau error = vmbus_ic_negomsg(sc, data, &dlen,
1105c072c8eSSepherosa Ziehau VMBUS_SHUTDOWN_FWVER, VMBUS_SHUTDOWN_MSGVER);
1115c072c8eSSepherosa Ziehau if (error)
1125c072c8eSSepherosa Ziehau return;
1135c072c8eSSepherosa Ziehau break;
1145c072c8eSSepherosa Ziehau
1155c072c8eSSepherosa Ziehau case VMBUS_ICMSG_TYPE_SHUTDOWN:
1165c072c8eSSepherosa Ziehau if (dlen < VMBUS_ICMSG_SHUTDOWN_SIZE_MIN) {
1175c072c8eSSepherosa Ziehau device_printf(sc->ic_dev, "invalid shutdown len %d\n",
1185c072c8eSSepherosa Ziehau dlen);
1195c072c8eSSepherosa Ziehau return;
1205c072c8eSSepherosa Ziehau }
1215c072c8eSSepherosa Ziehau msg = data;
1225c072c8eSSepherosa Ziehau
1235c072c8eSSepherosa Ziehau /* XXX ic_flags definition? */
1245c072c8eSSepherosa Ziehau if (msg->ic_haltflags == 0 || msg->ic_haltflags == 1) {
1255c072c8eSSepherosa Ziehau device_printf(sc->ic_dev, "shutdown requested\n");
1265c072c8eSSepherosa Ziehau hdr->ic_status = VMBUS_ICMSG_STATUS_OK;
1275c072c8eSSepherosa Ziehau do_shutdown = 1;
1285c072c8eSSepherosa Ziehau } else {
1295c072c8eSSepherosa Ziehau device_printf(sc->ic_dev, "unknown shutdown flags "
1305c072c8eSSepherosa Ziehau "0x%08x\n", msg->ic_haltflags);
1315c072c8eSSepherosa Ziehau hdr->ic_status = VMBUS_ICMSG_STATUS_FAIL;
1325c072c8eSSepherosa Ziehau }
1335c072c8eSSepherosa Ziehau break;
1345c072c8eSSepherosa Ziehau
1355c072c8eSSepherosa Ziehau default:
1365c072c8eSSepherosa Ziehau device_printf(sc->ic_dev, "got 0x%08x icmsg\n", hdr->ic_type);
1375c072c8eSSepherosa Ziehau break;
1385c072c8eSSepherosa Ziehau }
1395c072c8eSSepherosa Ziehau
1405c072c8eSSepherosa Ziehau /*
1415c072c8eSSepherosa Ziehau * Send response by echoing the request back.
1425c072c8eSSepherosa Ziehau */
1435c072c8eSSepherosa Ziehau vmbus_ic_sendresp(sc, chan, data, dlen, xactid);
1445c072c8eSSepherosa Ziehau
1455c072c8eSSepherosa Ziehau if (do_shutdown)
1465c072c8eSSepherosa Ziehau shutdown_nice(RB_POWEROFF);
1475c072c8eSSepherosa Ziehau }
1485c072c8eSSepherosa Ziehau
1495c072c8eSSepherosa Ziehau static int
vmbus_shutdown_probe(device_t dev)1505c072c8eSSepherosa Ziehau vmbus_shutdown_probe(device_t dev)
1515c072c8eSSepherosa Ziehau {
1525c072c8eSSepherosa Ziehau
1535c072c8eSSepherosa Ziehau return (vmbus_ic_probe(dev, vmbus_shutdown_descs));
1545c072c8eSSepherosa Ziehau }
1555c072c8eSSepherosa Ziehau
1565c072c8eSSepherosa Ziehau static int
vmbus_shutdown_attach(device_t dev)1575c072c8eSSepherosa Ziehau vmbus_shutdown_attach(device_t dev)
1585c072c8eSSepherosa Ziehau {
1595c072c8eSSepherosa Ziehau
1605c072c8eSSepherosa Ziehau return (vmbus_ic_attach(dev, vmbus_shutdown_cb));
1615c072c8eSSepherosa Ziehau }
162