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

MySQL数据库

开发平台:

Visual C++

  1. /* Copyright (C) 2000-2004 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.
  5.    There are special exceptions to the terms and conditions of the GPL as it
  6.    is applied to this software. View the full text of the exception in file
  7.    EXCEPTIONS-CLIENT in the directory of this software distribution.
  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.    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., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA */
  15. #include <my_global.h>
  16. #if defined(THREAD)
  17. #include <my_pthread.h> /* because of signal() */
  18. #endif
  19. #include "mysql.h"
  20. #include "mysql_version.h"
  21. #include "mysqld_error.h"
  22. #include <my_sys.h>
  23. #include <mysys_err.h>
  24. #include <m_string.h>
  25. #include <m_ctype.h>
  26. #include <my_net.h>
  27. #include <errmsg.h>
  28. #include <violite.h>
  29. #include <sys/stat.h>
  30. #include <signal.h>
  31. #include <errno.h>
  32. #if defined(OS2)
  33. #  include <sys/un.h>
  34. #elif defined(__NETWARE__)
  35. #include <netdb.h>
  36. #include <sys/select.h>
  37. #include <sys/utsname.h>
  38. #elif !defined( __WIN__)
  39. #include <sys/resource.h>
  40. #ifdef HAVE_SYS_UN_H
  41. #  include <sys/un.h>
  42. #endif
  43. #include <netdb.h>
  44. #ifdef HAVE_SELECT_H
  45. #  include <select.h>
  46. #endif
  47. #ifdef HAVE_SYS_SELECT_H
  48. #include <sys/select.h>
  49. #endif
  50. #include <sys/utsname.h>
  51. #endif /* __WIN__ */
  52. #ifndef INADDR_NONE
  53. #define INADDR_NONE -1
  54. #endif
  55. #define RES_BUF_SHIFT 5
  56. #define NET_BUF_SIZE  2048
  57. MYSQL_MANAGER*  STDCALL mysql_manager_init(MYSQL_MANAGER* con)
  58. {
  59.   int net_buf_size=NET_BUF_SIZE;
  60.   if (!con)
  61.   {
  62.     if (!(con=(MYSQL_MANAGER*)my_malloc(sizeof(*con)+net_buf_size,
  63. MYF(MY_WME|MY_ZEROFILL))))
  64.       return 0;
  65.     con->free_me=1;
  66.     con->net_buf=(char*)con+sizeof(*con);    
  67.   }
  68.   else
  69.   {
  70.     bzero((char*)con,sizeof(*con));
  71.     if (!(con->net_buf=my_malloc(net_buf_size,MYF(0))))
  72.       return 0;
  73.   }
  74.   con->net_buf_pos=con->net_data_end=con->net_buf;
  75.   con->net_buf_size=net_buf_size;
  76.   return con;
  77. }
  78. MYSQL_MANAGER*  STDCALL mysql_manager_connect(MYSQL_MANAGER* con,
  79.       const char* host,
  80.       const char* user,
  81.       const char* passwd,
  82.       unsigned int port)
  83. {
  84.   my_socket sock;
  85.   struct sockaddr_in sock_addr;
  86.   in_addr_t ip_addr;
  87.   char msg_buf[MAX_MYSQL_MANAGER_MSG];
  88.   int msg_len;
  89.   Vio* vio;
  90.   my_bool not_used;
  91.   if (!host)
  92.     host="localhost";
  93.   if (!user)
  94.     user="root";
  95.   if (!passwd)
  96.     passwd="";
  97.   if ((sock=(my_socket)socket(AF_INET,SOCK_STREAM,0)) == INVALID_SOCKET)
  98.   {
  99.     con->last_errno=errno;
  100.     strmov(con->last_error,"Cannot create socket");
  101.     goto err;
  102.   }
  103.   if (!(vio=vio_new(sock,VIO_TYPE_TCPIP,FALSE)))
  104.   {
  105.     con->last_errno=ENOMEM;
  106.     strmov(con->last_error,"Cannot create network I/O object");
  107.     goto err;
  108.   }
  109.   vio_blocking(vio, TRUE, &not_used);
  110.   my_net_init(&con->net,vio);
  111.   bzero((char*) &sock_addr,sizeof(sock_addr));
  112.   sock_addr.sin_family = AF_INET;
  113.   if ((int) (ip_addr = inet_addr(host)) != (int) INADDR_NONE)
  114.   {
  115.     memcpy_fixed(&sock_addr.sin_addr,&ip_addr,sizeof(ip_addr));
  116.   }
  117.   else
  118.   {
  119.     int tmp_errno;
  120.     struct hostent tmp_hostent,*hp;
  121.     char buff2[GETHOSTBYNAME_BUFF_SIZE];
  122.     hp = my_gethostbyname_r(host,&tmp_hostent,buff2,sizeof(buff2),
  123.     &tmp_errno);
  124.     if (!hp)
  125.     {
  126.       con->last_errno=tmp_errno;
  127.       sprintf(con->last_error,"Could not resolve host '%-.64s'",host);
  128.       my_gethostbyname_r_free();
  129.       goto err;
  130.     }
  131.     memcpy(&sock_addr.sin_addr,hp->h_addr, (size_t) hp->h_length);
  132.     my_gethostbyname_r_free();
  133.   }
  134.   sock_addr.sin_port = (ushort) htons((ushort) port);
  135.   if (my_connect(sock,(struct sockaddr *) &sock_addr, sizeof(sock_addr),
  136.  0))
  137.   {
  138.     con->last_errno=errno;
  139.     sprintf(con->last_error ,"Could not connect to %-.64s", host);
  140.     goto err;
  141.   }
  142.   /* read the greating */
  143.   if (my_net_read(&con->net) == packet_error)
  144.   {
  145.     con->last_errno=errno;
  146.     strmov(con->last_error,"Read error on socket");
  147.     goto err;
  148.   }
  149.   sprintf(msg_buf,"%-.16s %-.16sn",user,passwd);
  150.   msg_len=strlen(msg_buf);
  151.   if (my_net_write(&con->net,msg_buf,msg_len) || net_flush(&con->net))
  152.   {
  153.     con->last_errno=con->net.last_errno;
  154.     strmov(con->last_error,"Write error on socket");
  155.     goto err;
  156.   }
  157.   if (my_net_read(&con->net) == packet_error)
  158.   {
  159.     con->last_errno=errno;
  160.     strmov(con->last_error,"Read error on socket");
  161.     goto err;
  162.   }
  163.   if ((con->cmd_status=atoi((char*) con->net.read_pos)) != MANAGER_OK)
  164.   {
  165.     strmov(con->last_error,"Access denied");
  166.     goto err;
  167.   }
  168.   if (!my_multi_malloc(MYF(0), &con->host, (uint)strlen(host)+1,
  169.        &con->user, (uint)strlen(user)+1,
  170.        &con->passwd, (uint)strlen(passwd)+1,
  171.        NullS))
  172.   {
  173.     con->last_errno=ENOMEM;
  174.     strmov(con->last_error,"Out of memory");
  175.     goto err;
  176.   }
  177.   strmov(con->host,host);
  178.   strmov(con->user,user);
  179.   strmov(con->passwd,passwd);
  180.   return con;
  181. err:
  182.   {
  183.     my_bool free_me=con->free_me;
  184.     con->free_me=0;
  185.     mysql_manager_close(con);
  186.     con->free_me=free_me;
  187.   }
  188.   return 0;
  189. }
  190. void STDCALL mysql_manager_close(MYSQL_MANAGER* con)
  191. {
  192.   /*
  193.     No need to free con->user and con->passwd, because they were
  194.     allocated in my_multimalloc() along with con->host, freeing
  195.     con->hosts frees the whole block
  196.   */
  197.   my_free((gptr)con->host,MYF(MY_ALLOW_ZERO_PTR));
  198.   net_end(&con->net);
  199.   if (con->free_me)
  200.     my_free((gptr)con,MYF(0));
  201. }
  202. int STDCALL mysql_manager_command(MYSQL_MANAGER* con,const char* cmd,
  203.   int cmd_len)
  204. {
  205.   if (!cmd_len)
  206.     cmd_len=strlen(cmd);
  207.   if (my_net_write(&con->net,(char*)cmd,cmd_len) || net_flush(&con->net))
  208.   {
  209.     con->last_errno=errno;
  210.     strmov(con->last_error,"Write error on socket");
  211.     return 1;
  212.   }
  213.   con->eof=0;
  214.   return 0;
  215. }
  216. int STDCALL mysql_manager_fetch_line(MYSQL_MANAGER* con, char* res_buf,
  217.      int res_buf_size)
  218. {
  219.   char* res_buf_end=res_buf+res_buf_size;
  220.   char* net_buf=(char*) con->net.read_pos, *net_buf_end;
  221.   int res_buf_shift=RES_BUF_SHIFT;
  222.   ulong num_bytes;
  223.   if (res_buf_size<RES_BUF_SHIFT)
  224.   {
  225.     con->last_errno=ENOMEM;
  226.     strmov(con->last_error,"Result buffer too small");
  227.     return 1;
  228.   }
  229.   if ((num_bytes=my_net_read(&con->net)) == packet_error)
  230.   {
  231.     con->last_errno=errno;
  232.     strmov(con->last_error,"socket read failed");
  233.     return 1;
  234.   }
  235.   net_buf_end=net_buf+num_bytes;
  236.   if ((con->eof=(net_buf[3]==' ')))
  237.     res_buf_shift--;
  238.   net_buf+=res_buf_shift;
  239.   res_buf_end[-1]=0;
  240.   for (;net_buf<net_buf_end && res_buf < res_buf_end;res_buf++,net_buf++)
  241.   {
  242.     if ((*res_buf=*net_buf) == 'r')
  243.     {
  244.       *res_buf=0;
  245.       break;
  246.     }
  247.   }
  248.   return 0;
  249. }