[b7d3cc34] | 1 | /********************************************************************\ |
---|
| 2 | * BitlBee -- An IRC to other IM-networks gateway * |
---|
| 3 | * * |
---|
| 4 | * Copyright 2002-2004 Wilmer van der Gaast and others * |
---|
| 5 | \********************************************************************/ |
---|
| 6 | |
---|
[26fdfc5] | 7 | /* SSL module - OpenSSL version */ |
---|
[b7d3cc34] | 8 | |
---|
| 9 | /* |
---|
| 10 | This program is free software; you can redistribute it and/or modify |
---|
| 11 | it under the terms of the GNU General Public License as published by |
---|
| 12 | the Free Software Foundation; either version 2 of the License, or |
---|
| 13 | (at your option) any later version. |
---|
| 14 | |
---|
| 15 | This program is distributed in the hope that it will be useful, |
---|
| 16 | but WITHOUT ANY WARRANTY; without even the implied warranty of |
---|
| 17 | MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
---|
| 18 | GNU General Public License for more details. |
---|
| 19 | |
---|
| 20 | You should have received a copy of the GNU General Public License with |
---|
| 21 | the Debian GNU/Linux distribution in /usr/share/common-licenses/GPL; |
---|
| 22 | if not, write to the Free Software Foundation, Inc., 59 Temple Place, |
---|
| 23 | Suite 330, Boston, MA 02111-1307 USA |
---|
| 24 | */ |
---|
| 25 | |
---|
| 26 | #include <openssl/crypto.h> |
---|
| 27 | #include <openssl/rand.h> |
---|
| 28 | #include <openssl/x509.h> |
---|
| 29 | #include <openssl/pem.h> |
---|
| 30 | #include <openssl/ssl.h> |
---|
| 31 | #include <openssl/err.h> |
---|
| 32 | |
---|
| 33 | #include "proxy.h" |
---|
| 34 | #include "ssl_client.h" |
---|
| 35 | #include "sock.h" |
---|
| 36 | |
---|
[701acdd4] | 37 | int ssl_errno = 0; |
---|
| 38 | |
---|
[b7d3cc34] | 39 | static gboolean initialized = FALSE; |
---|
| 40 | |
---|
| 41 | struct scd |
---|
| 42 | { |
---|
[26fdfc5] | 43 | ssl_input_function func; |
---|
[b7d3cc34] | 44 | gpointer data; |
---|
| 45 | int fd; |
---|
| 46 | gboolean established; |
---|
| 47 | |
---|
[26fdfc5] | 48 | int inpa; |
---|
| 49 | int lasterr; /* Necessary for SSL_get_error */ |
---|
[b7d3cc34] | 50 | SSL *ssl; |
---|
| 51 | SSL_CTX *ssl_ctx; |
---|
| 52 | }; |
---|
| 53 | |
---|
[309cb9e] | 54 | static gboolean ssl_connected( gpointer data, gint source, b_input_condition cond ); |
---|
[f920d9e] | 55 | static gboolean ssl_starttls_real( gpointer data, gint source, b_input_condition cond ); |
---|
| 56 | static gboolean ssl_handshake( gpointer data, gint source, b_input_condition cond ); |
---|
[b7d3cc34] | 57 | |
---|
| 58 | |
---|
[9730d72] | 59 | void ssl_init( void ) |
---|
[ba5add7] | 60 | { |
---|
| 61 | initialized = TRUE; |
---|
[8e9e2b7] | 62 | SSL_library_init(); |
---|
| 63 | // SSLeay_add_ssl_algorithms(); |
---|
[ba5add7] | 64 | } |
---|
| 65 | |
---|
[26fdfc5] | 66 | void *ssl_connect( char *host, int port, ssl_input_function func, gpointer data ) |
---|
[b7d3cc34] | 67 | { |
---|
| 68 | struct scd *conn = g_new0( struct scd, 1 ); |
---|
| 69 | |
---|
| 70 | conn->fd = proxy_connect( host, port, ssl_connected, conn ); |
---|
| 71 | if( conn->fd < 0 ) |
---|
| 72 | { |
---|
| 73 | g_free( conn ); |
---|
[f920d9e] | 74 | return NULL; |
---|
[b7d3cc34] | 75 | } |
---|
| 76 | |
---|
[8a2221a7] | 77 | conn->func = func; |
---|
| 78 | conn->data = data; |
---|
| 79 | conn->inpa = -1; |
---|
| 80 | |
---|
[f920d9e] | 81 | return conn; |
---|
| 82 | } |
---|
| 83 | |
---|
| 84 | void *ssl_starttls( int fd, ssl_input_function func, gpointer data ) |
---|
| 85 | { |
---|
| 86 | struct scd *conn = g_new0( struct scd, 1 ); |
---|
| 87 | |
---|
| 88 | conn->fd = fd; |
---|
| 89 | conn->func = func; |
---|
| 90 | conn->data = data; |
---|
| 91 | conn->inpa = -1; |
---|
| 92 | |
---|
| 93 | /* This function should be called via a (short) timeout instead of |
---|
| 94 | directly from here, because these SSL calls are *supposed* to be |
---|
| 95 | *completely* asynchronous and not ready yet when this function |
---|
| 96 | (or *_connect, for examle) returns. Also, errors are reported via |
---|
| 97 | the callback function, not via this function's return value. |
---|
| 98 | |
---|
| 99 | In short, doing things like this makes the rest of the code a lot |
---|
| 100 | simpler. */ |
---|
| 101 | |
---|
| 102 | b_timeout_add( 1, ssl_starttls_real, conn ); |
---|
| 103 | |
---|
| 104 | return conn; |
---|
| 105 | } |
---|
| 106 | |
---|
| 107 | static gboolean ssl_starttls_real( gpointer data, gint source, b_input_condition cond ) |
---|
| 108 | { |
---|
| 109 | struct scd *conn = data; |
---|
| 110 | |
---|
[e046390] | 111 | return ssl_connected( conn, conn->fd, B_EV_IO_WRITE ); |
---|
[f920d9e] | 112 | } |
---|
| 113 | |
---|
| 114 | static gboolean ssl_connected( gpointer data, gint source, b_input_condition cond ) |
---|
| 115 | { |
---|
| 116 | struct scd *conn = data; |
---|
| 117 | SSL_METHOD *meth; |
---|
| 118 | |
---|
| 119 | if( source == -1 ) |
---|
| 120 | goto ssl_connected_failure; |
---|
| 121 | |
---|
[b7d3cc34] | 122 | if( !initialized ) |
---|
| 123 | { |
---|
[ba5add7] | 124 | ssl_init(); |
---|
[b7d3cc34] | 125 | } |
---|
| 126 | |
---|
| 127 | meth = TLSv1_client_method(); |
---|
| 128 | conn->ssl_ctx = SSL_CTX_new( meth ); |
---|
| 129 | if( conn->ssl_ctx == NULL ) |
---|
[f920d9e] | 130 | goto ssl_connected_failure; |
---|
[b7d3cc34] | 131 | |
---|
| 132 | conn->ssl = SSL_new( conn->ssl_ctx ); |
---|
| 133 | if( conn->ssl == NULL ) |
---|
[f920d9e] | 134 | goto ssl_connected_failure; |
---|
[b7d3cc34] | 135 | |
---|
[309cb9e] | 136 | /* We can do at least the handshake with non-blocking I/O */ |
---|
[26fdfc5] | 137 | sock_make_nonblocking( conn->fd ); |
---|
[b7d3cc34] | 138 | SSL_set_fd( conn->ssl, conn->fd ); |
---|
| 139 | |
---|
[26fdfc5] | 140 | return ssl_handshake( data, source, cond ); |
---|
[f920d9e] | 141 | |
---|
| 142 | ssl_connected_failure: |
---|
| 143 | conn->func( conn->data, NULL, cond ); |
---|
| 144 | |
---|
| 145 | if( conn->ssl ) |
---|
| 146 | { |
---|
| 147 | SSL_shutdown( conn->ssl ); |
---|
| 148 | SSL_free( conn->ssl ); |
---|
| 149 | } |
---|
| 150 | if( conn->ssl_ctx ) |
---|
| 151 | { |
---|
| 152 | SSL_CTX_free( conn->ssl_ctx ); |
---|
| 153 | } |
---|
| 154 | if( source >= 0 ) closesocket( source ); |
---|
| 155 | g_free( conn ); |
---|
| 156 | |
---|
| 157 | return FALSE; |
---|
| 158 | |
---|
[26fdfc5] | 159 | } |
---|
| 160 | |
---|
[309cb9e] | 161 | static gboolean ssl_handshake( gpointer data, gint source, b_input_condition cond ) |
---|
[26fdfc5] | 162 | { |
---|
| 163 | struct scd *conn = data; |
---|
| 164 | int st; |
---|
| 165 | |
---|
| 166 | if( ( st = SSL_connect( conn->ssl ) ) < 0 ) |
---|
| 167 | { |
---|
| 168 | conn->lasterr = SSL_get_error( conn->ssl, st ); |
---|
| 169 | if( conn->lasterr != SSL_ERROR_WANT_READ && conn->lasterr != SSL_ERROR_WANT_WRITE ) |
---|
[f920d9e] | 170 | { |
---|
| 171 | conn->func( conn->data, NULL, cond ); |
---|
| 172 | |
---|
| 173 | SSL_shutdown( conn->ssl ); |
---|
| 174 | SSL_free( conn->ssl ); |
---|
| 175 | SSL_CTX_free( conn->ssl_ctx ); |
---|
| 176 | |
---|
| 177 | if( source >= 0 ) closesocket( source ); |
---|
| 178 | g_free( conn ); |
---|
| 179 | |
---|
| 180 | return FALSE; |
---|
| 181 | } |
---|
[26fdfc5] | 182 | |
---|
[309cb9e] | 183 | conn->inpa = b_input_add( conn->fd, ssl_getdirection( conn ), ssl_handshake, data ); |
---|
| 184 | return FALSE; |
---|
[26fdfc5] | 185 | } |
---|
[b7d3cc34] | 186 | |
---|
| 187 | conn->established = TRUE; |
---|
[26fdfc5] | 188 | sock_make_blocking( conn->fd ); /* For now... */ |
---|
[b7d3cc34] | 189 | conn->func( conn->data, conn, cond ); |
---|
[309cb9e] | 190 | return FALSE; |
---|
[b7d3cc34] | 191 | } |
---|
| 192 | |
---|
| 193 | int ssl_read( void *conn, char *buf, int len ) |
---|
| 194 | { |
---|
[26fdfc5] | 195 | int st; |
---|
| 196 | |
---|
[b7d3cc34] | 197 | if( !((struct scd*)conn)->established ) |
---|
[26fdfc5] | 198 | { |
---|
| 199 | ssl_errno = SSL_NOHANDSHAKE; |
---|
| 200 | return -1; |
---|
| 201 | } |
---|
| 202 | |
---|
| 203 | st = SSL_read( ((struct scd*)conn)->ssl, buf, len ); |
---|
[b7d3cc34] | 204 | |
---|
[26fdfc5] | 205 | ssl_errno = SSL_OK; |
---|
| 206 | if( st <= 0 ) |
---|
| 207 | { |
---|
| 208 | ((struct scd*)conn)->lasterr = SSL_get_error( ((struct scd*)conn)->ssl, st ); |
---|
| 209 | if( ((struct scd*)conn)->lasterr == SSL_ERROR_WANT_READ || ((struct scd*)conn)->lasterr == SSL_ERROR_WANT_WRITE ) |
---|
| 210 | ssl_errno = SSL_AGAIN; |
---|
| 211 | } |
---|
| 212 | |
---|
[327af51] | 213 | if( 0 && getenv( "BITLBEE_DEBUG" ) && st > 0 ) write( 1, buf, st ); |
---|
[50b8978] | 214 | |
---|
[26fdfc5] | 215 | return st; |
---|
[b7d3cc34] | 216 | } |
---|
| 217 | |
---|
| 218 | int ssl_write( void *conn, const char *buf, int len ) |
---|
| 219 | { |
---|
[26fdfc5] | 220 | int st; |
---|
| 221 | |
---|
[b7d3cc34] | 222 | if( !((struct scd*)conn)->established ) |
---|
[26fdfc5] | 223 | { |
---|
| 224 | ssl_errno = SSL_NOHANDSHAKE; |
---|
| 225 | return -1; |
---|
| 226 | } |
---|
| 227 | |
---|
| 228 | st = SSL_write( ((struct scd*)conn)->ssl, buf, len ); |
---|
[b7d3cc34] | 229 | |
---|
[327af51] | 230 | if( 0 && getenv( "BITLBEE_DEBUG" ) && st > 0 ) write( 1, buf, st ); |
---|
[50b8978] | 231 | |
---|
[26fdfc5] | 232 | ssl_errno = SSL_OK; |
---|
| 233 | if( st <= 0 ) |
---|
| 234 | { |
---|
| 235 | ((struct scd*)conn)->lasterr = SSL_get_error( ((struct scd*)conn)->ssl, st ); |
---|
| 236 | if( ((struct scd*)conn)->lasterr == SSL_ERROR_WANT_READ || ((struct scd*)conn)->lasterr == SSL_ERROR_WANT_WRITE ) |
---|
| 237 | ssl_errno = SSL_AGAIN; |
---|
| 238 | } |
---|
| 239 | |
---|
| 240 | return st; |
---|
[b7d3cc34] | 241 | } |
---|
| 242 | |
---|
[8a2221a7] | 243 | int ssl_pending( void *conn ) |
---|
| 244 | { |
---|
| 245 | return ( ((struct scd*)conn) && ((struct scd*)conn)->established ) ? |
---|
| 246 | SSL_pending( ((struct scd*)conn)->ssl ) > 0 : 0; |
---|
| 247 | } |
---|
| 248 | |
---|
[b7d3cc34] | 249 | void ssl_disconnect( void *conn_ ) |
---|
| 250 | { |
---|
| 251 | struct scd *conn = conn_; |
---|
| 252 | |
---|
[26fdfc5] | 253 | if( conn->inpa != -1 ) |
---|
[309cb9e] | 254 | b_event_remove( conn->inpa ); |
---|
[26fdfc5] | 255 | |
---|
[b7d3cc34] | 256 | if( conn->established ) |
---|
| 257 | SSL_shutdown( conn->ssl ); |
---|
| 258 | |
---|
| 259 | closesocket( conn->fd ); |
---|
| 260 | |
---|
| 261 | SSL_free( conn->ssl ); |
---|
| 262 | SSL_CTX_free( conn->ssl_ctx ); |
---|
| 263 | g_free( conn ); |
---|
| 264 | } |
---|
| 265 | |
---|
| 266 | int ssl_getfd( void *conn ) |
---|
| 267 | { |
---|
| 268 | return( ((struct scd*)conn)->fd ); |
---|
| 269 | } |
---|
[26fdfc5] | 270 | |
---|
[ba9edaa] | 271 | b_input_condition ssl_getdirection( void *conn ) |
---|
[26fdfc5] | 272 | { |
---|
[e046390] | 273 | return( ((struct scd*)conn)->lasterr == SSL_ERROR_WANT_WRITE ? B_EV_IO_WRITE : B_EV_IO_READ ); |
---|
[26fdfc5] | 274 | } |
---|
[523fb23] | 275 | |
---|
| 276 | size_t ssl_des3_encrypt(const unsigned char *key, size_t key_len, const unsigned char *input, size_t input_len, const unsigned char *iv, unsigned char **res) |
---|
| 277 | { |
---|
| 278 | int output_length = 0; |
---|
[50b8978] | 279 | EVP_CIPHER_CTX ctx; |
---|
[523fb23] | 280 | |
---|
| 281 | *res = g_new0(unsigned char, 72); |
---|
| 282 | |
---|
| 283 | /* Don't set key or IV because we will modify the parameters */ |
---|
| 284 | EVP_CIPHER_CTX_init(&ctx); |
---|
| 285 | EVP_CipherInit_ex(&ctx, EVP_des_ede3_cbc(), NULL, NULL, NULL, 1); |
---|
| 286 | EVP_CIPHER_CTX_set_key_length(&ctx, key_len); |
---|
| 287 | EVP_CIPHER_CTX_set_padding(&ctx, 0); |
---|
| 288 | /* We finished modifying parameters so now we can set key and IV */ |
---|
| 289 | EVP_CipherInit_ex(&ctx, NULL, NULL, key, iv, 1); |
---|
| 290 | EVP_CipherUpdate(&ctx, *res, &output_length, input, input_len); |
---|
| 291 | EVP_CipherFinal_ex(&ctx, *res, &output_length); |
---|
| 292 | EVP_CIPHER_CTX_cleanup(&ctx); |
---|
[50b8978] | 293 | //EVP_cleanup(); |
---|
[523fb23] | 294 | |
---|
| 295 | return output_length; |
---|
| 296 | } |
---|