continue implementation; first tests with recursion returned IPs but then added cance...

[pingcheck] / src / dns / hostaddress.h

/*
The software in this package is distributed under the GNU General
Public License version 2 (with a special exception described below).

A copy of GNU General Public License (GPL) is included in this distribution,
in the file COPYING.GPL.

As a special exception, if other files instantiate templates or use macros
or inline functions from this file, or you compile this file and link it
with other works to produce a work based on this file, this file
does not by itself cause the resulting work to be covered
by the GNU General Public License.

However the source code for this file must still be made available
in accordance with section (3) of the GNU General Public License.

This exception does not invalidate any other reasons why a work based
on this file might be covered by the GNU General Public License.
*/
#ifndef HOST_ADDRESS_H
#define HOST_ADDRESS_H

#include <stdint.h>

#include <list>

#include <boost/asio.hpp>
#include <boost/serialization/access.hpp>
#include <boost/serialization/nvp.hpp>
#include <boost/serialization/split_member.hpp>
#include <boost/date_time/posix_time/posix_time.hpp>

#include "dns/timetolive.h"

//-----------------------------------------------------------------------------
// HostAddress
//-----------------------------------------------------------------------------

class HostAddress
{
public:
    HostAddress();
    HostAddress(
            const boost::asio::ip::address &ip,
            uint32_t ttl
    );
    ~HostAddress();

    boost::asio::ip::address get_ip() const;
    void set_ip( const boost::asio::ip::address &ip );

    TimeToLive get_ttl() const;
    void set_ttl( const TimeToLive &ttl );

private:
    /// IP address of the host
    boost::asio::ip::address Ip;
    /// time-to-live of the host IP
    TimeToLive Ttl;

    /// serialization: needs to be implemented here or will not compile
    // since there is no serialize in ip::address nor ptime,
    // need to save in other formats here
    friend class boost::serialization::access;
    template<class Archive>
    void save(Archive & ar, const unsigned int version) const
    {
        std::string ip = Ip.to_string();
        std::string ttl_creation_time = boost::posix_time::to_iso_string(
                                                                Ttl.TtlSetTime);
        ar & BOOST_SERIALIZATION_NVP(ip);
        ar & BOOST_SERIALIZATION_NVP(Ttl.Ttl);
        ar & BOOST_SERIALIZATION_NVP(ttl_creation_time);
    }

    template<class Archive>
    void load(Archive & ar, const unsigned int version)
    {
        std::string ip;
        uint32_t ttl_seconds;
        std::string ttl_creation_time;
        ar & BOOST_SERIALIZATION_NVP(ip);
        ar & BOOST_SERIALIZATION_NVP(ttl_seconds);
        ar & BOOST_SERIALIZATION_NVP(ttl_creation_time);

        // now convert to Ip and Ttl
        Ip = boost::asio::ip::address::from_string(ip);
        Ttl = TimeToLive();
        Ttl.Ttl = ttl_seconds;
        Ttl.TtlSetTime = boost::posix_time::from_iso_string(ttl_creation_time);
    }

    BOOST_SERIALIZATION_SPLIT_MEMBER()
};

//-----------------------------------------------------------------------------
// HostAddressList
//-----------------------------------------------------------------------------

typedef std::list<HostAddress> HostAddressList;

#endif // HOST_ADDRESS_H