]> git.ipfire.org Git - thirdparty/suricata-update.git/commitdiff
Use HTTP-Headers for checksum request
authorJascha Sticher <jascha.sticher@ausecus.com>
Fri, 16 Jul 2021 13:15:29 +0000 (15:15 +0200)
committerJason Ish <jason.ish@oisf.net>
Thu, 7 Oct 2021 18:16:40 +0000 (12:16 -0600)
suricata/update/main.py

index b1f08191cb1beaab84f319f2965f3ac78dc5aaaa..eed2805b3b7df66d37681b263ba56e9840aefe94 100644 (file)
@@ -106,12 +106,13 @@ class Fetch:
 
     def check_checksum(self, tmp_filename, url):
         try:
-            checksum_url = url + ".md5"
+            checksum_url = url[0] + ".md5"
+            net_arg=(checksum_url,url[1])
             local_checksum = hashlib.md5(
                 open(tmp_filename, "rb").read()).hexdigest().strip()
             remote_checksum_buf = io.BytesIO()
             logger.info("Checking %s." % (checksum_url))
-            net.get(checksum_url, remote_checksum_buf)
+            net.get(net_arg, remote_checksum_buf)
             remote_checksum = remote_checksum_buf.getvalue().decode().strip()
             logger.debug("Local checksum=|%s|; remote checksum=|%s|" % (
                 local_checksum, remote_checksum))
@@ -174,7 +175,7 @@ class Fetch:
                     url)
                 return self.extract_files(tmp_filename)
             if checksum:
-                if self.check_checksum(tmp_filename, url):
+                if self.check_checksum(tmp_filename, net_arg):
                     logger.info("Remote checksum has not changed. "
                                 "Not fetching.")
                     return self.extract_files(tmp_filename)