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

MySQL数据库

开发平台:

Visual C++

  1. /* Copyright (C) 2000 MySQL AB
  2.    This program is free software; you can redistribute it and/or modify
  3.    it under the terms of the GNU General Public License as published by
  4.    the Free Software Foundation; either version 2 of the License, or
  5.    (at your option) any later version.
  6.    This program is distributed in the hope that it will be useful,
  7.    but WITHOUT ANY WARRANTY; without even the implied warranty of
  8.    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  9.    GNU General Public License for more details.
  10.    You should have received a copy of the GNU General Public License
  11.    along with this program; if not, write to the Free Software
  12.    Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA */
  13. /* By Jani Tolonen, 2001-04-20, MySQL Development Team */
  14. #define CHECK_VERSION "2.4.4"
  15. #include "client_priv.h"
  16. #include <m_ctype.h>
  17. #include <mysql_version.h>
  18. #include <mysqld_error.h>
  19. #include <sslopt-vars.h>
  20. /* Exit codes */
  21. #define EX_USAGE 1
  22. #define EX_MYSQLERR 2
  23. static MYSQL mysql_connection, *sock = 0;
  24. static my_bool opt_alldbs = 0, opt_check_only_changed = 0, opt_extended = 0,
  25.                opt_compress = 0, opt_databases = 0, opt_fast = 0,
  26.                opt_medium_check = 0, opt_quick = 0, opt_all_in_1 = 0,
  27.                opt_silent = 0, opt_auto_repair = 0, ignore_errors = 0,
  28.                tty_password = 0, opt_frm = 0;
  29. static uint verbose = 0, opt_mysql_port=0;
  30. static my_string opt_mysql_unix_port = 0;
  31. static char *opt_password = 0, *current_user = 0, 
  32.     *default_charset = (char *)MYSQL_DEFAULT_CHARSET_NAME,
  33.     *current_host = 0;
  34. static int first_error = 0;
  35. DYNAMIC_ARRAY tables4repair;
  36. #ifdef HAVE_SMEM
  37. static char *shared_memory_base_name=0;
  38. #endif
  39. static uint opt_protocol=0;
  40. static CHARSET_INFO *charset_info= &my_charset_latin1;
  41. enum operations {DO_CHECK, DO_REPAIR, DO_ANALYZE, DO_OPTIMIZE};
  42. static struct my_option my_long_options[] =
  43. {
  44.   {"all-databases", 'A',
  45.    "Check all the databases. This will be same as  --databases with all databases selected.",
  46.    (gptr*) &opt_alldbs, (gptr*) &opt_alldbs, 0, GET_BOOL, NO_ARG, 0, 0, 0, 0,
  47.    0, 0},
  48.   {"analyze", 'a', "Analyze given tables.", 0, 0, 0, GET_NO_ARG, NO_ARG, 0, 0,
  49.    0, 0, 0, 0},
  50.   {"all-in-1", '1',
  51.    "Instead of issuing one query for each table, use one query per database, naming all tables in the database in a comma-separated list.",
  52.    (gptr*) &opt_all_in_1, (gptr*) &opt_all_in_1, 0, GET_BOOL, NO_ARG, 0, 0, 0,
  53.    0, 0, 0},
  54. #ifdef __NETWARE__
  55.   {"auto-close", OPT_AUTO_CLOSE, "Auto close the screen on exit for Netware.",
  56.    0, 0, 0, GET_NO_ARG, NO_ARG, 0, 0, 0, 0, 0, 0},
  57. #endif
  58.   {"auto-repair", OPT_AUTO_REPAIR,
  59.    "If a checked table is corrupted, automatically fix it. Repairing will be done after all tables have been checked, if corrupted ones were found.",
  60.    (gptr*) &opt_auto_repair, (gptr*) &opt_auto_repair, 0, GET_BOOL, NO_ARG, 0,
  61.    0, 0, 0, 0, 0},
  62.   {"character-sets-dir", OPT_CHARSETS_DIR,
  63.    "Directory where character sets are.", (gptr*) &charsets_dir,
  64.    (gptr*) &charsets_dir, 0, GET_STR, REQUIRED_ARG, 0, 0, 0, 0, 0, 0},
  65.   {"check", 'c', "Check table for errors.", 0, 0, 0, GET_NO_ARG, NO_ARG, 0, 0,
  66.    0, 0, 0, 0},
  67.   {"check-only-changed", 'C',
  68.    "Check only tables that have changed since last check or haven't been closed properly.",
  69.    0, 0, 0, GET_NO_ARG, NO_ARG, 0, 0, 0, 0, 0, 0},
  70.   {"compress", OPT_COMPRESS, "Use compression in server/client protocol.",
  71.    (gptr*) &opt_compress, (gptr*) &opt_compress, 0, GET_BOOL, NO_ARG, 0, 0, 0,
  72.    0, 0, 0},
  73.   {"databases", 'B',
  74.    "To check several databases. Note the difference in usage; In this case no tables are given. All name arguments are regarded as databasenames.",
  75.    (gptr*) &opt_databases, (gptr*) &opt_databases, 0, GET_BOOL, NO_ARG,
  76.    0, 0, 0, 0, 0, 0},
  77. #ifdef DBUG_OFF
  78.   {"debug", '#', "This is a non-debug version. Catch this and exit.",
  79.    0, 0, 0, GET_DISABLED, OPT_ARG, 0, 0, 0, 0, 0, 0},
  80. #else
  81.   {"debug", '#', "Output debug log. Often this is 'd:t:o,filename'.",
  82.    0, 0, 0, GET_STR, OPT_ARG, 0, 0, 0, 0, 0, 0},
  83. #endif
  84.   {"default-character-set", OPT_DEFAULT_CHARSET,
  85.    "Set the default character set.", (gptr*) &default_charset,
  86.    (gptr*) &default_charset, 0, GET_STR, REQUIRED_ARG, 0, 0, 0, 0, 0, 0},
  87.   {"fast",'F', "Check only tables that haven't been closed properly.",
  88.    (gptr*) &opt_fast, (gptr*) &opt_fast, 0, GET_BOOL, NO_ARG, 0, 0, 0, 0, 0,
  89.    0},
  90.   {"force", 'f', "Continue even if we get an sql-error.",
  91.    (gptr*) &ignore_errors, (gptr*) &ignore_errors, 0, GET_BOOL, NO_ARG, 0, 0,
  92.    0, 0, 0, 0},
  93.   {"extended", 'e',
  94.    "If you are using this option with CHECK TABLE, it will ensure that the table is 100 percent consistent, but will take a long time. If you are using this option with REPAIR TABLE, it will force using old slow repair with keycache method, instead of much faster repair by sorting.",
  95.    (gptr*) &opt_extended, (gptr*) &opt_extended, 0, GET_BOOL, NO_ARG, 0, 0, 0,
  96.    0, 0, 0},
  97.   {"help", '?', "Display this help message and exit.", 0, 0, 0, GET_NO_ARG,
  98.    NO_ARG, 0, 0, 0, 0, 0, 0},
  99.   {"host",'h', "Connect to host.", (gptr*) &current_host,
  100.    (gptr*) &current_host, 0, GET_STR_ALLOC, REQUIRED_ARG, 0, 0, 0, 0, 0, 0},
  101.   {"medium-check", 'm',
  102.    "Faster than extended-check, but only finds 99.99 percent of all errors. Should be good enough for most cases.",
  103.    0, 0, 0, GET_NO_ARG, NO_ARG, 0, 0, 0, 0, 0, 0},
  104.   {"optimize", 'o', "Optimize table.", 0, 0, 0, GET_NO_ARG, NO_ARG, 0, 0, 0, 0,
  105.    0, 0},
  106.   {"password", 'p',
  107.    "Password to use when connecting to server. If password is not given it's solicited on the tty.",
  108.    0, 0, 0, GET_STR, OPT_ARG, 0, 0, 0, 0, 0, 0},
  109. #ifdef __WIN__
  110.   {"pipe", 'W', "Use named pipes to connect to server.", 0, 0, 0, GET_NO_ARG,
  111.    NO_ARG, 0, 0, 0, 0, 0, 0},
  112. #endif
  113.   {"port", 'P', "Port number to use for connection.", (gptr*) &opt_mysql_port,
  114.    (gptr*) &opt_mysql_port, 0, GET_UINT, REQUIRED_ARG, MYSQL_PORT, 0, 0, 0, 0,
  115.    0},
  116.   {"protocol", OPT_MYSQL_PROTOCOL, "The protocol of connection (tcp,socket,pipe,memory).",
  117.    0, 0, 0, GET_STR,  REQUIRED_ARG, 0, 0, 0, 0, 0, 0},
  118.   {"quick", 'q',
  119.    "If you are using this option with CHECK TABLE, it prevents the check from scanning the rows to check for wrong links. This is the fastest check. If you are using this option with REPAIR TABLE, it will try to repair only the index tree. This is the fastest repair method for a table.",
  120.    (gptr*) &opt_quick, (gptr*) &opt_quick, 0, GET_BOOL, NO_ARG, 0, 0, 0, 0, 0,
  121.    0},
  122.   {"repair", 'r',
  123.    "Can fix almost anything except unique keys that aren't unique.",
  124.    0, 0, 0, GET_NO_ARG, NO_ARG, 0, 0, 0, 0, 0, 0},
  125. #ifdef HAVE_SMEM
  126.   {"shared-memory-base-name", OPT_SHARED_MEMORY_BASE_NAME,
  127.    "Base name of shared memory.", (gptr*) &shared_memory_base_name, (gptr*) &shared_memory_base_name,
  128.    0, GET_STR_ALLOC, REQUIRED_ARG, 0, 0, 0, 0, 0, 0},
  129. #endif
  130.   {"silent", 's', "Print only error messages.", (gptr*) &opt_silent,
  131.    (gptr*) &opt_silent, 0, GET_BOOL, NO_ARG, 0, 0, 0, 0, 0, 0},
  132.   {"socket", 'S', "Socket file to use for connection.",
  133.    (gptr*) &opt_mysql_unix_port, (gptr*) &opt_mysql_unix_port, 0, GET_STR,
  134.    REQUIRED_ARG, 0, 0, 0, 0, 0, 0},
  135. #include <sslopt-longopts.h>
  136.   {"tables", OPT_TABLES, "Overrides option --databases (-B).", 0, 0, 0,
  137.    GET_NO_ARG, NO_ARG, 0, 0, 0, 0, 0, 0},
  138. #ifndef DONT_ALLOW_USER_CHANGE
  139.   {"user", 'u', "User for login if not current user.", (gptr*) &current_user,
  140.    (gptr*) &current_user, 0, GET_STR, REQUIRED_ARG, 0, 0, 0, 0, 0, 0},
  141. #endif
  142.   {"use-frm", OPT_FRM,
  143.    "When used with REPAIR, get table structure from .frm file, so the table can be repaired even if .MYI header is corrupted.",
  144.    (gptr*) &opt_frm, (gptr*) &opt_frm, 0, GET_BOOL, NO_ARG, 0, 0, 0, 0, 0,
  145.    0},
  146.   {"verbose", 'v', "Print info about the various stages.", 0, 0, 0, GET_NO_ARG,
  147.    NO_ARG, 0, 0, 0, 0, 0, 0},
  148.   {"version", 'V', "Output version information and exit.", 0, 0, 0, GET_NO_ARG,
  149.    NO_ARG, 0, 0, 0, 0, 0, 0},
  150.   {0, 0, 0, 0, 0, 0, GET_NO_ARG, NO_ARG, 0, 0, 0, 0, 0, 0}
  151. };
  152. static const char *load_default_groups[] = { "mysqlcheck", "client", 0 };
  153. static void print_version(void);
  154. static void usage(void);
  155. static int get_options(int *argc, char ***argv);
  156. static int process_all_databases();
  157. static int process_databases(char **db_names);
  158. static int process_selected_tables(char *db, char **table_names, int tables);
  159. static int process_all_tables_in_db(char *database);
  160. static int use_db(char *database);
  161. static int handle_request_for_tables(char *tables, uint length);
  162. static int dbConnect(char *host, char *user,char *passwd);
  163. static void dbDisconnect(char *host);
  164. static void DBerror(MYSQL *mysql, const char *when);
  165. static void safe_exit(int error);
  166. static void print_result();
  167. static char *fix_table_name(char *dest, char *src);
  168. int what_to_do = 0;
  169. #include <help_start.h>
  170. static void print_version(void)
  171. {
  172.   printf("%s  Ver %s Distrib %s, for %s (%s)n", my_progname, CHECK_VERSION,
  173.    MYSQL_SERVER_VERSION, SYSTEM_TYPE, MACHINE_TYPE);
  174.   NETWARE_SET_SCREEN_MODE(1);
  175. } /* print_version */
  176. static void usage(void)
  177. {
  178.   print_version();
  179.   puts("By Jani Tolonen, 2001-04-20, MySQL Development Teamn");
  180.   puts("This software comes with ABSOLUTELY NO WARRANTY. This is free software,n");
  181.   puts("and you are welcome to modify and redistribute it under the GPL license.n");
  182.   puts("This program can be used to CHECK (-c,-m,-C), REPAIR (-r), ANALYZE (-a)");
  183.   puts("or OPTIMIZE (-o) tables. Some of the options (like -e or -q) can be");
  184.   puts("used at the same time. Not all options are supported by all storage engines.");
  185.   puts("Please consult the MySQL manual for latest information about the");
  186.   puts("above. The options -c,-r,-a and -o are exclusive to each other, which");
  187.   puts("means that the last option will be used, if several was specified.n");
  188.   puts("The option -c will be used by default, if none was specified. You");
  189.   puts("can change the default behavior by making a symbolic link, or");
  190.   puts("copying this file somewhere with another name, the alternatives are:");
  191.   puts("mysqlrepair:   The default option will be -r");
  192.   puts("mysqlanalyze:  The default option will be -a");
  193.   puts("mysqloptimize: The default option will be -on");
  194.   printf("Usage: %s [OPTIONS] database [tables]n", my_progname);
  195.   printf("OR     %s [OPTIONS] --databases DB1 [DB2 DB3...]n",
  196.  my_progname);
  197.   printf("OR     %s [OPTIONS] --all-databasesn", my_progname);
  198.   print_defaults("my", load_default_groups);
  199.   my_print_help(my_long_options);
  200.   my_print_variables(my_long_options);
  201. } /* usage */
  202. #include <help_end.h>
  203. static my_bool
  204. get_one_option(int optid, const struct my_option *opt __attribute__((unused)),
  205.        char *argument)
  206. {
  207.   switch(optid) {
  208. #ifdef __NETWARE__
  209.   case OPT_AUTO_CLOSE:
  210.     setscreenmode(SCR_AUTOCLOSE_ON_EXIT);
  211.     break;
  212. #endif
  213.   case 'a':
  214.     what_to_do = DO_ANALYZE;
  215.     break;
  216.   case 'c':
  217.     what_to_do = DO_CHECK;
  218.     break;
  219.   case 'C':
  220.     what_to_do = DO_CHECK;
  221.     opt_check_only_changed = 1;
  222.     break;
  223.   case 'I': /* Fall through */
  224.   case '?':
  225.     usage();
  226.     exit(0);
  227.   case 'm':
  228.     what_to_do = DO_CHECK;
  229.     opt_medium_check = 1;
  230.     break;
  231.   case 'o':
  232.     what_to_do = DO_OPTIMIZE;
  233.     break;
  234.   case 'p':
  235.     if (argument)
  236.     {
  237.       char *start = argument;
  238.       my_free(opt_password, MYF(MY_ALLOW_ZERO_PTR));
  239.       opt_password = my_strdup(argument, MYF(MY_FAE));
  240.       while (*argument) *argument++= 'x'; /* Destroy argument */
  241.       if (*start)
  242. start[1] = 0;                             /* Cut length of argument */
  243.       tty_password= 0;
  244.     }
  245.     else
  246.       tty_password = 1;
  247.     break;
  248.   case 'r':
  249.     what_to_do = DO_REPAIR;
  250.     break;
  251.   case 'W':
  252. #ifdef __WIN__
  253.     opt_protocol = MYSQL_PROTOCOL_PIPE;
  254. #endif
  255.     break;
  256.   case '#':
  257.     DBUG_PUSH(argument ? argument : "d:t:o");
  258.     break;
  259. #include <sslopt-case.h>
  260.   case OPT_TABLES:
  261.     opt_databases = 0;
  262.     break;
  263.   case 'v':
  264.     verbose++;
  265.     break;
  266.   case 'V': print_version(); exit(0);
  267.   case OPT_MYSQL_PROTOCOL:
  268.   {
  269.     if ((opt_protocol= find_type(argument, &sql_protocol_typelib,0)) <= 0)
  270.     {
  271.       fprintf(stderr, "Unknown option to protocol: %sn", argument);
  272.       exit(1);
  273.     }
  274.     break;
  275.   }
  276.   }
  277.   return 0;
  278. }
  279. static int get_options(int *argc, char ***argv)
  280. {
  281.   int ho_error;
  282.   if (*argc == 1)
  283.   {
  284.     usage();
  285.     exit(0);
  286.   }
  287.   load_defaults("my", load_default_groups, argc, argv);
  288.   if ((ho_error=handle_options(argc, argv, my_long_options, get_one_option)))
  289.     exit(ho_error);
  290.   if (!what_to_do)
  291.   {
  292.     int pnlen = strlen(my_progname);
  293.     if (pnlen < 6) /* name too short */
  294.       what_to_do = DO_CHECK;
  295.     else if (!strcmp("repair", my_progname + pnlen - 6))
  296.       what_to_do = DO_REPAIR;
  297.     else if (!strcmp("analyze", my_progname + pnlen - 7))
  298.       what_to_do = DO_ANALYZE;
  299.     else if  (!strcmp("optimize", my_progname + pnlen - 8))
  300.       what_to_do = DO_OPTIMIZE;
  301.     else
  302.       what_to_do = DO_CHECK;
  303.   }
  304.   /* TODO: This variable is not yet used */
  305.   if (strcmp(default_charset, charset_info->csname) &&
  306.       !(charset_info= get_charset_by_csname(default_charset, 
  307.        MY_CS_PRIMARY, MYF(MY_WME))))
  308.       exit(1);
  309.   if (*argc > 0 && opt_alldbs)
  310.   {
  311.     printf("You should give only options, no arguments at all, with optionn");
  312.     printf("--all-databases. Please see %s --help for more information.n",
  313.    my_progname);
  314.     return 1;
  315.   }
  316.   if (*argc < 1 && !opt_alldbs)
  317.   {
  318.     printf("You forgot to give the arguments! Please see %s --helpn",
  319.    my_progname);
  320.     printf("for more information.n");
  321.     return 1;
  322.   }
  323.   if (tty_password)
  324.     opt_password = get_tty_password(NullS);
  325.   return(0);
  326. } /* get_options */
  327. static int process_all_databases()
  328. {
  329.   MYSQL_ROW row;
  330.   MYSQL_RES *tableres;
  331.   int result = 0;
  332.   if (mysql_query(sock, "SHOW DATABASES") ||
  333.       !(tableres = mysql_store_result(sock)))
  334.   {
  335.     my_printf_error(0, "Error: Couldn't execute 'SHOW DATABASES': %s",
  336.     MYF(0), mysql_error(sock));
  337.     return 1;
  338.   }
  339.   while ((row = mysql_fetch_row(tableres)))
  340.   {
  341.     if (process_all_tables_in_db(row[0]))
  342.       result = 1;
  343.   }
  344.   return result;
  345. }
  346. /* process_all_databases */
  347. static int process_databases(char **db_names)
  348. {
  349.   int result = 0;
  350.   for ( ; *db_names ; db_names++)
  351.   {
  352.     if (process_all_tables_in_db(*db_names))
  353.       result = 1;
  354.   }
  355.   return result;
  356. } /* process_databases */
  357. static int process_selected_tables(char *db, char **table_names, int tables)
  358. {
  359.   if (use_db(db))
  360.     return 1;
  361.   if (opt_all_in_1)
  362.   {
  363.     /* 
  364.       We need table list in form `a`, `b`, `c`
  365.       that's why we need 4 more chars added to to each table name
  366.       space is for more readable output in logs and in case of error
  367.     */   
  368.     char *table_names_comma_sep, *end;
  369.     int i, tot_length = 0;
  370.     for (i = 0; i < tables; i++)
  371.       tot_length += strlen(*(table_names + i)) + 4;
  372.     if (!(table_names_comma_sep = (char *)
  373.   my_malloc((sizeof(char) * tot_length) + 4, MYF(MY_WME))))
  374.       return 1;
  375.     for (end = table_names_comma_sep + 1; tables > 0;
  376.  tables--, table_names++)
  377.     {
  378.       end= fix_table_name(end, *table_names);
  379.       *end++= ',';
  380.     }
  381.     *--end = 0;
  382.     handle_request_for_tables(table_names_comma_sep + 1, tot_length - 1);
  383.     my_free(table_names_comma_sep, MYF(0));
  384.   }
  385.   else
  386.     for (; tables > 0; tables--, table_names++)
  387.       handle_request_for_tables(*table_names, strlen(*table_names));
  388.   return 0;
  389. } /* process_selected_tables */
  390. static char *fix_table_name(char *dest, char *src)
  391. {
  392.   char *db_sep;
  393.   *dest++= '`';
  394.   if ((db_sep= strchr(src, '.')))
  395.   {
  396.     dest= strmake(dest, src, (uint) (db_sep - src));
  397.     dest= strmov(dest, "`.`");
  398.     src= db_sep + 1;
  399.   }
  400.   dest= strxmov(dest, src, "`", NullS);
  401.   return dest;
  402. }
  403. static int process_all_tables_in_db(char *database)
  404. {
  405.   MYSQL_RES *res;
  406.   MYSQL_ROW row;
  407.   LINT_INIT(res);
  408.   if (use_db(database))
  409.     return 1;
  410.   if (mysql_query(sock, "SHOW TABLES") ||
  411. !((res= mysql_store_result(sock))))
  412.     return 1;
  413.   if (opt_all_in_1)
  414.   {
  415.     /*
  416.       We need table list in form `a`, `b`, `c`
  417.       that's why we need 4 more chars added to to each table name
  418.       space is for more readable output in logs and in case of error
  419.      */
  420.     char *tables, *end;
  421.     uint tot_length = 0;
  422.     while ((row = mysql_fetch_row(res)))
  423.       tot_length += strlen(row[0]) + 4;
  424.     mysql_data_seek(res, 0);
  425.     if (!(tables=(char *) my_malloc(sizeof(char)*tot_length+4, MYF(MY_WME))))
  426.     {
  427.       mysql_free_result(res);
  428.       return 1;
  429.     }
  430.     for (end = tables + 1; (row = mysql_fetch_row(res)) ;)
  431.     {
  432.       end= fix_table_name(end, row[0]);
  433.       *end++= ',';
  434.     }
  435.     *--end = 0;
  436.     if (tot_length)
  437.       handle_request_for_tables(tables + 1, tot_length - 1);
  438.     my_free(tables, MYF(0));
  439.   }
  440.   else
  441.   {
  442.     while ((row = mysql_fetch_row(res)))
  443.       handle_request_for_tables(row[0], strlen(row[0]));
  444.   }
  445.   mysql_free_result(res);
  446.   return 0;
  447. } /* process_all_tables_in_db */
  448. static int use_db(char *database)
  449. {
  450.   if (mysql_select_db(sock, database))
  451.   {
  452.     DBerror(sock, "when selecting the database");
  453.     return 1;
  454.   }
  455.   return 0;
  456. } /* use_db */
  457. static int handle_request_for_tables(char *tables, uint length)
  458. {
  459.   char *query, *end, options[100], message[100];
  460.   uint query_length= 0;
  461.   const char *op = 0;
  462.   options[0] = 0;
  463.   end = options;
  464.   switch (what_to_do) {
  465.   case DO_CHECK:
  466.     op = "CHECK";
  467.     if (opt_quick)              end = strmov(end, " QUICK");
  468.     if (opt_fast)               end = strmov(end, " FAST");
  469.     if (opt_medium_check)       end = strmov(end, " MEDIUM"); /* Default */
  470.     if (opt_extended)           end = strmov(end, " EXTENDED");
  471.     if (opt_check_only_changed) end = strmov(end, " CHANGED");
  472.     break;
  473.   case DO_REPAIR:
  474.     op = "REPAIR";
  475.     if (opt_quick)              end = strmov(end, " QUICK");
  476.     if (opt_extended)           end = strmov(end, " EXTENDED");
  477.     if (opt_frm)                end = strmov(end, " USE_FRM");
  478.     break;
  479.   case DO_ANALYZE:
  480.     op = "ANALYZE";
  481.     break;
  482.   case DO_OPTIMIZE:
  483.     op = "OPTIMIZE";
  484.     break;
  485.   }
  486.   if (!(query =(char *) my_malloc((sizeof(char)*(length+110)), MYF(MY_WME))))
  487.     return 1;
  488.   if (opt_all_in_1)
  489.   {
  490.     /* No backticks here as we added them before */
  491.     query_length= my_sprintf(query,
  492.      (query, "%s TABLE %s %s", op, tables, options));
  493.   }
  494.   else
  495.   {
  496.     char *ptr;
  497.     ptr= strmov(strmov(query, op), " TABLE ");
  498.     ptr= fix_table_name(ptr, tables);
  499.     ptr= strxmov(ptr, " ", options, NullS);
  500.     query_length= (uint) (ptr - query);
  501.   }
  502.   if (mysql_real_query(sock, query, query_length))
  503.   {
  504.     sprintf(message, "when executing '%s TABLE ... %s'", op, options);
  505.     DBerror(sock, message);
  506.     return 1;
  507.   }
  508.   print_result();
  509.   my_free(query, MYF(0));
  510.   return 0;
  511. }
  512. static void print_result()
  513. {
  514.   MYSQL_RES *res;
  515.   MYSQL_ROW row;
  516.   char prev[NAME_LEN*2+2];
  517.   uint i;
  518.   my_bool found_error=0;
  519.   res = mysql_use_result(sock);
  520.   prev[0] = '';
  521.   for (i = 0; (row = mysql_fetch_row(res)); i++)
  522.   {
  523.     int changed = strcmp(prev, row[0]);
  524.     my_bool status = !strcmp(row[2], "status");
  525.     if (status)
  526.     {
  527.       /*
  528.         if there was an error with the table, we have --auto-repair set,
  529.         and this isn't a repair op, then add the table to the tables4repair
  530.         list
  531.       */
  532.       if (found_error && opt_auto_repair && what_to_do != DO_REPAIR &&
  533.   strcmp(row[3],"OK"))
  534. insert_dynamic(&tables4repair, prev);
  535.       found_error=0;
  536.       if (opt_silent)
  537. continue;
  538.     }
  539.     if (status && changed)
  540.       printf("%-50s %s", row[0], row[3]);
  541.     else if (!status && changed)
  542.     {
  543.       printf("%sn%-9s: %s", row[0], row[2], row[3]);
  544.       if (strcmp(row[2],"note"))
  545. found_error=1;
  546.     }
  547.     else
  548.       printf("%-9s: %s", row[2], row[3]);
  549.     strmov(prev, row[0]);
  550.     putchar('n');
  551.   }
  552.   /* add the last table to be repaired to the list */
  553.   if (found_error && opt_auto_repair && what_to_do != DO_REPAIR)
  554.     insert_dynamic(&tables4repair, prev);
  555.   mysql_free_result(res);
  556. }
  557. static int dbConnect(char *host, char *user, char *passwd)
  558. {
  559.   DBUG_ENTER("dbConnect");
  560.   if (verbose)
  561.   {
  562.     fprintf(stderr, "# Connecting to %s...n", host ? host : "localhost");
  563.   }
  564.   mysql_init(&mysql_connection);
  565.   if (opt_compress)
  566.     mysql_options(&mysql_connection, MYSQL_OPT_COMPRESS, NullS);
  567. #ifdef HAVE_OPENSSL
  568.   if (opt_use_ssl)
  569.     mysql_ssl_set(&mysql_connection, opt_ssl_key, opt_ssl_cert, opt_ssl_ca,
  570.   opt_ssl_capath, opt_ssl_cipher);
  571. #endif
  572.   if (opt_protocol)
  573.     mysql_options(&mysql_connection,MYSQL_OPT_PROTOCOL,(char*)&opt_protocol);
  574. #ifdef HAVE_SMEM
  575.   if (shared_memory_base_name)
  576.     mysql_options(&mysql_connection,MYSQL_SHARED_MEMORY_BASE_NAME,shared_memory_base_name);
  577. #endif
  578.   if (!(sock = mysql_real_connect(&mysql_connection, host, user, passwd,
  579.          NULL, opt_mysql_port, opt_mysql_unix_port, 0)))
  580.   {
  581.     DBerror(&mysql_connection, "when trying to connect");
  582.     return 1;
  583.   }
  584.   return 0;
  585. } /* dbConnect */
  586. static void dbDisconnect(char *host)
  587. {
  588.   if (verbose)
  589.     fprintf(stderr, "# Disconnecting from %s...n", host ? host : "localhost");
  590.   mysql_close(sock);
  591. } /* dbDisconnect */
  592. static void DBerror(MYSQL *mysql, const char *when)
  593. {
  594.   DBUG_ENTER("DBerror");
  595.   my_printf_error(0,"Got error: %d: %s %s", MYF(0),
  596.   mysql_errno(mysql), mysql_error(mysql), when);
  597.   safe_exit(EX_MYSQLERR);
  598.   DBUG_VOID_RETURN;
  599. } /* DBerror */
  600. static void safe_exit(int error)
  601. {
  602.   if (!first_error)
  603.     first_error= error;
  604.   if (ignore_errors)
  605.     return;
  606.   if (sock)
  607.     mysql_close(sock);
  608.   exit(error);
  609. }
  610. int main(int argc, char **argv)
  611. {
  612.   MY_INIT(argv[0]);
  613.   /*
  614.   ** Check out the args
  615.   */
  616.   if (get_options(&argc, &argv))
  617.   {
  618.     my_end(0);
  619.     exit(EX_USAGE);
  620.   }
  621.   if (dbConnect(current_host, current_user, opt_password))
  622.     exit(EX_MYSQLERR);
  623.   if (opt_auto_repair &&
  624.       my_init_dynamic_array(&tables4repair, sizeof(char)*(NAME_LEN*2+2),16,64))
  625.   {
  626.     first_error = 1;
  627.     goto end;
  628.   }
  629.   if (opt_alldbs)
  630.     process_all_databases();
  631.   /* Only one database and selected table(s) */
  632.   else if (argc > 1 && !opt_databases)
  633.     process_selected_tables(*argv, (argv + 1), (argc - 1));
  634.   /* One or more databases, all tables */
  635.   else
  636.     process_databases(argv);
  637.   if (opt_auto_repair)
  638.   {
  639.     uint i;
  640.     if (!opt_silent && tables4repair.elements)
  641.       puts("nRepairing tables");
  642.     what_to_do = DO_REPAIR;
  643.     for (i = 0; i < tables4repair.elements ; i++)
  644.     {
  645.       char *name= (char*) dynamic_array_ptr(&tables4repair, i);
  646.       handle_request_for_tables(name, strlen(name));
  647.     }
  648.   }
  649.  end:
  650.   dbDisconnect(current_host);
  651.   if (opt_auto_repair)
  652.     delete_dynamic(&tables4repair);
  653.   my_free(opt_password, MYF(MY_ALLOW_ZERO_PTR));
  654. #ifdef HAVE_SMEM
  655.   my_free(shared_memory_base_name,MYF(MY_ALLOW_ZERO_PTR));
  656. #endif
  657.   my_end(0);
  658.   return(first_error!=0);
  659. } /* main */