From 24ae0ba1682e26b406e960a0e2cef82b4d39ea32 Mon Sep 17 00:00:00 2001
From: Markus Birth <mbirth@gmail.com>
Date: Thu, 2 Nov 2017 02:02:13 +0100
Subject: [PATCH] Dump all XML responses to files for later upload.

---
 .gitignore |  1 +
 tcllib.py  | 14 ++++++++++++++
 2 files changed, 15 insertions(+)

diff --git a/.gitignore b/.gitignore
index db84f99..bfc1f68 100644
--- a/.gitignore
+++ b/.gitignore
@@ -1,3 +1,4 @@
 /__pycache__/
+/logs/
 /.nvimlog
 /header*.bin
diff --git a/tcllib.py b/tcllib.py
index 5ca388d..2ccddbf 100644
--- a/tcllib.py
+++ b/tcllib.py
@@ -6,7 +6,9 @@ import argparse
 import base64
 import binascii
 import enum
+import errno
 import hashlib
+import os
 import platform
 import random
 import time
@@ -135,6 +137,17 @@ class FotaCheck:
         elif last_duration > avg_duration + 0.5:
             self.master_server_downvote()
 
+    def write_dump(self, data):
+        outfile = "logs/{}.xml".format(self.get_salt())
+        if not os.path.exists(os.path.dirname(outfile)):
+            try:
+                os.makedirs(os.path.dirname(outfile))
+            except OSError as e:
+                if e.errno != errno.EEXIST:
+                    raise
+        with open(outfile, "w") as f:
+            f.write(data)
+
     def do_check(self, https=True, timeout=10, max_tries=5):
         protocol = "https://" if https else "http://"
         url = protocol + self.g2master + "/check.php"
@@ -162,6 +175,7 @@ class FotaCheck:
                 last_response = req
                 if req.status_code == 200:
                     self.master_server_vote_on_time(reqtime, reqtime_avg)
+                    self.write_dump(req.text)
                     return req.text
                 elif req.status_code == 204:
                     self.master_server_vote_on_time(reqtime, reqtime_avg)