OLD | NEW |
| 1 # Authors: |
| 2 # Trevor Perrin |
| 3 # Dave Baggett (Arcode Corporation) - MD5 support for MAC_SSL |
| 4 # |
| 5 # See the LICENSE file for legal information regarding use of this file. |
| 6 |
1 """Miscellaneous helper functions.""" | 7 """Miscellaneous helper functions.""" |
2 | 8 |
3 from utils.compat import * | 9 from .utils.compat import * |
4 from utils.cryptomath import * | 10 from .utils.cryptomath import * |
5 | 11 |
6 import hmac | 12 import hmac |
7 | 13 |
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] | 14 #1024, 1536, 2048, 3072, 4096, 6144, and 8192 bit groups] |
21 goodGroupParameters = [(2,0xEEAF0AB9ADB38DD69C33F80AFA8FC5E86072618775FF3C0B9EA2
314C9C256576D674DF7496EA81D3383B4813D692C6E0E0D5D8E250B98BE48E495C1D6089DAD15DC7
D7B46154D6B6CE8EF4AD69B15D4982559B297BCF1885C529F566660E57EC68EDBC3C05726CC02FD4
CBF4976EAA9AFD5138FE8376435B9FC61D2FC0EB06E3),\ | 15 goodGroupParameters = [(2,0xEEAF0AB9ADB38DD69C33F80AFA8FC5E86072618775FF3C0B9EA2
314C9C256576D674DF7496EA81D3383B4813D692C6E0E0D5D8E250B98BE48E495C1D6089DAD15DC7
D7B46154D6B6CE8EF4AD69B15D4982559B297BCF1885C529F566660E57EC68EDBC3C05726CC02FD4
CBF4976EAA9AFD5138FE8376435B9FC61D2FC0EB06E3),\ |
22 (2,0x9DEF3CAFB939277AB1F12A8617A47BBBDBA51DF499AC4C80BEEE
A9614B19CC4D5F4F5F556E27CBDE51C6A94BE4607A291558903BA0D0F84380B655BB9A22E8DCDF02
8A7CEC67F0D08134B1C8B97989149B609E0BE3BAB63D47548381DBC5B1FC764E3F4B53DD9DA1158B
FD3E2B9C8CF56EDF019539349627DB2FD53D24B7C48665772E437D6C7F8CE442734AF7CCB7AE837C
264AE3A9BEB87F8A2FE9B8B5292E5A021FFF5E91479E8CE7A28C2442C6F315180F93499A234DCF76
E3FED135F9BB),\ | 16 (2,0x9DEF3CAFB939277AB1F12A8617A47BBBDBA51DF499AC4C80BEEE
A9614B19CC4D5F4F5F556E27CBDE51C6A94BE4607A291558903BA0D0F84380B655BB9A22E8DCDF02
8A7CEC67F0D08134B1C8B97989149B609E0BE3BAB63D47548381DBC5B1FC764E3F4B53DD9DA1158B
FD3E2B9C8CF56EDF019539349627DB2FD53D24B7C48665772E437D6C7F8CE442734AF7CCB7AE837C
264AE3A9BEB87F8A2FE9B8B5292E5A021FFF5E91479E8CE7A28C2442C6F315180F93499A234DCF76
E3FED135F9BB),\ |
23 (2,0xAC6BDB41324A9A9BF166DE5E1389582FAF72B6651987EE07FC31
92943DB56050A37329CBB4A099ED8193E0757767A13DD52312AB4B03310DCD7F48A9DA04FD50E808
3969EDB767B0CF6095179A163AB3661A05FBD5FAAAE82918A9962F0B93B855F97993EC975EEAA80D
740ADBF4FF747359D041D5C33EA71D281E446B14773BCA97B43A23FB801676BD207A436C6481F1D2
B9078717461A5B9D32E688F87748544523B524B0D57D5EA77A2775D2ECFA032CFBDBF52FB3786160
279004E57AE6AF874E7303CE53299CCC041C7BC308D82A5698F3A8D0C38271AE35F8E9DBFBB694B5
C803D89F7AE435DE236D525F54759B65E372FCD68EF20FA7111F9E4AFF73),\ | 17 (2,0xAC6BDB41324A9A9BF166DE5E1389582FAF72B6651987EE07FC31
92943DB56050A37329CBB4A099ED8193E0757767A13DD52312AB4B03310DCD7F48A9DA04FD50E808
3969EDB767B0CF6095179A163AB3661A05FBD5FAAAE82918A9962F0B93B855F97993EC975EEAA80D
740ADBF4FF747359D041D5C33EA71D281E446B14773BCA97B43A23FB801676BD207A436C6481F1D2
B9078717461A5B9D32E688F87748544523B524B0D57D5EA77A2775D2ECFA032CFBDBF52FB3786160
279004E57AE6AF874E7303CE53299CCC041C7BC308D82A5698F3A8D0C38271AE35F8E9DBFBB694B5
C803D89F7AE435DE236D525F54759B65E372FCD68EF20FA7111F9E4AFF73),\ |
24 (2,0xFFFFFFFFFFFFFFFFC90FDAA22168C234C4C6628B80DC1CD12902
4E088A67CC74020BBEA63B139B22514A08798E3404DDEF9519B3CD3A431B302B0A6DF25F14374FE1
356D6D51C245E485B576625E7EC6F44C42E9A637ED6B0BFF5CB6F406B7EDEE386BFB5A899FA5AE9F
24117C4B1FE649286651ECE45B3DC2007CB8A163BF0598DA48361C55D39A69163FA8FD24CF5F8365
5D23DCA3AD961C62F356208552BB9ED529077096966D670C354E4ABC9804F1746C08CA18217C3290
5E462E36CE3BE39E772C180E86039B2783A2EC07A28FB5C55DF06F4C52C9DE2BCBF6955817183995
497CEA956AE515D2261898FA051015728E5A8AAAC42DAD33170D04507A33A85521ABDF1CBA64ECFB
850458DBEF0A8AEA71575D060C7DB3970F85A6E1E4C7ABF5AE8CDB0933D71E8C94E04A25619DCEE3
D2261AD2EE6BF12FFA06D98A0864D87602733EC86A64521F2B18177B200CBBE117577A615D6C7709
88C0BAD946E208E24FA074E5AB3143DB5BFCE0FD108E4B82D120A93AD2CAFFFFFFFFFFFFFFFF),\ | 18 (2,0xFFFFFFFFFFFFFFFFC90FDAA22168C234C4C6628B80DC1CD12902
4E088A67CC74020BBEA63B139B22514A08798E3404DDEF9519B3CD3A431B302B0A6DF25F14374FE1
356D6D51C245E485B576625E7EC6F44C42E9A637ED6B0BFF5CB6F406B7EDEE386BFB5A899FA5AE9F
24117C4B1FE649286651ECE45B3DC2007CB8A163BF0598DA48361C55D39A69163FA8FD24CF5F8365
5D23DCA3AD961C62F356208552BB9ED529077096966D670C354E4ABC9804F1746C08CA18217C3290
5E462E36CE3BE39E772C180E86039B2783A2EC07A28FB5C55DF06F4C52C9DE2BCBF6955817183995
497CEA956AE515D2261898FA051015728E5A8AAAC42DAD33170D04507A33A85521ABDF1CBA64ECFB
850458DBEF0A8AEA71575D060C7DB3970F85A6E1E4C7ABF5AE8CDB0933D71E8C94E04A25619DCEE3
D2261AD2EE6BF12FFA06D98A0864D87602733EC86A64521F2B18177B200CBBE117577A615D6C7709
88C0BAD946E208E24FA074E5AB3143DB5BFCE0FD108E4B82D120A93AD2CAFFFFFFFFFFFFFFFF),\ |
25 (5,0xFFFFFFFFFFFFFFFFC90FDAA22168C234C4C6628B80DC1CD12902
4E088A67CC74020BBEA63B139B22514A08798E3404DDEF9519B3CD3A431B302B0A6DF25F14374FE1
356D6D51C245E485B576625E7EC6F44C42E9A637ED6B0BFF5CB6F406B7EDEE386BFB5A899FA5AE9F
24117C4B1FE649286651ECE45B3DC2007CB8A163BF0598DA48361C55D39A69163FA8FD24CF5F8365
5D23DCA3AD961C62F356208552BB9ED529077096966D670C354E4ABC9804F1746C08CA18217C3290
5E462E36CE3BE39E772C180E86039B2783A2EC07A28FB5C55DF06F4C52C9DE2BCBF6955817183995
497CEA956AE515D2261898FA051015728E5A8AAAC42DAD33170D04507A33A85521ABDF1CBA64ECFB
850458DBEF0A8AEA71575D060C7DB3970F85A6E1E4C7ABF5AE8CDB0933D71E8C94E04A25619DCEE3
D2261AD2EE6BF12FFA06D98A0864D87602733EC86A64521F2B18177B200CBBE117577A615D6C7709
88C0BAD946E208E24FA074E5AB3143DB5BFCE0FD108E4B82D120A92108011A723C12A787E6D78871
9A10BDBA5B2699C327186AF4E23C1A946834B6150BDA2583E9CA2AD44CE8DBBBC2DB04DE8EF92E8E
FC141FBECAA6287C59474E6BC05D99B2964FA090C3A2233BA186515BE7ED1F612970CEE2D7AFB81B
DD762170481CD0069127D5B05AA993B4EA988D8FDDC186FFB7DC90A6C08F4DF435C934063199FFFF
FFFFFFFFFFFF),\ | 19 (5,0xFFFFFFFFFFFFFFFFC90FDAA22168C234C4C6628B80DC1CD12902
4E088A67CC74020BBEA63B139B22514A08798E3404DDEF9519B3CD3A431B302B0A6DF25F14374FE1
356D6D51C245E485B576625E7EC6F44C42E9A637ED6B0BFF5CB6F406B7EDEE386BFB5A899FA5AE9F
24117C4B1FE649286651ECE45B3DC2007CB8A163BF0598DA48361C55D39A69163FA8FD24CF5F8365
5D23DCA3AD961C62F356208552BB9ED529077096966D670C354E4ABC9804F1746C08CA18217C3290
5E462E36CE3BE39E772C180E86039B2783A2EC07A28FB5C55DF06F4C52C9DE2BCBF6955817183995
497CEA956AE515D2261898FA051015728E5A8AAAC42DAD33170D04507A33A85521ABDF1CBA64ECFB
850458DBEF0A8AEA71575D060C7DB3970F85A6E1E4C7ABF5AE8CDB0933D71E8C94E04A25619DCEE3
D2261AD2EE6BF12FFA06D98A0864D87602733EC86A64521F2B18177B200CBBE117577A615D6C7709
88C0BAD946E208E24FA074E5AB3143DB5BFCE0FD108E4B82D120A92108011A723C12A787E6D78871
9A10BDBA5B2699C327186AF4E23C1A946834B6150BDA2583E9CA2AD44CE8DBBBC2DB04DE8EF92E8E
FC141FBECAA6287C59474E6BC05D99B2964FA090C3A2233BA186515BE7ED1F612970CEE2D7AFB81B
DD762170481CD0069127D5B05AA993B4EA988D8FDDC186FFB7DC90A6C08F4DF435C934063199FFFF
FFFFFFFFFFFF),\ |
26 (5,0xFFFFFFFFFFFFFFFFC90FDAA22168C234C4C6628B80DC1CD12902
4E088A67CC74020BBEA63B139B22514A08798E3404DDEF9519B3CD3A431B302B0A6DF25F14374FE1
356D6D51C245E485B576625E7EC6F44C42E9A637ED6B0BFF5CB6F406B7EDEE386BFB5A899FA5AE9F
24117C4B1FE649286651ECE45B3DC2007CB8A163BF0598DA48361C55D39A69163FA8FD24CF5F8365
5D23DCA3AD961C62F356208552BB9ED529077096966D670C354E4ABC9804F1746C08CA18217C3290
5E462E36CE3BE39E772C180E86039B2783A2EC07A28FB5C55DF06F4C52C9DE2BCBF6955817183995
497CEA956AE515D2261898FA051015728E5A8AAAC42DAD33170D04507A33A85521ABDF1CBA64ECFB
850458DBEF0A8AEA71575D060C7DB3970F85A6E1E4C7ABF5AE8CDB0933D71E8C94E04A25619DCEE3
D2261AD2EE6BF12FFA06D98A0864D87602733EC86A64521F2B18177B200CBBE117577A615D6C7709
88C0BAD946E208E24FA074E5AB3143DB5BFCE0FD108E4B82D120A92108011A723C12A787E6D78871
9A10BDBA5B2699C327186AF4E23C1A946834B6150BDA2583E9CA2AD44CE8DBBBC2DB04DE8EF92E8E
FC141FBECAA6287C59474E6BC05D99B2964FA090C3A2233BA186515BE7ED1F612970CEE2D7AFB81B
DD762170481CD0069127D5B05AA993B4EA988D8FDDC186FFB7DC90A6C08F4DF435C93402849236C3
FAB4D27C7026C1D4DCB2602646DEC9751E763DBA37BDF8FF9406AD9E530EE5DB382F413001AEB06A
53ED9027D831179727B0865A8918DA3EDBEBCF9B14ED44CE6CBACED4BB1BDB7F1447E6CC254B3320
51512BD7AF426FB8F401378CD2BF5983CA01C64B92ECF032EA15D1721D03F482D7CE6E74FEF6D55E
702F46980C82B5A84031900B1C9E59E7C97FBEC7E8F323A97A7E36CC88BE0F1D45B7FF585AC54BD4
07B22B4154AACC8F6D7EBF48E1D814CC5ED20F8037E0A79715EEF29BE32806A1D58BB7C5DA76F550
AA3D8A1FBFF0EB19CCB1A313D55CDA56C9EC2EF29632387FE8D76E3C0468043E8F663F4860EE12BF
2D5B0B7474D6E694F91E6DCC4024FFFFFFFFFFFFFFFF),\ | 20 (5,0xFFFFFFFFFFFFFFFFC90FDAA22168C234C4C6628B80DC1CD12902
4E088A67CC74020BBEA63B139B22514A08798E3404DDEF9519B3CD3A431B302B0A6DF25F14374FE1
356D6D51C245E485B576625E7EC6F44C42E9A637ED6B0BFF5CB6F406B7EDEE386BFB5A899FA5AE9F
24117C4B1FE649286651ECE45B3DC2007CB8A163BF0598DA48361C55D39A69163FA8FD24CF5F8365
5D23DCA3AD961C62F356208552BB9ED529077096966D670C354E4ABC9804F1746C08CA18217C3290
5E462E36CE3BE39E772C180E86039B2783A2EC07A28FB5C55DF06F4C52C9DE2BCBF6955817183995
497CEA956AE515D2261898FA051015728E5A8AAAC42DAD33170D04507A33A85521ABDF1CBA64ECFB
850458DBEF0A8AEA71575D060C7DB3970F85A6E1E4C7ABF5AE8CDB0933D71E8C94E04A25619DCEE3
D2261AD2EE6BF12FFA06D98A0864D87602733EC86A64521F2B18177B200CBBE117577A615D6C7709
88C0BAD946E208E24FA074E5AB3143DB5BFCE0FD108E4B82D120A92108011A723C12A787E6D78871
9A10BDBA5B2699C327186AF4E23C1A946834B6150BDA2583E9CA2AD44CE8DBBBC2DB04DE8EF92E8E
FC141FBECAA6287C59474E6BC05D99B2964FA090C3A2233BA186515BE7ED1F612970CEE2D7AFB81B
DD762170481CD0069127D5B05AA993B4EA988D8FDDC186FFB7DC90A6C08F4DF435C93402849236C3
FAB4D27C7026C1D4DCB2602646DEC9751E763DBA37BDF8FF9406AD9E530EE5DB382F413001AEB06A
53ED9027D831179727B0865A8918DA3EDBEBCF9B14ED44CE6CBACED4BB1BDB7F1447E6CC254B3320
51512BD7AF426FB8F401378CD2BF5983CA01C64B92ECF032EA15D1721D03F482D7CE6E74FEF6D55E
702F46980C82B5A84031900B1C9E59E7C97FBEC7E8F323A97A7E36CC88BE0F1D45B7FF585AC54BD4
07B22B4154AACC8F6D7EBF48E1D814CC5ED20F8037E0A79715EEF29BE32806A1D58BB7C5DA76F550
AA3D8A1FBFF0EB19CCB1A313D55CDA56C9EC2EF29632387FE8D76E3C0468043E8F663F4860EE12BF
2D5B0B7474D6E694F91E6DCC4024FFFFFFFFFFFFFFFF),\ |
27 (5,0xFFFFFFFFFFFFFFFFC90FDAA22168C234C4C6628B80DC1CD12902
4E088A67CC74020BBEA63B139B22514A08798E3404DDEF9519B3CD3A431B302B0A6DF25F14374FE1
356D6D51C245E485B576625E7EC6F44C42E9A637ED6B0BFF5CB6F406B7EDEE386BFB5A899FA5AE9F
24117C4B1FE649286651ECE45B3DC2007CB8A163BF0598DA48361C55D39A69163FA8FD24CF5F8365
5D23DCA3AD961C62F356208552BB9ED529077096966D670C354E4ABC9804F1746C08CA18217C3290
5E462E36CE3BE39E772C180E86039B2783A2EC07A28FB5C55DF06F4C52C9DE2BCBF6955817183995
497CEA956AE515D2261898FA051015728E5A8AAAC42DAD33170D04507A33A85521ABDF1CBA64ECFB
850458DBEF0A8AEA71575D060C7DB3970F85A6E1E4C7ABF5AE8CDB0933D71E8C94E04A25619DCEE3
D2261AD2EE6BF12FFA06D98A0864D87602733EC86A64521F2B18177B200CBBE117577A615D6C7709
88C0BAD946E208E24FA074E5AB3143DB5BFCE0FD108E4B82D120A92108011A723C12A787E6D78871
9A10BDBA5B2699C327186AF4E23C1A946834B6150BDA2583E9CA2AD44CE8DBBBC2DB04DE8EF92E8E
FC141FBECAA6287C59474E6BC05D99B2964FA090C3A2233BA186515BE7ED1F612970CEE2D7AFB81B
DD762170481CD0069127D5B05AA993B4EA988D8FDDC186FFB7DC90A6C08F4DF435C93402849236C3
FAB4D27C7026C1D4DCB2602646DEC9751E763DBA37BDF8FF9406AD9E530EE5DB382F413001AEB06A
53ED9027D831179727B0865A8918DA3EDBEBCF9B14ED44CE6CBACED4BB1BDB7F1447E6CC254B3320
51512BD7AF426FB8F401378CD2BF5983CA01C64B92ECF032EA15D1721D03F482D7CE6E74FEF6D55E
702F46980C82B5A84031900B1C9E59E7C97FBEC7E8F323A97A7E36CC88BE0F1D45B7FF585AC54BD4
07B22B4154AACC8F6D7EBF48E1D814CC5ED20F8037E0A79715EEF29BE32806A1D58BB7C5DA76F550
AA3D8A1FBFF0EB19CCB1A313D55CDA56C9EC2EF29632387FE8D76E3C0468043E8F663F4860EE12BF
2D5B0B7474D6E694F91E6DBE115974A3926F12FEE5E438777CB6A932DF8CD8BEC4D073B931BA3BC8
32B68D9DD300741FA7BF8AFC47ED2576F6936BA424663AAB639C5AE4F5683423B4742BF1C978238F
16CBE39D652DE3FDB8BEFC848AD922222E04A4037C0713EB57A81A23F0C73473FC646CEA306B4BCB
C8862F8385DDFA9D4B7FA2C087E879683303ED5BDD3A062B3CF5B3A278A66D2A13F83F44F82DDF31
0EE074AB6A364597E899A0255DC164F31CC50846851DF9AB48195DED7EA1B1D510BD7EE74D73FAF3
6BC31ECFA268359046F4EB879F924009438B481C6CD7889A002ED5EE382BC9190DA6FC026E479558
E4475677E9AA9E3050E2765694DFC81F56E880B96E7160C980DD98EDD3DFFFFFFFFFFFFFFFFF)] | 21 (5,0xFFFFFFFFFFFFFFFFC90FDAA22168C234C4C6628B80DC1CD12902
4E088A67CC74020BBEA63B139B22514A08798E3404DDEF9519B3CD3A431B302B0A6DF25F14374FE1
356D6D51C245E485B576625E7EC6F44C42E9A637ED6B0BFF5CB6F406B7EDEE386BFB5A899FA5AE9F
24117C4B1FE649286651ECE45B3DC2007CB8A163BF0598DA48361C55D39A69163FA8FD24CF5F8365
5D23DCA3AD961C62F356208552BB9ED529077096966D670C354E4ABC9804F1746C08CA18217C3290
5E462E36CE3BE39E772C180E86039B2783A2EC07A28FB5C55DF06F4C52C9DE2BCBF6955817183995
497CEA956AE515D2261898FA051015728E5A8AAAC42DAD33170D04507A33A85521ABDF1CBA64ECFB
850458DBEF0A8AEA71575D060C7DB3970F85A6E1E4C7ABF5AE8CDB0933D71E8C94E04A25619DCEE3
D2261AD2EE6BF12FFA06D98A0864D87602733EC86A64521F2B18177B200CBBE117577A615D6C7709
88C0BAD946E208E24FA074E5AB3143DB5BFCE0FD108E4B82D120A92108011A723C12A787E6D78871
9A10BDBA5B2699C327186AF4E23C1A946834B6150BDA2583E9CA2AD44CE8DBBBC2DB04DE8EF92E8E
FC141FBECAA6287C59474E6BC05D99B2964FA090C3A2233BA186515BE7ED1F612970CEE2D7AFB81B
DD762170481CD0069127D5B05AA993B4EA988D8FDDC186FFB7DC90A6C08F4DF435C93402849236C3
FAB4D27C7026C1D4DCB2602646DEC9751E763DBA37BDF8FF9406AD9E530EE5DB382F413001AEB06A
53ED9027D831179727B0865A8918DA3EDBEBCF9B14ED44CE6CBACED4BB1BDB7F1447E6CC254B3320
51512BD7AF426FB8F401378CD2BF5983CA01C64B92ECF032EA15D1721D03F482D7CE6E74FEF6D55E
702F46980C82B5A84031900B1C9E59E7C97FBEC7E8F323A97A7E36CC88BE0F1D45B7FF585AC54BD4
07B22B4154AACC8F6D7EBF48E1D814CC5ED20F8037E0A79715EEF29BE32806A1D58BB7C5DA76F550
AA3D8A1FBFF0EB19CCB1A313D55CDA56C9EC2EF29632387FE8D76E3C0468043E8F663F4860EE12BF
2D5B0B7474D6E694F91E6DBE115974A3926F12FEE5E438777CB6A932DF8CD8BEC4D073B931BA3BC8
32B68D9DD300741FA7BF8AFC47ED2576F6936BA424663AAB639C5AE4F5683423B4742BF1C978238F
16CBE39D652DE3FDB8BEFC848AD922222E04A4037C0713EB57A81A23F0C73473FC646CEA306B4BCB
C8862F8385DDFA9D4B7FA2C087E879683303ED5BDD3A062B3CF5B3A278A66D2A13F83F44F82DDF31
0EE074AB6A364597E899A0255DC164F31CC50846851DF9AB48195DED7EA1B1D510BD7EE74D73FAF3
6BC31ECFA268359046F4EB879F924009438B481C6CD7889A002ED5EE382BC9190DA6FC026E479558
E4475677E9AA9E3050E2765694DFC81F56E880B96E7160C980DD98EDD3DFFFFFFFFFFFFFFFFF)] |
28 | 22 |
29 def P_hash(hashModule, secret, seed, length): | 23 def P_hash(macFunc, secret, seed, length): |
30 bytes = createByteArrayZeros(length) | 24 bytes = bytearray(length) |
31 secret = bytesToString(secret) | |
32 seed = bytesToString(seed) | |
33 A = seed | 25 A = seed |
34 index = 0 | 26 index = 0 |
35 while 1: | 27 while 1: |
36 A = hmac.HMAC(secret, A, hashModule).digest() | 28 A = macFunc(secret, A) |
37 output = hmac.HMAC(secret, A+seed, hashModule).digest() | 29 output = macFunc(secret, A + seed) |
38 for c in output: | 30 for c in output: |
39 if index >= length: | 31 if index >= length: |
40 return bytes | 32 return bytes |
41 bytes[index] = ord(c) | 33 bytes[index] = c |
42 index += 1 | 34 index += 1 |
43 return bytes | 35 return bytes |
44 | 36 |
45 def PRF(secret, label, seed, length): | 37 def PRF(secret, label, seed, length): |
46 #Split the secret into left and right halves | 38 #Split the secret into left and right halves |
| 39 # which may share a byte if len is odd |
47 S1 = secret[ : int(math.ceil(len(secret)/2.0))] | 40 S1 = secret[ : int(math.ceil(len(secret)/2.0))] |
48 S2 = secret[ int(math.floor(len(secret)/2.0)) : ] | 41 S2 = secret[ int(math.floor(len(secret)/2.0)) : ] |
49 | 42 |
50 #Run the left half through P_MD5 and the right half through P_SHA1 | 43 #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) | 44 p_md5 = P_hash(HMAC_MD5, S1, label + seed, length) |
52 p_sha1 = P_hash(sha, S2, concatArrays(stringToBytes(label), seed), length) | 45 p_sha1 = P_hash(HMAC_SHA1, S2, label + seed, length) |
53 | 46 |
54 #XOR the output values and return the result | 47 #XOR the output values and return the result |
55 for x in range(length): | 48 for x in range(length): |
56 p_md5[x] ^= p_sha1[x] | 49 p_md5[x] ^= p_sha1[x] |
57 return p_md5 | 50 return p_md5 |
58 | 51 |
59 | 52 |
60 def PRF_SSL(secret, seed, length): | 53 def PRF_SSL(secret, seed, length): |
61 secretStr = bytesToString(secret) | 54 bytes = bytearray(length) |
62 seedStr = bytesToString(seed) | |
63 bytes = createByteArrayZeros(length) | |
64 index = 0 | 55 index = 0 |
65 for x in range(26): | 56 for x in range(26): |
66 A = chr(ord('A')+x) * (x+1) # 'A', 'BB', 'CCC', etc.. | 57 A = bytearray([ord('A')+x] * (x+1)) # 'A', 'BB', 'CCC', etc.. |
67 input = secretStr + sha.sha(A + secretStr + seedStr).digest() | 58 input = secret + SHA1(A + secret + seed) |
68 output = md5.md5(input).digest() | 59 output = MD5(input) |
69 for c in output: | 60 for c in output: |
70 if index >= length: | 61 if index >= length: |
71 return bytes | 62 return bytes |
72 bytes[index] = ord(c) | 63 bytes[index] = c |
73 index += 1 | 64 index += 1 |
74 return bytes | 65 return bytes |
75 | 66 |
| 67 def calcMasterSecret(version, premasterSecret, clientRandom, serverRandom): |
| 68 if version == (3,0): |
| 69 masterSecret = PRF_SSL(premasterSecret, |
| 70 clientRandom + serverRandom, 48) |
| 71 elif version in ((3,1), (3,2)): |
| 72 masterSecret = PRF(premasterSecret, b"master secret", |
| 73 clientRandom + serverRandom, 48) |
| 74 else: |
| 75 raise AssertionError() |
| 76 return masterSecret |
| 77 |
| 78 |
76 def makeX(salt, username, password): | 79 def makeX(salt, username, password): |
77 if len(username)>=256: | 80 if len(username)>=256: |
78 raise ValueError("username too long") | 81 raise ValueError("username too long") |
79 if len(salt)>=256: | 82 if len(salt)>=256: |
80 raise ValueError("salt too long") | 83 raise ValueError("salt too long") |
81 return stringToNumber(sha.sha(salt + sha.sha(username + ":" + password)\ | 84 innerHashResult = SHA1(username + bytearray(b":") + password) |
82 .digest()).digest()) | 85 outerHashResult = SHA1(salt + innerHashResult) |
| 86 return bytesToNumber(outerHashResult) |
83 | 87 |
84 #This function is used by VerifierDB.makeVerifier | 88 #This function is used by VerifierDB.makeVerifier |
85 def makeVerifier(username, password, bits): | 89 def makeVerifier(username, password, bits): |
86 bitsIndex = {1024:0, 1536:1, 2048:2, 3072:3, 4096:4, 6144:5, 8192:6}[bits] | 90 bitsIndex = {1024:0, 1536:1, 2048:2, 3072:3, 4096:4, 6144:5, 8192:6}[bits] |
87 g,N = goodGroupParameters[bitsIndex] | 91 g,N = goodGroupParameters[bitsIndex] |
88 salt = bytesToString(getRandomBytes(16)) | 92 salt = getRandomBytes(16) |
89 x = makeX(salt, username, password) | 93 x = makeX(salt, username, password) |
90 verifier = powMod(g, x, N) | 94 verifier = powMod(g, x, N) |
91 return N, g, salt, verifier | 95 return N, g, salt, verifier |
92 | 96 |
93 def PAD(n, x): | 97 def PAD(n, x): |
94 nLength = len(numberToString(n)) | 98 nLength = len(numberToByteArray(n)) |
95 s = numberToString(x) | 99 b = numberToByteArray(x) |
96 if len(s) < nLength: | 100 if len(b) < nLength: |
97 s = ("\0" * (nLength-len(s))) + s | 101 b = (b"\0" * (nLength-len(b))) + b |
98 return s | 102 return b |
99 | 103 |
100 def makeU(N, A, B): | 104 def makeU(N, A, B): |
101 return stringToNumber(sha.sha(PAD(N, A) + PAD(N, B)).digest()) | 105 return bytesToNumber(SHA1(PAD(N, A) + PAD(N, B))) |
102 | 106 |
103 def makeK(N, g): | 107 def makeK(N, g): |
104 return stringToNumber(sha.sha(numberToString(N) + PAD(N, g)).digest()) | 108 return bytesToNumber(SHA1(numberToByteArray(N) + PAD(N, g))) |
| 109 |
| 110 def createHMAC(k, digestmod=hashlib.sha1): |
| 111 return hmac.HMAC(k, digestmod=digestmod) |
| 112 |
| 113 def createMAC_SSL(k, digestmod=None): |
| 114 mac = MAC_SSL() |
| 115 mac.create(k, digestmod=digestmod) |
| 116 return mac |
105 | 117 |
106 | 118 |
107 """ | 119 class MAC_SSL(object): |
108 MAC_SSL | 120 def create(self, k, digestmod=None): |
109 Modified from Python HMAC by Trevor | 121 self.digestmod = digestmod or hashlib.sha1 |
110 """ | 122 # Repeat pad bytes 48 times for MD5; 40 times for other hash functions. |
| 123 self.digest_size = 16 if (self.digestmod is hashlib.md5) else 20 |
| 124 repeat = 40 if self.digest_size == 20 else 48 |
| 125 opad = b"\x5C" * repeat |
| 126 ipad = b"\x36" * repeat |
111 | 127 |
112 class MAC_SSL: | 128 self.ohash = self.digestmod(k + opad) |
113 """MAC_SSL class. | 129 self.ihash = self.digestmod(k + ipad) |
114 | 130 |
115 This supports the API for Cryptographic Hash Functions (PEP 247). | 131 def update(self, m): |
116 """ | 132 self.ihash.update(m) |
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 | 133 |
153 def copy(self): | 134 def copy(self): |
154 """Return a separate copy of this hashing object. | 135 new = MAC_SSL() |
155 | 136 new.ihash = self.ihash.copy() |
156 An update to this copy won't affect the original object. | 137 new.ohash = self.ohash.copy() |
157 """ | 138 new.digestmod = self.digestmod |
158 other = MAC_SSL(None) #TREVNEW - for faster copying | 139 new.digest_size = self.digest_size |
159 other.digest_size = self.digest_size #TREVNEW | 140 return new |
160 other.digestmod = self.digestmod | |
161 other.inner = self.inner.copy() | |
162 other.outer = self.outer.copy() | |
163 return other | |
164 | 141 |
165 def digest(self): | 142 def digest(self): |
166 """Return the hash value of this hashing object. | 143 ohash2 = self.ohash.copy() |
167 | 144 ohash2.update(self.ihash.digest()) |
168 This returns a string containing 8-bit data. The object is | 145 return bytearray(ohash2.digest()) |
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())]) | |
OLD | NEW |