Converted ChecksumRequest to new style.

tcllib/requests/__init__.py

@@ -2,6 +2,7 @@
 
 from .checkrequest import CheckRequest
 from .downloadrequest import DownloadRequest
+from .checksumrequest import ChecksumRequest
 from .runner import *
 from .serverselector import *
 from .dumpmgr import write_info_if_dumps_found

tcllib/requests/checksumrequest.py

@@ -0,0 +1,42 @@
+# -*- coding: utf-8 -*-
+
+from collections import OrderedDict
+import json
+from .. import credentials, devices
+from .tclrequest import TclRequest
+from .tclresult import ChecksumResult
+
+class ChecksumRequest(TclRequest):
+    def __init__(self, address, file_uri):
+        super().__init__()
+        # NOTE: THIS HAS TO BE RUN ON AN ENCSLAVE
+        self.uri = "/checksum.php"
+        self.method = "POST"
+        self.address = address
+        self.file_uri = file_uri
+
+    def get_headers(self):
+        return {"User-Agent": "tcl"}
+
+    def get_params(self):
+        params = OrderedDict()
+        params.update(credentials.get_creds2())
+        payload = {self.address: self.file_uri}
+        payload_json = json.dumps(payload)
+        params["address"] = bytes(payload_json, "utf-8")
+        return params
+
+    def is_done(self, http_status: int, contents: str) -> bool:
+        if http_status == 200:
+            # <ENCRYPT_FOOTER>2abfa6f6507044fec995efede5d818e62a0b19b5</ENCRYPT_FOOTER> means ERROR (invalid ADDRESS!)
+            if "<ENCRYPT_FOOTER>2abfa6f6507044fec995efede5d818e62a0b19b5</ENCRYPT_FOOTER>" in contents:
+                self.error = "INVALID URI: {}".format(self.file_uri)
+                self.success = False
+                return True
+            self.response = contents
+            self.result = ChecksumResult(contents)
+            self.success = True
+            return True
+        self.error = "HTTP {}".format(http_status)
+        self.success = False
+        return True

tcllib/requests/tclresult.py

@@ -52,3 +52,13 @@ class DownloadResult(TclResult):
         self.slaves = [s.text for s in slave_list]
         self.encslaves = [s.text for s in enc_list]
         self.s3_slaves = [s.text for s in s3_slave_list]
+
+class ChecksumResult(TclResult):
+    def __init__(self, xml: str):
+        super().__init__(xml)
+        root = ElementTree.fromstring(xml)
+        file = root.find("FILE_CHECKSUM_LIST").find("FILE")
+        self.file_addr = file.find("ADDRESS").text
+        self.sha1_enc_footer = file.find("ENCRYPT_FOOTER").text
+        self.sha1_footer = file.find("FOOTER").text
+        self.sha1_body = file.find("BODY").text