From 93c1be4c5fb1f3b5cf190e182f9988227aa8c09f Mon Sep 17 00:00:00 2001 From: Christian Kellner Date: Fri, 24 Jul 2020 20:03:22 +0200 Subject: [PATCH] remoteloop: use high level message dispatcher Use the new `BaseAPI._message` high level message dispatcher that is more convenient to use. --- osbuild/remoteloop.py | 14 ++++++-------- 1 file changed, 6 insertions(+), 8 deletions(-) diff --git a/osbuild/remoteloop.py b/osbuild/remoteloop.py index 27a664a6..498e5623 100644 --- a/osbuild/remoteloop.py +++ b/osbuild/remoteloop.py @@ -70,16 +70,14 @@ class LoopServer(api.BaseAPI): self.devs.append(lo) return lo.devname - def _dispatch(self, server): - args, fds, addr = server.recv() - - fd = fds[args["fd"]] - dir_fd = fds[args["dir_fd"]] - offset = args.get("offset") - sizelimit = args.get("sizelimit") + def _message(self, msg, fds, sock, addr): + fd = fds[msg["fd"]] + dir_fd = fds[msg["dir_fd"]] + offset = msg.get("offset") + sizelimit = msg.get("sizelimit") devname = self._create_device(fd, dir_fd, offset, sizelimit) - server.send({"devname": devname}, destination=addr) + sock.send({"devname": devname}, destination=addr) fds.close() def _cleanup(self):