MY_CONFIG_SET,

MY_CONFIG_REMOVE,

MY_CONFIG_PRINT,

MY_CONFIG_RESET,

MY_CONFIG_HELP

const int id;

const char *name;

const char *description;

my_option *options;

bool (*get_one_option_func)(int optid, const struct my_option *opt,

char *argument);

const struct my_option *opt

[[maybe_unused]],

char *argument [[maybe_unused]]) {

switch (optid) {

case '#':

DBUG_PUSH(argument ? argument : "d:t:o,/tmp/mysql_config_editor.trace");

break;

case 'V':

print_version();

exit(0);

break;

case '?':

usage_program();

exit(0);

break;

}

return false;

char *) {

switch (optid) {

case 'p':

tty_password = true;

break;

case 'G':

if (login_path_specified) {

/* Error, we do not support multiple login paths. */

my_perror(

"Error: Use of multiple login paths is not supported. "

"Exiting..");

return true;

}

login_path_specified = true;

break;

case '?':

usage_command(MY_CONFIG_SET);

exit(0);

break;

}

return false;

const struct my_option *, char *) {

switch (optid) {

case 'G':

if (login_path_specified) {

/* Error, we do not support multiple login paths. */

my_perror(

"Error: Use of multiple login paths is not supported. "

"Exiting..");

return true;

}

login_path_specified = true;

break;

case '?':

usage_command(MY_CONFIG_REMOVE);

exit(0);

break;

}

return false;

char *) {

switch (optid) {

case 'G':

if (login_path_specified) {

/* Error, we do not support multiple login paths. */

my_perror(

"Error: Use of multiple login paths is not supported. "

"Exiting..");

return true;

}

login_path_specified = true;

break;

case '?':

usage_command(MY_CONFIG_PRINT);

exit(0);

break;

}

return false;

char *) {

switch (optid) {

case '?':

usage_command(MY_CONFIG_RESET);

exit(0);

break;

}

return false;

MY_INIT(argv[0]);

DBUG_TRACE;

int command, rc = 0;

command = do_handle_options(argc, argv);

if (command > -1) rc = execute_commands(command);

if (rc != 0) {

my_perror("operation failed.");

return 1;

}

return 0;

DBUG_TRACE;

const char *command_list[MAX_COMMAND_LIMIT + 1];

char **saved_argv = argv;

char **argv_cmd;

char *ptr; /* for free. */

int argc_cmd;

int rc, i, command = -1;

if (argc < 2) {

usage_program();

exit(1);

}

if (!(ptr = (char *)my_malloc(PSI_NOT_INSTRUMENTED,

(argc + 2) * sizeof(char *), MYF(MY_WME))))

goto error;

/* Handle program options. */

/* Prepare a list of supported commands to be used by my_handle_options(). */

for (i = 0; (command_data[i].name != nullptr) && (i < MAX_COMMAND_LIMIT); i++)

command_list[i] = command_data[i].name;

command_list[i] = nullptr;

if ((rc = my_handle_options(&argc, &argv, my_program_long_options,

my_program_get_one_option, command_list, false)))

exit(rc);

if (argc == 0) /* No command specified. */

goto done;

for (i = 0; command_data[i].name != nullptr; i++) {

if (!strcmp(command_data[i].name, argv[0])) {

command = i;

break;

}

}

if (command == -1) goto error;

/* Handle command options. */

argv_cmd = (char **)ptr;

argc_cmd = argc + 1;

/* Prepare a command line (argv) using the rest of the options. */

argv_cmd[0] = saved_argv[0];

memcpy((uchar *)(argv_cmd + 1), (uchar *)(argv), (argc * sizeof(char *)));

argv_cmd[argc_cmd] = nullptr;

if ((rc = handle_options(&argc_cmd, &argv_cmd, command_data[command].options,

command_data[command].get_one_option_func)))

exit(rc);

/* Do not allow multiple commands. */

if (argc_cmd > 1) goto error;

done:

my_free(ptr);

return command;

error:

my_free(ptr);

usage_program();

exit(1);

DBUG_TRACE;

int rc = 0;

if ((rc = check_and_create_login_file())) goto done;

switch (command_data[command].id) {

case MY_CONFIG_SET:

verbose_msg("Executing set command.\n");

rc = set_command();

break;

case MY_CONFIG_REMOVE:

verbose_msg("Executing remove command.\n");

rc = remove_command();

break;

case MY_CONFIG_PRINT:

verbose_msg("Executing print command.\n");

rc = print_command();

break;

case MY_CONFIG_RESET:

verbose_msg("Resetting login file.\n");

rc = reset_login_file(true);

break;

case MY_CONFIG_HELP:

verbose_msg("Printing usage info.\n");

usage_program();

break;

default:

my_perror("unknown command.");

exit(1);

}

done:

my_close(g_fd, MYF(MY_WME));

return rc;

DBUG_TRACE;

DYNAMIC_STRING file_buf, path_buf;

init_dynamic_string(&path_buf, "", MY_LINE_MAX);

init_dynamic_string(&file_buf, "", file_size);

if (tty_password) opt_password = get_tty_password(NullS);

if (file_size) {

if (read_and_decrypt_file(&file_buf) == -1) goto error;

}

dynstr_append(&path_buf, "["); /* --login=path */

if (opt_login_path)

dynstr_append(&path_buf, opt_login_path);

else

dynstr_append(&path_buf, "client");

dynstr_append(&path_buf, "]");

if (opt_user) /* --user */

{

dynstr_append(&path_buf, "\nuser = ");

dynstr_append_quoted(&path_buf, "\"", 1, opt_user, NullS);

}

if (opt_password) /* --password */

{

dynstr_append(&path_buf, "\npassword = ");

dynstr_append_quoted(&path_buf, "\"", 1, opt_password, NullS);

}

if (opt_host) /* --host */

{

dynstr_append(&path_buf, "\nhost = ");

dynstr_append_quoted(&path_buf, "\"", 1, opt_host, NullS);

}

if (opt_socket) {

dynstr_append(&path_buf, "\nsocket = ");

dynstr_append_quoted(&path_buf, "\"", 1, opt_socket, NullS);

}

if (opt_port) {

dynstr_append(&path_buf, "\nport = ");

dynstr_append(&path_buf, opt_port);

}

dynstr_append(&path_buf, "\n");

/* Warn if login path already exists */

if (opt_warn && ((locate_login_path(&file_buf, opt_login_path)) != nullptr)) {

int choice;

printf(

"WARNING : \'%s\' path already exists and will be "

"overwritten. \n Continue? (Press y|Y for Yes, any "

"other key for No) : ",

opt_login_path);

choice = getchar();

if (choice != (int)'y' && choice != (int)'Y') goto done; /* skip */

}

/* Remove the login path. */

remove_login_path(&file_buf, opt_login_path);

/* Append the new login path to the file buffer. */

dynstr_append(&file_buf, path_buf.str);

if (encrypt_and_write_file(&file_buf) == -1) goto error;

done:

dynstr_free(&file_buf);

dynstr_free(&path_buf);

return 0;

error:

dynstr_free(&file_buf);

dynstr_free(&path_buf);

return -1;

DBUG_TRACE;

DYNAMIC_STRING file_buf, path_buf;

init_dynamic_string(&path_buf, "", MY_LINE_MAX);

init_dynamic_string(&file_buf, "", file_size);

if (file_size) {

if (read_and_decrypt_file(&file_buf) == -1) goto error;

} else

goto done; /* Nothing to remove, skip.. */

/* Warn if no login path is specified. */

if (opt_warn && ((locate_login_path(&file_buf, opt_login_path)) != nullptr) &&

(login_path_specified == false)) {

int choice;

printf(

"WARNING : No login path specified, so options from the default "

"login path will be removed.\nContinue? (Press y|Y for Yes, "

"any other key for No) : ");

choice = getchar();

if (choice != (int)'y' && choice != (int)'Y') goto done; /* skip */

}

remove_options(&file_buf, opt_login_path);

if (encrypt_and_write_file(&file_buf) == -1) goto error;

done:

dynstr_free(&file_buf);

dynstr_free(&path_buf);

return 0;

error:

dynstr_free(&file_buf);

dynstr_free(&path_buf);

return -1;

DBUG_TRACE;

DYNAMIC_STRING file_buf;

init_dynamic_string(&file_buf, "", file_size);

if (file_size) {

if (read_and_decrypt_file(&file_buf) == -1) goto error;

} else

goto done; /* Nothing to print, skip..*/

print_login_path(&file_buf, opt_login_path);

done:

dynstr_free(&file_buf);

return 0;

error:

dynstr_free(&file_buf);

return -1;

DBUG_TRACE;

MY_STAT stat_info;

#ifdef _WIN32

#define S_IRUSR 00400

#define S_IWUSR 00200

#define S_IRWXU 00700

#define S_IRWXG 00070

#define S_IRWXO 00007

#endif

const int access_flag = O_RDWR;

const ushort create_mode = (S_IRUSR | S_IWUSR);

/* Get the login file name. */

if (!my_default_get_login_file(my_login_file, sizeof(my_login_file))) {

my_perror("failed to set login file name");

goto error;

}

/*

#ifdef _WIN32

if (!(getenv("MYSQL_TEST_LOGIN_FILE"))) {

/* Check if 'MySQL' directory is in place. */

MY_STAT stat_info_dir;

char login_dir[FN_REFLEN];

size_t size;

dirname_part(login_dir, my_login_file, &size);

/* Remove the trailing '\' */

if (is_directory_separator(login_dir[--size])) login_dir[size] = 0;

/* Now check if directory exists? */

if (my_stat(login_dir, &stat_info_dir, MYF(0))) {

verbose_msg("%s directory exists.\n", login_dir);

} else {

/* Create the login directory. */

verbose_msg("%s directory doesn't exist, creating it.\n", login_dir);

if (my_mkdir(login_dir, 0, MYF(0))) {

my_perror("failed to create the directory");

goto error;

}

}

}

#endif

/* Check for login file's existence and permissions (0600). */

if (my_stat(my_login_file, &stat_info, MYF(0))) {

verbose_msg("File exists.\n");

file_size = (size_t)stat_info.st_size;

#ifdef _WIN32

if (1)

#else

if (!(stat_info.st_mode & (S_IXUSR | S_IRWXG | S_IRWXO)))

#endif

{

verbose_msg("File has the required permission.\nOpening the file.\n");

if ((g_fd = my_open(my_login_file, access_flag, MYF(MY_WME))) == -1) {

my_perror("couldn't open the file");

goto error;

}

} else {

verbose_msg("File does not have the required permission.\n");

printf(

"WARNING : Login file does not have the required"

" file permissions.\nPlease set the mode to 600 &"

" run the command again.\n");

goto error;

}

} else {

verbose_msg("File does not exist.\nCreating login file.\n");

if ((g_fd = my_create(my_login_file, create_mode, access_flag,

MYF(MY_WME))) == -1) {

my_perror("couldn't create the login file");

goto error;

} else {

verbose_msg("Login file created.\n");

my_close(g_fd, MYF(MY_WME));

verbose_msg("Opening the file.\n");

if ((g_fd = my_open(my_login_file, access_flag, MYF(MY_WME))) == -1) {

my_perror("couldn't open the file");

goto error;

}

file_size = 0;

}

}

if (file_size == 0) {

if (generate_login_key()) goto error;

if (add_header() == -1) goto error;

} else {

if (read_login_key() == -1) goto error;

}

return false;

error:

return true;

}

static void print_login_path(DYNAMIC_STRING *file_buf, const char *path_name) {

DBUG_TRACE;

char *start = nullptr, *end = nullptr, temp = '\0';

if (file_buf->length == 0) goto done; /* Nothing to print. */

if (opt_all) {

start = file_buf->str;

end = file_buf->str + file_buf->length;

} else {

start = locate_login_path(file_buf, path_name);

if (!start) /* login path not found, skip..*/

goto done;

end = strstr(start, "\n[");

}

if (end) {

temp = *end;

*end = '\0';

}

mask_password_and_print(start);

if (temp != '\0') *end = temp;

done:

return;

}

static void mask_password_and_print(char *buf) {

DBUG_TRACE;

const char *password_str = "\npassword = ", *mask = "*****";

char *next = nullptr;

while ((next = strstr(buf, password_str)) != nullptr) {

while (*buf != 0 && buf != next) putc(*(buf++), stdout);

printf("%s", password_str);

printf("%s\n", mask);

if (*buf == '\n') /* Move past \n' */

buf++;

/* Ignore the password. */

while (*buf && *(buf++) != '\n') {

}

if (!opt_all) break;

}

/* Now print the rest of the buffer. */

while (*buf) putc(*(buf++), stdout);

// And a new line.. if required.

if (*(buf - 1) != '\n') putc('\n', stdout);

}

static void remove_options(DYNAMIC_STRING *file_buf, const char *path_name) {

/* If nope of the options are specified remove the entire path. */

if (!opt_remove_host && !opt_remove_pass && !opt_remove_user &&

!opt_remove_socket && !opt_remove_port) {

remove_login_path(file_buf, path_name);

return;

}

if (opt_remove_user) remove_option(file_buf, path_name, "user");

if (opt_remove_pass) remove_option(file_buf, path_name, "password");

if (opt_remove_host) remove_option(file_buf, path_name, "host");

if (opt_remove_socket) remove_option(file_buf, path_name, "socket");

if (opt_remove_port) remove_option(file_buf, path_name, "port");

}

static void remove_option(DYNAMIC_STRING *file_buf, const char *path_name,

const char *option_name) {

DBUG_TRACE;

char *start = nullptr, *end = nullptr;

char *search_str;

size_t search_len, shift_len;

bool option_found = false;

search_str = (char *)my_malloc(PSI_NOT_INSTRUMENTED,

(uint)strlen(option_name) + 2, MYF(MY_WME));

sprintf(search_str, "\n%s", option_name);

if ((start = locate_login_path(file_buf, path_name)) == nullptr)

/* login path was not found, skip.. */

goto done;

end = strstr(start, "\n["); /* Next path. */

if (end)

search_len = end - start;

else

search_len = file_buf->length - (start - file_buf->str);

while (search_len > 1) {

if (!strncmp(start, search_str, strlen(search_str))) {

/* Option found. */

end = start;

while (*(++end) != '\n') {

}

option_found = true;

break;

} else {

/* Move to next line. */

while ((--search_len > 1) && (*(++start) != '\n')) {

}

}

}

if (option_found) {

shift_len = file_buf->length - (end - file_buf->str);

file_buf->length -= (end - start);

while (shift_len--) *(start++) = *(end++);

*start = '\0';

}

done:

my_free(search_str);

}

static void remove_login_path(DYNAMIC_STRING *file_buf, const char *path_name) {

DBUG_TRACE;

char *start = nullptr, *end = nullptr;

int to_move, len, diff;

if ((start = locate_login_path(file_buf, path_name)) == nullptr)

/* login path was not found, skip.. */

goto done;

end = strstr(start, "\n[");

if (end) {

end++; /* Move past '\n' */

len = ((diff = (start - end)) > 0) ? diff : -diff;

to_move = file_buf->length - (end - file_buf->str);

} else {

*start = '\0';

file_buf->length = ((diff = (file_buf->str - start)) > 0) ? diff : -diff;

goto done;

}

while (to_move--) *(start++) = *(end++);

*start = '\0';

file_buf->length -= len;

done:

return;

}

static int reset_login_file(bool gen_key) {

DBUG_TRACE;

if (my_chsize(g_fd, 0, 0, MYF(MY_WME))) {

verbose_msg("Error while truncating the file.\n");

goto error;

}

/* Seek to the beginning of the file. */

if (my_seek(g_fd, 0L, SEEK_SET, MYF(MY_WME) == MY_FILEPOS_ERROR))

goto error; /* Error. */

if (gen_key && generate_login_key()) goto error; /* Generate a new key. */

if (add_header() == -1) goto error;

return 0;

error:

return 0;

}

static char *locate_login_path(DYNAMIC_STRING *file_buf,