用于EagleEye3.0 规则集漏报和误报测试的示例项目,项目收集于github和gitee
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 
 
 

856 lines
28 KiB

/*
Copyright (c) 2000, 2019, Oracle and/or its affiliates. All rights reserved.
This program is free software; you can redistribute it and/or modify
it under the terms of the GNU General Public License, version 2.0,
as published by the Free Software Foundation.
This program is also distributed with certain software (including
but not limited to OpenSSL) that is licensed under separate terms,
as designated in a particular file or component or in included license
documentation. The authors of MySQL hereby grant you an additional
permission to link the program and your derivative works with the
separately licensed software that they have included with MySQL.
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, version 2.0, 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., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
*/
/* Show databases, tables or columns */
#include <mysql.h>
#include <mysqld_error.h>
#include <signal.h>
#include <stdarg.h>
#include <stdio.h>
#include <stdlib.h>
#include <sys/types.h>
#include "caching_sha2_passwordopt-vars.h"
#include "client/client_priv.h"
#include "compression.h"
#include "m_ctype.h"
#include "m_string.h"
#include "my_alloc.h"
#include "my_dbug.h"
#include "my_default.h"
#include "my_inttypes.h"
#include "my_macros.h"
#include "my_sys.h"
#include "mysql/service_mysql_alloc.h"
#include "print_version.h"
#include "sslopt-vars.h"
#include "typelib.h"
#include "welcome_copyright_notice.h" /* ORACLE_WELCOME_COPYRIGHT_NOTICE */
static char *host = 0, *opt_password = 0, *user = 0;
static bool opt_show_keys = 0, opt_compress = 0, opt_count = 0, opt_status = 0;
static bool tty_password = 0, opt_table_type = 0;
static bool debug_info_flag = 0, debug_check_flag = 0;
static uint my_end_arg = 0;
static uint opt_verbose = 0;
static const char *default_charset = MYSQL_AUTODETECT_CHARSET_NAME;
static char *opt_plugin_dir = 0, *opt_default_auth = 0;
static uint opt_enable_cleartext_plugin = 0;
static bool using_opt_enable_cleartext_plugin = 0;
static uint opt_zstd_compress_level = default_zstd_compression_level;
static char *opt_compress_algorithm = nullptr;
#if defined(_WIN32)
static char *shared_memory_base_name = 0;
#endif
static uint opt_protocol = 0;
static char *opt_bind_addr = NULL;
static void get_options(int *argc, char ***argv);
static uint opt_mysql_port = 0;
static int list_dbs(MYSQL *mysql, const char *wild);
static int list_tables(MYSQL *mysql, const char *db, const char *table);
static int list_table_status(MYSQL *mysql, const char *db, const char *table);
static int list_fields(MYSQL *mysql, const char *db, const char *table,
const char *field);
static void print_header(const char *header, size_t head_length, ...);
static void print_row(const char *header, size_t head_length, ...);
static void print_trailer(size_t length, ...);
static void print_res_header(MYSQL_RES *result);
static void print_res_top(MYSQL_RES *result);
static void print_res_row(MYSQL_RES *result, MYSQL_ROW cur);
static const char *load_default_groups[] = {"mysqlshow", "client", 0};
static char *opt_mysql_unix_port = 0;
int main(int argc, char **argv) {
int error;
bool first_argument_uses_wildcards = 0;
char *wild;
MYSQL mysql;
MY_INIT(argv[0]);
my_getopt_use_args_separator = true;
MEM_ROOT alloc{PSI_NOT_INSTRUMENTED, 512};
if (load_defaults("my", load_default_groups, &argc, &argv, &alloc)) exit(1);
my_getopt_use_args_separator = false;
get_options(&argc, &argv);
wild = 0;
if (argc) {
char *pos = argv[argc - 1], *to;
for (to = pos; *pos; pos++, to++) {
switch (*pos) {
case '*':
*pos = '%';
first_argument_uses_wildcards = 1;
break;
case '?':
*pos = '_';
first_argument_uses_wildcards = 1;
break;
case '%':
case '_':
first_argument_uses_wildcards = 1;
break;
case '\\':
pos++;
default:
break;
}
*to = *pos;
}
*to = *pos; /* just to copy a '\0' if '\\' was used */
}
if (first_argument_uses_wildcards)
wild = argv[--argc];
else if (argc == 3) /* We only want one field */
wild = argv[--argc];
if (argc > 2) {
fprintf(stderr, "%s: Too many arguments\n", my_progname);
exit(1);
}
mysql_init(&mysql);
if (opt_compress) mysql_options(&mysql, MYSQL_OPT_COMPRESS, NullS);
if (SSL_SET_OPTIONS(&mysql)) {
fprintf(stderr, "%s", SSL_SET_OPTIONS_ERROR);
exit(1);
}
if (opt_protocol)
mysql_options(&mysql, MYSQL_OPT_PROTOCOL, (char *)&opt_protocol);
if (opt_bind_addr) mysql_options(&mysql, MYSQL_OPT_BIND, opt_bind_addr);
#if defined(_WIN32)
if (shared_memory_base_name)
mysql_options(&mysql, MYSQL_SHARED_MEMORY_BASE_NAME,
shared_memory_base_name);
#endif
mysql_options(&mysql, MYSQL_SET_CHARSET_NAME, default_charset);
if (opt_compress_algorithm)
mysql_options(&mysql, MYSQL_OPT_COMPRESSION_ALGORITHMS,
opt_compress_algorithm);
mysql_options(&mysql, MYSQL_OPT_ZSTD_COMPRESSION_LEVEL,
&opt_zstd_compress_level);
if (opt_plugin_dir && *opt_plugin_dir)
mysql_options(&mysql, MYSQL_PLUGIN_DIR, opt_plugin_dir);
if (opt_default_auth && *opt_default_auth)
mysql_options(&mysql, MYSQL_DEFAULT_AUTH, opt_default_auth);
if (using_opt_enable_cleartext_plugin)
mysql_options(&mysql, MYSQL_ENABLE_CLEARTEXT_PLUGIN,
(char *)&opt_enable_cleartext_plugin);
mysql_options(&mysql, MYSQL_OPT_CONNECT_ATTR_RESET, 0);
mysql_options4(&mysql, MYSQL_OPT_CONNECT_ATTR_ADD, "program_name",
"mysqlshow");
set_server_public_key(&mysql);
set_get_server_public_key_option(&mysql);
if (!(mysql_real_connect(&mysql, host, user, opt_password,
(first_argument_uses_wildcards) ? "" : argv[0],
opt_mysql_port, opt_mysql_unix_port, 0))) {
fprintf(stderr, "%s: %s\n", my_progname, mysql_error(&mysql));
exit(1);
}
mysql.reconnect = 1;
switch (argc) {
case 0:
error = list_dbs(&mysql, wild);
break;
case 1:
if (opt_status)
error = list_table_status(&mysql, argv[0], wild);
else
error = list_tables(&mysql, argv[0], wild);
break;
default:
if (opt_status && !wild)
error = list_table_status(&mysql, argv[0], argv[1]);
else
error = list_fields(&mysql, argv[0], argv[1], wild);
break;
}
mysql_close(&mysql); /* Close & free connection */
my_free(opt_password);
#if defined(_WIN32)
my_free(shared_memory_base_name);
#endif
mysql_server_end();
my_end(my_end_arg);
exit(error ? 1 : 0);
}
static struct my_option my_long_options[] = {
{"bind-address", 0, "IP address to bind to.", (uchar **)&opt_bind_addr,
(uchar **)&opt_bind_addr, 0, GET_STR, REQUIRED_ARG, 0, 0, 0, 0, 0, 0},
{"character-sets-dir", 'c', "Directory for character set files.",
&charsets_dir, &charsets_dir, 0, GET_STR, REQUIRED_ARG, 0, 0, 0, 0, 0, 0},
{"default-character-set", OPT_DEFAULT_CHARSET,
"Set the default character set.", &default_charset, &default_charset, 0,
GET_STR, REQUIRED_ARG, 0, 0, 0, 0, 0, 0},
{"count", OPT_COUNT,
"Show number of rows per table (may be slow for non-MyISAM tables).",
&opt_count, &opt_count, 0, GET_BOOL, NO_ARG, 0, 0, 0, 0, 0, 0},
{"compress", 'C', "Use compression in server/client protocol.",
&opt_compress, &opt_compress, 0, GET_BOOL, NO_ARG, 0, 0, 0, 0, 0, 0},
{"debug", '#', "Output debug log. Often this is 'd:t:o,filename'.", 0, 0, 0,
GET_STR, OPT_ARG, 0, 0, 0, 0, 0, 0},
{"debug-check", OPT_DEBUG_CHECK,
"Check memory and open file usage at exit.", &debug_check_flag,
&debug_check_flag, 0, GET_BOOL, NO_ARG, 0, 0, 0, 0, 0, 0},
{"debug-info", OPT_DEBUG_INFO, "Print some debug info at exit.",
&debug_info_flag, &debug_info_flag, 0, GET_BOOL, NO_ARG, 0, 0, 0, 0, 0, 0},
{"default_auth", OPT_DEFAULT_AUTH,
"Default authentication client-side plugin to use.", &opt_default_auth,
&opt_default_auth, 0, GET_STR, REQUIRED_ARG, 0, 0, 0, 0, 0, 0},
{"enable_cleartext_plugin", OPT_ENABLE_CLEARTEXT_PLUGIN,
"Enable/disable the clear text authentication plugin.",
&opt_enable_cleartext_plugin, &opt_enable_cleartext_plugin, 0, GET_BOOL,
OPT_ARG, 0, 0, 0, 0, 0, 0},
{"help", '?', "Display this help and exit.", 0, 0, 0, GET_NO_ARG, NO_ARG, 0,
0, 0, 0, 0, 0},
{"host", 'h', "Connect to host.", &host, &host, 0, GET_STR, REQUIRED_ARG, 0,
0, 0, 0, 0, 0},
{"status", 'i', "Shows a lot of extra information about each table.",
&opt_status, &opt_status, 0, GET_BOOL, NO_ARG, 0, 0, 0, 0, 0, 0},
{"keys", 'k', "Show keys for table.", &opt_show_keys, &opt_show_keys, 0,
GET_BOOL, NO_ARG, 0, 0, 0, 0, 0, 0},
{"password", 'p',
"Password to use when connecting to server. If password is not given, "
"it's "
"solicited on the tty.",
0, 0, 0, GET_PASSWORD, OPT_ARG, 0, 0, 0, 0, 0, 0},
{"plugin_dir", OPT_PLUGIN_DIR, "Directory for client-side plugins.",
&opt_plugin_dir, &opt_plugin_dir, 0, GET_STR, REQUIRED_ARG, 0, 0, 0, 0, 0,
0},
{"port", 'P',
"Port number to use for connection or 0 for default to, in "
"order of preference, my.cnf, $MYSQL_TCP_PORT, "
#if MYSQL_PORT_DEFAULT == 0
"/etc/services, "
#endif
"built-in default (" STRINGIFY_ARG(MYSQL_PORT) ").",
&opt_mysql_port, &opt_mysql_port, 0, GET_UINT, REQUIRED_ARG, 0, 0, 0, 0, 0,
0},
#ifdef _WIN32
{"pipe", 'W', "Use named pipes to connect to server.", 0, 0, 0, GET_NO_ARG,
NO_ARG, 0, 0, 0, 0, 0, 0},
#endif
{"protocol", OPT_MYSQL_PROTOCOL,
"The protocol to use for connection (tcp, socket, pipe, memory).", 0, 0, 0,
GET_STR, REQUIRED_ARG, 0, 0, 0, 0, 0, 0},
#if defined(_WIN32)
{"shared-memory-base-name", OPT_SHARED_MEMORY_BASE_NAME,
"Base name of shared memory.", &shared_memory_base_name,
&shared_memory_base_name, 0, GET_STR_ALLOC, REQUIRED_ARG, 0, 0, 0, 0, 0,
0},
#endif
{"show-table-type", 't', "Show table type column.", &opt_table_type,
&opt_table_type, 0, GET_BOOL, NO_ARG, 0, 0, 0, 0, 0, 0},
{"socket", 'S', "The socket file to use for connection.",
&opt_mysql_unix_port, &opt_mysql_unix_port, 0, GET_STR, REQUIRED_ARG, 0, 0,
0, 0, 0, 0},
#include "caching_sha2_passwordopt-longopts.h"
#include "sslopt-longopts.h"
{"user", 'u', "User for login if not current user.", &user, &user, 0,
GET_STR, REQUIRED_ARG, 0, 0, 0, 0, 0, 0},
{"verbose", 'v',
"More verbose output; you can use this multiple times to get even more "
"verbose output.",
0, 0, 0, GET_NO_ARG, NO_ARG, 0, 0, 0, 0, 0, 0},
{"version", 'V', "Output version information and exit.", 0, 0, 0,
GET_NO_ARG, NO_ARG, 0, 0, 0, 0, 0, 0},
{"compression-algorithms", 0,
"Use compression algorithm in server/client protocol. Valid values "
"are any combination of 'zstd','zlib','uncompressed'.",
&opt_compress_algorithm, &opt_compress_algorithm, 0, GET_STR, REQUIRED_ARG,
0, 0, 0, 0, 0, 0},
{"zstd-compression-level", 0,
"Use this compression level in the client/server protocol, in case "
"--compression-algorithms=zstd. Valid range is between 1 and 22, "
"inclusive. Default is 3.",
&opt_zstd_compress_level, &opt_zstd_compress_level, 0, GET_UINT,
REQUIRED_ARG, 3, 1, 22, 0, 0, 0},
{0, 0, 0, 0, 0, 0, GET_NO_ARG, NO_ARG, 0, 0, 0, 0, 0, 0}};
static void usage(void) {
print_version();
puts(ORACLE_WELCOME_COPYRIGHT_NOTICE("2000"));
puts(
"Shows the structure of a MySQL database (databases, tables, and "
"columns).\n");
printf("Usage: %s [OPTIONS] [database [table [column]]]\n", my_progname);
puts(
"\n\
If last argument contains a shell or SQL wildcard (*,?,% or _) then only\n\
what\'s matched by the wildcard is shown.\n\
If no database is given then all matching databases are shown.\n\
If no table is given, then all matching tables in database are shown.\n\
If no column is given, then all matching columns and column types in table\n\
are shown.");
print_defaults("my", load_default_groups);
my_print_help(my_long_options);
my_print_variables(my_long_options);
}
extern "C" {
static bool get_one_option(int optid, const struct my_option *opt,
char *argument) {
switch (optid) {
case 'v':
opt_verbose++;
break;
case 'p':
if (argument == disabled_my_option) {
// Don't require password
static char empty_password[] = {'\0'};
DBUG_ASSERT(empty_password[0] ==
'\0'); // Check that it has not been overwritten
argument = empty_password;
}
if (argument) {
char *start = argument;
my_free(opt_password);
opt_password = my_strdup(PSI_NOT_INSTRUMENTED, argument, MYF(MY_FAE));
while (*argument) *argument++ = 'x'; /* Destroy argument */
if (*start) start[1] = 0; /* Cut length of argument */
tty_password = 0;
} else
tty_password = 1;
break;
case 'W':
#ifdef _WIN32
opt_protocol = MYSQL_PROTOCOL_PIPE;
#endif
break;
case (int)OPT_ENABLE_CLEARTEXT_PLUGIN:
using_opt_enable_cleartext_plugin = true;
break;
case OPT_MYSQL_PROTOCOL:
opt_protocol =
find_type_or_exit(argument, &sql_protocol_typelib, opt->name);
break;
case '#':
DBUG_PUSH(argument ? argument : "d:t:o");
debug_check_flag = 1;
break;
#include "sslopt-case.h"
case 'V':
print_version();
exit(0);
case '?':
case 'I': /* Info */
usage();
exit(0);
}
return 0;
}
} // extern "C"
static void get_options(int *argc, char ***argv) {
int ho_error;
if ((ho_error = handle_options(argc, argv, my_long_options, get_one_option)))
exit(ho_error);
if (tty_password) opt_password = get_tty_password(NullS);
if (opt_count) {
/*
We need to set verbose to 2 as we need to change the output to include
the number-of-rows column
*/
opt_verbose = 2;
}
if (debug_info_flag) my_end_arg = MY_CHECK_ERROR | MY_GIVE_INFO;
if (debug_check_flag) my_end_arg = MY_CHECK_ERROR;
return;
}
static int list_dbs(MYSQL *mysql, const char *wild) {
const char *header;
size_t length = 0;
uint counter = 0;
ulong rowcount = 0L;
char tables[NAME_LEN + 1], rows[NAME_LEN + 1];
char query[NAME_LEN + 100];
MYSQL_FIELD *field;
MYSQL_RES *result;
MYSQL_ROW row = NULL, rrow;
if (!(result = mysql_list_dbs(mysql, wild))) {
fprintf(stderr, "%s: Cannot list databases: %s\n", my_progname,
mysql_error(mysql));
return 1;
}
/*
If a wildcard was used, but there was only one row and it's name is an
exact match, we'll assume they really wanted to see the contents of that
database. This is because it is fairly common for database names to
contain the underscore (_), like INFORMATION_SCHEMA.
*/
if (wild && mysql_num_rows(result) == 1) {
row = mysql_fetch_row(result);
if (!my_strcasecmp(&my_charset_latin1, row[0], wild)) {
mysql_free_result(result);
if (opt_status)
return list_table_status(mysql, wild, NULL);
else
return list_tables(mysql, wild, NULL);
}
}
if (wild) printf("Wildcard: %s\n", wild);
header = "Databases";
length = strlen(header);
field = mysql_fetch_field(result);
if (length < field->max_length) length = field->max_length;
if (!opt_verbose)
print_header(header, length, NullS);
else if (opt_verbose == 1)
print_header(header, length, "Tables", 6, NullS);
else
print_header(header, length, "Tables", 6, "Total Rows", 12, NullS);
/* The first row may have already been read up above. */
while (row || (row = mysql_fetch_row(result))) {
counter++;
if (opt_verbose) {
if (!(mysql_select_db(mysql, row[0]))) {
MYSQL_RES *tresult = mysql_list_tables(mysql, (char *)NULL);
if (mysql_affected_rows(mysql) > 0) {
sprintf(tables, "%6lu", (ulong)mysql_affected_rows(mysql));
rowcount = 0;
if (opt_verbose > 1) {
/* Print the count of tables and rows for each database */
MYSQL_ROW trow;
while ((trow = mysql_fetch_row(tresult))) {
snprintf(query, sizeof(query), "SELECT COUNT(*) FROM `%s`",
trow[0]);
if (!(mysql_query(mysql, query))) {
MYSQL_RES *rresult;
if ((rresult = mysql_store_result(mysql))) {
rrow = mysql_fetch_row(rresult);
rowcount += (ulong)my_strtoull(rrow[0], (char **)0, 10);
mysql_free_result(rresult);
}
}
}
sprintf(rows, "%12lu", rowcount);
}
} else {
sprintf(tables, "%6d", 0);
sprintf(rows, "%12d", 0);
}
mysql_free_result(tresult);
} else {
my_stpcpy(tables, "N/A");
my_stpcpy(rows, "N/A");
}
}
if (!opt_verbose)
print_row(row[0], length, 0);
else if (opt_verbose == 1)
print_row(row[0], length, tables, 6, NullS);
else
print_row(row[0], length, tables, 6, rows, 12, NullS);
row = NULL;
}
print_trailer(length, (opt_verbose > 0 ? 6 : 0), (opt_verbose > 1 ? 12 : 0),
0);
if (counter && opt_verbose)
printf("%u row%s in set.\n", counter, (counter > 1) ? "s" : "");
mysql_free_result(result);
return 0;
}
static int list_tables(MYSQL *mysql, const char *db, const char *table) {
const char *header;
size_t head_length;
uint counter = 0;
char query[NAME_LEN + 100], rows[NAME_LEN], fields[16];
MYSQL_FIELD *field;
MYSQL_RES *result = nullptr;
MYSQL_ROW row, rrow;
if (mysql_select_db(mysql, db)) {
fprintf(stderr, "%s: Cannot connect to db %s: %s\n", my_progname, db,
mysql_error(mysql));
return 1;
}
if (table) {
/*
We just hijack the 'rows' variable for a bit to store the escaped
table name
*/
mysql_real_escape_string_quote(mysql, rows, table,
(unsigned long)strlen(table), '\'');
snprintf(query, sizeof(query), "show%s tables like '%s'",
opt_table_type ? " full" : "", rows);
} else
snprintf(query, sizeof(query), "show%s tables",
opt_table_type ? " full" : "");
if (mysql_query(mysql, query) || !(result = mysql_store_result(mysql))) {
fprintf(stderr, "%s: Cannot list tables in %s: %s\n", my_progname, db,
mysql_error(mysql));
exit(1);
}
printf("Database: %s", db);
if (table) printf(" Wildcard: %s", table);
putchar('\n');
header = "Tables";
head_length = strlen(header);
field = mysql_fetch_field(result);
if (head_length < field->max_length) head_length = field->max_length;
if (opt_table_type) {
if (!opt_verbose)
print_header(header, head_length, "table_type", 10, NullS);
else if (opt_verbose == 1)
print_header(header, head_length, "table_type", 10, "Columns", 8, NullS);
else {
print_header(header, head_length, "table_type", 10, "Columns", 8,
"Total Rows", 10, NullS);
}
} else {
if (!opt_verbose)
print_header(header, head_length, NullS);
else if (opt_verbose == 1)
print_header(header, head_length, "Columns", 8, NullS);
else
print_header(header, head_length, "Columns", 8, "Total Rows", 10, NullS);
}
while ((row = mysql_fetch_row(result))) {
counter++;
if (opt_verbose > 0) {
if (!(mysql_select_db(mysql, db))) {
MYSQL_RES *rresult = mysql_list_fields(mysql, row[0], NULL);
ulong rowcount = 0L;
if (!rresult) {
my_stpcpy(fields, "N/A");
my_stpcpy(rows, "N/A");
} else {
sprintf(fields, "%8u", (uint)mysql_num_fields(rresult));
mysql_free_result(rresult);
if (opt_verbose > 1) {
/* Print the count of rows for each table */
snprintf(query, sizeof(query), "SELECT COUNT(*) FROM `%s`", row[0]);
if (!(mysql_query(mysql, query))) {
if ((rresult = mysql_store_result(mysql))) {
rrow = mysql_fetch_row(rresult);
rowcount += (unsigned long)my_strtoull(rrow[0], (char **)0, 10);
mysql_free_result(rresult);
}
sprintf(rows, "%10lu", rowcount);
} else
sprintf(rows, "%10d", 0);
}
}
} else {
my_stpcpy(fields, "N/A");
my_stpcpy(rows, "N/A");
}
}
if (opt_table_type) {
if (!opt_verbose)
print_row(row[0], head_length, row[1], 10, NullS);
else if (opt_verbose == 1)
print_row(row[0], head_length, row[1], 10, fields, 8, NullS);
else
print_row(row[0], head_length, row[1], 10, fields, 8, rows, 10, NullS);
} else {
if (!opt_verbose)
print_row(row[0], head_length, NullS);
else if (opt_verbose == 1)
print_row(row[0], head_length, fields, 8, NullS);
else
print_row(row[0], head_length, fields, 8, rows, 10, NullS);
}
}
print_trailer(
head_length, (opt_table_type ? 10 : opt_verbose > 0 ? 8 : 0),
(opt_table_type ? (opt_verbose > 0 ? 8 : 0) : (opt_verbose > 1 ? 10 : 0)),
!opt_table_type ? 0 : opt_verbose > 1 ? 10 : 0, 0);
if (counter && opt_verbose)
printf("%u row%s in set.\n\n", counter, (counter > 1) ? "s" : "");
mysql_free_result(result);
return 0;
}
static int list_table_status(MYSQL *mysql, const char *db, const char *wild) {
char query[NAME_LEN + 100];
size_t len;
MYSQL_RES *result;
MYSQL_ROW row;
len = sizeof(query);
len -= snprintf(query, len, "show table status from `%s`", db);
if (wild && wild[0] && len)
strxnmov(query + strlen(query), len - 1, " like '", wild, "'", NullS);
if (mysql_query(mysql, query) || !(result = mysql_store_result(mysql))) {
fprintf(stderr, "%s: Cannot get status for db: %s, table: %s: %s\n",
my_progname, db, wild ? wild : "", mysql_error(mysql));
if (mysql_errno(mysql) == ER_PARSE_ERROR)
fprintf(stderr,
"This error probably means that your MySQL server doesn't "
"support the\n\'show table status' command.\n");
return 1;
}
printf("Database: %s", db);
if (wild) printf(" Wildcard: %s", wild);
putchar('\n');
print_res_header(result);
while ((row = mysql_fetch_row(result))) print_res_row(result, row);
print_res_top(result);
mysql_free_result(result);
return 0;
}
/*
list fields uses field interface as an example of how to parse
a MYSQL FIELD
*/
static int list_fields(MYSQL *mysql, const char *db, const char *table,
const char *wild) {
char query[NAME_LEN + 100];
size_t len;
MYSQL_RES *result;
MYSQL_ROW row;
ulong rows = 0;
if (mysql_select_db(mysql, db)) {
fprintf(stderr, "%s: Cannot connect to db: %s: %s\n", my_progname, db,
mysql_error(mysql));
return 1;
}
if (opt_count) {
snprintf(query, sizeof(query), "select count(*) from `%s`", table);
if (mysql_query(mysql, query) || !(result = mysql_store_result(mysql))) {
fprintf(stderr, "%s: Cannot get record count for db: %s, table: %s: %s\n",
my_progname, db, table, mysql_error(mysql));
return 1;
}
row = mysql_fetch_row(result);
rows = (ulong)my_strtoull(row[0], (char **)0, 10);
mysql_free_result(result);
}
len = sizeof(query);
len -= snprintf(query, len, "show /*!32332 FULL */ columns from `%s`", table);
if (wild && wild[0] && len)
strxnmov(query + strlen(query), len - 1, " like '", wild, "'", NullS);
if (mysql_query(mysql, query) || !(result = mysql_store_result(mysql))) {
fprintf(stderr, "%s: Cannot list columns in db: %s, table: %s: %s\n",
my_progname, db, table, mysql_error(mysql));
return 1;
}
printf("Database: %s Table: %s", db, table);
if (opt_count) printf(" Rows: %lu", rows);
if (wild && wild[0]) printf(" Wildcard: %s", wild);
putchar('\n');
print_res_header(result);
while ((row = mysql_fetch_row(result))) print_res_row(result, row);
print_res_top(result);
mysql_free_result(result);
if (opt_show_keys) {
snprintf(query, sizeof(query), "show keys from `%s`", table);
if (mysql_query(mysql, query) || !(result = mysql_store_result(mysql))) {
fprintf(stderr, "%s: Cannot list keys in db: %s, table: %s: %s\n",
my_progname, db, table, mysql_error(mysql));
return 1;
}
if (mysql_num_rows(result)) {
print_res_header(result);
while ((row = mysql_fetch_row(result))) print_res_row(result, row);
print_res_top(result);
} else
puts("Table has no keys");
mysql_free_result(result);
}
return 0;
}
/*****************************************************************************
General functions to print a nice ascii-table from data
*****************************************************************************/
static void print_header(const char *header, size_t head_length, ...) {
va_list args;
size_t length, i, str_length, pre_space;
const char *field;
va_start(args, head_length);
putchar('+');
field = header;
length = head_length;
for (;;) {
for (i = 0; i < length + 2; i++) putchar('-');
putchar('+');
if (!(field = va_arg(args, char *))) break;
length = va_arg(args, uint);
}
va_end(args);
putchar('\n');
va_start(args, head_length);
field = header;
length = head_length;
putchar('|');
for (;;) {
str_length = strlen(field);
if (str_length > length) str_length = length + 1;
pre_space = ((length - str_length) / 2) + 1;
for (i = 0; i < pre_space; i++) putchar(' ');
for (i = 0; i < str_length; i++) putchar(field[i]);
length = length + 2 - str_length - pre_space;
for (i = 0; i < length; i++) putchar(' ');
putchar('|');
if (!(field = va_arg(args, char *))) break;
length = va_arg(args, uint);
}
va_end(args);
putchar('\n');
va_start(args, head_length);
putchar('+');
field = header;
length = head_length;
for (;;) {
for (i = 0; i < length + 2; i++) putchar('-');
putchar('+');
if (!(field = va_arg(args, char *))) break;
length = va_arg(args, uint);
}
va_end(args);
putchar('\n');
}
static void print_row(const char *header, size_t head_length, ...) {
va_list args;
const char *field;
size_t i, length, field_length;
va_start(args, head_length);
field = header;
length = head_length;
for (;;) {
putchar('|');
putchar(' ');
fputs(field, stdout);
field_length = strlen(field);
for (i = field_length; i <= length; i++) putchar(' ');
if (!(field = va_arg(args, char *))) break;
length = va_arg(args, uint);
}
va_end(args);
putchar('|');
putchar('\n');
}
static void print_trailer(size_t head_length, ...) {
va_list args;
size_t length, i;
va_start(args, head_length);
length = head_length;
putchar('+');
for (;;) {
for (i = 0; i < length + 2; i++) putchar('-');
putchar('+');
if (!(length = va_arg(args, uint))) break;
}
va_end(args);
putchar('\n');
}
static void print_res_header(MYSQL_RES *result) {
MYSQL_FIELD *field;
print_res_top(result);
mysql_field_seek(result, 0);
putchar('|');
while ((field = mysql_fetch_field(result))) {
printf(" %-*s|", (int)field->max_length + 1, field->name);
}
putchar('\n');
print_res_top(result);
}
static void print_res_top(MYSQL_RES *result) {
uint i, length;
MYSQL_FIELD *field;
putchar('+');
mysql_field_seek(result, 0);
while ((field = mysql_fetch_field(result))) {
if ((length = strlen(field->name)) > field->max_length)
field->max_length = length;
else
length = field->max_length;
for (i = length + 2; i-- > 0;) putchar('-');
putchar('+');
}
putchar('\n');
}
static void print_res_row(MYSQL_RES *result, MYSQL_ROW cur) {
uint i, length;
MYSQL_FIELD *field;
putchar('|');
mysql_field_seek(result, 0);
for (i = 0; i < mysql_num_fields(result); i++) {
field = mysql_fetch_field(result);
length = field->max_length;
printf(" %-*s|", length + 1, cur[i] ? (char *)cur[i] : "");
}
putchar('\n');
}