xref: /llvm-project/lldb/unittests/Host/SocketAddressTest.cpp (revision b9c1b51e45b845debb76d8658edabca70ca56079)
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 }
17 
18 using namespace lldb_private;
19 
20 TEST_F(SocketAddressTest, Set) {
21   SocketAddress sa;
22   ASSERT_TRUE(sa.SetToLocalhost(AF_INET, 1138));
23   ASSERT_STREQ("127.0.0.1", sa.GetIPAddress().c_str());
24   ASSERT_EQ(1138, sa.GetPort());
25 
26   ASSERT_TRUE(sa.SetToAnyAddress(AF_INET, 0));
27   ASSERT_STREQ("0.0.0.0", sa.GetIPAddress().c_str());
28   ASSERT_EQ(0, sa.GetPort());
29 
30   ASSERT_TRUE(sa.SetToLocalhost(AF_INET6, 1139));
31   ASSERT_TRUE(sa.GetIPAddress() == "::1" ||
32               sa.GetIPAddress() == "0:0:0:0:0:0:0:1")
33       << "Address was: " << sa.GetIPAddress();
34   ASSERT_EQ(1139, sa.GetPort());
35 }
36 
37 #ifdef _WIN32
38 
39 // we need to test our inet_ntop implementation for Windows XP
40 const char *inet_ntop(int af, const void *src, char *dst, socklen_t size);
41 
42 TEST_F(SocketAddressTest, inet_ntop) {
43   const uint8_t address4[4] = {255, 0, 1, 100};
44   const uint8_t address6[16] = {0, 1, 2,  3,  4,  5,  6,   7,
45                                 8, 9, 10, 11, 12, 13, 255, 0};
46 
47   char buffer[INET6_ADDRSTRLEN];
48   memset(buffer, 'x', sizeof(buffer));
49   EXPECT_STREQ("1:203:405:607:809:a0b:c0d:ff00",
50                inet_ntop(AF_INET6, address6, buffer, sizeof(buffer)));
51   memset(buffer, 'x', sizeof(buffer));
52   EXPECT_STREQ("1:203:405:607:809:a0b:c0d:ff00",
53                inet_ntop(AF_INET6, address6, buffer, 31));
54   memset(buffer, 'x', sizeof(buffer));
55   EXPECT_STREQ(nullptr, inet_ntop(AF_INET6, address6, buffer, 0));
56   memset(buffer, 'x', sizeof(buffer));
57   EXPECT_STREQ(nullptr, inet_ntop(AF_INET6, address6, buffer, 30));
58 
59   memset(buffer, 'x', sizeof(buffer));
60   EXPECT_STREQ("255.0.1.100",
61                inet_ntop(AF_INET, address4, buffer, sizeof(buffer)));
62   memset(buffer, 'x', sizeof(buffer));
63   EXPECT_STREQ("255.0.1.100", inet_ntop(AF_INET, address4, buffer, 12));
64   memset(buffer, 'x', sizeof(buffer));
65   EXPECT_STREQ(nullptr, inet_ntop(AF_INET, address4, buffer, 0));
66   memset(buffer, 'x', sizeof(buffer));
67   EXPECT_STREQ(nullptr, inet_ntop(AF_INET, address4, buffer, 11));
68 }
69 
70 #endif
71