Sakurai/pdf.js
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Merge pull request #16959 from Snuffleupagus/rm-CipherTransformFactory-closure

Browse Source 
Remove the closure from the `CipherTransformFactory` class
This commit is contained in:
Tim van der Meij 2023-09-17 11:42:20 +02:00 committed by GitHub
commit 98671c3618
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 200 additions and 213 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

83
src/core/crypto.js
View File

@ -1418,14 +1418,14 @@ class CipherTransform {
} }
} }
const CipherTransformFactory = (function CipherTransformFactoryClosure() { class CipherTransformFactory {
const defaultPasswordBytes = new Uint8Array([ static #defaultPasswordBytes = new Uint8Array([
0x28, 0xbf, 0x4e, 0x5e, 0x4e, 0x75, 0x8a, 0x41, 0x64, 0x00, 0x4e, 0x56, 0x28, 0xbf, 0x4e, 0x5e, 0x4e, 0x75, 0x8a, 0x41, 0x64, 0x00, 0x4e, 0x56,
0xff, 0xfa, 0x01, 0x08, 0x2e, 0x2e, 0x00, 0xb6, 0xd0, 0x68, 0x3e, 0x80, 0xff, 0xfa, 0x01, 0x08, 0x2e, 0x2e, 0x00, 0xb6, 0xd0, 0x68, 0x3e, 0x80,
0x2f, 0x0c, 0xa9, 0xfe, 0x64, 0x53, 0x69, 0x7a, 0x2f, 0x0c, 0xa9, 0xfe, 0x64, 0x53, 0x69, 0x7a,
]); ]);
function createEncryptionKey20( #createEncryptionKey20(
revision, revision,
password, password,
ownerPassword, ownerPassword,
@ -1471,7 +1471,7 @@ const CipherTransformFactory = (function CipherTransformFactoryClosure() {
return null; return null;
} }
function prepareKeyData( #prepareKeyData(
fileId, fileId,
password, password,
ownerPassword, ownerPassword,
@ -1494,7 +1494,7 @@ const CipherTransformFactory = (function CipherTransformFactoryClosure() {
} }
j = 0; j = 0;
while (i < 32) { while (i < 32) {
hashData[i++] = defaultPasswordBytes[j++]; hashData[i++] = CipherTransformFactory.#defaultPasswordBytes[j++];
} }
// as now the padded password in the hashData[0..i] // as now the padded password in the hashData[0..i]
for (j = 0, n = ownerPassword.length; j < n; ++j) { for (j = 0, n = ownerPassword.length; j < n; ++j) {
@ -1525,7 +1525,7 @@ const CipherTransformFactory = (function CipherTransformFactoryClosure() {
if (revision >= 3) { if (revision >= 3) {
for (i = 0; i < 32; ++i) { for (i = 0; i < 32; ++i) {
hashData[i] = defaultPasswordBytes[i]; hashData[i] = CipherTransformFactory.#defaultPasswordBytes[i];
} }
for (j = 0, n = fileId.length; j < n; ++j) { for (j = 0, n = fileId.length; j < n; ++j) {
hashData[i++] = fileId[j]; hashData[i++] = fileId[j];
@ -1548,7 +1548,9 @@ const CipherTransformFactory = (function CipherTransformFactoryClosure() {
} }
} else { } else {
cipher = new ARCFourCipher(encryptionKey); cipher = new ARCFourCipher(encryptionKey);
checkData = cipher.encryptBlock(defaultPasswordBytes); checkData = cipher.encryptBlock(
CipherTransformFactory.#defaultPasswordBytes
);
for (j = 0, n = checkData.length; j < n; ++j) { for (j = 0, n = checkData.length; j < n; ++j) {
if (userPassword[j] !== checkData[j]) { if (userPassword[j] !== checkData[j]) {
return null; return null;
@ -1558,7 +1560,7 @@ const CipherTransformFactory = (function CipherTransformFactoryClosure() {
return encryptionKey; return encryptionKey;
} }
function decodeUserPassword(password, ownerPassword, revision, keyLength) { #decodeUserPassword(password, ownerPassword, revision, keyLength) {
const hashData = new Uint8Array(32); const hashData = new Uint8Array(32);
let i = 0; let i = 0;
const n = Math.min(32, password.length); const n = Math.min(32, password.length);
@ -1567,7 +1569,7 @@ const CipherTransformFactory = (function CipherTransformFactoryClosure() {
} }
let j = 0; let j = 0;
while (i < 32) { while (i < 32) {
hashData[i++] = defaultPasswordBytes[j++]; hashData[i++] = CipherTransformFactory.#defaultPasswordBytes[j++];
} }
let hash = calculateMD5(hashData, 0, i); let hash = calculateMD5(hashData, 0, i);
const keyLengthInBytes = keyLength >> 3; const keyLengthInBytes = keyLength >> 3;
@ -1595,9 +1597,7 @@ const CipherTransformFactory = (function CipherTransformFactoryClosure() {
return userPassword; return userPassword;
} }
const identityName = Name.get("Identity"); #buildObjectKey(num, gen, encryptionKey, isAes = false) {
function buildObjectKey(num, gen, encryptionKey, isAes = false) {
const key = new Uint8Array(encryptionKey.length + 9); const key = new Uint8Array(encryptionKey.length + 9);
const n = encryptionKey.length; const n = encryptionKey.length;
let i; let i;
@ -1619,44 +1619,41 @@ const CipherTransformFactory = (function CipherTransformFactoryClosure() {
return hash.subarray(0, Math.min(encryptionKey.length + 5, 16)); return hash.subarray(0, Math.min(encryptionKey.length + 5, 16));
} }
function buildCipherConstructor(cf, name, num, gen, key) { #buildCipherConstructor(cf, name, num, gen, key) {
if (!(name instanceof Name)) { if (!(name instanceof Name)) {
throw new FormatError("Invalid crypt filter name."); throw new FormatError("Invalid crypt filter name.");
} }
const self = this;
const cryptFilter = cf.get(name.name); const cryptFilter = cf.get(name.name);
let cfm; const cfm = cryptFilter?.get("CFM");
if (cryptFilter !== null && cryptFilter !== undefined) {
cfm = cryptFilter.get("CFM");
}
if (!cfm || cfm.name === "None") { if (!cfm || cfm.name === "None") {
return function cipherTransformFactoryBuildCipherConstructorNone() { return function () {
return new NullCipher(); return new NullCipher();
}; };
} }
if (cfm.name === "V2") { if (cfm.name === "V2") {
return function cipherTransformFactoryBuildCipherConstructorV2() { return function () {
return new ARCFourCipher( return new ARCFourCipher(
buildObjectKey(num, gen, key, /* isAes = */ false) self.#buildObjectKey(num, gen, key, /* isAes = */ false)
); );
}; };
} }
if (cfm.name === "AESV2") { if (cfm.name === "AESV2") {
return function cipherTransformFactoryBuildCipherConstructorAESV2() { return function () {
return new AES128Cipher( return new AES128Cipher(
buildObjectKey(num, gen, key, /* isAes = */ true) self.#buildObjectKey(num, gen, key, /* isAes = */ true)
); );
}; };
} }
if (cfm.name === "AESV3") { if (cfm.name === "AESV3") {
return function cipherTransformFactoryBuildCipherConstructorAESV3() { return function () {
return new AES256Cipher(key); return new AES256Cipher(key);
}; };
} }
throw new FormatError("Unknown crypto method"); throw new FormatError("Unknown crypto method");
} }
// eslint-disable-next-line no-shadow
class CipherTransformFactory {
constructor(dict, fileId, password) { constructor(dict, fileId, password) {
const filter = dict.get("Filter"); const filter = dict.get("Filter");
if (!isName(filter, "Standard")) { if (!isName(filter, "Standard")) {
@ -1667,10 +1664,7 @@ const CipherTransformFactory = (function CipherTransformFactoryClosure() {
const algorithm = dict.get("V"); const algorithm = dict.get("V");
if ( if (
!Number.isInteger(algorithm) || !Number.isInteger(algorithm) ||
(algorithm !== 1 && (algorithm !== 1 && algorithm !== 2 && algorithm !== 4 && algorithm !== 5)
algorithm !== 2 &&
algorithm !== 4 &&
algorithm !== 5)
) { ) {
throw new FormatError("unsupported encryption algorithm"); throw new FormatError("unsupported encryption algorithm");
} }
@ -1698,11 +1692,7 @@ const CipherTransformFactory = (function CipherTransformFactoryClosure() {
} }
} }
} }
if ( if (!Number.isInteger(keyLength) || keyLength < 40 || keyLength % 8 !== 0) {
!Number.isInteger(keyLength) ||
keyLength < 40 ||
keyLength % 8 !== 0
) {
throw new FormatError("invalid key length"); throw new FormatError("invalid key length");
} }
@ -1736,7 +1726,7 @@ const CipherTransformFactory = (function CipherTransformFactoryClosure() {
let encryptionKey; let encryptionKey;
if (algorithm !== 5) { if (algorithm !== 5) {
encryptionKey = prepareKeyData( encryptionKey = this.#prepareKeyData(
fileIdBytes, fileIdBytes,
passwordBytes, passwordBytes,
ownerPassword, ownerPassword,
@ -1755,7 +1745,7 @@ const CipherTransformFactory = (function CipherTransformFactoryClosure() {
const ownerEncryption = stringToBytes(dict.get("OE")); const ownerEncryption = stringToBytes(dict.get("OE"));
const userEncryption = stringToBytes(dict.get("UE")); const userEncryption = stringToBytes(dict.get("UE"));
const perms = stringToBytes(dict.get("Perms")); const perms = stringToBytes(dict.get("Perms"));
encryptionKey = createEncryptionKey20( encryptionKey = this.#createEncryptionKey20(
revision, revision,
passwordBytes, passwordBytes,
ownerPassword, ownerPassword,
@ -1777,13 +1767,13 @@ const CipherTransformFactory = (function CipherTransformFactoryClosure() {
); );
} else if (!encryptionKey && password) { } else if (!encryptionKey && password) {
// Attempting use the password as an owner password // Attempting use the password as an owner password
const decodedPassword = decodeUserPassword( const decodedPassword = this.#decodeUserPassword(
passwordBytes, passwordBytes,
ownerPassword, ownerPassword,
revision, revision,
keyLength keyLength
); );
encryptionKey = prepareKeyData( encryptionKey = this.#prepareKeyData(
fileIdBytes, fileIdBytes,
decodedPassword, decodedPassword,
ownerPassword, ownerPassword,
@ -1814,8 +1804,8 @@ const CipherTransformFactory = (function CipherTransformFactoryClosure() {
cf.suppressEncryption = true; cf.suppressEncryption = true;
} }
this.cf = cf; this.cf = cf;
this.stmf = dict.get("StmF") || identityName; this.stmf = dict.get("StmF") || Name.get("Identity");
this.strf = dict.get("StrF") || identityName; this.strf = dict.get("StrF") || Name.get("Identity");
this.eff = dict.get("EFF") || this.stmf; this.eff = dict.get("EFF") || this.stmf;
} }
} }
@ -1823,14 +1813,14 @@ const CipherTransformFactory = (function CipherTransformFactoryClosure() {
createCipherTransform(num, gen) { createCipherTransform(num, gen) {
if (this.algorithm === 4 || this.algorithm === 5) { if (this.algorithm === 4 || this.algorithm === 5) {
return new CipherTransform( return new CipherTransform(
buildCipherConstructor( this.#buildCipherConstructor(
this.cf, this.cf,
this.strf, this.strf,
num, num,
gen, gen,
this.encryptionKey this.encryptionKey
), ),
buildCipherConstructor( this.#buildCipherConstructor(
this.cf, this.cf,
this.stmf, this.stmf,
num, num,
@ -1840,21 +1830,18 @@ const CipherTransformFactory = (function CipherTransformFactoryClosure() {
); );
} }
// algorithms 1 and 2 // algorithms 1 and 2
const key = buildObjectKey( const key = this.#buildObjectKey(
num, num,
gen, gen,
this.encryptionKey, this.encryptionKey,
/* isAes = */ false /* isAes = */ false
); );
const cipherConstructor = function buildCipherCipherConstructor() { const cipherConstructor = function () {
return new ARCFourCipher(key); return new ARCFourCipher(key);
}; };
return new CipherTransform(cipherConstructor, cipherConstructor); return new CipherTransform(cipherConstructor, cipherConstructor);
} }
} }
return CipherTransformFactory;
})();
export { export {
AES128Cipher, AES128Cipher,