2 #include <netinet/in.h>
4 #include <sys/socket.h>
12 #ifndef SO_MAX_PACING_RATE
13 #define SO_MAX_PACING_RATE 47
18 Client::Client(int sock)
21 connect_time(time(NULL)),
22 state(Client::READING_REQUEST),
24 header_or_error_bytes_sent(0),
30 request.reserve(1024);
32 // Find the remote address, and convert it to ASCII.
34 socklen_t addr_len = sizeof(addr);
36 if (getpeername(sock, reinterpret_cast<sockaddr *>(&addr), &addr_len) == -1) {
37 log_perror("getpeername");
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");
52 if (inet_ntop(addr.sin6_family, &addr.sin6_addr, buf, sizeof(buf)) == NULL) {
53 log_perror("inet_ntop");
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()),
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())
76 if (stream != NULL && stream->mark_pool != NULL) {
77 fwmark = stream->mark_pool->get_mark();
79 fwmark = 0; // No mark.
81 if (setsockopt(sock, SOL_SOCKET, SO_MARK, &fwmark, sizeof(fwmark)) == -1) {
83 log_perror("setsockopt(SO_MARK)");
88 if (setsockopt(sock, SOL_SOCKET, SO_MAX_PACING_RATE, &stream->pacing_rate, sizeof(stream->pacing_rate)) == -1) {
89 if (stream->pacing_rate != ~0U) {
90 log_perror("setsockopt(SO_MAX_PACING_RATE)");
96 ClientProto Client::serialize() const
98 ClientProto serialized;
99 serialized.set_sock(sock);
100 serialized.set_remote_addr(remote_addr);
101 serialized.set_connect_time(connect_time);
102 serialized.set_state(state);
103 serialized.set_request(request);
104 serialized.set_url(url);
105 serialized.set_header_or_error(header_or_error);
106 serialized.set_header_or_error_bytes_sent(serialized.header_or_error_bytes_sent());
107 serialized.set_stream_pos(stream_pos);
108 serialized.set_bytes_sent(bytes_sent);
109 serialized.set_bytes_lost(bytes_lost);
110 serialized.set_num_loss_events(num_loss_events);
114 ClientStats Client::get_stats() const
123 stats.fwmark = fwmark;
124 stats.remote_addr = remote_addr;
125 stats.connect_time = connect_time;
126 stats.bytes_sent = bytes_sent;
127 stats.bytes_lost = bytes_lost;
128 stats.num_loss_events = num_loss_events;