block/nbd-client.h

   1 #ifndef NBD_CLIENT_H
   2 #define NBD_CLIENT_H
   3
   4 #include "qemu-common.h"
   5 #include "block/nbd.h"
   6 #include "block/block_int.h"
   7 #include "io/channel-socket.h"
   8
   9 /* #define DEBUG_NBD */
  10
  11 #if defined(DEBUG_NBD)
  12 #define logout(fmt, ...) \
  13     fprintf(stderr, "nbd\t%-24s" fmt, __func__, ##__VA_ARGS__)
  14 #else
  15 #define logout(fmt, ...) ((void)0)
  16 #endif
  17
  18 #define MAX_NBD_REQUESTS    16
  19
  20 typedef struct {
  21     Coroutine *coroutine;
  22     uint64_t offset;        /* original offset of the request */
  23     bool receiving;         /* waiting for read_reply_co? */
  24 } NBDClientRequest;
  25
  26 typedef struct NBDClientSession {
  27     QIOChannelSocket *sioc; /* The master data channel */
  28     QIOChannel *ioc; /* The current I/O channel which may differ (eg TLS) */
  29     NBDExportInfo info;
  30
  31     CoMutex send_mutex;
  32     CoQueue free_sema;
  33     Coroutine *read_reply_co;
  34     int in_flight;
  35
  36     NBDClientRequest requests[MAX_NBD_REQUESTS];
  37     NBDReply reply;
  38     bool quit;
  39 } NBDClientSession;
  40
  41 NBDClientSession *nbd_get_client_session(BlockDriverState *bs);
  42
  43 int nbd_client_init(BlockDriverState *bs,
  44                     QIOChannelSocket *sock,
  45                     const char *export_name,
  46                     QCryptoTLSCreds *tlscreds,
  47                     const char *hostname,
  48                     const char *x_dirty_bitmap,
  49                     Error **errp);
  50 void nbd_client_close(BlockDriverState *bs);
  51
  52 int nbd_client_co_pdiscard(BlockDriverState *bs, int64_t offset, int bytes);
  53 int nbd_client_co_flush(BlockDriverState *bs);
  54 int nbd_client_co_pwritev(BlockDriverState *bs, uint64_t offset,
  55                           uint64_t bytes, QEMUIOVector *qiov, int flags);
  56 int nbd_client_co_pwrite_zeroes(BlockDriverState *bs, int64_t offset,
  57                                 int bytes, BdrvRequestFlags flags);
  58 int nbd_client_co_preadv(BlockDriverState *bs, uint64_t offset,
  59                          uint64_t bytes, QEMUIOVector *qiov, int flags);
  60
  61 void nbd_client_detach_aio_context(BlockDriverState *bs);
  62 void nbd_client_attach_aio_context(BlockDriverState *bs,
  63                                    AioContext *new_context);
  64
  65 int coroutine_fn nbd_client_co_block_status(BlockDriverState *bs,
  66                                             bool want_zero,
  67                                             int64_t offset, int64_t bytes,
  68                                             int64_t *pnum, int64_t *map,
  69                                             BlockDriverState **file);
  70
  71 #endif /* NBD_CLIENT_H */