Introduced new class IPHelper which should perform DNS and webcheck IP queries.

[bpdyndnsd] / src / iphelper.cpp

/** @file
 * @brief IPHelper class implementation. This class represents a Helper to get the actual IP.
 *
 *
 *
 * @copyright Intra2net AG
 * @license GPLv2
*/

#include "iphelper.h"
#include <boost/asio.hpp>

using namespace std;

namespace net = boost::asio;

/**
 * Default constructor.
 */
IPHelper::IPHelper()
    : Hostname("")
    , WebcheckUrl("")
    , WebcheckUrlAlt("")
    , Log(new Logger)
    , UseIPv6(false)
{
}


/**
 * Constructor.
 */
IPHelper::IPHelper(Logger::Ptr _log)
    : Hostname("")
    , WebcheckUrl("")
    , WebcheckUrlAlt("")
    , Log(new Logger)
    , UseIPv6(false)
{
    Log = _log;
}


/**
 * Default destructor
 */
IPHelper::~IPHelper()
{
}


/**
 * Initializes the ip helper.
 */
int IPHelper::init_hostname()
{
    Hostname = net::ip::host_name();

    Log->print_hostname(Hostname);

    return 0;
}


/**
 * Get the actual IP of this host through a conventional DNS query or through a IP webcheck URL if configured so.
 * @return A string representation of the actual IP in dotted format or an empty string if something went wrong.
 */
string IPHelper::get_actual_ip() const
{
    if ( WebcheckUrl == "" )
    {
        return dns_query();
    }
    else
    {
        return webcheck_ip();
    }
    return "";
}

void IPHelper::set_hostname(const string& _hostname)
{
    Hostname = _hostname;
}


/**
 * Get the actual IP of this host through a DNS query.
 * @return A string representation of the actual IP in dotted format or an empty string if something went wrong.
 */
string IPHelper::dns_query() const
{
    string ip_addr_v4 = "";
    string ip_addr_v6 = "";

    try
    {
        // BOOST asio isn't the simplest way to perform a DNS lookup, but it works just fine.
        net::io_service io_service;
        net::ip::tcp::resolver resolver(io_service);
        net::ip::tcp::resolver::query query(Hostname, "0");
        net::ip::tcp::resolver::iterator endpoint_iterator = resolver.resolve(query);
        net::ip::tcp::resolver::iterator end;
        while(endpoint_iterator != end)
        {
            net::ip::tcp::endpoint endpoint = endpoint_iterator->endpoint();
            net::ip::address ip = endpoint.address();
            if ( ip.is_v4() )
                ip_addr_v4 = ip.to_string();
            else if ( ip.is_v6() )
                ip_addr_v6 = ip.to_string();
            Log->print_own_ip(ip_addr_v4, ip_addr_v6);
            endpoint_iterator++;
        }
        io_service.reset();
    }
    catch(exception& e)
    {
        Log->print_error_hostname_to_ip(e.what(),Hostname);
        return "";
    }

    if ( (UseIPv6 == true) && (ip_addr_v6 != "") )
        return ip_addr_v6;

    return ip_addr_v4;
}


/**
 * Get the actual IP of this host through a IP webcheck URL.
 * @return A string representation of the actual IP in dotted format or an empty string if something went wrong.
 */
string IPHelper::webcheck_ip() const
{
    string ip_addr = "127.0.0.1";

    return ip_addr;
}


/**
 * Setter for both members WebcheckUrl and WebcheckUrlAlt
 * @param _webcheck_url The primary webcheck URL.
 * @param _webcheck_url_alt The fallback webcheck URL.
 */
void IPHelper::set_webcheck_urls(std::string& _webcheck_url, std::string& _webcheck_url_alt)
{

}