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

MySQL数据库

开发平台:

Visual C++

  1. /* Copyright (C) 2000 MySQL AB & MySQL Finland AB & TCX DataKonsult AB & Sasha
  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 "mysql_priv.h"
  14. #ifdef HAVE_REPLICATION
  15. #include "repl_failsafe.h"
  16. #include "sql_repl.h"
  17. #include "slave.h"
  18. #include "log_event.h"
  19. #include <mysql.h>
  20. #define SLAVE_LIST_CHUNK 128
  21. #define SLAVE_ERRMSG_SIZE (FN_REFLEN+64)
  22. RPL_STATUS rpl_status=RPL_NULL;
  23. pthread_mutex_t LOCK_rpl_status;
  24. pthread_cond_t COND_rpl_status;
  25. HASH slave_list;
  26. const char *rpl_role_type[] = {"MASTER","SLAVE",NullS};
  27. TYPELIB rpl_role_typelib = {array_elements(rpl_role_type)-1,"",
  28.     rpl_role_type, NULL};
  29. const char* rpl_status_type[]=
  30. {
  31.   "AUTH_MASTER","ACTIVE_SLAVE","IDLE_SLAVE", "LOST_SOLDIER","TROOP_SOLDIER",
  32.   "RECOVERY_CAPTAIN","NULL",NullS
  33. };
  34. TYPELIB rpl_status_typelib= {array_elements(rpl_status_type)-1,"",
  35.      rpl_status_type, NULL};
  36. static Slave_log_event* find_slave_event(IO_CACHE* log,
  37.  const char* log_file_name,
  38.  char* errmsg);
  39. /*
  40.   All of the functions defined in this file which are not used (the ones to
  41.   handle failsafe) are not used; their code has not been updated for more than
  42.   one year now so should be considered as BADLY BROKEN. Do not enable it.
  43.   The used functions (to handle LOAD DATA FROM MASTER, plus some small
  44.   functions like register_slave()) are working.
  45. */
  46. static int init_failsafe_rpl_thread(THD* thd)
  47. {
  48.   DBUG_ENTER("init_failsafe_rpl_thread");
  49.   /*
  50.     thd->bootstrap is to report errors barely to stderr; if this code is
  51.     enable again one day, one should check if bootstrap is still needed (maybe
  52.     this thread has no other error reporting method).
  53.   */
  54.   thd->system_thread = thd->bootstrap = 1;
  55.   thd->host_or_ip= "";
  56.   my_net_init(&thd->net, 0);
  57.   thd->net.read_timeout = slave_net_timeout;
  58.   thd->max_client_packet_length=thd->net.max_packet;
  59.   thd->master_access= ~(ulong)0;
  60.   thd->priv_user = 0;
  61.   pthread_mutex_lock(&LOCK_thread_count);
  62.   thd->thread_id = thread_id++;
  63.   pthread_mutex_unlock(&LOCK_thread_count);
  64.   if (init_thr_lock() || thd->store_globals())
  65.   {
  66.     close_connection(thd, ER_OUT_OF_RESOURCES, 1); // is this needed?
  67.     statistic_increment(aborted_connects,&LOCK_status);
  68.     end_thread(thd,0);
  69.     DBUG_RETURN(-1);
  70.   }
  71. #if !defined(__WIN__) && !defined(OS2) && !defined(__NETWARE__)
  72.   sigset_t set;
  73.   VOID(sigemptyset(&set)); // Get mask in use
  74.   VOID(pthread_sigmask(SIG_UNBLOCK,&set,&thd->block_signals));
  75. #endif
  76.   thd->mem_root->free= thd->mem_root->used= 0;
  77.   if (thd->variables.max_join_size == HA_POS_ERROR)
  78.     thd->options|= OPTION_BIG_SELECTS;
  79.   thd->proc_info="Thread initialized";
  80.   thd->version=refresh_version;
  81.   thd->set_time();
  82.   DBUG_RETURN(0);
  83. }
  84. void change_rpl_status(RPL_STATUS from_status, RPL_STATUS to_status)
  85. {
  86.   pthread_mutex_lock(&LOCK_rpl_status);
  87.   if (rpl_status == from_status || rpl_status == RPL_ANY)
  88.     rpl_status = to_status;
  89.   pthread_cond_signal(&COND_rpl_status);
  90.   pthread_mutex_unlock(&LOCK_rpl_status);
  91. }
  92. #define get_object(p, obj) 
  93. {
  94.   uint len = (uint)*p++;  
  95.   if (p + len > p_end || len >= sizeof(obj)) 
  96.     goto err; 
  97.   strmake(obj,(char*) p,len); 
  98.   p+= len; 
  99. }
  100. static inline int cmp_master_pos(Slave_log_event* sev, LEX_MASTER_INFO* mi)
  101. {
  102.   return cmp_master_pos(sev->master_log, sev->master_pos, mi->log_file_name,
  103. mi->pos);
  104. }
  105. void unregister_slave(THD* thd, bool only_mine, bool need_mutex)
  106. {
  107.   if (thd->server_id)
  108.   {
  109.     if (need_mutex)
  110.       pthread_mutex_lock(&LOCK_slave_list);
  111.     SLAVE_INFO* old_si;
  112.     if ((old_si = (SLAVE_INFO*)hash_search(&slave_list,
  113.    (byte*)&thd->server_id, 4)) &&
  114. (!only_mine || old_si->thd == thd))
  115.     hash_delete(&slave_list, (byte*)old_si);
  116.     if (need_mutex)
  117.       pthread_mutex_unlock(&LOCK_slave_list);
  118.   }
  119. }
  120. /*
  121.   Register slave in 'slave_list' hash table
  122.   RETURN VALUES
  123.   0 ok
  124.   1 Error.   Error message sent to client
  125. */
  126. int register_slave(THD* thd, uchar* packet, uint packet_length)
  127. {
  128.   int res;
  129.   SLAVE_INFO *si;
  130.   uchar *p= packet, *p_end= packet + packet_length;
  131.   if (check_access(thd, REPL_SLAVE_ACL, any_db,0,0,0))
  132.     return 1;
  133.   if (!(si = (SLAVE_INFO*)my_malloc(sizeof(SLAVE_INFO), MYF(MY_WME))))
  134.     goto err2;
  135.   thd->server_id= si->server_id= uint4korr(p);
  136.   p+= 4;
  137.   get_object(p,si->host);
  138.   get_object(p,si->user);
  139.   get_object(p,si->password);
  140.   if (p+10 > p_end)
  141.     goto err;
  142.   si->port= uint2korr(p);
  143.   p += 2;
  144.   si->rpl_recovery_rank= uint4korr(p);
  145.   p += 4;
  146.   if (!(si->master_id= uint4korr(p)))
  147.     si->master_id= server_id;
  148.   si->thd= thd;
  149.   pthread_mutex_lock(&LOCK_slave_list);
  150.   unregister_slave(thd,0,0);
  151.   res= my_hash_insert(&slave_list, (byte*) si);
  152.   pthread_mutex_unlock(&LOCK_slave_list);
  153.   return res;
  154. err:
  155.   my_free((gptr) si, MYF(MY_WME));
  156.   my_message(ER_UNKNOWN_ERROR, "Wrong parameters to function register_slave",
  157.      MYF(0));
  158. err2:
  159.   send_error(thd);
  160.   return 1;
  161. }
  162. extern "C" uint32
  163. *slave_list_key(SLAVE_INFO* si, uint* len,
  164. my_bool not_used __attribute__((unused)))
  165. {
  166.   *len = 4;
  167.   return &si->server_id;
  168. }
  169. extern "C" void slave_info_free(void *s)
  170. {
  171.   my_free((gptr) s, MYF(MY_WME));
  172. }
  173. void init_slave_list()
  174. {
  175.   hash_init(&slave_list, system_charset_info, SLAVE_LIST_CHUNK, 0, 0,
  176.     (hash_get_key) slave_list_key, (hash_free_key) slave_info_free, 0);
  177.   pthread_mutex_init(&LOCK_slave_list, MY_MUTEX_INIT_FAST);
  178. }
  179. void end_slave_list()
  180. {
  181.   /* No protection by a mutex needed as we are only called at shutdown */
  182.   if (hash_inited(&slave_list))
  183.   {
  184.     hash_free(&slave_list);
  185.     pthread_mutex_destroy(&LOCK_slave_list);
  186.   }
  187. }
  188. static int find_target_pos(LEX_MASTER_INFO *mi, IO_CACHE *log, char *errmsg)
  189. {
  190.   my_off_t log_pos =     (my_off_t) mi->pos;
  191.   uint32 target_server_id = mi->server_id;
  192.   for (;;)
  193.   {
  194.     Log_event* ev;
  195.     if (!(ev = Log_event::read_log_event(log, (pthread_mutex_t*) 0, 0)))
  196.     {
  197.       if (log->error > 0)
  198. strmov(errmsg, "Binary log truncated in the middle of event");
  199.       else if (log->error < 0)
  200. strmov(errmsg, "I/O error reading binary log");
  201.       else
  202. strmov(errmsg, "Could not find target event in the binary log");
  203.       return 1;
  204.     }
  205.     if (ev->log_pos >= log_pos && ev->server_id == target_server_id)
  206.     {
  207.       delete ev;
  208.       mi->pos = my_b_tell(log);
  209.       return 0;
  210.     }
  211.     delete ev;
  212.   }
  213.   /* Impossible */
  214. }
  215. /* 
  216.   Before 4.0.15 we had a member of THD called log_pos, it was meant for
  217.   failsafe replication code in repl_failsafe.cc which is disabled until
  218.   it is reworked. Event's log_pos used to be preserved through 
  219.   log-slave-updates to make code in repl_failsafe.cc work (this 
  220.   function, SHOW NEW MASTER); but on the other side it caused unexpected
  221.   values in Exec_Master_Log_Pos in A->B->C replication setup, 
  222.   synchronization problems in master_pos_wait(), ... So we 
  223.   (Dmitri & Guilhem) removed it.
  224.   
  225.   So for now this function is broken. 
  226. */
  227. int translate_master(THD* thd, LEX_MASTER_INFO* mi, char* errmsg)
  228. {
  229.   LOG_INFO linfo;
  230.   char last_log_name[FN_REFLEN];
  231.   IO_CACHE log;
  232.   File file = -1, last_file = -1;
  233.   pthread_mutex_t *log_lock;
  234.   const char* errmsg_p;
  235.   Slave_log_event* sev = 0;
  236.   my_off_t last_pos = 0;
  237.   int error = 1;
  238.   int cmp_res;
  239.   LINT_INIT(cmp_res);
  240.   DBUG_ENTER("translate_master");
  241.   if (!mysql_bin_log.is_open())
  242.   {
  243.     strmov(errmsg,"Binary log is not open");
  244.     DBUG_RETURN(1);
  245.   }
  246.   if (!server_id_supplied)
  247.   {
  248.     strmov(errmsg, "Misconfigured master - server id was not set");
  249.     DBUG_RETURN(1);
  250.   }
  251.   if (mysql_bin_log.find_log_pos(&linfo, NullS, 1))
  252.   {
  253.     strmov(errmsg,"Could not find first log");
  254.     DBUG_RETURN(1);
  255.   }
  256.   thd->current_linfo = &linfo;
  257.   bzero((char*) &log,sizeof(log));
  258.   log_lock = mysql_bin_log.get_log_lock();
  259.   pthread_mutex_lock(log_lock);
  260.   for (;;)
  261.   {
  262.     if ((file=open_binlog(&log, linfo.log_file_name, &errmsg_p)) < 0)
  263.     {
  264.       strmov(errmsg, errmsg_p);
  265.       goto err;
  266.     }
  267.     if (!(sev = find_slave_event(&log, linfo.log_file_name, errmsg)))
  268.       goto err;
  269.     cmp_res = cmp_master_pos(sev, mi);
  270.     delete sev;
  271.     if (!cmp_res)
  272.     {
  273.       /* Copy basename */
  274.       fn_format(mi->log_file_name, linfo.log_file_name, "","",1);
  275.       mi->pos = my_b_tell(&log);
  276.       goto mi_inited;
  277.     }
  278.     else if (cmp_res > 0)
  279.     {
  280.       if (!last_pos)
  281.       {
  282. strmov(errmsg,
  283.        "Slave event in first log points past the target position");
  284. goto err;
  285.       }
  286.       end_io_cache(&log);
  287.       (void) my_close(file, MYF(MY_WME));
  288.       if (init_io_cache(&log, (file = last_file), IO_SIZE, READ_CACHE, 0, 0,
  289. MYF(MY_WME)))
  290.       {
  291. errmsg[0] = 0;
  292. goto err;
  293.       }
  294.       break;
  295.     }
  296.     strmov(last_log_name, linfo.log_file_name);
  297.     last_pos = my_b_tell(&log);
  298.     switch (mysql_bin_log.find_next_log(&linfo, 1)) {
  299.     case LOG_INFO_EOF:
  300.       if (last_file >= 0)
  301.        (void)my_close(last_file, MYF(MY_WME));
  302.       last_file = -1;
  303.       goto found_log;
  304.     case 0:
  305.       break;
  306.     default:
  307.       strmov(errmsg, "Error reading log index");
  308.       goto err;
  309.     }
  310.     end_io_cache(&log);
  311.     if (last_file >= 0)
  312.      (void) my_close(last_file, MYF(MY_WME));
  313.     last_file = file;
  314.   }
  315. found_log:
  316.   my_b_seek(&log, last_pos);
  317.   if (find_target_pos(mi,&log,errmsg))
  318.     goto err;
  319.   fn_format(mi->log_file_name, last_log_name, "","",1);  /* Copy basename */
  320. mi_inited:
  321.   error = 0;
  322. err:
  323.   pthread_mutex_unlock(log_lock);
  324.   end_io_cache(&log);
  325.   pthread_mutex_lock(&LOCK_thread_count);
  326.   thd->current_linfo = 0;
  327.   pthread_mutex_unlock(&LOCK_thread_count);
  328.   if (file >= 0)
  329.     (void) my_close(file, MYF(MY_WME));
  330.   if (last_file >= 0 && last_file != file)
  331.     (void) my_close(last_file, MYF(MY_WME));
  332.   DBUG_RETURN(error);
  333. }
  334. /*
  335.   Caller must delete result when done
  336. */
  337. static Slave_log_event* find_slave_event(IO_CACHE* log,
  338.  const char* log_file_name,
  339.  char* errmsg)
  340. {
  341.   Log_event* ev;
  342.   int i;
  343.   bool slave_event_found = 0;
  344.   LINT_INIT(ev);
  345.   for (i = 0; i < 2; i++)
  346.   {
  347.     if (!(ev = Log_event::read_log_event(log, (pthread_mutex_t*)0, 0)))
  348.     {
  349.       my_snprintf(errmsg, SLAVE_ERRMSG_SIZE,
  350.   "Error reading event in log '%s'",
  351.   (char*)log_file_name);
  352.       return 0;
  353.     }
  354.     if (ev->get_type_code() == SLAVE_EVENT)
  355.     {
  356.       slave_event_found = 1;
  357.       break;
  358.     }
  359.     delete ev;
  360.   }
  361.   if (!slave_event_found)
  362.   {
  363.     my_snprintf(errmsg, SLAVE_ERRMSG_SIZE,
  364. "Could not find slave event in log '%s'",
  365. (char*)log_file_name);
  366.     return 0;
  367.   }
  368.   return (Slave_log_event*)ev;
  369. }
  370. /*
  371.    This function is broken now. See comment for translate_master().
  372.  */
  373. int show_new_master(THD* thd)
  374. {
  375.   Protocol *protocol= thd->protocol;
  376.   DBUG_ENTER("show_new_master");
  377.   List<Item> field_list;
  378.   char errmsg[SLAVE_ERRMSG_SIZE];
  379.   LEX_MASTER_INFO* lex_mi= &thd->lex->mi;
  380.   errmsg[0]=0; // Safety
  381.   if (translate_master(thd, lex_mi, errmsg))
  382.   {
  383.     if (errmsg[0])
  384.       my_error(ER_ERROR_WHEN_EXECUTING_COMMAND, MYF(0),
  385.        "SHOW NEW MASTER", errmsg);
  386.     DBUG_RETURN(-1);
  387.   }
  388.   else
  389.   {
  390.     field_list.push_back(new Item_empty_string("Log_name", 20));
  391.     field_list.push_back(new Item_return_int("Log_pos", 10,
  392.      MYSQL_TYPE_LONGLONG));
  393.     if (protocol->send_fields(&field_list, 1))
  394.       DBUG_RETURN(-1);
  395.     protocol->prepare_for_resend();
  396.     protocol->store(lex_mi->log_file_name, &my_charset_bin);
  397.     protocol->store((ulonglong) lex_mi->pos);
  398.     if (protocol->write())
  399.       DBUG_RETURN(-1);
  400.     send_eof(thd);
  401.     DBUG_RETURN(0);
  402.   }
  403. }
  404. /*
  405.   Asks the master for the list of its other connected slaves.
  406.   This is for failsafe replication: 
  407.   in order for failsafe replication to work, the servers involved in
  408.   replication must know of each other. We accomplish this by having each
  409.   slave report to the master how to reach it, and on connection, each
  410.   slave receives information about where the other slaves are.
  411.   SYNOPSIS
  412.     update_slave_list()
  413.     mysql           pre-existing connection to the master
  414.     mi              master info
  415.   NOTES
  416.     mi is used only to give detailed error messages which include the
  417.     hostname/port of the master, the username used by the slave to connect to
  418.     the master.
  419.     If the user used by the slave to connect to the master does not have the
  420.     REPLICATION SLAVE privilege, it will pop in this function because
  421.     SHOW SLAVE HOSTS will fail on the master.
  422.   RETURN VALUES
  423.     1           error
  424.     0           success
  425.  */
  426. int update_slave_list(MYSQL* mysql, MASTER_INFO* mi)
  427. {
  428.   MYSQL_RES* res=0;
  429.   MYSQL_ROW row;
  430.   const char* error=0;
  431.   bool have_auth_info;
  432.   int port_ind;
  433.   DBUG_ENTER("update_slave_list");
  434.   if (mysql_real_query(mysql,"SHOW SLAVE HOSTS",16) ||
  435.       !(res = mysql_store_result(mysql)))
  436.   {
  437.     error= mysql_error(mysql);
  438.     goto err;
  439.   }
  440.   switch (mysql_num_fields(res)) {
  441.   case 5:
  442.     have_auth_info = 0;
  443.     port_ind=2;
  444.     break;
  445.   case 7:
  446.     have_auth_info = 1;
  447.     port_ind=4;
  448.     break;
  449.   default:
  450.     error= "the master returned an invalid number of fields for SHOW SLAVE 
  451. HOSTS";
  452.     goto err;
  453.   }
  454.   pthread_mutex_lock(&LOCK_slave_list);
  455.   while ((row= mysql_fetch_row(res)))
  456.   {
  457.     uint32 server_id;
  458.     SLAVE_INFO* si, *old_si;
  459.     server_id = atoi(row[0]);
  460.     if ((old_si= (SLAVE_INFO*)hash_search(&slave_list,
  461.   (byte*)&server_id,4)))
  462.       si = old_si;
  463.     else
  464.     {
  465.       if (!(si = (SLAVE_INFO*)my_malloc(sizeof(SLAVE_INFO), MYF(MY_WME))))
  466.       {
  467. error= "the slave is out of memory";
  468. pthread_mutex_unlock(&LOCK_slave_list);
  469. goto err;
  470.       }
  471.       si->server_id = server_id;
  472.       my_hash_insert(&slave_list, (byte*)si);
  473.     }
  474.     strmake(si->host, row[1], sizeof(si->host)-1);
  475.     si->port = atoi(row[port_ind]);
  476.     si->rpl_recovery_rank = atoi(row[port_ind+1]);
  477.     si->master_id = atoi(row[port_ind+2]);
  478.     if (have_auth_info)
  479.     {
  480.       strmake(si->user, row[2], sizeof(si->user)-1);
  481.       strmake(si->password, row[3], sizeof(si->password)-1);
  482.     }
  483.   }
  484.   pthread_mutex_unlock(&LOCK_slave_list);
  485. err:
  486.   if (res)
  487.     mysql_free_result(res);
  488.   if (error)
  489.   {
  490.     sql_print_error("While trying to obtain the list of slaves from the master 
  491. '%s:%d', user '%s' got the following error: '%s'", 
  492.                     mi->host, mi->port, mi->user, error);
  493.     DBUG_RETURN(1);
  494.   }
  495.   DBUG_RETURN(0);
  496. }
  497. int find_recovery_captain(THD* thd, MYSQL* mysql)
  498. {
  499.   return 0;
  500. }
  501. pthread_handler_decl(handle_failsafe_rpl,arg)
  502. {
  503.   DBUG_ENTER("handle_failsafe_rpl");
  504.   THD *thd = new THD;
  505.   thd->thread_stack = (char*)&thd;
  506.   MYSQL* recovery_captain = 0;
  507.   const char* msg;
  508.   pthread_detach_this_thread();
  509.   if (init_failsafe_rpl_thread(thd) || !(recovery_captain=mysql_init(0)))
  510.   {
  511.     sql_print_error("Could not initialize failsafe replication thread");
  512.     goto err;
  513.   }
  514.   pthread_mutex_lock(&LOCK_rpl_status);
  515.   msg= thd->enter_cond(&COND_rpl_status,
  516.                        &LOCK_rpl_status, "Waiting for request");
  517.   while (!thd->killed && !abort_loop)
  518.   {
  519.     bool break_req_chain = 0;
  520.     pthread_cond_wait(&COND_rpl_status, &LOCK_rpl_status);
  521.     thd->proc_info="Processing request";
  522.     while (!break_req_chain)
  523.     {
  524.       switch (rpl_status) {
  525.       case RPL_LOST_SOLDIER:
  526. if (find_recovery_captain(thd, recovery_captain))
  527.   rpl_status=RPL_TROOP_SOLDIER;
  528. else
  529.   rpl_status=RPL_RECOVERY_CAPTAIN;
  530. break_req_chain=1; /* for now until other states are implemented */
  531. break;
  532.       default:
  533. break_req_chain=1;
  534. break;
  535.       }
  536.     }
  537.   }
  538.   thd->exit_cond(msg);
  539. err:
  540.   if (recovery_captain)
  541.     mysql_close(recovery_captain);
  542.   delete thd;
  543.   my_thread_end();
  544.   pthread_exit(0);
  545.   DBUG_RETURN(0);
  546. }
  547. int show_slave_hosts(THD* thd)
  548. {
  549.   List<Item> field_list;
  550.   Protocol *protocol= thd->protocol;
  551.   DBUG_ENTER("show_slave_hosts");
  552.   field_list.push_back(new Item_return_int("Server_id", 10,
  553.    MYSQL_TYPE_LONG));
  554.   field_list.push_back(new Item_empty_string("Host", 20));
  555.   if (opt_show_slave_auth_info)
  556.   {
  557.     field_list.push_back(new Item_empty_string("User",20));
  558.     field_list.push_back(new Item_empty_string("Password",20));
  559.   }
  560.   field_list.push_back(new Item_return_int("Port", 7, MYSQL_TYPE_LONG));
  561.   field_list.push_back(new Item_return_int("Rpl_recovery_rank", 7,
  562.    MYSQL_TYPE_LONG));
  563.   field_list.push_back(new Item_return_int("Master_id", 10,
  564.    MYSQL_TYPE_LONG));
  565.   if (protocol->send_fields(&field_list, 1))
  566.     DBUG_RETURN(-1);
  567.   pthread_mutex_lock(&LOCK_slave_list);
  568.   for (uint i = 0; i < slave_list.records; ++i)
  569.   {
  570.     SLAVE_INFO* si = (SLAVE_INFO*) hash_element(&slave_list, i);
  571.     protocol->prepare_for_resend();
  572.     protocol->store((uint32) si->server_id);
  573.     protocol->store(si->host, &my_charset_bin);
  574.     if (opt_show_slave_auth_info)
  575.     {
  576.       protocol->store(si->user, &my_charset_bin);
  577.       protocol->store(si->password, &my_charset_bin);
  578.     }
  579.     protocol->store((uint32) si->port);
  580.     protocol->store((uint32) si->rpl_recovery_rank);
  581.     protocol->store((uint32) si->master_id);
  582.     if (protocol->write())
  583.     {
  584.       pthread_mutex_unlock(&LOCK_slave_list);
  585.       DBUG_RETURN(-1);
  586.     }
  587.   }
  588.   pthread_mutex_unlock(&LOCK_slave_list);
  589.   send_eof(thd);
  590.   DBUG_RETURN(0);
  591. }
  592. int connect_to_master(THD *thd, MYSQL* mysql, MASTER_INFO* mi)
  593. {
  594.   DBUG_ENTER("connect_to_master");
  595.   if (!mi->host || !*mi->host) /* empty host */
  596.   {
  597.     strmov(mysql->net.last_error, "Master is not configured");
  598.     DBUG_RETURN(1);
  599.   }
  600.   mysql_options(mysql, MYSQL_OPT_CONNECT_TIMEOUT, (char *) &slave_net_timeout);
  601.   mysql_options(mysql, MYSQL_OPT_READ_TIMEOUT, (char *) &slave_net_timeout);
  602. #ifdef HAVE_OPENSSL
  603.   if (mi->ssl)
  604.     mysql_ssl_set(mysql, 
  605.         mi->ssl_key[0]?mi->ssl_key:0,
  606.         mi->ssl_cert[0]?mi->ssl_cert:0,
  607.         mi->ssl_ca[0]?mi->ssl_ca:0, 
  608.         mi->ssl_capath[0]?mi->ssl_capath:0,
  609.         mi->ssl_cipher[0]?mi->ssl_cipher:0);
  610. #endif
  611.     
  612.   mysql_options(mysql, MYSQL_SET_CHARSET_NAME, default_charset_info->csname);
  613.   mysql_options(mysql, MYSQL_SET_CHARSET_DIR, (char *) charsets_dir);
  614.   if (!mysql_real_connect(mysql, mi->host, mi->user, mi->password, 0,
  615. mi->port, 0, 0))
  616.     DBUG_RETURN(1);
  617.   DBUG_RETURN(0);
  618. }
  619. static inline void cleanup_mysql_results(MYSQL_RES* db_res,
  620.  MYSQL_RES** cur, MYSQL_RES** start)
  621. {
  622.   for (; cur >= start; --cur)
  623.   {
  624.     if (*cur)
  625.       mysql_free_result(*cur);
  626.   }
  627.   mysql_free_result(db_res);
  628. }
  629. static int fetch_db_tables(THD *thd, MYSQL *mysql, const char *db,
  630.    MYSQL_RES *table_res, MASTER_INFO *mi)
  631. {
  632.   MYSQL_ROW row;
  633.   for (row = mysql_fetch_row(table_res); row;
  634.        row = mysql_fetch_row(table_res))
  635.   {
  636.     TABLE_LIST table;
  637.     const char* table_name= row[0];
  638.     int error;
  639.     if (table_rules_on)
  640.     {
  641.       bzero((char*) &table, sizeof(table)); //just for safe
  642.       table.db= (char*) db;
  643.       table.real_name= (char*) table_name;
  644.       table.updating= 1;
  645.       if (!tables_ok(thd, &table))
  646. continue;
  647.     }
  648.     /* download master's table and overwrite slave's table */
  649.     if ((error= fetch_master_table(thd, db, table_name, mi, mysql, 1)))
  650.       return error;
  651.   }
  652.   return 0;
  653. }
  654. /*
  655.   Load all MyISAM tables from master to this slave.
  656.   REQUIREMENTS
  657.    - No active transaction (flush_relay_log_info would not work in this case)
  658. */
  659. int load_master_data(THD* thd)
  660. {
  661.   MYSQL mysql;
  662.   MYSQL_RES* master_status_res = 0;
  663.   int error = 0;
  664.   const char* errmsg=0;
  665.   int restart_thread_mask;
  666.   HA_CREATE_INFO create_info;
  667.   mysql_init(&mysql);
  668.   /*
  669.     We do not want anyone messing with the slave at all for the entire
  670.     duration of the data load.
  671.   */
  672.   pthread_mutex_lock(&LOCK_active_mi);
  673.   lock_slave_threads(active_mi);
  674.   init_thread_mask(&restart_thread_mask,active_mi,0 /*not inverse*/);
  675.   if (restart_thread_mask &&
  676.       (error=terminate_slave_threads(active_mi,restart_thread_mask,
  677.      1 /*skip lock*/)))
  678.   {
  679.     send_error(thd,error);
  680.     unlock_slave_threads(active_mi);
  681.     pthread_mutex_unlock(&LOCK_active_mi);
  682.     return 1;
  683.   }
  684.   
  685.   if (connect_to_master(thd, &mysql, active_mi))
  686.   {
  687.     net_printf(thd, error= ER_CONNECT_TO_MASTER,
  688.        mysql_error(&mysql));
  689.     goto err;
  690.   }
  691.   // now that we are connected, get all database and tables in each
  692.   {
  693.     MYSQL_RES *db_res, **table_res, **table_res_end, **cur_table_res;
  694.     uint num_dbs;
  695.     if (mysql_real_query(&mysql, "SHOW DATABASES", 14) ||
  696. !(db_res = mysql_store_result(&mysql)))
  697.     {
  698.       net_printf(thd, error = ER_QUERY_ON_MASTER,
  699.  mysql_error(&mysql));
  700.       goto err;
  701.     }
  702.     if (!(num_dbs = (uint) mysql_num_rows(db_res)))
  703.       goto err;
  704.     /*
  705.       In theory, the master could have no databases at all
  706.       and run with skip-grant
  707.     */
  708.     if (!(table_res = (MYSQL_RES**)thd->alloc(num_dbs * sizeof(MYSQL_RES*))))
  709.     {
  710.       net_printf(thd, error = ER_OUTOFMEMORY);
  711.       goto err;
  712.     }
  713.     /*
  714.       This is a temporary solution until we have online backup
  715.       capabilities - to be replaced once online backup is working
  716.       we wait to issue FLUSH TABLES WITH READ LOCK for as long as we
  717.       can to minimize the lock time.
  718.     */
  719.     if (mysql_real_query(&mysql, "FLUSH TABLES WITH READ LOCK", 27) ||
  720. mysql_real_query(&mysql, "SHOW MASTER STATUS",18) ||
  721. !(master_status_res = mysql_store_result(&mysql)))
  722.     {
  723.       net_printf(thd, error = ER_QUERY_ON_MASTER,
  724.  mysql_error(&mysql));
  725.       goto err;
  726.     }
  727.     /*
  728.       Go through every table in every database, and if the replication
  729.       rules allow replicating it, get it
  730.     */
  731.     table_res_end = table_res + num_dbs;
  732.     for (cur_table_res = table_res; cur_table_res < table_res_end;
  733.  cur_table_res++)
  734.     {
  735.       // since we know how many rows we have, this can never be NULL
  736.       MYSQL_ROW row = mysql_fetch_row(db_res);
  737.       char* db = row[0];
  738.       /*
  739. Do not replicate databases excluded by rules. We also test
  740. replicate_wild_*_table rules (replicate_wild_ignore_table='db1.%' will
  741. be considered as "ignore the 'db1' database as a whole, as it already
  742. works for CREATE DATABASE and DROP DATABASE).
  743. Also skip 'mysql' database - in most cases the user will
  744. mess up and not exclude mysql database with the rules when
  745. he actually means to - in this case, he is up for a surprise if
  746. his priv tables get dropped and downloaded from master
  747. TODO - add special option, not enabled
  748. by default, to allow inclusion of mysql database into load
  749. data from master
  750.       */
  751.       if (!db_ok(db, replicate_do_db, replicate_ignore_db) ||
  752.           !db_ok_with_wild_table(db) ||
  753.   !strcmp(db,"mysql"))
  754.       {
  755. *cur_table_res = 0;
  756. continue;
  757.       }
  758.       bzero((char*) &create_info, sizeof(create_info));
  759.       create_info.options= HA_LEX_CREATE_IF_NOT_EXISTS;
  760.       if (mysql_create_db(thd, db, &create_info, 1))
  761.       {
  762. send_error(thd, 0, 0);
  763. cleanup_mysql_results(db_res, cur_table_res - 1, table_res);
  764. goto err;
  765.       }
  766.       if (mysql_select_db(&mysql, db) ||
  767.   mysql_real_query(&mysql, "SHOW TABLES", 11) ||
  768.   !(*cur_table_res = mysql_store_result(&mysql)))
  769.       {
  770. net_printf(thd, error = ER_QUERY_ON_MASTER,
  771.    mysql_error(&mysql));
  772. cleanup_mysql_results(db_res, cur_table_res - 1, table_res);
  773. goto err;
  774.       }
  775.       if ((error = fetch_db_tables(thd,&mysql,db,*cur_table_res,active_mi)))
  776.       {
  777. // we do not report the error - fetch_db_tables handles it
  778. cleanup_mysql_results(db_res, cur_table_res, table_res);
  779. goto err;
  780.       }
  781.     }
  782.     cleanup_mysql_results(db_res, cur_table_res - 1, table_res);
  783.     // adjust replication coordinates from the master
  784.     if (master_status_res)
  785.     {
  786.       MYSQL_ROW row = mysql_fetch_row(master_status_res);
  787.       /*
  788. We need this check because the master may not be running with
  789. log-bin, but it will still allow us to do all the steps
  790. of LOAD DATA FROM MASTER - no reason to forbid it, really,
  791. although it does not make much sense for the user to do it
  792.       */
  793.       if (row && row[0] && row[1])
  794.       {
  795.         /*
  796.           If the slave's master info is not inited, we init it, then we write
  797.           the new coordinates to it. Must call init_master_info() *before*
  798.           setting active_mi, because init_master_info() sets active_mi with
  799.           defaults.
  800.         */
  801.         int error;
  802.         if (init_master_info(active_mi, master_info_file, relay_log_info_file, 
  803.      0, (SLAVE_IO | SLAVE_SQL)))
  804.           send_error(thd, ER_MASTER_INFO);
  805. strmake(active_mi->master_log_name, row[0],
  806. sizeof(active_mi->master_log_name));
  807. active_mi->master_log_pos= my_strtoll10(row[1], (char**) 0, &error);
  808.         /* at least in recent versions, the condition below should be false */
  809. if (active_mi->master_log_pos < BIN_LOG_HEADER_SIZE)
  810.   active_mi->master_log_pos = BIN_LOG_HEADER_SIZE;
  811.         /*
  812.           Relay log's IO_CACHE may not be inited (even if we are sure that some
  813.           host was specified; there could have been a problem when replication
  814.           started, which led to relay log's IO_CACHE to not be inited.
  815.         */
  816. flush_master_info(active_mi, 0);
  817.       }
  818.       mysql_free_result(master_status_res);
  819.     }
  820.     if (mysql_real_query(&mysql, "UNLOCK TABLES", 13))
  821.     {
  822.       net_printf(thd, error = ER_QUERY_ON_MASTER,
  823.  mysql_error(&mysql));
  824.       goto err;
  825.     }
  826.   }
  827.   thd->proc_info="purging old relay logs";
  828.   if (purge_relay_logs(&active_mi->rli,thd,
  829.        0 /* not only reset, but also reinit */,
  830.        &errmsg))
  831.   {
  832.     send_error(thd, 0, "Failed purging old relay logs");
  833.     unlock_slave_threads(active_mi);
  834.     pthread_mutex_unlock(&LOCK_active_mi);
  835.     return 1;
  836.   }
  837.   pthread_mutex_lock(&active_mi->rli.data_lock);
  838.   active_mi->rli.group_master_log_pos = active_mi->master_log_pos;
  839.   strmake(active_mi->rli.group_master_log_name,active_mi->master_log_name,
  840.   sizeof(active_mi->rli.group_master_log_name)-1);
  841.   /*
  842.      Cancel the previous START SLAVE UNTIL, as the fact to download
  843.      a new copy logically makes UNTIL irrelevant.
  844.   */
  845.   clear_until_condition(&active_mi->rli);
  846.   /*
  847.     No need to update rli.event* coordinates, they will be when the slave
  848.     threads start ; only rli.group* coordinates are necessary here.
  849.   */
  850.   flush_relay_log_info(&active_mi->rli);
  851.   pthread_cond_broadcast(&active_mi->rli.data_cond);
  852.   pthread_mutex_unlock(&active_mi->rli.data_lock);
  853.   thd->proc_info = "starting slave";
  854.   if (restart_thread_mask)
  855.   {
  856.     error=start_slave_threads(0 /* mutex not needed */,
  857.       1 /* wait for start */,
  858.       active_mi,master_info_file,relay_log_info_file,
  859.       restart_thread_mask);
  860.   }
  861. err:
  862.   unlock_slave_threads(active_mi);
  863.   pthread_mutex_unlock(&LOCK_active_mi);
  864.   thd->proc_info = 0;
  865.   mysql_close(&mysql); // safe to call since we always do mysql_init()
  866.   if (!error)
  867.     send_ok(thd);
  868.   return error;
  869. }
  870. #endif /* HAVE_REPLICATION */