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
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
20typedef struct {
21 Coroutine *coroutine;
22 uint64_t offset;
23 bool receiving;
24} NBDClientRequest;
25
26typedef struct NBDClientSession {
27 QIOChannelSocket *sioc;
28 QIOChannel *ioc;
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
41NBDClientSession *nbd_get_client_session(BlockDriverState *bs);
42
43int 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);
50void nbd_client_close(BlockDriverState *bs);
51
52int nbd_client_co_pdiscard(BlockDriverState *bs, int64_t offset, int bytes);
53int nbd_client_co_flush(BlockDriverState *bs);
54int nbd_client_co_pwritev(BlockDriverState *bs, uint64_t offset,
55 uint64_t bytes, QEMUIOVector *qiov, int flags);
56int nbd_client_co_pwrite_zeroes(BlockDriverState *bs, int64_t offset,
57 int bytes, BdrvRequestFlags flags);
58int nbd_client_co_preadv(BlockDriverState *bs, uint64_t offset,
59 uint64_t bytes, QEMUIOVector *qiov, int flags);
60
61void nbd_client_detach_aio_context(BlockDriverState *bs);
62void nbd_client_attach_aio_context(BlockDriverState *bs,
63 AioContext *new_context);
64
65int 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
72