1/* 2 * libjingle 3 * Copyright 2006, Google Inc. 4 * 5 * Redistribution and use in source and binary forms, with or without 6 * modification, are permitted provided that the following conditions are met: 7 * 8 * 1. Redistributions of source code must retain the above copyright notice, 9 * this list of conditions and the following disclaimer. 10 * 2. Redistributions in binary form must reproduce the above copyright notice, 11 * this list of conditions and the following disclaimer in the documentation 12 * and/or other materials provided with the distribution. 13 * 3. The name of the author may not be used to endorse or promote products 14 * derived from this software without specific prior written permission. 15 * 16 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR IMPLIED 17 * WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF 18 * MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO 19 * EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, 20 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, 21 * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; 22 * OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, 23 * WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR 24 * OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF 25 * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 26 */ 27 28#include "talk/base/gunit.h" 29#include "talk/base/nethelpers.h" 30#include "talk/base/physicalsocketserver.h" 31#include "talk/base/testclient.h" 32#include "talk/base/testechoserver.h" 33#include "talk/base/thread.h" 34 35using namespace talk_base; 36 37void TestUdpInternal(const SocketAddress& loopback) { 38 Thread *main = Thread::Current(); 39 AsyncSocket* socket = main->socketserver() 40 ->CreateAsyncSocket(loopback.family(), SOCK_DGRAM); 41 socket->Bind(loopback); 42 43 TestClient client(new AsyncUDPSocket(socket)); 44 SocketAddress addr = client.address(), from; 45 EXPECT_EQ(3, client.SendTo("foo", 3, addr)); 46 EXPECT_TRUE(client.CheckNextPacket("foo", 3, &from)); 47 EXPECT_EQ(from, addr); 48 EXPECT_TRUE(client.CheckNoPacket()); 49} 50 51void TestTcpInternal(const SocketAddress& loopback) { 52 Thread *main = Thread::Current(); 53 TestEchoServer server(main, loopback); 54 55 AsyncSocket* socket = main->socketserver() 56 ->CreateAsyncSocket(loopback.family(), SOCK_STREAM); 57 AsyncTCPSocket* tcp_socket = AsyncTCPSocket::Create( 58 socket, loopback, server.address()); 59 ASSERT_TRUE(tcp_socket != NULL); 60 61 TestClient client(tcp_socket); 62 SocketAddress addr = client.address(), from; 63 EXPECT_TRUE(client.CheckConnected()); 64 EXPECT_EQ(3, client.Send("foo", 3)); 65 EXPECT_TRUE(client.CheckNextPacket("foo", 3, &from)); 66 EXPECT_EQ(from, server.address()); 67 EXPECT_TRUE(client.CheckNoPacket()); 68} 69 70// Tests whether the TestClient can send UDP to itself. 71TEST(TestClientTest, TestUdpIPv4) { 72 TestUdpInternal(SocketAddress("127.0.0.1", 0)); 73} 74 75TEST(TestClientTest, TestUdpIPv6) { 76 if (HasIPv6Enabled()) { 77 TestUdpInternal(SocketAddress("::1", 0)); 78 } else { 79 LOG(LS_INFO) << "Skipping IPv6 test."; 80 } 81} 82 83// Tests whether the TestClient can connect to a server and exchange data. 84TEST(TestClientTest, TestTcpIPv4) { 85 TestTcpInternal(SocketAddress("127.0.0.1", 0)); 86} 87 88TEST(TestClientTest, TestTcpIPv6) { 89 if (HasIPv6Enabled()) { 90 TestTcpInternal(SocketAddress("::1", 0)); 91 } else { 92 LOG(LS_INFO) << "Skipping IPv6 test."; 93 } 94} 95