mariadb/vio/viotest-sslconnect.cc

/* Copyright (c) 2000 MySQL AB
   Use is subject to license terms.

   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., 51 Franklin St, Fifth Floor, Boston, MA 02110-1335  USA */

/* 
**  Virtual I/O library
**  Written by Andrei Errapart <andreie@no.spam.ee>
*/

#include	"all.h"

#include	<sys/types.h>
#include	<sys/socket.h>
#include	<netinet/in.h>
#include	<stdio.h>
#include	<unistd.h>


void
fatal_error(	const char*	r)
{
	perror(r);
	exit(0);
}

void
print_usage()
{
	printf("viotest-sslconnect: testing SSL virtual IO. Usage:\n");
	printf("viotest-sslconnect key cert\n");
}

int
main(	int	argc,
	char**	argv)
{
	char*	key = 0;
	char*	cert = 0;

	if (argc<3)
	{
		print_usage();
		return 1;
	}

	char		ip[4] = {127, 0, 0, 1};
	unsigned long	addr = (unsigned long)
			((unsigned long)ip[0]<<24L)|
			((unsigned long)ip[1]<<16L)|
			((unsigned long)ip[2]<< 8L)|
			((unsigned long)ip[3]);
	int	fd = socket(PF_INET, SOCK_STREAM, IPPROTO_TCP);
	if (fd<0)
		fatal_error("socket");
	struct sockaddr_in	sa;
	sa.sin_family = AF_INET;
	sa.sin_port=htons(4433);
	sa.sin_addr.s_addr=htonl(addr);
	int	sa_size = sizeof sa;
	if (connect(fd, reinterpret_cast<const sockaddr*>(&sa), sa_size)==-1)
		fatal_error("connect");
	key = argv[1];
	cert = argv[2];
	printf("Key  : %s\n", key);
	printf("Cert : %s\n", cert);

	VIO_NS::VioSSLConnectorFd*	ssl_connector = new VIO_NS::VioSSLConnectorFd(cert, key,0,0);

	VIO_NS::VioSSL*	vio = ssl_connector->connect(fd);

	char	xbuf[100];
	int	r = vio->read(xbuf, sizeof(xbuf));
	if (r<=0) {
		delete ssl_connector;
		delete vio;
		fatal_error("client:SSL_read");
	}
	xbuf[r] = 0;
	printf("client:got %s\n", xbuf);
	delete vio;
	delete ssl_connector;
	return 0;
}