Compare commits
3 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
| 8c8c873e73 | |||
| 692f65b7f0 | |||
| 1c4200a73a |
125
otp.c
125
otp.c
@@ -50,6 +50,11 @@ static char current_pads_dir[512] = DEFAULT_PADS_DIR;
|
|||||||
int main(int argc, char* argv[]);
|
int main(int argc, char* argv[]);
|
||||||
int interactive_mode(void);
|
int interactive_mode(void);
|
||||||
int command_line_mode(int argc, char* argv[]);
|
int command_line_mode(int argc, char* argv[]);
|
||||||
|
int pipe_mode(int argc, char* argv[], const char* piped_text);
|
||||||
|
|
||||||
|
// Stdin detection functions
|
||||||
|
int has_stdin_data(void);
|
||||||
|
char* read_stdin_text(void);
|
||||||
|
|
||||||
// OTP thumb drive detection function
|
// OTP thumb drive detection function
|
||||||
int detect_otp_thumb_drive(char* otp_drive_path, size_t path_size);
|
int detect_otp_thumb_drive(char* otp_drive_path, size_t path_size);
|
||||||
@@ -124,7 +129,18 @@ int main(int argc, char* argv[]) {
|
|||||||
current_pads_dir[sizeof(current_pads_dir) - 1] = '\0';
|
current_pads_dir[sizeof(current_pads_dir) - 1] = '\0';
|
||||||
}
|
}
|
||||||
|
|
||||||
if (argc == 1) {
|
// Check for piped input
|
||||||
|
if (argc == 1 && has_stdin_data()) {
|
||||||
|
// No arguments but has piped data - enter pipe mode for interactive pad selection
|
||||||
|
char* piped_text = read_stdin_text();
|
||||||
|
if (piped_text) {
|
||||||
|
int result = pipe_mode(argc, argv, piped_text);
|
||||||
|
free(piped_text);
|
||||||
|
return result;
|
||||||
|
}
|
||||||
|
// If reading stdin failed, fall back to interactive mode
|
||||||
|
return interactive_mode();
|
||||||
|
} else if (argc == 1) {
|
||||||
return interactive_mode();
|
return interactive_mode();
|
||||||
} else {
|
} else {
|
||||||
return command_line_mode(argc, argv);
|
return command_line_mode(argc, argv);
|
||||||
@@ -158,6 +174,7 @@ int interactive_mode(void) {
|
|||||||
handle_pads_menu();
|
handle_pads_menu();
|
||||||
break;
|
break;
|
||||||
case 'X':
|
case 'X':
|
||||||
|
case 'Q':
|
||||||
printf("Goodbye!\n");
|
printf("Goodbye!\n");
|
||||||
return 0;
|
return 0;
|
||||||
default:
|
default:
|
||||||
@@ -2322,6 +2339,94 @@ void get_pad_path(const char* chksum, char* pad_path, char* state_path) {
|
|||||||
snprintf(state_path, 1024, "%s/%s.state", current_pads_dir, chksum);
|
snprintf(state_path, 1024, "%s/%s.state", current_pads_dir, chksum);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Stdin detection functions implementation
|
||||||
|
int has_stdin_data(void) {
|
||||||
|
// Check if stdin is a pipe/redirect (not a terminal)
|
||||||
|
if (!isatty(STDIN_FILENO)) {
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
char* read_stdin_text(void) {
|
||||||
|
size_t capacity = 4096;
|
||||||
|
size_t length = 0;
|
||||||
|
char* buffer = malloc(capacity);
|
||||||
|
|
||||||
|
if (!buffer) {
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
|
||||||
|
char chunk[1024];
|
||||||
|
while (fgets(chunk, sizeof(chunk), stdin)) {
|
||||||
|
size_t chunk_len = strlen(chunk);
|
||||||
|
|
||||||
|
// Ensure we have enough capacity
|
||||||
|
while (length + chunk_len >= capacity) {
|
||||||
|
capacity *= 2;
|
||||||
|
char* new_buffer = realloc(buffer, capacity);
|
||||||
|
if (!new_buffer) {
|
||||||
|
free(buffer);
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
buffer = new_buffer;
|
||||||
|
}
|
||||||
|
|
||||||
|
strcpy(buffer + length, chunk);
|
||||||
|
length += chunk_len;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Remove trailing newline if present
|
||||||
|
if (length > 0 && buffer[length - 1] == '\n') {
|
||||||
|
buffer[length - 1] = '\0';
|
||||||
|
length--;
|
||||||
|
}
|
||||||
|
|
||||||
|
// If empty, free and return NULL
|
||||||
|
if (length == 0) {
|
||||||
|
free(buffer);
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
|
||||||
|
return buffer;
|
||||||
|
}
|
||||||
|
|
||||||
|
int pipe_mode(int argc, char* argv[], const char* piped_text) {
|
||||||
|
(void)argc; // Suppress unused parameter warning
|
||||||
|
(void)argv; // Suppress unused parameter warning
|
||||||
|
|
||||||
|
printf("Piped text received: \"%s\"\n\n", piped_text);
|
||||||
|
|
||||||
|
// List available pads for selection
|
||||||
|
int pad_count = list_available_pads();
|
||||||
|
if (pad_count == 0) {
|
||||||
|
printf("No pads available. Generate a pad first.\n");
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Reopen stdin from the controlling terminal for interactive input
|
||||||
|
FILE* tty = fopen("/dev/tty", "r");
|
||||||
|
if (!tty) {
|
||||||
|
printf("Error: Cannot open terminal for input\n");
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
printf("\nEnter pad selection (number, checksum, or prefix): ");
|
||||||
|
fflush(stdout);
|
||||||
|
|
||||||
|
char pad_input[MAX_HASH_LENGTH];
|
||||||
|
if (!fgets(pad_input, sizeof(pad_input), tty)) {
|
||||||
|
printf("Error: Failed to read pad selection\n");
|
||||||
|
fclose(tty);
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
pad_input[strcspn(pad_input, "\n")] = 0;
|
||||||
|
fclose(tty);
|
||||||
|
|
||||||
|
// Encrypt the piped text
|
||||||
|
return encrypt_text(pad_input, piped_text);
|
||||||
|
}
|
||||||
|
|
||||||
// OTP thumb drive detection function implementation
|
// OTP thumb drive detection function implementation
|
||||||
int detect_otp_thumb_drive(char* otp_drive_path, size_t path_size) {
|
int detect_otp_thumb_drive(char* otp_drive_path, size_t path_size) {
|
||||||
const char* mount_dirs[] = {"/media", "/run/media", "/mnt", NULL};
|
const char* mount_dirs[] = {"/media", "/run/media", "/mnt", NULL};
|
||||||
@@ -3045,10 +3150,15 @@ int handle_pads_menu(void) {
|
|||||||
|
|
||||||
char input[10];
|
char input[10];
|
||||||
if (fgets(input, sizeof(input), stdin)) {
|
if (fgets(input, sizeof(input), stdin)) {
|
||||||
char choice = toupper(input[0]);
|
char choice = toupper(input[0]);
|
||||||
if (choice == 'G') {
|
if (choice == 'G') {
|
||||||
return handle_generate_menu();
|
int result = handle_generate_menu();
|
||||||
|
if (result == 0) {
|
||||||
|
// After successful pad generation, return to pads menu
|
||||||
|
return handle_pads_menu();
|
||||||
}
|
}
|
||||||
|
return result;
|
||||||
|
}
|
||||||
}
|
}
|
||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
@@ -3117,7 +3227,12 @@ int handle_pads_menu(void) {
|
|||||||
|
|
||||||
// Handle actions first
|
// Handle actions first
|
||||||
if (toupper(input[0]) == 'G') {
|
if (toupper(input[0]) == 'G') {
|
||||||
return handle_generate_menu();
|
int result = handle_generate_menu();
|
||||||
|
if (result == 0) {
|
||||||
|
// After successful pad generation, return to pads menu
|
||||||
|
return handle_pads_menu();
|
||||||
|
}
|
||||||
|
return result;
|
||||||
} else if (toupper(input[0]) == 'B') {
|
} else if (toupper(input[0]) == 'B') {
|
||||||
return 0; // Back to main menu
|
return 0; // Back to main menu
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user