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_pad_path(chksum_hex, pad_path, state_path);
|
||||
|
||||
// Try rename first (works for same filesystem)
|
||||
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);
|
||||
return 1;
|
||||
}
|
||||
|
||||
// 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) {
|
||||
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
|
||||
if (strstr(media_start, "/run/media/")) {
|
||||
else if (strstr(media_start, "/run/media/")) {
|
||||
char* next_slash = strchr(path_after_media, '/');
|
||||
if (next_slash) {
|
||||
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)
|
||||
char* label_end = strchr(path_after_media, '/');
|
||||
char usb_label[32];
|
||||
if (label_end) {
|
||||
size_t label_len = label_end - path_after_media;
|
||||
if (label_len > 11) label_len = 11; // Max 11 chars for display
|
||||
strncpy(result, path_after_media, label_len);
|
||||
result[label_len] = '\0';
|
||||
if (label_len > sizeof(usb_label) - 1) label_len = sizeof(usb_label) - 1;
|
||||
strncpy(usb_label, path_after_media, label_len);
|
||||
usb_label[label_len] = '\0';
|
||||
} else {
|
||||
// USB label is the last part
|
||||
strncpy(result, path_after_media, result_size - 1);
|
||||
result[result_size - 1] = '\0';
|
||||
strncpy(usb_label, path_after_media, sizeof(usb_label) - 1);
|
||||
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;
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user