Compare commits
2 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
| 1c9e2ee527 | |||
| 8401e14ae0 |
60
otp.c
60
otp.c
@@ -1127,10 +1127,39 @@ int generate_pad_with_entropy(uint64_t size_bytes, int display_progress, int use
|
|||||||
// Get final paths in pads directory
|
// Get final paths in pads directory
|
||||||
get_pad_path(chksum_hex, pad_path, state_path);
|
get_pad_path(chksum_hex, pad_path, state_path);
|
||||||
|
|
||||||
|
// Try rename first (works for same filesystem)
|
||||||
if (rename(temp_filename, pad_path) != 0) {
|
if (rename(temp_filename, pad_path) != 0) {
|
||||||
printf("Error: Cannot move pad file to pads directory\n");
|
// If rename fails, try copy and delete (works across filesystems)
|
||||||
|
FILE* temp_file = fopen(temp_filename, "rb");
|
||||||
|
FILE* dest_file = fopen(pad_path, "wb");
|
||||||
|
|
||||||
|
if (!temp_file || !dest_file) {
|
||||||
|
printf("Error: Cannot copy pad file to pads directory\n");
|
||||||
|
if (temp_file) fclose(temp_file);
|
||||||
|
if (dest_file) fclose(dest_file);
|
||||||
|
unlink(temp_filename);
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Copy file in chunks
|
||||||
|
unsigned char copy_buffer[64 * 1024];
|
||||||
|
size_t bytes_read;
|
||||||
|
while ((bytes_read = fread(copy_buffer, 1, sizeof(copy_buffer), temp_file)) > 0) {
|
||||||
|
if (fwrite(copy_buffer, 1, bytes_read, dest_file) != bytes_read) {
|
||||||
|
printf("Error: Failed to copy pad file to pads directory\n");
|
||||||
|
fclose(temp_file);
|
||||||
|
fclose(dest_file);
|
||||||
|
unlink(temp_filename);
|
||||||
|
unlink(pad_path);
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fclose(temp_file);
|
||||||
|
fclose(dest_file);
|
||||||
|
|
||||||
|
// Remove temporary file after successful copy
|
||||||
unlink(temp_filename);
|
unlink(temp_filename);
|
||||||
return 1;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// Set pad file to read-only
|
// Set pad file to read-only
|
||||||
@@ -3175,8 +3204,15 @@ void get_directory_display(const char* file_path, char* result, size_t result_si
|
|||||||
if (path_after_media) {
|
if (path_after_media) {
|
||||||
path_after_media++; // Skip the slash
|
path_after_media++; // Skip the slash
|
||||||
|
|
||||||
|
// For /media/user/LABEL pattern, skip the username to get to the drive label
|
||||||
|
if (strstr(media_start, "/media/")) {
|
||||||
|
char* next_slash = strchr(path_after_media, '/');
|
||||||
|
if (next_slash) {
|
||||||
|
path_after_media = next_slash + 1;
|
||||||
|
}
|
||||||
|
}
|
||||||
// For /run/media/user/LABEL pattern, skip the username
|
// For /run/media/user/LABEL pattern, skip the username
|
||||||
if (strstr(media_start, "/run/media/")) {
|
else if (strstr(media_start, "/run/media/")) {
|
||||||
char* next_slash = strchr(path_after_media, '/');
|
char* next_slash = strchr(path_after_media, '/');
|
||||||
if (next_slash) {
|
if (next_slash) {
|
||||||
path_after_media = next_slash + 1;
|
path_after_media = next_slash + 1;
|
||||||
@@ -3185,15 +3221,23 @@ void get_directory_display(const char* file_path, char* result, size_t result_si
|
|||||||
|
|
||||||
// Extract just the USB label (up to next slash or end)
|
// Extract just the USB label (up to next slash or end)
|
||||||
char* label_end = strchr(path_after_media, '/');
|
char* label_end = strchr(path_after_media, '/');
|
||||||
|
char usb_label[32];
|
||||||
if (label_end) {
|
if (label_end) {
|
||||||
size_t label_len = label_end - path_after_media;
|
size_t label_len = label_end - path_after_media;
|
||||||
if (label_len > 11) label_len = 11; // Max 11 chars for display
|
if (label_len > sizeof(usb_label) - 1) label_len = sizeof(usb_label) - 1;
|
||||||
strncpy(result, path_after_media, label_len);
|
strncpy(usb_label, path_after_media, label_len);
|
||||||
result[label_len] = '\0';
|
usb_label[label_len] = '\0';
|
||||||
} else {
|
} else {
|
||||||
// USB label is the last part
|
// USB label is the last part
|
||||||
strncpy(result, path_after_media, result_size - 1);
|
strncpy(usb_label, path_after_media, sizeof(usb_label) - 1);
|
||||||
result[result_size - 1] = '\0';
|
usb_label[sizeof(usb_label) - 1] = '\0';
|
||||||
|
}
|
||||||
|
|
||||||
|
// Format with USB: prefix, limiting total length to fit in result
|
||||||
|
snprintf(result, result_size, "USB:%s", usb_label);
|
||||||
|
// Truncate if too long
|
||||||
|
if (strlen(result) > 11) {
|
||||||
|
result[11] = '\0';
|
||||||
}
|
}
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user