From 13c16418075fc46d6f762332418fed9a628027d6 Mon Sep 17 00:00:00 2001
From: Luke Hackwell <luke.hackwell@arm.com>
Date: Wed, 21 Jan 2026 10:08:29 +0000
Subject: [PATCH 1/2] feat: print the host and port where coordinator listens

Print that exporter is ready.
---
 labgrid/remote/coordinator.py | 6 +++++-
 labgrid/remote/exporter.py    | 1 +
 2 files changed, 6 insertions(+), 1 deletion(-)

diff --git a/labgrid/remote/coordinator.py b/labgrid/remote/coordinator.py
index 385d471f9..fef6952b0 100644
--- a/labgrid/remote/coordinator.py
+++ b/labgrid/remote/coordinator.py
@@ -1214,7 +1214,7 @@ async def serve(listen, cleanup) -> None:
     except ImportError:
         logging.info("Module grpcio-channelz not available")
 
-    server.add_insecure_port(listen)
+    bound = server.add_insecure_port(listen)
     logging.debug("Starting server")
     await server.start()
 
@@ -1230,6 +1230,10 @@ async def server_graceful_shutdown():
 
     cleanup.append(server_graceful_shutdown())
     logging.info("Coordinator ready")
+    host, sep, port = listen.rpartition(":")
+    if not sep or not port.isdigit():
+        host = listen
+    print(f"listening on {host}:{bound}", flush=True)
     await server.wait_for_termination()
 
 
diff --git a/labgrid/remote/exporter.py b/labgrid/remote/exporter.py
index b69166863..48593da2a 100755
--- a/labgrid/remote/exporter.py
+++ b/labgrid/remote/exporter.py
@@ -958,6 +958,7 @@ async def message_pump(self):
                 logging.debug("received message %s", out_message)
                 kind = out_message.WhichOneof("kind")
                 if kind == "hello":
+                    print("exporter ready", flush=True)
                     logging.info("connected to coordinator version %s", out_message.hello.version)
                 elif kind == "set_acquired_request":
                     logging.debug("acquire request")

From 9bf81b01197796068b0b9f1684d8bd1d06509951 Mon Sep 17 00:00:00 2001
From: Luke Hackwell <luke.hackwell@arm.com>
Date: Wed, 21 Jan 2026 10:09:02 +0000
Subject: [PATCH 2/2] fix: gracefully shutdown coordinator and exporter on
 SIGTERM and SIGINT

---
 labgrid/remote/coordinator.py |  7 +++++++
 labgrid/remote/exporter.py    | 20 ++++++++++++++++++--
 2 files changed, 25 insertions(+), 2 deletions(-)

diff --git a/labgrid/remote/coordinator.py b/labgrid/remote/coordinator.py
index fef6952b0..80597abe3 100644
--- a/labgrid/remote/coordinator.py
+++ b/labgrid/remote/coordinator.py
@@ -9,6 +9,7 @@
 from contextlib import contextmanager
 import copy
 import random
+import signal
 
 import attr
 import grpc
@@ -1228,7 +1229,13 @@ async def server_graceful_shutdown():
         # existing RPCs to continue within the grace period.
         await server.stop(5)
 
+    def callback():
+        asyncio.ensure_future(server_graceful_shutdown())
+
     cleanup.append(server_graceful_shutdown())
+    loop = asyncio.get_event_loop()
+    loop.add_signal_handler(signal.SIGINT, callback)
+    loop.add_signal_handler(signal.SIGTERM, callback)
     logging.info("Coordinator ready")
     host, sep, port = listen.rpartition(":")
     if not sep or not port.isdigit():
diff --git a/labgrid/remote/exporter.py b/labgrid/remote/exporter.py
index 48593da2a..eb232b7f9 100755
--- a/labgrid/remote/exporter.py
+++ b/labgrid/remote/exporter.py
@@ -8,6 +8,7 @@
 import os
 import os.path
 import traceback
+import signal
 import shutil
 import subprocess
 from urllib.parse import urlsplit
@@ -941,8 +942,11 @@ async def run(self) -> None:
         for task in pending:
             task.cancel()
 
-        await self.pump_task
-        await self.poll_task
+        try:
+            await self.pump_task
+            await self.poll_task
+        except asyncio.CancelledError:
+            return
 
     def send_started(self):
         msg = labgrid_coordinator_pb2.ExporterInMessage()
@@ -1067,6 +1071,11 @@ async def poll(self):
             except Exception:  # pylint: disable=broad-except
                 traceback.print_exc(file=sys.stderr)
 
+    async def stop(self):
+        if None is self.poll_task:
+            return
+        self.poll_task.cancel()
+
     async def add_resource(self, group_name, resource_name, cls, params):
         """Add a resource to the exporter and update status on the coordinator"""
         print(f"add resource {group_name}/{resource_name}: {cls}/{params}")
@@ -1109,6 +1118,13 @@ async def amain(config) -> bool:
     if inspect:
         inspect.exporter = exporter
 
+    def _stop():
+        asyncio.ensure_future(exporter.stop())
+
+    loop = asyncio.get_event_loop()
+    loop.add_signal_handler(signal.SIGINT, _stop)
+    loop.add_signal_handler(signal.SIGTERM, _stop)
+
     await exporter.run()