unittests/Host/SocketAddressTest.cpp

   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