Merge remote-tracking branch 'remotes/dgilbert/tags/pull-migration-20200925a' into...
[qemu.git] / io / channel-tls.c
1 /*
2 * QEMU I/O channels TLS driver
3 *
4 * Copyright (c) 2015 Red Hat, Inc.
5 *
6 * This library is free software; you can redistribute it and/or
7 * modify it under the terms of the GNU Lesser General Public
8 * License as published by the Free Software Foundation; either
9 * version 2 of the License, or (at your option) any later version.
10 *
11 * This library is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14 * Lesser General Public License for more details.
15 *
16 * You should have received a copy of the GNU Lesser General Public
17 * License along with this library; if not, see <http://www.gnu.org/licenses/>.
18 *
19 */
20
21 #include "qemu/osdep.h"
22 #include "qapi/error.h"
23 #include "qemu/module.h"
24 #include "io/channel-tls.h"
25 #include "trace.h"
26
27
28 static ssize_t qio_channel_tls_write_handler(const char *buf,
29 size_t len,
30 void *opaque)
31 {
32 QIOChannelTLS *tioc = QIO_CHANNEL_TLS(opaque);
33 ssize_t ret;
34
35 ret = qio_channel_write(tioc->master, buf, len, NULL);
36 if (ret == QIO_CHANNEL_ERR_BLOCK) {
37 errno = EAGAIN;
38 return -1;
39 } else if (ret < 0) {
40 errno = EIO;
41 return -1;
42 }
43 return ret;
44 }
45
46 static ssize_t qio_channel_tls_read_handler(char *buf,
47 size_t len,
48 void *opaque)
49 {
50 QIOChannelTLS *tioc = QIO_CHANNEL_TLS(opaque);
51 ssize_t ret;
52
53 ret = qio_channel_read(tioc->master, buf, len, NULL);
54 if (ret == QIO_CHANNEL_ERR_BLOCK) {
55 errno = EAGAIN;
56 return -1;
57 } else if (ret < 0) {
58 errno = EIO;
59 return -1;
60 }
61 return ret;
62 }
63
64
65 QIOChannelTLS *
66 qio_channel_tls_new_server(QIOChannel *master,
67 QCryptoTLSCreds *creds,
68 const char *aclname,
69 Error **errp)
70 {
71 QIOChannelTLS *ioc;
72
73 ioc = QIO_CHANNEL_TLS(object_new(TYPE_QIO_CHANNEL_TLS));
74
75 ioc->master = master;
76 object_ref(OBJECT(master));
77
78 ioc->session = qcrypto_tls_session_new(
79 creds,
80 NULL,
81 aclname,
82 QCRYPTO_TLS_CREDS_ENDPOINT_SERVER,
83 errp);
84 if (!ioc->session) {
85 goto error;
86 }
87
88 qcrypto_tls_session_set_callbacks(
89 ioc->session,
90 qio_channel_tls_write_handler,
91 qio_channel_tls_read_handler,
92 ioc);
93
94 trace_qio_channel_tls_new_server(ioc, master, creds, aclname);
95 return ioc;
96
97 error:
98 object_unref(OBJECT(ioc));
99 return NULL;
100 }
101
102 QIOChannelTLS *
103 qio_channel_tls_new_client(QIOChannel *master,
104 QCryptoTLSCreds *creds,
105 const char *hostname,
106 Error **errp)
107 {
108 QIOChannelTLS *tioc;
109 QIOChannel *ioc;
110
111 tioc = QIO_CHANNEL_TLS(object_new(TYPE_QIO_CHANNEL_TLS));
112 ioc = QIO_CHANNEL(tioc);
113
114 tioc->master = master;
115 if (qio_channel_has_feature(master, QIO_CHANNEL_FEATURE_SHUTDOWN)) {
116 qio_channel_set_feature(ioc, QIO_CHANNEL_FEATURE_SHUTDOWN);
117 }
118 object_ref(OBJECT(master));
119
120 tioc->session = qcrypto_tls_session_new(
121 creds,
122 hostname,
123 NULL,
124 QCRYPTO_TLS_CREDS_ENDPOINT_CLIENT,
125 errp);
126 if (!tioc->session) {
127 goto error;
128 }
129
130 qcrypto_tls_session_set_callbacks(
131 tioc->session,
132 qio_channel_tls_write_handler,
133 qio_channel_tls_read_handler,
134 tioc);
135
136 trace_qio_channel_tls_new_client(tioc, master, creds, hostname);
137 return tioc;
138
139 error:
140 object_unref(OBJECT(tioc));
141 return NULL;
142 }
143
144 struct QIOChannelTLSData {
145 QIOTask *task;
146 GMainContext *context;
147 };
148 typedef struct QIOChannelTLSData QIOChannelTLSData;
149
150 static gboolean qio_channel_tls_handshake_io(QIOChannel *ioc,
151 GIOCondition condition,
152 gpointer user_data);
153
154 static void qio_channel_tls_handshake_task(QIOChannelTLS *ioc,
155 QIOTask *task,
156 GMainContext *context)
157 {
158 Error *err = NULL;
159 QCryptoTLSSessionHandshakeStatus status;
160
161 if (qcrypto_tls_session_handshake(ioc->session, &err) < 0) {
162 trace_qio_channel_tls_handshake_fail(ioc);
163 qio_task_set_error(task, err);
164 qio_task_complete(task);
165 return;
166 }
167
168 status = qcrypto_tls_session_get_handshake_status(ioc->session);
169 if (status == QCRYPTO_TLS_HANDSHAKE_COMPLETE) {
170 trace_qio_channel_tls_handshake_complete(ioc);
171 if (qcrypto_tls_session_check_credentials(ioc->session,
172 &err) < 0) {
173 trace_qio_channel_tls_credentials_deny(ioc);
174 qio_task_set_error(task, err);
175 } else {
176 trace_qio_channel_tls_credentials_allow(ioc);
177 }
178 qio_task_complete(task);
179 } else {
180 GIOCondition condition;
181 QIOChannelTLSData *data = g_new0(typeof(*data), 1);
182
183 data->task = task;
184 data->context = context;
185
186 if (context) {
187 g_main_context_ref(context);
188 }
189
190 if (status == QCRYPTO_TLS_HANDSHAKE_SENDING) {
191 condition = G_IO_OUT;
192 } else {
193 condition = G_IO_IN;
194 }
195
196 trace_qio_channel_tls_handshake_pending(ioc, status);
197 qio_channel_add_watch_full(ioc->master,
198 condition,
199 qio_channel_tls_handshake_io,
200 data,
201 NULL,
202 context);
203 }
204 }
205
206
207 static gboolean qio_channel_tls_handshake_io(QIOChannel *ioc,
208 GIOCondition condition,
209 gpointer user_data)
210 {
211 QIOChannelTLSData *data = user_data;
212 QIOTask *task = data->task;
213 GMainContext *context = data->context;
214 QIOChannelTLS *tioc = QIO_CHANNEL_TLS(
215 qio_task_get_source(task));
216
217 g_free(data);
218 qio_channel_tls_handshake_task(tioc, task, context);
219
220 if (context) {
221 g_main_context_unref(context);
222 }
223
224 return FALSE;
225 }
226
227 void qio_channel_tls_handshake(QIOChannelTLS *ioc,
228 QIOTaskFunc func,
229 gpointer opaque,
230 GDestroyNotify destroy,
231 GMainContext *context)
232 {
233 QIOTask *task;
234
235 task = qio_task_new(OBJECT(ioc),
236 func, opaque, destroy);
237
238 trace_qio_channel_tls_handshake_start(ioc);
239 qio_channel_tls_handshake_task(ioc, task, context);
240 }
241
242
243 static void qio_channel_tls_init(Object *obj G_GNUC_UNUSED)
244 {
245 }
246
247
248 static void qio_channel_tls_finalize(Object *obj)
249 {
250 QIOChannelTLS *ioc = QIO_CHANNEL_TLS(obj);
251
252 object_unref(OBJECT(ioc->master));
253 qcrypto_tls_session_free(ioc->session);
254 }
255
256
257 static ssize_t qio_channel_tls_readv(QIOChannel *ioc,
258 const struct iovec *iov,
259 size_t niov,
260 int **fds,
261 size_t *nfds,
262 Error **errp)
263 {
264 QIOChannelTLS *tioc = QIO_CHANNEL_TLS(ioc);
265 size_t i;
266 ssize_t got = 0;
267
268 for (i = 0 ; i < niov ; i++) {
269 ssize_t ret = qcrypto_tls_session_read(tioc->session,
270 iov[i].iov_base,
271 iov[i].iov_len);
272 if (ret < 0) {
273 if (errno == EAGAIN) {
274 if (got) {
275 return got;
276 } else {
277 return QIO_CHANNEL_ERR_BLOCK;
278 }
279 } else if (errno == ECONNABORTED &&
280 (tioc->shutdown & QIO_CHANNEL_SHUTDOWN_READ)) {
281 return 0;
282 }
283
284 error_setg_errno(errp, errno,
285 "Cannot read from TLS channel");
286 return -1;
287 }
288 got += ret;
289 if (ret < iov[i].iov_len) {
290 break;
291 }
292 }
293 return got;
294 }
295
296
297 static ssize_t qio_channel_tls_writev(QIOChannel *ioc,
298 const struct iovec *iov,
299 size_t niov,
300 int *fds,
301 size_t nfds,
302 Error **errp)
303 {
304 QIOChannelTLS *tioc = QIO_CHANNEL_TLS(ioc);
305 size_t i;
306 ssize_t done = 0;
307
308 for (i = 0 ; i < niov ; i++) {
309 ssize_t ret = qcrypto_tls_session_write(tioc->session,
310 iov[i].iov_base,
311 iov[i].iov_len);
312 if (ret <= 0) {
313 if (errno == EAGAIN) {
314 if (done) {
315 return done;
316 } else {
317 return QIO_CHANNEL_ERR_BLOCK;
318 }
319 }
320
321 error_setg_errno(errp, errno,
322 "Cannot write to TLS channel");
323 return -1;
324 }
325 done += ret;
326 if (ret < iov[i].iov_len) {
327 break;
328 }
329 }
330 return done;
331 }
332
333 static int qio_channel_tls_set_blocking(QIOChannel *ioc,
334 bool enabled,
335 Error **errp)
336 {
337 QIOChannelTLS *tioc = QIO_CHANNEL_TLS(ioc);
338
339 return qio_channel_set_blocking(tioc->master, enabled, errp);
340 }
341
342 static void qio_channel_tls_set_delay(QIOChannel *ioc,
343 bool enabled)
344 {
345 QIOChannelTLS *tioc = QIO_CHANNEL_TLS(ioc);
346
347 qio_channel_set_delay(tioc->master, enabled);
348 }
349
350 static void qio_channel_tls_set_cork(QIOChannel *ioc,
351 bool enabled)
352 {
353 QIOChannelTLS *tioc = QIO_CHANNEL_TLS(ioc);
354
355 qio_channel_set_cork(tioc->master, enabled);
356 }
357
358 static int qio_channel_tls_shutdown(QIOChannel *ioc,
359 QIOChannelShutdown how,
360 Error **errp)
361 {
362 QIOChannelTLS *tioc = QIO_CHANNEL_TLS(ioc);
363
364 tioc->shutdown |= how;
365
366 return qio_channel_shutdown(tioc->master, how, errp);
367 }
368
369 static int qio_channel_tls_close(QIOChannel *ioc,
370 Error **errp)
371 {
372 QIOChannelTLS *tioc = QIO_CHANNEL_TLS(ioc);
373
374 return qio_channel_close(tioc->master, errp);
375 }
376
377 static void qio_channel_tls_set_aio_fd_handler(QIOChannel *ioc,
378 AioContext *ctx,
379 IOHandler *io_read,
380 IOHandler *io_write,
381 void *opaque)
382 {
383 QIOChannelTLS *tioc = QIO_CHANNEL_TLS(ioc);
384
385 qio_channel_set_aio_fd_handler(tioc->master, ctx, io_read, io_write, opaque);
386 }
387
388 static GSource *qio_channel_tls_create_watch(QIOChannel *ioc,
389 GIOCondition condition)
390 {
391 QIOChannelTLS *tioc = QIO_CHANNEL_TLS(ioc);
392
393 return qio_channel_create_watch(tioc->master, condition);
394 }
395
396 QCryptoTLSSession *
397 qio_channel_tls_get_session(QIOChannelTLS *ioc)
398 {
399 return ioc->session;
400 }
401
402 static void qio_channel_tls_class_init(ObjectClass *klass,
403 void *class_data G_GNUC_UNUSED)
404 {
405 QIOChannelClass *ioc_klass = QIO_CHANNEL_CLASS(klass);
406
407 ioc_klass->io_writev = qio_channel_tls_writev;
408 ioc_klass->io_readv = qio_channel_tls_readv;
409 ioc_klass->io_set_blocking = qio_channel_tls_set_blocking;
410 ioc_klass->io_set_delay = qio_channel_tls_set_delay;
411 ioc_klass->io_set_cork = qio_channel_tls_set_cork;
412 ioc_klass->io_close = qio_channel_tls_close;
413 ioc_klass->io_shutdown = qio_channel_tls_shutdown;
414 ioc_klass->io_create_watch = qio_channel_tls_create_watch;
415 ioc_klass->io_set_aio_fd_handler = qio_channel_tls_set_aio_fd_handler;
416 }
417
418 static const TypeInfo qio_channel_tls_info = {
419 .parent = TYPE_QIO_CHANNEL,
420 .name = TYPE_QIO_CHANNEL_TLS,
421 .instance_size = sizeof(QIOChannelTLS),
422 .instance_init = qio_channel_tls_init,
423 .instance_finalize = qio_channel_tls_finalize,
424 .class_init = qio_channel_tls_class_init,
425 };
426
427 static void qio_channel_tls_register_types(void)
428 {
429 type_register_static(&qio_channel_tls_info);
430 }
431
432 type_init(qio_channel_tls_register_types);