//===-- SocketAddressTest.cpp -----------------------------------*- C++ -*-===//
//
//                     The LLVM Compiler Infrastructure
//
// This file is distributed under the University of Illinois Open Source
// License. See LICENSE.TXT for details.
//
//===----------------------------------------------------------------------===//


#include "gtest/gtest.h"

#include "lldb/Host/SocketAddress.h"

namespace
{
    class SocketAddressTest: public ::testing::Test
    {
    };
}

using namespace lldb_private;

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));
#ifdef _WIN32
    ASSERT_STREQ ("0:0:0:0:0:0:0:1", sa.GetIPAddress ().c_str ());
#else
    ASSERT_STREQ ("::1", sa.GetIPAddress ().c_str ());
#endif
    ASSERT_EQ (1139, sa.GetPort ());
}

#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