git.m6w6.name Git - awesomized/libmemcached/blob - libtest/server.h

   1 /*  vim:expandtab:shiftwidth=2:tabstop=2:smarttab:
   2  *
   3  *  libtest
   4  *
   5  *  Copyright (C) 2011 Data Differential, http://datadifferential.com/
   6  *
   7  *  This library is free software; you can redistribute it and/or
   8  *  modify it under the terms of the GNU Lesser General Public
   9  *  License as published by the Free Software Foundation; either
  10  *  version 3 of the License, or (at your option) any later version.
  11  *
  12  *  This library is distributed in the hope that it will be useful,
  13  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
  14  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  15  *  Lesser General Public License for more details.
  16  *
  17  *  You should have received a copy of the GNU Lesser General Public
  18  *  License along with this library; if not, write to the Free Software
  19  *  Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
  20  */
  21
  22 #pragma once
  23
  24 #include <cassert>
  25 #include <cstdio>
  26 #include <cstring>
  27 #include <netdb.h>
  28 #include <netinet/in.h>
  29 #include <string>
  30 #include <unistd.h>
  31 #include <vector>
  32
  33 namespace libtest {
  34
  35 struct Server {
  36 private:
  37   bool _is_socket;
  38   std::string _socket;
  39   std::string _pid_file;
  40   std::string _log_file;
  41   std::string _base_command; // executable command which include libtool, valgrind, gdb, etc
  42   std::string _running; // Current string being used for system()
  43   pid_t _pid;
  44
  45 protected:
  46   in_port_t _port;
  47   std::string _hostname;
  48   std::string _extra_args;
  49
  50 public:
  51   Server(const std::string& hostname, const in_port_t port_arg, const bool is_socket_arg= false);
  52
  53   virtual ~Server();
  54
  55   virtual const char *name()= 0;
  56   virtual const char *executable()= 0;
  57   virtual const char *port_option()= 0;
  58   virtual const char *pid_file_option()= 0;
  59   virtual const char *daemon_file_option()= 0;
  60   virtual const char *log_file_option()= 0;
  61   virtual bool is_libtool()= 0;
  62
  63   virtual const char *socket_file_option() const
  64   {
  65     return NULL;
  66   }
  67
  68   virtual bool broken_pid_file()
  69   {
  70     return false;
  71   }
  72
  73   const std::string& pid_file() const
  74   {
  75     return _pid_file;
  76   }
  77
  78   const std::string& base_command() const
  79   {
  80     return _base_command;
  81   }
  82
  83   const std::string& log_file() const
  84   {
  85     return _log_file;
  86   }
  87
  88   const std::string& hostname() const
  89   {
  90     return _hostname;
  91   }
  92
  93   const std::string& socket() const
  94   {
  95     return _socket;
  96   }
  97
  98   bool has_socket() const
  99   {
 100     return _is_socket;
 101   }
 102
 103   bool cycle();
 104
 105   virtual bool ping()= 0;
 106
 107   virtual pid_t get_pid(bool error_is_ok= false)= 0;
 108
 109   virtual bool build(int argc, const char *argv[])= 0;
 110
 111   in_port_t port() const
 112   {
 113     return _port;
 114   }
 115
 116   bool has_port() const
 117   {
 118     return (_port != 0);
 119   }
 120
 121   // Reset a server if another process has killed the server
 122   void reset()
 123   {
 124     _pid= -1;
 125     _pid_file.clear();
 126     _log_file.clear();
 127   }
 128
 129   void set_extra_args(const std::string &arg);
 130
 131   bool args(std::string& options);
 132
 133   pid_t pid();
 134
 135   pid_t pid() const
 136   {
 137     return _pid;
 138   }
 139
 140   bool has_pid() const
 141   {
 142     return (_pid > 1);
 143   }
 144
 145   bool check_pid(pid_t pid_arg) const
 146   {
 147     return (pid_arg > 1);
 148   }
 149
 150   bool is_socket() const
 151   {
 152     return _hostname[0] == '/';
 153   }
 154
 155   const std::string running() const
 156   {
 157     return _running;
 158   }
 159
 160   std::string log_and_pid();
 161
 162   bool kill(pid_t pid_arg);
 163   bool start();
 164   bool command(std::string& command_arg);
 165
 166 protected:
 167   void nap();
 168
 169 private:
 170   bool is_helgrind() const;
 171   bool is_valgrind() const;
 172   bool is_debug() const;
 173   bool set_log_file();
 174   bool set_pid_file();
 175   bool set_socket_file();
 176   void rebuild_base_command();
 177   void reset_pid();
 178 };
 179
 180 std::ostream& operator<<(std::ostream& output, const libtest::Server &arg);
 181
 182 class server_startup_st
 183 {
 184 private:
 185   std::string server_list;
 186   bool _socket;
 187
 188 public:
 189
 190   uint8_t udp;
 191   std::vector<Server *> servers;
 192
 193   server_startup_st() :
 194     _socket(false),
 195     udp(0)
 196   { }
 197
 198   bool start_socket_server(const std::string& server_type, const in_port_t try_port, int argc, const char *argv[]);
 199
 200   std::string option_string() const;
 201
 202   size_t count() const
 203   {
 204     return servers.size();
 205   }
 206
 207   bool is_debug() const;
 208   bool is_helgrind() const;
 209   bool is_valgrind() const;
 210
 211   bool socket()
 212   {
 213     return _socket;
 214   }
 215
 216   void set_socket()
 217   {
 218     _socket= true;
 219   }
 220
 221
 222   void shutdown(bool remove= false);
 223   void push_server(Server *);
 224   Server *pop_server();
 225
 226   ~server_startup_st();
 227 };
 228
 229 bool server_startup(server_startup_st&, const std::string&, in_port_t try_port, int argc, const char *argv[]);
 230
 231 } // namespace libtest
 232
 233