Ensure that RSA signatures have the correct length

third_party/tlslite/tlslite/utils/RSAKey.py

 """Abstract class for RSA."""
 
 from cryptomath import *
 
 
 class RSAKey:
     """This is an abstract base class for RSA keys.
 
     Particular implementations of RSA keys, such as
     L{OpenSSL_RSAKey.OpenSSL_RSAKey},
@@ skipping 99 matching lines @@
         @rtype: L{array.array} of unsigned bytes.
         @return: A PKCS1 signature on the passed-in data.
         """
         if not self.hasPrivateKey():
             raise AssertionError()
         paddedBytes = self._addPKCS1Padding(bytes, 1)
         m = bytesToNumber(paddedBytes)
         if m >= self.n:
             raise ValueError()
         c = self._rawPrivateKeyOp(m)
-        sigBytes = numberToBytes(c)
+        sigBytes = numberToBytes(c, numBytes(self.n))
         return sigBytes
 
     def verify(self, sigBytes, bytes):
         """Verify the passed-in bytes with the signature.
 
         This verifies a PKCS1 signature on the passed-in data.
 
         @type sigBytes: L{array.array} of unsigned bytes
         @param sigBytes: A PKCS1 signature.
 
@@ skipping 124 matching lines @@
         else:
             raise AssertionError()
 
         #NOTE: To be proper, we should add [0,blockType].  However,
         #the zero is lost when the returned padding is converted
         #to a number, so we don't even bother with it.  Also,
         #adding it would cause a misalignment in verify()
         padding = createByteArraySequence([blockType] + pad + [0])
         paddedBytes = padding + bytes
         return paddedBytes