First steps in fine tuning and improving error handling.

[bpdyndnsd] / src / service_gnudip.cpp

/** @file
 * @brief GNUDIP Service class implementation. This class represents the GNUDIP service.
 *
 *
 *
 * @copyright Intra2net AG
 * @license GPLv2
*/

#include "service_gnudip.h"
#include "util.h"

#include <time.h>
#include <boost/foreach.hpp>
#include <boost/regex.hpp>

using namespace std;


/**
 * Default Constructor, needed for object serialization.
 */
ServiceGnudip::ServiceGnudip()
{
}


/**
 * Constructor.
 * @param _hostname The hostname to update
 * @param _login The login name.
 * @param _password The corresponding password.
 */
ServiceGnudip::ServiceGnudip(const string& _protocol, const string& _gnudip_server, const string& _hostname, const string& _login, const string& _password, const Logger::Ptr& _logger, const int _update_interval, const int _max_updates_within_interval, const int _dns_cache_ttl, const string& _proxy, const int _proxy_port)
    : GnudipServer(_gnudip_server)
{
    if ( _update_interval == -1 )        // If _update_interval is default po::option_desc (not specified via config)
        set_update_interval(0);              // use default protocol value
    else
        set_update_interval(_update_interval);

    if ( _max_updates_within_interval == -1 )
        set_max_updates_within_interval(0);
    else
        set_max_updates_within_interval(_max_updates_within_interval);

    if ( _dns_cache_ttl == -1 )
        set_dns_cache_ttl(60);
    else
        set_dns_cache_ttl(_dns_cache_ttl);

    set_protocol(_protocol);
    set_hostname(_hostname);
    set_login(_login);
    set_password(_password);
    set_logger(_logger);

    // create http helper class
    HTTPHelper::Ptr _http_help(new HTTPHelper(_logger,_proxy,_proxy_port,_login,_password));
    HTTPHelp.swap(_http_help);

    BaseUrl = assemble_base_url(get_hostname(),_gnudip_server);
}


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


/**
 * Assemble the dyndns update url from the given fqhn
 * @param hostname The fqhn hostname to update IP for.
 * @return The assembled update url without IP.
 */
string ServiceGnudip::assemble_base_url(const string& fqhn, const string& gnudip_server) const
{
    string base_url;

    base_url = "http://";
    base_url.append(gnudip_server);
    base_url.append("/gnudip/cgi-bin/gdipupdt.cgi");

    return base_url;
}


/**
 * Parses the data received from the initial request, which should contain salt, time and sign.
 * @param curl_data The complete received curl data.
 * @return A map with salt, time and sign or an empty map.
 */
map<string,string> ServiceGnudip::parse_initial_request(const string& curl_data) const
{
    map<string,string> response;

    // regex for salt
    boost::regex expr_salt("<meta name=\"salt\" content=\"(.*)\">");
    // regex for time
    boost::regex expr_time("<meta name=\"time\" content=\"(.*)\">");
    // regex for sign
    boost::regex expr_sign("<meta name=\"sign\" content=\"(.*)\">");

    boost::smatch matches;

    // Get the salt out of received http data
    if ( boost::regex_search(curl_data,matches,expr_salt) )
    {
        response.insert(pair<string,string>("salt",matches[1]));
        get_logger()->print_regex_match(expr_salt.str(),matches[1]);
    }
    else
    {
        get_logger()->print_no_regex_match(expr_salt.str(),curl_data);
        response.clear();
        return response;
    }

    // Get the time out of received http data
    if ( boost::regex_search(curl_data,matches,expr_time) )
    {
        response.insert(pair<string,string>("time",matches[1]));
        get_logger()->print_regex_match(expr_salt.str(),matches[1]);
    }
    else
    {
        get_logger()->print_no_regex_match(expr_salt.str(),curl_data);
        response.clear();
        return response;
    }

    // Get the sign out of received http data
    if ( boost::regex_search(curl_data,matches,expr_sign) )
    {
        response.insert(pair<string,string>("sign",matches[1]));
        get_logger()->print_regex_match(expr_salt.str(),matches[1]);
    }
    else
    {
        get_logger()->print_no_regex_match(expr_salt.str(),curl_data);
        response.clear();
        return response;
    }

    return response;
}


/**
 * Get the assembled update url.
 * @param salt Salt from the initial request
 * @param time Time from the initial request
 * @param sign Sign from the initial request
 * @param secret Computed md5 secret in HEX
 * @param ip IP to update
 * @return The assembled update url.
 */
string ServiceGnudip::assemble_update_url(const string& salt, const string& time, const string& sign, const string& secret, const string& ip) const
{
        string url = BaseUrl;

        url.append("?salt=");
        url.append(salt);
        url.append("&time=");
        url.append(time);
        url.append("&sign=");
        url.append(sign);
        url.append("&user=");
        url.append(get_login());
        url.append("&domn=");
        url.append(get_hostname());
        url.append("&pass=");
        url.append(secret);
        url.append("&reqc=0&addr=");
        url.append(ip);

        return url;
}


/**
 * Performs the Service update.
 * @param ip IP Address to set.
 * @return 0 if all is fine, -1 otherwise.
 */
int ServiceGnudip::perform_update(const std::string& ip)
{
    if ( HTTPHelp->is_initialized() == true )
    {
        // initial request
        long http_status_code = HTTPHelp->http_get(BaseUrl);

        get_logger()->print_http_status_code(BaseUrl,http_status_code);

        if ( http_status_code == 200 )
        {
            // Get the received http data which should contain the salt, time and sign
            string curl_data = HTTPHelp->get_curl_data();

            // Parse salt, time and sign out of the received data
            map<string,string> salt_time_sign = parse_initial_request(curl_data);

            if ( salt_time_sign.empty() )
            {
                get_logger()->print_could_not_parse_received_data(curl_data);
                return -1;
            }

            // at this point we have salt, time and sign parsed successfully
            string salt, time, sign;

            map<string,string>::iterator iter = salt_time_sign.find("salt");
            if ( iter != salt_time_sign.end() )
                salt = iter->second;

            iter = salt_time_sign.find("time");
            if ( iter != salt_time_sign.end() )
                time = iter->second;

            iter = salt_time_sign.find("sign");
            if ( iter != salt_time_sign.end() )
                sign = iter->second;

            if ( salt.empty() || time.empty() || sign.empty() )
                get_logger()->print_could_not_get_initial_gnudip_data();

            // compute md5 sum from users password and get the HEX representation
            string pw_md5_hex;
            try
            {
                pw_md5_hex = Util::compute_md5_digest(get_password());
            }
            catch ( invalid_argument e )
            {
                get_logger()->print_exception_md5_sum(e.what());
                return -1;
            }

            // append "." and salt and compute md5 sum and get the HEX representation
            pw_md5_hex.append(".");
            pw_md5_hex.append(salt);

            string secret;
            try
            {
                secret = Util::compute_md5_digest(pw_md5_hex);
            }
            catch ( invalid_argument e )
            {
                get_logger()->print_exception_md5_sum(e.what());
                return -1;
            }

            // Now its time to issue the second http_get operation
            string url = assemble_update_url(salt, time, sign, secret, ip);

            // perform the update operation
            http_status_code = HTTPHelp->http_get(url);

            get_logger()->print_http_status_code(url,http_status_code);

            if ( http_status_code == 200 )
            {
                // parse the update request return code
                string update_return_code = HTTPHelp->get_curl_data();
                if ( update_return_code == "0" )
                {
                    return 0;
                }
                else if ( update_return_code == "1" )
                {
                    get_logger()->print_service_not_authorized(url,get_login(),get_password());
                }
                else
                {
                    get_logger()->print_update_failure(url,update_return_code);
                }
            }
            else
            {
                // second http get operation (update) was not successful
                get_logger()->print_update_failure(url,http_status_code);
            }
        }
        else
        {
            // first http get operation was not successful
            get_logger()->print_update_failure(BaseUrl,http_status_code);
        }
    }
    else
    {
        get_logger()->print_service_not_initialized(BaseUrl);
    }
    return -1;
}