From ce3d18412c9cbd8259f1dac84f83a039436adf91 Mon Sep 17 00:00:00 2001
From: isaacs <i@izs.me>
Date: Wed, 14 Aug 2013 22:14:49 -0700
Subject: [PATCH] http: Write hex/base64 chunks properly

This removes a dubious performance "optimization" where strings body
chunks were concatenated to one another (and to the headers) without any
regard for their encoding.
---
 lib/_http_outgoing.js              | 34 ++++++-------------
 test/simple/test-http-hex-write.js | 53 ++++++++++++++++++++++++++++++
 2 files changed, 63 insertions(+), 24 deletions(-)
 create mode 100644 test/simple/test-http-hex-write.js

diff --git a/lib/_http_outgoing.js b/lib/_http_outgoing.js
index 7cfa92c9c3..18ba517cfb 100644
--- a/lib/_http_outgoing.js
+++ b/lib/_http_outgoing.js
@@ -115,7 +115,9 @@ OutgoingMessage.prototype._send = function(data, encoding) {
   // the same packet. Future versions of Node are going to take care of
   // this at a lower level and in a more general way.
   if (!this._headerSent) {
-    if (util.isString(data)) {
+    if (util.isString(data) &&
+        encoding !== 'hex' &&
+        encoding !== 'base64') {
       data = this._header + data;
     } else {
       this.output.unshift(this._header);
@@ -162,25 +164,6 @@ OutgoingMessage.prototype._writeRaw = function(data, encoding) {
 
 
 OutgoingMessage.prototype._buffer = function(data, encoding) {
-  if (data.length === 0) return;
-
-  var length = this.output.length;
-
-  if (length === 0 || !util.isString(data)) {
-    this.output.push(data);
-    this.outputEncodings.push(encoding);
-    return false;
-  }
-
-  var lastEncoding = this.outputEncodings[length - 1];
-  var lastData = this.output[length - 1];
-
-  if ((encoding && lastEncoding === encoding) ||
-      (!encoding && data.constructor === lastData.constructor)) {
-    this.output[length - 1] = lastData + data;
-    return false;
-  }
-
   this.output.push(data);
   this.outputEncodings.push(encoding);
 
@@ -421,13 +404,16 @@ OutgoingMessage.prototype.write = function(chunk, encoding) {
       ret = this._send(chunk, encoding);
     } else {
       // buffer, or a non-toString-friendly encoding
-      len = chunk.length;
+      if (util.isString(chunk))
+        len = Buffer.byteLength(chunk, encoding);
+      else
+        len = chunk.length;
 
       if (this.connection)
         this.connection.cork();
-      this._send(len.toString(16));
+      this._send(len.toString(16), 'ascii');
       this._send(crlf_buf);
-      this._send(chunk);
+      this._send(chunk, encoding);
       ret = this._send(crlf_buf);
       if (this.connection)
         this.connection.uncork();
@@ -493,7 +479,7 @@ OutgoingMessage.prototype.end = function(data, encoding) {
   }
 
   if (this.chunkedEncoding) {
-    ret = this._send('0\r\n' + this._trailer + '\r\n'); // Last chunk.
+    ret = this._send('0\r\n' + this._trailer + '\r\n', 'ascii');
   } else {
     // Force a flush, HACK.
     ret = this._send('');
diff --git a/test/simple/test-http-hex-write.js b/test/simple/test-http-hex-write.js
new file mode 100644
index 0000000000..21a93df53e
--- /dev/null
+++ b/test/simple/test-http-hex-write.js
@@ -0,0 +1,53 @@
+// Copyright Joyent, Inc. and other Node contributors.
+//
+// Permission is hereby granted, free of charge, to any person obtaining a
+// copy of this software and associated documentation files (the
+// "Software"), to deal in the Software without restriction, including
+// without limitation the rights to use, copy, modify, merge, publish,
+// distribute, sublicense, and/or sell copies of the Software, and to permit
+// persons to whom the Software is furnished to do so, subject to the
+// following conditions:
+//
+// The above copyright notice and this permission notice shall be included
+// in all copies or substantial portions of the Software.
+//
+// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
+// OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+// MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN
+// NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM,
+// DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR
+// OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE
+// USE OR OTHER DEALINGS IN THE SOFTWARE.
+
+var common = require('../common');
+var assert = require('assert');
+
+var http = require('http');
+
+var expect = 'hex\nutf8\n';
+var data = '';
+var ended = false;
+
+process.on('exit', function() {
+  assert(ended);
+  assert.equal(data, expect);
+  console.log('ok');
+});
+
+http.createServer(function(q, s) {
+  s.setHeader('content-length', expect.length);
+  s.write('6865780a', 'hex');
+  s.write('utf8\n');
+  s.end();
+  this.close();
+}).listen(common.PORT, function() {
+  http.request({ port: common.PORT }).on('response', function(res) {
+    res.setEncoding('ascii');
+    res.on('data', function(c) {
+      data += c;
+    });
+    res.on('end', function() {
+      ended = true;
+    });
+  }).end();
+});