From f14b09b10db42962a4f7f0a4d7f9cf2ea8feef4f Mon Sep 17 00:00:00 2001 From: Joursoir Date: Tue, 23 Mar 2021 16:24:50 +0000 Subject: move cmd to 'exec-cmd.c', fix some memory leak, macros changes --- src/exec-cmd.c | 495 +++++++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 495 insertions(+) create mode 100644 src/exec-cmd.c (limited to 'src/exec-cmd.c') diff --git a/src/exec-cmd.c b/src/exec-cmd.c new file mode 100644 index 0000000..6aced93 --- /dev/null +++ b/src/exec-cmd.c @@ -0,0 +1,495 @@ +/*** + This file is part of LockPassword + Copyright (C) 2020-2021 Aleksandr D. Goncharov (Joursoir) + + This program is free software: you can redistribute it and/or modify + it under the terms of the GNU General Public License as published by + the Free Software Foundation, either version 3 of the License, or + (at your option) any later version. + + 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 for more details. + + You should have received a copy of the GNU General Public License + along with this program. If not, see . +***/ + +#include +#include +#include +#include +#include +#include +#include +#include +#include + +#include "exec-cmd.h" +#include "constants.h" +#include "easydir.h" +#include "implementation.h" +#include "exec-cmd.h" +#include "tree.h" + +int cmd_init(int argc, char *argv[]) +{ + const char description[] = "init gpg-key\n"; + int retval = 0, result; + char *gpg_key = argv[2]; + if(gpg_key == NULL) + usageprint("%s", description); + + // create .gpg-key in storage + FILE *filekey = fopen(GPGKEY_FILE, "w"); + if(!filekey) + errprint_r(1, "%s\n", strerror(errno)); + + result = fputs(gpg_key, filekey); + if(result == EOF) + errprint_ptr(&retval, 1, "%s\n", strerror(errno)); + else + printf("LockPassword initialized for %s\n", gpg_key); + + fclose(filekey); + return retval; +} + +int cmd_insert(int argc, char *argv[]) +{ + const char description[] = "insert [-ecf] passname\n"; + int retval = 0, result; + int flag_echo = 0, flag_force = 0, flag_copy = 0; + const struct option long_options[] = { + {"echo", no_argument, NULL, 'e'}, + {"force", no_argument, NULL, 'f'}, + {"copy", no_argument, NULL, 'c'}, + {NULL, 0, NULL, 0} + }; + + while((result = getopt_long(argc, argv, "efc", long_options, NULL)) != -1) { + switch(result) { + case 'e': { flag_echo = 1; break; } + case 'f': { flag_force = 1; break; } + case 'c': { flag_copy = 1; break; } + default: usageprint("%s", description); + } + } + + if(optind < argc) optind++; // for skip "insert" + dbgprint("passname: %s\n", argv[optind]); + + char *path = argv[optind]; + if(path == NULL) + usageprint("%s", description); + + result = check_sneaky_paths(path); + if(result) + errprint_r(1, "You have used forbidden paths\n"); + + if(file_exist(path) == F_ISFILE) { + if(!flag_force) { + if(overwrite_answer(path) != 'y') + return 1; + } + } + + char *f_pass, *s_pass; + do { // START DO + + if(!flag_echo) { + visible_enter(0); + + printf("Type your password: "); + f_pass = get_input(minlen_pass, maxlen_pass); + printf("\n"); + if(f_pass == NULL) { + errprint_ptr(&retval, 1, "Incorrect password\n"); + break; + } + + printf("Type your password again: "); + s_pass = get_input(minlen_pass, maxlen_pass); + printf("\n"); + if(s_pass == NULL) { + errprint_ptr(&retval, 1, "Incorrect password\n"); + break; + } + + if(strcmp(f_pass, s_pass) != 0) { + errprint_ptr(&retval, 1, "Password do not match\n"); + break; + } + } + else { + printf("Type your password: "); + f_pass = get_input(minlen_pass, maxlen_pass); + if(f_pass == NULL) { + errprint_ptr(&retval, 1, "Incorrect password\n"); + break; + } + } + + result = insert_pass(path, f_pass); + if(result) { + errprint_ptr(&retval, 1, "Can't add password to LockPassword\n"); + break; + } + if(flag_copy) + copy_outside(f_pass); + + printf("Password added successfully for %s\n", path); + + } while(0); // END DO + + visible_enter(1); + if(f_pass) + free(f_pass); + if(s_pass) + free(s_pass); + return retval; +} + +int cmd_edit(int argc, char *argv[]) +{ + const char description[] = "edit passname\n"; + int result, fd, pid, len_pass, save_errno; + /* We expect tmpfs to be mounted at /dev/shm */ + char path_tmpfile[] = "/dev/shm/lpass.XXXXXX"; + char *editor, *password; + char *path = argv[2]; + if(!path) + usageprint("%s", description); + + result = check_sneaky_paths(path); + if(result) + errprint_r(1, "You have used forbidden paths\n"); + + result = file_exist(path); + if(result == F_NOEXIST) + errprint_r(1, "No such file exists\n"); + if(result == F_ISDIR) + errprint_r(1, "It's a directory\n"); + + editor = getenv("EDITOR"); + if(!editor) + editor = STD_TEXT_EDITOR; + + password = get_password(path); + if(password == NULL) + errprint_r(1, "Decrypt password failed\n"); + + fd = mkstemp(path_tmpfile); + if(fd == -1) { + free(password); + errprint_r(1, "Create temporary file failed: %s\n", strerror(errno)); + } + dbgprint("tmp file: %s\n", path_tmpfile); + + len_pass = strlen(password); + result = write(fd, password, len_pass); + free(password); + close(fd); + if(result != len_pass) { + unlink(path_tmpfile); + errprint_r(1, "Write password to temporary file failed\n"); + } + + // fork for text editor + char *editor_arg[] = {editor, path_tmpfile, NULL}; + pid = fork(); + if(pid == -1) { + unlink(path_tmpfile); + errprint_r(1, "Start %s failed: %s\n", editor, strerror(errno)); + } + if(pid == 0) { /* new process */ + execvp(editor, editor_arg); + perror(editor); + exit(1); + } + wait(&pid); + + fd = open(path_tmpfile, O_RDONLY); + if(fd == -1) { + unlink(path_tmpfile); + errprint_r(1, "Open temporary file failed: %s\n", strerror(errno)); + } + + password = malloc(sizeof(char) * (maxlen_pass + 1)); + len_pass = read(fd, password, maxlen_pass); + save_errno = errno; + close(fd); + unlink(path_tmpfile); + if(len_pass < minlen_pass) { + free(password); + if(len_pass == -1) + errprint_r(1, "Read temporary file failed: %s\n", strerror(save_errno)); + else + errprint_r(1, "Min. password length is %d\n", minlen_pass); + } + password[len_pass-1] = '\0'; + dbgprint("new pass: %s\n", password); + + // encrypt + result = insert_pass(path, password); + free(password); + if(result) + errprint_r(1, "Can't add password to LockPassword\n"); + + return 0; +} + +int cmd_generate(int argc, char *argv[]) +{ + const char description[] = "generate [-l=pass-length] [-f] passname\n"; + int pass_length = stdlen_pass; + int flag_force = 0, flag_copy = 0, result; + const struct option long_options[] = { + {"length", required_argument, NULL, 'l'}, + {"force", no_argument, NULL, 'f'}, + {"copy", no_argument, NULL, 'c'}, + {NULL, 0, NULL, 0} + }; + + while((result = getopt_long(argc, argv, "l:fc", long_options, NULL)) != -1) { + switch(result) { + // if optarg - incorrect number, atoi return 0 + case 'l': { pass_length = atoi(optarg); break; } + case 'f': { flag_force = 1; break; } + case 'c': { flag_copy = 1; break; } + default: usageprint("%s", description); + } + } + + if(optind < argc) optind++; // for skip "generate" + dbgprint("passname: %s\n", argv[optind]); + + char *path = argv[optind]; + if(path == NULL) + usageprint("%s", description); + + if(pass_length < minlen_pass || pass_length > maxlen_pass) + errprint_r(1, "You typed an incorrect length\n"); + + result = check_sneaky_paths(path); + if(result) + errprint_r(1, "You have used forbidden paths\n"); + + result = file_exist(path); + if(result == F_ISFILE) { + if(!flag_force) { + if(overwrite_answer(path) != 'y') + return 1; + } + } + else if(result == F_ISDIR) + errprint_r(1, "You can't generate password for directory\n"); + + // generate password + char *g_pass = gen_password(pass_length); + + result = insert_pass(path, g_pass); + if(result) { + free(g_pass); + errprint_r(1, "Can't add password to LockPassword\n"); + } + + if(flag_copy) + copy_outside(g_pass); + else + printf("Generated password: %s\n", g_pass); + printf("Password added successfully for %s\n", path); + free(g_pass); + return 0; +} + +int cmd_remove(int argc, char *argv[]) +{ + const char description[] = "rm passname\n"; + int result; + char *path = argv[2]; + if(!path) + usageprint("%s", description); + + result = check_sneaky_paths(path); + if(result) + errprint_r(1, "You have used forbidden paths\n"); + + result = file_exist(path); + if(result == F_NOEXIST) + errprint_r(1, "No such file exists\n"); + if(result == F_ISDIR) { + if(count_dir_entries(path) != 0) + errprint_r(1, "Directory not empty\n"); + } + + result = remove(path); + if(result) + errprint_r(1, "%s\n", strerror(errno)); + return 0; +} + +int cmd_move(int argc, char *argv[]) +{ + /* we have a two situation: + 1) mv file file + 2) mv file directory */ + + const char description[] = "mv [-f] old-path new-path\n"; + const struct option long_options[] = { + {"force", no_argument, NULL, 'f'}, + {NULL, 0, NULL, 0} + }; + + int result, flag_force = 0; + while((result = getopt_long(argc, argv, "f", long_options, NULL)) != -1) { + switch(result) { + case 'f': { flag_force = 1; break; } + default: usageprint("%s", description); + } + } + + if(optind < argc) optind++; // for skip "move" + if(!argv[optind] || !argv[optind+1]) + usageprint("%s", description); + + char *old_path = argv[optind]; + char *new_path = argv[optind+1]; + dbgprint("old-path = %s\n", old_path); + dbgprint("new-path = %s\n", new_path); + + result = check_sneaky_paths(old_path); + if(result) + errprint_r(1, "You have used forbidden paths\n"); + result = file_exist(old_path); + if(result == F_NOEXIST) + errprint_r(1, "No such file exists\n"); + + result = check_sneaky_paths(new_path); + if(result) + errprint_r(1, "You have used forbidden paths\n"); + result = file_exist(new_path); + if(result != F_NOEXIST) { + if(!flag_force) { + if(overwrite_answer(new_path) != 'y') + return 1; + } + } + + result = rename(old_path, new_path); + if(result) + errprint_r(1, "%s\n", strerror(errno)); + return 0; +} + +int cmd_help(int argc, char *argv[]) +{ + printf("Synopsis:\n" + "\tlpass [command] [arguments] ...\n" + + "Commands:\n" + "\tinit gpg-key\n" + "\t\tInitialize the password manager using the passed gpg-key.\n" + "\tinsert [-e, --echo] [-c, --copy] [-f, --force] passname\n" + "\t\tAdd the specified passname to the password manager.\n" + "\tedit [-t, --text-editor=text-editor] passname\n" + "\t\tOpen the specified passname in a text editor, waiting for changes.\n" + "\tgenerate [-l, --length=pass-length] [-c, --copy] [-f, --force] passname\n" + "\t\tGenerate a random password and write it in passname.\n" + "\tmv [-f, --force] old-path new-path\n" + "\t\tMove/rename old-path to new-path.\n" + "\trm passname\n" + "\t\tRemove the passname you specified from the password manager.\n" + "\thelp\n" + "\t\tPrint help information about commands and the application itself.\n" + "\tversion\n" + "\t\tPrint version information.\n" + + "\nMore information may be found in the lpass(1) man page.\n"); + return 0; +} + +int cmd_version(int argc, char *argv[]) +{ + printf("LockPassword v%s\n" + "Release date: %s\n\n" + "Copyright (C) 2020-2021 Aleksandr D. Goncharov (Joursoir)\n" + "License: GNU GPL version 3\n" + "This is free software: you are free to change and redistribute it.\n" + "This program comes with ABSOLUTELY NO WARRANTY.\n", + VERSION, DATE_RELEASE); + return 0; +} + +int cmd_showtree(int argc, char *argv[]) +{ + const char description[] = "[-c] [passname]\n"; + int flag_copy = 0; + int retval = 0, result; + char *path; + const struct option long_options[] = { + {"copy", no_argument, NULL, 'c'}, + {NULL, 0, NULL, 0} + }; + + while((result = getopt_long(argc, argv, "c", long_options, NULL)) != -1) { + switch(result) { + case 'c': { flag_copy = 1; break; } + default: usageprint("%s", description); + } + } + + if(argv[optind]) { + result = check_sneaky_paths(argv[optind]); + if(result) + errprint_r(1, "You have used forbidden paths\n"); + path = malloc(sizeof(char) * (strlen(argv[optind]) + 1)); + strcpy(path, argv[optind]); + } + else { + path = malloc(sizeof(char) * 2); + strcpy(path, "."); + } + + do { // START_DO + + result = file_exist(path); + if(result == F_ISDIR) + { + if(flag_copy) { + errprint_ptr(&retval, 1, + "You must type a passname, not a directory\n"); + break; + } + + if(strcmp(path, ".") == 0) + printf("Password Manager\n"); + else + printf("Password Manager/%s\n", path); + tree(path, ""); + } + else if(result == F_ISFILE) + { + char *pass = get_password(path); + if(!pass) { + errprint_ptr(&retval, 1, "Decrypt password failed\n"); + break; + } + + if(flag_copy) + copy_outside(pass); + else + printf("%s\n", pass); + + free(pass); + } + else + errprint_ptr(&retval, 1, + "This path is not in the password storage\n"); + + } while(0); // END_DO + + free(path); + return retval; +} -- cgit v1.2.3-18-g5258