git.sesse.net Git - cubemap/blob - client.cpp

   1 #include <arpa/inet.h>
   2 #include <netinet/in.h>
   3 #include <stdint.h>
   4 #include <sys/socket.h>
   5
   6 #include "client.h"
   7 #include "log.h"
   8 #include "markpool.h"
   9 #include "state.pb.h"
  10 #include "stream.h"
  11
  12 #ifndef SO_MAX_PACING_RATE
  13 #define SO_MAX_PACING_RATE 47
  14 #endif
  15
  16 using namespace std;
  17
  18 Client::Client(int sock)
  19         : sock(sock),
  20           fwmark(0),
  21           connect_time(time(NULL)),
  22           state(Client::READING_REQUEST),
  23           stream(NULL),
  24           header_or_error_bytes_sent(0),
  25           stream_pos(0),
  26           bytes_sent(0),
  27           bytes_lost(0),
  28           num_loss_events(0)
  29 {
  30         request.reserve(1024);
  31
  32         // Find the remote address, and convert it to ASCII.
  33         sockaddr_in6 addr;
  34         socklen_t addr_len = sizeof(addr);
  35
  36         if (getpeername(sock, reinterpret_cast<sockaddr *>(&addr), &addr_len) == -1) {
  37                 log_perror("getpeername");
  38                 remote_addr = "";
  39                 return;
  40         }
  41
  42         char buf[INET6_ADDRSTRLEN];
  43         if (IN6_IS_ADDR_V4MAPPED(&addr.sin6_addr)) {
  44                 // IPv4 address, really.
  45                 if (inet_ntop(AF_INET, &addr.sin6_addr.s6_addr32[3], buf, sizeof(buf)) == NULL) {
  46                         log_perror("inet_ntop");
  47                         remote_addr = "";
  48                 } else {
  49                         remote_addr = buf;
  50                 }
  51         } else {
  52                 if (inet_ntop(addr.sin6_family, &addr.sin6_addr, buf, sizeof(buf)) == NULL) {
  53                         log_perror("inet_ntop");
  54                         remote_addr = "";
  55                 } else {
  56                         remote_addr = buf;
  57                 }
  58         }
  59 }
  60
  61 Client::Client(const ClientProto &serialized, Stream *stream)
  62         : sock(serialized.sock()),
  63           remote_addr(serialized.remote_addr()),
  64           connect_time(serialized.connect_time()),
  65           state(State(serialized.state())),
  66           request(serialized.request()),
  67           url(serialized.url()),
  68           stream(stream),
  69           header_or_error(serialized.header_or_error()),
  70           header_or_error_bytes_sent(serialized.header_or_error_bytes_sent()),
  71           stream_pos(serialized.stream_pos()),
  72           bytes_sent(serialized.bytes_sent()),
  73           bytes_lost(serialized.bytes_lost()),
  74           num_loss_events(serialized.num_loss_events())
  75 {
  76         if (stream != NULL && stream->mark_pool != NULL) {
  77                 fwmark = stream->mark_pool->get_mark();
  78         } else {
  79                 fwmark = 0;  // No mark.
  80         }
  81         if (setsockopt(sock, SOL_SOCKET, SO_MARK, &fwmark, sizeof(fwmark)) == -1) {
  82                 if (fwmark != 0) {
  83                         log_perror("setsockopt(SO_MARK)");
  84                 }
  85                 fwmark = 0;
  86         }
  87         if (setsockopt(sock, SOL_SOCKET, SO_MAX_PACING_RATE, &stream->pacing_rate, sizeof(stream->pacing_rate)) == -1) {
  88                 if (stream->pacing_rate != ~0U) {
  89                         log_perror("setsockopt(SO_MAX_PACING_RATE)");
  90                 }
  91         }
  92 }
  93
  94 ClientProto Client::serialize() const
  95 {
  96         ClientProto serialized;
  97         serialized.set_sock(sock);
  98         serialized.set_remote_addr(remote_addr);
  99         serialized.set_connect_time(connect_time);
 100         serialized.set_state(state);
 101         serialized.set_request(request);
 102         serialized.set_url(url);
 103         serialized.set_header_or_error(header_or_error);
 104         serialized.set_header_or_error_bytes_sent(serialized.header_or_error_bytes_sent());
 105         serialized.set_stream_pos(stream_pos);
 106         serialized.set_bytes_sent(bytes_sent);
 107         serialized.set_bytes_lost(bytes_lost);
 108         serialized.set_num_loss_events(num_loss_events);
 109         return serialized;
 110 }
 111
 112 ClientStats Client::get_stats() const
 113 {
 114         ClientStats stats;
 115         if (url.empty()) {
 116                 stats.url = "-";
 117         } else {
 118                 stats.url = url;
 119         }
 120         stats.sock = sock;
 121         stats.fwmark = fwmark;
 122         stats.remote_addr = remote_addr;
 123         stats.connect_time = connect_time;
 124         stats.bytes_sent = bytes_sent;
 125         stats.bytes_lost = bytes_lost;
 126         stats.num_loss_events = num_loss_events;
 127         return stats;
 128 }