Fixing the length given by HTTP headers

yurydelendik · yurydelendik · commit aa00fe44c441 · 2012-08-22T13:20:50.000-05:00
diff --git a/extensions/firefox/components/PdfStreamConverter.js b/extensions/firefox/components/PdfStreamConverter.js
@@ -132,28 +132,28 @@ function PdfDataListener(length) {
 
 PdfDataListener.prototype = {
   set: function PdfDataListener_set(chunk, offset) {
-    if (this.length < 0) {
-      var willBeLoaded = this.loaded + chunk.length;
+    var willBeLoaded = this.loaded + chunk.length;
+    if (this.length >= 0 && this.length < willBeLoaded) {
+      this.length = -1; // reset the length, server is giving incorrect one
+    }
+    if (this.length < 0 && this.data.length < willBeLoaded) {
       // data length is unknown and new chunk will not fit in the existing
       // buffer, resizing the buffer by doubling the its last length
-      if (this.data.length < willBeLoaded) {
-        var newLength = this.data.length;
-        for (; newLength < willBeLoaded; newLength *= 2) {}
-        var newData = new Uint8Array(newLength);
-        newData.set(this.data);
-        this.data = newData;
-      }
-      this.data.set(chunk, this.loaded);
-      this.loaded = willBeLoaded;
-    } else {
-      this.data.set(chunk, offset);
-      this.loaded = offset + chunk.length;
-      this.onprogress(this.loaded, this.length);
+      var newLength = this.data.length;
+      for (; newLength < willBeLoaded; newLength *= 2) {}
+      var newData = new Uint8Array(newLength);
+      newData.set(this.data);
+      this.data = newData;
     }
+    this.data.set(chunk, this.loaded);
+    this.loaded = willBeLoaded;
+    if (this.length >= 0)
+      this.onprogress(this.loaded, this.length);
   },
   getData: function PdfDataListener_getData() {
-    var data = this.length >= 0 ? this.data :
-                                  this.data.subarray(0, this.loaded);
+    var data = this.data;
+    if (this.loaded != data.length)
+      data = data.subarray(0, this.loaded);
     delete this.data; // releasing temporary storage
     return data;
   },
