From 0d1e50ababa29939f18fecd8d29a03a3b3972e86 Mon Sep 17 00:00:00 2001
From: Markus Birth <mbirth@gmail.com>
Date: Sat, 10 Feb 2018 03:12:24 +0100
Subject: [PATCH] Converted ChecksumRequest to new style.

---
 tclcheck.py                        | 21 +++++++++------
 tcllib/requests/__init__.py        |  1 +
 tcllib/requests/checksumrequest.py | 42 ++++++++++++++++++++++++++++++
 tcllib/requests/tclresult.py       | 10 +++++++
 4 files changed, 66 insertions(+), 8 deletions(-)
 create mode 100644 tcllib/requests/checksumrequest.py

diff --git a/tclcheck.py b/tclcheck.py
index c5a83c9..6ae2eeb 100755
--- a/tclcheck.py
+++ b/tclcheck.py
@@ -12,7 +12,7 @@ import sys
 import tcllib
 import tcllib.argparser
 from tcllib.devices import Device
-from tcllib.requests import RequestRunner, CheckRequest, DownloadRequest, ServerSelector, write_info_if_dumps_found
+from tcllib.requests import RequestRunner, CheckRequest, DownloadRequest, ChecksumRequest, ServerSelector, write_info_if_dumps_found
 from tcllib.xmltools import pretty_xml
 
 
@@ -88,15 +88,20 @@ dlrres = dlr.get_result()
 print(dlrres.pretty_xml())
 
 if dlrres.encslaves:
-    chksum_xml = fc.do_checksum(random.choice(encslaves), fileurl, fileurl)
-    print(pretty_xml(chksum_xml))
-    file_addr, sha1_body, sha1_enc_footer, sha1_footer = fc.parse_checksum(chksum_xml)
+    cksrunner = RequestRunner(ServerSelector(dlrres.encslaves), https=False)
+    cks = ChecksumRequest(dlrres.fileurl, dlrres.fileurl)
+    cksrunner.run(cks)
+    if not cks.success:
+        print("{}".format(cks.error))
+        sys.exit(4)
+    cksres = cks.get_result()
+    print(cksres.pretty_xml())
 
-for s in slaves:
-    print("http://{}{}".format(s, fileurl))
+for s in dlrres.slaves:
+    print("http://{}{}".format(s, dlrres.fileurl))
 
-for s in s3_slaves:
-    print("http://{}{}".format(s, s3_fileurl))
+for s in dlrres.s3_slaves:
+    print("http://{}{}".format(s, dlrres.s3_fileurl))
 
 if dev.mode == dev.MODE_STATES["FULL"]:
     header = fc.do_encrypt_header(random.choice(encslaves), fileurl)
diff --git a/tcllib/requests/__init__.py b/tcllib/requests/__init__.py
index 8e0148e..93e7cd3 100644
--- a/tcllib/requests/__init__.py
+++ b/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
diff --git a/tcllib/requests/checksumrequest.py b/tcllib/requests/checksumrequest.py
new file mode 100644
index 0000000..2ec2f89
--- /dev/null
+++ b/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
diff --git a/tcllib/requests/tclresult.py b/tcllib/requests/tclresult.py
index 7899b4d..146fba0 100644
--- a/tcllib/requests/tclresult.py
+++ b/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