home *** CD-ROM | disk | FTP | other *** search
/ Chip 2001 January / Chip_2001-01_cd1.bin / tema / mysql / mysql-3.23.28g-win-source.exe / client / mysqlshow.c < prev    next >
C/C++ Source or Header  |  2000-11-14  |  15KB  |  611 lines
  1. /* Copyright (C) 2000 MySQL AB & MySQL Finland AB & TCX DataKonsult AB
  2.    
  3.    This program is free software; you can redistribute it and/or modify
  4.    it under the terms of the GNU General Public License as published by
  5.    the Free Software Foundation; either version 2 of the License, or
  6.    (at your option) any later version.
  7.    
  8.    This program is distributed in the hope that it will be useful,
  9.    but WITHOUT ANY WARRANTY; without even the implied warranty of
  10.    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  11.    GNU General Public License for more details.
  12.    
  13.    You should have received a copy of the GNU General Public License
  14.    along with this program; if not, write to the Free Software
  15.    Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA */
  16.  
  17. /* Show databases, tables or columns */
  18.  
  19. #define SHOW_VERSION "8.2"
  20.  
  21. #include <global.h>
  22. #include <my_sys.h>
  23. #include <m_string.h>
  24. #include "mysql.h"
  25. #include "mysql_version.h"
  26. #include "mysqld_error.h"
  27. #include <signal.h>
  28. #include <stdarg.h>
  29. #include <getopt.h>
  30.  
  31. static my_string host=0,opt_password=0,user=0;
  32. static my_bool opt_show_keys=0,opt_compress=0,opt_status=0;
  33.  
  34. static void get_options(int *argc,char ***argv);
  35. static uint opt_mysql_port=0;
  36. static int list_dbs(MYSQL *mysql,const char *wild);
  37. static int list_tables(MYSQL *mysql,const char *db,const char *table);
  38. static int list_table_status(MYSQL *mysql,const char *db,const char *table);
  39. static int list_fields(MYSQL *mysql,const char *db,const char *table,
  40.                const char *field);
  41. static void print_header(const char *header,uint head_length,...);
  42. static void print_row(const char *header,uint head_length,...);
  43. static void print_trailer(uint length,...);
  44. static void print_res_header(MYSQL_RES *result);
  45. static void print_res_top(MYSQL_RES *result);
  46. static void print_res_row(MYSQL_RES *result,MYSQL_ROW cur);
  47.  
  48. static const char *load_default_groups[]= { "mysqlshow","client",0 };
  49. static my_string opt_mysql_unix_port=0;
  50. #include "sslopt-vars.h"
  51.  
  52. int main(int argc, char **argv)
  53. {
  54.   int error;
  55.   char *wild;
  56.   MYSQL mysql;
  57.   MY_INIT(argv[0]);
  58.   load_defaults("my",load_default_groups,&argc,&argv);
  59.   get_options(&argc,&argv);
  60.  
  61.   wild=0;
  62.   if (argc && strcont(argv[argc-1],"*?%_"))
  63.   {
  64.     char *pos;
  65.  
  66.     wild=argv[--argc];
  67.     for (pos=wild ; *pos ; pos++)
  68.     {                    /* Unix wildcards to sql  */
  69.       if (*pos == '*')
  70.     *pos='%';
  71.       else if (*pos == '?')
  72.     *pos='_';
  73.     }
  74.   }
  75.   else if (argc == 3)            /* We only want one field */
  76.     wild=argv[--argc];
  77.  
  78.   if (argc > 2)
  79.   {
  80.     fprintf(stderr,"%s: Too many arguments\n",my_progname);
  81.     exit(1);
  82.   }
  83.   mysql_init(&mysql);
  84.   if (opt_compress)
  85.     mysql_options(&mysql,MYSQL_OPT_COMPRESS,NullS);
  86. #ifdef HAVE_OPENSSL
  87.   if (opt_use_ssl)
  88.     mysql_ssl_set(&mysql, opt_ssl_key, opt_ssl_cert, opt_ssl_ca,
  89.           opt_ssl_capath);
  90. #endif
  91.   if (!(mysql_real_connect(&mysql,host,user,opt_password,
  92.                argv[0],opt_mysql_port,opt_mysql_unix_port,
  93.                0)))
  94.   {
  95.     fprintf(stderr,"%s: %s\n",my_progname,mysql_error(&mysql));
  96.     exit(1);
  97.   }
  98.  
  99.   switch (argc)
  100.   {
  101.   case 0:  error=list_dbs(&mysql,wild); break;
  102.   case 1:
  103.     if (opt_status)
  104.       error=list_table_status(&mysql,argv[0],wild);
  105.     else
  106.       error=list_tables(&mysql,argv[0],wild);
  107.     break;
  108.   default:
  109.     if (opt_status && ! wild)
  110.       error=list_table_status(&mysql,argv[0],argv[1]);
  111.     else
  112.       error=list_fields(&mysql,argv[0],argv[1],wild);
  113.     break;
  114.   }
  115.   mysql_close(&mysql);            /* Close & free connection */
  116.   if (opt_password)
  117.     my_free(opt_password,MYF(0));
  118.   my_end(0);
  119.   exit(error ? 1 : 0);
  120.   return 0;                /* No compiler warnings */
  121. }
  122.  
  123.  
  124. static struct option long_options[] =
  125. {
  126.   {"character-sets-dir", required_argument, 0, 'c'},
  127.   {"compress",    no_argument,       0, 'C'},
  128.   {"debug",    optional_argument, 0, '#'},
  129.   {"help",    no_argument,       0, '?'},
  130.   {"host",    required_argument, 0, 'h'},
  131.   {"status",    no_argument,       0, 'i'},
  132.   {"keys",    no_argument,       0, 'k'},
  133.   {"password",    optional_argument, 0, 'p'},
  134.   {"port",    required_argument, 0, 'P'},
  135. #ifdef __WIN__
  136.   {"pipe",    no_argument,       0, 'W'},
  137. #endif
  138.   {"socket",    required_argument, 0, 'S'},
  139. #include "sslopt-longopts.h"
  140. #ifndef DONT_ALLOW_USER_CHANGE
  141.   {"user",    required_argument, 0, 'u'},
  142. #endif
  143.   {"version",    no_argument,       0, 'V'},
  144.   {0, 0, 0, 0}
  145. };
  146.  
  147.  
  148. static void print_version(void)
  149. {
  150.   printf("%s  Ver %s Distrib %s, for %s (%s)\n",my_progname,SHOW_VERSION,
  151.      MYSQL_SERVER_VERSION,SYSTEM_TYPE,MACHINE_TYPE);
  152. }
  153.  
  154. static void usage(void)
  155. {
  156.   print_version();
  157.   puts("Copyright (C) 2000 MySQL AB & MySQL Finland AB & TCX DataKonsult AB");
  158.   puts("This software comes with ABSOLUTELY NO WARRANTY. This is free software,\nand you are welcome to modify and redistribute it under the GPL license\n");
  159.   puts("Shows the structure of a mysql database (databases,tables and columns)\n");
  160.   printf("Usage: %s [OPTIONS] [database [table [column]]]\n",my_progname);
  161.   printf("\n\
  162.   -#, --debug=...       output debug log. Often this is 'd:t:o,filename`\n\
  163.   -?, --help        display this help and exit\n\
  164.   -c, --character-sets-dir=...\n\
  165.                         Directory where character sets are\n\
  166.   -C, --compress        Use compression in server/client protocol\n\
  167.   -h, --host=...    connect to host\n\
  168.   -i, --status        Shows a lot of extra information about each table\n\
  169.   -k, --keys        show keys for table\n\
  170.   -p, --password[=...]    password to use when connecting to server\n\
  171.             If password is not given it's asked from the tty.\n");
  172. #ifdef __WIN__
  173.   puts("-W, --pipe        Use named pipes to connect to server");
  174. #endif
  175.   printf("\
  176.   -P  --port=...    Port number to use for connection\n\
  177.   -S  --socket=...    Socket file to use for connection\n");
  178. #include "sslopt-usage.h"
  179. #ifndef DONT_ALLOW_USER_CHANGE
  180.   printf("\
  181.   -u, --user=#        user for login if not current user\n");
  182. #endif
  183.   printf("\
  184.   -V, --version        output version information and exit\n");
  185.  
  186.   puts("\n\
  187. If last argument contains a shell or SQL wildcard (*,?,% or _) then only\n\
  188. what\'s matched by the wildcard is shown.\n\
  189. If no database is given then all matching databases are shown.\n\
  190. If no table is given then all matching tables in database are shown\n\
  191. If no column is given then all matching columns and columntypes in table\n\
  192. are shown");
  193.   print_defaults("my",load_default_groups);
  194. }
  195.  
  196.  
  197. static void
  198. get_options(int *argc,char ***argv)
  199. {
  200.   int c,option_index;
  201.   my_bool tty_password=0;
  202.  
  203.   while ((c=getopt_long(*argc,*argv,"c:h:p::u:#::P:S:Ck?VWi",long_options,
  204.             &option_index)) != EOF)
  205.   {
  206.     switch(c) {
  207.     case 'C':
  208.       opt_compress=1;
  209.       break;
  210.     case 'c':
  211.       charsets_dir= optarg;
  212.       break;
  213.     case 'h':
  214.       host = optarg;
  215.       break;
  216.     case 'i':
  217.       opt_status=1;
  218.       break;
  219.     case 'k':
  220.       opt_show_keys=1;
  221.       break;
  222.     case 'p':
  223.       if (optarg)
  224.       {
  225.     char *start=optarg;
  226.     my_free(opt_password,MYF(MY_ALLOW_ZERO_PTR));
  227.     opt_password=my_strdup(optarg,MYF(MY_FAE));
  228.     while (*optarg) *optarg++= 'x';        /* Destroy argument */
  229.     if (*start)
  230.       start[1]=0;                /* Cut length of argument */
  231.       }
  232.       else
  233.     tty_password=1;
  234.       break;
  235. #ifndef DONT_ALLOW_USER_CHANGE
  236.     case 'u':
  237.       user=optarg;
  238.       break;
  239. #endif
  240.     case 'P':
  241.       opt_mysql_port= (unsigned int) atoi(optarg);
  242.       break;
  243.     case 'S':
  244.       opt_mysql_unix_port= optarg;
  245.       break;
  246.     case 'W':
  247. #ifdef __WIN__
  248.       opt_mysql_unix_port=MYSQL_NAMEDPIPE;
  249. #endif
  250.       break;
  251. #include "sslopt-case.h"
  252.     case '#':
  253.       DBUG_PUSH(optarg ? optarg : "d:t:o");
  254.       break;
  255.     case 'V':
  256.       print_version();
  257.       exit(0);
  258.       break;
  259.     default:
  260.       fprintf(stderr,"Illegal option character '%c'\n",opterr);
  261.       /* Fall throught */
  262.     case '?':
  263.     case 'I':                    /* Info */
  264.       usage();
  265.       exit(0);
  266.     }
  267.   }
  268.   (*argc)-=optind;
  269.   (*argv)+=optind;
  270.   if (tty_password)
  271.     opt_password=get_tty_password(NullS);
  272.   return;
  273. }
  274.  
  275.  
  276. static int
  277. list_dbs(MYSQL *mysql,const char *wild)
  278. {
  279.   const char *header;
  280.   uint length;
  281.   MYSQL_FIELD *field;
  282.   MYSQL_RES *result;
  283.   MYSQL_ROW row;
  284.  
  285.   if (!(result=mysql_list_dbs(mysql,wild)))
  286.   {
  287.     fprintf(stderr,"%s: Cannot list databases: %s\n",my_progname,
  288.         mysql_error(mysql));
  289.     return 1;
  290.   }
  291.   if (wild)
  292.     printf("Wildcard: %s\n",wild);
  293.  
  294.   header="Databases";
  295.   length=(uint) strlen(header);
  296.   field=mysql_fetch_field(result);
  297.   if (length < field->max_length)
  298.     length=field->max_length;
  299.  
  300.   print_header(header,length,NullS);
  301.   while ((row = mysql_fetch_row(result)))
  302.     print_row(row[0],length,0);
  303.   print_trailer(length,0);
  304.   mysql_free_result(result);
  305.   return 0;
  306. }
  307.  
  308.  
  309. static int
  310. list_tables(MYSQL *mysql,const char *db,const char *table)
  311. {
  312.   const char *header;
  313.   uint head_length;
  314.   MYSQL_FIELD *field;
  315.   MYSQL_RES *result;
  316.   MYSQL_ROW row;
  317.  
  318.   if (mysql_select_db(mysql,db))
  319.   {
  320.     fprintf(stderr,"%s: Cannot connect to db %s: %s\n",my_progname,db,
  321.         mysql_error(mysql));
  322.     return 1;
  323.   }
  324.   if (!(result=mysql_list_tables(mysql,table)))
  325.   {
  326.     fprintf(stderr,"%s: Cannot list tables in %s: %s\n",my_progname,db,
  327.         mysql_error(mysql));
  328.     exit(1);
  329.   }
  330.   printf("Database: %s",db);
  331.   if (table)
  332.     printf("  Wildcard: %s",table);
  333.   putchar('\n');
  334.  
  335.   header="Tables";
  336.   head_length=(uint) strlen(header);
  337.   field=mysql_fetch_field(result);
  338.   if (head_length < field->max_length)
  339.     head_length=field->max_length;
  340.  
  341.   print_header(header,head_length,NullS);
  342.   while ((row = mysql_fetch_row(result)))
  343.     print_row(row[0],head_length,0);
  344.   print_trailer(head_length,0);
  345.   mysql_free_result(result);
  346.   return 0;
  347. }
  348.  
  349. static int
  350. list_table_status(MYSQL *mysql,const char *db,const char *wild)
  351. {
  352.   char query[1024],*end;
  353.   MYSQL_RES *result;
  354.   MYSQL_ROW row;
  355.  
  356.   end=strxmov(query,"show table status from ",db,NullS);
  357.   if (wild && wild[0])
  358.     strxmov(end," like '",wild,"'",NullS);
  359.   if (mysql_query(mysql,query) || !(result=mysql_store_result(mysql)))
  360.   {
  361.     fprintf(stderr,"%s: Cannot get status for db: %s, table: %s: %s\n",
  362.         my_progname,db,wild ? wild : "",mysql_error(mysql));
  363.     if (mysql_errno(mysql) == ER_PARSE_ERROR)
  364.       fprintf(stderr,"This error probably means that your MySQL server doesn't support the\n\'show table status' command.\n");
  365.     return 1;
  366.   }
  367.  
  368.   printf("Database: %s",db);
  369.   if (wild)
  370.     printf("  Wildcard: %s",wild);
  371.   putchar('\n');
  372.  
  373.   print_res_header(result);
  374.   while ((row=mysql_fetch_row(result)))
  375.     print_res_row(result,row);
  376.   print_res_top(result);
  377.   mysql_free_result(result);
  378.   return 0;
  379. }
  380.  
  381. /*
  382. ** list fields uses field interface as an example of how to parse
  383. ** a MYSQL FIELD
  384. */
  385.  
  386. static int
  387. list_fields(MYSQL *mysql,const char *db,const char *table,
  388.         const char *wild)
  389. {
  390.   char query[1024],*end;
  391.   MYSQL_RES *result;
  392.   MYSQL_ROW row;
  393.  
  394.   if (mysql_select_db(mysql,db))
  395.   {
  396.     fprintf(stderr,"%s: Cannot connect to db: %s: %s\n",my_progname,db,
  397.         mysql_error(mysql));
  398.     return 1;
  399.   }
  400.   end=strmov(strmov(query,"show columns from "),table);
  401.   if (wild && wild[0])
  402.     strxmov(end," like '",wild,"'",NullS);
  403.   if (mysql_query(mysql,query) || !(result=mysql_store_result(mysql)))
  404.   {
  405.     fprintf(stderr,"%s: Cannot list columns in db: %s, table: %s: %s\n",
  406.         my_progname,db,table,mysql_error(mysql));
  407.     return 1;
  408.   }
  409.  
  410.   printf("Database: %s  Table: %s  Rows: %lu", db,table,
  411.      (ulong) mysql->extra_info);
  412.   if (wild && wild[0])
  413.     printf("  Wildcard: %s",wild);
  414.   putchar('\n');
  415.  
  416.   print_res_header(result);
  417.   while ((row=mysql_fetch_row(result)))
  418.     print_res_row(result,row);
  419.   print_res_top(result);
  420.   if (opt_show_keys)
  421.   {
  422.     end=strmov(strmov(query,"show keys from "),table);
  423.     if (mysql_query(mysql,query) || !(result=mysql_store_result(mysql)))
  424.     {
  425.       fprintf(stderr,"%s: Cannot list keys in db: %s, table: %s: %s\n",
  426.           my_progname,db,table,mysql_error(mysql));
  427.       return 1;
  428.     }
  429.     if (mysql_num_rows(result))
  430.     {
  431.       print_res_header(result);
  432.       while ((row=mysql_fetch_row(result)))
  433.     print_res_row(result,row);
  434.       print_res_top(result);
  435.     }
  436.     else
  437.       puts("Table has no keys");
  438.   }
  439.   mysql_free_result(result);
  440.   return 0;
  441. }
  442.  
  443.  
  444. /*****************************************************************************
  445. ** General functions to print a nice ascii-table from data
  446. *****************************************************************************/
  447.  
  448. static void
  449. print_header(const char *header,uint head_length,...)
  450. {
  451.   va_list args;
  452.   uint length,i,str_length,pre_space;
  453.   const char *field;
  454.  
  455.   va_start(args,head_length);
  456.   putchar('+');
  457.   field=header; length=head_length;
  458.   for (;;)
  459.   {
  460.     for (i=0 ; i < length+2 ; i++)
  461.       putchar('-');
  462.     putchar('+');
  463.     if (!(field=va_arg(args,my_string)))
  464.       break;
  465.     length=va_arg(args,uint);
  466.   }
  467.   va_end(args);
  468.   putchar('\n');
  469.  
  470.   va_start(args,head_length);
  471.   field=header; length=head_length;
  472.   putchar('|');
  473.   for (;;)
  474.   {
  475.     str_length=(uint) strlen(field);
  476.     if (str_length > length)
  477.       str_length=length+1;
  478.     pre_space=(uint) (((int) length-(int) str_length)/2)+1;
  479.     for (i=0 ; i < pre_space ; i++)
  480.       putchar(' ');
  481.     for (i = 0 ; i < str_length ; i++)
  482.       putchar(field[i]);
  483.     length=length+2-str_length-pre_space;
  484.     for (i=0 ; i < length ; i++)
  485.       putchar(' ');
  486.     putchar('|');
  487.     if (!(field=va_arg(args,my_string)))
  488.       break;
  489.     length=va_arg(args,uint);
  490.   }
  491.   va_end(args);
  492.   putchar('\n');
  493.  
  494.   va_start(args,head_length);
  495.   putchar('+');
  496.   field=header; length=head_length;
  497.   for (;;)
  498.   {
  499.     for (i=0 ; i < length+2 ; i++)
  500.       putchar('-');
  501.     putchar('+');
  502.     if (!(field=va_arg(args,my_string)))
  503.       break;
  504.     length=va_arg(args,uint);
  505.   }
  506.   va_end(args);
  507.   putchar('\n');
  508. }
  509.  
  510.  
  511. static void
  512. print_row(const char *header,uint head_length,...)
  513. {
  514.   va_list args;
  515.   const char *field;
  516.   uint i,length,field_length;
  517.  
  518.   va_start(args,head_length);
  519.   field=header; length=head_length;
  520.   for (;;)
  521.   {
  522.     putchar('|');
  523.     putchar(' ');
  524.     fputs(field,stdout);
  525.     field_length=(uint) strlen(field);
  526.     for (i=field_length ; i <= length ; i++)
  527.       putchar(' ');
  528.     if (!(field=va_arg(args,my_string)))
  529.       break;
  530.     length=va_arg(args,uint);
  531.   }
  532.   va_end(args);
  533.   putchar('|');
  534.   putchar('\n');
  535. }
  536.  
  537.  
  538. static void
  539. print_trailer(uint head_length,...)
  540. {
  541.   va_list args;
  542.   uint length,i;
  543.  
  544.   va_start(args,head_length);
  545.   length=head_length;
  546.   putchar('+');
  547.   for (;;)
  548.   {
  549.     for (i=0 ; i < length+2 ; i++)
  550.       putchar('-');
  551.     putchar('+');
  552.     if (!(length=va_arg(args,uint)))
  553.       break;
  554.   }
  555.   va_end(args);
  556.   putchar('\n');
  557. }
  558.  
  559.  
  560. static void print_res_header(MYSQL_RES *result)
  561. {
  562.   MYSQL_FIELD *field;
  563.  
  564.   print_res_top(result);
  565.   mysql_field_seek(result,0);
  566.   putchar('|');
  567.   while ((field = mysql_fetch_field(result)))
  568.   {
  569.     printf(" %-*s|",field->max_length+1,field->name);
  570.   }
  571.   putchar('\n');
  572.   print_res_top(result);
  573. }
  574.  
  575.  
  576. static void print_res_top(MYSQL_RES *result)
  577. {
  578.   uint i,length;
  579.   MYSQL_FIELD *field;
  580.  
  581.   putchar('+');
  582.   mysql_field_seek(result,0);
  583.   while((field = mysql_fetch_field(result)))
  584.   {
  585.     if ((length=(uint) strlen(field->name)) > field->max_length)
  586.       field->max_length=length;
  587.     else
  588.       length=field->max_length;
  589.     for (i=length+2 ; i--> 0 ; )
  590.       putchar('-');
  591.     putchar('+');
  592.   }
  593.   putchar('\n');
  594. }
  595.  
  596.  
  597. static void print_res_row(MYSQL_RES *result,MYSQL_ROW cur)
  598. {
  599.   uint i,length;
  600.   MYSQL_FIELD *field;
  601.   putchar('|');
  602.   mysql_field_seek(result,0);
  603.   for (i=0 ; i < mysql_num_fields(result); i++)
  604.   {
  605.     field = mysql_fetch_field(result);
  606.     length=field->max_length;
  607.     printf(" %-*s|",length+1,cur[i] ? (char*) cur[i] : "");
  608.   }
  609.   putchar('\n');
  610. }
  611.