unbound-dhcp-leases-bridge: Replace leases file atomically

[ipfire-2.x.git] / config / unbound / unbound-dhcp-leases-bridge

diff --git a/config/unbound/unbound-dhcp-leases-bridge b/config/unbound/unbound-dhcp-leases-bridge
index 54cd8135ba9fea2192b03a8266b0b61402f3bf1f..a8cd837bbfbf17b08cf03cd77eb5c11f7b56233d 100644 (file)
--- a/config/unbound/unbound-dhcp-leases-bridge
+++ b/config/unbound/unbound-dhcp-leases-bridge
@@ -25,9 +25,11 @@ import daemon
 import ipaddress
 import logging
 import logging.handlers
 import ipaddress
 import logging
 import logging.handlers

+import os

 import re
 import signal
 import subprocess
 import re
 import signal
 import subprocess

+import tempfile

 
 import inotify.adapters
 
 
 import inotify.adapters
 


@@ -519,11 +521,15 @@ class UnboundConfigWriter(object):
                                self._cached_leases.append(l)
 
        def write_dhcp_leases(self, leases):
                                self._cached_leases.append(l)
 
        def write_dhcp_leases(self, leases):

-               with open(self.path, "w") as f:
+               with tempfile.NamedTemporaryFile(mode="w", delete=False) as f:
+                       filename = f.name
+

                        for l in leases:
                                for rr in l.rrset:
                                        f.write("local-data: \"%s\"\n" % " ".join(rr))
 
                        for l in leases:
                                for rr in l.rrset:
                                        f.write("local-data: \"%s\"\n" % " ".join(rr))
 

+               os.rename(filename, self.path)
+

        def _control(self, *args):
                command = ["unbound-control"]
                command.extend(args)
        def _control(self, *args):
                command = ["unbound-control"]
                command.extend(args)