ported double tracking from TC
This commit is contained in:
291
deps/mysqllite/vio/vio.c
vendored
Normal file
291
deps/mysqllite/vio/vio.c
vendored
Normal file
@@ -0,0 +1,291 @@
|
||||
/* Copyright (C) 2000 MySQL AB
|
||||
|
||||
This program is free software; you can redistribute it and/or modify
|
||||
it under the terms of the GNU General Public License as published by
|
||||
the Free Software Foundation; version 2 of the License.
|
||||
|
||||
This program is distributed in the hope that it will be useful,
|
||||
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
GNU General Public License for more details.
|
||||
|
||||
You should have received a copy of the GNU General Public License
|
||||
along with this program; if not, write to the Free Software
|
||||
Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA */
|
||||
|
||||
/*
|
||||
Note that we can't have assertion on file descriptors; The reason for
|
||||
this is that during mysql shutdown, another thread can close a file
|
||||
we are working on. In this case we should just return read errors from
|
||||
the file descriptior.
|
||||
*/
|
||||
|
||||
#include "vio_priv.h"
|
||||
|
||||
#if defined(__WIN__) || defined(HAVE_SMEM)
|
||||
|
||||
/**
|
||||
Stub poll_read method that defaults to indicate that there
|
||||
is data to read.
|
||||
|
||||
Used for named pipe and shared memory VIO types.
|
||||
|
||||
@param vio Unused.
|
||||
@param timeout Unused.
|
||||
|
||||
@retval FALSE There is data to read.
|
||||
*/
|
||||
|
||||
static my_bool no_poll_read(Vio *vio __attribute__((unused)),
|
||||
uint timeout __attribute__((unused)))
|
||||
{
|
||||
return FALSE;
|
||||
}
|
||||
|
||||
#endif
|
||||
|
||||
static my_bool has_no_data(Vio *vio __attribute__((unused)))
|
||||
{
|
||||
return FALSE;
|
||||
}
|
||||
|
||||
/*
|
||||
* Helper to fill most of the Vio* with defaults.
|
||||
*/
|
||||
|
||||
static void vio_init(Vio* vio, enum enum_vio_type type,
|
||||
my_socket sd, HANDLE hPipe, uint flags)
|
||||
{
|
||||
DBUG_ENTER("vio_init");
|
||||
DBUG_PRINT("enter", ("type: %d sd: %d flags: %d", type, sd, flags));
|
||||
|
||||
#ifndef HAVE_VIO_READ_BUFF
|
||||
flags&= ~VIO_BUFFERED_READ;
|
||||
#endif
|
||||
bzero((char*) vio, sizeof(*vio));
|
||||
vio->type = type;
|
||||
vio->sd = sd;
|
||||
vio->hPipe = hPipe;
|
||||
vio->localhost= flags & VIO_LOCALHOST;
|
||||
if ((flags & VIO_BUFFERED_READ) &&
|
||||
!(vio->read_buffer= (char*)my_malloc(VIO_READ_BUFFER_SIZE, MYF(MY_WME))))
|
||||
flags&= ~VIO_BUFFERED_READ;
|
||||
#ifdef _WIN32
|
||||
if (type == VIO_TYPE_NAMEDPIPE)
|
||||
{
|
||||
vio->viodelete =vio_delete;
|
||||
vio->vioerrno =vio_errno;
|
||||
vio->read =vio_read_pipe;
|
||||
vio->write =vio_write_pipe;
|
||||
vio->fastsend =vio_fastsend;
|
||||
vio->viokeepalive =vio_keepalive;
|
||||
vio->should_retry =vio_should_retry;
|
||||
vio->was_interrupted=vio_was_interrupted;
|
||||
vio->vioclose =vio_close_pipe;
|
||||
vio->peer_addr =vio_peer_addr;
|
||||
vio->vioblocking =vio_blocking;
|
||||
vio->is_blocking =vio_is_blocking;
|
||||
|
||||
vio->poll_read =no_poll_read;
|
||||
vio->is_connected =vio_is_connected_pipe;
|
||||
vio->has_data =has_no_data;
|
||||
|
||||
vio->timeout=vio_win32_timeout;
|
||||
/* Set default timeout */
|
||||
vio->read_timeout_ms= INFINITE;
|
||||
vio->write_timeout_ms= INFINITE;
|
||||
vio->pipe_overlapped.hEvent= CreateEvent(NULL, TRUE, FALSE, NULL);
|
||||
DBUG_VOID_RETURN;
|
||||
}
|
||||
#endif
|
||||
#ifdef HAVE_SMEM
|
||||
if (type == VIO_TYPE_SHARED_MEMORY)
|
||||
{
|
||||
vio->viodelete =vio_delete;
|
||||
vio->vioerrno =vio_errno;
|
||||
vio->read =vio_read_shared_memory;
|
||||
vio->write =vio_write_shared_memory;
|
||||
vio->fastsend =vio_fastsend;
|
||||
vio->viokeepalive =vio_keepalive;
|
||||
vio->should_retry =vio_should_retry;
|
||||
vio->was_interrupted=vio_was_interrupted;
|
||||
vio->vioclose =vio_close_shared_memory;
|
||||
vio->peer_addr =vio_peer_addr;
|
||||
vio->vioblocking =vio_blocking;
|
||||
vio->is_blocking =vio_is_blocking;
|
||||
|
||||
vio->poll_read =no_poll_read;
|
||||
vio->is_connected =vio_is_connected_shared_memory;
|
||||
vio->has_data =has_no_data;
|
||||
|
||||
/* Currently, shared memory is on Windows only, hence the below is ok*/
|
||||
vio->timeout= vio_win32_timeout;
|
||||
/* Set default timeout */
|
||||
vio->read_timeout_ms= INFINITE;
|
||||
vio->write_timeout_ms= INFINITE;
|
||||
DBUG_VOID_RETURN;
|
||||
}
|
||||
#endif
|
||||
#ifdef HAVE_OPENSSL
|
||||
if (type == VIO_TYPE_SSL)
|
||||
{
|
||||
vio->viodelete =vio_ssl_delete;
|
||||
vio->vioerrno =vio_errno;
|
||||
vio->read =vio_ssl_read;
|
||||
vio->write =vio_ssl_write;
|
||||
vio->fastsend =vio_fastsend;
|
||||
vio->viokeepalive =vio_keepalive;
|
||||
vio->should_retry =vio_should_retry;
|
||||
vio->was_interrupted=vio_was_interrupted;
|
||||
vio->vioclose =vio_ssl_close;
|
||||
vio->peer_addr =vio_peer_addr;
|
||||
vio->vioblocking =vio_ssl_blocking;
|
||||
vio->is_blocking =vio_is_blocking;
|
||||
vio->timeout =vio_timeout;
|
||||
vio->poll_read =vio_poll_read;
|
||||
vio->is_connected =vio_is_connected;
|
||||
vio->has_data =vio_ssl_has_data;
|
||||
DBUG_VOID_RETURN;
|
||||
}
|
||||
#endif /* HAVE_OPENSSL */
|
||||
vio->viodelete =vio_delete;
|
||||
vio->vioerrno =vio_errno;
|
||||
vio->read= (flags & VIO_BUFFERED_READ) ? vio_read_buff : vio_read;
|
||||
vio->write =vio_write;
|
||||
vio->fastsend =vio_fastsend;
|
||||
vio->viokeepalive =vio_keepalive;
|
||||
vio->should_retry =vio_should_retry;
|
||||
vio->was_interrupted =vio_was_interrupted;
|
||||
vio->vioclose =vio_close;
|
||||
vio->peer_addr =vio_peer_addr;
|
||||
vio->vioblocking =vio_blocking;
|
||||
vio->is_blocking =vio_is_blocking;
|
||||
vio->timeout =vio_timeout;
|
||||
vio->poll_read =vio_poll_read;
|
||||
vio->is_connected =vio_is_connected;
|
||||
vio->has_data= (flags & VIO_BUFFERED_READ) ?
|
||||
vio_buff_has_data : has_no_data;
|
||||
DBUG_VOID_RETURN;
|
||||
}
|
||||
|
||||
|
||||
/* Reset initialized VIO to use with another transport type */
|
||||
|
||||
void vio_reset(Vio* vio, enum enum_vio_type type,
|
||||
my_socket sd, HANDLE hPipe, uint flags)
|
||||
{
|
||||
my_free(vio->read_buffer);
|
||||
vio_init(vio, type, sd, hPipe, flags);
|
||||
}
|
||||
|
||||
|
||||
/* Open the socket or TCP/IP connection and read the fnctl() status */
|
||||
|
||||
Vio *vio_new(my_socket sd, enum enum_vio_type type, uint flags)
|
||||
{
|
||||
Vio *vio;
|
||||
DBUG_ENTER("vio_new");
|
||||
DBUG_PRINT("enter", ("sd: %d", sd));
|
||||
if ((vio = (Vio*) my_malloc(sizeof(*vio),MYF(MY_WME))))
|
||||
{
|
||||
vio_init(vio, type, sd, 0, flags);
|
||||
sprintf(vio->desc,
|
||||
(vio->type == VIO_TYPE_SOCKET ? "socket (%d)" : "TCP/IP (%d)"),
|
||||
vio->sd);
|
||||
#if !defined(__WIN__)
|
||||
#if !defined(NO_FCNTL_NONBLOCK)
|
||||
/*
|
||||
We call fcntl() to set the flags and then immediately read them back
|
||||
to make sure that we and the system are in agreement on the state of
|
||||
things.
|
||||
|
||||
An example of why we need to do this is FreeBSD (and apparently some
|
||||
other BSD-derived systems, like Mac OS X), where the system sometimes
|
||||
reports that the socket is set for non-blocking when it really will
|
||||
block.
|
||||
*/
|
||||
fcntl(sd, F_SETFL, 0);
|
||||
vio->fcntl_mode= fcntl(sd, F_GETFL);
|
||||
#elif defined(HAVE_SYS_IOCTL_H) /* hpux */
|
||||
/* Non blocking sockets doesn't work good on HPUX 11.0 */
|
||||
(void) ioctl(sd,FIOSNBIO,0);
|
||||
vio->fcntl_mode &= ~O_NONBLOCK;
|
||||
#endif
|
||||
#else /* !defined(__WIN__) */
|
||||
{
|
||||
/* set to blocking mode by default */
|
||||
ulong arg=0, r;
|
||||
r = ioctlsocket(sd,FIONBIO,(void*) &arg);
|
||||
vio->fcntl_mode &= ~O_NONBLOCK;
|
||||
}
|
||||
#endif
|
||||
}
|
||||
DBUG_RETURN(vio);
|
||||
}
|
||||
|
||||
|
||||
#ifdef __WIN__
|
||||
|
||||
Vio *vio_new_win32pipe(HANDLE hPipe)
|
||||
{
|
||||
Vio *vio;
|
||||
DBUG_ENTER("vio_new_handle");
|
||||
if ((vio = (Vio*) my_malloc(sizeof(Vio),MYF(MY_WME))))
|
||||
{
|
||||
vio_init(vio, VIO_TYPE_NAMEDPIPE, 0, hPipe, VIO_LOCALHOST);
|
||||
strmov(vio->desc, "named pipe");
|
||||
}
|
||||
DBUG_RETURN(vio);
|
||||
}
|
||||
|
||||
#ifdef HAVE_SMEM
|
||||
Vio *vio_new_win32shared_memory(HANDLE handle_file_map, HANDLE handle_map,
|
||||
HANDLE event_server_wrote, HANDLE event_server_read,
|
||||
HANDLE event_client_wrote, HANDLE event_client_read,
|
||||
HANDLE event_conn_closed)
|
||||
{
|
||||
Vio *vio;
|
||||
DBUG_ENTER("vio_new_win32shared_memory");
|
||||
if ((vio = (Vio*) my_malloc(sizeof(Vio),MYF(MY_WME))))
|
||||
{
|
||||
vio_init(vio, VIO_TYPE_SHARED_MEMORY, 0, 0, VIO_LOCALHOST);
|
||||
vio->handle_file_map= handle_file_map;
|
||||
vio->handle_map= handle_map;
|
||||
vio->event_server_wrote= event_server_wrote;
|
||||
vio->event_server_read= event_server_read;
|
||||
vio->event_client_wrote= event_client_wrote;
|
||||
vio->event_client_read= event_client_read;
|
||||
vio->event_conn_closed= event_conn_closed;
|
||||
vio->shared_memory_remain= 0;
|
||||
vio->shared_memory_pos= handle_map;
|
||||
strmov(vio->desc, "shared memory");
|
||||
}
|
||||
DBUG_RETURN(vio);
|
||||
}
|
||||
#endif
|
||||
#endif
|
||||
|
||||
|
||||
void vio_delete(Vio* vio)
|
||||
{
|
||||
if (!vio)
|
||||
return; /* It must be safe to delete null pointers. */
|
||||
|
||||
if (vio->type != VIO_CLOSED)
|
||||
vio->vioclose(vio);
|
||||
my_free(vio->read_buffer);
|
||||
my_free(vio);
|
||||
}
|
||||
|
||||
|
||||
/*
|
||||
Cleanup memory allocated by vio or the
|
||||
components below it when application finish
|
||||
|
||||
*/
|
||||
void vio_end(void)
|
||||
{
|
||||
#ifdef HAVE_YASSL
|
||||
yaSSL_CleanUp();
|
||||
#endif
|
||||
}
|
||||
69
deps/mysqllite/vio/vio_priv.h
vendored
Normal file
69
deps/mysqllite/vio/vio_priv.h
vendored
Normal file
@@ -0,0 +1,69 @@
|
||||
#ifndef VIO_PRIV_INCLUDED
|
||||
#define VIO_PRIV_INCLUDED
|
||||
|
||||
/* Copyright (C) 2003 MySQL AB
|
||||
|
||||
This program is free software; you can redistribute it and/or modify
|
||||
it under the terms of the GNU General Public License as published by
|
||||
the Free Software Foundation; version 2 of the License.
|
||||
|
||||
This program is distributed in the hope that it will be useful,
|
||||
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
GNU General Public License for more details.
|
||||
|
||||
You should have received a copy of the GNU General Public License
|
||||
along with this program; if not, write to the Free Software
|
||||
Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA */
|
||||
|
||||
/* Structures and functions private to the vio package */
|
||||
|
||||
#define DONT_MAP_VIO
|
||||
#include <my_global.h>
|
||||
#include <mysql_com.h>
|
||||
#include <my_sys.h>
|
||||
#include <m_string.h>
|
||||
#include <violite.h>
|
||||
|
||||
#ifndef __WIN__
|
||||
#include <sys/socket.h>
|
||||
#include <netdb.h>
|
||||
#endif
|
||||
|
||||
#ifdef _WIN32
|
||||
void vio_win32_timeout(Vio *vio, uint which, uint timeout);
|
||||
#endif
|
||||
|
||||
#ifdef __WIN__
|
||||
size_t vio_read_pipe(Vio *vio, uchar * buf, size_t size);
|
||||
size_t vio_write_pipe(Vio *vio, const uchar * buf, size_t size);
|
||||
my_bool vio_is_connected_pipe(Vio *vio);
|
||||
int vio_close_pipe(Vio * vio);
|
||||
#endif
|
||||
|
||||
#ifdef HAVE_SMEM
|
||||
size_t vio_read_shared_memory(Vio *vio, uchar * buf, size_t size);
|
||||
size_t vio_write_shared_memory(Vio *vio, const uchar * buf, size_t size);
|
||||
my_bool vio_is_connected_shared_memory(Vio *vio);
|
||||
int vio_close_shared_memory(Vio * vio);
|
||||
#endif
|
||||
|
||||
void vio_timeout(Vio *vio,uint which, uint timeout);
|
||||
my_bool vio_buff_has_data(Vio *vio);
|
||||
|
||||
#ifdef HAVE_OPENSSL
|
||||
#include "my_net.h" /* needed because of struct in_addr */
|
||||
|
||||
size_t vio_ssl_read(Vio *vio,uchar* buf, size_t size);
|
||||
size_t vio_ssl_write(Vio *vio,const uchar* buf, size_t size);
|
||||
|
||||
/* When the workday is over... */
|
||||
int vio_ssl_close(Vio *vio);
|
||||
void vio_ssl_delete(Vio *vio);
|
||||
|
||||
int vio_ssl_blocking(Vio *vio, my_bool set_blocking_mode, my_bool *old_mode);
|
||||
|
||||
my_bool vio_ssl_has_data(Vio *vio);
|
||||
|
||||
#endif /* HAVE_OPENSSL */
|
||||
#endif /* VIO_PRIV_INCLUDED */
|
||||
1100
deps/mysqllite/vio/viosocket.c
vendored
Normal file
1100
deps/mysqllite/vio/viosocket.c
vendored
Normal file
File diff suppressed because it is too large
Load Diff
252
deps/mysqllite/vio/viossl.c
vendored
Normal file
252
deps/mysqllite/vio/viossl.c
vendored
Normal file
@@ -0,0 +1,252 @@
|
||||
/* Copyright (C) 2000 MySQL AB
|
||||
|
||||
This program is free software; you can redistribute it and/or modify
|
||||
it under the terms of the GNU General Public License as published by
|
||||
the Free Software Foundation; version 2 of the License.
|
||||
|
||||
This program is distributed in the hope that it will be useful,
|
||||
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
GNU General Public License for more details.
|
||||
|
||||
You should have received a copy of the GNU General Public License
|
||||
along with this program; if not, write to the Free Software
|
||||
Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA */
|
||||
|
||||
/*
|
||||
Note that we can't have assertion on file descriptors; The reason for
|
||||
this is that during mysql shutdown, another thread can close a file
|
||||
we are working on. In this case we should just return read errors from
|
||||
the file descriptior.
|
||||
*/
|
||||
|
||||
#include "vio_priv.h"
|
||||
|
||||
#ifdef HAVE_OPENSSL
|
||||
|
||||
static void
|
||||
report_errors(SSL* ssl)
|
||||
{
|
||||
unsigned long l;
|
||||
const char *file;
|
||||
const char *data;
|
||||
int line, flags;
|
||||
#ifndef DBUG_OFF
|
||||
char buf[512];
|
||||
#endif
|
||||
|
||||
DBUG_ENTER("report_errors");
|
||||
|
||||
while ((l= ERR_get_error_line_data(&file,&line,&data,&flags)))
|
||||
{
|
||||
DBUG_PRINT("error", ("OpenSSL: %s:%s:%d:%s\n", ERR_error_string(l,buf),
|
||||
file,line,(flags&ERR_TXT_STRING)?data:"")) ;
|
||||
}
|
||||
|
||||
if (ssl)
|
||||
DBUG_PRINT("error", ("error: %s",
|
||||
ERR_error_string(SSL_get_error(ssl, l), buf)));
|
||||
|
||||
DBUG_PRINT("info", ("socket_errno: %d", socket_errno));
|
||||
DBUG_VOID_RETURN;
|
||||
}
|
||||
|
||||
|
||||
size_t vio_ssl_read(Vio *vio, uchar* buf, size_t size)
|
||||
{
|
||||
size_t r;
|
||||
DBUG_ENTER("vio_ssl_read");
|
||||
DBUG_PRINT("enter", ("sd: %d buf: 0x%lx size: %u ssl: 0x%lx",
|
||||
vio->sd, (long) buf, (uint) size, (long) vio->ssl_arg));
|
||||
|
||||
r= SSL_read((SSL*) vio->ssl_arg, buf, size);
|
||||
#ifndef DBUG_OFF
|
||||
if (r == (size_t) -1)
|
||||
report_errors((SSL*) vio->ssl_arg);
|
||||
#endif
|
||||
DBUG_PRINT("exit", ("%u", (uint) r));
|
||||
DBUG_RETURN(r);
|
||||
}
|
||||
|
||||
|
||||
size_t vio_ssl_write(Vio *vio, const uchar* buf, size_t size)
|
||||
{
|
||||
size_t r;
|
||||
DBUG_ENTER("vio_ssl_write");
|
||||
DBUG_PRINT("enter", ("sd: %d buf: 0x%lx size: %u", vio->sd,
|
||||
(long) buf, (uint) size));
|
||||
|
||||
r= SSL_write((SSL*) vio->ssl_arg, buf, size);
|
||||
#ifndef DBUG_OFF
|
||||
if (r == (size_t) -1)
|
||||
report_errors((SSL*) vio->ssl_arg);
|
||||
#endif
|
||||
DBUG_PRINT("exit", ("%u", (uint) r));
|
||||
DBUG_RETURN(r);
|
||||
}
|
||||
|
||||
|
||||
int vio_ssl_close(Vio *vio)
|
||||
{
|
||||
int r= 0;
|
||||
SSL *ssl= (SSL*)vio->ssl_arg;
|
||||
DBUG_ENTER("vio_ssl_close");
|
||||
|
||||
if (ssl)
|
||||
{
|
||||
/*
|
||||
THE SSL standard says that SSL sockets must send and receive a close_notify
|
||||
alert on socket shutdown to avoid truncation attacks. However, this can
|
||||
cause problems since we often hold a lock during shutdown and this IO can
|
||||
take an unbounded amount of time to complete. Since our packets are self
|
||||
describing with length, we aren't vunerable to these attacks. Therefore,
|
||||
we just shutdown by closing the socket (quiet shutdown).
|
||||
*/
|
||||
SSL_set_quiet_shutdown(ssl, 1);
|
||||
|
||||
switch ((r= SSL_shutdown(ssl))) {
|
||||
case 1:
|
||||
/* Shutdown successful */
|
||||
break;
|
||||
case 0:
|
||||
/*
|
||||
Shutdown not yet finished - since the socket is going to
|
||||
be closed there is no need to call SSL_shutdown() a second
|
||||
time to wait for the other side to respond
|
||||
*/
|
||||
break;
|
||||
default: /* Shutdown failed */
|
||||
DBUG_PRINT("vio_error", ("SSL_shutdown() failed, error: %d",
|
||||
SSL_get_error(ssl, r)));
|
||||
break;
|
||||
}
|
||||
}
|
||||
DBUG_RETURN(vio_close(vio));
|
||||
}
|
||||
|
||||
|
||||
void vio_ssl_delete(Vio *vio)
|
||||
{
|
||||
if (!vio)
|
||||
return; /* It must be safe to delete null pointer */
|
||||
|
||||
if (vio->type == VIO_TYPE_SSL)
|
||||
vio_ssl_close(vio); /* Still open, close connection first */
|
||||
|
||||
if (vio->ssl_arg)
|
||||
{
|
||||
SSL_free((SSL*) vio->ssl_arg);
|
||||
vio->ssl_arg= 0;
|
||||
}
|
||||
|
||||
vio_delete(vio);
|
||||
}
|
||||
|
||||
|
||||
static int ssl_do(struct st_VioSSLFd *ptr, Vio *vio, long timeout,
|
||||
int (*connect_accept_func)(SSL*))
|
||||
{
|
||||
SSL *ssl;
|
||||
my_bool unused;
|
||||
my_bool was_blocking;
|
||||
|
||||
DBUG_ENTER("ssl_do");
|
||||
DBUG_PRINT("enter", ("ptr: 0x%lx, sd: %d ctx: 0x%lx",
|
||||
(long) ptr, vio->sd, (long) ptr->ssl_context));
|
||||
|
||||
/* Set socket to blocking if not already set */
|
||||
vio_blocking(vio, 1, &was_blocking);
|
||||
|
||||
if (!(ssl= SSL_new(ptr->ssl_context)))
|
||||
{
|
||||
DBUG_PRINT("error", ("SSL_new failure"));
|
||||
report_errors(ssl);
|
||||
vio_blocking(vio, was_blocking, &unused);
|
||||
DBUG_RETURN(1);
|
||||
}
|
||||
DBUG_PRINT("info", ("ssl: 0x%lx timeout: %ld", (long) ssl, timeout));
|
||||
SSL_clear(ssl);
|
||||
SSL_SESSION_set_timeout(SSL_get_session(ssl), timeout);
|
||||
SSL_set_fd(ssl, vio->sd);
|
||||
|
||||
if (connect_accept_func(ssl) < 1)
|
||||
{
|
||||
DBUG_PRINT("error", ("SSL_connect/accept failure"));
|
||||
report_errors(ssl);
|
||||
SSL_free(ssl);
|
||||
vio_blocking(vio, was_blocking, &unused);
|
||||
DBUG_RETURN(1);
|
||||
}
|
||||
|
||||
/*
|
||||
Connection succeeded. Install new function handlers,
|
||||
change type, set sd to the fd used when connecting
|
||||
and set pointer to the SSL structure
|
||||
*/
|
||||
vio_reset(vio, VIO_TYPE_SSL, SSL_get_fd(ssl), 0, 0);
|
||||
vio->ssl_arg= (void*)ssl;
|
||||
|
||||
#ifndef DBUG_OFF
|
||||
{
|
||||
/* Print some info about the peer */
|
||||
X509 *cert;
|
||||
char buf[512];
|
||||
|
||||
DBUG_PRINT("info",("SSL connection succeeded"));
|
||||
DBUG_PRINT("info",("Using cipher: '%s'" , SSL_get_cipher_name(ssl)));
|
||||
|
||||
if ((cert= SSL_get_peer_certificate (ssl)))
|
||||
{
|
||||
DBUG_PRINT("info",("Peer certificate:"));
|
||||
X509_NAME_oneline(X509_get_subject_name(cert), buf, sizeof(buf));
|
||||
DBUG_PRINT("info",("\t subject: '%s'", buf));
|
||||
X509_NAME_oneline(X509_get_issuer_name(cert), buf, sizeof(buf));
|
||||
DBUG_PRINT("info",("\t issuer: '%s'", buf));
|
||||
X509_free(cert);
|
||||
}
|
||||
else
|
||||
DBUG_PRINT("info",("Peer does not have certificate."));
|
||||
|
||||
if (SSL_get_shared_ciphers(ssl, buf, sizeof(buf)))
|
||||
{
|
||||
DBUG_PRINT("info",("shared_ciphers: '%s'", buf));
|
||||
}
|
||||
else
|
||||
DBUG_PRINT("info",("no shared ciphers!"));
|
||||
}
|
||||
#endif
|
||||
|
||||
DBUG_RETURN(0);
|
||||
}
|
||||
|
||||
|
||||
int sslaccept(struct st_VioSSLFd *ptr, Vio *vio, long timeout)
|
||||
{
|
||||
DBUG_ENTER("sslaccept");
|
||||
DBUG_RETURN(ssl_do(ptr, vio, timeout, SSL_accept));
|
||||
}
|
||||
|
||||
|
||||
int sslconnect(struct st_VioSSLFd *ptr, Vio *vio, long timeout)
|
||||
{
|
||||
DBUG_ENTER("sslconnect");
|
||||
DBUG_RETURN(ssl_do(ptr, vio, timeout, SSL_connect));
|
||||
}
|
||||
|
||||
|
||||
int vio_ssl_blocking(Vio *vio __attribute__((unused)),
|
||||
my_bool set_blocking_mode,
|
||||
my_bool *old_mode)
|
||||
{
|
||||
/* Mode is always blocking */
|
||||
*old_mode= 1;
|
||||
/* Return error if we try to change to non_blocking mode */
|
||||
return (set_blocking_mode ? 0 : 1);
|
||||
}
|
||||
|
||||
my_bool vio_ssl_has_data(Vio *vio)
|
||||
{
|
||||
return SSL_pending(vio->ssl_arg) > 0 ? TRUE : FALSE;
|
||||
}
|
||||
|
||||
#endif /* HAVE_OPENSSL */
|
||||
315
deps/mysqllite/vio/viosslfactories.c
vendored
Normal file
315
deps/mysqllite/vio/viosslfactories.c
vendored
Normal file
@@ -0,0 +1,315 @@
|
||||
/* Copyright (C) 2000 MySQL AB
|
||||
|
||||
This program is free software; you can redistribute it and/or modify
|
||||
it under the terms of the GNU General Public License as published by
|
||||
the Free Software Foundation; version 2 of the License.
|
||||
|
||||
This program is distributed in the hope that it will be useful,
|
||||
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
GNU General Public License for more details.
|
||||
|
||||
You should have received a copy of the GNU General Public License
|
||||
along with this program; if not, write to the Free Software
|
||||
Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA */
|
||||
|
||||
#include "vio_priv.h"
|
||||
|
||||
#ifdef HAVE_OPENSSL
|
||||
|
||||
static my_bool ssl_algorithms_added = FALSE;
|
||||
static my_bool ssl_error_strings_loaded= FALSE;
|
||||
|
||||
static unsigned char dh512_p[]=
|
||||
{
|
||||
0xDA,0x58,0x3C,0x16,0xD9,0x85,0x22,0x89,0xD0,0xE4,0xAF,0x75,
|
||||
0x6F,0x4C,0xCA,0x92,0xDD,0x4B,0xE5,0x33,0xB8,0x04,0xFB,0x0F,
|
||||
0xED,0x94,0xEF,0x9C,0x8A,0x44,0x03,0xED,0x57,0x46,0x50,0xD3,
|
||||
0x69,0x99,0xDB,0x29,0xD7,0x76,0x27,0x6B,0xA2,0xD3,0xD4,0x12,
|
||||
0xE2,0x18,0xF4,0xDD,0x1E,0x08,0x4C,0xF6,0xD8,0x00,0x3E,0x7C,
|
||||
0x47,0x74,0xE8,0x33,
|
||||
};
|
||||
|
||||
static unsigned char dh512_g[]={
|
||||
0x02,
|
||||
};
|
||||
|
||||
static DH *get_dh512(void)
|
||||
{
|
||||
DH *dh;
|
||||
if ((dh=DH_new()))
|
||||
{
|
||||
dh->p=BN_bin2bn(dh512_p,sizeof(dh512_p),NULL);
|
||||
dh->g=BN_bin2bn(dh512_g,sizeof(dh512_g),NULL);
|
||||
if (! dh->p || ! dh->g)
|
||||
{
|
||||
DH_free(dh);
|
||||
dh=0;
|
||||
}
|
||||
}
|
||||
return(dh);
|
||||
}
|
||||
|
||||
|
||||
static void
|
||||
report_errors()
|
||||
{
|
||||
unsigned long l;
|
||||
const char* file;
|
||||
const char* data;
|
||||
int line,flags;
|
||||
|
||||
DBUG_ENTER("report_errors");
|
||||
|
||||
while ((l=ERR_get_error_line_data(&file,&line,&data,&flags)) != 0)
|
||||
{
|
||||
#ifndef DBUG_OFF /* Avoid warning */
|
||||
char buf[200];
|
||||
DBUG_PRINT("error", ("OpenSSL: %s:%s:%d:%s\n", ERR_error_string(l,buf),
|
||||
file,line,(flags & ERR_TXT_STRING) ? data : "")) ;
|
||||
#endif
|
||||
}
|
||||
DBUG_VOID_RETURN;
|
||||
}
|
||||
|
||||
static const char*
|
||||
ssl_error_string[] =
|
||||
{
|
||||
"No error",
|
||||
"Unable to get certificate",
|
||||
"Unable to get private key",
|
||||
"Private key does not match the certificate public key"
|
||||
"SSL_CTX_set_default_verify_paths failed",
|
||||
"Failed to set ciphers to use",
|
||||
"SSL_CTX_new failed"
|
||||
};
|
||||
|
||||
const char*
|
||||
sslGetErrString(enum enum_ssl_init_error e)
|
||||
{
|
||||
DBUG_ASSERT(SSL_INITERR_NOERROR < e && e < SSL_INITERR_LASTERR);
|
||||
return ssl_error_string[e];
|
||||
}
|
||||
|
||||
static int
|
||||
vio_set_cert_stuff(SSL_CTX *ctx, const char *cert_file, const char *key_file,
|
||||
enum enum_ssl_init_error* error)
|
||||
{
|
||||
DBUG_ENTER("vio_set_cert_stuff");
|
||||
DBUG_PRINT("enter", ("ctx: 0x%lx cert_file: %s key_file: %s",
|
||||
(long) ctx, cert_file, key_file));
|
||||
if (cert_file)
|
||||
{
|
||||
if (SSL_CTX_use_certificate_file(ctx, cert_file, SSL_FILETYPE_PEM) <= 0)
|
||||
{
|
||||
*error= SSL_INITERR_CERT;
|
||||
DBUG_PRINT("error",("%s from file '%s'", sslGetErrString(*error), cert_file));
|
||||
DBUG_EXECUTE("error", ERR_print_errors_fp(DBUG_FILE););
|
||||
fprintf(stderr, "SSL error: %s from '%s'\n", sslGetErrString(*error),
|
||||
cert_file);
|
||||
fflush(stderr);
|
||||
DBUG_RETURN(1);
|
||||
}
|
||||
|
||||
if (!key_file)
|
||||
key_file= cert_file;
|
||||
|
||||
if (SSL_CTX_use_PrivateKey_file(ctx, key_file, SSL_FILETYPE_PEM) <= 0)
|
||||
{
|
||||
*error= SSL_INITERR_KEY;
|
||||
DBUG_PRINT("error", ("%s from file '%s'", sslGetErrString(*error), key_file));
|
||||
DBUG_EXECUTE("error", ERR_print_errors_fp(DBUG_FILE););
|
||||
fprintf(stderr, "SSL error: %s from '%s'\n", sslGetErrString(*error),
|
||||
key_file);
|
||||
fflush(stderr);
|
||||
DBUG_RETURN(1);
|
||||
}
|
||||
|
||||
/*
|
||||
If we are using DSA, we can copy the parameters from the private key
|
||||
Now we know that a key and cert have been set against the SSL context
|
||||
*/
|
||||
if (!SSL_CTX_check_private_key(ctx))
|
||||
{
|
||||
*error= SSL_INITERR_NOMATCH;
|
||||
DBUG_PRINT("error", ("%s",sslGetErrString(*error)));
|
||||
DBUG_EXECUTE("error", ERR_print_errors_fp(DBUG_FILE););
|
||||
fprintf(stderr, "SSL error: %s\n", sslGetErrString(*error));
|
||||
fflush(stderr);
|
||||
DBUG_RETURN(1);
|
||||
}
|
||||
}
|
||||
DBUG_RETURN(0);
|
||||
}
|
||||
|
||||
|
||||
static void check_ssl_init()
|
||||
{
|
||||
if (!ssl_algorithms_added)
|
||||
{
|
||||
ssl_algorithms_added= TRUE;
|
||||
SSL_library_init();
|
||||
OpenSSL_add_all_algorithms();
|
||||
|
||||
}
|
||||
|
||||
if (!ssl_error_strings_loaded)
|
||||
{
|
||||
ssl_error_strings_loaded= TRUE;
|
||||
SSL_load_error_strings();
|
||||
}
|
||||
}
|
||||
|
||||
/************************ VioSSLFd **********************************/
|
||||
static struct st_VioSSLFd *
|
||||
new_VioSSLFd(const char *key_file, const char *cert_file,
|
||||
const char *ca_file, const char *ca_path,
|
||||
const char *cipher, SSL_METHOD *method,
|
||||
enum enum_ssl_init_error* error)
|
||||
{
|
||||
DH *dh;
|
||||
struct st_VioSSLFd *ssl_fd;
|
||||
DBUG_ENTER("new_VioSSLFd");
|
||||
DBUG_PRINT("enter",
|
||||
("key_file: '%s' cert_file: '%s' ca_file: '%s' ca_path: '%s' "
|
||||
"cipher: '%s'",
|
||||
key_file ? key_file : "NULL",
|
||||
cert_file ? cert_file : "NULL",
|
||||
ca_file ? ca_file : "NULL",
|
||||
ca_path ? ca_path : "NULL",
|
||||
cipher ? cipher : "NULL"));
|
||||
|
||||
check_ssl_init();
|
||||
|
||||
if (!(ssl_fd= ((struct st_VioSSLFd*)
|
||||
my_malloc(sizeof(struct st_VioSSLFd),MYF(0)))))
|
||||
DBUG_RETURN(0);
|
||||
|
||||
if (!(ssl_fd->ssl_context= SSL_CTX_new(method)))
|
||||
{
|
||||
*error= SSL_INITERR_MEMFAIL;
|
||||
DBUG_PRINT("error", ("%s", sslGetErrString(*error)));
|
||||
report_errors();
|
||||
my_free(ssl_fd);
|
||||
DBUG_RETURN(0);
|
||||
}
|
||||
|
||||
/*
|
||||
Set the ciphers that can be used
|
||||
NOTE: SSL_CTX_set_cipher_list will return 0 if
|
||||
none of the provided ciphers could be selected
|
||||
*/
|
||||
if (cipher &&
|
||||
SSL_CTX_set_cipher_list(ssl_fd->ssl_context, cipher) == 0)
|
||||
{
|
||||
*error= SSL_INITERR_CIPHERS;
|
||||
DBUG_PRINT("error", ("%s", sslGetErrString(*error)));
|
||||
report_errors();
|
||||
SSL_CTX_free(ssl_fd->ssl_context);
|
||||
my_free(ssl_fd);
|
||||
DBUG_RETURN(0);
|
||||
}
|
||||
|
||||
/* Load certs from the trusted ca */
|
||||
if (SSL_CTX_load_verify_locations(ssl_fd->ssl_context, ca_file, ca_path) == 0)
|
||||
{
|
||||
DBUG_PRINT("warning", ("SSL_CTX_load_verify_locations failed"));
|
||||
if (SSL_CTX_set_default_verify_paths(ssl_fd->ssl_context) == 0)
|
||||
{
|
||||
*error= SSL_INITERR_BAD_PATHS;
|
||||
DBUG_PRINT("error", ("%s", sslGetErrString(*error)));
|
||||
report_errors();
|
||||
SSL_CTX_free(ssl_fd->ssl_context);
|
||||
my_free(ssl_fd);
|
||||
DBUG_RETURN(0);
|
||||
}
|
||||
}
|
||||
|
||||
if (vio_set_cert_stuff(ssl_fd->ssl_context, cert_file, key_file, error))
|
||||
{
|
||||
DBUG_PRINT("error", ("vio_set_cert_stuff failed"));
|
||||
report_errors();
|
||||
SSL_CTX_free(ssl_fd->ssl_context);
|
||||
my_free(ssl_fd);
|
||||
DBUG_RETURN(0);
|
||||
}
|
||||
|
||||
/* DH stuff */
|
||||
dh=get_dh512();
|
||||
SSL_CTX_set_tmp_dh(ssl_fd->ssl_context, dh);
|
||||
DH_free(dh);
|
||||
|
||||
DBUG_PRINT("exit", ("OK 1"));
|
||||
|
||||
DBUG_RETURN(ssl_fd);
|
||||
}
|
||||
|
||||
|
||||
/************************ VioSSLConnectorFd **********************************/
|
||||
struct st_VioSSLFd *
|
||||
new_VioSSLConnectorFd(const char *key_file, const char *cert_file,
|
||||
const char *ca_file, const char *ca_path,
|
||||
const char *cipher)
|
||||
{
|
||||
struct st_VioSSLFd *ssl_fd;
|
||||
int verify= SSL_VERIFY_PEER;
|
||||
enum enum_ssl_init_error dummy;
|
||||
|
||||
/*
|
||||
Turn off verification of servers certificate if both
|
||||
ca_file and ca_path is set to NULL
|
||||
*/
|
||||
if (ca_file == 0 && ca_path == 0)
|
||||
verify= SSL_VERIFY_NONE;
|
||||
|
||||
if (!(ssl_fd= new_VioSSLFd(key_file, cert_file, ca_file,
|
||||
ca_path, cipher, TLSv1_client_method(), &dummy)))
|
||||
{
|
||||
return 0;
|
||||
}
|
||||
|
||||
/* Init the VioSSLFd as a "connector" ie. the client side */
|
||||
|
||||
SSL_CTX_set_verify(ssl_fd->ssl_context, verify, NULL);
|
||||
|
||||
return ssl_fd;
|
||||
}
|
||||
|
||||
|
||||
/************************ VioSSLAcceptorFd **********************************/
|
||||
struct st_VioSSLFd *
|
||||
new_VioSSLAcceptorFd(const char *key_file, const char *cert_file,
|
||||
const char *ca_file, const char *ca_path,
|
||||
const char *cipher, enum enum_ssl_init_error* error)
|
||||
{
|
||||
struct st_VioSSLFd *ssl_fd;
|
||||
int verify= SSL_VERIFY_PEER | SSL_VERIFY_CLIENT_ONCE;
|
||||
if (!(ssl_fd= new_VioSSLFd(key_file, cert_file, ca_file,
|
||||
ca_path, cipher, TLSv1_server_method(), error)))
|
||||
{
|
||||
return 0;
|
||||
}
|
||||
/* Init the the VioSSLFd as a "acceptor" ie. the server side */
|
||||
|
||||
/* Set max number of cached sessions, returns the previous size */
|
||||
SSL_CTX_sess_set_cache_size(ssl_fd->ssl_context, 128);
|
||||
|
||||
SSL_CTX_set_verify(ssl_fd->ssl_context, verify, NULL);
|
||||
|
||||
/*
|
||||
Set session_id - an identifier for this server session
|
||||
Use the ssl_fd pointer
|
||||
*/
|
||||
SSL_CTX_set_session_id_context(ssl_fd->ssl_context,
|
||||
(const unsigned char *)ssl_fd,
|
||||
sizeof(ssl_fd));
|
||||
|
||||
return ssl_fd;
|
||||
}
|
||||
|
||||
void free_vio_ssl_acceptor_fd(struct st_VioSSLFd *fd)
|
||||
{
|
||||
SSL_CTX_free(fd->ssl_context);
|
||||
my_free(fd);
|
||||
}
|
||||
#endif /* HAVE_OPENSSL */
|
||||
Reference in New Issue
Block a user