2 #include <netinet/in.h>
4 #include <sys/socket.h>
14 Client::Client(int sock)
17 connect_time(time(NULL)),
18 state(Client::READING_REQUEST),
20 header_or_error_bytes_sent(0),
26 request.reserve(1024);
28 // Find the remote address, and convert it to ASCII.
30 socklen_t addr_len = sizeof(addr);
32 if (getpeername(sock, reinterpret_cast<sockaddr *>(&addr), &addr_len) == -1) {
33 log_perror("getpeername");
36 char buf[INET6_ADDRSTRLEN];
37 if (inet_ntop(addr.sin6_family, &addr.sin6_addr, buf, sizeof(buf)) == NULL) {
38 log_perror("inet_ntop");
46 Client::Client(const ClientProto &serialized, Stream *stream)
47 : sock(serialized.sock()),
48 remote_addr(serialized.remote_addr()),
49 connect_time(serialized.connect_time()),
50 state(State(serialized.state())),
51 request(serialized.request()),
52 stream_id(serialized.stream_id()),
54 header_or_error(serialized.header_or_error()),
55 header_or_error_bytes_sent(serialized.header_or_error_bytes_sent()),
56 stream_pos(serialized.stream_pos()),
57 bytes_sent(serialized.bytes_sent()),
58 bytes_lost(serialized.bytes_lost()),
59 num_loss_events(serialized.num_loss_events())
61 if (stream != NULL && stream->mark_pool != NULL) {
62 fwmark = stream->mark_pool->get_mark();
64 fwmark = 0; // No mark.
66 if (setsockopt(sock, SOL_SOCKET, SO_MARK, &fwmark, sizeof(fwmark)) == -1) {
68 log_perror("setsockopt(SO_MARK)");
74 ClientProto Client::serialize() const
76 ClientProto serialized;
77 serialized.set_sock(sock);
78 serialized.set_remote_addr(remote_addr);
79 serialized.set_connect_time(connect_time);
80 serialized.set_state(state);
81 serialized.set_request(request);
82 serialized.set_stream_id(stream_id);
83 serialized.set_header_or_error(header_or_error);
84 serialized.set_header_or_error_bytes_sent(serialized.header_or_error_bytes_sent());
85 serialized.set_stream_pos(stream_pos);
86 serialized.set_bytes_sent(bytes_sent);
87 serialized.set_bytes_lost(bytes_lost);
88 serialized.set_num_loss_events(num_loss_events);
92 ClientStats Client::get_stats() const
95 if (stream_id.empty()) {
96 stats.stream_id = "-";
98 stats.stream_id = stream_id;
101 stats.fwmark = fwmark;
102 stats.remote_addr = remote_addr;
103 stats.connect_time = connect_time;
104 stats.bytes_sent = bytes_sent;
105 stats.bytes_lost = bytes_lost;
106 stats.num_loss_events = num_loss_events;