From d3968de3b4b578e0bc1bfa283fa5034be93fe028 Mon Sep 17 00:00:00 2001
From: Bjoern Kerler <info@revskills.de>
Date: Sat, 27 Aug 2022 16:10:16 +0200
Subject: [PATCH] Fix missing totalsectors init

---
 edlclient/Library/firehose.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/edlclient/Library/firehose.py b/edlclient/Library/firehose.py
index ff0a791..488c48a 100755
--- a/edlclient/Library/firehose.py
+++ b/edlclient/Library/firehose.py
@@ -48,6 +48,7 @@ class nand_partition:
         self.partitiontblsector = None
         self.parent = parent
         self.storage_info = {}
+        self.totalsectors = None
 
     def parse(self, partdata):
         self.partentries = {}
@@ -889,7 +890,7 @@ class firehose(metaclass=LogBase):
                     else:
                         self.error("Error on EDL Authentification")
                         return False
-                elif "MaxPayloadSizeToTargetInBytes" in line:
+                elif "MaxPayloadSizeToTargetInBytes" in rsp.data:
                     try:
                         self.cfg.MemoryName = rsp.data["MemoryName"]
                         self.cfg.MaxPayloadSizeToTargetInBytes = int(rsp.data["MaxPayloadSizeToTargetInBytes"])