block/nbd-client.h

   1 #ifndef NBD_CLIENT_H
   2 #define NBD_CLIENT_H
   3
   4 #include "block/nbd.h"
   5 #include "block/block_int.h"
   6 #include "io/channel-socket.h"
   7
   8 /* #define DEBUG_NBD */
   9
  10 #if defined(DEBUG_NBD)
  11 #define logout(fmt, ...) \
  12     fprintf(stderr, "nbd\t%-24s" fmt, __func__, ##__VA_ARGS__)
  13 #else
  14 #define logout(fmt, ...) ((void)0)
  15 #endif
  16
  17 #define MAX_NBD_REQUESTS    16
  18
  19 typedef struct {
  20     Coroutine *coroutine;
  21     uint64_t offset;        /* original offset of the request */
  22     bool receiving;         /* waiting for connection_co? */
  23 } NBDClientRequest;
  24
  25 typedef struct NBDClientSession {
  26     QIOChannelSocket *sioc; /* The master data channel */
  27     QIOChannel *ioc; /* The current I/O channel which may differ (eg TLS) */
  28     NBDExportInfo info;
  29
  30     CoMutex send_mutex;
  31     CoQueue free_sema;
  32     Coroutine *connection_co;
  33     int in_flight;
  34
  35     NBDClientRequest requests[MAX_NBD_REQUESTS];
  36     NBDReply reply;
  37     BlockDriverState *bs;
  38     bool quit;
  39 } NBDClientSession;
  40
  41 NBDClientSession *nbd_get_client_session(BlockDriverState *bs);
  42
  43 int nbd_client_init(BlockDriverState *bs,
  44                     SocketAddress *saddr,
  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 */