From 4bc7025309bc5b5263833635dbcebabac9bb8d30 Mon Sep 17 00:00:00 2001
From: Robert Nagy <ronagy@icloud.com>
Date: Sun, 26 Apr 2020 11:16:43 +0200
Subject: [PATCH] stream: write should throw on unknown encoding

Validate encoding passed to write(chunk, encoding, cb) and
throw if it is invalid.

PR-URL: https://github.com/nodejs/node/pull/33075
Reviewed-By: James M Snell <jasnell@gmail.com>
Reviewed-By: Matteo Collina <matteo.collina@gmail.com>
---
 lib/_stream_writable.js                       |  2 ++
 .../test-stream-writable-write-error.js       | 23 +++++++++++++------
 2 files changed, 18 insertions(+), 7 deletions(-)

diff --git a/lib/_stream_writable.js b/lib/_stream_writable.js
index 317333a0b762c3..4e0f40b432a213 100644
--- a/lib/_stream_writable.js
+++ b/lib/_stream_writable.js
@@ -268,6 +268,8 @@ Writable.prototype.write = function(chunk, encoding, cb) {
   } else {
     if (!encoding)
       encoding = state.defaultEncoding;
+    else if (encoding !== 'buffer' && !Buffer.isEncoding(encoding))
+      throw new ERR_UNKNOWN_ENCODING(encoding);
     if (typeof cb !== 'function')
       cb = nop;
   }
diff --git a/test/parallel/test-stream-writable-write-error.js b/test/parallel/test-stream-writable-write-error.js
index eb9b1db06b772b..069e32e1be8e3e 100644
--- a/test/parallel/test-stream-writable-write-error.js
+++ b/test/parallel/test-stream-writable-write-error.js
@@ -4,17 +4,17 @@ const assert = require('assert');
 
 const { Writable } = require('stream');
 
-function expectError(w, arg, code, sync) {
+function expectError(w, args, code, sync) {
   if (sync) {
     if (code) {
-      assert.throws(() => w.write(arg), { code });
+      assert.throws(() => w.write(...args), { code });
     } else {
-      w.write(arg);
+      w.write(...args);
     }
   } else {
     let errorCalled = false;
     let ticked = false;
-    w.write(arg, common.mustCall((err) => {
+    w.write(...args, common.mustCall((err) => {
       assert.strictEqual(ticked, true);
       assert.strictEqual(errorCalled, false);
       assert.strictEqual(err.code, code);
@@ -34,7 +34,7 @@ function test(autoDestroy) {
       _write() {}
     });
     w.end();
-    expectError(w, 'asd', 'ERR_STREAM_WRITE_AFTER_END');
+    expectError(w, ['asd'], 'ERR_STREAM_WRITE_AFTER_END');
   }
 
   {
@@ -50,7 +50,7 @@ function test(autoDestroy) {
       autoDestroy,
       _write() {}
     });
-    expectError(w, null, 'ERR_STREAM_NULL_VALUES', true);
+    expectError(w, [null], 'ERR_STREAM_NULL_VALUES', true);
   }
 
   {
@@ -58,7 +58,16 @@ function test(autoDestroy) {
       autoDestroy,
       _write() {}
     });
-    expectError(w, {}, 'ERR_INVALID_ARG_TYPE', true);
+    expectError(w, [{}], 'ERR_INVALID_ARG_TYPE', true);
+  }
+
+  {
+    const w = new Writable({
+      decodeStrings: false,
+      autoDestroy,
+      _write() {}
+    });
+    expectError(w, ['asd', 'noencoding'], 'ERR_UNKNOWN_ENCODING', true);
   }
 }