xref: /freebsd/usr.sbin/bhyve/pci_virtio_rnd.c (revision 9268022b74279434ed6300244e3f977e56a8ceb5)
1 /*-
2  * Copyright (c) 2014 Nahanni Systems Inc.
3  * All rights reserved.
4  *
5  * Redistribution and use in source and binary forms, with or without
6  * modification, are permitted provided that the following conditions
7  * are met:
8  * 1. Redistributions of source code must retain the above copyright
9  *    notice, this list of conditions and the following disclaimer
10  *    in this position and unchanged.
11  * 2. Redistributions in binary form must reproduce the above copyright
12  *    notice, this list of conditions and the following disclaimer in the
13  *    documentation and/or other materials provided with the distribution.
14  *
15  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
16  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
17  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
18  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
19  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
20  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
21  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
22  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
23  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
24  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
25  * SUCH DAMAGE.
26  */
27 
28 /*
29  * virtio entropy device emulation.
30  * Randomness is sourced from /dev/random which does not block
31  * once it has been seeded at bootup.
32  */
33 
34 #include <sys/cdefs.h>
35 __FBSDID("$FreeBSD$");
36 
37 #include <sys/param.h>
38 #include <sys/linker_set.h>
39 #include <sys/uio.h>
40 
41 #include <errno.h>
42 #include <fcntl.h>
43 #include <stdio.h>
44 #include <stdlib.h>
45 #include <string.h>
46 #include <unistd.h>
47 #include <assert.h>
48 #include <pthread.h>
49 
50 #include "bhyverun.h"
51 #include "pci_emul.h"
52 #include "virtio.h"
53 
54 #define VTRND_RINGSZ	64
55 
56 
57 static int pci_vtrnd_debug;
58 #define DPRINTF(params) if (pci_vtrnd_debug) printf params
59 #define WPRINTF(params) printf params
60 
61 /*
62  * Per-device softc
63  */
64 struct pci_vtrnd_softc {
65 	struct virtio_softc vrsc_vs;
66 	struct vqueue_info  vrsc_vq;
67 	pthread_mutex_t     vrsc_mtx;
68 	uint64_t            vrsc_cfg;
69 	int                 vrsc_fd;
70 };
71 
72 static void pci_vtrnd_reset(void *);
73 static void pci_vtrnd_notify(void *, struct vqueue_info *);
74 
75 static struct virtio_consts vtrnd_vi_consts = {
76 	"vtrnd",		/* our name */
77 	1,			/* we support 1 virtqueue */
78 	0,			/* config reg size */
79 	pci_vtrnd_reset,	/* reset */
80 	pci_vtrnd_notify,	/* device-wide qnotify */
81 	NULL,			/* read virtio config */
82 	NULL,			/* write virtio config */
83 	NULL,			/* apply negotiated features */
84 	0,			/* our capabilities */
85 };
86 
87 
88 static void
89 pci_vtrnd_reset(void *vsc)
90 {
91 	struct pci_vtrnd_softc *sc;
92 
93 	sc = vsc;
94 
95 	DPRINTF(("vtrnd: device reset requested !\n"));
96 	vi_reset_dev(&sc->vrsc_vs);
97 }
98 
99 
100 static void
101 pci_vtrnd_notify(void *vsc, struct vqueue_info *vq)
102 {
103 	struct iovec iov;
104 	struct pci_vtrnd_softc *sc;
105 	int len;
106 
107 	sc = vsc;
108 
109 	vq_startchains(vq);
110 
111 	if (sc->vrsc_fd < 0) {
112 		vq_endchains(vq, 0);
113 		return;
114 	}
115 
116 	while (vq_has_descs(vq)) {
117 		vq_getchain(vq, &iov, 1, NULL);
118 
119 		len = read(sc->vrsc_fd, iov.iov_base, iov.iov_len);
120 
121 		DPRINTF(("vtrnd: vtrnd_notify(): %d\r\n", len));
122 
123 		/* Catastrophe if unable to read from /dev/random */
124 		assert(len > 0);
125 
126 		/*
127 		 * Release this chain and handle more
128 		 */
129 		vq_relchain(vq, len);
130 	}
131 	vq_endchains(vq, 1);	/* Generate interrupt if appropriate. */
132 }
133 
134 
135 static int
136 pci_vtrnd_init(struct vmctx *ctx, struct pci_devinst *pi, char *opts)
137 {
138 	struct pci_vtrnd_softc *sc;
139 	int fd;
140 	int len;
141 	uint8_t v;
142 
143 	/*
144 	 * Should always be able to open /dev/random.
145 	 */
146 	fd = open("/dev/random", O_RDONLY | O_NONBLOCK);
147 
148 	assert(fd >= 0);
149 
150 	/*
151 	 * Check that device is seeded and non-blocking.
152 	 */
153 	len = read(fd, &v, sizeof(v));
154 	if (len <= 0) {
155 		WPRINTF(("vtrnd: /dev/random not ready, read(): %d", len));
156 		return (1);
157 	}
158 
159 	sc = calloc(1, sizeof(struct pci_vtrnd_softc));
160 
161 	vi_softc_linkup(&sc->vrsc_vs, &vtrnd_vi_consts, sc, pi, &sc->vrsc_vq);
162 	sc->vrsc_vs.vs_mtx = &sc->vrsc_mtx;
163 
164 	sc->vrsc_vq.vq_qsize = VTRND_RINGSZ;
165 
166 	/* keep /dev/random opened while emulating */
167 	sc->vrsc_fd = fd;
168 
169 	/* initialize config space */
170 	pci_set_cfgdata16(pi, PCIR_DEVICE, VIRTIO_DEV_RANDOM);
171 	pci_set_cfgdata16(pi, PCIR_VENDOR, VIRTIO_VENDOR);
172 	pci_set_cfgdata8(pi, PCIR_CLASS, PCIC_CRYPTO);
173 	pci_set_cfgdata16(pi, PCIR_SUBDEV_0, VIRTIO_TYPE_ENTROPY);
174 
175 	if (vi_intr_init(&sc->vrsc_vs, 1, fbsdrun_virtio_msix()))
176 		return (1);
177 	vi_set_io_bar(&sc->vrsc_vs, 0);
178 
179 	return (0);
180 }
181 
182 
183 struct pci_devemu pci_de_vrnd = {
184 	.pe_emu =	"virtio-rnd",
185 	.pe_init =	pci_vtrnd_init,
186 	.pe_barwrite =	vi_pci_write,
187 	.pe_barread =	vi_pci_read
188 };
189 PCI_EMUL_SET(pci_de_vrnd);
190