[NaCl SDK] nacl_io: implement set/getsockopt for TCP_NODELAY.

native_client_sdk/src/libraries/nacl_io/mount_node_tcp.cc

 // Copyright (c) 2013 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 
 #include "nacl_io/ossocket.h"
 #ifdef PROVIDES_SOCKET_API
 
 #include <assert.h>
 #include <errno.h>
@@ skipping 254 matching lines @@
 
     node->ConnectDone_Locked();
   }
 
  protected:
   ScopedEventEmitterTCP emitter_;
 };
 
 MountNodeTCP::MountNodeTCP(Mount* mount)
     : MountNodeSocket(mount),
-      emitter_(new EventEmitterTCP(kDefaultFifoSize, kDefaultFifoSize)) {
+      emitter_(new EventEmitterTCP(kDefaultFifoSize, kDefaultFifoSize)),
+      tcp_nodelay_(false) {
   emitter_->AttachStream(this);
 }
 
 MountNodeTCP::MountNodeTCP(Mount* mount, PP_Resource socket)
     : MountNodeSocket(mount, socket),
-      emitter_(new EventEmitterTCP(kDefaultFifoSize, kDefaultFifoSize)) {
+      emitter_(new EventEmitterTCP(kDefaultFifoSize, kDefaultFifoSize)),
+      tcp_nodelay_(false) {
   emitter_->AttachStream(this);
 }
 
 void MountNodeTCP::Destroy() {
   emitter_->DetachStream();
   MountNodeSocket::Destroy();
 }
 
 Error MountNodeTCP::Init(int open_flags) {
   Error err = MountNodeSocket::Init(open_flags);
@@ skipping 21 matching lines @@
 
 EventEmitter* MountNodeTCP::GetEventEmitter() {
   return emitter_.get();
 }
 
 void MountNodeTCP::SetError_Locked(int pp_error_num) {
   MountNodeSocket::SetError_Locked(pp_error_num);
   emitter_->SetError_Locked();
 }
 
+Error MountNodeTCP::GetSockOpt(int lvl,
+                               int optname,
+                               void* optval,
+                               socklen_t* len) {
+  if (lvl == IPPROTO_TCP && optname == TCP_NODELAY) {
+    int value = tcp_nodelay_;
+    socklen_t value_len = sizeof(value);
+    int copy_bytes = std::min(value_len, *len);
+    memcpy(optval, &value, copy_bytes);
+    *len = value_len;
+    return 0;
+  }
+
+  return MountNodeSocket::GetSockOpt(lvl, optname, optval, len);
+}
+
+
+Error MountNodeTCP::SetNoDelay() {

[binji, 2013/12/02 20:55:32]

should this be called SetNoDelay_Locked? Or does it work properly without
holding the lock?

[Sam Clegg, 2013/12/02 22:55:44]

Done.

+  if (!IsConnected())
+    return 0;
+
+  int32_t error = TCPInterface()->SetOption(socket_resource_,
+      PP_TCPSOCKET_OPTION_NO_DELAY,
+      PP_MakeBool(tcp_nodelay_ ? PP_TRUE : PP_FALSE),
+      PP_BlockUntilComplete());
+  return PPErrorToErrno(error);
+}
+
+Error MountNodeTCP::SetSockOpt(int lvl,
+                               int optname,
+                               const void* optval,
+                               socklen_t len) {
+  if (lvl == IPPROTO_TCP && optname == TCP_NODELAY) {
+    if (len < sizeof(int))
+      return EINVAL;
+    tcp_nodelay_ = *static_cast<const int*>(optval) ? true : false;

[binji, 2013/12/02 20:55:32]

nit: I'd prefer

*static_cast<const int*>(optval) != 0

here instead

[Sam Clegg, 2013/12/02 22:55:44]

Done.

+    return SetNoDelay();
+  }
+
+  return MountNodeSocket::SetSockOpt(lvl, optname, optval, len);
+}
+
 void MountNodeTCP::QueueAccept() {
   MountStream::Work* work = new TCPAcceptWork(mount_stream(), emitter_);
   mount_stream()->EnqueueWork(work);
 }
 
 void MountNodeTCP::QueueConnect() {
   MountStream::Work* work = new TCPConnectWork(mount_stream(), emitter_);
   mount_stream()->EnqueueWork(work);
 }
 
@@ skipping 124 matching lines @@
 
 void MountNodeTCP::ConnectDone_Locked() {
   local_addr_ = TCPInterface()->GetLocalAddress(socket_resource_);
 
   // Now that we are connected, we can start sending and receiving.
   ClearStreamFlags(SSF_CONNECTING | SSF_CAN_CONNECT);
   SetStreamFlags(SSF_CAN_SEND | SSF_CAN_RECV);
 
   emitter_->ConnectDone_Locked();
 
+  // The NODELAY option cannot be set in PPAPI before the socket
+  // is connected, but setsockopt() might have already set it.
+  SetNoDelay();
+
   // Begin the input pump
   QueueInput();
 }
 
 void MountNodeTCP::ConnectFailed_Locked() {
   mount_->ppapi()->ReleaseResource(remote_addr_);
   remote_addr_ = 0;
 }
 
 Error MountNodeTCP::Listen(int backlog) {
@@ skipping 36 matching lines @@
   if (emitter_->GetError_Locked())
     return EPIPE;
   *out_len = emitter_->WriteOut_Locked((char*)buf, len);
   return 0;
 }
 
 
 }  // namespace nacl_io
 
 #endif  // PROVIDES_SOCKET_API