From: Jascha Sticher <jascha.sticher@ausecus.com>
Date: Fri, 16 Jul 2021 13:15:29 +0000 (+0200)
Subject: Use HTTP-Headers for checksum request
X-Git-Tag: 1.3.0rc1~51
X-Git-Url: http://git.ipfire.org/cgi-bin/gitweb.cgi?a=commitdiff_plain;h=74c3e6d2396176a0c64af54cf4680496f38749b8;p=thirdparty%2Fsuricata-update.git

Use HTTP-Headers for checksum request
---

diff --git a/suricata/update/main.py b/suricata/update/main.py
index b1f0819..eed2805 100644
--- a/suricata/update/main.py
+++ b/suricata/update/main.py
@@ -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)