init project

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