tools/qubesd-query: handle events-based methods

Pass the data as it arrives, instead of buffering it all. QubesOS/qubes-issues#2622
2017-04-10 01:47:04 +02:00 · 2017-04-10 01:47:04 +02:00 · 7e4bc4249d
commit 7e4bc4249d
parent 40a86efd66
1 changed files with 4 additions and 2 deletions
--- a/qubes/tools/qubesd_query.py
+++ b/qubes/tools/qubesd_query.py
@ -54,8 +54,10 @@ def qubesd_client(socket, payload, *args):
    writer.write_eof()

    try:
-        data = yield from reader.read()
-        sys.stdout.buffer.write(data)  # pylint: disable=no-member
+        while not reader.at_eof():
+            data = yield from reader.read(4096)
+            sys.stdout.buffer.write(data)  # pylint: disable=no-member
+            sys.stdout.flush()
    except asyncio.CancelledError:
        return
    finally: