developer.intra2net.com Git - bpdyndnsd/blob - src/service_tzo.cpp

   1 /** @file
   2  * @brief TZO Service class implementation. This class represents the TZO service.
   3  *
   4  *
   5  *
   6  * @copyright Intra2net AG
   7  * @license GPLv2
   8 */
   9
  10 #include "service_tzo.h"
  11
  12 #include <time.h>
  13 #include <boost/foreach.hpp>
  14 #include <boost/algorithm/string.hpp>
  15
  16 namespace ba = boost::algorithm;
  17
  18 using namespace std;
  19
  20
  21 /**
  22  * Default Constructor, needed for object serialization.
  23  */
  24 ServiceTzo::ServiceTzo()
  25 {
  26 }
  27
  28
  29 /**
  30  * Constructor.
  31  * @param _hostname The hostname to update
  32  * @param _login The login name.
  33  * @param _password The corresponding password.
  34  */
  35 ServiceTzo::ServiceTzo(const string& _protocol, 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)
  36 {
  37     if ( _update_interval == -1 )        // If _update_interval is default po::option_desc (not specified via config)
  38         set_update_interval(0);              // use default protocol value
  39     else
  40         set_update_interval(_update_interval);
  41
  42     if ( _max_updates_within_interval == -1 )
  43         set_max_updates_within_interval(0);
  44     else
  45         set_max_updates_within_interval(_max_updates_within_interval);
  46
  47     if ( _dns_cache_ttl == -1 )
  48         set_dns_cache_ttl(60);
  49     else
  50         set_dns_cache_ttl(_dns_cache_ttl);
  51
  52     set_protocol(_protocol);
  53     set_hostname(_hostname);
  54     set_login(_login);
  55     set_password(_password);
  56     set_logger(_logger);
  57
  58     // create http helper class
  59     HTTPHelper::Ptr _http_help(new HTTPHelper(_logger,_proxy,_proxy_port));
  60     HTTPHelp = _http_help;
  61     _http_help.reset();
  62
  63     BaseUrl = assemble_base_url(get_hostname(),get_login(),get_password());
  64 }
  65
  66
  67 /**
  68  * Default destructor
  69  */
  70 ServiceTzo::~ServiceTzo()
  71 {
  72 }
  73
  74
  75 /**
  76  * Assemble the dyns update url from the given fqhn
  77  * @param hostname The fqhn hostname to update IP for.
  78  * @param username The username to use.
  79  * @param hostname The password to use.
  80  * @return The assembled update url without IP.
  81  */
  82 string ServiceTzo::assemble_base_url(const string& fqhn, const string& username, const string& password) const
  83 {
  84     string base_url;
  85
  86     base_url = "http://rh.tzo.com";
  87     base_url.append("/webclient/tzoperl.html?system=tzodns&info=1&tzoname=");
  88     base_url.append(fqhn);
  89     base_url.append("&email=");
  90     base_url.append(username);
  91     base_url.append("&tzokey=");
  92     base_url.append(password);
  93     base_url.append("&ipaddress=");
  94
  95     return base_url;
  96 }
  97
  98
  99 /**
 100  * Performs the Service update.
 101  * @param ip IP Address to set.
 102  * @return 0 if all is fine, -1 otherwise.
 103  */
 104 int ServiceTzo::perform_update(const std::string& ip)
 105 {
 106     string url = BaseUrl;
 107     url.append(ip);
 108
 109     long http_status_code = HTTPHelp->http_get(url);
 110
 111     get_logger()->print_http_status_code(url,http_status_code);
 112
 113     // HTTP operation completed successful.
 114     // Now we have to parse the data received by curl,
 115     // cause http status code is not significant for dyns update errors
 116     if ( http_status_code == 200 )
 117     {
 118         // Get the received http data and parse the status code.
 119         string curl_data = HTTPHelp->get_curl_data();
 120         string status_code = parse_status_code(curl_data);
 121
 122         if ( status_code == "200" )
 123         {
 124             return 0;
 125         }
 126         else if ( status_code == "401" )
 127         {
 128             get_logger()->print_http_not_authorized(url,get_login(),get_password());
 129         }
 130         else
 131         {
 132             get_logger()->print_update_failure(url,curl_data);
 133         }
 134     }
 135     else
 136     {
 137         get_logger()->print_update_failure(url,http_status_code);
 138     }
 139
 140     return -1;
 141 }
 142
 143
 144 /**
 145  * Get the status code from the returned http data
 146  * @param data The returned http data.
 147  * @return The status code.
 148  */
 149 string ServiceTzo::parse_status_code(const string& data) const
 150 {
 151     list<string> tokens;
 152     ba::split(tokens,data,boost::is_any_of(" "));
 153     return tokens.front();
 154 }
 155
 156
 157 /**
 158  * Serialize function needed by boost/serialization to define which members should be stored as the object state.
 159  * @param ar Archive
 160  * @param version Version
 161  */
 162 template<class Archive>
 163 void ServiceTzo::serialize(Archive & ar, const unsigned int version)
 164 {
 165     ar & boost::serialization::base_object<Service>(*this);
 166 }