| 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 | #include <tuple> |
| 16 | |
| 17 | class SrvRec |
| 18 | { |
| 19 | public: |
| 20 | std::string target; |
| 21 | u_int16_t priority; |
| 22 | u_int16_t weight; |
| 23 | u_int16_t port; |
| 24 | |
| 25 | // each server is assigned a interval [start, end] in the space of [0, max] |
| 26 | int random_number_range_start; |
| 27 | int random_number_range_end; |
| 28 | int random_number_range_max; |
| 29 | |
| 30 | bool operator<(SrvRec const &other) const { |
| 31 | return this->priority < other.priority; |
| 32 | } |
| 33 | bool operator==(SrvRec const &other) const { |
| 34 | return std::tie(target, priority, weight, port) == std::tie(other.target, other.priority, other.weight, other.port); |
| 35 | } |
| 36 | |
| 37 | SrvRec(std::string const Target, u_int16_t const Priority, |
| 38 | u_int16_t const Weight, u_int16_t const Port) : |
| 39 | target(Target), priority(Priority), weight(Weight), port(Port), |
| 40 | random_number_range_start(0), random_number_range_end(0), |
| 41 | random_number_range_max(0) {} |
| 42 | }; |
| 43 | |
| 44 | /** \brief Get SRV records from host/port (builds the query string internally) |
| 45 | */ |
| 46 | bool GetSrvRecords(std::string name, std::vector<SrvRec> &Result); |
| 47 | |
| 48 | /** \brief Get SRV records for query string like: _http._tcp.example.com |
| 49 | */ |
| 50 | bool GetSrvRecords(std::string host, int port, std::vector<SrvRec> &Result); |
| 51 | |
| 52 | /** \brief Pop a single SRV record from the vector of SrvRec taking |
| 53 | * priority and weight into account |
| 54 | */ |
| 55 | SrvRec PopFromSrvRecs(std::vector<SrvRec> &Recs); |
| 56 | |
| 57 | #endif |