blob: d4c9fa55250d0bc33c73d9950b581e2afd7b8d38 [file] [log] [blame]
Jean-Paul Calderone5ef86512008-04-26 19:06:28 -04001# Copyright (C) Jean-Paul Calderone 2008, All rights reserved
2#
3# Stress tester for thread-related bugs in global_info_callback in
4# src/ssl/context.c. In 0.7 and earlier, this will somewhat reliably
5# segfault or abort after a few dozen to a few thousand iterations on an SMP
6# machine (generally not on a UP machine) due to uses of Python/C API
7# without holding the GIL.
8
9from itertools import count
10from threading import Thread
11from socket import socket
12
13from OpenSSL.SSL import Context, TLSv1_METHOD, Connection, WantReadError
14from OpenSSL.crypto import FILETYPE_PEM, load_certificate, load_privatekey
15
16cleartextPrivateKeyPEM = (
17 "-----BEGIN RSA PRIVATE KEY-----\n"
18 "MIICXAIBAAKBgQDaemNe1syksAbFFpF3aoOrZ18vB/IQNZrAjFqXPv9iieJm7+Tc\n"
19 "g+lA/v0qmoEKrpT2xfwxXmvZwBNM4ZhyRC3DPIFEyJV7/3IA1p5iuMY/GJI1VIgn\n"
20 "aikQCnrsyxtaRpsMBeZRniaVzcUJ+XnEdFGEjlo+k0xlwfVclDEMwgpXAQIDAQAB\n"
21 "AoGBALi0a7pMQqqgnriVAdpBVJveQtxSDVWi2/gZMKVZfzNheuSnv4amhtaKPKJ+\n"
22 "CMZtHkcazsE2IFvxRN/kgato9H3gJqq8nq2CkdpdLNVKBoxiCtkLfutdY4SQLtoY\n"
23 "USN7exk131pchsAJXYlR6mCW+ZP+E523cNwpPgsyKxVbmXSBAkEA9470fy2W0jFM\n"
24 "taZFslpntKSzbvn6JmdtjtvWrM1bBaeeqFiGBuQFYg46VaCUaeRWYw02jmYAsDYh\n"
25 "ZQavmXThaQJBAOHtlAQ0IJJEiMZr6vtVPH32fmbthSv1AUSYPzKqdlQrUnOXPQXu\n"
26 "z70cFoLG1TvPF5rBxbOkbQ/s8/ka5ZjPfdkCQCeC7YsO36+UpsWnUCBzRXITh4AC\n"
27 "7eYLQ/U1KUJTVF/GrQ/5cQrQgftwgecAxi9Qfmk4xqhbp2h4e0QAmS5I9WECQH02\n"
28 "0QwrX8nxFeTytr8pFGezj4a4KVCdb2B3CL+p3f70K7RIo9d/7b6frJI6ZL/LHQf2\n"
29 "UP4pKRDkgKsVDx7MELECQGm072/Z7vmb03h/uE95IYJOgY4nfmYs0QKA9Is18wUz\n"
30 "DpjfE33p0Ha6GO1VZRIQoqE24F8o5oimy3BEjryFuw4=\n"
31 "-----END RSA PRIVATE KEY-----\n")
32
33
34cleartextCertificatePEM = (
35 "-----BEGIN CERTIFICATE-----\n"
36 "MIICfTCCAeYCAQEwDQYJKoZIhvcNAQEEBQAwgYYxCzAJBgNVBAYTAlVTMRkwFwYD\n"
37 "VQQDExBweW9wZW5zc2wuc2YubmV0MREwDwYDVQQHEwhOZXcgWW9yazESMBAGA1UE\n"
38 "ChMJUHlPcGVuU1NMMREwDwYDVQQIEwhOZXcgWW9yazEQMA4GCSqGSIb3DQEJARYB\n"
39 "IDEQMA4GA1UECxMHVGVzdGluZzAeFw0wODAzMjUxOTA0MTNaFw0wOTAzMjUxOTA0\n"
40 "MTNaMIGGMQswCQYDVQQGEwJVUzEZMBcGA1UEAxMQcHlvcGVuc3NsLnNmLm5ldDER\n"
41 "MA8GA1UEBxMITmV3IFlvcmsxEjAQBgNVBAoTCVB5T3BlblNTTDERMA8GA1UECBMI\n"
42 "TmV3IFlvcmsxEDAOBgkqhkiG9w0BCQEWASAxEDAOBgNVBAsTB1Rlc3RpbmcwgZ8w\n"
43 "DQYJKoZIhvcNAQEBBQADgY0AMIGJAoGBANp6Y17WzKSwBsUWkXdqg6tnXy8H8hA1\n"
44 "msCMWpc+/2KJ4mbv5NyD6UD+/SqagQqulPbF/DFea9nAE0zhmHJELcM8gUTIlXv/\n"
45 "cgDWnmK4xj8YkjVUiCdqKRAKeuzLG1pGmwwF5lGeJpXNxQn5ecR0UYSOWj6TTGXB\n"
46 "9VyUMQzCClcBAgMBAAEwDQYJKoZIhvcNAQEEBQADgYEAmm0Vzvv1O91WLl2LnF2P\n"
47 "q55LJdOnJbCCXIgxLdoVmvYAz1ZJq1eGKgKWI5QLgxiSzJLEU7KK//aVfiZzoCd5\n"
48 "RipBiEEMEV4eAY317bHPwPP+4Bj9t0l8AsDLseC5vLRHgxrLEu3bn08DYx6imB5Q\n"
49 "UBj849/xpszEM7BhwKE0GiQ=\n"
50 "-----END CERTIFICATE-----\n")
51
52count = count()
53def go():
54 port = socket()
55 port.bind(('', 0))
56 port.listen(1)
57
58 called = []
59 def info(conn, where, ret):
60 print count.next()
61 called.append(None)
62 context = Context(TLSv1_METHOD)
63 context.set_info_callback(info)
64 context.use_certificate(
65 load_certificate(FILETYPE_PEM, cleartextCertificatePEM))
66 context.use_privatekey(
67 load_privatekey(FILETYPE_PEM, cleartextPrivateKeyPEM))
68
69 while 1:
70 client = socket()
71 client.setblocking(False)
72 client.connect_ex(port.getsockname())
73
74 clientSSL = Connection(Context(TLSv1_METHOD), client)
75 clientSSL.set_connect_state()
76
77 server, ignored = port.accept()
78 server.setblocking(False)
79
80 serverSSL = Connection(context, server)
81 serverSSL.set_accept_state()
82
83 del called[:]
84 while not called:
85 for ssl in clientSSL, serverSSL:
86 try:
87 ssl.do_handshake()
88 except WantReadError:
89 pass
90
91
92threads = [Thread(target=go, args=()) for i in xrange(2)]
93for th in threads:
94 th.start()
95for th in threads:
96 th.join()