test-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. #if defined(HAVE_OPENSSL) && !defined(__NETWARE__)
  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,-";
  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
  41. main(int argc, char** argv)
  42. {
  43.   char* server_key = 0, *server_cert = 0;
  44.   char* client_key = 0, *client_cert = 0;
  45.   char* ca_file = 0, *ca_path = 0;
  46.   char* cipher=0;
  47.   int child_pid,sv[2];
  48.   my_bool unused;
  49.   struct st_VioSSLAcceptorFd* ssl_acceptor=0;
  50.   struct st_VioSSLConnectorFd* ssl_connector=0; 
  51.   Vio* client_vio=0, *server_vio=0;
  52.   MY_INIT(argv[0]);
  53.   DBUG_PROCESS(argv[0]);
  54.   DBUG_PUSH(default_dbug_option);
  55.   if (argc<5)
  56.   {
  57.     print_usage();
  58.     return 1;
  59.   }
  60.   server_key = argv[1];
  61.   server_cert = argv[2];
  62.   client_key = argv[3];
  63.   client_cert = argv[4];
  64.   if (argc>5)
  65.     ca_file = argv[5];
  66.   if (argc>6)
  67.     ca_path = argv[6];
  68.   printf("Server key/cert : %s/%sn", server_key, server_cert);
  69.   printf("Client key/cert : %s/%sn", client_key, client_cert);
  70.   if (ca_file!=0)
  71.     printf("CAfile          : %sn", ca_file);
  72.   if (ca_path!=0)
  73.     printf("CApath          : %sn", ca_path);
  74.   if (socketpair(PF_UNIX, SOCK_STREAM, IPPROTO_IP, sv)==-1)
  75.     fatal_error("socketpair");
  76.   ssl_acceptor = new_VioSSLAcceptorFd(server_key, server_cert, ca_file,
  77.       ca_path, cipher);
  78.   ssl_connector = new_VioSSLConnectorFd(client_key, client_cert, ca_file,
  79. ca_path, cipher);
  80.   client_vio = (struct st_vio*)my_malloc(sizeof(struct st_vio),MYF(0));
  81.   client_vio->sd = sv[0];
  82.   client_vio->vioblocking(client_vio, 0, &unused);
  83.   sslconnect(ssl_connector,client_vio,60L);
  84.   server_vio = (struct st_vio*)my_malloc(sizeof(struct st_vio),MYF(0));
  85.   server_vio->sd = sv[1];
  86.   server_vio->vioblocking(client_vio, 0, &unused);
  87.   sslaccept(ssl_acceptor,server_vio,60L);
  88.   printf("Socketpair: %d , %dn", client_vio->sd, server_vio->sd);
  89.   child_pid = fork();
  90.   if (child_pid==-1) {
  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_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.   }
  111.   else
  112.   {
  113.     const char* s = "Huhuhuh";
  114.     int r = vio_write(server_vio,(gptr)s, strlen(s));
  115.     if (r<=0) {
  116.       my_free((gptr)ssl_acceptor,MYF(0));
  117.       my_free((gptr)ssl_connector,MYF(0));
  118.       fatal_error("server:SSL_write");
  119.     }
  120.     my_free((gptr)server_vio,MYF(0));
  121.     my_free((gptr)ssl_acceptor,MYF(0));
  122.     my_free((gptr)ssl_connector,MYF(0));
  123.   }
  124.   return 0;
  125. }
  126. #else /* HAVE_OPENSSL */
  127. int main() {
  128. return 0;
  129. }
  130. #endif /* HAVE_OPENSSL */