nbd/nbd-internal.h

   1 /*
   2  * NBD Internal Declarations
   3  *
   4  * Copyright (C) 2016 Red Hat, Inc.
   5  *
   6  * This work is licensed under the terms of the GNU GPL, version 2 or later.
   7  * See the COPYING file in the top-level directory.
   8  */
   9
  10 #ifndef NBD_INTERNAL_H
  11 #define NBD_INTERNAL_H
  12 #include "block/nbd.h"
  13 #include "sysemu/block-backend.h"
  14 #include "io/channel-tls.h"
  15
  16 #include "qemu/coroutine.h"
  17 #include "qemu/iov.h"
  18
  19 #ifndef _WIN32
  20 #include <sys/ioctl.h>
  21 #endif
  22 #if defined(__sun__) || defined(__HAIKU__)
  23 #include <sys/ioccom.h>
  24 #endif
  25
  26 #ifdef __linux__
  27 #include <linux/fs.h>
  28 #endif
  29
  30 #include "qemu/bswap.h"
  31 #include "qemu/queue.h"
  32 #include "qemu/main-loop.h"
  33
  34 /* This is all part of the "official" NBD API.
  35  *
  36  * The most up-to-date documentation is available at:
  37  * https://github.com/yoe/nbd/blob/master/doc/proto.md
  38  */
  39
  40 /* Size of all NBD_OPT_*, without payload */
  41 #define NBD_REQUEST_SIZE            (4 + 2 + 2 + 8 + 8 + 4)
  42 /* Size of all NBD_REP_* sent in answer to most NBD_OPT_*, without payload */
  43 #define NBD_REPLY_SIZE              (4 + 4 + 8)
  44 /* Size of reply to NBD_OPT_EXPORT_NAME */
  45 #define NBD_REPLY_EXPORT_NAME_SIZE  (8 + 2 + 124)
  46 /* Size of oldstyle negotiation */
  47 #define NBD_OLDSTYLE_NEGOTIATE_SIZE (8 + 8 + 8 + 4 + 124)
  48
  49 #define NBD_REQUEST_MAGIC       0x25609513
  50 #define NBD_REPLY_MAGIC         0x67446698
  51 #define NBD_OPTS_MAGIC          0x49484156454F5054LL
  52 #define NBD_CLIENT_MAGIC        0x0000420281861253LL
  53 #define NBD_REP_MAGIC           0x0003e889045565a9LL
  54
  55 #define NBD_SET_SOCK            _IO(0xab, 0)
  56 #define NBD_SET_BLKSIZE         _IO(0xab, 1)
  57 #define NBD_SET_SIZE            _IO(0xab, 2)
  58 #define NBD_DO_IT               _IO(0xab, 3)
  59 #define NBD_CLEAR_SOCK          _IO(0xab, 4)
  60 #define NBD_CLEAR_QUE           _IO(0xab, 5)
  61 #define NBD_PRINT_DEBUG         _IO(0xab, 6)
  62 #define NBD_SET_SIZE_BLOCKS     _IO(0xab, 7)
  63 #define NBD_DISCONNECT          _IO(0xab, 8)
  64 #define NBD_SET_TIMEOUT         _IO(0xab, 9)
  65 #define NBD_SET_FLAGS           _IO(0xab, 10)
  66
  67 /* NBD errors are based on errno numbers, so there is a 1:1 mapping,
  68  * but only a limited set of errno values is specified in the protocol.
  69  * Everything else is squashed to EINVAL.
  70  */
  71 #define NBD_SUCCESS    0
  72 #define NBD_EPERM      1
  73 #define NBD_EIO        5
  74 #define NBD_ENOMEM     12
  75 #define NBD_EINVAL     22
  76 #define NBD_ENOSPC     28
  77 #define NBD_ESHUTDOWN  108
  78
  79 /* nbd_read_eof
  80  * Tries to read @size bytes from @ioc.
  81  * Returns 1 on success
  82  *         0 on eof, when no data was read (errp is not set)
  83  *         negative errno on failure (errp is set)
  84  */
  85 static inline int nbd_read_eof(QIOChannel *ioc, void *buffer, size_t size,
  86                                Error **errp)
  87 {
  88     struct iovec iov = { .iov_base = buffer, .iov_len = size };
  89     ssize_t ret;
  90
  91     /* Sockets are kept in blocking mode in the negotiation phase.  After
  92      * that, a non-readable socket simply means that another thread stole
  93      * our request/reply.  Synchronization is done with recv_coroutine, so
  94      * that this is coroutine-safe.
  95      */
  96
  97     assert(size);
  98
  99     ret = nbd_rwv(ioc, &iov, 1, size, true, errp);
 100     if (ret <= 0) {
 101         return ret;
 102     }
 103
 104     if (ret != size) {
 105         error_setg(errp, "End of file");
 106         return -EINVAL;
 107     }
 108
 109     return 1;
 110 }
 111
 112 /* nbd_read
 113  * Reads @size bytes from @ioc. Returns 0 on success.
 114  */
 115 static inline int nbd_read(QIOChannel *ioc, void *buffer, size_t size,
 116                            Error **errp)
 117 {
 118     int ret = nbd_read_eof(ioc, buffer, size, errp);
 119
 120     if (ret == 0) {
 121         ret = -EINVAL;
 122         error_setg(errp, "End of file");
 123     }
 124
 125     return ret < 0 ? ret : 0;
 126 }
 127
 128 /* nbd_write
 129  * Writes @size bytes to @ioc. Returns 0 on success.
 130  */
 131 static inline int nbd_write(QIOChannel *ioc, const void *buffer, size_t size,
 132                             Error **errp)
 133 {
 134     struct iovec iov = { .iov_base = (void *) buffer, .iov_len = size };
 135
 136     ssize_t ret = nbd_rwv(ioc, &iov, 1, size, false, errp);
 137
 138     assert(ret < 0 || ret == size);
 139
 140     return ret < 0 ? ret : 0;
 141 }
 142
 143 struct NBDTLSHandshakeData {
 144     GMainLoop *loop;
 145     bool complete;
 146     Error *error;
 147 };
 148
 149
 150 void nbd_tls_handshake(QIOTask *task,
 151                        void *opaque);
 152 const char *nbd_opt_lookup(uint32_t opt);
 153 const char *nbd_rep_lookup(uint32_t rep);
 154 const char *nbd_info_lookup(uint16_t info);
 155 const char *nbd_cmd_lookup(uint16_t info);
 156
 157 int nbd_drop(QIOChannel *ioc, size_t size, Error **errp);
 158
 159 #endif