From 061f2075cf81017cdb40de80533ba18746743c94 Mon Sep 17 00:00:00 2001
From: isaacs <i@izs.me>
Date: Thu, 11 Oct 2012 15:53:11 -0700
Subject: string_decoder: Add 'end' method, do base64 properly

---
 lib/fs.js             |  5 +++++
 lib/http.js           | 10 ++++++++++
 lib/net.js            |  5 +++++
 lib/string_decoder.js | 26 ++++++++++++++++++++++++++
 lib/tls.js            |  5 +++++
 5 files changed, 51 insertions(+)

(limited to 'lib')

diff --git a/lib/fs.js b/lib/fs.js
index e8a9aa4f727..8041ab620df 100644
--- a/lib/fs.js
+++ b/lib/fs.js
@@ -1402,6 +1402,11 @@ ReadStream.prototype._read = function() {
     }
 
     if (bytesRead === 0) {
+      if (this._decoder) {
+        var ret = this._decoder.end();
+        if (ret)
+          this.emit('data', ret);
+      }
       self.emit('end');
       self.destroy();
       return;
diff --git a/lib/http.js b/lib/http.js
index 4aa8f5cb40f..3c8b60944c5 100644
--- a/lib/http.js
+++ b/lib/http.js
@@ -363,6 +363,11 @@ IncomingMessage.prototype._emitData = function(d) {
 
 IncomingMessage.prototype._emitEnd = function() {
   if (!this._endEmitted) {
+    if (this._decoder) {
+      var ret = this._decoder.end();
+      if (ret)
+        this.emit('data', ret);
+    }
     this.emit('end');
   }
 
@@ -1859,6 +1864,11 @@ Client.prototype.request = function(method, path, headers) {
   // but it will get removed when we remove this legacy interface.
   c.on('socket', function(s) {
     s.on('end', function() {
+      if (self._decoder) {
+        var ret = self._decoder.end();
+        if (ret)
+          self.emit('data', ret);
+      }
       self.emit('end');
     });
   });
diff --git a/lib/net.js b/lib/net.js
index 0decfe963c8..46d7d0b173a 100644
--- a/lib/net.js
+++ b/lib/net.js
@@ -414,6 +414,11 @@ function onread(buffer, offset, length) {
     if (!self.writable) self._destroy();
 
     if (!self.allowHalfOpen) self.end();
+    if (self._decoder) {
+      var ret = self._decoder.end();
+      if (ret)
+        self.emit('data', ret);
+    }
     if (self._events && self._events['end']) self.emit('end');
     if (self.onend) self.onend();
   } else {
diff --git a/lib/string_decoder.js b/lib/string_decoder.js
index 879e5906474..31d4b247021 100644
--- a/lib/string_decoder.js
+++ b/lib/string_decoder.js
@@ -32,6 +32,11 @@ var StringDecoder = exports.StringDecoder = function(encoding) {
       this.surrogateSize = 2;
       this.detectIncompleteChar = utf16DetectIncompleteChar;
       break;
+    case 'base64':
+      // Base-64 stores 3 bytes in 4 chars, and pads the remainder.
+      this.surrogateSize = 3;
+      this.detectIncompleteChar = base64DetectIncompleteChar;
+      break;
     default:
       this.write = passThroughWrite;
       return;
@@ -145,6 +150,21 @@ StringDecoder.prototype.detectIncompleteChar = function(buffer) {
   return i;
 };
 
+StringDecoder.prototype.end = function(buffer) {
+  var res = '';
+  if (buffer && buffer.length)
+    res = this.write(buffer);
+
+  if (this.charReceived) {
+    var cr = this.charReceived;
+    var buf = this.charBuffer;
+    var enc = this.encoding;
+    res += buf.slice(0, cr).toString(enc);
+  }
+
+  return res;
+};
+
 function passThroughWrite(buffer) {
   return buffer.toString(this.encoding);
 }
@@ -154,3 +174,9 @@ function utf16DetectIncompleteChar(buffer) {
   this.charLength = incomplete ? 2 : 0;
   return incomplete;
 }
+
+function base64DetectIncompleteChar(buffer) {
+  var incomplete = this.charReceived = buffer.length % 3;
+  this.charLength = incomplete ? 3 : 0;
+  return incomplete;
+}
diff --git a/lib/tls.js b/lib/tls.js
index 1fe4f789993..ed6c5a7d44e 100644
--- a/lib/tls.js
+++ b/lib/tls.js
@@ -967,6 +967,11 @@ SecurePair.prototype.destroy = function() {
     self.cleartext.writable = self.cleartext.readable = false;
 
     process.nextTick(function() {
+      if (self.cleartext._decoder) {
+        var ret = self.cleartext._decoder.end();
+        if (ret)
+          self.cleartext.emit('data', ret);
+      }
       self.cleartext.emit('end');
       self.encrypted.emit('close');
       self.cleartext.emit('close');
-- 
cgit v1.2.3