summaryrefslogtreecommitdiffstats
path: root/application/basilisk/extensions/pdfjs/content/PdfStreamConverter.jsm
diff options
context:
space:
mode:
authorMatt A. Tobin <email@mattatobin.com>2018-02-02 03:32:58 -0500
committerMatt A. Tobin <email@mattatobin.com>2018-02-02 03:32:58 -0500
commite72ef92b5bdc43cd2584198e2e54e951b70299e8 (patch)
tree01ceb4a897c33eca9e7ccf2bc3aefbe530169fe5 /application/basilisk/extensions/pdfjs/content/PdfStreamConverter.jsm
parent0d19b77d3eaa5b8d837bf52c19759e68e42a1c4c (diff)
downloadUXP-e72ef92b5bdc43cd2584198e2e54e951b70299e8.tar
UXP-e72ef92b5bdc43cd2584198e2e54e951b70299e8.tar.gz
UXP-e72ef92b5bdc43cd2584198e2e54e951b70299e8.tar.lz
UXP-e72ef92b5bdc43cd2584198e2e54e951b70299e8.tar.xz
UXP-e72ef92b5bdc43cd2584198e2e54e951b70299e8.zip
Add Basilisk
Diffstat (limited to 'application/basilisk/extensions/pdfjs/content/PdfStreamConverter.jsm')
-rw-r--r--application/basilisk/extensions/pdfjs/content/PdfStreamConverter.jsm1035
1 files changed, 1035 insertions, 0 deletions
diff --git a/application/basilisk/extensions/pdfjs/content/PdfStreamConverter.jsm b/application/basilisk/extensions/pdfjs/content/PdfStreamConverter.jsm
new file mode 100644
index 000000000..7e5e2fbe6
--- /dev/null
+++ b/application/basilisk/extensions/pdfjs/content/PdfStreamConverter.jsm
@@ -0,0 +1,1035 @@
+/* Copyright 2012 Mozilla Foundation
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+/* globals Components, Services, XPCOMUtils, NetUtil, PrivateBrowsingUtils,
+ dump, NetworkManager, PdfJsTelemetry, PdfjsContentUtils */
+
+'use strict';
+
+var EXPORTED_SYMBOLS = ['PdfStreamConverter'];
+
+const Cc = Components.classes;
+const Ci = Components.interfaces;
+const Cr = Components.results;
+const Cu = Components.utils;
+// True only if this is the version of pdf.js that is included with firefox.
+const MOZ_CENTRAL = JSON.parse('true');
+const PDFJS_EVENT_ID = 'pdf.js.message';
+const PDF_CONTENT_TYPE = 'application/pdf';
+const PREF_PREFIX = 'pdfjs';
+const PDF_VIEWER_WEB_PAGE = 'resource://pdf.js/web/viewer.html';
+const MAX_NUMBER_OF_PREFS = 50;
+const MAX_STRING_PREF_LENGTH = 128;
+
+Cu.import('resource://gre/modules/XPCOMUtils.jsm');
+Cu.import('resource://gre/modules/Services.jsm');
+Cu.import('resource://gre/modules/NetUtil.jsm');
+
+XPCOMUtils.defineLazyModuleGetter(this, 'NetworkManager',
+ 'resource://pdf.js/PdfJsNetwork.jsm');
+
+XPCOMUtils.defineLazyModuleGetter(this, 'PrivateBrowsingUtils',
+ 'resource://gre/modules/PrivateBrowsingUtils.jsm');
+
+XPCOMUtils.defineLazyModuleGetter(this, 'PdfJsTelemetry',
+ 'resource://pdf.js/PdfJsTelemetry.jsm');
+
+XPCOMUtils.defineLazyModuleGetter(this, 'PdfjsContentUtils',
+ 'resource://pdf.js/PdfjsContentUtils.jsm');
+
+var Svc = {};
+XPCOMUtils.defineLazyServiceGetter(Svc, 'mime',
+ '@mozilla.org/mime;1',
+ 'nsIMIMEService');
+
+function getContainingBrowser(domWindow) {
+ return domWindow.QueryInterface(Ci.nsIInterfaceRequestor)
+ .getInterface(Ci.nsIWebNavigation)
+ .QueryInterface(Ci.nsIDocShell)
+ .chromeEventHandler;
+}
+
+function getFindBar(domWindow) {
+ if (PdfjsContentUtils.isRemote) {
+ throw new Error('FindBar is not accessible from the content process.');
+ }
+ try {
+ var browser = getContainingBrowser(domWindow);
+ var tabbrowser = browser.getTabBrowser();
+ var tab = tabbrowser.getTabForBrowser(browser);
+ return tabbrowser.getFindBar(tab);
+ } catch (e) {
+ // Suppress errors for PDF files opened in the bookmark sidebar, see
+ // https://bugzilla.mozilla.org/show_bug.cgi?id=1248959.
+ return null;
+ }
+}
+
+function getBoolPref(pref, def) {
+ try {
+ return Services.prefs.getBoolPref(pref);
+ } catch (ex) {
+ return def;
+ }
+}
+
+function getIntPref(pref, def) {
+ try {
+ return Services.prefs.getIntPref(pref);
+ } catch (ex) {
+ return def;
+ }
+}
+
+function getStringPref(pref, def) {
+ try {
+ return Services.prefs.getComplexValue(pref, Ci.nsISupportsString).data;
+ } catch (ex) {
+ return def;
+ }
+}
+
+function log(aMsg) {
+ if (!getBoolPref(PREF_PREFIX + '.pdfBugEnabled', false)) {
+ return;
+ }
+ var msg = 'PdfStreamConverter.js: ' + (aMsg.join ? aMsg.join('') : aMsg);
+ Services.console.logStringMessage(msg);
+ dump(msg + '\n');
+}
+
+function getDOMWindow(aChannel) {
+ var requestor = aChannel.notificationCallbacks ?
+ aChannel.notificationCallbacks :
+ aChannel.loadGroup.notificationCallbacks;
+ var win = requestor.getInterface(Components.interfaces.nsIDOMWindow);
+ return win;
+}
+
+function getLocalizedStrings(path) {
+ var stringBundle = Cc['@mozilla.org/intl/stringbundle;1'].
+ getService(Ci.nsIStringBundleService).
+ createBundle('chrome://pdf.js/locale/' + path);
+
+ var map = {};
+ var enumerator = stringBundle.getSimpleEnumeration();
+ while (enumerator.hasMoreElements()) {
+ var string = enumerator.getNext().QueryInterface(Ci.nsIPropertyElement);
+ var key = string.key, property = 'textContent';
+ var i = key.lastIndexOf('.');
+ if (i >= 0) {
+ property = key.substring(i + 1);
+ key = key.substring(0, i);
+ }
+ if (!(key in map)) {
+ map[key] = {};
+ }
+ map[key][property] = string.value;
+ }
+ return map;
+}
+function getLocalizedString(strings, id, property) {
+ property = property || 'textContent';
+ if (id in strings) {
+ return strings[id][property];
+ }
+ return id;
+}
+
+// PDF data storage
+function PdfDataListener(length) {
+ this.length = length; // less than 0, if length is unknown
+ this.buffer = null;
+ this.loaded = 0;
+}
+
+PdfDataListener.prototype = {
+ append: function PdfDataListener_append(chunk) {
+ // In most of the cases we will pass data as we receive it, but at the
+ // beginning of the loading we may accumulate some data.
+ if (!this.buffer) {
+ this.buffer = new Uint8Array(chunk);
+ } else {
+ var buffer = this.buffer;
+ var newBuffer = new Uint8Array(buffer.length + chunk.length);
+ newBuffer.set(buffer);
+ newBuffer.set(chunk, buffer.length);
+ this.buffer = newBuffer;
+ }
+ this.loaded += chunk.length;
+ if (this.length >= 0 && this.length < this.loaded) {
+ this.length = -1; // reset the length, server is giving incorrect one
+ }
+ this.onprogress(this.loaded, this.length >= 0 ? this.length : void 0);
+ },
+ readData: function PdfDataListener_readData() {
+ var result = this.buffer;
+ this.buffer = null;
+ return result;
+ },
+ finish: function PdfDataListener_finish() {
+ this.isDataReady = true;
+ if (this.oncompleteCallback) {
+ this.oncompleteCallback(this.readData());
+ }
+ },
+ error: function PdfDataListener_error(errorCode) {
+ this.errorCode = errorCode;
+ if (this.oncompleteCallback) {
+ this.oncompleteCallback(null, errorCode);
+ }
+ },
+ onprogress: function() {},
+ get oncomplete() {
+ return this.oncompleteCallback;
+ },
+ set oncomplete(value) {
+ this.oncompleteCallback = value;
+ if (this.isDataReady) {
+ value(this.readData());
+ }
+ if (this.errorCode) {
+ value(null, this.errorCode);
+ }
+ }
+};
+
+// All the priviledged actions.
+function ChromeActions(domWindow, contentDispositionFilename) {
+ this.domWindow = domWindow;
+ this.contentDispositionFilename = contentDispositionFilename;
+ this.telemetryState = {
+ documentInfo: false,
+ firstPageInfo: false,
+ streamTypesUsed: [],
+ fontTypesUsed: [],
+ startAt: Date.now()
+ };
+}
+
+ChromeActions.prototype = {
+ isInPrivateBrowsing: function() {
+ return PrivateBrowsingUtils.isContentWindowPrivate(this.domWindow);
+ },
+ download: function(data, sendResponse) {
+ var self = this;
+ var originalUrl = data.originalUrl;
+ var blobUrl = data.blobUrl || originalUrl;
+ // The data may not be downloaded so we need just retry getting the pdf with
+ // the original url.
+ var originalUri = NetUtil.newURI(originalUrl);
+ var filename = data.filename;
+ if (typeof filename !== 'string' ||
+ (!/\.pdf$/i.test(filename) && !data.isAttachment)) {
+ filename = 'document.pdf';
+ }
+ var blobUri = NetUtil.newURI(blobUrl);
+ var extHelperAppSvc =
+ Cc['@mozilla.org/uriloader/external-helper-app-service;1'].
+ getService(Ci.nsIExternalHelperAppService);
+
+ var docIsPrivate = this.isInPrivateBrowsing();
+ var netChannel = NetUtil.newChannel({
+ uri: blobUri,
+ loadUsingSystemPrincipal: true,
+ });
+ if ('nsIPrivateBrowsingChannel' in Ci &&
+ netChannel instanceof Ci.nsIPrivateBrowsingChannel) {
+ netChannel.setPrivate(docIsPrivate);
+ }
+ NetUtil.asyncFetch(netChannel, function(aInputStream, aResult) {
+ if (!Components.isSuccessCode(aResult)) {
+ if (sendResponse) {
+ sendResponse(true);
+ }
+ return;
+ }
+ // Create a nsIInputStreamChannel so we can set the url on the channel
+ // so the filename will be correct.
+ var channel = Cc['@mozilla.org/network/input-stream-channel;1'].
+ createInstance(Ci.nsIInputStreamChannel);
+ channel.QueryInterface(Ci.nsIChannel);
+ try {
+ // contentDisposition/contentDispositionFilename is readonly before FF18
+ channel.contentDisposition = Ci.nsIChannel.DISPOSITION_ATTACHMENT;
+ if (self.contentDispositionFilename && !data.isAttachment) {
+ channel.contentDispositionFilename = self.contentDispositionFilename;
+ } else {
+ channel.contentDispositionFilename = filename;
+ }
+ } catch (e) {}
+ channel.setURI(originalUri);
+ channel.loadInfo = netChannel.loadInfo;
+ channel.contentStream = aInputStream;
+ if ('nsIPrivateBrowsingChannel' in Ci &&
+ channel instanceof Ci.nsIPrivateBrowsingChannel) {
+ channel.setPrivate(docIsPrivate);
+ }
+
+ var listener = {
+ extListener: null,
+ onStartRequest: function(aRequest, aContext) {
+ var loadContext = self.domWindow
+ .QueryInterface(Ci.nsIInterfaceRequestor)
+ .getInterface(Ci.nsIWebNavigation)
+ .QueryInterface(Ci.nsILoadContext);
+ this.extListener = extHelperAppSvc.doContent(
+ (data.isAttachment ? 'application/octet-stream' :
+ 'application/pdf'),
+ aRequest, loadContext, false);
+ this.extListener.onStartRequest(aRequest, aContext);
+ },
+ onStopRequest: function(aRequest, aContext, aStatusCode) {
+ if (this.extListener) {
+ this.extListener.onStopRequest(aRequest, aContext, aStatusCode);
+ }
+ // Notify the content code we're done downloading.
+ if (sendResponse) {
+ sendResponse(false);
+ }
+ },
+ onDataAvailable: function(aRequest, aContext, aInputStream, aOffset,
+ aCount) {
+ this.extListener.onDataAvailable(aRequest, aContext, aInputStream,
+ aOffset, aCount);
+ }
+ };
+
+ channel.asyncOpen2(listener);
+ });
+ },
+ getLocale: function() {
+ return getStringPref('general.useragent.locale', 'en-US');
+ },
+ getStrings: function(data) {
+ try {
+ // Lazy initialization of localizedStrings
+ if (!('localizedStrings' in this)) {
+ this.localizedStrings = getLocalizedStrings('viewer.properties');
+ }
+ var result = this.localizedStrings[data];
+ return JSON.stringify(result || null);
+ } catch (e) {
+ log('Unable to retrieve localized strings: ' + e);
+ return 'null';
+ }
+ },
+ supportsIntegratedFind: function() {
+ // Integrated find is only supported when we're not in a frame
+ if (this.domWindow.frameElement !== null) {
+ return false;
+ }
+
+ // ... and we are in a child process
+ if (PdfjsContentUtils.isRemote) {
+ return true;
+ }
+
+ // ... or when the new find events code exists.
+ var findBar = getFindBar(this.domWindow);
+ return !!findBar && ('updateControlState' in findBar);
+ },
+ supportsDocumentFonts: function() {
+ var prefBrowser = getIntPref('browser.display.use_document_fonts', 1);
+ var prefGfx = getBoolPref('gfx.downloadable_fonts.enabled', true);
+ return (!!prefBrowser && prefGfx);
+ },
+ supportsDocumentColors: function() {
+ return getIntPref('browser.display.document_color_use', 0) !== 2;
+ },
+ supportedMouseWheelZoomModifierKeys: function() {
+ return {
+ ctrlKey: getIntPref('mousewheel.with_control.action', 3) === 3,
+ metaKey: getIntPref('mousewheel.with_meta.action', 1) === 3,
+ };
+ },
+ reportTelemetry: function (data) {
+ var probeInfo = JSON.parse(data);
+ switch (probeInfo.type) {
+ case 'documentInfo':
+ if (!this.telemetryState.documentInfo) {
+ PdfJsTelemetry.onDocumentVersion(probeInfo.version | 0);
+ PdfJsTelemetry.onDocumentGenerator(probeInfo.generator | 0);
+ if (probeInfo.formType) {
+ PdfJsTelemetry.onForm(probeInfo.formType === 'acroform');
+ }
+ this.telemetryState.documentInfo = true;
+ }
+ break;
+ case 'pageInfo':
+ if (!this.telemetryState.firstPageInfo) {
+ var duration = Date.now() - this.telemetryState.startAt;
+ PdfJsTelemetry.onTimeToView(duration);
+ this.telemetryState.firstPageInfo = true;
+ }
+ break;
+ case 'documentStats':
+ // documentStats can be called several times for one documents.
+ // if stream/font types are reported, trying not to submit the same
+ // enumeration value multiple times.
+ var documentStats = probeInfo.stats;
+ if (!documentStats || typeof documentStats !== 'object') {
+ break;
+ }
+ var i, streamTypes = documentStats.streamTypes;
+ if (Array.isArray(streamTypes)) {
+ var STREAM_TYPE_ID_LIMIT = 20;
+ for (i = 0; i < STREAM_TYPE_ID_LIMIT; i++) {
+ if (streamTypes[i] &&
+ !this.telemetryState.streamTypesUsed[i]) {
+ PdfJsTelemetry.onStreamType(i);
+ this.telemetryState.streamTypesUsed[i] = true;
+ }
+ }
+ }
+ var fontTypes = documentStats.fontTypes;
+ if (Array.isArray(fontTypes)) {
+ var FONT_TYPE_ID_LIMIT = 20;
+ for (i = 0; i < FONT_TYPE_ID_LIMIT; i++) {
+ if (fontTypes[i] &&
+ !this.telemetryState.fontTypesUsed[i]) {
+ PdfJsTelemetry.onFontType(i);
+ this.telemetryState.fontTypesUsed[i] = true;
+ }
+ }
+ }
+ break;
+ case 'print':
+ PdfJsTelemetry.onPrint();
+ break;
+ }
+ },
+ fallback: function(args, sendResponse) {
+ var featureId = args.featureId;
+ var url = args.url;
+
+ var self = this;
+ var domWindow = this.domWindow;
+ var strings = getLocalizedStrings('chrome.properties');
+ var message;
+ if (featureId === 'forms') {
+ message = getLocalizedString(strings, 'unsupported_feature_forms');
+ } else {
+ message = getLocalizedString(strings, 'unsupported_feature');
+ }
+ PdfJsTelemetry.onFallback();
+ PdfjsContentUtils.displayWarning(domWindow, message,
+ getLocalizedString(strings, 'open_with_different_viewer'),
+ getLocalizedString(strings, 'open_with_different_viewer', 'accessKey'));
+
+ let winmm = domWindow.QueryInterface(Ci.nsIInterfaceRequestor)
+ .getInterface(Ci.nsIDocShell)
+ .QueryInterface(Ci.nsIInterfaceRequestor)
+ .getInterface(Ci.nsIContentFrameMessageManager);
+
+ winmm.addMessageListener('PDFJS:Child:fallbackDownload',
+ function fallbackDownload(msg) {
+ let data = msg.data;
+ sendResponse(data.download);
+
+ winmm.removeMessageListener('PDFJS:Child:fallbackDownload',
+ fallbackDownload);
+ });
+ },
+ updateFindControlState: function(data) {
+ if (!this.supportsIntegratedFind()) {
+ return;
+ }
+ // Verify what we're sending to the findbar.
+ var result = data.result;
+ var findPrevious = data.findPrevious;
+ var findPreviousType = typeof findPrevious;
+ if ((typeof result !== 'number' || result < 0 || result > 3) ||
+ (findPreviousType !== 'undefined' && findPreviousType !== 'boolean')) {
+ return;
+ }
+
+ var winmm = this.domWindow.QueryInterface(Ci.nsIInterfaceRequestor)
+ .getInterface(Ci.nsIDocShell)
+ .QueryInterface(Ci.nsIInterfaceRequestor)
+ .getInterface(Ci.nsIContentFrameMessageManager);
+
+ winmm.sendAsyncMessage('PDFJS:Parent:updateControlState', data);
+ },
+ setPreferences: function(prefs, sendResponse) {
+ var defaultBranch = Services.prefs.getDefaultBranch(PREF_PREFIX + '.');
+ var numberOfPrefs = 0;
+ var prefValue, prefName;
+ for (var key in prefs) {
+ if (++numberOfPrefs > MAX_NUMBER_OF_PREFS) {
+ log('setPreferences - Exceeded the maximum number of preferences ' +
+ 'that is allowed to be set at once.');
+ break;
+ } else if (!defaultBranch.getPrefType(key)) {
+ continue;
+ }
+ prefValue = prefs[key];
+ prefName = (PREF_PREFIX + '.' + key);
+ switch (typeof prefValue) {
+ case 'boolean':
+ PdfjsContentUtils.setBoolPref(prefName, prefValue);
+ break;
+ case 'number':
+ PdfjsContentUtils.setIntPref(prefName, prefValue);
+ break;
+ case 'string':
+ if (prefValue.length > MAX_STRING_PREF_LENGTH) {
+ log('setPreferences - Exceeded the maximum allowed length ' +
+ 'for a string preference.');
+ } else {
+ PdfjsContentUtils.setStringPref(prefName, prefValue);
+ }
+ break;
+ }
+ }
+ if (sendResponse) {
+ sendResponse(true);
+ }
+ },
+ getPreferences: function(prefs, sendResponse) {
+ var defaultBranch = Services.prefs.getDefaultBranch(PREF_PREFIX + '.');
+ var currentPrefs = {}, numberOfPrefs = 0;
+ var prefValue, prefName;
+ for (var key in prefs) {
+ if (++numberOfPrefs > MAX_NUMBER_OF_PREFS) {
+ log('getPreferences - Exceeded the maximum number of preferences ' +
+ 'that is allowed to be fetched at once.');
+ break;
+ } else if (!defaultBranch.getPrefType(key)) {
+ continue;
+ }
+ prefValue = prefs[key];
+ prefName = (PREF_PREFIX + '.' + key);
+ switch (typeof prefValue) {
+ case 'boolean':
+ currentPrefs[key] = getBoolPref(prefName, prefValue);
+ break;
+ case 'number':
+ currentPrefs[key] = getIntPref(prefName, prefValue);
+ break;
+ case 'string':
+ currentPrefs[key] = getStringPref(prefName, prefValue);
+ break;
+ }
+ }
+ if (sendResponse) {
+ sendResponse(JSON.stringify(currentPrefs));
+ } else {
+ return JSON.stringify(currentPrefs);
+ }
+ }
+};
+
+var RangedChromeActions = (function RangedChromeActionsClosure() {
+ /**
+ * This is for range requests
+ */
+ function RangedChromeActions(
+ domWindow, contentDispositionFilename, originalRequest,
+ rangeEnabled, streamingEnabled, dataListener) {
+
+ ChromeActions.call(this, domWindow, contentDispositionFilename);
+ this.dataListener = dataListener;
+ this.originalRequest = originalRequest;
+ this.rangeEnabled = rangeEnabled;
+ this.streamingEnabled = streamingEnabled;
+
+ this.pdfUrl = originalRequest.URI.spec;
+ this.contentLength = originalRequest.contentLength;
+
+ // Pass all the headers from the original request through
+ var httpHeaderVisitor = {
+ headers: {},
+ visitHeader: function(aHeader, aValue) {
+ if (aHeader === 'Range') {
+ // When loading the PDF from cache, firefox seems to set the Range
+ // request header to fetch only the unfetched portions of the file
+ // (e.g. 'Range: bytes=1024-'). However, we want to set this header
+ // manually to fetch the PDF in chunks.
+ return;
+ }
+ this.headers[aHeader] = aValue;
+ }
+ };
+ if (originalRequest.visitRequestHeaders) {
+ originalRequest.visitRequestHeaders(httpHeaderVisitor);
+ }
+
+ var self = this;
+ var xhr_onreadystatechange = function xhr_onreadystatechange() {
+ if (this.readyState === 1) { // LOADING
+ var netChannel = this.channel;
+ if ('nsIPrivateBrowsingChannel' in Ci &&
+ netChannel instanceof Ci.nsIPrivateBrowsingChannel) {
+ var docIsPrivate = self.isInPrivateBrowsing();
+ netChannel.setPrivate(docIsPrivate);
+ }
+ }
+ };
+ var getXhr = function getXhr() {
+ const XMLHttpRequest = Components.Constructor(
+ '@mozilla.org/xmlextras/xmlhttprequest;1');
+ var xhr = new XMLHttpRequest();
+ xhr.addEventListener('readystatechange', xhr_onreadystatechange);
+ return xhr;
+ };
+
+ this.networkManager = new NetworkManager(this.pdfUrl, {
+ httpHeaders: httpHeaderVisitor.headers,
+ getXhr: getXhr
+ });
+
+ // If we are in range request mode, this means we manually issued xhr
+ // requests, which we need to abort when we leave the page
+ domWindow.addEventListener('unload', function unload(e) {
+ domWindow.removeEventListener(e.type, unload);
+ self.abortLoading();
+ });
+ }
+
+ RangedChromeActions.prototype = Object.create(ChromeActions.prototype);
+ var proto = RangedChromeActions.prototype;
+ proto.constructor = RangedChromeActions;
+
+ proto.initPassiveLoading = function RangedChromeActions_initPassiveLoading() {
+ var self = this;
+ var data;
+ if (!this.streamingEnabled) {
+ this.originalRequest.cancel(Cr.NS_BINDING_ABORTED);
+ this.originalRequest = null;
+ data = this.dataListener.readData();
+ this.dataListener = null;
+ } else {
+ data = this.dataListener.readData();
+
+ this.dataListener.onprogress = function (loaded, total) {
+ self.domWindow.postMessage({
+ pdfjsLoadAction: 'progressiveRead',
+ loaded: loaded,
+ total: total,
+ chunk: self.dataListener.readData()
+ }, '*');
+ };
+ this.dataListener.oncomplete = function () {
+ self.dataListener = null;
+ };
+ }
+
+ this.domWindow.postMessage({
+ pdfjsLoadAction: 'supportsRangedLoading',
+ rangeEnabled: this.rangeEnabled,
+ streamingEnabled: this.streamingEnabled,
+ pdfUrl: this.pdfUrl,
+ length: this.contentLength,
+ data: data
+ }, '*');
+
+ return true;
+ };
+
+ proto.requestDataRange = function RangedChromeActions_requestDataRange(args) {
+ if (!this.rangeEnabled) {
+ return;
+ }
+
+ var begin = args.begin;
+ var end = args.end;
+ var domWindow = this.domWindow;
+ // TODO(mack): Support error handler. We're not currently not handling
+ // errors from chrome code for non-range requests, so this doesn't
+ // seem high-pri
+ this.networkManager.requestRange(begin, end, {
+ onDone: function RangedChromeActions_onDone(args) {
+ domWindow.postMessage({
+ pdfjsLoadAction: 'range',
+ begin: args.begin,
+ chunk: args.chunk
+ }, '*');
+ },
+ onProgress: function RangedChromeActions_onProgress(evt) {
+ domWindow.postMessage({
+ pdfjsLoadAction: 'rangeProgress',
+ loaded: evt.loaded,
+ }, '*');
+ }
+ });
+ };
+
+ proto.abortLoading = function RangedChromeActions_abortLoading() {
+ this.networkManager.abortAllRequests();
+ if (this.originalRequest) {
+ this.originalRequest.cancel(Cr.NS_BINDING_ABORTED);
+ this.originalRequest = null;
+ }
+ this.dataListener = null;
+ };
+
+ return RangedChromeActions;
+})();
+
+var StandardChromeActions = (function StandardChromeActionsClosure() {
+
+ /**
+ * This is for a single network stream
+ */
+ function StandardChromeActions(domWindow, contentDispositionFilename,
+ originalRequest, dataListener) {
+
+ ChromeActions.call(this, domWindow, contentDispositionFilename);
+ this.originalRequest = originalRequest;
+ this.dataListener = dataListener;
+ }
+
+ StandardChromeActions.prototype = Object.create(ChromeActions.prototype);
+ var proto = StandardChromeActions.prototype;
+ proto.constructor = StandardChromeActions;
+
+ proto.initPassiveLoading =
+ function StandardChromeActions_initPassiveLoading() {
+
+ if (!this.dataListener) {
+ return false;
+ }
+
+ var self = this;
+
+ this.dataListener.onprogress = function ChromeActions_dataListenerProgress(
+ loaded, total) {
+ self.domWindow.postMessage({
+ pdfjsLoadAction: 'progress',
+ loaded: loaded,
+ total: total
+ }, '*');
+ };
+
+ this.dataListener.oncomplete =
+ function StandardChromeActions_dataListenerComplete(data, errorCode) {
+ self.domWindow.postMessage({
+ pdfjsLoadAction: 'complete',
+ data: data,
+ errorCode: errorCode
+ }, '*');
+
+ self.dataListener = null;
+ self.originalRequest = null;
+ };
+
+ return true;
+ };
+
+ proto.abortLoading = function StandardChromeActions_abortLoading() {
+ if (this.originalRequest) {
+ this.originalRequest.cancel(Cr.NS_BINDING_ABORTED);
+ this.originalRequest = null;
+ }
+ this.dataListener = null;
+ };
+
+ return StandardChromeActions;
+})();
+
+// Event listener to trigger chrome privileged code.
+function RequestListener(actions) {
+ this.actions = actions;
+}
+// Receive an event and synchronously or asynchronously responds.
+RequestListener.prototype.receive = function(event) {
+ var message = event.target;
+ var doc = message.ownerDocument;
+ var action = event.detail.action;
+ var data = event.detail.data;
+ var sync = event.detail.sync;
+ var actions = this.actions;
+ if (!(action in actions)) {
+ log('Unknown action: ' + action);
+ return;
+ }
+ var response;
+ if (sync) {
+ response = actions[action].call(this.actions, data);
+ event.detail.response = Cu.cloneInto(response, doc.defaultView);
+ } else {
+ if (!event.detail.responseExpected) {
+ doc.documentElement.removeChild(message);
+ response = null;
+ } else {
+ response = function sendResponse(response) {
+ try {
+ var listener = doc.createEvent('CustomEvent');
+ let detail = Cu.cloneInto({ response: response }, doc.defaultView);
+ listener.initCustomEvent('pdf.js.response', true, false, detail);
+ return message.dispatchEvent(listener);
+ } catch (e) {
+ // doc is no longer accessible because the requestor is already
+ // gone. unloaded content cannot receive the response anyway.
+ return false;
+ }
+ };
+ }
+ actions[action].call(this.actions, data, response);
+ }
+};
+
+// Forwards events from the eventElement to the contentWindow only if the
+// content window matches the currently selected browser window.
+function FindEventManager(contentWindow) {
+ this.contentWindow = contentWindow;
+ this.winmm = contentWindow.QueryInterface(Ci.nsIInterfaceRequestor)
+ .getInterface(Ci.nsIDocShell)
+ .QueryInterface(Ci.nsIInterfaceRequestor)
+ .getInterface(Ci.nsIContentFrameMessageManager);
+}
+
+FindEventManager.prototype.bind = function() {
+ var unload = function(e) {
+ this.unbind();
+ this.contentWindow.removeEventListener(e.type, unload);
+ }.bind(this);
+ this.contentWindow.addEventListener('unload', unload);
+
+ // We cannot directly attach listeners to for the find events
+ // since the FindBar is in the parent process. Instead we're
+ // asking the PdfjsChromeUtils to do it for us and forward
+ // all the find events to us.
+ this.winmm.sendAsyncMessage('PDFJS:Parent:addEventListener');
+ this.winmm.addMessageListener('PDFJS:Child:handleEvent', this);
+};
+
+FindEventManager.prototype.receiveMessage = function(msg) {
+ var detail = msg.data.detail;
+ var type = msg.data.type;
+ var contentWindow = this.contentWindow;
+
+ detail = Cu.cloneInto(detail, contentWindow);
+ var forward = contentWindow.document.createEvent('CustomEvent');
+ forward.initCustomEvent(type, true, true, detail);
+ contentWindow.dispatchEvent(forward);
+};
+
+FindEventManager.prototype.unbind = function() {
+ this.winmm.sendAsyncMessage('PDFJS:Parent:removeEventListener');
+};
+
+function PdfStreamConverter() {
+}
+
+PdfStreamConverter.prototype = {
+
+ // properties required for XPCOM registration:
+ classID: Components.ID('{d0c5195d-e798-49d4-b1d3-9324328b2291}'),
+ classDescription: 'pdf.js Component',
+ contractID: '@mozilla.org/streamconv;1?from=application/pdf&to=*/*',
+
+ classID2: Components.ID('{d0c5195d-e798-49d4-b1d3-9324328b2292}'),
+ contractID2: '@mozilla.org/streamconv;1?from=application/pdf&to=text/html',
+
+ QueryInterface: XPCOMUtils.generateQI([
+ Ci.nsISupports,
+ Ci.nsIStreamConverter,
+ Ci.nsIStreamListener,
+ Ci.nsIRequestObserver
+ ]),
+
+ /*
+ * This component works as such:
+ * 1. asyncConvertData stores the listener
+ * 2. onStartRequest creates a new channel, streams the viewer
+ * 3. If range requests are supported:
+ * 3.1. Leave the request open until the viewer is ready to switch to
+ * range requests.
+ *
+ * If range rquests are not supported:
+ * 3.1. Read the stream as it's loaded in onDataAvailable to send
+ * to the viewer
+ *
+ * The convert function just returns the stream, it's just the synchronous
+ * version of asyncConvertData.
+ */
+
+ // nsIStreamConverter::convert
+ convert: function(aFromStream, aFromType, aToType, aCtxt) {
+ throw Cr.NS_ERROR_NOT_IMPLEMENTED;
+ },
+
+ // nsIStreamConverter::asyncConvertData
+ asyncConvertData: function(aFromType, aToType, aListener, aCtxt) {
+ // Store the listener passed to us
+ this.listener = aListener;
+ },
+
+ // nsIStreamListener::onDataAvailable
+ onDataAvailable: function(aRequest, aContext, aInputStream, aOffset, aCount) {
+ if (!this.dataListener) {
+ return;
+ }
+
+ var binaryStream = this.binaryStream;
+ binaryStream.setInputStream(aInputStream);
+ var chunk = binaryStream.readByteArray(aCount);
+ this.dataListener.append(chunk);
+ },
+
+ // nsIRequestObserver::onStartRequest
+ onStartRequest: function(aRequest, aContext) {
+ // Setup the request so we can use it below.
+ var isHttpRequest = false;
+ try {
+ aRequest.QueryInterface(Ci.nsIHttpChannel);
+ isHttpRequest = true;
+ } catch (e) {}
+
+ var rangeRequest = false;
+ var streamRequest = false;
+ if (isHttpRequest) {
+ var contentEncoding = 'identity';
+ try {
+ contentEncoding = aRequest.getResponseHeader('Content-Encoding');
+ } catch (e) {}
+
+ var acceptRanges;
+ try {
+ acceptRanges = aRequest.getResponseHeader('Accept-Ranges');
+ } catch (e) {}
+
+ var hash = aRequest.URI.ref;
+ var isPDFBugEnabled = getBoolPref(PREF_PREFIX + '.pdfBugEnabled', false);
+ rangeRequest = contentEncoding === 'identity' &&
+ acceptRanges === 'bytes' &&
+ aRequest.contentLength >= 0 &&
+ !getBoolPref(PREF_PREFIX + '.disableRange', false) &&
+ (!isPDFBugEnabled ||
+ hash.toLowerCase().indexOf('disablerange=true') < 0);
+ streamRequest = contentEncoding === 'identity' &&
+ aRequest.contentLength >= 0 &&
+ !getBoolPref(PREF_PREFIX + '.disableStream', false) &&
+ (!isPDFBugEnabled ||
+ hash.toLowerCase().indexOf('disablestream=true') < 0);
+ }
+
+ aRequest.QueryInterface(Ci.nsIChannel);
+
+ aRequest.QueryInterface(Ci.nsIWritablePropertyBag);
+
+ var contentDispositionFilename;
+ try {
+ contentDispositionFilename = aRequest.contentDispositionFilename;
+ } catch (e) {}
+
+ // Change the content type so we don't get stuck in a loop.
+ aRequest.setProperty('contentType', aRequest.contentType);
+ aRequest.contentType = 'text/html';
+ if (isHttpRequest) {
+ // We trust PDF viewer, using no CSP
+ aRequest.setResponseHeader('Content-Security-Policy', '', false);
+ aRequest.setResponseHeader('Content-Security-Policy-Report-Only', '',
+ false);
+ // The viewer does not need to handle HTTP Refresh header.
+ aRequest.setResponseHeader('Refresh', '', false);
+ }
+
+ PdfJsTelemetry.onViewerIsUsed();
+ PdfJsTelemetry.onDocumentSize(aRequest.contentLength);
+
+ // Creating storage for PDF data
+ var contentLength = aRequest.contentLength;
+ this.dataListener = new PdfDataListener(contentLength);
+ this.binaryStream = Cc['@mozilla.org/binaryinputstream;1']
+ .createInstance(Ci.nsIBinaryInputStream);
+
+ // Create a new channel that is viewer loaded as a resource.
+ var channel = NetUtil.newChannel({
+ uri: PDF_VIEWER_WEB_PAGE,
+ loadUsingSystemPrincipal: true,
+ });
+
+ var listener = this.listener;
+ var dataListener = this.dataListener;
+ // Proxy all the request observer calls, when it gets to onStopRequest
+ // we can get the dom window. We also intentionally pass on the original
+ // request(aRequest) below so we don't overwrite the original channel and
+ // trigger an assertion.
+ var proxy = {
+ onStartRequest: function(request, context) {
+ listener.onStartRequest(aRequest, aContext);
+ },
+ onDataAvailable: function(request, context, inputStream, offset, count) {
+ listener.onDataAvailable(aRequest, aContext, inputStream,
+ offset, count);
+ },
+ onStopRequest: function(request, context, statusCode) {
+ // We get the DOM window here instead of before the request since it
+ // may have changed during a redirect.
+ var domWindow = getDOMWindow(channel);
+ var actions;
+ if (rangeRequest || streamRequest) {
+ actions = new RangedChromeActions(
+ domWindow, contentDispositionFilename, aRequest,
+ rangeRequest, streamRequest, dataListener);
+ } else {
+ actions = new StandardChromeActions(
+ domWindow, contentDispositionFilename, aRequest, dataListener);
+ }
+ var requestListener = new RequestListener(actions);
+ domWindow.addEventListener(PDFJS_EVENT_ID, function(event) {
+ requestListener.receive(event);
+ }, false, true);
+ if (actions.supportsIntegratedFind()) {
+ var findEventManager = new FindEventManager(domWindow);
+ findEventManager.bind();
+ }
+ listener.onStopRequest(aRequest, aContext, statusCode);
+
+ if (domWindow.frameElement) {
+ var isObjectEmbed = domWindow.frameElement.tagName !== 'IFRAME' ||
+ domWindow.frameElement.className === 'previewPluginContentFrame';
+ PdfJsTelemetry.onEmbed(isObjectEmbed);
+ }
+ }
+ };
+
+ // Keep the URL the same so the browser sees it as the same.
+ channel.originalURI = aRequest.URI;
+ channel.loadGroup = aRequest.loadGroup;
+ channel.loadInfo.originAttributes = aRequest.loadInfo.originAttributes;
+
+ // We can use the resource principal when data is fetched by the chrome,
+ // e.g. useful for NoScript. Make make sure we reuse the origin attributes
+ // from the request channel to keep isolation consistent.
+ var ssm = Cc['@mozilla.org/scriptsecuritymanager;1']
+ .getService(Ci.nsIScriptSecurityManager);
+ var uri = NetUtil.newURI(PDF_VIEWER_WEB_PAGE);
+ var resourcePrincipal =
+ ssm.createCodebasePrincipal(uri, aRequest.loadInfo.originAttributes);
+ aRequest.owner = resourcePrincipal;
+
+ channel.asyncOpen2(proxy);
+ },
+
+ // nsIRequestObserver::onStopRequest
+ onStopRequest: function(aRequest, aContext, aStatusCode) {
+ if (!this.dataListener) {
+ // Do nothing
+ return;
+ }
+
+ if (Components.isSuccessCode(aStatusCode)) {
+ this.dataListener.finish();
+ } else {
+ this.dataListener.error(aStatusCode);
+ }
+ delete this.dataListener;
+ delete this.binaryStream;
+ }
+};
+