ConvertStreamToPem2.py

#!/usr/bin/env python

from scan.TLSConstructs import *
import os, sys, os.path
import traceback

# like ConvertStreamToPem.py, but writes a series of concatenated PEMS to
# stdout.

out = sys.stdout

USAGE = "ConvertStreamToPem2.py <.result file> \n"
def convert(in_name):
  try:
    f = open(in_name, 'rb')
    data = f.read()
    f.close()
    
    count = 0
    recs = TLSRecord.parse(data)
    certs = []
    for rec in recs:
      if rec.ContentType == 'handshake':
        handshake_messages = Handshake.parse(rec.data)
        for hand_msg in handshake_messages:
          if hand_msg.msg_type == 'certificate':
            cert_msg = Certificate.parse(hand_msg.body)
            data_read = 0
            while data_read < cert_msg.list_length:
              cert = ASNCert.parse(cert_msg.list_data[data_read:])
              data_read += cert.cert_length + 3
              certs.append(cert)
    for certificate in certs:
      try:
        count = count + 1
        out.write('-----BEGIN CERTIFICATE-----\n')
        out.write(certificate.cert.encode('base64'))
        out.write('-----END CERTIFICATE-----\n')
      except:
        traceback.print_exc()
        print "error with certificate %d in %s" % (count, in_name)
  except:
    print "Error attempting to process", in_name

if __name__ == "__main__":
  if len(sys.argv) != 2:
    print USAGE
  else:
    convert(sys.argv[1])