/*

 * Licensed to the Apache Software Foundation (ASF) under one or more

 * contributor license agreements.  See the NOTICE file distributed with

 * this work for additional information regarding copyright ownership.

 * The ASF licenses this file to You under the Apache License, Version 2.0

 * (the "License"); you may not use this file except in compliance with

 * the License.  You may obtain a copy of the License at

 *

 *      http://www.apache.org/licenses/LICENSE-2.0

 *

 * Unless required by applicable law or agreed to in writing, software

 * distributed under the License is distributed on an "AS IS" BASIS,

 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.

 * See the License for the specific language governing permissions and

 * limitations under the License.

 */

#include <log4cxx/logstring.h>

#include <log4cxx/helpers/inetaddress.h>

#include <log4cxx/helpers/loglog.h>

#include <log4cxx/helpers/transcoder.h>

#include <log4cxx/helpers/pool.h>

#include "apr_network_io.h"

using namespace LOG4CXX_NS;

using namespace LOG4CXX_NS::helpers;

IMPLEMENT_LOG4CXX_OBJECT(InetAddress)

struct InetAddress::InetAddressPrivate{

  LogString ipAddrString;

  LogString hostNameString;

};

UnknownHostException::UnknownHostException(const LogString& msg1)

  : Exception(msg1)

{

}

UnknownHostException::UnknownHostException(const UnknownHostException& src)

  : Exception(src)

{

}

UnknownHostException& UnknownHostException::operator=(const UnknownHostException& src)

{

  Exception::operator=(src);

  return *this;

}

InetAddress::InetAddress(const LogString& hostName, const LogString& hostAddr)

  : m_priv(std::make_unique<InetAddressPrivate>())

{

  m_priv->ipAddrString = hostAddr;

  m_priv->hostNameString = hostName;

}

InetAddress::~InetAddress(){}

/** Determines all the IP addresses of a host, given the host's name.

*/

std::vector<InetAddressPtr> InetAddress::getAllByName(const LogString& host)

{

  LOG4CXX_ENCODE_CHAR(encodedHost, host);

  // retrieve information about the given host

  Pool addrPool;

  apr_sockaddr_t* address = 0;

  apr_status_t status =

    apr_sockaddr_info_get(&address, encodedHost.c_str(),

      APR_INET, 0, 0, addrPool.getAPRPool());

  if (status != APR_SUCCESS)

  {

    LogString msg(LOG4CXX_STR("Cannot get information about host: "));

    msg.append(host);

    LogLog::error(msg);

    throw UnknownHostException(msg);

  }

  std::vector<InetAddressPtr> result;

  apr_sockaddr_t* currentAddr = address;

  while (currentAddr != NULL)

  {

    // retrieve the IP address of this InetAddress.

    LogString ipAddrString;

    char* ipAddr;

    status = apr_sockaddr_ip_get(&ipAddr, currentAddr);

    if (status == APR_SUCCESS)

    {

      std::string ip(ipAddr);

      Transcoder::decode(ip, ipAddrString);

    }

    // retrieve the host name of this InetAddress.

    LogString hostNameString;

    char* hostName;

    status = apr_getnameinfo(&hostName, currentAddr, 0);

    if (status == APR_SUCCESS)

    {

      std::string host(hostName);

      Transcoder::decode(host, hostNameString);

    }

    result.push_back(std::make_shared<InetAddress>(hostNameString, ipAddrString));

    currentAddr = currentAddr->next;

  }

  return result;

}

/** Determines the IP address of a host, given the host's name.

*/

InetAddressPtr InetAddress::getByName(const LogString& host)

{

  InetAddressPtr result;

  auto address = getAllByName(host);

  if (!address.empty())

    result = address.front();

  return result;

}

/** Returns the IP address string "%d.%d.%d.%d".

*/

LogString InetAddress::getHostAddress() const

{

  return m_priv->ipAddrString;

}

/** Gets the host name for this IP address.

*/

LogString InetAddress::getHostName() const

{

  return m_priv->hostNameString;

}

/** Returns the local host.

*/

InetAddressPtr InetAddress::getLocalHost()

{

  return getByName(LOG4CXX_STR("127.0.0.1"));

}

InetAddressPtr InetAddress::anyAddress()

{

  // APR_ANYADDR does not work with the LOG4CXX_STR macro

  return getByName(LOG4CXX_STR("0.0.0.0"));

}

/** Converts this IP address to a String.

*/

LogString InetAddress::toString() const

{

  LogString rv(getHostName());

  if (!rv.empty())

    rv.append(LOG4CXX_STR("/"));

  rv.append(getHostAddress());

  return rv;

}