mini_client.cc
上传用户:tsgydb
上传日期:2007-04-14
资源大小:10674k
文件大小:23k
源码类别:

MySQL数据库

开发平台:

Visual C++

  1. /* Copyright (C) 2000 MySQL AB & MySQL Finland AB & TCX DataKonsult AB
  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; either version 2 of the License, or
  6.    (at your option) any later version.
  7.    
  8.    This program is distributed in the hope that it will be useful,
  9.    but WITHOUT ANY WARRANTY; without even the implied warranty of
  10.    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  11.    GNU General Public License for more details.
  12.    
  13.    You should have received a copy of the GNU General Public License
  14.    along with this program; if not, write to the Free Software
  15.    Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA */
  16. /*
  17.  mini MySQL client to be included into the server to do server to server
  18.  commincation by Sasha Pachev
  19.  Note: all file-global symbols must begin with mc_ , even the static ones, just
  20.  in case we decide to make them external at some point
  21.  */
  22. #define DONT_USE_RAID
  23. #if defined(__WIN__) || defined(WIN32)
  24. #include <winsock.h>
  25. #include <odbcinst.h>
  26. #endif
  27. #include <global.h>
  28. #include <my_sys.h>
  29. #include <mysys_err.h>
  30. #include <m_string.h>
  31. #include <m_ctype.h>
  32. #include "mysql.h"
  33. #include "mini_client.h"
  34. #include "mysql_version.h"
  35. #include "mysqld_error.h"
  36. #include "errmsg.h"
  37. #include <violite.h>
  38. extern "C" { // Because of SCO 3.2V4.2
  39. #include <sys/stat.h>
  40. #include <signal.h>
  41. #ifdef  HAVE_PWD_H
  42. #include <pwd.h>
  43. #endif
  44. #if !defined(MSDOS) && !defined(__WIN__)
  45. #include <sys/socket.h>
  46. #include <netinet/in.h>
  47. #include <arpa/inet.h>
  48. #include <netdb.h>
  49. #ifdef HAVE_SELECT_H
  50. #  include <select.h>
  51. #endif
  52. #ifdef HAVE_SYS_SELECT_H
  53. #include <sys/select.h>
  54. #endif
  55. #endif
  56. #ifdef HAVE_SYS_UN_H
  57. #  include <sys/un.h>
  58. #endif
  59. #if defined(THREAD) && !defined(__WIN__)
  60. #include <my_pthread.h> /* because of signal() */
  61. #endif
  62. #ifndef INADDR_NONE
  63. #define INADDR_NONE -1
  64. #endif
  65. }
  66. static void mc_end_server(MYSQL *mysql);
  67. static int mc_sock_connect(File s, const struct sockaddr *name, uint namelen, uint to);
  68. static void mc_free_old_query(MYSQL *mysql);
  69. #define CLIENT_CAPABILITIES (CLIENT_LONG_PASSWORD | CLIENT_LONG_FLAG | CLIENT_LOCAL_FILES)
  70. #if defined(MSDOS) || defined(__WIN__)
  71. #define ERRNO WSAGetLastError()
  72. #define perror(A)
  73. #else
  74. #include <sys/errno.h>
  75. #define ERRNO errno
  76. #define SOCKET_ERROR -1
  77. #define closesocket(A) close(A)
  78. #endif
  79. #ifdef __WIN__
  80. static my_bool is_NT(void)
  81. {
  82.   char *os=getenv("OS");
  83.   return (os && !strcmp(os, "Windows_NT")) ? 1 : 0;
  84. }
  85. #endif
  86. /*
  87. ** Create a named pipe connection
  88. */
  89. #ifdef __WIN__
  90. HANDLE create_named_pipe(NET *net, uint connect_timeout, char **arg_host,
  91.  char **arg_unix_socket)
  92. {
  93.   HANDLE hPipe=INVALID_HANDLE_VALUE;
  94.   char szPipeName [ 257 ];
  95.   DWORD dwMode;
  96.   int i;
  97.   my_bool testing_named_pipes=0;
  98.   char *host= *arg_host, *unix_socket= *arg_unix_socket;
  99.   if (!host || !strcmp(host,LOCAL_HOST))
  100.     host=LOCAL_HOST_NAMEDPIPE;
  101.   sprintf( szPipeName, "\\%s\pipe\%s", host, unix_socket);
  102.   DBUG_PRINT("info",("Server name: '%s'.  Named Pipe: %s",
  103.      host, unix_socket));
  104.   for (i=0 ; i < 100 ; i++) /* Don't retry forever */
  105.   {
  106.     if ((hPipe = CreateFile(szPipeName,
  107.     GENERIC_READ | GENERIC_WRITE,
  108.     0,
  109.     NULL,
  110.     OPEN_EXISTING,
  111.     0,
  112.     NULL )) != INVALID_HANDLE_VALUE)
  113.       break;
  114.     if (GetLastError() != ERROR_PIPE_BUSY)
  115.     {
  116.       net->last_errno=CR_NAMEDPIPEOPEN_ERROR;
  117.       sprintf(net->last_error,ER(net->last_errno),host, unix_socket,
  118.       (ulong) GetLastError());
  119.       return INVALID_HANDLE_VALUE;
  120.     }
  121.     /* wait for for an other instance */
  122.     if (! WaitNamedPipe(szPipeName, connect_timeout*1000) )
  123.     {
  124.       net->last_errno=CR_NAMEDPIPEWAIT_ERROR;
  125.       sprintf(net->last_error,ER(net->last_errno),host, unix_socket,
  126.       (ulong) GetLastError());
  127.       return INVALID_HANDLE_VALUE;
  128.     }
  129.   }
  130.   if (hPipe == INVALID_HANDLE_VALUE)
  131.   {
  132.     net->last_errno=CR_NAMEDPIPEOPEN_ERROR;
  133.     sprintf(net->last_error,ER(net->last_errno),host, unix_socket,
  134.     (ulong) GetLastError());
  135.     return INVALID_HANDLE_VALUE;
  136.   }
  137.   dwMode = PIPE_READMODE_BYTE | PIPE_WAIT;
  138.   if ( !SetNamedPipeHandleState(hPipe, &dwMode, NULL, NULL) )
  139.   {
  140.     CloseHandle( hPipe );
  141.     net->last_errno=CR_NAMEDPIPESETSTATE_ERROR;
  142.     sprintf(net->last_error,ER(net->last_errno),host, unix_socket,
  143.     (ulong) GetLastError());
  144.     return INVALID_HANDLE_VALUE;
  145.   }
  146.   *arg_host=host ; *arg_unix_socket=unix_socket; /* connect arg */
  147.   return (hPipe);
  148. }
  149. #endif
  150. /****************************************************************************
  151. ** Init MySQL structure or allocate one
  152. ****************************************************************************/
  153. MYSQL * STDCALL
  154. mc_mysql_init(MYSQL *mysql)
  155. {
  156.   init_client_errs();
  157.   if (!mysql)
  158.   {
  159.     if (!(mysql=(MYSQL*) my_malloc(sizeof(*mysql),MYF(MY_WME | MY_ZEROFILL))))
  160.       return 0;
  161.     mysql->free_me=1;
  162.     mysql->net.vio = 0;
  163.   }
  164.   else
  165.     bzero((char*) (mysql),sizeof(*(mysql)));
  166. #ifdef __WIN__
  167.   mysql->options.connect_timeout=20;
  168. #endif
  169.   return mysql;
  170. }
  171. /**************************************************************************
  172. ** Shut down connection
  173. **************************************************************************/
  174. static void
  175. mc_end_server(MYSQL *mysql)
  176. {
  177.   DBUG_ENTER("mc_end_server");
  178.   if (mysql->net.vio != 0)
  179.   {
  180.     DBUG_PRINT("info",("Net: %s", vio_description(mysql->net.vio)));
  181.     vio_delete(mysql->net.vio);
  182.     mysql->net.vio= 0;          /* Marker */
  183.   }
  184.   net_end(&mysql->net);
  185.   mc_free_old_query(mysql);
  186.   DBUG_VOID_RETURN;
  187. }
  188. static void mc_free_old_query(MYSQL *mysql)
  189. {
  190.   DBUG_ENTER("mc_free_old_query");
  191.   if (mysql->fields)
  192.     free_root(&mysql->field_alloc,MYF(0));
  193.   else
  194.     init_alloc_root(&mysql->field_alloc,8192,0); /* Assume rowlength < 8192 */
  195.   mysql->fields=0;
  196.   mysql->field_count=0; /* For API */
  197.   DBUG_VOID_RETURN;
  198. }
  199. /****************************************************************************
  200. * A modified version of connect().  mc_sock_connect() allows you to specify
  201. * a timeout value, in seconds, that we should wait until we
  202. * derermine we can't connect to a particular host.  If timeout is 0,
  203. * mc_sock_connect() will behave exactly like connect().
  204. *
  205. * Base version coded by Steve Bernacki, Jr. <steve@navinet.net>
  206. *****************************************************************************/
  207. static int mc_sock_connect(my_socket s, const struct sockaddr *name,
  208.    uint namelen, uint to)
  209. {
  210. #if defined(__WIN__)
  211.   return connect(s, (struct sockaddr*) name, namelen);
  212. #else
  213.   int flags, res, s_err;
  214.   SOCKOPT_OPTLEN_TYPE s_err_size = sizeof(uint);
  215.   fd_set sfds;
  216.   struct timeval tv;
  217.   /* If they passed us a timeout of zero, we should behave
  218.    * exactly like the normal connect() call does.
  219.    */
  220.   if (to == 0)
  221.     return connect(s, (struct sockaddr*) name, namelen);
  222.   flags = fcntl(s, F_GETFL, 0);   /* Set socket to not block */
  223. #ifdef O_NONBLOCK
  224.   fcntl(s, F_SETFL, flags | O_NONBLOCK);  /* and save the flags..  */
  225. #endif
  226.   res = connect(s, (struct sockaddr*) name, namelen);
  227.   s_err = errno; /* Save the error... */
  228.   fcntl(s, F_SETFL, flags);
  229.   if ((res != 0) && (s_err != EINPROGRESS))
  230.   {
  231.     errno = s_err; /* Restore it */
  232.     return(-1);
  233.   }
  234.   if (res == 0) /* Connected quickly! */
  235.     return(0);
  236.   /* Otherwise, our connection is "in progress."  We can use
  237.    * the select() call to wait up to a specified period of time
  238.    * for the connection to suceed.  If select() returns 0
  239.    * (after waiting howevermany seconds), our socket never became
  240.    * writable (host is probably unreachable.)  Otherwise, if
  241.    * select() returns 1, then one of two conditions exist:
  242.    *
  243.    * 1. An error occured.  We use getsockopt() to check for this.
  244.    * 2. The connection was set up sucessfully: getsockopt() will
  245.    * return 0 as an error.
  246.    *
  247.    * Thanks goes to Andrew Gierth <andrew@erlenstar.demon.co.uk>
  248.    * who posted this method of timing out a connect() in
  249.    * comp.unix.programmer on August 15th, 1997.
  250.    */
  251.   FD_ZERO(&sfds);
  252.   FD_SET(s, &sfds);
  253.   tv.tv_sec = (long) to;
  254.   tv.tv_usec = 0;
  255. #ifdef HPUX
  256.   res = select(s+1, NULL, (int*) &sfds, NULL, &tv);
  257. #else
  258.   res = select(s+1, NULL, &sfds, NULL, &tv);
  259. #endif
  260.   if (res <= 0) /* Never became writable */
  261.     return(-1);
  262.   /* select() returned something more interesting than zero, let's
  263.    * see if we have any errors.  If the next two statements pass,
  264.    * we've got an open socket!
  265.    */
  266.   s_err=0;
  267.   if (getsockopt(s, SOL_SOCKET, SO_ERROR, (char*) &s_err, &s_err_size) != 0)
  268.     return(-1);
  269.   if (s_err)
  270.   { // getsockopt() could succeed
  271.     errno = s_err;
  272.     return(-1); // but return an error...
  273.   }
  274.   return(0); /* It's all good! */
  275. #endif
  276. }
  277. /*****************************************************************************
  278. ** read a packet from server. Give error message if socket was down
  279. ** or packet is an error message
  280. *****************************************************************************/
  281. uint STDCALL
  282. mc_net_safe_read(MYSQL *mysql)
  283. {
  284.   NET *net= &mysql->net;
  285.   uint len=0;
  286.   if (net->vio != 0)
  287.     len=my_net_read(net);
  288.   if (len == packet_error || len == 0)
  289.   {
  290.     DBUG_PRINT("error",("Wrong connection or packet. fd: %s  len: %d",
  291. vio_description(net->vio),len));
  292.     if(errno != EINTR)
  293.       {
  294.         mc_end_server(mysql);
  295.         net->last_errno=CR_SERVER_LOST;
  296.         strmov(net->last_error,ER(net->last_errno));
  297.       }
  298.     return(packet_error);
  299.   }
  300.   if (net->read_pos[0] == 255)
  301.   {
  302.     if (len > 3)
  303.     {
  304.       char *pos=(char*) net->read_pos+1;
  305.       if (mysql->protocol_version > 9)
  306.       { /* New client protocol */
  307. net->last_errno=uint2korr(pos);
  308. pos+=2;
  309. len-=2;
  310. if(!net->last_errno)
  311.   net->last_errno = CR_UNKNOWN_ERROR;
  312.       }
  313.       else
  314.       {
  315. net->last_errno=CR_UNKNOWN_ERROR;
  316. len--;
  317.       }
  318.       (void) strmake(net->last_error,(char*) pos,
  319.      min(len,sizeof(net->last_error)-1));
  320.     }
  321.     else
  322.     {
  323.       net->last_errno=CR_UNKNOWN_ERROR;
  324.       (void) strmov(net->last_error,ER(net->last_errno));
  325.     }
  326.     DBUG_PRINT("error",("Got error: %d (%s)", net->last_errno,
  327. net->last_error));
  328.     return(packet_error);
  329.   }
  330.   return len;
  331. }
  332. char * STDCALL mc_mysql_error(MYSQL *mysql)
  333. {
  334.   return (mysql)->net.last_error;
  335. }
  336. int STDCALL mc_mysql_errno(MYSQL *mysql)
  337. {
  338.   return (mysql)->net.last_errno;
  339. }
  340. my_bool STDCALL mc_mysql_reconnect(MYSQL *mysql)
  341. {
  342.   MYSQL tmp_mysql;
  343.   DBUG_ENTER("mc_mysql_reconnect");
  344.   mc_mysql_init(&tmp_mysql);
  345.   tmp_mysql.options=mysql->options;
  346.   if (!mc_mysql_connect(&tmp_mysql,mysql->host,mysql->user,mysql->passwd,
  347.   mysql->db, mysql->port, mysql->unix_socket,
  348.   mysql->client_flag))
  349.     {
  350.       mc_mysql_close(&tmp_mysql); 
  351.       DBUG_RETURN(1);
  352.     }
  353.   tmp_mysql.free_me=mysql->free_me;
  354.   mysql->free_me=0;
  355.   bzero((char*) &mysql->options,sizeof(&mysql->options));
  356.   mc_mysql_close(mysql);
  357.   *mysql=tmp_mysql;
  358.   net_clear(&mysql->net);
  359.   mysql->affected_rows= ~(my_ulonglong) 0;
  360.   DBUG_RETURN(0);
  361. }
  362. int STDCALL
  363. mc_simple_command(MYSQL *mysql,enum enum_server_command command,
  364.   const char *arg, uint length, my_bool skipp_check)
  365. {
  366.   NET *net= &mysql->net;
  367.   int result= -1;
  368.   if (mysql->net.vio == 0)
  369.   { /* Do reconnect if possible */
  370.     if (mc_mysql_reconnect(mysql))
  371.     {
  372.       net->last_errno=CR_SERVER_GONE_ERROR;
  373.       strmov(net->last_error,ER(net->last_errno));
  374.       goto end;
  375.     }
  376.   }
  377.   if (mysql->status != MYSQL_STATUS_READY)
  378.   {
  379.     strmov(net->last_error,ER(mysql->net.last_errno=CR_COMMANDS_OUT_OF_SYNC));
  380.     goto end;
  381.   }
  382.   mysql->net.last_error[0]=0;
  383.   mysql->net.last_errno=0;
  384.   mysql->info=0;
  385.   mysql->affected_rows= ~(my_ulonglong) 0;
  386.   net_clear(net); /* Clear receive buffer */
  387.   if (!arg)
  388.     arg="";
  389.   if (net_write_command(net,(uchar) command,arg,
  390. length ? length :(uint) strlen(arg)))
  391.   {
  392.     DBUG_PRINT("error",("Can't send command to server. Error: %d",errno));
  393.     mc_end_server(mysql);
  394.     if (mc_mysql_reconnect(mysql) ||
  395. net_write_command(net,(uchar) command,arg,
  396.   length ? length :(uint) strlen(arg)))
  397.     {
  398.       net->last_errno=CR_SERVER_GONE_ERROR;
  399.       strmov(net->last_error,ER(net->last_errno));
  400.       goto end;
  401.     }
  402.   }
  403.   result=0;
  404.   if (!skipp_check)
  405.     result= ((mysql->packet_length=mc_net_safe_read(mysql)) == packet_error ?
  406.      -1 : 0);
  407.  end:
  408.   return result;
  409. }
  410. MYSQL * STDCALL
  411. mc_mysql_connect(MYSQL *mysql,const char *host, const char *user,
  412.    const char *passwd, const char *db,
  413.    uint port, const char *unix_socket,uint client_flag)
  414. {
  415.   char buff[100],*end,*host_info;
  416.   my_socket sock;
  417.   ulong ip_addr;
  418.   struct sockaddr_in sock_addr;
  419.   uint pkt_length;
  420.   NET *net= &mysql->net;
  421. #ifdef __WIN__
  422.   HANDLE hPipe=INVALID_HANDLE_VALUE;
  423. #endif
  424. #ifdef HAVE_SYS_UN_H
  425.   struct sockaddr_un UNIXaddr;
  426. #endif
  427.   DBUG_ENTER("mysql_real_connect");
  428.   DBUG_PRINT("enter",("host: %s  db: %s  user: %s",
  429.       host ? host : "(Null)",
  430.       db ? db : "(Null)",
  431.       user ? user : "(Null)"));
  432.   bzero((char*) &mysql->options,sizeof(mysql->options));
  433.   net->vio = 0; /* If something goes wrong */
  434.   mysql->charset=default_charset_info;  /* Set character set */
  435.   if (!port)
  436.     port = MYSQL_PORT; /* Should always be set by mysqld */
  437.   if (!unix_socket)
  438.     unix_socket=MYSQL_UNIX_ADDR;
  439.   mysql->reconnect=1; /* Reconnect as default */
  440.   /*
  441.   ** Grab a socket and connect it to the server
  442.   */
  443. #if defined(HAVE_SYS_UN_H)
  444.   if (!host || !strcmp(host,LOCAL_HOST))
  445.   {
  446.     host=LOCAL_HOST;
  447.     host_info=(char*) ER(CR_LOCALHOST_CONNECTION);
  448.     DBUG_PRINT("info",("Using UNIX sock '%s'",unix_socket));
  449.     if ((sock = socket(AF_UNIX,SOCK_STREAM,0)) == SOCKET_ERROR)
  450.     {
  451.       net->last_errno=CR_SOCKET_CREATE_ERROR;
  452.       sprintf(net->last_error,ER(net->last_errno),ERRNO);
  453.       goto error;
  454.     }
  455.     net->vio = vio_new(sock, VIO_TYPE_SOCKET, TRUE);
  456.     bzero((char*) &UNIXaddr,sizeof(UNIXaddr));
  457.     UNIXaddr.sun_family = AF_UNIX;
  458.     strmov(UNIXaddr.sun_path, unix_socket);
  459.     if (mc_sock_connect(sock,(struct sockaddr *) &UNIXaddr, sizeof(UNIXaddr),
  460. mysql->options.connect_timeout) <0)
  461.     {
  462.       DBUG_PRINT("error",("Got error %d on connect to local server",ERRNO));
  463.       net->last_errno=CR_CONNECTION_ERROR;
  464.       sprintf(net->last_error,ER(net->last_errno),unix_socket,ERRNO);
  465.       goto error;
  466.     }
  467.   }
  468.   else
  469. #elif defined(__WIN__)
  470.   {
  471.     if ((unix_socket ||
  472.  !host && is_NT() ||
  473.  host && !strcmp(host,LOCAL_HOST_NAMEDPIPE) ||
  474.  mysql->options.named_pipe || !have_tcpip))
  475.     {
  476.       sock=0;
  477.       if ((hPipe=create_named_pipe(net, mysql->options.connect_timeout,
  478.    (char**) &host, (char**) &unix_socket)) ==
  479.   INVALID_HANDLE_VALUE)
  480.       {
  481. DBUG_PRINT("error",
  482.    ("host: '%s'  socket: '%s'  named_pipe: %d  have_tcpip: %d",
  483.     host ? host : "<null>",
  484.     unix_socket ? unix_socket : "<null>",
  485.     (int) mysql->options.named_pipe,
  486.     (int) have_tcpip));
  487. if (mysql->options.named_pipe ||
  488.     (host && !strcmp(host,LOCAL_HOST_NAMEDPIPE)) ||
  489.     (unix_socket && !strcmp(unix_socket,MYSQL_NAMEDPIPE)))
  490.   goto error; /* User only requested named pipes */
  491. /* Try also with TCP/IP */
  492.       }
  493.       else
  494.       {
  495. net->vio=vio_new_win32pipe(hPipe);
  496. sprintf(host_info=buff, ER(CR_NAMEDPIPE_CONNECTION), host,
  497. unix_socket);
  498.       }
  499.     }
  500.   }
  501.   if (hPipe == INVALID_HANDLE_VALUE)
  502. #endif
  503.   {
  504.     unix_socket=0; /* This is not used */
  505.     if (!host)
  506.       host=LOCAL_HOST;
  507.     sprintf(host_info=buff,ER(CR_TCP_CONNECTION),host);
  508.     DBUG_PRINT("info",("Server name: '%s'.  TCP sock: %d", host,port));
  509.     if ((sock = socket(AF_INET,SOCK_STREAM,0)) == SOCKET_ERROR)
  510.     {
  511.       net->last_errno=CR_IPSOCK_ERROR;
  512.       sprintf(net->last_error,ER(net->last_errno),ERRNO);
  513.       goto error;
  514.     }
  515.     net->vio = vio_new(sock,VIO_TYPE_TCPIP,FALSE);
  516.     bzero((char*) &sock_addr,sizeof(sock_addr));
  517.     sock_addr.sin_family = AF_INET;
  518.     /*
  519.     ** The server name may be a host name or IP address
  520.     */
  521.     if ((int) (ip_addr = inet_addr(host)) != (int) INADDR_NONE)
  522.     {
  523.       memcpy_fixed(&sock_addr.sin_addr,&ip_addr,sizeof(ip_addr));
  524.     }
  525.     else
  526. #if defined(HAVE_GETHOSTBYNAME_R) && defined(_REENTRANT) && defined(THREAD)
  527.     {
  528.       int tmp_errno;
  529.       struct hostent tmp_hostent,*hp;
  530.       char buff2[GETHOSTBYNAME_BUFF_SIZE];
  531.       hp = my_gethostbyname_r(host,&tmp_hostent,buff2,sizeof(buff2),
  532.       &tmp_errno);
  533.       if (!hp)
  534.       {
  535. net->last_errno=CR_UNKNOWN_HOST;
  536. sprintf(net->last_error, ER(CR_UNKNOWN_HOST), host, tmp_errno);
  537. goto error;
  538.       }
  539.       memcpy(&sock_addr.sin_addr,hp->h_addr, (size_t) hp->h_length);
  540.     }
  541. #else
  542.     {
  543.       struct hostent *hp;
  544.       if (!(hp=gethostbyname(host)))
  545.       {
  546. net->last_errno=CR_UNKNOWN_HOST;
  547. sprintf(net->last_error, ER(CR_UNKNOWN_HOST), host, errno);
  548. goto error;
  549.       }
  550.       memcpy(&sock_addr.sin_addr,hp->h_addr, (size_t) hp->h_length);
  551.     }
  552. #endif
  553.     sock_addr.sin_port = (ushort) htons((ushort) port);
  554.     if (mc_sock_connect(sock,(struct sockaddr *) &sock_addr, sizeof(sock_addr),
  555.  mysql->options.connect_timeout) <0)
  556.     {
  557.       DBUG_PRINT("error",("Got error %d on connect to '%s'",ERRNO,host));
  558.       net->last_errno= CR_CONN_HOST_ERROR;
  559.       sprintf(net->last_error ,ER(CR_CONN_HOST_ERROR), host, ERRNO);
  560.       goto error;
  561.     }
  562.   }
  563.   if (!net->vio || my_net_init(net, net->vio))
  564.   {
  565.     vio_delete(net->vio);
  566.     net->vio = 0; // safety
  567.     net->last_errno=CR_OUT_OF_MEMORY;
  568.     strmov(net->last_error,ER(net->last_errno));
  569.     goto error;
  570.   }
  571.   vio_keepalive(net->vio,TRUE);
  572.   /* Get version info */
  573.   mysql->protocol_version= PROTOCOL_VERSION; /* Assume this */
  574.   if ((pkt_length=mc_net_safe_read(mysql)) == packet_error)
  575.     goto error;
  576.   /* Check if version of protocoll matches current one */
  577.   mysql->protocol_version= net->read_pos[0];
  578.   DBUG_DUMP("packet",(char*) net->read_pos,10);
  579.   DBUG_PRINT("info",("mysql protocol version %d, server=%d",
  580.      PROTOCOL_VERSION, mysql->protocol_version));
  581.   if (mysql->protocol_version != PROTOCOL_VERSION &&
  582.       mysql->protocol_version != PROTOCOL_VERSION-1)
  583.   {
  584.     net->last_errno= CR_VERSION_ERROR;
  585.     sprintf(net->last_error, ER(CR_VERSION_ERROR), mysql->protocol_version,
  586.     PROTOCOL_VERSION);
  587.     goto error;
  588.   }
  589.   end=strend((char*) net->read_pos+1);
  590.   mysql->thread_id=uint4korr(end+1);
  591.   end+=5;
  592.   strmake(mysql->scramble_buff,end,8);
  593.   if (pkt_length > (uint) (end+9 - (char*) net->read_pos))
  594.     mysql->server_capabilities=uint2korr(end+9);
  595.   /* Save connection information */
  596.   if (!user) user="";
  597.   if (!passwd) passwd="";
  598.   if (!my_multi_malloc(MYF(0),
  599.        &mysql->host_info, (uint) strlen(host_info)+1,
  600.        &mysql->host,      (uint) strlen(host)+1,
  601.        &mysql->unix_socket,
  602.        unix_socket ? (uint) strlen(unix_socket)+1 : (uint) 1,
  603.        &mysql->server_version,
  604.        (uint) (end - (char*) net->read_pos),
  605.        NullS) ||
  606.       !(mysql->user=my_strdup(user,MYF(0))) ||
  607.       !(mysql->passwd=my_strdup(passwd,MYF(0))))
  608.   {
  609.     strmov(net->last_error, ER(net->last_errno=CR_OUT_OF_MEMORY));
  610.     goto error;
  611.   }
  612.   strmov(mysql->host_info,host_info);
  613.   strmov(mysql->host,host);
  614.   if (unix_socket)
  615.     strmov(mysql->unix_socket,unix_socket);
  616.   else
  617.     mysql->unix_socket=0;
  618.   strmov(mysql->server_version,(char*) net->read_pos+1);
  619.   mysql->port=port;
  620.   mysql->client_flag=client_flag | mysql->options.client_flag;
  621.   DBUG_PRINT("info",("Server version = '%s'  capabilites: %ld",
  622.      mysql->server_version,mysql->server_capabilities));
  623.   /* Send client information for access check */
  624.   client_flag|=CLIENT_CAPABILITIES;
  625. #ifdef HAVE_OPENSSL
  626.   if (mysql->options.use_ssl)
  627.     client_flag|=CLIENT_SSL;
  628. #endif /* HAVE_OPENSSL */
  629.   if (db)
  630.     client_flag|=CLIENT_CONNECT_WITH_DB;
  631. #ifdef HAVE_COMPRESS
  632.   if (mysql->server_capabilities & CLIENT_COMPRESS &&
  633.       (mysql->options.compress || client_flag & CLIENT_COMPRESS))
  634.     client_flag|=CLIENT_COMPRESS; /* We will use compression */
  635.   else
  636. #endif
  637.     client_flag&= ~CLIENT_COMPRESS;
  638. #ifdef HAVE_OPENSSL
  639.   if ((mysql->server_capabilities & CLIENT_SSL) &&
  640.       (mysql->options.use_ssl || (client_flag & CLIENT_SSL)))
  641.   {
  642.     DBUG_PRINT("info", ("Changing IO layer to SSL"));
  643.     client_flag |= CLIENT_SSL;
  644.   }
  645.   else
  646.   {
  647.     if (client_flag & CLIENT_SSL)
  648.     {
  649.       DBUG_PRINT("info", ("Leaving IO layer intact because server doesn't support SSL"));
  650.     }
  651.     client_flag &= ~CLIENT_SSL;
  652.   }
  653. #endif /* HAVE_OPENSSL */
  654.   int2store(buff,client_flag);
  655.   mysql->client_flag=client_flag;
  656. #ifdef HAVE_OPENSSL
  657.   /* Oops.. are we careful enough to not send ANY information */
  658.   /* without encryption? */
  659.   if (client_flag & CLIENT_SSL)
  660.   {
  661.     if (my_net_write(net,buff,(uint) (2)) || net_flush(net))
  662.       goto error;
  663.     /* Do the SSL layering. */
  664.     DBUG_PRINT("info", ("IO layer change in progress..."));
  665.     VioSSLConnectorFd* connector_fd = (VioSSLConnectorFd*)
  666.       (mysql->connector_fd);
  667.     VioSocket* vio_socket = (VioSocket*)(mysql->net.vio);
  668.     VioSSL* vio_ssl =    connector_fd->connect(vio_socket);
  669.     mysql->net.vio =         (NetVio*)(vio_ssl);
  670.   }
  671. #endif /* HAVE_OPENSSL */
  672.   int3store(buff+2,max_allowed_packet);
  673.   if (user && user[0])
  674.     strmake(buff+5,user,32);
  675.   else
  676.     {
  677.       user = getenv("USER");
  678.       if(!user) user = "mysql";
  679.        strmov((char*) buff+5, user );
  680.     }
  681.   DBUG_PRINT("info",("user: %s",buff+5));
  682.   end=scramble(strend(buff+5)+1, mysql->scramble_buff, passwd,
  683.        (my_bool) (mysql->protocol_version == 9));
  684.   if (db)
  685.   {
  686.     end=strmov(end+1,db);
  687.     mysql->db=my_strdup(db,MYF(MY_WME));
  688.   }
  689.   if (my_net_write(net,buff,(uint) (end-buff)) || net_flush(net) ||
  690.       mc_net_safe_read(mysql) == packet_error)
  691.     goto error;
  692.   if (client_flag & CLIENT_COMPRESS) /* We will use compression */
  693.     net->compress=1;
  694.   DBUG_PRINT("exit",("Mysql handler: %lx",mysql));
  695.   DBUG_RETURN(mysql);
  696. error:
  697.   DBUG_PRINT("error",("message: %u (%s)",net->last_errno,net->last_error));
  698.   {
  699.     /* Free alloced memory */
  700.     my_bool free_me=mysql->free_me;
  701.     mc_end_server(mysql);
  702.     mysql->free_me=0;
  703.     mc_mysql_close(mysql);
  704.     mysql->free_me=free_me;
  705.   }
  706.   DBUG_RETURN(0);
  707. }
  708. /*************************************************************************
  709. ** Send a QUIT to the server and close the connection
  710. ** If handle is alloced by mysql connect free it.
  711. *************************************************************************/
  712. void STDCALL
  713. mc_mysql_close(MYSQL *mysql)
  714. {
  715.   DBUG_ENTER("mysql_close");
  716.   if (mysql) /* Some simple safety */
  717.   {
  718.     if (mysql->net.vio != 0)
  719.     {
  720.       mc_free_old_query(mysql);
  721.       mysql->status=MYSQL_STATUS_READY; /* Force command */
  722.       mc_simple_command(mysql,COM_QUIT,NullS,0,1);
  723.       mc_end_server(mysql);
  724.     }
  725.     my_free((gptr) mysql->host_info,MYF(MY_ALLOW_ZERO_PTR));
  726.     my_free(mysql->user,MYF(MY_ALLOW_ZERO_PTR));
  727.     my_free(mysql->passwd,MYF(MY_ALLOW_ZERO_PTR));
  728.     my_free(mysql->db,MYF(MY_ALLOW_ZERO_PTR));
  729.     /* Clear pointers for better safety */
  730.     mysql->host_info=mysql->user=mysql->passwd=mysql->db=0;
  731.     bzero((char*) &mysql->options,sizeof(mysql->options));
  732.     mysql->net.vio = 0;
  733. #ifdef HAVE_OPENSSL
  734.     ((VioConnectorFd*)(mysql->connector_fd))->delete();
  735.     mysql->connector_fd = 0;
  736. #endif /* HAVE_OPENSSL */
  737.     if (mysql->free_me)
  738.       my_free((gptr) mysql,MYF(0));
  739.   }
  740.   DBUG_VOID_RETURN;
  741. }