DwmRDAPFetcher.hh

Go to the documentation of this file.

//===========================================================================
// @(#) $Name:$
// @(#) $Id: DwmRDAPFetcher.hh 9156 2017-04-19 05:36:59Z dwm $
//===========================================================================
//  Copyright (c) Daniel W. McRobb 2017
//  All rights reserved.
//
//  Redistribution and use in source and binary forms, with or without
//  modification, are permitted provided that the following conditions
//  are met:
//
//  1. Redistributions of source code must retain the above copyright
//     notice, this list of conditions and the following disclaimer.
//  2. Redistributions in binary form must reproduce the above copyright
//     notice, this list of conditions and the following disclaimer in the
//     documentation and/or other materials provided with the distribution.
//  3. The names of the authors and copyright holders may not be used to
//     endorse or promote products derived from this software without
//     specific prior written permission.
//
//  IN NO EVENT SHALL DANIEL W. MCROBB BE LIABLE TO ANY PARTY FOR
//  DIRECT, INDIRECT, SPECIAL, INCIDENTAL, OR CONSEQUENTIAL DAMAGES,
//  INCLUDING LOST PROFITS, ARISING OUT OF THE USE OF THIS SOFTWARE,
//  EVEN IF DANIEL W. MCROBB HAS BEEN ADVISED OF THE POSSIBILITY OF SUCH
//  DAMAGE.
//
//  THE SOFTWARE PROVIDED HEREIN IS ON AN "AS IS" BASIS, AND
//  DANIEL W. MCROBB HAS NO OBLIGATION TO PROVIDE MAINTENANCE, SUPPORT,
//  UPDATES, ENHANCEMENTS, OR MODIFICATIONS. DANIEL W. MCROBB MAKES NO
//  REPRESENTATIONS AND EXTENDS NO WARRANTIES OF ANY KIND, EITHER
//  IMPLIED OR EXPRESS, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
//  WARRANTIES OF MERCHANTABILITY OR FITNESS FOR A PARTICULAR PURPOSE,
//  OR THAT THE USE OF THIS SOFTWARE WILL NOT INFRINGE ANY PATENT,
//  TRADEMARK OR OTHER RIGHTS.
//===========================================================================

//---------------------------------------------------------------------------
//---------------------------------------------------------------------------

#ifndef _DWMRDAPFETCHER_HH_
#define _DWMRDAPFETCHER_HH_

#include <map>
#include <string>
#include <vector>

#include "DwmIpv4Prefix.hh"
#include "DwmSocket.hh"
#include "DwmTimeValue64.hh"
#include "DwmRDAPRequestMessage.hh"
#include "DwmRDAPResponseMessage.hh"

namespace Dwm {

  namespace RDAP {

    //------------------------------------------------------------------------
    //------------------------------------------------------------------------
    class FetchedEntry
    {
    public:
      //----------------------------------------------------------------------
      //----------------------------------------------------------------------
      FetchedEntry() = default;
      
      //----------------------------------------------------------------------
      //----------------------------------------------------------------------
      FetchedEntry(const Ipv4Prefix & prefix, const std::string & country,
                   const TimeValue64 & lastUpdated);

      //----------------------------------------------------------------------
      //----------------------------------------------------------------------
      const Ipv4Prefix & Prefix() const;
      
      //----------------------------------------------------------------------
      //----------------------------------------------------------------------
      const Ipv4Prefix & Prefix(const Ipv4Prefix & prefix);
      
      //----------------------------------------------------------------------
      //----------------------------------------------------------------------
      const std::string & Country() const;
      
      //----------------------------------------------------------------------
      //----------------------------------------------------------------------
      const std::string & Country(const std::string & country);
      
      //----------------------------------------------------------------------
      //----------------------------------------------------------------------
      const TimeValue64 & LastUpdated() const;
      
      //----------------------------------------------------------------------
      //----------------------------------------------------------------------
      const TimeValue64 & LastUpdated(const TimeValue64 & lastUpdated);
      
    private:
      Ipv4Prefix   _prefix;
      std::string  _country;
      TimeValue64  _lastUpdated;
    };

    typedef std::map<Ipv4Address,FetchedEntry>  FetchedEntryMap;
    
    //------------------------------------------------------------------------
    //------------------------------------------------------------------------
    class Fetcher
    {
    public:
      //----------------------------------------------------------------------
      //----------------------------------------------------------------------
      Fetcher();
      
      //----------------------------------------------------------------------
      //----------------------------------------------------------------------
      bool OpenSession(const std::string & server,
                       const std::string & privKeyPath,
                       const std::string & knownServicesPath);

      //----------------------------------------------------------------------
      //----------------------------------------------------------------------
      ~Fetcher();

      //----------------------------------------------------------------------
      //----------------------------------------------------------------------
      FetchedEntryMap GetEntries(const std::vector<Ipv4Address> & addrs);

      //----------------------------------------------------------------------
      //----------------------------------------------------------------------
      std::pair<bool,FetchedEntry> GetEntry(const Ipv4Address & addr);
      
      //----------------------------------------------------------------------
      //----------------------------------------------------------------------
      void CloseSession();

    private:
      Socket       _rdapSocket;
      std::string  _agreedKey;

      bool RequestResponse(const RequestMessage & request,
                           ResponseMessage & response);
      static bool ResponseEntryHasData(const Json::Value & jv);
    };
    
  }  // namespace RDAP

}  // namespace Dwm

#endif  // _DWMRDAPFETCHER_HH_