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): |
| 15 | # This obviously has to be updated |
Jim Shaver | b2ff5be | 2015-04-30 08:26:29 -0400 | [diff] [blame^] | 16 | certsubject = crypto.X509Name(cert.get_subject()) |
| 17 | commonname = certsubject.commonName |
| 18 | print('Got certificate: ' + commonname) |
Jean-Paul Calderone | 897bc25 | 2008-02-18 20:50:23 -0500 | [diff] [blame] | 19 | return ok |
| 20 | |
| 21 | if len(sys.argv) < 3: |
Jim Shaver | 8a4a7ae | 2015-04-29 01:17:33 -0400 | [diff] [blame] | 22 | print('Usage: python client.py HOST PORT') |
Jean-Paul Calderone | 897bc25 | 2008-02-18 20:50:23 -0500 | [diff] [blame] | 23 | sys.exit(1) |
| 24 | |
| 25 | dir = os.path.dirname(sys.argv[0]) |
| 26 | if dir == '': |
| 27 | dir = os.curdir |
| 28 | |
| 29 | # Initialize context |
| 30 | ctx = SSL.Context(SSL.SSLv23_METHOD) |
| 31 | ctx.set_verify(SSL.VERIFY_PEER, verify_cb) # Demand a certificate |
| 32 | ctx.use_privatekey_file (os.path.join(dir, 'client.pkey')) |
| 33 | ctx.use_certificate_file(os.path.join(dir, 'client.cert')) |
| 34 | ctx.load_verify_locations(os.path.join(dir, 'CA.cert')) |
| 35 | |
| 36 | # Set up client |
| 37 | sock = SSL.Connection(ctx, socket.socket(socket.AF_INET, socket.SOCK_STREAM)) |
| 38 | sock.connect((sys.argv[1], int(sys.argv[2]))) |
| 39 | |
| 40 | while 1: |
| 41 | line = sys.stdin.readline() |
| 42 | if line == '': |
| 43 | break |
| 44 | try: |
| 45 | sock.send(line) |
Jim Shaver | b2ff5be | 2015-04-30 08:26:29 -0400 | [diff] [blame^] | 46 | sys.stdout.write(sock.recv(1024).decode('utf-8')) |
Jean-Paul Calderone | 897bc25 | 2008-02-18 20:50:23 -0500 | [diff] [blame] | 47 | sys.stdout.flush() |
| 48 | except SSL.Error: |
Jim Shaver | 71ad368 | 2015-04-29 00:09:14 -0400 | [diff] [blame] | 49 | print('Connection died unexpectedly') |
Jean-Paul Calderone | 897bc25 | 2008-02-18 20:50:23 -0500 | [diff] [blame] | 50 | break |
| 51 | |
| 52 | |
| 53 | sock.shutdown() |
| 54 | sock.close() |