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 NbdClientSession {
21 QIOChannelSocket *sioc;
22 QIOChannel *ioc;
23 uint32_t nbdflags;
24 off_t size;
25
26 CoMutex send_mutex;
27 CoMutex free_sema;
28 Coroutine *send_coroutine;
29 int in_flight;
30
31 Coroutine *recv_coroutine[MAX_NBD_REQUESTS];
32 struct nbd_reply reply;
33
34 bool is_unix;
35} NbdClientSession;
36
37NbdClientSession *nbd_get_client_session(BlockDriverState *bs);
38
39int nbd_client_init(BlockDriverState *bs,
40 QIOChannelSocket *sock,
41 const char *export_name,
42 QCryptoTLSCreds *tlscreds,
43 const char *hostname,
44 Error **errp);
45void nbd_client_close(BlockDriverState *bs);
46
47int nbd_client_co_discard(BlockDriverState *bs, int64_t sector_num,
48 int nb_sectors);
49int nbd_client_co_flush(BlockDriverState *bs);
50int nbd_client_co_writev(BlockDriverState *bs, int64_t sector_num,
51 int nb_sectors, QEMUIOVector *qiov, int *flags);
52int nbd_client_co_readv(BlockDriverState *bs, int64_t sector_num,
53 int nb_sectors, QEMUIOVector *qiov);
54
55void nbd_client_detach_aio_context(BlockDriverState *bs);
56void nbd_client_attach_aio_context(BlockDriverState *bs,
57 AioContext *new_context);
58
59#endif
60