Fix typed array output

foliojs · Feb 15, 2015 · f69afbe · f69afbe
1 parent f4fb95e
commit f69afbe
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 2 deletions.
diff --git a/index.js b/index.js
@@ -347,7 +347,14 @@ function tinf_uncompress(source, dest) {
 
   } while (!bfinal);
 
-  return d.dest.slice(0, d.destLen);
+  if (d.destLen < d.dest.length) {
+    if (typeof d.dest.slice === 'function')
+      return d.dest.slice(0, d.destLen);
+    else
+      return d.dest.subarray(0, d.destLen);
+  }
+
+  return d.dest;
 }
 
 /* -------------------- *

diff --git a/test/index.js b/test/index.js
@@ -60,9 +60,16 @@ describe('tiny-inflate', function() {
     assert.deepEqual(out, uncompressed);
   });
 
-  it('should handle uncompressed blocks', function() {
+  it('should handle fixed huffman blocks', function() {
     var out = new Buffer(uncompressed.length);
     inflate(fixed, out);
     assert.deepEqual(out, uncompressed);
   });
+
+  it('should handle typed arrays', function() {
+    var input = new Uint8Array(compressed);
+    var out = new Uint8Array(uncompressed.length);
+    inflate(input, out);
+    assert.deepEqual(out, new Uint8Array(uncompressed));
+  });
 });