Merge pull request #10974 from Snuffleupagus/refactor-get-fingerprint

Simplify the `PDFDocument.fingerprint` method slightly
This commit is contained in:
Tim van der Meij 2019-07-15 22:29:37 +02:00 committed by GitHub
commit bf60fe88d0
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
4 changed files with 43 additions and 26 deletions

View File

@ -218,6 +218,12 @@ class ChunkedStream {
} }
getByteRange(begin, end) { getByteRange(begin, end) {
if (begin < 0) {
begin = 0;
}
if (end > this.end) {
end = this.end;
}
this.ensureRange(begin, end); this.ensureRange(begin, end);
return this.bytes.subarray(begin, end); return this.bytes.subarray(begin, end);
} }

View File

@ -598,20 +598,16 @@ class PDFDocument {
idArray[0] !== EMPTY_FINGERPRINT) { idArray[0] !== EMPTY_FINGERPRINT) {
hash = stringToBytes(idArray[0]); hash = stringToBytes(idArray[0]);
} else { } else {
if (this.stream.ensureRange) { hash = calculateMD5(this.stream.getByteRange(0, FINGERPRINT_FIRST_BYTES),
this.stream.ensureRange(0, 0, FINGERPRINT_FIRST_BYTES);
Math.min(FINGERPRINT_FIRST_BYTES, this.stream.end));
}
hash = calculateMD5(this.stream.bytes.subarray(0,
FINGERPRINT_FIRST_BYTES), 0, FINGERPRINT_FIRST_BYTES);
} }
let fingerprint = ''; const fingerprintBuf = [];
for (let i = 0, ii = hash.length; i < ii; i++) { for (let i = 0, ii = hash.length; i < ii; i++) {
const hex = hash[i].toString(16); const hex = hash[i].toString(16);
fingerprint += (hex.length === 1 ? '0' + hex : hex); fingerprintBuf.push(hex.padStart(2, '0'));
} }
return shadow(this, 'fingerprint', fingerprint); return shadow(this, 'fingerprint', fingerprintBuf.join(''));
} }
_getLinearizationPage(pageIndex) { _getLinearizationPage(pageIndex) {

View File

@ -19,7 +19,9 @@
* license. * license.
*/ */
import { FormatError, isSpace, stringToBytes } from '../shared/util'; import {
FormatError, isSpace, stringToBytes, unreachable
} from '../shared/util';
import { isDict } from './primitives'; import { isDict } from './primitives';
var Stream = (function StreamClosure() { var Stream = (function StreamClosure() {
@ -92,6 +94,17 @@ var Stream = (function StreamClosure() {
this.pos -= bytes.length; this.pos -= bytes.length;
return bytes; return bytes;
}, },
getByteRange(begin, end) {
if (begin < 0) {
begin = 0;
}
if (end > this.end) {
end = this.end;
}
return this.bytes.subarray(begin, end);
},
skip: function Stream_skip(n) { skip: function Stream_skip(n) {
if (!n) { if (!n) {
n = 1; n = 1;
@ -236,6 +249,11 @@ var DecodeStream = (function DecodeStreamClosure() {
} }
return new Stream(this.buffer, start, length, dict); return new Stream(this.buffer, start, length, dict);
}, },
getByteRange(begin, end) {
unreachable('Should not call DecodeStream.getByteRange');
},
skip: function DecodeStream_skip(n) { skip: function DecodeStream_skip(n) {
if (!n) { if (!n) {
n = 1; n = 1;

View File

@ -413,9 +413,7 @@ describe('api', function() {
expect(doc.numPages).toEqual(3); expect(doc.numPages).toEqual(3);
}); });
it('gets fingerprint', function() { it('gets fingerprint', function() {
var fingerprint = doc.fingerprint; expect(doc.fingerprint).toEqual('ea8b35919d6279a369e835bde778611b');
expect(typeof fingerprint).toEqual('string');
expect(fingerprint.length > 0).toEqual(true);
}); });
it('gets page', function(done) { it('gets page', function(done) {
var promise = doc.getPage(1); var promise = doc.getPage(1);
@ -934,23 +932,22 @@ describe('api', function() {
}); });
it('checks that fingerprints are unique', function(done) { it('checks that fingerprints are unique', function(done) {
var loadingTask1 = getDocument(buildGetDocumentParams('issue4436r.pdf')); const loadingTask1 = getDocument(
buildGetDocumentParams('issue4436r.pdf'));
const loadingTask2 = getDocument(buildGetDocumentParams('issue4575.pdf'));
var loadingTask2 = getDocument(buildGetDocumentParams('issue4575.pdf')); Promise.all([
loadingTask1.promise,
var promises = [loadingTask1.promise, loadingTask2.promise
loadingTask2.promise]; ]).then(function(data) {
Promise.all(promises).then(function (data) { const fingerprint1 = data[0].fingerprint;
var fingerprint1 = data[0].fingerprint; const fingerprint2 = data[1].fingerprint;
expect(typeof fingerprint1).toEqual('string');
expect(fingerprint1.length > 0).toEqual(true);
var fingerprint2 = data[1].fingerprint;
expect(typeof fingerprint2).toEqual('string');
expect(fingerprint2.length > 0).toEqual(true);
expect(fingerprint1).not.toEqual(fingerprint2); expect(fingerprint1).not.toEqual(fingerprint2);
expect(fingerprint1).toEqual('2f695a83d6e7553c24fc08b7ac69712d');
expect(fingerprint2).toEqual('04c7126b34a46b6d4d6e7a1eff7edcb6');
Promise.all([ Promise.all([
loadingTask1.destroy(), loadingTask1.destroy(),
loadingTask2.destroy() loadingTask2.destroy()