apt-pkg/contrib/srvrec.h

   1 // -*- mode: cpp; mode: fold -*-
   2 // Description                                                          /*{{{*/
   3 /* ######################################################################
   4
   5    SRV record support
   6
   7    ##################################################################### */
   8                                                                         /*}}}*/
   9 #ifndef SRVREC_H
  10 #define SRVREC_H
  11
  12 #include <arpa/nameser.h>
  13 #include <vector>
  14 #include <string>
  15
  16 class SrvRec
  17 {
  18  public:
  19    std::string target;
  20    u_int16_t priority;
  21    u_int16_t weight;
  22    u_int16_t port;
  23
  24    // each server is assigned a interval [start, end] in the space of [0, max]
  25    int random_number_range_start;
  26    int random_number_range_end;
  27    int random_number_range_max;
  28
  29    bool operator<(SrvRec const &other) const {
  30       return this->priority < other.priority;
  31    }
  32    bool operator==(SrvRec const &other) const;
  33
  34    SrvRec(std::string const Target, u_int16_t const Priority,
  35          u_int16_t const Weight, u_int16_t const Port) :
  36       target(Target), priority(Priority), weight(Weight), port(Port),
  37       random_number_range_start(0), random_number_range_end(0),
  38       random_number_range_max(0) {}
  39 };
  40
  41 /** \brief Get SRV records from host/port (builds the query string internally)
  42  */
  43 bool GetSrvRecords(std::string name, std::vector<SrvRec> &Result);
  44
  45 /** \brief Get SRV records for query string like: _http._tcp.example.com
  46  */
  47 bool GetSrvRecords(std::string host, int port, std::vector<SrvRec> &Result);
  48
  49 /** \brief Pop a single SRV record from the vector of SrvRec taking
  50  *         priority and weight into account
  51  */
  52 SrvRec PopFromSrvRecs(std::vector<SrvRec> &Recs);
  53
  54 #endif