silo/stats_server.cc

   1 #include <system_error>
   2 #include <thread>
   3
   4 #include <unistd.h>
   5 #include <sys/socket.h>
   6 #include <sys/un.h>
   7
   8 #include "counter.h"
   9 #include "stats_server.h"
  10 #include "util.h"
  11
  12 using namespace std;
  13 using namespace util;
  14
  15 stats_server::stats_server(const string &sockfile)
  16   : sockfile_(sockfile) {}
  17
  18 void
  19 stats_server::serve_forever()
  20 {
  21   int fd = socket(AF_UNIX, SOCK_STREAM, 0);
  22   if (fd < 0)
  23     throw system_error(errno, system_category(),
  24         "creating UNIX domain socket");
  25
  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());
  33
  34   if (::bind(fd, (struct sockaddr *) &addr, sizeof(addr)) < 0)
  35     throw system_error(errno, system_category(),
  36         "binding to " + sockfile_);
  37
  38   if (listen(fd, 5) < 0)
  39     throw system_error(errno, system_category(),
  40         "listening on " + sockfile_);
  41
  42   for (;;) {
  43     int cfd = accept(fd, nullptr, 0);
  44     if (cfd < 0)
  45       throw system_error(errno, system_category(), "accept failed");
  46     thread(&stats_server::serve_client, this, cfd).detach();
  47   }
  48 }
  49
  50
  51 bool
  52 stats_server::handle_cmd_get_counter_value(const string &name, packet &pkt)
  53 {
  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));
  59   return true;
  60 }
  61
  62 void
  63 stats_server::serve_client(int fd)
  64 {
  65   packet pkt;
  66   string scratch;
  67   for (;;) {
  68     int r = pkt.recvpkt(fd);
  69     if (r == EOF) {
  70       cerr << "client disconnected" << endl;
  71       return;
  72     }
  73     if (r) {
  74       perror("recv- dropping connection");
  75       return;
  76     }
  77     INVARIANT(pkt.size());
  78     switch (pkt.data()[0]) {
  79     case static_cast<uint8_t>(stats_command::GET_COUNTER_VALUE):
  80       {
  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;
  84           return;
  85         }
  86         pkt.sendpkt(fd);
  87         break;
  88       }
  89     default:
  90       cerr << "bad command- dropping connection" << endl;
  91       return;
  92     }
  93   }
  94 }