1 //===-- SocketAddressTest.cpp -----------------------------------*- C++ -*-===//
2 //
3 //                     The LLVM Compiler Infrastructure
4 //
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
7 //
8 //===----------------------------------------------------------------------===//
9 
10 #include "gtest/gtest.h"
11 
12 #include "lldb/Host/SocketAddress.h"
13 
14 namespace {
15 class SocketAddressTest : public testing::Test {
16 public:
17   static void SetUpTestCase() {
18 #ifdef _MSC_VER
19     WSADATA data;
20     ASSERT_EQ(0, WSAStartup(MAKEWORD(2, 2), &data));
21 #endif
22   }
23   static void TearDownTestCase() {
24 #ifdef _MSC_VER
25     ASSERT_EQ(0, WSACleanup());
26 #endif
27   }
28 };
29 } // namespace
30 
31 using namespace lldb_private;
32 
33 TEST_F(SocketAddressTest, Set) {
34   SocketAddress sa;
35   ASSERT_TRUE(sa.SetToLocalhost(AF_INET, 1138));
36   ASSERT_STREQ("127.0.0.1", sa.GetIPAddress().c_str());
37   ASSERT_EQ(1138, sa.GetPort());
38 
39   ASSERT_TRUE(sa.SetToAnyAddress(AF_INET, 0));
40   ASSERT_STREQ("0.0.0.0", sa.GetIPAddress().c_str());
41   ASSERT_EQ(0, sa.GetPort());
42 
43   ASSERT_TRUE(sa.SetToLocalhost(AF_INET6, 1139));
44   ASSERT_TRUE(sa.GetIPAddress() == "::1" ||
45               sa.GetIPAddress() == "0:0:0:0:0:0:0:1")
46       << "Address was: " << sa.GetIPAddress();
47   ASSERT_EQ(1139, sa.GetPort());
48 }
49 
50 TEST_F(SocketAddressTest, GetAddressInfo) {
51   auto addr = SocketAddress::GetAddressInfo("127.0.0.1", nullptr, AF_UNSPEC,
52                                             SOCK_STREAM, IPPROTO_TCP);
53   ASSERT_EQ(1u, addr.size());
54   EXPECT_EQ(AF_INET, addr[0].GetFamily());
55   EXPECT_EQ("127.0.0.1", addr[0].GetIPAddress());
56 }
57 
58 #ifdef _WIN32
59 
60 // we need to test our inet_ntop implementation for Windows XP
61 const char *inet_ntop(int af, const void *src, char *dst, socklen_t size);
62 
63 TEST_F(SocketAddressTest, inet_ntop) {
64   const uint8_t address4[4] = {255, 0, 1, 100};
65   const uint8_t address6[16] = {0, 1, 2,  3,  4,  5,  6,   7,
66                                 8, 9, 10, 11, 12, 13, 255, 0};
67 
68   char buffer[INET6_ADDRSTRLEN];
69   memset(buffer, 'x', sizeof(buffer));
70   EXPECT_STREQ("1:203:405:607:809:a0b:c0d:ff00",
71                inet_ntop(AF_INET6, address6, buffer, sizeof(buffer)));
72   memset(buffer, 'x', sizeof(buffer));
73   EXPECT_STREQ("1:203:405:607:809:a0b:c0d:ff00",
74                inet_ntop(AF_INET6, address6, buffer, 31));
75   memset(buffer, 'x', sizeof(buffer));
76   EXPECT_STREQ(nullptr, inet_ntop(AF_INET6, address6, buffer, 0));
77   memset(buffer, 'x', sizeof(buffer));
78   EXPECT_STREQ(nullptr, inet_ntop(AF_INET6, address6, buffer, 30));
79 
80   memset(buffer, 'x', sizeof(buffer));
81   EXPECT_STREQ("255.0.1.100",
82                inet_ntop(AF_INET, address4, buffer, sizeof(buffer)));
83   memset(buffer, 'x', sizeof(buffer));
84   EXPECT_STREQ("255.0.1.100", inet_ntop(AF_INET, address4, buffer, 12));
85   memset(buffer, 'x', sizeof(buffer));
86   EXPECT_STREQ(nullptr, inet_ntop(AF_INET, address4, buffer, 0));
87   memset(buffer, 'x', sizeof(buffer));
88   EXPECT_STREQ(nullptr, inet_ntop(AF_INET, address4, buffer, 11));
89 }
90 
91 #endif
92