Mirrors/Stirling-PDF
Mirrors/Stirling-PDF
1
0
Fork 0
mirror of https://github.com/Frooodle/Stirling-PDF.git synced 2025-01-19 00:07:17 +01:00
Code Issues Packages Projects Releases Wiki Activity

Check if file is encrypted without password

Browse Source
This commit is contained in:
Reece Browne 2024-12-09 13:20:08 +00:00
parent 6ee6254f5a
commit 1d6511b043
2 changed files with 60 additions and 36 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

80
src/main/resources/static/js/multitool/DecryptFiles.js
View File

@ -1,34 +1,30 @@
export class DecryptFile {
async decryptFile(file) {
async decryptFile(file, requiresPassword) {
try {
const password = prompt(`${window.translations.passwordPrompt}`);
if (password === null) {
// User cancelled
console.error(`Password prompt cancelled for PDF: ${file.name}`);
this.showErrorBanner(
`${window.translations.cancelled.replace('{0}', file.name)}`,
'',
`${window.translations.unexpectedError}`
);
return null; // No file to return
}
if (!password) {
// No password provided
console.error(`No password provided for encrypted PDF: ${file.name}`);
this.showErrorBanner(
`${window.translations.noPassword.replace('{0}', file.name)}`,
'',
`${window.translations.unexpectedError}`
);
return null; // No file to return
}
const formData = new FormData();
formData.append('fileInput', file);
formData.append('password', password);
if (requiresPassword) {
const password = prompt(`${window.translations.passwordPrompt}`);
if (password === null) {
// User cancelled
console.error(`Password prompt cancelled for PDF: ${file.name}`);
return null; // No file to return
}
if (!password) {
// No password provided
console.error(`No password provided for encrypted PDF: ${file.name}`);
this.showErrorBanner(
`${window.translations.noPassword.replace('{0}', file.name)}`,
'',
`${window.translations.unexpectedError}`
);
return null; // No file to return
}
formData.append('password', password);
}
// Send decryption request
const response = await fetch('/api/v1/security/remove-password', {
method: 'POST',
@ -64,17 +60,35 @@ export class DecryptFile {
async checkFileEncrypted(file) {
try {
pdfjsLib.GlobalWorkerOptions.workerSrc = './pdfjs-legacy/pdf.worker.mjs';
const arrayBuffer = await file.arrayBuffer(); // Convert file to ArrayBuffer
await pdfjsLib.getDocument({
data: arrayBuffer,
password: '',
}).promise;
const arrayBuffer = await file.arrayBuffer();
const arrayBufferForPdfLib = arrayBuffer.slice(0);
return false; // File is not encrypted
const loadingTask = pdfjsLib.getDocument({
data: arrayBuffer,
});
await loadingTask.promise;
try {
//Uses PDFLib.PDFDocument to check if unpassworded but encrypted
const pdfDoc = await PDFLib.PDFDocument.load(arrayBufferForPdfLib);
return {isEncrypted: false, requiresPassword: false};
} catch (error) {
if (error.message.includes('Input document to `PDFDocument.load` is encrypted')) {
return {isEncrypted: true, requiresPassword: false};
}
console.error('Error checking encryption:', error);
throw new Error('Failed to determine if the file is encrypted.');
}
} catch (error) {
if (error.name === 'PasswordException') {
return true; // File is encrypted
if (error.code === pdfjsLib.PasswordResponses.NEED_PASSWORD) {
return {isEncrypted: true, requiresPassword: true};
} else if (error.code === pdfjsLib.PasswordResponses.INCORRECT_PASSWORD) {
return {isEncrypted: true, requiresPassword: false};
}
}
console.error('Error checking encryption:', error);
throw new Error('Failed to determine if the file is encrypted.');
}

16
src/main/resources/static/js/multitool/PdfContainer.js
View File

@ -178,9 +178,19 @@ class PdfContainer {
try {
let decryptedFile = files[i];
if (decryptedFile.type === 'application/pdf' && (await this.decryptFile.checkFileEncrypted(decryptedFile))) {
decryptedFile = await this.decryptFile.decryptFile(decryptedFile);
let isEncrypted = false;
let requiresPassword = false;
await this.decryptFile
.checkFileEncrypted(decryptedFile)
.then((result) => {
isEncrypted = result.isEncrypted;
requiresPassword = result.requiresPassword;
})
.catch((error) => {
console.error(error);
});
if (decryptedFile.type === 'application/pdf' && isEncrypted) {
decryptedFile = await this.decryptFile.decryptFile(decryptedFile, requiresPassword);
if (!decryptedFile) {
throw new Error('File decryption failed.');
}