200 lines
6.5 KiB
Diff
200 lines
6.5 KiB
Diff
>From a962453ed73a671f566cc94858dc21ab694cc85f Mon Sep 17 00:00:00 2001
|
|
From: Amit Shah <amit.shah@redhat.com>
|
|
Date: Mon, 21 Mar 2011 22:00:27 +0100
|
|
Subject: [PATCH 05/28] char: Update send_all() to handle nonblocking chardev
|
|
write requests
|
|
|
|
The send_all function is modified to return to the caller in case the
|
|
driver cannot handle any more data. It returns -EAGAIN or
|
|
WSAEWOULDBLOCK on non-Windows and Windows platforms respectively. This
|
|
is only done when the caller sets a callback function handler indicating
|
|
it's not interested in blocking till the driver has written out all the
|
|
data.
|
|
|
|
Currently there's no driver or caller that supports this. Future
|
|
commits will add such capability.
|
|
|
|
Signed-off-by: Amit Shah <amit.shah@redhat.com>
|
|
---
|
|
net/socket.c | 4 +-
|
|
qemu-char.c | 80 ++++++++++++++++++++++++++++++++++++++++++++++++++++----
|
|
qemu_socket.h | 2 +-
|
|
3 files changed, 77 insertions(+), 9 deletions(-)
|
|
|
|
diff --git a/net/socket.c b/net/socket.c
|
|
index 11fe5f3..14706fc 100644
|
|
--- a/net/socket.c
|
|
+++ b/net/socket.c
|
|
@@ -56,8 +56,8 @@ static ssize_t net_socket_receive(VLANClientState *nc, const uint8_t *buf, size_
|
|
uint32_t len;
|
|
len = htonl(size);
|
|
|
|
- send_all(s->fd, (const uint8_t *)&len, sizeof(len));
|
|
- return send_all(s->fd, buf, size);
|
|
+ send_all(NULL, s->fd, (const uint8_t *)&len, sizeof(len));
|
|
+ return send_all(NULL, s->fd, buf, size);
|
|
}
|
|
|
|
static ssize_t net_socket_receive_dgram(VLANClientState *nc, const uint8_t *buf, size_t size)
|
|
diff --git a/qemu-char.c b/qemu-char.c
|
|
index bc07b20..5dbf063 100644
|
|
--- a/qemu-char.c
|
|
+++ b/qemu-char.c
|
|
@@ -513,7 +513,7 @@ static CharDriverState *qemu_chr_open_mux(CharDriverState *drv)
|
|
|
|
|
|
#ifdef _WIN32
|
|
-int send_all(int fd, const void *buf, int len1)
|
|
+static int do_send(int fd, const void *buf, int len1, bool nonblock)
|
|
{
|
|
int ret, len;
|
|
|
|
@@ -521,9 +521,14 @@ int send_all(int fd, const void *buf, int len1)
|
|
while (len > 0) {
|
|
ret = send(fd, buf, len, 0);
|
|
if (ret < 0) {
|
|
+ if (nonblock && len1 - len) {
|
|
+ return len1 - len;
|
|
+ }
|
|
errno = WSAGetLastError();
|
|
if (errno != WSAEWOULDBLOCK) {
|
|
return -1;
|
|
+ } else if (errno == WSAEWOULDBLOCK && nonblock) {
|
|
+ return WSAEWOULDBLOCK;
|
|
}
|
|
} else if (ret == 0) {
|
|
break;
|
|
@@ -537,7 +542,7 @@ int send_all(int fd, const void *buf, int len1)
|
|
|
|
#else
|
|
|
|
-int send_all(int fd, const void *_buf, int len1)
|
|
+static int do_send(int fd, const void *_buf, int len1, bool nonblock)
|
|
{
|
|
int ret, len;
|
|
const uint8_t *buf = _buf;
|
|
@@ -546,8 +551,15 @@ int send_all(int fd, const void *_buf, int len1)
|
|
while (len > 0) {
|
|
ret = write(fd, buf, len);
|
|
if (ret < 0) {
|
|
- if (errno != EINTR && errno != EAGAIN)
|
|
+ if (nonblock && len1 - len) {
|
|
+ return len1 - len;
|
|
+ }
|
|
+ if (errno == EAGAIN && nonblock) {
|
|
+ return -EAGAIN;
|
|
+ }
|
|
+ if (errno != EINTR && errno != EAGAIN) {
|
|
return -1;
|
|
+ }
|
|
} else if (ret == 0) {
|
|
break;
|
|
} else {
|
|
@@ -559,6 +571,55 @@ int send_all(int fd, const void *_buf, int len1)
|
|
}
|
|
#endif /* !_WIN32 */
|
|
|
|
+int send_all(CharDriverState *chr, int fd, const void *_buf, int len1)
|
|
+{
|
|
+ int ret, eagain_errno;
|
|
+ bool nonblock;
|
|
+
|
|
+ if (chr && chr->write_blocked) {
|
|
+ /*
|
|
+ * We don't handle this situation: the caller should not send
|
|
+ * us data while we're blocked.
|
|
+ *
|
|
+ * We could buffer this data here but that'll only encourage
|
|
+ * bad behaviour on part of the callers.
|
|
+ *
|
|
+ * Also, the data already in fd's buffers isn't easily
|
|
+ * migratable. If we want full migration support, all the
|
|
+ * data landing here needs to be buffered and on migration,
|
|
+ * anything that's unsent needs to be transferred to the
|
|
+ * dest. machine (which again isn't a very good way of solving
|
|
+ * the problem, as the src may become writable just during
|
|
+ * migration and the reader could receive some data twice,
|
|
+ * essentially corrupting the data).
|
|
+ */
|
|
+ abort();
|
|
+ }
|
|
+
|
|
+ nonblock = false;
|
|
+ /*
|
|
+ * Ensure the char backend is able to receive and handle the
|
|
+ * 'write unblocked' event before we turn on nonblock support.
|
|
+ */
|
|
+ if (chr && chr->chr_enable_write_fd_handler && chr->chr_write_unblocked) {
|
|
+ nonblock = true;
|
|
+ }
|
|
+ ret = do_send(fd, _buf, len1, nonblock);
|
|
+
|
|
+#ifdef _WIN32
|
|
+ eagain_errno = WSAEWOULDBLOCK;
|
|
+#else
|
|
+ eagain_errno = -EAGAIN;
|
|
+#endif
|
|
+
|
|
+ if (nonblock && (ret == eagain_errno || (ret >= 0 && ret < len1))) {
|
|
+ /* Update fd handler to wake up when chr becomes writable */
|
|
+ chr->chr_enable_write_fd_handler(chr);
|
|
+ chr->write_blocked = true;
|
|
+ }
|
|
+ return ret;
|
|
+}
|
|
+
|
|
#ifndef _WIN32
|
|
|
|
typedef struct {
|
|
@@ -572,7 +633,7 @@ static int stdio_nb_clients = 0;
|
|
static int fd_chr_write(CharDriverState *chr, const uint8_t *buf, int len)
|
|
{
|
|
FDCharDriver *s = chr->opaque;
|
|
- return send_all(s->fd_out, buf, len);
|
|
+ return send_all(chr, s->fd_out, buf, len);
|
|
}
|
|
|
|
static int fd_chr_read_poll(void *opaque)
|
|
@@ -897,7 +958,7 @@ static int pty_chr_write(CharDriverState *chr, const uint8_t *buf, int len)
|
|
pty_chr_update_read_handler(chr);
|
|
return 0;
|
|
}
|
|
- return send_all(s->fd, buf, len);
|
|
+ return send_all(chr, s->fd, buf, len);
|
|
}
|
|
|
|
static int pty_chr_read_poll(void *opaque)
|
|
@@ -1979,8 +2040,15 @@ static void tcp_closed(void *opaque)
|
|
static int tcp_chr_write(CharDriverState *chr, const uint8_t *buf, int len)
|
|
{
|
|
TCPCharDriver *s = chr->opaque;
|
|
+
|
|
if (s->connected) {
|
|
- return send_all(s->fd, buf, len);
|
|
+ int ret;
|
|
+
|
|
+ ret = send_all(chr, s->fd, buf, len);
|
|
+ if (ret == -1 && errno == EPIPE) {
|
|
+ tcp_closed(chr);
|
|
+ }
|
|
+ return ret;
|
|
} else {
|
|
/* XXX: indicate an error ? */
|
|
return len;
|
|
diff --git a/qemu_socket.h b/qemu_socket.h
|
|
index 180e4db..6f453e5 100644
|
|
--- a/qemu_socket.h
|
|
+++ b/qemu_socket.h
|
|
@@ -36,7 +36,7 @@ int inet_aton(const char *cp, struct in_addr *ia);
|
|
int qemu_socket(int domain, int type, int protocol);
|
|
int qemu_accept(int s, struct sockaddr *addr, socklen_t *addrlen);
|
|
void socket_set_nonblock(int fd);
|
|
-int send_all(int fd, const void *buf, int len1);
|
|
+int send_all(CharDriverState *chr, int fd, const void *buf, int len1);
|
|
|
|
/* New, ipv6-ready socket helper functions, see qemu-sockets.c */
|
|
int inet_listen_opts(QemuOpts *opts, int port_offset);
|
|
--
|
|
1.7.5.1
|
|
|