Jean-Paul Calderone | 3de9f62 | 2008-03-12 14:12:19 -0400 | [diff] [blame] | 1 | # -*- coding: latin-1 -*- |
Jean-Paul Calderone | 897bc25 | 2008-02-18 20:50:23 -0500 | [diff] [blame] | 2 | # |
Jean-Paul Calderone | de0a71e | 2011-03-02 19:55:11 -0500 | [diff] [blame] | 3 | # Copyright (C) AB Strakt |
| 4 | # Copyright (C) Jean-Paul Calderone |
| 5 | # See LICENSE for details. |
Jean-Paul Calderone | 8b63d45 | 2008-03-21 18:31:12 -0400 | [diff] [blame] | 6 | |
Jean-Paul Calderone | 897bc25 | 2008-02-18 20:50:23 -0500 | [diff] [blame] | 7 | """ |
| 8 | Simple SSL client, using blocking I/O |
| 9 | """ |
| 10 | |
Jim Shaver | b2ff5be | 2015-04-30 08:26:29 -0400 | [diff] [blame] | 11 | from OpenSSL import SSL, crypto |
Jean-Paul Calderone | 897bc25 | 2008-02-18 20:50:23 -0500 | [diff] [blame] | 12 | import sys, os, select, socket |
| 13 | |
| 14 | def verify_cb(conn, cert, errnum, depth, ok): |
Jim Shaver | b2ff5be | 2015-04-30 08:26:29 -0400 | [diff] [blame] | 15 | certsubject = crypto.X509Name(cert.get_subject()) |
| 16 | commonname = certsubject.commonName |
| 17 | print('Got certificate: ' + commonname) |
Jean-Paul Calderone | 897bc25 | 2008-02-18 20:50:23 -0500 | [diff] [blame] | 18 | return ok |
| 19 | |
| 20 | if len(sys.argv) < 3: |
Jim Shaver | 8a4a7ae | 2015-04-29 01:17:33 -0400 | [diff] [blame] | 21 | print('Usage: python client.py HOST PORT') |
Jean-Paul Calderone | 897bc25 | 2008-02-18 20:50:23 -0500 | [diff] [blame] | 22 | sys.exit(1) |
| 23 | |
| 24 | dir = os.path.dirname(sys.argv[0]) |
| 25 | if dir == '': |
| 26 | dir = os.curdir |
| 27 | |
| 28 | # Initialize context |
| 29 | ctx = SSL.Context(SSL.SSLv23_METHOD) |
| 30 | ctx.set_verify(SSL.VERIFY_PEER, verify_cb) # Demand a certificate |
| 31 | ctx.use_privatekey_file (os.path.join(dir, 'client.pkey')) |
| 32 | ctx.use_certificate_file(os.path.join(dir, 'client.cert')) |
| 33 | ctx.load_verify_locations(os.path.join(dir, 'CA.cert')) |
| 34 | |
| 35 | # Set up client |
| 36 | sock = SSL.Connection(ctx, socket.socket(socket.AF_INET, socket.SOCK_STREAM)) |
| 37 | sock.connect((sys.argv[1], int(sys.argv[2]))) |
| 38 | |
| 39 | while 1: |
| 40 | line = sys.stdin.readline() |
| 41 | if line == '': |
| 42 | break |
| 43 | try: |
| 44 | sock.send(line) |
Jim Shaver | b2ff5be | 2015-04-30 08:26:29 -0400 | [diff] [blame] | 45 | sys.stdout.write(sock.recv(1024).decode('utf-8')) |
Jean-Paul Calderone | 897bc25 | 2008-02-18 20:50:23 -0500 | [diff] [blame] | 46 | sys.stdout.flush() |
| 47 | except SSL.Error: |
Jim Shaver | 71ad368 | 2015-04-29 00:09:14 -0400 | [diff] [blame] | 48 | print('Connection died unexpectedly') |
Jean-Paul Calderone | 897bc25 | 2008-02-18 20:50:23 -0500 | [diff] [blame] | 49 | break |
| 50 | |
| 51 | |
| 52 | sock.shutdown() |
| 53 | sock.close() |