| //===-- SocketAddressTest.cpp ---------------------------------------------===// |
| // |
| // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. |
| // See https://llvm.org/LICENSE.txt for license information. |
| // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception |
| // |
| //===----------------------------------------------------------------------===// |
| |
| #include "lldb/Host/SocketAddress.h" |
| #include "TestingSupport/SubsystemRAII.h" |
| #include "lldb/Host/Socket.h" |
| #include "llvm/Testing/Support/Error.h" |
| |
| #include "gtest/gtest.h" |
| |
| using namespace lldb_private; |
| |
| namespace { |
| class SocketAddressTest : public testing::Test { |
| public: |
| SubsystemRAII<Socket> subsystems; |
| }; |
| } // namespace |
| |
| TEST_F(SocketAddressTest, Set) { |
| SocketAddress sa; |
| ASSERT_TRUE(sa.SetToLocalhost(AF_INET, 1138)); |
| ASSERT_STREQ("127.0.0.1", sa.GetIPAddress().c_str()); |
| ASSERT_EQ(1138, sa.GetPort()); |
| |
| ASSERT_TRUE(sa.SetToAnyAddress(AF_INET, 0)); |
| ASSERT_STREQ("0.0.0.0", sa.GetIPAddress().c_str()); |
| ASSERT_EQ(0, sa.GetPort()); |
| |
| ASSERT_TRUE(sa.SetToLocalhost(AF_INET6, 1139)); |
| ASSERT_TRUE(sa.GetIPAddress() == "::1" || |
| sa.GetIPAddress() == "0:0:0:0:0:0:0:1") |
| << "Address was: " << sa.GetIPAddress(); |
| ASSERT_EQ(1139, sa.GetPort()); |
| } |
| |
| TEST_F(SocketAddressTest, GetAddressInfo) { |
| auto addr = SocketAddress::GetAddressInfo("127.0.0.1", nullptr, AF_UNSPEC, |
| SOCK_STREAM, IPPROTO_TCP); |
| ASSERT_EQ(1u, addr.size()); |
| EXPECT_EQ(AF_INET, addr[0].GetFamily()); |
| EXPECT_EQ("127.0.0.1", addr[0].GetIPAddress()); |
| } |
| |
| #ifdef _WIN32 |
| |
| // we need to test our inet_ntop implementation for Windows XP |
| const char *inet_ntop(int af, const void *src, char *dst, socklen_t size); |
| |
| TEST_F(SocketAddressTest, inet_ntop) { |
| const uint8_t address4[4] = {255, 0, 1, 100}; |
| const uint8_t address6[16] = {0, 1, 2, 3, 4, 5, 6, 7, |
| 8, 9, 10, 11, 12, 13, 255, 0}; |
| |
| char buffer[INET6_ADDRSTRLEN]; |
| memset(buffer, 'x', sizeof(buffer)); |
| EXPECT_STREQ("1:203:405:607:809:a0b:c0d:ff00", |
| inet_ntop(AF_INET6, address6, buffer, sizeof(buffer))); |
| memset(buffer, 'x', sizeof(buffer)); |
| EXPECT_STREQ("1:203:405:607:809:a0b:c0d:ff00", |
| inet_ntop(AF_INET6, address6, buffer, 31)); |
| memset(buffer, 'x', sizeof(buffer)); |
| EXPECT_STREQ(nullptr, inet_ntop(AF_INET6, address6, buffer, 0)); |
| memset(buffer, 'x', sizeof(buffer)); |
| EXPECT_STREQ(nullptr, inet_ntop(AF_INET6, address6, buffer, 30)); |
| |
| memset(buffer, 'x', sizeof(buffer)); |
| EXPECT_STREQ("255.0.1.100", |
| inet_ntop(AF_INET, address4, buffer, sizeof(buffer))); |
| memset(buffer, 'x', sizeof(buffer)); |
| EXPECT_STREQ("255.0.1.100", inet_ntop(AF_INET, address4, buffer, 12)); |
| memset(buffer, 'x', sizeof(buffer)); |
| EXPECT_STREQ(nullptr, inet_ntop(AF_INET, address4, buffer, 0)); |
| memset(buffer, 'x', sizeof(buffer)); |
| EXPECT_STREQ(nullptr, inet_ntop(AF_INET, address4, buffer, 11)); |
| } |
| |
| #endif |