include/network.hpp

   1 #ifndef NETWORK_H
   2 #define NETWORK_H
   3
   4 #include <netinet/in.h>
   5 #include <arpa/inet.h>
   6 #include <sys/un.h>
   7 #include <sys/socket.h>
   8 #include <sys/stat.h>
   9 #include <sys/types.h>
  10
  11 namespace Sinkhole
  12 {
  13         /** Retrieve a pointer to the network buffer for recv() etc to read into.
  14          * @param The pointer and the size it points to
  15          */
  16         extern std::pair<char *, size_t> NetBuffer();
  17
  18         /** A sockaddr union used to combine IPv4 and IPv6 sockaddrs
  19          */
  20         union sockaddrs
  21         {
  22                 sockaddr sa;
  23                 sockaddr_in sa4;
  24                 sockaddr_in6 sa6;
  25                 sockaddr_un un;
  26
  27                 /** Construct the object, sets everything to 0
  28                  */
  29                 sockaddrs();
  30
  31                 /** Memset the object to 0
  32                  */
  33                 void clear();
  34
  35                 /** Get the size of the sockaddr we represent
  36                  * @return The size
  37                  */
  38                 size_t size() const;
  39
  40                 /** Get the port represented by this addr
  41                  * @return The port, or -1 on fail
  42                  */
  43                 int port() const;
  44
  45                 /** Get the address represented by this addr
  46                  * @return The address
  47                  */
  48                 std::string addr() const;
  49
  50                 /** Check if this sockaddr has data in it
  51                  */
  52                 bool operator()() const;
  53
  54                 /** Compares this sockaddr with another. Compares address type, port, and address
  55                  * @return true if they are the same
  56                  */
  57                 bool operator==(const sockaddrs &other) const;
  58                 /* The same as above but not */
  59                 inline bool operator!=(const sockaddrs &other) const { return !(*this == other); }
  60
  61                 /** Compares this sockaddr with another.
  62                  * @return true if this is less than the other
  63                  */
  64                 bool operator<(const sockaddrs &other) const;
  65
  66                 /** The equivalent of inet_pton
  67                  * @param type AF_INET or AF_INET6
  68                  * @param address The address to place in the sockaddr structures
  69                  * @param pport An option port to include in the  sockaddr structures
  70                  * @throws A socket exception if given invalid IPs
  71                  */
  72                 void pton(int type, const std::string &address, int pport = 0);
  73
  74                 /** The equivalent of inet_ntop
  75                  * @param type AF_INET or AF_INET6
  76                  * @param address The in_addr or in_addr6 structure
  77                  * @throws A socket exception if given an invalid structure
  78                  */
  79                 void ntop(int type, const void *src);
  80         };
  81
  82         class cidr
  83         {
  84                 static const unsigned char AF_INET_LEN;
  85                 static const unsigned char AF_INET6_LEN;
  86                 sockaddrs addr;
  87                 std::string cidr_ip;
  88                 unsigned char cidr_len;
  89          public:
  90                 cidr(const std::string &ip);
  91                 cidr(const std::string &ip, unsigned char len);
  92                 std::string mask() const;
  93                 bool match(sockaddrs &other);
  94                 bool operator==(const cidr &other) const;
  95                 /* The same as above but not */
  96                 inline bool operator!=(const cidr &other) const { return !(*this == other); }
  97                 bool operator<(const cidr &other) const;
  98         };
  99 }
 100
 101 #endif
 102