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);