Add support for modifying TCP and UDP socket buffer size in nacl_io

native_client_sdk/src/tests/nacl_io_socket_test/socket_test.cc

 // Copyright 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 <arpa/inet.h>
 #include <errno.h>
 #include <fcntl.h>
 #include <netinet/in.h>
 #include <pthread.h>
 #include <sys/types.h>
@@ skipping 493 matching lines @@
   ASSERT_GT(sock1_, -1);
   ASSERT_EQ(0, ki_setsockopt(sock1_, SOL_SOCKET, SO_REUSEADDR, &value, len));
 
   value = 0;
   len = sizeof(value);
   ASSERT_EQ(0, ki_getsockopt(sock1_, SOL_SOCKET, SO_REUSEADDR, &value, &len));
   ASSERT_EQ(1, value);
   ASSERT_EQ(sizeof(int), len);
 }
 
+TEST_F(SocketTest, Sockopt_BUFSIZE) {
+  int option = 1024*1024;
+  socklen_t len = sizeof(option);
+
+  sock1_ = ki_socket(AF_INET, SOCK_DGRAM, 0);
+  ASSERT_GT(sock1_, -1);
+  ASSERT_EQ(0, Bind(sock1_, LOCAL_HOST, PORT1));
+  ASSERT_EQ(0, ki_setsockopt(sock1_, SOL_SOCKET, SO_RCVBUF, &option, len))
+    << "failed with: " << strerror(errno);
+  ASSERT_EQ(0, ki_setsockopt(sock1_, SOL_SOCKET, SO_SNDBUF, &option, len))
+    << "failed with: " << strerror(errno);
+
+  sock2_ = socket(AF_INET, SOCK_STREAM, 0);

[CodeByThePound, 2014/06/02 19:38:18]

I think you meant to call ki_socket here, instead of socket?

+  ASSERT_GT(sock2_, -1);
+  ASSERT_EQ(0, Bind(sock2_, LOCAL_HOST, PORT2));
+
+  ASSERT_EQ(0, ki_setsockopt(sock2_, SOL_SOCKET, SO_RCVBUF, &option, len))
+    << "failed with: " << strerror(errno);
+  ASSERT_EQ(0, ki_setsockopt(sock2_, SOL_SOCKET, SO_SNDBUF, &option, len))
+    << "failed with: " << strerror(errno);
+}
+
 // The size of the data to send is deliberately chosen to be
 // larger than the TCP buffer in nacl_io.
 // TODO(sbc): use ioctl to discover the actual buffer size at
 // runtime.
 #define LARGE_SEND_BYTES (800 * 1024)
 TEST_F(SocketTestWithServer, LargeSend) {
   char* outbuf = (char*)malloc(LARGE_SEND_BYTES);
   char* inbuf = (char*)malloc(LARGE_SEND_BYTES);
   int bytes_sent = 0;
   int bytes_received = 0;
@@ skipping 282 matching lines @@
   while (remainder > 0) {
     int rtn = ki_recv(new_sock, buffer, remainder, 0);
     ASSERT_GT(rtn, 0);
     remainder -= rtn;
   }
 
   ASSERT_EQ(0, ki_close(new_sock));
 }
 
 #endif  // PROVIDES_SOCKET_API