1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
|
# This file is dual licensed under the terms of the Apache License, Version
# 2.0, and the BSD License. See the LICENSE file in the root of this repository
# for complete details.
from __future__ import absolute_import, division, print_function
INCLUDES = """
#include <openssl/ssl.h>
#include <openssl/x509.h>
#include <openssl/x509_vfy.h>
#include <openssl/crypto.h>
#ifdef _WIN32
#define WIN32_LEAN_AND_MEAN
#include <Windows.h>
#include <Wincrypt.h>
#include <Winsock2.h>
#else
#include <stdio.h>
#include <stdlib.h>
#include <pthread.h>
#endif
"""
TYPES = """
typedef struct {
char *password;
int length;
int called;
int error;
int maxsize;
} CRYPTOGRAPHY_PASSWORD_DATA;
"""
FUNCTIONS = """
int Cryptography_setup_ssl_threads(void);
int Cryptography_pem_password_cb(char *, int, int, void *);
"""
CUSTOMIZATIONS = """
/* This code is derived from the locking code found in the Python _ssl module's
locking callback for OpenSSL.
Copyright 2001-2016 Python Software Foundation; All Rights Reserved.
It has been subsequently modified to use cross platform locking without
using CPython APIs by Armin Rigo of the PyPy project.
*/
#if CRYPTOGRAPHY_OPENSSL_LESS_THAN_110
#ifdef _WIN32
typedef CRITICAL_SECTION Cryptography_mutex;
static __inline void cryptography_mutex_init(Cryptography_mutex *mutex) {
InitializeCriticalSection(mutex);
}
static __inline void cryptography_mutex_lock(Cryptography_mutex *mutex) {
EnterCriticalSection(mutex);
}
static __inline void cryptography_mutex_unlock(Cryptography_mutex *mutex) {
LeaveCriticalSection(mutex);
}
#else
typedef pthread_mutex_t Cryptography_mutex;
#define ASSERT_STATUS(call) \
if ((call) != 0) { \
perror("Fatal error in callback initialization: " #call); \
abort(); \
}
static inline void cryptography_mutex_init(Cryptography_mutex *mutex) {
#if !defined(pthread_mutexattr_default)
# define pthread_mutexattr_default ((pthread_mutexattr_t *)NULL)
#endif
ASSERT_STATUS(pthread_mutex_init(mutex, pthread_mutexattr_default));
}
static inline void cryptography_mutex_lock(Cryptography_mutex *mutex) {
ASSERT_STATUS(pthread_mutex_lock(mutex));
}
static inline void cryptography_mutex_unlock(Cryptography_mutex *mutex) {
ASSERT_STATUS(pthread_mutex_unlock(mutex));
}
#endif
static int _ssl_locks_count = 0;
static Cryptography_mutex *_ssl_locks = NULL;
static void _ssl_thread_locking_function(int mode, int n, const char *file,
int line) {
/* this function is needed to perform locking on shared data
structures. (Note that OpenSSL uses a number of global data
structures that will be implicitly shared whenever multiple
threads use OpenSSL.) Multi-threaded applications will
crash at random if it is not set.
locking_function() must be able to handle up to
CRYPTO_num_locks() different mutex locks. It sets the n-th
lock if mode & CRYPTO_LOCK, and releases it otherwise.
file and line are the file number of the function setting the
lock. They can be useful for debugging.
*/
if ((_ssl_locks == NULL) ||
(n < 0) || (n >= _ssl_locks_count)) {
return;
}
if (mode & CRYPTO_LOCK) {
cryptography_mutex_lock(_ssl_locks + n);
} else {
cryptography_mutex_unlock(_ssl_locks + n);
}
}
static void init_mutexes(void) {
int i;
for (i = 0; i < _ssl_locks_count; i++) {
cryptography_mutex_init(_ssl_locks + i);
}
}
int Cryptography_setup_ssl_threads(void) {
if (_ssl_locks == NULL) {
_ssl_locks_count = CRYPTO_num_locks();
_ssl_locks = calloc(_ssl_locks_count, sizeof(Cryptography_mutex));
if (_ssl_locks == NULL) {
return 0;
}
init_mutexes();
CRYPTO_set_locking_callback(_ssl_thread_locking_function);
#ifndef _WIN32
pthread_atfork(NULL, NULL, &init_mutexes);
#endif
}
return 1;
}
#else
int (*Cryptography_setup_ssl_threads)(void) = NULL;
#endif
typedef struct {
char *password;
int length;
int called;
int error;
int maxsize;
} CRYPTOGRAPHY_PASSWORD_DATA;
int Cryptography_pem_password_cb(char *buf, int size,
int rwflag, void *userdata) {
/* The password cb is only invoked if OpenSSL decides the private
key is encrypted. So this path only occurs if it needs a password */
CRYPTOGRAPHY_PASSWORD_DATA *st = (CRYPTOGRAPHY_PASSWORD_DATA *)userdata;
st->called += 1;
st->maxsize = size;
if (st->length == 0) {
st->error = -1;
return 0;
} else if (st->length < size) {
memcpy(buf, st->password, st->length);
return st->length;
} else {
st->error = -2;
return 0;
}
}
"""
|
