1 /* Copyright (c) 2000, 2011, Oracle and/or its affiliates. All rights reserved.
2
3 This program is free software; you can redistribute it and/or modify
4 it under the terms of the GNU General Public License as published by
5 the Free Software Foundation; version 2 of the License.
6
7 This program is distributed in the hope that it will be useful,
8 but WITHOUT ANY WARRANTY; without even the implied warranty of
9 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
10 GNU General Public License for more details.
11
12 You should have received a copy of the GNU General Public License
13 along with this program; if not, write to the Free Software
14 Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1335 USA */
15
16 #include <my_global.h>
17 #if defined(HAVE_OPENSSL)
18 #include <my_sys.h>
19 #include <m_string.h>
20 #include <m_ctype.h>
21 #include "mysql.h"
22 #include "errmsg.h"
23 #include <my_dir.h>
24 #include <my_getopt.h>
25 #include <signal.h>
26 #include <violite.h>
27
28 const char *VER="0.2";
29
30
31 #ifndef DBUG_OFF
32 const char *default_dbug_option="d:t:O,-";
33 #endif
34
35 void
fatal_error(const char * r)36 fatal_error( const char* r)
37 {
38 perror(r);
39 exit(0);
40 }
41
42 void
print_usage()43 print_usage()
44 {
45 printf("viossl-test: testing SSL virtual IO. Usage:\n");
46 printf("viossl-test server-key server-cert client-key client-cert [CAfile] [CApath]\n");
47 }
48
49
50 int
main(int argc,char ** argv)51 main(int argc, char** argv)
52 {
53 char* server_key = 0, *server_cert = 0;
54 char* client_key = 0, *client_cert = 0;
55 char* ca_file = 0, *ca_path = 0;
56 char* cipher=0;
57 int child_pid,sv[2];
58 my_bool unused;
59 struct st_VioSSLFd* ssl_acceptor= 0;
60 struct st_VioSSLFd* ssl_connector= 0;
61 Vio* client_vio=0, *server_vio=0;
62 enum enum_ssl_init_error ssl_init_error;
63 unsigned long ssl_error;
64
65 MY_INIT(argv[0]);
66 DBUG_PROCESS(argv[0]);
67 DBUG_PUSH(default_dbug_option);
68
69 if (argc<5)
70 {
71 print_usage();
72 return 1;
73 }
74
75 server_key = argv[1];
76 server_cert = argv[2];
77 client_key = argv[3];
78 client_cert = argv[4];
79 if (argc>5)
80 ca_file = argv[5];
81 if (argc>6)
82 ca_path = argv[6];
83 printf("Server key/cert : %s/%s\n", server_key, server_cert);
84 printf("Client key/cert : %s/%s\n", client_key, client_cert);
85 if (ca_file!=0)
86 printf("CAfile : %s\n", ca_file);
87 if (ca_path!=0)
88 printf("CApath : %s\n", ca_path);
89
90
91 if (socketpair(PF_UNIX, SOCK_STREAM, IPPROTO_IP, sv)==-1)
92 fatal_error("socketpair");
93
94 ssl_acceptor = new_VioSSLAcceptorFd(server_key, server_cert, ca_file,
95 ca_path, cipher);
96 ssl_connector = new_VioSSLConnectorFd(client_key, client_cert, ca_file,
97 ca_path, cipher, &ssl_init_error);
98
99 client_vio = (struct st_vio*)my_malloc(sizeof(struct st_vio),MYF(0));
100 client_vio->sd = sv[0];
101 client_vio->vioblocking(client_vio, 0, &unused);
102 sslconnect(ssl_connector,client_vio,60L,&ssl_error);
103 server_vio = (struct st_vio*)my_malloc(sizeof(struct st_vio),MYF(0));
104 server_vio->sd = sv[1];
105 server_vio->vioblocking(client_vio, 0, &unused);
106 sslaccept(ssl_acceptor,server_vio,60L, &ssl_error);
107
108 printf("Socketpair: %d , %d\n", client_vio->sd, server_vio->sd);
109
110 child_pid = fork();
111 if (child_pid==-1) {
112 my_free(ssl_acceptor);
113 my_free(ssl_connector);
114 fatal_error("fork");
115 }
116 if (child_pid==0)
117 {
118 /* child, therefore, client */
119 char xbuf[100];
120 int r = vio_read(client_vio,xbuf, sizeof(xbuf));
121 if (r<=0) {
122 my_free(ssl_acceptor);
123 my_free(ssl_connector);
124 fatal_error("client:SSL_read");
125 }
126 xbuf[r] = 0;
127 printf("client:got %s\n", xbuf);
128 my_free(client_vio);
129 my_free(ssl_acceptor);
130 my_free(ssl_connector);
131 }
132 else
133 {
134 const char* s = "Huhuhuh";
135 int r = vio_write(server_vio,(uchar*)s, strlen(s));
136 if (r<=0) {
137 my_free(ssl_acceptor);
138 my_free(ssl_connector);
139 fatal_error("server:SSL_write");
140 }
141 my_free(server_vio);
142 my_free(ssl_acceptor);
143 my_free(ssl_connector);
144 }
145 return 0;
146 }
147 #else /* HAVE_OPENSSL */
148
main()149 int main() {
150 return 0;
151 }
152 #endif /* HAVE_OPENSSL */
153