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

   1 /*
   2  * Copyright (C) 2011 Data Differential, http://datadifferential.com/
   3  * Copyright (C) 2006-2009 Brian Aker
   4  * All rights reserved.
   5  *
   6  * Use and distribution licensed under the BSD license.  See
   7  * the COPYING file in the parent directory for full text.
   8  */
   9
  10 #pragma once
  11
  12 #include <cassert>
  13 #include <cstdio>
  14 #include <cstring>
  15 #include <netdb.h>
  16 #include <netinet/in.h>
  17 #include <string>
  18 #include <unistd.h>
  19 #include <vector>
  20
  21 namespace libtest {
  22
  23 struct Server {
  24 private:
  25   bool _is_socket;
  26   std::string _socket;
  27   std::string _pid_file;
  28   std::string _log_file;
  29   std::string _base_command; // executable command which include libtool, valgrind, gdb, etc
  30   std::string _running; // Current string being used for system()
  31   pid_t _pid;
  32
  33 protected:
  34   in_port_t _port;
  35   std::string _hostname;
  36   std::string _extra_args;
  37
  38 public:
  39   Server(const std::string& hostname, const in_port_t port_arg, const bool is_socket_arg= false);
  40
  41   virtual ~Server();
  42
  43   virtual const char *name()= 0;
  44   virtual const char *executable()= 0;
  45   virtual const char *port_option()= 0;
  46   virtual const char *pid_file_option()= 0;
  47   virtual const char *daemon_file_option()= 0;
  48   virtual const char *log_file_option()= 0;
  49   virtual bool is_libtool()= 0;
  50
  51   virtual const char *socket_file_option() const
  52   {
  53     return NULL;
  54   }
  55
  56   virtual bool broken_pid_file()
  57   {
  58     return false;
  59   }
  60
  61   const std::string& pid_file() const
  62   {
  63     return _pid_file;
  64   }
  65
  66   const std::string& base_command() const
  67   {
  68     return _base_command;
  69   }
  70
  71   const std::string& log_file() const
  72   {
  73     return _log_file;
  74   }
  75
  76   const std::string& hostname() const
  77   {
  78     return _hostname;
  79   }
  80
  81   const std::string& socket() const
  82   {
  83     return _socket;
  84   }
  85
  86   bool has_socket() const
  87   {
  88     return _is_socket;
  89   }
  90
  91   bool cycle();
  92
  93   virtual bool ping()= 0;
  94
  95   virtual pid_t get_pid(bool error_is_ok= false)= 0;
  96
  97   virtual bool build(int argc, const char *argv[])= 0;
  98
  99   in_port_t port() const
 100   {
 101     return _port;
 102   }
 103
 104   bool has_port() const
 105   {
 106     return (_port != 0);
 107   }
 108
 109   // Reset a server if another process has killed the server
 110   void reset()
 111   {
 112     _pid= -1;
 113     _pid_file.clear();
 114     _log_file.clear();
 115   }
 116
 117   void set_extra_args(const std::string &arg);
 118
 119   bool args(std::string& options);
 120
 121   pid_t pid();
 122
 123   pid_t pid() const
 124   {
 125     return _pid;
 126   }
 127
 128   bool has_pid() const
 129   {
 130     return (_pid > 1);
 131   }
 132
 133   bool check_pid(pid_t pid_arg) const
 134   {
 135     return (pid_arg > 1);
 136   }
 137
 138   bool is_socket() const
 139   {
 140     return _hostname[0] == '/';
 141   }
 142
 143   const std::string running() const
 144   {
 145     return _running;
 146   }
 147
 148   std::string log_and_pid();
 149
 150   bool kill(pid_t pid_arg);
 151   bool start();
 152   bool command(std::string& command_arg);
 153
 154 protected:
 155   void nap();
 156
 157 private:
 158   bool is_helgrind() const;
 159   bool is_valgrind() const;
 160   bool is_debug() const;
 161   bool set_log_file();
 162   bool set_pid_file();
 163   bool set_socket_file();
 164   void rebuild_base_command();
 165   void reset_pid();
 166 };
 167
 168 std::ostream& operator<<(std::ostream& output, const libtest::Server &arg);
 169
 170 class server_startup_st
 171 {
 172 private:
 173   std::string server_list;
 174   bool _socket;
 175
 176 public:
 177
 178   uint8_t udp;
 179   std::vector<Server *> servers;
 180
 181   server_startup_st() :
 182     _socket(false),
 183     udp(0)
 184   { }
 185
 186   bool start_socket_server(const std::string& server_type, const in_port_t try_port, int argc, const char *argv[]);
 187
 188   std::string option_string() const;
 189
 190   size_t count() const
 191   {
 192     return servers.size();
 193   }
 194
 195   bool is_debug() const;
 196   bool is_helgrind() const;
 197   bool is_valgrind() const;
 198
 199   bool socket()
 200   {
 201     return _socket;
 202   }
 203
 204   void set_socket()
 205   {
 206     _socket= true;
 207   }
 208
 209
 210   void shutdown(bool remove= false);
 211   void push_server(Server *);
 212   Server *pop_server();
 213
 214   ~server_startup_st();
 215 };
 216
 217 bool server_startup(server_startup_st&, const std::string&, in_port_t try_port, int argc, const char *argv[]);
 218
 219 } // namespace libtest
 220
 221