Index: test/cctest/test-socket.cc |
diff --git a/test/cctest/test-socket.cc b/test/cctest/test-socket.cc |
new file mode 100644 |
index 0000000000000000000000000000000000000000..47d8b178312646e7c6e8c81c552ef5ac605ae824 |
--- /dev/null |
+++ b/test/cctest/test-socket.cc |
@@ -0,0 +1,176 @@ |
+// Copyright 2009 the V8 project authors. All rights reserved. |
+// Redistribution and use in source and binary forms, with or without |
+// modification, are permitted provided that the following conditions are |
+// met: |
+// |
+// * Redistributions of source code must retain the above copyright |
+// notice, this list of conditions and the following disclaimer. |
+// * Redistributions in binary form must reproduce the above |
+// copyright notice, this list of conditions and the following |
+// disclaimer in the documentation and/or other materials provided |
+// with the distribution. |
+// * Neither the name of Google Inc. nor the names of its |
+// contributors may be used to endorse or promote products derived |
+// from this software without specific prior written permission. |
+// |
+// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS |
+// "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT |
+// LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR |
+// A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT |
+// OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, |
+// SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT |
+// LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, |
+// DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY |
+// THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT |
+// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE |
+// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
+ |
+#include "v8.h" |
+#include "platform.h" |
+#include "platform/socket.h" |
+#include "cctest.h" |
+ |
+ |
+using namespace ::v8::internal; |
+ |
+ |
+class SocketListenerThread : public Thread { |
+ public: |
+ SocketListenerThread(int port, int data_size) |
+ : Thread("SocketListenerThread"), |
+ port_(port), |
+ data_size_(data_size), |
+ server_(NULL), |
+ client_(NULL), |
+ listening_(0) { |
+ data_ = new char[data_size_]; |
+ } |
+ ~SocketListenerThread() { |
+ // Close both sockets. |
+ delete client_; |
+ delete server_; |
+ delete[] data_; |
+ } |
+ |
+ void Run(); |
+ void WaitForListening() { listening_.Wait(); } |
+ char* data() { return data_; } |
+ |
+ private: |
+ int port_; |
+ char* data_; |
+ int data_size_; |
+ Socket* server_; // Server socket used for bind/accept. |
+ Socket* client_; // Single client connection used by the test. |
+ Semaphore listening_; // Signalled when the server socket is in listen mode. |
+}; |
+ |
+ |
+void SocketListenerThread::Run() { |
+ bool ok; |
+ |
+ // Create the server socket and bind it to the requested port. |
+ server_ = new Socket; |
+ server_->SetReuseAddress(true); |
+ CHECK(server_ != NULL); |
+ ok = server_->Bind(port_); |
+ CHECK(ok); |
+ |
+ // Listen for new connections. |
+ ok = server_->Listen(1); |
+ CHECK(ok); |
+ listening_.Signal(); |
+ |
+ // Accept a connection. |
+ client_ = server_->Accept(); |
+ CHECK(client_ != NULL); |
+ |
+ // Read the expected niumber of bytes of data. |
+ int bytes_read = 0; |
+ while (bytes_read < data_size_) { |
+ bytes_read += client_->Receive(data_ + bytes_read, data_size_ - bytes_read); |
+ } |
+} |
+ |
+ |
+static bool SendAll(Socket* socket, const char* data, int len) { |
+ int sent_len = 0; |
+ while (sent_len < len) { |
+ int status = socket->Send(data, len); |
+ if (status <= 0) { |
+ return false; |
+ } |
+ sent_len += status; |
+ } |
+ return true; |
+} |
+ |
+ |
+static void SendAndReceive(int port, char *data, int len) { |
+ static const char* kLocalhost = "localhost"; |
+ |
+ bool ok; |
+ |
+ // Make a string with the port number. |
+ const int kPortBuferLen = 6; |
+ char port_str[kPortBuferLen]; |
+ OS::SNPrintF(Vector<char>(port_str, kPortBuferLen), "%d", port); |
+ |
+ // Create a socket listener. |
+ SocketListenerThread* listener = new SocketListenerThread(port, len); |
+ listener->Start(); |
+ listener->WaitForListening(); |
+ |
+ // Connect and write some data. |
+ Socket* client = new Socket; |
+ CHECK(client != NULL); |
+ ok = client->Connect(kLocalhost, port_str); |
+ CHECK(ok); |
+ |
+ // Send all the data. |
+ ok = SendAll(client, data, len); |
+ CHECK(ok); |
+ |
+ // Wait until data is received. |
+ listener->Join(); |
+ |
+ // Check that data received is the same as data send. |
+ for (int i = 0; i < len; i++) { |
+ CHECK(data[i] == listener->data()[i]); |
+ } |
+ |
+ // Close the client before the listener to avoid TIME_WAIT issues. |
+ client->Shutdown(); |
+ delete client; |
+ delete listener; |
+} |
+ |
+ |
+TEST(Socket) { |
+ // Make sure this port is not used by other tests to allow tests to run in |
+ // parallel. |
+ static const int kPort = 5859 + FlagDependentPortOffset(); |
+ |
+ // Send and receive some data. |
+ static const int kBufferSizeSmall = 20; |
+ char small_data[kBufferSizeSmall + 1] = "1234567890abcdefghij"; |
+ SendAndReceive(kPort, small_data, kBufferSizeSmall); |
+ |
+ // Send and receive some more data. |
+ static const int kBufferSizeMedium = 10000; |
+ char* medium_data = new char[kBufferSizeMedium]; |
+ for (int i = 0; i < kBufferSizeMedium; i++) { |
+ medium_data[i] = i % 256; |
+ } |
+ SendAndReceive(kPort, medium_data, kBufferSizeMedium); |
+ delete[] medium_data; |
+ |
+ // Send and receive even more data. |
+ static const int kBufferSizeLarge = 1000000; |
+ char* large_data = new char[kBufferSizeLarge]; |
+ for (int i = 0; i < kBufferSizeLarge; i++) { |
+ large_data[i] = i % 256; |
+ } |
+ SendAndReceive(kPort, large_data, kBufferSizeLarge); |
+ delete[] large_data; |
+} |