From b55ce66a1b89ac347bdf5ebd38b744494a5c10d6 Mon Sep 17 00:00:00 2001
From: whitequark <whitequark@whitequark.org>
Date: Tue, 22 May 2018 21:31:34 +0000
Subject: [PATCH] Reinstate accidentally removed aqctl_corelog.

Closes #1008.
---
 artiq/frontend/aqctl_corelog.py | 81 +++++++++++++++++++++++++++++++++
 1 file changed, 81 insertions(+)
 create mode 100755 artiq/frontend/aqctl_corelog.py

diff --git a/artiq/frontend/aqctl_corelog.py b/artiq/frontend/aqctl_corelog.py
new file mode 100755
index 000000000..7cee0b974
--- /dev/null
+++ b/artiq/frontend/aqctl_corelog.py
@@ -0,0 +1,81 @@
+#!/usr/bin/env python3
+
+import argparse
+import asyncio
+import struct
+import logging
+import re
+
+from artiq.tools import *
+from artiq.protocols.pc_rpc import Server
+from artiq.protocols.logging import log_with_name
+from artiq.coredevice.comm_mgmt import Request, Reply
+
+
+def get_argparser():
+    parser = argparse.ArgumentParser(
+        description="ARTIQ controller for core device logs")
+    simple_network_args(parser, 1068)
+    parser.add_argument("core_addr",
+                        help="hostname or IP address of the core device")
+    return parser
+
+
+class PingTarget:
+    def ping(self):
+        return True
+
+
+async def get_logs(host):
+    reader, writer = await asyncio.open_connection(host, 1380)
+    writer.write(b"ARTIQ management\n")
+    writer.write(struct.pack("B", Request.PullLog.value))
+    await writer.drain()
+
+    while True:
+        length, = struct.unpack(">l", await reader.readexactly(4))
+        log = await reader.readexactly(length)
+
+        for line in log.decode("utf-8").splitlines():
+            m = re.match(r"^\[.+?\] (TRACE|DEBUG| INFO| WARN|ERROR)\((.+?)\): (.+)$", line)
+            levelname = m.group(1)
+            if levelname == 'TRACE':
+                level = logging.TRACE
+            elif levelname == 'DEBUG':
+                level = logging.DEBUG
+            elif levelname == ' INFO':
+                level = logging.INFO
+            elif levelname == ' WARN':
+                level = logging.WARN
+            elif levelname == 'ERROR':
+                level = logging.ERROR
+            name = 'firmware.' + m.group(2).replace('::', '.')
+            text = m.group(3)
+            log_with_name(name, level, text)
+
+
+def main():
+    args = get_argparser().parse_args()
+
+    loop = asyncio.get_event_loop()
+    try:
+        get_logs_task = asyncio.ensure_future(get_logs(args.core_addr))
+        try:
+            server = Server({"corelog": PingTarget()}, None, True)
+            loop.run_until_complete(server.start(bind_address_from_args(args), args.port))
+            try:
+                multiline_log_config(logging.TRACE)
+                loop.run_until_complete(server.wait_terminate())
+            finally:
+                loop.run_until_complete(server.stop())
+        finally:
+            get_logs_task.cancel()
+            try:
+                loop.run_until_complete(get_logs_task)
+            except asyncio.CancelledError:
+                pass
+    finally:
+        loop.close()
+
+if __name__ == "__main__":
+    main()