提交 3777b09f 编写于 作者: P Paolo Bonzini

qemu-nbd: remove data_size argument to nbd_trip

The size of the buffer is in practice part of the protocol.
Signed-off-by: NPaolo Bonzini <pbonzini@redhat.com>
上级 94607e7a
...@@ -585,7 +585,7 @@ static int nbd_send_reply(int csock, struct nbd_reply *reply) ...@@ -585,7 +585,7 @@ static int nbd_send_reply(int csock, struct nbd_reply *reply)
int nbd_trip(BlockDriverState *bs, int csock, off_t size, int nbd_trip(BlockDriverState *bs, int csock, off_t size,
uint64_t dev_offset, uint32_t nbdflags, uint64_t dev_offset, uint32_t nbdflags,
uint8_t *data, int data_size) uint8_t *data)
{ {
struct nbd_request request; struct nbd_request request;
struct nbd_reply reply; struct nbd_reply reply;
...@@ -596,9 +596,9 @@ int nbd_trip(BlockDriverState *bs, int csock, off_t size, ...@@ -596,9 +596,9 @@ int nbd_trip(BlockDriverState *bs, int csock, off_t size,
if (nbd_receive_request(csock, &request) == -1) if (nbd_receive_request(csock, &request) == -1)
return -1; return -1;
if (request.len + NBD_REPLY_SIZE > data_size) { if (request.len + NBD_REPLY_SIZE > NBD_BUFFER_SIZE) {
LOG("len (%u) is larger than max len (%u)", LOG("len (%u) is larger than max len (%u)",
request.len + NBD_REPLY_SIZE, data_size); request.len + NBD_REPLY_SIZE, NBD_BUFFER_SIZE);
errno = EINVAL; errno = EINVAL;
return -1; return -1;
} }
......
...@@ -57,6 +57,8 @@ enum { ...@@ -57,6 +57,8 @@ enum {
#define NBD_DEFAULT_PORT 10809 #define NBD_DEFAULT_PORT 10809
#define NBD_BUFFER_SIZE (1024*1024)
size_t nbd_wr_sync(int fd, void *buffer, size_t size, bool do_read); size_t nbd_wr_sync(int fd, void *buffer, size_t size, bool do_read);
int tcp_socket_outgoing(const char *address, uint16_t port); int tcp_socket_outgoing(const char *address, uint16_t port);
int tcp_socket_incoming(const char *address, uint16_t port); int tcp_socket_incoming(const char *address, uint16_t port);
...@@ -72,7 +74,7 @@ int nbd_init(int fd, int csock, uint32_t flags, off_t size, size_t blocksize); ...@@ -72,7 +74,7 @@ int nbd_init(int fd, int csock, uint32_t flags, off_t size, size_t blocksize);
int nbd_send_request(int csock, struct nbd_request *request); int nbd_send_request(int csock, struct nbd_request *request);
int nbd_receive_reply(int csock, struct nbd_reply *reply); int nbd_receive_reply(int csock, struct nbd_reply *reply);
int nbd_trip(BlockDriverState *bs, int csock, off_t size, uint64_t dev_offset, int nbd_trip(BlockDriverState *bs, int csock, off_t size, uint64_t dev_offset,
uint32_t nbdflags, uint8_t *data, int data_size); uint32_t nbdflags, uint8_t *data);
int nbd_client(int fd); int nbd_client(int fd);
int nbd_disconnect(int fd); int nbd_disconnect(int fd);
......
...@@ -35,8 +35,6 @@ ...@@ -35,8 +35,6 @@
#define SOCKET_PATH "/var/lock/qemu-nbd-%s" #define SOCKET_PATH "/var/lock/qemu-nbd-%s"
#define NBD_BUFFER_SIZE (1024*1024)
static int sigterm_wfd; static int sigterm_wfd;
static int verbose; static int verbose;
static char *device; static char *device;
...@@ -541,7 +539,7 @@ int main(int argc, char **argv) ...@@ -541,7 +539,7 @@ int main(int argc, char **argv)
for (i = 1; i < nb_fds && ret; i++) { for (i = 1; i < nb_fds && ret; i++) {
if (FD_ISSET(sharing_fds[i], &fds)) { if (FD_ISSET(sharing_fds[i], &fds)) {
if (nbd_trip(bs, sharing_fds[i], fd_size, dev_offset, if (nbd_trip(bs, sharing_fds[i], fd_size, dev_offset,
nbdflags, data, NBD_BUFFER_SIZE) != 0) { nbdflags, data) != 0) {
close(sharing_fds[i]); close(sharing_fds[i]);
nb_fds--; nb_fds--;
sharing_fds[i] = sharing_fds[nb_fds]; sharing_fds[i] = sharing_fds[nb_fds];
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册