blob: a9a0effeca860581b4609826e96ce767090b0ff6 [file] [log] [blame]
Jean-Paul Calderone3de9f622008-03-12 14:12:19 -04001# -*- coding: latin-1 -*-
Jean-Paul Calderone897bc252008-02-18 20:50:23 -05002#
Jean-Paul Calderonede0a71e2011-03-02 19:55:11 -05003# Copyright (C) AB Strakt
4# Copyright (C) Jean-Paul Calderone
5# See LICENSE for details.
Jean-Paul Calderone8b63d452008-03-21 18:31:12 -04006
Jean-Paul Calderone897bc252008-02-18 20:50:23 -05007"""
8Simple SSL client, using blocking I/O
9"""
10
Jim Shaverb2ff5be2015-04-30 08:26:29 -040011from OpenSSL import SSL, crypto
Jean-Paul Calderone897bc252008-02-18 20:50:23 -050012import sys, os, select, socket
13
14def verify_cb(conn, cert, errnum, depth, ok):
15 # This obviously has to be updated
Jim Shaverb2ff5be2015-04-30 08:26:29 -040016 certsubject = crypto.X509Name(cert.get_subject())
17 commonname = certsubject.commonName
18 print('Got certificate: ' + commonname)
Jean-Paul Calderone897bc252008-02-18 20:50:23 -050019 return ok
20
21if len(sys.argv) < 3:
Jim Shaver8a4a7ae2015-04-29 01:17:33 -040022 print('Usage: python client.py HOST PORT')
Jean-Paul Calderone897bc252008-02-18 20:50:23 -050023 sys.exit(1)
24
25dir = os.path.dirname(sys.argv[0])
26if dir == '':
27 dir = os.curdir
28
29# Initialize context
30ctx = SSL.Context(SSL.SSLv23_METHOD)
31ctx.set_verify(SSL.VERIFY_PEER, verify_cb) # Demand a certificate
32ctx.use_privatekey_file (os.path.join(dir, 'client.pkey'))
33ctx.use_certificate_file(os.path.join(dir, 'client.cert'))
34ctx.load_verify_locations(os.path.join(dir, 'CA.cert'))
35
36# Set up client
37sock = SSL.Connection(ctx, socket.socket(socket.AF_INET, socket.SOCK_STREAM))
38sock.connect((sys.argv[1], int(sys.argv[2])))
39
40while 1:
41 line = sys.stdin.readline()
42 if line == '':
43 break
44 try:
45 sock.send(line)
Jim Shaverb2ff5be2015-04-30 08:26:29 -040046 sys.stdout.write(sock.recv(1024).decode('utf-8'))
Jean-Paul Calderone897bc252008-02-18 20:50:23 -050047 sys.stdout.flush()
48 except SSL.Error:
Jim Shaver71ad3682015-04-29 00:09:14 -040049 print('Connection died unexpectedly')
Jean-Paul Calderone897bc252008-02-18 20:50:23 -050050 break
51
52
53sock.shutdown()
54sock.close()