-/*
- * Copyright (C) 2011 Data Differential, http://datadifferential.com/
- * Copyright (C) 2006-2009 Brian Aker
- * All rights reserved.
+/* vim:expandtab:shiftwidth=2:tabstop=2:smarttab:
+ *
+ * Data Differential YATL (i.e. libtest) library
+ *
+ * Copyright (C) 2012 Data Differential, http://datadifferential.com/
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions are
+ * met:
+ *
+ * * Redistributions of source code must retain the above copyright
+ * notice, this list of conditions and the following disclaimer.
+ *
+ * * Redistributions in binary form must reproduce the above
+ * copyright notice, this list of conditions and the following disclaimer
+ * in the documentation and/or other materials provided with the
+ * distribution.
+ *
+ * * The names of its contributors may not be used to endorse or
+ * promote products derived from this software without specific prior
+ * written permission.
+ *
+ * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
+ * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
+ * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
+ * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
+ * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
+ * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
+ * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
+ * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
+ * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
+ * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
+ * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
*
- * Use and distribution licensed under the BSD license. See
- * the COPYING file in the parent directory for full text.
*/
#pragma once
+#include <libtest/cmdline.h>
+
#include <cassert>
#include <cstdio>
#include <cstring>
struct Server {
private:
+ typedef std::vector< std::pair<std::string, std::string> > Options;
+
+private:
+ uint64_t _magic;
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;
std::string _extra_args;
public:
- Server(const std::string& hostname, const in_port_t port_arg, const bool is_socket_arg= false);
+ Server(const std::string& hostname, const in_port_t port_arg,
+ const std::string& executable, const bool _is_libtool,
+ const bool is_socket_arg= false);
virtual ~Server();
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;
- virtual const char *socket_file_option() const
+ 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);
+ }
+ }
+
+ virtual bool has_log_file_option() const
+ {
+ return false;
+ }
+
+ virtual void log_file_option(Application& app, const std::string& arg)
+ {
+ if (arg.empty() == false)
+ {
+ std::string buffer("--log-file=");
+ buffer+= arg;
+ app.add_option(buffer);
+ }
+ }
+
+ virtual void pid_file_option(Application& app, const std::string& arg)
+ {
+ if (arg.empty() == false)
+ {
+ std::string buffer("--pid-file=");
+ buffer+= arg;
+ app.add_option(buffer);
+ }
+ }
+
+ virtual bool has_port_option() const
+ {
+ 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 NULL;
+ return false;
}
virtual bool broken_pid_file()
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;
- virtual bool build(int 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 != 0);
}
+ virtual bool has_syslog() const
+ {
+ return false;
+ }
+
// Reset a server if another process has killed the server
void reset()
{
- _pid= -1;
_pid_file.clear();
_log_file.clear();
}
- void set_extra_args(const std::string &arg);
+ bool args(Application&);
- bool args(std::string& options);
+ pid_t pid() const;
- pid_t pid();
+ bool has_pid() const;
- pid_t pid() const
- {
- return _pid;
- }
-
- bool has_pid() const
- {
- return (_pid > 1);
- }
+ virtual bool wait_for_pidfile() const;
bool check_pid(pid_t pid_arg) const
{
return _running;
}
+ bool check();
+
std::string log_and_pid();
- bool kill(pid_t pid_arg);
+ bool kill();
bool start();
- bool command(std::string& command_arg);
+ bool command(libtest::Application& app);
+
+ bool validate();
protected:
- void nap();
+ bool set_pid_file();
+ Options _options;
+ Application _app;
private:
+ bool is_helgrind() const;
bool is_valgrind() const;
bool is_debug() const;
bool set_log_file();
- bool set_pid_file();
bool set_socket_file();
- void rebuild_base_command();
void reset_pid();
};
std::ostream& operator<<(std::ostream& output, const libtest::Server &arg);
-class server_startup_st
-{
-private:
- std::string server_list;
-
-public:
-
- uint8_t udp;
- std::vector<Server *> servers;
-
- server_startup_st() :
- udp(0)
- { }
-
- bool start_socket_server(const std::string& server_type, const in_port_t try_port, int argc, const char *argv[]);
-
- std::string option_string() const;
-
- size_t count() const
- {
- return servers.size();
- }
-
- bool is_debug() const;
- bool is_valgrind() const;
-
- void shutdown(bool remove= false);
- void push_server(Server *);
- Server *pop_server();
-
- ~server_startup_st();
-};
-
-bool server_startup(server_startup_st&, const std::string&, in_port_t try_port, int argc, const char *argv[]);
-
} // namespace libtest