src/third_party/tlslite/tlslite/mathtls.py

   1 """Miscellaneous helper functions."""
   2
   3 from utils.compat import *
   4 from utils.cryptomath import *
   5
   6 import hmac
   7
   8 # The sha module is deprecated in Python 2.6
   9 try:
  10     import sha
  11 except ImportError:
  12     from hashlib import sha1 as sha
  13
  14 # The md5 module is deprecated in Python 2.6
  15 try:
  16     import md5
  17 except ImportError:
  18     from hashlib import md5
  19
  20 #1024, 1536, 2048, 3072, 4096, 6144, and 8192 bit groups]
  21 goodGroupParameters = [(2,0xEEAF0AB9ADB38DD69C33F80AFA8FC5E86072618775FF3C0B9EA2314C9C256576D674DF7496EA81D3383B4813D692C6E0E0D5D8E250B98BE48E495C1D6089DAD15DC7D7B46154D6B6CE8EF4AD69B15D4982559B297BCF1885C529F566660E57EC68EDBC3C05726CC02FD4CBF4976EAA9AFD5138FE8376435B9FC61D2FC0EB06E3),\
  22                        (2,0x9DEF3CAFB939277AB1F12A8617A47BBBDBA51DF499AC4C80BEEEA9614B19CC4D5F4F5F556E27CBDE51C6A94BE4607A291558903BA0D0F84380B655BB9A22E8DCDF028A7CEC67F0D08134B1C8B97989149B609E0BE3BAB63D47548381DBC5B1FC764E3F4B53DD9DA1158BFD3E2B9C8CF56EDF019539349627DB2FD53D24B7C48665772E437D6C7F8CE442734AF7CCB7AE837C264AE3A9BEB87F8A2FE9B8B5292E5A021FFF5E91479E8CE7A28C2442C6F315180F93499A234DCF76E3FED135F9BB),\
  23                        (2,0xAC6BDB41324A9A9BF166DE5E1389582FAF72B6651987EE07FC3192943DB56050A37329CBB4A099ED8193E0757767A13DD52312AB4B03310DCD7F48A9DA04FD50E8083969EDB767B0CF6095179A163AB3661A05FBD5FAAAE82918A9962F0B93B855F97993EC975EEAA80D740ADBF4FF747359D041D5C33EA71D281E446B14773BCA97B43A23FB801676BD207A436C6481F1D2B9078717461A5B9D32E688F87748544523B524B0D57D5EA77A2775D2ECFA032CFBDBF52FB3786160279004E57AE6AF874E7303CE53299CCC041C7BC308D82A5698F3A8D0C38271AE35F8E9DBFBB694B5C803D89F7AE435DE236D525F54759B65E372FCD68EF20FA7111F9E4AFF73),\
  24                        (2,0xFFFFFFFFFFFFFFFFC90FDAA22168C234C4C6628B80DC1CD129024E088A67CC74020BBEA63B139B22514A08798E3404DDEF9519B3CD3A431B302B0A6DF25F14374FE1356D6D51C245E485B576625E7EC6F44C42E9A637ED6B0BFF5CB6F406B7EDEE386BFB5A899FA5AE9F24117C4B1FE649286651ECE45B3DC2007CB8A163BF0598DA48361C55D39A69163FA8FD24CF5F83655D23DCA3AD961C62F356208552BB9ED529077096966D670C354E4ABC9804F1746C08CA18217C32905E462E36CE3BE39E772C180E86039B2783A2EC07A28FB5C55DF06F4C52C9DE2BCBF6955817183995497CEA956AE515D2261898FA051015728E5A8AAAC42DAD33170D04507A33A85521ABDF1CBA64ECFB850458DBEF0A8AEA71575D060C7DB3970F85A6E1E4C7ABF5AE8CDB0933D71E8C94E04A25619DCEE3D2261AD2EE6BF12FFA06D98A0864D87602733EC86A64521F2B18177B200CBBE117577A615D6C770988C0BAD946E208E24FA074E5AB3143DB5BFCE0FD108E4B82D120A93AD2CAFFFFFFFFFFFFFFFF),\
  25                        (5,0xFFFFFFFFFFFFFFFFC90FDAA22168C234C4C6628B80DC1CD129024E088A67CC74020BBEA63B139B22514A08798E3404DDEF9519B3CD3A431B302B0A6DF25F14374FE1356D6D51C245E485B576625E7EC6F44C42E9A637ED6B0BFF5CB6F406B7EDEE386BFB5A899FA5AE9F24117C4B1FE649286651ECE45B3DC2007CB8A163BF0598DA48361C55D39A69163FA8FD24CF5F83655D23DCA3AD961C62F356208552BB9ED529077096966D670C354E4ABC9804F1746C08CA18217C32905E462E36CE3BE39E772C180E86039B2783A2EC07A28FB5C55DF06F4C52C9DE2BCBF6955817183995497CEA956AE515D2261898FA051015728E5A8AAAC42DAD33170D04507A33A85521ABDF1CBA64ECFB850458DBEF0A8AEA71575D060C7DB3970F85A6E1E4C7ABF5AE8CDB0933D71E8C94E04A25619DCEE3D2261AD2EE6BF12FFA06D98A0864D87602733EC86A64521F2B18177B200CBBE117577A615D6C770988C0BAD946E208E24FA074E5AB3143DB5BFCE0FD108E4B82D120A92108011A723C12A787E6D788719A10BDBA5B2699C327186AF4E23C1A946834B6150BDA2583E9CA2AD44CE8DBBBC2DB04DE8EF92E8EFC141FBECAA6287C59474E6BC05D99B2964FA090C3A2233BA186515BE7ED1F612970CEE2D7AFB81BDD762170481CD0069127D5B05AA993B4EA988D8FDDC186FFB7DC90A6C08F4DF435C934063199FFFFFFFFFFFFFFFF),\
  26                        (5,0xFFFFFFFFFFFFFFFFC90FDAA22168C234C4C6628B80DC1CD129024E088A67CC74020BBEA63B139B22514A08798E3404DDEF9519B3CD3A431B302B0A6DF25F14374FE1356D6D51C245E485B576625E7EC6F44C42E9A637ED6B0BFF5CB6F406B7EDEE386BFB5A899FA5AE9F24117C4B1FE649286651ECE45B3DC2007CB8A163BF0598DA48361C55D39A69163FA8FD24CF5F83655D23DCA3AD961C62F356208552BB9ED529077096966D670C354E4ABC9804F1746C08CA18217C32905E462E36CE3BE39E772C180E86039B2783A2EC07A28FB5C55DF06F4C52C9DE2BCBF6955817183995497CEA956AE515D2261898FA051015728E5A8AAAC42DAD33170D04507A33A85521ABDF1CBA64ECFB850458DBEF0A8AEA71575D060C7DB3970F85A6E1E4C7ABF5AE8CDB0933D71E8C94E04A25619DCEE3D2261AD2EE6BF12FFA06D98A0864D87602733EC86A64521F2B18177B200CBBE117577A615D6C770988C0BAD946E208E24FA074E5AB3143DB5BFCE0FD108E4B82D120A92108011A723C12A787E6D788719A10BDBA5B2699C327186AF4E23C1A946834B6150BDA2583E9CA2AD44CE8DBBBC2DB04DE8EF92E8EFC141FBECAA6287C59474E6BC05D99B2964FA090C3A2233BA186515BE7ED1F612970CEE2D7AFB81BDD762170481CD0069127D5B05AA993B4EA988D8FDDC186FFB7DC90A6C08F4DF435C93402849236C3FAB4D27C7026C1D4DCB2602646DEC9751E763DBA37BDF8FF9406AD9E530EE5DB382F413001AEB06A53ED9027D831179727B0865A8918DA3EDBEBCF9B14ED44CE6CBACED4BB1BDB7F1447E6CC254B332051512BD7AF426FB8F401378CD2BF5983CA01C64B92ECF032EA15D1721D03F482D7CE6E74FEF6D55E702F46980C82B5A84031900B1C9E59E7C97FBEC7E8F323A97A7E36CC88BE0F1D45B7FF585AC54BD407B22B4154AACC8F6D7EBF48E1D814CC5ED20F8037E0A79715EEF29BE32806A1D58BB7C5DA76F550AA3D8A1FBFF0EB19CCB1A313D55CDA56C9EC2EF29632387FE8D76E3C0468043E8F663F4860EE12BF2D5B0B7474D6E694F91E6DCC4024FFFFFFFFFFFFFFFF),\
  27                        (5,0xFFFFFFFFFFFFFFFFC90FDAA22168C234C4C6628B80DC1CD129024E088A67CC74020BBEA63B139B22514A08798E3404DDEF9519B3CD3A431B302B0A6DF25F14374FE1356D6D51C245E485B576625E7EC6F44C42E9A637ED6B0BFF5CB6F406B7EDEE386BFB5A899FA5AE9F24117C4B1FE649286651ECE45B3DC2007CB8A163BF0598DA48361C55D39A69163FA8FD24CF5F83655D23DCA3AD961C62F356208552BB9ED529077096966D670C354E4ABC9804F1746C08CA18217C32905E462E36CE3BE39E772C180E86039B2783A2EC07A28FB5C55DF06F4C52C9DE2BCBF6955817183995497CEA956AE515D2261898FA051015728E5A8AAAC42DAD33170D04507A33A85521ABDF1CBA64ECFB850458DBEF0A8AEA71575D060C7DB3970F85A6E1E4C7ABF5AE8CDB0933D71E8C94E04A25619DCEE3D2261AD2EE6BF12FFA06D98A0864D87602733EC86A64521F2B18177B200CBBE117577A615D6C770988C0BAD946E208E24FA074E5AB3143DB5BFCE0FD108E4B82D120A92108011A723C12A787E6D788719A10BDBA5B2699C327186AF4E23C1A946834B6150BDA2583E9CA2AD44CE8DBBBC2DB04DE8EF92E8EFC141FBECAA6287C59474E6BC05D99B2964FA090C3A2233BA186515BE7ED1F612970CEE2D7AFB81BDD762170481CD0069127D5B05AA993B4EA988D8FDDC186FFB7DC90A6C08F4DF435C93402849236C3FAB4D27C7026C1D4DCB2602646DEC9751E763DBA37BDF8FF9406AD9E530EE5DB382F413001AEB06A53ED9027D831179727B0865A8918DA3EDBEBCF9B14ED44CE6CBACED4BB1BDB7F1447E6CC254B332051512BD7AF426FB8F401378CD2BF5983CA01C64B92ECF032EA15D1721D03F482D7CE6E74FEF6D55E702F46980C82B5A84031900B1C9E59E7C97FBEC7E8F323A97A7E36CC88BE0F1D45B7FF585AC54BD407B22B4154AACC8F6D7EBF48E1D814CC5ED20F8037E0A79715EEF29BE32806A1D58BB7C5DA76F550AA3D8A1FBFF0EB19CCB1A313D55CDA56C9EC2EF29632387FE8D76E3C0468043E8F663F4860EE12BF2D5B0B7474D6E694F91E6DBE115974A3926F12FEE5E438777CB6A932DF8CD8BEC4D073B931BA3BC832B68D9DD300741FA7BF8AFC47ED2576F6936BA424663AAB639C5AE4F5683423B4742BF1C978238F16CBE39D652DE3FDB8BEFC848AD922222E04A4037C0713EB57A81A23F0C73473FC646CEA306B4BCBC8862F8385DDFA9D4B7FA2C087E879683303ED5BDD3A062B3CF5B3A278A66D2A13F83F44F82DDF310EE074AB6A364597E899A0255DC164F31CC50846851DF9AB48195DED7EA1B1D510BD7EE74D73FAF36BC31ECFA268359046F4EB879F924009438B481C6CD7889A002ED5EE382BC9190DA6FC026E479558E4475677E9AA9E3050E2765694DFC81F56E880B96E7160C980DD98EDD3DFFFFFFFFFFFFFFFFF)]
  28
  29 def P_hash(hashModule, secret, seed, length):
  30     bytes = createByteArrayZeros(length)
  31     secret = bytesToString(secret)
  32     seed = bytesToString(seed)
  33     A = seed
  34     index = 0
  35     while 1:
  36         A = hmac.HMAC(secret, A, hashModule).digest()
  37         output = hmac.HMAC(secret, A+seed, hashModule).digest()
  38         for c in output:
  39             if index >= length:
  40                 return bytes
  41             bytes[index] = ord(c)
  42             index += 1
  43     return bytes
  44
  45 def PRF(secret, label, seed, length):
  46     #Split the secret into left and right halves
  47     S1 = secret[ : int(math.ceil(len(secret)/2.0))]
  48     S2 = secret[ int(math.floor(len(secret)/2.0)) : ]
  49
  50     #Run the left half through P_MD5 and the right half through P_SHA1
  51     p_md5 = P_hash(md5, S1, concatArrays(stringToBytes(label), seed), length)
  52     p_sha1 = P_hash(sha, S2, concatArrays(stringToBytes(label), seed), length)
  53
  54     #XOR the output values and return the result
  55     for x in range(length):
  56         p_md5[x] ^= p_sha1[x]
  57     return p_md5
  58
  59
  60 def PRF_SSL(secret, seed, length):
  61     secretStr = bytesToString(secret)
  62     seedStr = bytesToString(seed)
  63     bytes = createByteArrayZeros(length)
  64     index = 0
  65     for x in range(26):
  66         A = chr(ord('A')+x) * (x+1) # 'A', 'BB', 'CCC', etc..
  67         input = secretStr + sha.sha(A + secretStr + seedStr).digest()
  68         output = md5.md5(input).digest()
  69         for c in output:
  70             if index >= length:
  71                 return bytes
  72             bytes[index] = ord(c)
  73             index += 1
  74     return bytes
  75
  76 def makeX(salt, username, password):
  77     if len(username)>=256:
  78         raise ValueError("username too long")
  79     if len(salt)>=256:
  80         raise ValueError("salt too long")
  81     return stringToNumber(sha.sha(salt + sha.sha(username + ":" + password)\
  82            .digest()).digest())
  83
  84 #This function is used by VerifierDB.makeVerifier
  85 def makeVerifier(username, password, bits):
  86     bitsIndex = {1024:0, 1536:1, 2048:2, 3072:3, 4096:4, 6144:5, 8192:6}[bits]
  87     g,N = goodGroupParameters[bitsIndex]
  88     salt = bytesToString(getRandomBytes(16))
  89     x = makeX(salt, username, password)
  90     verifier = powMod(g, x, N)
  91     return N, g, salt, verifier
  92
  93 def PAD(n, x):
  94     nLength = len(numberToString(n))
  95     s = numberToString(x)
  96     if len(s) < nLength:
  97         s = ("\0" * (nLength-len(s))) + s
  98     return s
  99
 100 def makeU(N, A, B):
 101   return stringToNumber(sha.sha(PAD(N, A) + PAD(N, B)).digest())
 102
 103 def makeK(N, g):
 104   return stringToNumber(sha.sha(numberToString(N) + PAD(N, g)).digest())
 105
 106
 107 """
 108 MAC_SSL
 109 Modified from Python HMAC by Trevor
 110 """
 111
 112 class MAC_SSL:
 113     """MAC_SSL class.
 114
 115     This supports the API for Cryptographic Hash Functions (PEP 247).
 116     """
 117
 118     def __init__(self, key, msg = None, digestmod = None):
 119         """Create a new MAC_SSL object.
 120
 121         key:       key for the keyed hash object.
 122         msg:       Initial input for the hash, if provided.
 123         digestmod: A module supporting PEP 247. Defaults to the md5 module.
 124         """
 125         if digestmod is None:
 126             import md5
 127             digestmod = md5
 128
 129         if key == None: #TREVNEW - for faster copying
 130             return      #TREVNEW
 131
 132         self.digestmod = digestmod
 133         self.outer = digestmod.new()
 134         self.inner = digestmod.new()
 135         self.digest_size = digestmod.digest_size
 136
 137         ipad = "\x36" * 40
 138         opad = "\x5C" * 40
 139
 140         self.inner.update(key)
 141         self.inner.update(ipad)
 142         self.outer.update(key)
 143         self.outer.update(opad)
 144         if msg is not None:
 145             self.update(msg)
 146
 147
 148     def update(self, msg):
 149         """Update this hashing object with the string msg.
 150         """
 151         self.inner.update(msg)
 152
 153     def copy(self):
 154         """Return a separate copy of this hashing object.
 155
 156         An update to this copy won't affect the original object.
 157         """
 158         other = MAC_SSL(None) #TREVNEW - for faster copying
 159         other.digest_size = self.digest_size #TREVNEW
 160         other.digestmod = self.digestmod
 161         other.inner = self.inner.copy()
 162         other.outer = self.outer.copy()
 163         return other
 164
 165     def digest(self):
 166         """Return the hash value of this hashing object.
 167
 168         This returns a string containing 8-bit data.  The object is
 169         not altered in any way by this function; you can continue
 170         updating the object after calling this function.
 171         """
 172         h = self.outer.copy()
 173         h.update(self.inner.digest())
 174         return h.digest()
 175
 176     def hexdigest(self):
 177         """Like digest(), but returns a string of hexadecimal digits instead.
 178         """
 179         return "".join([hex(ord(x))[2:].zfill(2)
 180                         for x in tuple(self.digest())])