mirror of
				https://github.com/MariaDB/server.git
				synced 2025-11-03 14:33:32 +03:00 
			
		
		
		
	
		
			
				
	
	
		
			804 lines
		
	
	
		
			23 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
			
		
		
	
	
			804 lines
		
	
	
		
			23 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
/* Copyright (C) 2000 MySQL AB & MySQL Finland AB & TCX DataKonsult AB
 | 
						|
   
 | 
						|
   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; either version 2 of the License, or
 | 
						|
   (at your option) any later version.
 | 
						|
   
 | 
						|
   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., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA */
 | 
						|
 | 
						|
/*
 | 
						|
 mini MySQL client to be included into the server to do server to server
 | 
						|
 commincation by Sasha Pachev
 | 
						|
 | 
						|
 Note: all file-global symbols must begin with mc_ , even the static ones, just
 | 
						|
 in case we decide to make them external at some point
 | 
						|
 */
 | 
						|
 | 
						|
#define DONT_USE_RAID
 | 
						|
#if defined(__WIN__) || defined(WIN32)
 | 
						|
#include <winsock.h>
 | 
						|
#include <odbcinst.h>
 | 
						|
#endif
 | 
						|
#include <global.h>
 | 
						|
#include <my_sys.h>
 | 
						|
#include <mysys_err.h>
 | 
						|
#include <m_string.h>
 | 
						|
#include <m_ctype.h>
 | 
						|
#include "mysql.h"
 | 
						|
#include "mini_client.h"
 | 
						|
#include "mysql_version.h"
 | 
						|
#include "mysqld_error.h"
 | 
						|
#include "errmsg.h"
 | 
						|
#include <violite.h>
 | 
						|
#include <sys/stat.h>
 | 
						|
#include <signal.h>
 | 
						|
#ifdef	 HAVE_PWD_H
 | 
						|
#include <pwd.h>
 | 
						|
#endif
 | 
						|
#if !defined(MSDOS) && !defined(__WIN__)
 | 
						|
#include <sys/socket.h>
 | 
						|
#include <netinet/in.h>
 | 
						|
#include <arpa/inet.h>
 | 
						|
#include <netdb.h>
 | 
						|
#ifdef HAVE_SELECT_H
 | 
						|
#  include <select.h>
 | 
						|
#endif
 | 
						|
#ifdef HAVE_SYS_SELECT_H
 | 
						|
#include <sys/select.h>
 | 
						|
#endif
 | 
						|
#endif
 | 
						|
#ifdef HAVE_SYS_UN_H
 | 
						|
#  include <sys/un.h>
 | 
						|
#endif
 | 
						|
#if defined(THREAD) && !defined(__WIN__)
 | 
						|
#include <my_pthread.h>				/* because of signal()	*/
 | 
						|
#endif
 | 
						|
#ifndef INADDR_NONE
 | 
						|
#define INADDR_NONE	-1
 | 
						|
#endif
 | 
						|
 | 
						|
 | 
						|
static void mc_end_server(MYSQL *mysql);
 | 
						|
static int mc_sock_connect(File s, const struct sockaddr *name, uint namelen, uint to);
 | 
						|
static void mc_free_old_query(MYSQL *mysql);
 | 
						|
 | 
						|
 | 
						|
#define CLIENT_CAPABILITIES	(CLIENT_LONG_PASSWORD | CLIENT_LONG_FLAG | CLIENT_LOCAL_FILES)
 | 
						|
 | 
						|
#if defined(MSDOS) || defined(__WIN__)
 | 
						|
#define ERRNO WSAGetLastError()
 | 
						|
#define perror(A)
 | 
						|
#else
 | 
						|
#include <sys/errno.h>
 | 
						|
#define ERRNO errno
 | 
						|
#define SOCKET_ERROR -1
 | 
						|
#define closesocket(A) close(A)
 | 
						|
#endif
 | 
						|
 | 
						|
#ifdef __WIN__
 | 
						|
static my_bool is_NT(void)
 | 
						|
{
 | 
						|
  char *os=getenv("OS");
 | 
						|
  return (os && !strcmp(os, "Windows_NT")) ? 1 : 0;
 | 
						|
}
 | 
						|
#endif
 | 
						|
 | 
						|
/*
 | 
						|
** Create a named pipe connection
 | 
						|
*/
 | 
						|
 | 
						|
#ifdef __WIN__
 | 
						|
 | 
						|
HANDLE create_named_pipe(NET *net, uint connect_timeout, char **arg_host,
 | 
						|
			 char **arg_unix_socket)
 | 
						|
{
 | 
						|
  HANDLE hPipe=INVALID_HANDLE_VALUE;
 | 
						|
  char szPipeName [ 257 ];
 | 
						|
  DWORD dwMode;
 | 
						|
  int i;
 | 
						|
  my_bool testing_named_pipes=0;
 | 
						|
  char *host= *arg_host, *unix_socket= *arg_unix_socket;
 | 
						|
 | 
						|
  if (!host || !strcmp(host,LOCAL_HOST))
 | 
						|
    host=LOCAL_HOST_NAMEDPIPE;
 | 
						|
 | 
						|
  sprintf( szPipeName, "\\\\%s\\pipe\\%s", host, unix_socket);
 | 
						|
  DBUG_PRINT("info",("Server name: '%s'.  Named Pipe: %s",
 | 
						|
		     host, unix_socket));
 | 
						|
 | 
						|
  for (i=0 ; i < 100 ; i++)			/* Don't retry forever */
 | 
						|
  {
 | 
						|
    if ((hPipe = CreateFile(szPipeName,
 | 
						|
			    GENERIC_READ | GENERIC_WRITE,
 | 
						|
			    0,
 | 
						|
			    NULL,
 | 
						|
			    OPEN_EXISTING,
 | 
						|
			    0,
 | 
						|
			    NULL )) != INVALID_HANDLE_VALUE)
 | 
						|
      break;
 | 
						|
    if (GetLastError() != ERROR_PIPE_BUSY)
 | 
						|
    {
 | 
						|
      net->last_errno=CR_NAMEDPIPEOPEN_ERROR;
 | 
						|
      sprintf(net->last_error,ER(net->last_errno),host, unix_socket,
 | 
						|
	      (ulong) GetLastError());
 | 
						|
      return INVALID_HANDLE_VALUE;
 | 
						|
    }
 | 
						|
    /* wait for for an other instance */
 | 
						|
    if (! WaitNamedPipe(szPipeName, connect_timeout*1000) )
 | 
						|
    {
 | 
						|
      net->last_errno=CR_NAMEDPIPEWAIT_ERROR;
 | 
						|
      sprintf(net->last_error,ER(net->last_errno),host, unix_socket,
 | 
						|
	      (ulong) GetLastError());
 | 
						|
      return INVALID_HANDLE_VALUE;
 | 
						|
    }
 | 
						|
  }
 | 
						|
  if (hPipe == INVALID_HANDLE_VALUE)
 | 
						|
  {
 | 
						|
    net->last_errno=CR_NAMEDPIPEOPEN_ERROR;
 | 
						|
    sprintf(net->last_error,ER(net->last_errno),host, unix_socket,
 | 
						|
	    (ulong) GetLastError());
 | 
						|
    return INVALID_HANDLE_VALUE;
 | 
						|
  }
 | 
						|
  dwMode = PIPE_READMODE_BYTE | PIPE_WAIT;
 | 
						|
  if ( !SetNamedPipeHandleState(hPipe, &dwMode, NULL, NULL) )
 | 
						|
  {
 | 
						|
    CloseHandle( hPipe );
 | 
						|
    net->last_errno=CR_NAMEDPIPESETSTATE_ERROR;
 | 
						|
    sprintf(net->last_error,ER(net->last_errno),host, unix_socket,
 | 
						|
	    (ulong) GetLastError());
 | 
						|
    return INVALID_HANDLE_VALUE;
 | 
						|
  }
 | 
						|
  *arg_host=host ; *arg_unix_socket=unix_socket;	/* connect arg */
 | 
						|
  return (hPipe);
 | 
						|
}
 | 
						|
#endif
 | 
						|
 | 
						|
 | 
						|
/****************************************************************************
 | 
						|
** Init MySQL structure or allocate one
 | 
						|
****************************************************************************/
 | 
						|
 | 
						|
MYSQL * STDCALL
 | 
						|
mc_mysql_init(MYSQL *mysql)
 | 
						|
{
 | 
						|
  init_client_errs();
 | 
						|
  if (!mysql)
 | 
						|
  {
 | 
						|
    if (!(mysql=(MYSQL*) my_malloc(sizeof(*mysql),MYF(MY_WME | MY_ZEROFILL))))
 | 
						|
      return 0;
 | 
						|
    mysql->free_me=1;
 | 
						|
    mysql->net.vio = 0;
 | 
						|
  }
 | 
						|
  else
 | 
						|
    bzero((char*) (mysql),sizeof(*(mysql)));
 | 
						|
#ifdef __WIN__
 | 
						|
  mysql->options.connect_timeout=20;
 | 
						|
#endif
 | 
						|
  return mysql;
 | 
						|
}
 | 
						|
 | 
						|
/**************************************************************************
 | 
						|
** Shut down connection
 | 
						|
**************************************************************************/
 | 
						|
 | 
						|
static void
 | 
						|
mc_end_server(MYSQL *mysql)
 | 
						|
{
 | 
						|
  DBUG_ENTER("mc_end_server");
 | 
						|
  if (mysql->net.vio != 0)
 | 
						|
  {
 | 
						|
    DBUG_PRINT("info",("Net: %s", vio_description(mysql->net.vio)));
 | 
						|
    vio_delete(mysql->net.vio);
 | 
						|
    mysql->net.vio= 0;          /* Marker */
 | 
						|
  }
 | 
						|
  net_end(&mysql->net);
 | 
						|
  mc_free_old_query(mysql);
 | 
						|
  DBUG_VOID_RETURN;
 | 
						|
}
 | 
						|
 | 
						|
static void mc_free_old_query(MYSQL *mysql)
 | 
						|
{
 | 
						|
  DBUG_ENTER("mc_free_old_query");
 | 
						|
  if (mysql->fields)
 | 
						|
    free_root(&mysql->field_alloc,MYF(0));
 | 
						|
  else
 | 
						|
    init_alloc_root(&mysql->field_alloc,8192,0); /* Assume rowlength < 8192 */
 | 
						|
  mysql->fields=0;
 | 
						|
  mysql->field_count=0;				/* For API */
 | 
						|
  DBUG_VOID_RETURN;
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
/****************************************************************************
 | 
						|
* A modified version of connect().  mc_sock_connect() allows you to specify
 | 
						|
* a timeout value, in seconds, that we should wait until we
 | 
						|
* derermine we can't connect to a particular host.  If timeout is 0,
 | 
						|
* mc_sock_connect() will behave exactly like connect().
 | 
						|
*
 | 
						|
* Base version coded by Steve Bernacki, Jr. <steve@navinet.net>
 | 
						|
*****************************************************************************/
 | 
						|
 | 
						|
static int mc_sock_connect(my_socket s, const struct sockaddr *name,
 | 
						|
			   uint namelen, uint to)
 | 
						|
{
 | 
						|
#if defined(__WIN__)
 | 
						|
  return connect(s, (struct sockaddr*) name, namelen);
 | 
						|
#else
 | 
						|
  int flags, res, s_err;
 | 
						|
  size_socket s_err_size = sizeof(uint);
 | 
						|
  fd_set sfds;
 | 
						|
  struct timeval tv;
 | 
						|
 | 
						|
  /* If they passed us a timeout of zero, we should behave
 | 
						|
   * exactly like the normal connect() call does.
 | 
						|
   */
 | 
						|
 | 
						|
  if (to == 0)
 | 
						|
    return connect(s, (struct sockaddr*) name, namelen);
 | 
						|
 | 
						|
  flags = fcntl(s, F_GETFL, 0);		  /* Set socket to not block */
 | 
						|
#ifdef O_NONBLOCK
 | 
						|
  fcntl(s, F_SETFL, flags | O_NONBLOCK);  /* and save the flags..  */
 | 
						|
#endif
 | 
						|
 | 
						|
  res = connect(s, (struct sockaddr*) name, namelen);
 | 
						|
  s_err = errno;			/* Save the error... */
 | 
						|
  fcntl(s, F_SETFL, flags);
 | 
						|
  if ((res != 0) && (s_err != EINPROGRESS))
 | 
						|
  {
 | 
						|
    errno = s_err;			/* Restore it */
 | 
						|
    return(-1);
 | 
						|
  }
 | 
						|
  if (res == 0)				/* Connected quickly! */
 | 
						|
    return(0);
 | 
						|
 | 
						|
  /* Otherwise, our connection is "in progress."  We can use
 | 
						|
   * the select() call to wait up to a specified period of time
 | 
						|
   * for the connection to suceed.  If select() returns 0
 | 
						|
   * (after waiting howevermany seconds), our socket never became
 | 
						|
   * writable (host is probably unreachable.)  Otherwise, if
 | 
						|
   * select() returns 1, then one of two conditions exist:
 | 
						|
   *
 | 
						|
   * 1. An error occured.  We use getsockopt() to check for this.
 | 
						|
   * 2. The connection was set up sucessfully: getsockopt() will
 | 
						|
   * return 0 as an error.
 | 
						|
   *
 | 
						|
   * Thanks goes to Andrew Gierth <andrew@erlenstar.demon.co.uk>
 | 
						|
   * who posted this method of timing out a connect() in
 | 
						|
   * comp.unix.programmer on August 15th, 1997.
 | 
						|
   */
 | 
						|
 | 
						|
  FD_ZERO(&sfds);
 | 
						|
  FD_SET(s, &sfds);
 | 
						|
  tv.tv_sec = (long) to;
 | 
						|
  tv.tv_usec = 0;
 | 
						|
#ifdef HPUX
 | 
						|
  res = select(s+1, NULL, (int*) &sfds, NULL, &tv);
 | 
						|
#else
 | 
						|
  res = select(s+1, NULL, &sfds, NULL, &tv);
 | 
						|
#endif
 | 
						|
  if (res <= 0)					/* Never became writable */
 | 
						|
    return(-1);
 | 
						|
 | 
						|
  /* select() returned something more interesting than zero, let's
 | 
						|
   * see if we have any errors.  If the next two statements pass,
 | 
						|
   * we've got an open socket!
 | 
						|
   */
 | 
						|
 | 
						|
  s_err=0;
 | 
						|
  if (getsockopt(s, SOL_SOCKET, SO_ERROR, (char*) &s_err, &s_err_size) != 0)
 | 
						|
    return(-1);
 | 
						|
 | 
						|
  if (s_err)
 | 
						|
  {						/* getsockopt() could suceed */
 | 
						|
    errno = s_err;
 | 
						|
    return(-1);					/* but return an error... */
 | 
						|
  }
 | 
						|
  return(0);					/* It's all good! */
 | 
						|
#endif
 | 
						|
}
 | 
						|
 | 
						|
/*****************************************************************************
 | 
						|
** read a packet from server. Give error message if socket was down
 | 
						|
** or packet is an error message
 | 
						|
*****************************************************************************/
 | 
						|
 | 
						|
uint STDCALL
 | 
						|
mc_net_safe_read(MYSQL *mysql)
 | 
						|
{
 | 
						|
  NET *net= &mysql->net;
 | 
						|
  uint len=0;
 | 
						|
 | 
						|
  if (net->vio != 0)
 | 
						|
    len=my_net_read(net);
 | 
						|
 | 
						|
  if (len == packet_error || len == 0)
 | 
						|
  {
 | 
						|
    DBUG_PRINT("error",("Wrong connection or packet. fd: %s  len: %d",
 | 
						|
			vio_description(net->vio),len));
 | 
						|
    if(errno != EINTR)
 | 
						|
      {
 | 
						|
        mc_end_server(mysql);
 | 
						|
        net->last_errno=CR_SERVER_LOST;
 | 
						|
        strmov(net->last_error,ER(net->last_errno));
 | 
						|
      }	
 | 
						|
    return(packet_error);
 | 
						|
  }
 | 
						|
  if (net->read_pos[0] == 255)
 | 
						|
  {
 | 
						|
    if (len > 3)
 | 
						|
    {
 | 
						|
      char *pos=(char*) net->read_pos+1;
 | 
						|
      if (mysql->protocol_version > 9)
 | 
						|
      {						/* New client protocol */
 | 
						|
	net->last_errno=uint2korr(pos);
 | 
						|
	pos+=2;
 | 
						|
	len-=2;
 | 
						|
	if(!net->last_errno)
 | 
						|
	  net->last_errno = CR_UNKNOWN_ERROR;
 | 
						|
      }
 | 
						|
      else
 | 
						|
      {
 | 
						|
	net->last_errno=CR_UNKNOWN_ERROR;
 | 
						|
	len--;
 | 
						|
      }
 | 
						|
      (void) strmake(net->last_error,(char*) pos,
 | 
						|
		     min(len,sizeof(net->last_error)-1));
 | 
						|
    }
 | 
						|
    else
 | 
						|
    {
 | 
						|
      net->last_errno=CR_UNKNOWN_ERROR;
 | 
						|
      (void) strmov(net->last_error,ER(net->last_errno));
 | 
						|
    }
 | 
						|
    DBUG_PRINT("error",("Got error: %d (%s)", net->last_errno,
 | 
						|
			net->last_error));
 | 
						|
    return(packet_error);
 | 
						|
  }
 | 
						|
  return len;
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
char * STDCALL mc_mysql_error(MYSQL *mysql)
 | 
						|
{
 | 
						|
  return (mysql)->net.last_error;
 | 
						|
}
 | 
						|
 | 
						|
my_bool STDCALL mc_mysql_reconnect(MYSQL *mysql)
 | 
						|
{
 | 
						|
  MYSQL tmp_mysql;
 | 
						|
  DBUG_ENTER("mc_mysql_reconnect");
 | 
						|
 | 
						|
  mc_mysql_init(&tmp_mysql);
 | 
						|
  tmp_mysql.options=mysql->options;
 | 
						|
  if (!mc_mysql_connect(&tmp_mysql,mysql->host,mysql->user,mysql->passwd,
 | 
						|
			  mysql->db, mysql->port, mysql->unix_socket,
 | 
						|
			  mysql->client_flag))
 | 
						|
    DBUG_RETURN(1);
 | 
						|
  tmp_mysql.free_me=mysql->free_me;
 | 
						|
  mysql->free_me=0;
 | 
						|
  bzero((char*) &mysql->options,sizeof(&mysql->options));
 | 
						|
  mc_mysql_close(mysql);
 | 
						|
  *mysql=tmp_mysql;
 | 
						|
  net_clear(&mysql->net);
 | 
						|
  mysql->affected_rows= ~(my_ulonglong) 0;
 | 
						|
  DBUG_RETURN(0);
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
 | 
						|
int STDCALL
 | 
						|
mc_simple_command(MYSQL *mysql,enum enum_server_command command, const char *arg,
 | 
						|
	       uint length, my_bool skipp_check)
 | 
						|
{
 | 
						|
  NET *net= &mysql->net;
 | 
						|
  int result= -1;
 | 
						|
 | 
						|
  if (mysql->net.vio == 0)
 | 
						|
  {						/* Do reconnect if possible */
 | 
						|
    if (mc_mysql_reconnect(mysql))
 | 
						|
    {
 | 
						|
      net->last_errno=CR_SERVER_GONE_ERROR;
 | 
						|
      strmov(net->last_error,ER(net->last_errno));
 | 
						|
      goto end;
 | 
						|
    }
 | 
						|
  }
 | 
						|
  if (mysql->status != MYSQL_STATUS_READY)
 | 
						|
  {
 | 
						|
    strmov(net->last_error,ER(mysql->net.last_errno=CR_COMMANDS_OUT_OF_SYNC));
 | 
						|
    goto end;
 | 
						|
  }
 | 
						|
 | 
						|
  mysql->net.last_error[0]=0;
 | 
						|
  mysql->net.last_errno=0;
 | 
						|
  mysql->info=0;
 | 
						|
  mysql->affected_rows= ~(my_ulonglong) 0;
 | 
						|
  net_clear(net);			/* Clear receive buffer */
 | 
						|
  if (!arg)
 | 
						|
    arg="";
 | 
						|
 | 
						|
  if (net_write_command(net,(uchar) command,arg,
 | 
						|
			length ? length :(uint) strlen(arg)))
 | 
						|
  {
 | 
						|
    DBUG_PRINT("error",("Can't send command to server. Error: %d",errno));
 | 
						|
    mc_end_server(mysql);
 | 
						|
    if (mc_mysql_reconnect(mysql) ||
 | 
						|
	net_write_command(net,(uchar) command,arg,
 | 
						|
			  length ? length :(uint) strlen(arg)))
 | 
						|
    {
 | 
						|
      net->last_errno=CR_SERVER_GONE_ERROR;
 | 
						|
      strmov(net->last_error,ER(net->last_errno));
 | 
						|
      goto end;
 | 
						|
    }
 | 
						|
  }
 | 
						|
  result=0;
 | 
						|
  if (!skipp_check)
 | 
						|
    result= ((mysql->packet_length=mc_net_safe_read(mysql)) == packet_error ?
 | 
						|
	     -1 : 0);
 | 
						|
 end:
 | 
						|
  return result;
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
MYSQL * STDCALL
 | 
						|
mc_mysql_connect(MYSQL *mysql,const char *host, const char *user,
 | 
						|
		   const char *passwd, const char *db,
 | 
						|
		   uint port, const char *unix_socket,uint client_flag)
 | 
						|
{
 | 
						|
  char		buff[100],*end,*host_info;
 | 
						|
  my_socket	sock;
 | 
						|
  ulong		ip_addr;
 | 
						|
  struct	sockaddr_in sock_addr;
 | 
						|
  uint		pkt_length;
 | 
						|
  NET		*net= &mysql->net;
 | 
						|
#ifdef __WIN__
 | 
						|
  HANDLE	hPipe=INVALID_HANDLE_VALUE;
 | 
						|
#endif
 | 
						|
#ifdef HAVE_SYS_UN_H
 | 
						|
  struct	sockaddr_un UNIXaddr;
 | 
						|
#endif
 | 
						|
  DBUG_ENTER("mysql_real_connect");
 | 
						|
 | 
						|
  DBUG_PRINT("enter",("host: %s  db: %s  user: %s",
 | 
						|
		      host ? host : "(Null)",
 | 
						|
		      db ? db : "(Null)",
 | 
						|
		      user ? user : "(Null)"));
 | 
						|
 | 
						|
  bzero((char*) &mysql->options,sizeof(mysql->options));
 | 
						|
  net->vio = 0;				/* If something goes wrong */
 | 
						|
  mysql->charset=default_charset_info;  /* Set character set */
 | 
						|
  if (!port)
 | 
						|
    port = MYSQL_PORT;			/* Should always be set by mysqld */
 | 
						|
  if (!unix_socket)
 | 
						|
    unix_socket=MYSQL_UNIX_ADDR;
 | 
						|
 | 
						|
  mysql->reconnect=1;			/* Reconnect as default */
 | 
						|
 | 
						|
  /*
 | 
						|
  ** Grab a socket and connect it to the server
 | 
						|
  */
 | 
						|
 | 
						|
#if defined(HAVE_SYS_UN_H)
 | 
						|
  if (!host || !strcmp(host,LOCAL_HOST))
 | 
						|
  {
 | 
						|
    host=LOCAL_HOST;
 | 
						|
    host_info=(char*) ER(CR_LOCALHOST_CONNECTION);
 | 
						|
    DBUG_PRINT("info",("Using UNIX sock '%s'",unix_socket));
 | 
						|
    if ((sock = socket(AF_UNIX,SOCK_STREAM,0)) == SOCKET_ERROR)
 | 
						|
    {
 | 
						|
      net->last_errno=CR_SOCKET_CREATE_ERROR;
 | 
						|
      sprintf(net->last_error,ER(net->last_errno),ERRNO);
 | 
						|
      goto error;
 | 
						|
    }
 | 
						|
    net->vio = vio_new(sock, VIO_TYPE_SOCKET, TRUE);
 | 
						|
    bzero((char*) &UNIXaddr,sizeof(UNIXaddr));
 | 
						|
    UNIXaddr.sun_family = AF_UNIX;
 | 
						|
    strmov(UNIXaddr.sun_path, unix_socket);
 | 
						|
    if (mc_sock_connect(sock,(struct sockaddr *) &UNIXaddr, sizeof(UNIXaddr),
 | 
						|
			mysql->options.connect_timeout) <0)
 | 
						|
    {
 | 
						|
      DBUG_PRINT("error",("Got error %d on connect to local server",ERRNO));
 | 
						|
      net->last_errno=CR_CONNECTION_ERROR;
 | 
						|
      sprintf(net->last_error,ER(net->last_errno),unix_socket,ERRNO);
 | 
						|
      goto error;
 | 
						|
    }
 | 
						|
  }
 | 
						|
  else
 | 
						|
#elif defined(__WIN__)
 | 
						|
  {
 | 
						|
    if ((unix_socket ||
 | 
						|
	 !host && is_NT() ||
 | 
						|
	 host && !strcmp(host,LOCAL_HOST_NAMEDPIPE) ||
 | 
						|
	 mysql->options.named_pipe || !have_tcpip))
 | 
						|
    {
 | 
						|
      sock=0;
 | 
						|
      if ((hPipe=create_named_pipe(net, mysql->options.connect_timeout,
 | 
						|
				   (char**) &host, (char**) &unix_socket)) ==
 | 
						|
	  INVALID_HANDLE_VALUE)
 | 
						|
      {
 | 
						|
	DBUG_PRINT("error",
 | 
						|
		   ("host: '%s'  socket: '%s'  named_pipe: %d  have_tcpip: %d",
 | 
						|
		    host ? host : "<null>",
 | 
						|
		    unix_socket ? unix_socket : "<null>",
 | 
						|
		    (int) mysql->options.named_pipe,
 | 
						|
		    (int) have_tcpip));
 | 
						|
	if (mysql->options.named_pipe ||
 | 
						|
	    (host && !strcmp(host,LOCAL_HOST_NAMEDPIPE)) ||
 | 
						|
	    (unix_socket && !strcmp(unix_socket,MYSQL_NAMEDPIPE)))
 | 
						|
	  goto error;		/* User only requested named pipes */
 | 
						|
	/* Try also with TCP/IP */
 | 
						|
      }
 | 
						|
      else
 | 
						|
      {
 | 
						|
	net->vio=vio_new_win32pipe(hPipe);
 | 
						|
	sprintf(host_info=buff, ER(CR_NAMEDPIPE_CONNECTION), host,
 | 
						|
		unix_socket);
 | 
						|
      }
 | 
						|
    }
 | 
						|
  }
 | 
						|
  if (hPipe == INVALID_HANDLE_VALUE)
 | 
						|
#endif
 | 
						|
  {
 | 
						|
    unix_socket=0;				/* This is not used */
 | 
						|
    if (!host)
 | 
						|
      host=LOCAL_HOST;
 | 
						|
    sprintf(host_info=buff,ER(CR_TCP_CONNECTION),host);
 | 
						|
    DBUG_PRINT("info",("Server name: '%s'.  TCP sock: %d", host,port));
 | 
						|
    if ((sock = socket(AF_INET,SOCK_STREAM,0)) == SOCKET_ERROR)
 | 
						|
    {
 | 
						|
      net->last_errno=CR_IPSOCK_ERROR;
 | 
						|
      sprintf(net->last_error,ER(net->last_errno),ERRNO);
 | 
						|
      goto error;
 | 
						|
    }
 | 
						|
    net->vio = vio_new(sock,VIO_TYPE_TCPIP,FALSE);
 | 
						|
    bzero((char*) &sock_addr,sizeof(sock_addr));
 | 
						|
    sock_addr.sin_family = AF_INET;
 | 
						|
 | 
						|
    /*
 | 
						|
    ** The server name may be a host name or IP address
 | 
						|
    */
 | 
						|
 | 
						|
    if ((int) (ip_addr = inet_addr(host)) != (int) INADDR_NONE)
 | 
						|
    {
 | 
						|
      memcpy_fixed(&sock_addr.sin_addr,&ip_addr,sizeof(ip_addr));
 | 
						|
    }
 | 
						|
    else
 | 
						|
#if defined(HAVE_GETHOSTBYNAME_R) && defined(_REENTRANT) && defined(THREAD)
 | 
						|
    {
 | 
						|
      int tmp_errno;
 | 
						|
      struct hostent tmp_hostent,*hp;
 | 
						|
      char buff2[GETHOSTBYNAME_BUFF_SIZE];
 | 
						|
      hp = my_gethostbyname_r(host,&tmp_hostent,buff2,sizeof(buff2),
 | 
						|
			      &tmp_errno);
 | 
						|
      if (!hp)
 | 
						|
      {
 | 
						|
	net->last_errno=CR_UNKNOWN_HOST;
 | 
						|
	sprintf(net->last_error, ER(CR_UNKNOWN_HOST), host, tmp_errno);
 | 
						|
	goto error;
 | 
						|
      }
 | 
						|
      memcpy(&sock_addr.sin_addr,hp->h_addr, (size_t) hp->h_length);
 | 
						|
    }
 | 
						|
#else
 | 
						|
    {
 | 
						|
      struct hostent *hp;
 | 
						|
      if (!(hp=gethostbyname(host)))
 | 
						|
      {
 | 
						|
	net->last_errno=CR_UNKNOWN_HOST;
 | 
						|
	sprintf(net->last_error, ER(CR_UNKNOWN_HOST), host, errno);
 | 
						|
	goto error;
 | 
						|
      }
 | 
						|
      memcpy(&sock_addr.sin_addr,hp->h_addr, (size_t) hp->h_length);
 | 
						|
    }
 | 
						|
#endif
 | 
						|
    sock_addr.sin_port = (ushort) htons((ushort) port);
 | 
						|
    if (mc_sock_connect(sock,(struct sockaddr *) &sock_addr, sizeof(sock_addr),
 | 
						|
		 mysql->options.connect_timeout) <0)
 | 
						|
    {
 | 
						|
      DBUG_PRINT("error",("Got error %d on connect to '%s'",ERRNO,host));
 | 
						|
      net->last_errno= CR_CONN_HOST_ERROR;
 | 
						|
      sprintf(net->last_error ,ER(CR_CONN_HOST_ERROR), host, ERRNO);
 | 
						|
      goto error;
 | 
						|
    }
 | 
						|
  }
 | 
						|
 | 
						|
  if (!net->vio || my_net_init(net, net->vio))
 | 
						|
  {
 | 
						|
    vio_delete(net->vio);
 | 
						|
    net->last_errno=CR_OUT_OF_MEMORY;
 | 
						|
    strmov(net->last_error,ER(net->last_errno));
 | 
						|
    goto error;
 | 
						|
  }
 | 
						|
  vio_keepalive(net->vio,TRUE);
 | 
						|
 | 
						|
  /* Get version info */
 | 
						|
  mysql->protocol_version= PROTOCOL_VERSION;	/* Assume this */
 | 
						|
  if ((pkt_length=mc_net_safe_read(mysql)) == packet_error)
 | 
						|
    goto error;
 | 
						|
 | 
						|
  /* Check if version of protocoll matches current one */
 | 
						|
 | 
						|
  mysql->protocol_version= net->read_pos[0];
 | 
						|
  DBUG_DUMP("packet",(char*) net->read_pos,10);
 | 
						|
  DBUG_PRINT("info",("mysql protocol version %d, server=%d",
 | 
						|
		     PROTOCOL_VERSION, mysql->protocol_version));
 | 
						|
  if (mysql->protocol_version != PROTOCOL_VERSION &&
 | 
						|
      mysql->protocol_version != PROTOCOL_VERSION-1)
 | 
						|
  {
 | 
						|
    net->last_errno= CR_VERSION_ERROR;
 | 
						|
    sprintf(net->last_error, ER(CR_VERSION_ERROR), mysql->protocol_version,
 | 
						|
	    PROTOCOL_VERSION);
 | 
						|
    goto error;
 | 
						|
  }
 | 
						|
  end=strend((char*) net->read_pos+1);
 | 
						|
  mysql->thread_id=uint4korr(end+1);
 | 
						|
  end+=5;
 | 
						|
  strmake(mysql->scramble_buff,end,8);
 | 
						|
  if (pkt_length > (uint) (end+9 - (char*) net->read_pos))
 | 
						|
    mysql->server_capabilities=uint2korr(end+9);
 | 
						|
 | 
						|
  /* Save connection information */
 | 
						|
  if (!user) user="";
 | 
						|
  if (!passwd) passwd="";
 | 
						|
  if (!my_multi_malloc(MYF(0),
 | 
						|
		       &mysql->host_info, (uint) strlen(host_info)+1,
 | 
						|
		       &mysql->host,      (uint) strlen(host)+1,
 | 
						|
		       &mysql->unix_socket,
 | 
						|
		       unix_socket ? (uint) strlen(unix_socket)+1 : (uint) 1,
 | 
						|
		       &mysql->server_version,
 | 
						|
		       (uint) (end - (char*) net->read_pos),
 | 
						|
		       NullS) ||
 | 
						|
      !(mysql->user=my_strdup(user,MYF(0))) ||
 | 
						|
      !(mysql->passwd=my_strdup(passwd,MYF(0))))
 | 
						|
  {
 | 
						|
    strmov(net->last_error, ER(net->last_errno=CR_OUT_OF_MEMORY));
 | 
						|
    goto error;
 | 
						|
  }
 | 
						|
  strmov(mysql->host_info,host_info);
 | 
						|
  strmov(mysql->host,host);
 | 
						|
  if (unix_socket)
 | 
						|
    strmov(mysql->unix_socket,unix_socket);
 | 
						|
  else
 | 
						|
    mysql->unix_socket=0;
 | 
						|
  strmov(mysql->server_version,(char*) net->read_pos+1);
 | 
						|
  mysql->port=port;
 | 
						|
  mysql->client_flag=client_flag | mysql->options.client_flag;
 | 
						|
  DBUG_PRINT("info",("Server version = '%s'  capabilites: %ld",
 | 
						|
		     mysql->server_version,mysql->server_capabilities));
 | 
						|
 | 
						|
  /* Send client information for access check */
 | 
						|
  client_flag|=CLIENT_CAPABILITIES;
 | 
						|
 | 
						|
#ifdef HAVE_OPENSSL
 | 
						|
  if (mysql->options.use_ssl)
 | 
						|
    client_flag|=CLIENT_SSL;
 | 
						|
#endif /* HAVE_OPENSSL */
 | 
						|
 | 
						|
  if (db)
 | 
						|
    client_flag|=CLIENT_CONNECT_WITH_DB;
 | 
						|
#ifdef HAVE_COMPRESS
 | 
						|
  if (mysql->server_capabilities & CLIENT_COMPRESS &&
 | 
						|
      (mysql->options.compress || client_flag & CLIENT_COMPRESS))
 | 
						|
    client_flag|=CLIENT_COMPRESS;		/* We will use compression */
 | 
						|
  else
 | 
						|
#endif
 | 
						|
    client_flag&= ~CLIENT_COMPRESS;
 | 
						|
 | 
						|
#ifdef HAVE_OPENSSL
 | 
						|
  if ((mysql->server_capabilities & CLIENT_SSL) &&
 | 
						|
      (mysql->options.use_ssl || (client_flag & CLIENT_SSL)))
 | 
						|
  {
 | 
						|
    DBUG_PRINT("info", ("Changing IO layer to SSL"));
 | 
						|
    client_flag |= CLIENT_SSL;
 | 
						|
  }
 | 
						|
  else
 | 
						|
  {
 | 
						|
    if (client_flag & CLIENT_SSL)
 | 
						|
    {
 | 
						|
      DBUG_PRINT("info", ("Leaving IO layer intact because server doesn't support SSL"));
 | 
						|
    }
 | 
						|
    client_flag &= ~CLIENT_SSL;
 | 
						|
  }
 | 
						|
#endif /* HAVE_OPENSSL */
 | 
						|
 | 
						|
  int2store(buff,client_flag);
 | 
						|
  mysql->client_flag=client_flag;
 | 
						|
 | 
						|
#ifdef HAVE_OPENSSL
 | 
						|
  /* Oops.. are we careful enough to not send ANY information */
 | 
						|
  /* without encryption? */
 | 
						|
  if (client_flag & CLIENT_SSL)
 | 
						|
  {
 | 
						|
    if (my_net_write(net,buff,(uint) (2)) || net_flush(net))
 | 
						|
      goto error;
 | 
						|
    /* Do the SSL layering. */
 | 
						|
    DBUG_PRINT("info", ("IO layer change in progress..."));
 | 
						|
    VioSSLConnectorFd* connector_fd = (VioSSLConnectorFd*)
 | 
						|
      (mysql->connector_fd);
 | 
						|
    VioSocket*	vio_socket = (VioSocket*)(mysql->net.vio);
 | 
						|
    VioSSL*	vio_ssl =    connector_fd->connect(vio_socket);
 | 
						|
    mysql->net.vio =         (NetVio*)(vio_ssl);
 | 
						|
  }
 | 
						|
#endif /* HAVE_OPENSSL */
 | 
						|
 | 
						|
  int3store(buff+2,max_allowed_packet);
 | 
						|
  if (user && user[0])
 | 
						|
    strmake(buff+5,user,32);
 | 
						|
  else
 | 
						|
    {
 | 
						|
      user = getenv("USER");
 | 
						|
      if(!user) user = "mysql";
 | 
						|
       strmov((char*) buff+5, user );
 | 
						|
    }
 | 
						|
 | 
						|
  DBUG_PRINT("info",("user: %s",buff+5));
 | 
						|
  end=scramble(strend(buff+5)+1, mysql->scramble_buff, passwd,
 | 
						|
	       (my_bool) (mysql->protocol_version == 9));
 | 
						|
  if (db)
 | 
						|
  {
 | 
						|
    end=strmov(end+1,db);
 | 
						|
    mysql->db=my_strdup(db,MYF(MY_WME));
 | 
						|
  }
 | 
						|
  if (my_net_write(net,buff,(uint) (end-buff)) || net_flush(net) ||
 | 
						|
      mc_net_safe_read(mysql) == packet_error)
 | 
						|
    goto error;
 | 
						|
  if (client_flag & CLIENT_COMPRESS)		/* We will use compression */
 | 
						|
    net->compress=1;
 | 
						|
  DBUG_PRINT("exit",("Mysql handler: %lx",mysql));
 | 
						|
  DBUG_RETURN(mysql);
 | 
						|
 | 
						|
error:
 | 
						|
  DBUG_PRINT("error",("message: %u (%s)",net->last_errno,net->last_error));
 | 
						|
  {
 | 
						|
    /* Free alloced memory */
 | 
						|
    my_bool free_me=mysql->free_me;
 | 
						|
    mc_end_server(mysql);
 | 
						|
    mysql->free_me=0;
 | 
						|
    mc_mysql_close(mysql);
 | 
						|
    mysql->free_me=free_me;
 | 
						|
  }
 | 
						|
  DBUG_RETURN(0);
 | 
						|
}
 | 
						|
 | 
						|
/*************************************************************************
 | 
						|
** Send a QUIT to the server and close the connection
 | 
						|
** If handle is alloced by mysql connect free it.
 | 
						|
*************************************************************************/
 | 
						|
 | 
						|
void STDCALL
 | 
						|
mc_mysql_close(MYSQL *mysql)
 | 
						|
{
 | 
						|
  DBUG_ENTER("mysql_close");
 | 
						|
  if (mysql)					/* Some simple safety */
 | 
						|
  {
 | 
						|
    if (mysql->net.vio != 0)
 | 
						|
    {
 | 
						|
      mc_free_old_query(mysql);
 | 
						|
      mysql->status=MYSQL_STATUS_READY; /* Force command */
 | 
						|
      mc_simple_command(mysql,COM_QUIT,NullS,0,1);
 | 
						|
      mc_end_server(mysql);
 | 
						|
    }
 | 
						|
    my_free((gptr) mysql->host_info,MYF(MY_ALLOW_ZERO_PTR));
 | 
						|
    my_free(mysql->user,MYF(MY_ALLOW_ZERO_PTR));
 | 
						|
    my_free(mysql->passwd,MYF(MY_ALLOW_ZERO_PTR));
 | 
						|
    my_free(mysql->db,MYF(MY_ALLOW_ZERO_PTR));
 | 
						|
    /* Clear pointers for better safety */
 | 
						|
    mysql->host_info=mysql->user=mysql->passwd=mysql->db=0;
 | 
						|
    bzero((char*) &mysql->options,sizeof(mysql->options));
 | 
						|
    mysql->net.vio = 0;
 | 
						|
#ifdef HAVE_OPENSSL
 | 
						|
    ((VioConnectorFd*)(mysql->connector_fd))->delete();
 | 
						|
    mysql->connector_fd = 0;
 | 
						|
#endif /* HAVE_OPENSSL */
 | 
						|
    if (mysql->free_me)
 | 
						|
      my_free((gptr) mysql,MYF(0));
 | 
						|
  }
 | 
						|
  DBUG_VOID_RETURN;
 | 
						|
}
 |