Merge in trunk.
[m6w6/libmemcached] / libtest / server.h
index 198e0f6abdc3b8981abcd3aece67c3d63976ac93..b91f55b13de3124b4102961efadffcba053ac3b4 100644 (file)
-/*
- * Copyright (C) 2011 Data Differential, http://datadifferential.com/
- * Copyright (C) 2006-2009 Brian Aker
- * All rights reserved.
+/*  vim:expandtab:shiftwidth=2:tabstop=2:smarttab:
+ * 
+ *  libtest
  *
- * Use and distribution licensed under the BSD license.  See
- * the COPYING file in the parent directory for full text.
+ *  Copyright (C) 2011 Data Differential, http://datadifferential.com/
+ *
+ *  This library is free software; you can redistribute it and/or
+ *  modify it under the terms of the GNU Lesser General Public
+ *  License as published by the Free Software Foundation; either
+ *  version 3 of the License, or (at your option) any later version.
+ *
+ *  This library is distributed in the hope that it will be useful,
+ *  but WITHOUT ANY WARRANTY; without even the implied warranty of
+ *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+ *  Lesser General Public License for more details.
+ *
+ *  You should have received a copy of the GNU Lesser General Public
+ *  License along with this library; if not, write to the Free Software
+ *  Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
  */
 
 #pragma once
 
+#include <cassert>
+#include <cstdio>
 #include <cstring>
 #include <netdb.h>
 #include <netinet/in.h>
 #include <string>
 #include <unistd.h>
+#include <vector>
 
-#define SERVERS_TO_CREATE 5
-
-struct server_st;
-
-typedef pid_t (test_server_getpid)(server_st &);
-typedef bool (test_server_ping)(server_st &);
+namespace libtest {
 
-struct server_st {
+struct Server {
 private:
-  bool _used;
+  bool _is_socket;
+  std::string _socket;
+  std::string _sasl;
+  std::string _pid_file;
+  std::string _log_file;
+  std::string _base_command; // executable command which include libtool, valgrind, gdb, etc
+  std::string _running; // Current string being used for system()
   pid_t _pid;
+
+protected:
   in_port_t _port;
-  char pid_file[FILENAME_MAX]; // Did we start it, or was it just sitting there?
-  std::string _command;
-  test_server_getpid *__get_pid;
-  test_server_ping *__ping;
+  std::string _hostname;
+  std::string _extra_args;
 
 public:
+  Server(const std::string& hostname, const in_port_t port_arg, const bool is_socket_arg= false);
+
+  virtual ~Server();
 
-  char hostname[NI_MAXHOST];
+  virtual const char *name()= 0;
+  virtual const char *executable()= 0;
+  virtual const char *port_option()= 0;
+  virtual const char *pid_file_option()= 0;
+  virtual const char *daemon_file_option()= 0;
+  virtual const char *log_file_option()= 0;
+  virtual bool is_libtool()= 0;
 
-  server_st() :
-    _used(false),
-    _pid(-1),
-    _port(0),
-    __get_pid(NULL),
-    __ping(NULL)
+  virtual bool broken_socket_cleanup()
   {
-    pid_file[0]= 0;
-    strncpy(hostname, "localhost", sizeof(hostname));
+    return false;
   }
 
-  void set_methods(test_server_getpid *get_pid_arg, test_server_ping *ping_arg)
+  virtual const char *socket_file_option() const
   {
-    __get_pid= get_pid_arg;
-    __ping= ping_arg;
+    return NULL;
   }
 
-  bool ping()
+  virtual bool broken_pid_file()
   {
-    if (__ping)
-      return __ping(*this);
-
     return false;
   }
 
-  pid_t get_pid()
+  const std::string& pid_file() const
+  {
+    return _pid_file;
+  }
+
+  const std::string& base_command() const
+  {
+    return _base_command;
+  }
+
+  const std::string& log_file() const
   {
-    if (__get_pid)
-      return _pid= __get_pid(*this);
+    return _log_file;
+  }
 
-    return -1;
+  const std::string& hostname() const
+  {
+    return _hostname;
+  }
+
+  const std::string& socket() const
+  {
+    return _socket;
   }
 
-  void set_port(in_port_t arg)
+  bool has_socket() const
   {
-    _port= arg;
+    return _is_socket;
   }
 
+  bool cycle();
+
+  virtual bool ping()= 0;
+
+  virtual pid_t get_pid(bool error_is_ok= false)= 0;
+
+  virtual bool build(int argc, const char *argv[])= 0;
+
   in_port_t port() const
   {
     return _port;
@@ -84,72 +124,69 @@ public:
     return (_port != 0);
   }
 
-  void set_command(const char *arg)
+  // Reset a server if another process has killed the server
+  void reset()
   {
-    _command= arg;
+    _pid= -1;
+    _pid_file.clear();
+    _log_file.clear();
   }
 
-  void set_used()
-  {
-    _used= true;
-  }
+  void set_extra_args(const std::string &arg);
+
+  bool args(std::string& options);
 
   pid_t pid();
 
-  bool is_used() const
+  pid_t pid() const
   {
-    return _used;
+    return _pid;
   }
 
-  ~server_st();
-
-  bool has_pid()
+  bool has_pid() const
   {
     return (_pid > 1);
   }
 
+  bool wait_for_pidfile() const;
+
+  bool check_pid(pid_t pid_arg) const
+  {
+    return (pid_arg > 1);
+  }
+
   bool is_socket() const
   {
-    return hostname[0] == '/';
+    return _hostname[0] == '/';
   }
 
-  void set_hostname(const char *arg)
+  const std::string running() const
   {
-    strncpy(hostname, arg, sizeof(hostname));
+    return _running;
   }
 
-  bool kill();
+  std::string log_and_pid();
+
+  bool kill(pid_t pid_arg);
   bool start();
+  bool command(std::string& command_arg);
+
+protected:
+  void nap();
+  bool set_pid_file();
 
 private:
+  bool is_helgrind() const;
+  bool is_valgrind() const;
+  bool is_debug() const;
+  bool set_log_file();
+  bool set_socket_file();
+  void rebuild_base_command();
   void reset_pid();
 };
 
-std::ostream& operator<<(std::ostream& output, const server_st &arg);
-
-struct server_startup_st
-{
-  uint8_t count;
-  uint8_t udp;
-  std::string server_list;
-  server_st server[SERVERS_TO_CREATE];
-
-  server_startup_st() :
-    count(SERVERS_TO_CREATE),
-    udp(0)
-  { }
-
-  ~server_startup_st();
-};
-
-#ifdef __cplusplus
-extern "C" {
-#endif
+std::ostream& operator<<(std::ostream& output, const libtest::Server &arg);
 
+} // namespace libtest
 
-bool server_startup(server_startup_st *construct);
-void server_shutdown(server_startup_st *construct);
 
-#ifdef __cplusplus
-}
-#endif