1 #include <system_error>
5 #include <sys/socket.h>
9 #include "stats_server.h"
15 stats_server::stats_server(const string &sockfile)
16 : sockfile_(sockfile) {}
19 stats_server::serve_forever()
21 int fd = socket(AF_UNIX, SOCK_STREAM, 0);
23 throw system_error(errno, system_category(),
24 "creating UNIX domain socket");
26 struct sockaddr_un addr;
27 memset(&addr, 0, sizeof(addr));
28 addr.sun_family = AF_UNIX;
29 if (sockfile_.length() + 1 >= sizeof(addr.sun_path))
30 throw range_error("UNIX domain socket path too long");
31 strcpy(addr.sun_path, sockfile_.c_str());
32 unlink(sockfile_.c_str());
34 if (::bind(fd, (struct sockaddr *) &addr, sizeof(addr)) < 0)
35 throw system_error(errno, system_category(),
36 "binding to " + sockfile_);
38 if (listen(fd, 5) < 0)
39 throw system_error(errno, system_category(),
40 "listening on " + sockfile_);
43 int cfd = accept(fd, nullptr, 0);
45 throw system_error(errno, system_category(), "accept failed");
46 thread(&stats_server::serve_client, this, cfd).detach();
52 stats_server::handle_cmd_get_counter_value(const string &name, packet &pkt)
54 get_counter_value_t ret;
55 ret.timestamp_us_ = timer::cur_usec();
56 if (!event_counter::stat(name, ret.d_))
57 cerr << "could not find counter " << name << endl;
58 pkt.assign((const char *) &ret, sizeof(ret));
63 stats_server::serve_client(int fd)
68 int r = pkt.recvpkt(fd);
70 cerr << "client disconnected" << endl;
74 perror("recv- dropping connection");
77 INVARIANT(pkt.size());
78 switch (pkt.data()[0]) {
79 case static_cast<uint8_t>(stats_command::GET_COUNTER_VALUE):
81 scratch.assign(pkt.data() + 1, pkt.size() - 1);
82 if (!handle_cmd_get_counter_value(scratch, pkt)) {
83 cerr << "error on handle_cmd_get_counter_value(), dropping" << endl;
90 cerr << "bad command- dropping connection" << endl;