other micro optimisations

tlsfuzzer · Sep 23, 2020 · 71f7e24 · 71f7e24
1 parent 37d9e1a
commit 71f7e24
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/tlslite/utils/codec.py b/tlslite/utils/codec.py
@@ -318,11 +318,12 @@ def getFixBytes(self, lengthBytes):
 
         :rtype: bytearray
         """
-        if self.index + lengthBytes > len(self.bytes):
+        end = self.index + lengthBytes
+        if end > len(self.bytes):
             raise DecodeError("Read past end of buffer")
-        bytes = self.bytes[self.index : self.index+lengthBytes]
+        ret = self.bytes[self.index : end]
         self.index += lengthBytes
-        return bytes
+        return ret
 
     def skip_bytes(self, length):
         """Move the internal pointer ahead length bytes."""

diff --git a/tlslite/utils/rsakey.py b/tlslite/utils/rsakey.py
@@ -534,11 +534,10 @@ def _addPKCS1Padding(self, bytes, blockType):
             pad = bytearray(0)
             while len(pad) < padLength:
                 padBytes = getRandomBytes(padLength * 2)
-                pad = [b for b in padBytes if b != 0]
+                pad = [b for b in padBytes if b]
                 pad = pad[:padLength]
         else:
             raise AssertionError()
 
         padding = bytearray([0,blockType] + pad + [0])
-        paddedBytes = padding + bytes
-        return paddedBytes
+        return padding + bytes