Prevent stacktrace during SIGTERM signal and open_device call

Signed-off-by: Ronan Abhamon <ronan.abhamon@vates.tech>
xcp-ng · Nov 18, 2024 · ca6e310 · ca6e310
1 parent 54ae171
commit ca6e310
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/http_disk_server.py b/http_disk_server.py
@@ -425,6 +425,10 @@ def emit_server_ready():
                 disk_fd = open_device(disk, retry=False)
             else:
                 disk_fd = open_device(disk)
+
+            if SIGTERM_RECEIVED:
+                break
+
             is_block_device = stat.S_ISBLK(os.fstat(disk_fd).st_mode)
 
             HandlerClass = MakeRequestHandler(disk_fd, is_block_device)

diff --git a/setup.py b/setup.py
@@ -20,7 +20,7 @@
 
 setup(
     name="http-nbd-transfer",
-    version="1.4.0",
+    version="1.5.0",
     description="Set of tools to transfer NBD requests to an HTTP server",
     author="Ronan Abhamon <ronan.abhamon@vates.tech>",
     author_email="ronan.abhamon@vates.tech",