nbd/client: Add x-dirty-bitmap to query bitmap from server
[qemu.git] / 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 */