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
  33    SrvRec(std::string const Target, u_int16_t const Priority,
  34          u_int16_t const Weight, u_int16_t const Port) :
  35       target(Target), priority(Priority), weight(Weight), port(Port),
  36       random_number_range_start(0), random_number_range_end(0),
  37       random_number_range_max(0) {}
  38 };
  39
  40 /** \brief Get SRV records from host/port (builds the query string internally)
  41  */
  42 bool GetSrvRecords(std::string name, std::vector<SrvRec> &Result);
  43
  44 /** \brief Get SRV records for query string like: _http._tcp.example.com
  45  */
  46 bool GetSrvRecords(std::string host, int port, std::vector<SrvRec> &Result);
  47
  48 /** \brief Pop a single SRV record from the vector of SrvRec taking
  49  *         priority and weight into account
  50  */
  51 SrvRec PopFromSrvRecs(std::vector<SrvRec> &Recs);
  52
  53 #endif