diff mbox series

[4/8] trace-cmd: Check if connection is done when reading data in tracecmd_msg_read_data

Message ID 20190204070855.8921-5-kaslevs@vmware.com (mailing list archive)
State Accepted
Commit 7cd03e1cb38a701331ac5c93ac183306e22680c4
Headers show
Series trace-cmd protocol fixes | expand

Commit Message

Slavomir Kaslev Feb. 4, 2019, 7:08 a.m. UTC 
Now that tracecmd_msg_done/tracecmd_msg_set_done can be used in both server and
client context, check if the connection is shutting down in tracecmd_msg_read_data.

Signed-off-by: Slavomir Kaslev <kaslevs@vmware.com>
---
 tracecmd/trace-msg.c | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)
diff mbox series

Patch

diff --git a/tracecmd/trace-msg.c b/tracecmd/trace-msg.c
index ef2a6d8..f7ce863 100644
--- a/tracecmd/trace-msg.c
+++ b/tracecmd/trace-msg.c
@@ -603,7 +603,7 @@  int tracecmd_msg_read_data(struct tracecmd_msg_handle *msg_handle, int ofd)
 	ssize_t s;
 	int ret;
 
-	for (;;) {
+	while (!tracecmd_msg_done(msg_handle)) {
 		ret = tracecmd_msg_recv_wait(msg_handle->fd, &msg);
 		if (ret < 0) {
 			if (ret == -ETIMEDOUT)