viotest-ssl.c
上传用户:romrleung
上传日期:2022-05-23
资源大小:18897k
文件大小:4k
源码类别:

MySQL数据库

开发平台:

Visual C++

  1. /* Copyright (C) 2000 MySQL AB
  2.    This program is free software; you can redistribute it and/or modify
  3.    it under the terms of the GNU General Public License as published by
  4.    the Free Software Foundation; either version 2 of the License, or
  5.    (at your option) any later version.
  6.    This program is distributed in the hope that it will be useful,
  7.    but WITHOUT ANY WARRANTY; without even the implied warranty of
  8.    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  9.    GNU General Public License for more details.
  10.    You should have received a copy of the GNU General Public License
  11.    along with this program; if not, write to the Free Software
  12.    Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA */
  13. #include <my_global.h>
  14. #ifdef HAVE_OPENSSL
  15. #include <my_sys.h>
  16. #include <m_string.h>
  17. #include <m_ctype.h>
  18. #include "mysql.h"
  19. #include "errmsg.h"
  20. #include <my_dir.h>
  21. #include <my_getopt.h>
  22. #include <signal.h>
  23. #include <violite.h>
  24. const char *VER="0.2";
  25. #ifndef DBUG_OFF
  26. const char *default_dbug_option="d:t:O,/tmp/viotest-ssl.trace";
  27. #endif
  28. void
  29. fatal_error(const char *r)
  30. {
  31.   perror(r);
  32.   exit(0);
  33. }
  34. void
  35. print_usage()
  36. {
  37.   printf("viossl-test: testing SSL virtual IO. Usage:n");
  38.   printf("viossl-test server-key server-cert client-key client-cert [CAfile] [CApath]n");
  39. }
  40. int main(int argc, char **argv)
  41. {
  42.   char* server_key = 0;
  43.   char* server_cert = 0;
  44.   char* client_key = 0;
  45.   char* client_cert = 0;
  46.   char* ca_file = 0;
  47.   char* ca_path = 0;
  48.   int child_pid,sv[2];
  49.   struct st_VioSSLAcceptorFd* ssl_acceptor=0;
  50.   struct st_VioSSLConnectorFd* ssl_connector=0; 
  51.   Vio* client_vio=0;
  52.   Vio* server_vio=0;
  53.   MY_INIT(argv[0]);
  54.   DBUG_PROCESS(argv[0]);
  55.   DBUG_PUSH(default_dbug_option);
  56.   if (argc<5)
  57.   {
  58.     print_usage();
  59.     return 1;
  60.   }
  61.   server_key = argv[1];
  62.   server_cert = argv[2];
  63.   client_key = argv[3];
  64.   client_cert = argv[4];
  65.   if (argc>5)
  66.     ca_file = argv[5];
  67.   if (argc>6)
  68.     ca_path = argv[6];
  69.   printf("Server key/cert : %s/%sn", server_key, server_cert);
  70.   printf("Client key/cert : %s/%sn", client_key, client_cert);
  71.   if (ca_file!=0)
  72.     printf("CAfile          : %sn", ca_file);
  73.   if (ca_path!=0)
  74.     printf("CApath          : %sn", ca_path);
  75.   if (socketpair(PF_UNIX, SOCK_STREAM, IPPROTO_IP, sv)==-1)
  76.     fatal_error("socketpair");
  77.   ssl_acceptor = new_VioSSLAcceptorFd(server_key, server_cert, ca_file,
  78.       ca_path);
  79.   ssl_connector = new_VioSSLConnectorFd(client_key, client_cert, ca_file,
  80. ca_path);
  81.   client_vio = (Vio*)my_malloc(sizeof(struct st_vio),MYF(0));
  82.   client_vio->sd = sv[0];
  83.   sslconnect(ssl_connector,client_vio);
  84.   server_vio = (Vio*)my_malloc(sizeof(struct st_vio),MYF(0));
  85.   server_vio->sd = sv[1];
  86.   sslaccept(ssl_acceptor,server_vio);
  87.   printf("Socketpair: %d , %dn", client_vio->sd, server_vio->sd);
  88.   child_pid = fork();
  89.   if (child_pid==-1)
  90.   {
  91.     my_free((gptr)ssl_acceptor,MYF(0));
  92.     my_free((gptr)ssl_connector,MYF(0));
  93.     fatal_error("fork");
  94.   }
  95.   if (child_pid==0)
  96.   {
  97.     /* child, therefore, client */
  98.     char xbuf[100];
  99.     int r = vio_ssl_read(client_vio,xbuf, sizeof(xbuf));
  100.     if (r<=0) {
  101.       my_free((gptr)ssl_acceptor,MYF(0));
  102.       my_free((gptr)ssl_connector,MYF(0));
  103.       fatal_error("client:SSL_read");
  104.     }
  105.     xbuf[r] = 0;
  106.     printf("client:got %sn", xbuf);
  107.     my_free((gptr)client_vio,MYF(0));
  108.     my_free((gptr)ssl_acceptor,MYF(0));
  109.     my_free((gptr)ssl_connector,MYF(0));
  110.     sleep(1);
  111.   }
  112.   else
  113.   {
  114.     const char* s = "Huhuhuh";
  115.     int r = vio_ssl_write(server_vio,(gptr)s, strlen(s));
  116.     if (r<=0) {
  117.       my_free((gptr)ssl_acceptor,MYF(0));
  118.       my_free((gptr)ssl_connector,MYF(0));
  119.       fatal_error("server:SSL_write");
  120.     }
  121.     my_free((gptr)server_vio,MYF(0));
  122.     my_free((gptr)ssl_acceptor,MYF(0));
  123.     my_free((gptr)ssl_connector,MYF(0));
  124.     sleep(1);
  125.   }
  126.   return 0;
  127. }
  128. #else /* HAVE_OPENSSL */
  129. int main() {
  130. return 0;
  131. }
  132. #endif /* HAVE_OPENSSL */