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 NBDClientSession {
  21     QIOChannelSocket *sioc; /* The master data channel */
  22     QIOChannel *ioc; /* The current I/O channel which may differ (eg TLS) */
  23     uint16_t nbdflags;
  24     off_t size;
  25
  26     CoMutex send_mutex;
  27     CoQueue free_sema;
  28     Coroutine *read_reply_co;
  29     int in_flight;
  30
  31     Coroutine *recv_coroutine[MAX_NBD_REQUESTS];
  32     NBDReply reply;
  33 } NBDClientSession;
  34
  35 NBDClientSession *nbd_get_client_session(BlockDriverState *bs);
  36
  37 int nbd_client_init(BlockDriverState *bs,
  38                     QIOChannelSocket *sock,
  39                     const char *export_name,
  40                     QCryptoTLSCreds *tlscreds,
  41                     const char *hostname,
  42                     Error **errp);
  43 void nbd_client_close(BlockDriverState *bs);
  44
  45 int nbd_client_co_pdiscard(BlockDriverState *bs, int64_t offset, int count);
  46 int nbd_client_co_flush(BlockDriverState *bs);
  47 int nbd_client_co_pwritev(BlockDriverState *bs, uint64_t offset,
  48                           uint64_t bytes, QEMUIOVector *qiov, int flags);
  49 int nbd_client_co_pwrite_zeroes(BlockDriverState *bs, int64_t offset,
  50                                 int count, BdrvRequestFlags flags);
  51 int nbd_client_co_preadv(BlockDriverState *bs, uint64_t offset,
  52                          uint64_t bytes, QEMUIOVector *qiov, int flags);
  53
  54 void nbd_client_detach_aio_context(BlockDriverState *bs);
  55 void nbd_client_attach_aio_context(BlockDriverState *bs,
  56                                    AioContext *new_context);
  57
  58 #endif /* NBD_CLIENT_H */