stream_wrap: Add support to write binary strings

node::StringBytes::Write() has appropriate support to write strings with
'binary' encoding. So expose that API through StreamWrap and allow
inheriting classes to use it.

Signed-off-by: Trevor Norris <trev.norris@gmail.com>

5 files changed, 15 insertions, 0 deletions

diff --git a/src/pipe_wrap.cc b/src/pipe_wrap.cc
index 03156cc8f5..05472de5d9 100644
--- a/src/pipe_wrap.cc
+++ b/src/pipe_wrap.cc
@@ -104,6 +104,9 @@ void PipeWrap::Initialize(Handle<Object> target,
                             StreamWrap::WriteAsciiString);
   NODE_SET_PROTOTYPE_METHOD(t, "writeUtf8String", StreamWrap::WriteUtf8String);
   NODE_SET_PROTOTYPE_METHOD(t, "writeUcs2String", StreamWrap::WriteUcs2String);
+  NODE_SET_PROTOTYPE_METHOD(t,
+                            "writeBinaryString",
+                            StreamWrap::WriteBinaryString);
 
   NODE_SET_PROTOTYPE_METHOD(t, "bind", Bind);
   NODE_SET_PROTOTYPE_METHOD(t, "listen", Listen);
diff --git a/src/stream_wrap.cc b/src/stream_wrap.cc
index 5d6435101e..0f4da42ad0 100644
--- a/src/stream_wrap.cc
+++ b/src/stream_wrap.cc
@@ -499,6 +499,10 @@ void StreamWrap::WriteUcs2String(const FunctionCallbackInfo<Value>& args) {
   WriteStringImpl<UCS2>(args);
 }
 
+void StreamWrap::WriteBinaryString(const FunctionCallbackInfo<Value>& args) {
+  WriteStringImpl<BINARY>(args);
+}
+
 void StreamWrap::SetBlocking(const FunctionCallbackInfo<Value>& args) {
   Environment* env = Environment::GetCurrent(args.GetIsolate());
   HandleScope scope(env->isolate());
diff --git a/src/stream_wrap.h b/src/stream_wrap.h
index 3a7f6f8638..05b828ef8b 100644
--- a/src/stream_wrap.h
+++ b/src/stream_wrap.h
@@ -125,6 +125,8 @@ class StreamWrap : public HandleWrap {
   static void WriteAsciiString(const v8::FunctionCallbackInfo<v8::Value>& args);
   static void WriteUtf8String(const v8::FunctionCallbackInfo<v8::Value>& args);
   static void WriteUcs2String(const v8::FunctionCallbackInfo<v8::Value>& args);
+  static void WriteBinaryString(
+      const v8::FunctionCallbackInfo<v8::Value>& args);
 
   static void SetBlocking(const v8::FunctionCallbackInfo<v8::Value>& args);
 
diff --git a/src/tcp_wrap.cc b/src/tcp_wrap.cc
index 702dd4d805..09671d0095 100644
--- a/src/tcp_wrap.cc
+++ b/src/tcp_wrap.cc
@@ -111,6 +111,9 @@ void TCPWrap::Initialize(Handle<Object> target,
                             StreamWrap::WriteAsciiString);
   NODE_SET_PROTOTYPE_METHOD(t, "writeUtf8String", StreamWrap::WriteUtf8String);
   NODE_SET_PROTOTYPE_METHOD(t, "writeUcs2String", StreamWrap::WriteUcs2String);
+  NODE_SET_PROTOTYPE_METHOD(t,
+                            "writeBinaryString",
+                            StreamWrap::WriteBinaryString);
   NODE_SET_PROTOTYPE_METHOD(t, "writev", StreamWrap::Writev);
 
   NODE_SET_PROTOTYPE_METHOD(t, "open", Open);
diff --git a/src/tty_wrap.cc b/src/tty_wrap.cc
index dba545b5f4..34ee14dec7 100644
--- a/src/tty_wrap.cc
+++ b/src/tty_wrap.cc
@@ -78,6 +78,9 @@ void TTYWrap::Initialize(Handle<Object> target,
                             StreamWrap::WriteAsciiString);
   NODE_SET_PROTOTYPE_METHOD(t, "writeUtf8String", StreamWrap::WriteUtf8String);
   NODE_SET_PROTOTYPE_METHOD(t, "writeUcs2String", StreamWrap::WriteUcs2String);
+  NODE_SET_PROTOTYPE_METHOD(t,
+                            "writeBinaryString",
+                            StreamWrap::WriteBinaryString);
 
   NODE_SET_PROTOTYPE_METHOD(t, "getWindowSize", TTYWrap::GetWindowSize);
   NODE_SET_PROTOTYPE_METHOD(t, "setRawMode", SetRawMode);