-/*
- * 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 <libtest/cmdline.h>
+
+#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
+namespace libtest {
-struct server_st;
-
-typedef pid_t (test_server_getpid)(server_st &);
-typedef bool (test_server_ping)(server_st &);
+struct Server {
+private:
+ typedef std::vector< std::pair<std::string, std::string> > Options;
-struct server_st {
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();
+
+ virtual const char *name()= 0;
+ virtual const char *executable()= 0;
+ virtual const char *daemon_file_option()= 0;
+ virtual bool is_libtool()= 0;
+
+ virtual bool has_socket_file_option() const
+ {
+ return false;
+ }
+
+ virtual void socket_file_option(Application& app, const std::string& socket_arg)
+ {
+ if (socket_arg.empty() == false)
+ {
+ std::string buffer("--socket=");
+ buffer+= socket_arg;
+ app.add_option(buffer);
+ }
+ }
- char hostname[NI_MAXHOST];
+ bool has_log_file_option() const
+ {
+ return false;
+ }
- server_st() :
- _used(false),
- _pid(-1),
- _port(0),
- __get_pid(NULL),
- __ping(NULL)
+ virtual void log_file_option(Application& app, const std::string& arg)
{
- pid_file[0]= 0;
- strncpy(hostname, "localhost", sizeof(hostname));
+ if (arg.empty() == false)
+ {
+ std::string buffer("--log-file=");
+ buffer+= arg;
+ app.add_option(buffer);
+ }
}
- void set_methods(test_server_getpid *get_pid_arg, test_server_ping *ping_arg)
+ virtual void pid_file_option(Application& app, const std::string& arg)
{
- __get_pid= get_pid_arg;
- __ping= ping_arg;
+ if (arg.empty() == false)
+ {
+ std::string buffer("--pid-file=");
+ buffer+= arg;
+ app.add_option(buffer);
+ }
}
- bool ping()
+ virtual bool has_port_option() const
{
- if (__ping)
- return __ping(*this);
+ return false;
+ }
+ virtual void port_option(Application& app, in_port_t arg)
+ {
+ if (arg > 0)
+ {
+ char buffer[1024];
+ snprintf(buffer, sizeof(buffer), "--port=%d", int(arg));
+ app.add_option(buffer);
+ }
+ }
+
+ virtual bool broken_socket_cleanup()
+ {
return false;
}
- pid_t get_pid()
+ virtual bool broken_pid_file()
{
- if (__get_pid)
- return _pid= __get_pid(*this);
+ return false;
+ }
- return -1;
+ const std::string& pid_file() const
+ {
+ return _pid_file;
}
- void set_port(in_port_t arg)
+ const std::string& base_command() const
{
- _port= arg;
+ return _base_command;
}
+ const std::string& log_file() const
+ {
+ return _log_file;
+ }
+
+ const std::string& hostname() const
+ {
+ return _hostname;
+ }
+
+ const std::string& socket() const
+ {
+ return _socket;
+ }
+
+ bool has_socket() const
+ {
+ return _is_socket;
+ }
+
+ bool cycle();
+
+ virtual bool ping()= 0;
+
+ virtual pid_t get_pid(bool error_is_ok= false)= 0;
+
+ virtual bool build(size_t argc, const char *argv[])= 0;
+
+ void add_option(const std::string&);
+ void add_option(const std::string&, const std::string&);
+
in_port_t port() const
{
return _port;
return (_port != 0);
}
- void set_command(const char *arg)
+ virtual bool has_syslog() const
{
- _command= arg;
+ return false;
}
- void set_used()
+ // Reset a server if another process has killed the server
+ void reset()
{
- _used= true;
+ _pid= -1;
+ _pid_file.clear();
+ _log_file.clear();
}
+ bool args(Application&);
+
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(libtest::Application& app);
+
+protected:
+ bool set_pid_file();
+ Options _options;
private:
+ bool is_helgrind() const;
+ bool is_valgrind() const;
+ bool is_debug() const;
+ bool set_log_file();
+ bool set_socket_file();
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