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