aboutsummaryrefslogtreecommitdiffstats
path: root/main.c
diff options
context:
space:
mode:
Diffstat (limited to 'main.c')
-rw-r--r--main.c398
1 files changed, 398 insertions, 0 deletions
diff --git a/main.c b/main.c
new file mode 100644
index 0000000..5d579cd
--- /dev/null
+++ b/main.c
@@ -0,0 +1,398 @@
+#include <stdio.h>
+#include <termios.h>
+#include <string.h>
+#include <stdlib.h>
+#include <unistd.h>
+#include <sys/wait.h>
+#include <getopt.h>
+#include <errno.h>
+#include <dirent.h>
+#include <time.h>
+
+#include "easydir.h"
+#include "handerror.h"
+
+//#define DEBUG
+#define MAXLEN_PASSWORD 128
+#define FULLNAMEFILE "/.lock-password/Password-Bank"
+#define TREENAMEFILE "/.lock-password"
+#define ADDNAMETOFULL "/Password-Bank"
+#define NAMEFILE "Password-Bank"
+
+static char* get_password(char *path, char *password)
+{
+ FILE *filePass;
+ filePass = fopen(path, "r");
+ if(filePass == NULL) {
+ call_error(110);
+ }
+
+ char sign[MAXLEN_PASSWORD];
+ if(!fgets(sign, sizeof(sign), filePass)) {
+ call_error(111);
+ }
+
+ strcpy(password, sign);
+ fclose(filePass);
+ return password;
+}
+
+static void show_tree(char *path)
+{
+ if(opendir(path) != NULL) { // if it's directory
+ int pid;
+ pid = fork();
+ if(pid == -1) call_error(101);
+ if(pid == 0) { /* new process */
+ execlp("tree", "tree", "-C", "--noreport", path, NULL);
+ perror("tree");
+ exit(4);
+ }
+ wait(&pid);
+ }
+ else {
+ char password[MAXLEN_PASSWORD];
+ get_password(path, password);
+ printf("%s\n", password);
+ }
+}
+
+static void changePass(char *root_path, char *add_path, char *password)
+{
+ /* root_path = /home/[username]/
+ add_path = banks/france/[number]
+ main_path = banks/france
+ file_path = [number] */
+
+ char *main_path = malloc(sizeof(char) * strlen(add_path) + 1);
+ char *file_path = malloc(sizeof(char) * strlen(add_path) + 1);
+
+ if(splitPath(add_path, main_path, file_path) == NULL) {
+ print_error("[Error] The path you specified is incorrect\n");
+ }
+
+ int pass_buf = strlen(root_path) + strlen(add_path);
+ char *final_path = (char*) malloc(sizeof(char) * pass_buf + 1);
+
+ strcpy(final_path, root_path);
+ strcat(final_path, "/");
+ strcat(final_path, main_path);
+
+ if(chdir(final_path) != 0) {
+ call_error(107);
+ }
+
+ FILE *filePass;
+ filePass = fopen(file_path, "r+");
+ if(filePass == NULL) {
+ if(errno == ENOENT) { // file doesn't exist
+ print_error("Error: No such file exists\n");
+ }
+ call_error(114);
+ }
+ fputs(password, filePass);
+
+ free(main_path);
+ free(file_path);
+ fclose(filePass);
+}
+
+static void insertPass(char *root_path, char *add_path, char *password)
+{
+ /* root_path = /home/[username]/
+ add_path = banks/france/[number]
+ main_path = banks/france
+ file_path = [number] */
+
+ char *main_path = malloc(sizeof(char) * strlen(add_path) + 1);
+ char *file_path = malloc(sizeof(char) * strlen(add_path) + 1);
+
+ if(splitPath(add_path, main_path, file_path) == NULL) {
+ print_error("[Error] The path you specified is incorrect\n");
+ }
+
+ int pass_buf = strlen(root_path) + strlen(add_path);
+ char *final_path = (char*) malloc(sizeof(char) * pass_buf + 1);
+
+ strcpy(final_path, root_path);
+ strcat(final_path, "/");
+ strcat(final_path, main_path);
+
+ int pid = fork();
+ if(pid == -1) call_error(103);
+ if(pid == 0) { /* new process */
+ execlp("mkdir", "mkdir", "-p", main_path, NULL);
+ perror("mkdir");
+ exit(4);
+ }
+ wait(&pid);
+
+ if(chdir(final_path) != 0) {
+ call_error(107);
+ }
+
+ // create file, copy password there
+ FILE *filePass;
+ filePass = fopen(file_path, "w");
+ if(filePass == NULL) {
+ call_error(108);
+ }
+ fputs(password, filePass);
+
+ free(main_path);
+ free(file_path);
+ fclose(filePass);
+}
+
+static char *generate_password(char *dest, int amount)
+{
+ char allowed_symbols[] = {
+ 'A','E','I','J','O','U','B','C','D','F','G','H',
+ 'K','L','M','N','P','Q','R','S','T','V','W','X',
+ 'Y','Z','a','e','i','j','o','u','b','c','d','f',
+ 'g','h','k','l','m','n','p','q','r','s','t','v',
+ 'w','x','y','z','1','2','3','4','5','6','7','8',
+ '9','0','!','#','$',';','%','^',':','&','?','*',
+ '(',')','-','_','+','=','<', '>'
+ };
+ int max = sizeof(allowed_symbols);
+ srand(time(NULL));
+
+ char password[MAXLEN_PASSWORD];
+ for(int i=0; i < amount; i++)
+ {
+ char c = allowed_symbols[rand() % max];
+
+ password[i] = c;
+ password[i+1] = '\0';
+ }
+
+ strcpy(dest, password);
+ return dest;
+}
+
+static void nonvisible_enter(int status)
+{
+ struct termios term_settings;
+ tcgetattr(0, &term_settings); // get current settings
+ if(status == 1) {
+ term_settings.c_lflag &= ~ECHO; // flag reset
+ }
+ else {
+ term_settings.c_lflag |= ECHO;
+ }
+ tcsetattr(0, TCSANOW, &term_settings);
+}
+
+int main(int argc, char *argv[])
+{
+ if(!isatty(0)) { // stdin
+ print_error("Please, use a terminal to run this program\n");
+ }
+
+ #if defined(DEBUG)
+ char* short_options = "c:e:g:i:R:t";
+ #else
+ char* short_options = "c:e:g:i:R:";
+ #endif
+
+ struct option long_options[] = {
+ {"init", no_argument, NULL, 'I'},
+ {NULL, 0, NULL, 0}
+ };
+
+ int rootBuf = strlen(getenv("HOME")) + strlen(FULLNAMEFILE);
+ char *rootPath = (char *) malloc(sizeof(char) * rootBuf + 1); // +1 for '\0'
+
+ int path_init = 0;
+ strcpy(rootPath, getenv("HOME"));
+ strcat(rootPath, TREENAMEFILE);
+
+ if(chdir(rootPath) == 0) path_init = 1;
+
+ int option_index = -1, result = -1;
+ result = getopt_long(argc, argv, short_options, long_options, &option_index);
+ if(!path_init) {
+ if(result != 'I') // if doesn't init
+ {
+ if(errno == ENOENT) { // file doesn't exist
+ print_error("Before starting work, you must initialize LockPassword\n\
+ Use: lpass --init\n");
+ }
+ call_error(102);
+ }
+ }
+ else {
+ if(result == 'I') {
+ print_error("You already initialized LockPassword\n");
+ }
+ else if(result != -1)
+ {
+ strcat(rootPath, ADDNAMETOFULL); // complements to full path
+ if(chdir(rootPath) != 0) {
+ call_error(112);
+ }
+ }
+ }
+
+ if(result != -1)
+ {
+ switch(result) {
+ case 'c': {
+ printf("Coming soon...\n");
+ break;
+ }
+ case 'e': {
+ nonvisible_enter(1); /* change terminal work */
+
+ /* enter password */
+ char pass_one[MAXLEN_PASSWORD], pass_two[MAXLEN_PASSWORD];
+
+ printf("Please type your new password: ");
+ if(!fgets(pass_one, sizeof(pass_one), stdin)) {
+ nonvisible_enter(0);
+ print_error("Unexpected end of file\n");
+ }
+ printf("\n");
+
+ printf("Please type your new password again: ");
+ if(!fgets(pass_two, sizeof(pass_two), stdin)) {
+ nonvisible_enter(0);
+ print_error("Unexpected end of file\n");
+ }
+ printf("\n");
+
+ nonvisible_enter(0);
+
+ if(strcmp(pass_one, pass_two) == 0) {
+ printf("Password correct\n");
+
+ changePass(rootPath, optarg, pass_one);
+ printf("Password updated successfully for %s\n", optarg);
+ }
+ else printf("Passwords do not match\n");
+
+ break;
+ }
+ case 'g': {
+ int n_symbols = 0;
+ if(strcmp(optarg, argv[argc-1]) == 0)
+ n_symbols = 8;
+ else n_symbols = atoi(optarg);
+
+
+ if(n_symbols < 1 || n_symbols > 127) {
+ print_error("Error: you typed an incorrect number");
+ }
+
+ /* generate password */
+ char gpass[MAXLEN_PASSWORD];
+ generate_password(gpass, n_symbols);
+
+ insertPass(rootPath, argv[argc-1], gpass);
+ printf("Generated password: %s\n", gpass);
+ printf("Password added successfully for %s\n", argv[argc-1]);
+
+ break;
+ }
+ case 'i': {
+ nonvisible_enter(1); /* change terminal work */
+
+ /* enter password */
+ char pass_one[MAXLEN_PASSWORD], pass_two[MAXLEN_PASSWORD];
+
+ printf("Please type your password: ");
+ if(!fgets(pass_one, sizeof(pass_one), stdin)) {
+ nonvisible_enter(0);
+ print_error("Unexpected end of file\n");
+ }
+ printf("\n");
+
+ printf("Please type your password again: ");
+ if(!fgets(pass_two, sizeof(pass_two), stdin)) {
+ nonvisible_enter(0);
+ print_error("Unexpected end of file\n");
+ }
+ printf("\n");
+
+ nonvisible_enter(0);
+
+ if(strcmp(pass_one, pass_two) == 0) {
+ printf("Password correct\n");
+
+ insertPass(rootPath, optarg, pass_one);
+ printf("Password added successfully for %s\n", optarg);
+ }
+ else printf("Passwords do not match\n");
+
+ break;
+ }
+ case 'R': {
+ char *main_path = malloc(sizeof(char) * strlen(optarg) + 1);
+ char *file_path = malloc(sizeof(char) * strlen(optarg) + 1);
+
+ if(splitPath(optarg, main_path, file_path) == NULL) { // check correct input
+ print_error("[Error] The path you specified is incorrect\n");
+ }
+
+ if(delete_file(optarg)) {
+ delete_emptydir(main_path);
+ }
+
+ free(main_path);
+ free(file_path);
+ break;
+ }
+ case 'I': {
+ int pid;
+
+ // create direction:
+ pid = fork();
+ if(pid == -1) call_error(100);
+ if(pid == 0) { /* new process */
+ execlp("mkdir", "mkdir", "-vp", rootPath, NULL);
+ perror("mkdir");
+ exit(4);
+ }
+ wait(&pid);
+ printf("LockPassword initialized\n");
+ break;
+ }
+
+ #if defined(DEBUG)
+ case 't':
+ {
+ char *main_path = malloc(sizeof(char) * strlen(optarg) + 1);
+ char *file_path = malloc(sizeof(char) * strlen(optarg) + 1);
+
+ if(splitPath(optarg, main_path, file_path) == NULL)
+ printf("NULL\n");
+ break;
+ }
+ #endif
+
+ default: break;
+
+ }
+ }
+ else
+ {
+ char *ptr;
+ if(strcmp(argv[argc-1], argv[0]) == 0) {
+ ptr = malloc(sizeof(char) * strlen(NAMEFILE) + 1);
+ strcpy(ptr, NAMEFILE);
+ }
+ else {
+ int buff_tree = strlen(NAMEFILE) + strlen(argv[argc-1]);
+ ptr = malloc(sizeof(char) * buff_tree + 1 + 1);
+ strcpy(ptr, NAMEFILE);
+ strcat(ptr, "/");
+ strcat(ptr, argv[argc-1]);
+ }
+ show_tree(ptr);
+
+ free(ptr);
+ }
+
+ return 0;
+} \ No newline at end of file
generated by cgit v1.2.3-18-g5258 (git 2.32.0) at 2024-12-22 12:46:57 +0000