X-Git-Url: https://git.m6w6.name/?a=blobdiff_plain;f=libtest%2Fserver.h;h=b193fb8ffd21f617d79a178a0477f4e17bb090c0;hb=8cc058d16abc8b75b675173c6c416c5ac7a30849;hp=123d1c3255f6d85af7b9b4ee2b26ef35c3b55099;hpb=58c279d58e2a44562f729e93e301dfedf42f530b;p=m6w6%2Flibmemcached diff --git a/libtest/server.h b/libtest/server.h index 123d1c32..b193fb8f 100644 --- a/libtest/server.h +++ b/libtest/server.h @@ -21,6 +21,8 @@ #pragma once +#include + #include #include #include @@ -33,6 +35,9 @@ namespace libtest { struct Server { +private: + typedef std::vector< std::pair > Options; + private: bool _is_socket; std::string _socket; @@ -55,20 +60,67 @@ public: 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 bool broken_socket_cleanup() + virtual bool has_socket_file_option() const { return false; } - virtual const char *socket_file_option() const + virtual void socket_file_option(Application& app, const std::string& socket_arg) { - return NULL; + if (socket_arg.empty() == false) + { + std::string buffer("--socket="); + buffer+= socket_arg; + app.add_option(buffer); + } + } + + 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 false; } virtual bool broken_pid_file() @@ -112,7 +164,10 @@ public: virtual pid_t get_pid(bool error_is_ok= false)= 0; - virtual bool build(int argc, const char *argv[])= 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 { @@ -124,6 +179,11 @@ public: return (_port != 0); } + virtual bool has_syslog() const + { + return false; + } + // Reset a server if another process has killed the server void reset() { @@ -132,9 +192,7 @@ public: _log_file.clear(); } - void set_extra_args(const std::string &arg); - - bool args(std::string& options); + bool args(Application&); pid_t pid(); @@ -148,6 +206,8 @@ public: return (_pid > 1); } + bool wait_for_pidfile() const; + bool check_pid(pid_t pid_arg) const { return (pid_arg > 1); @@ -167,11 +227,11 @@ public: bool kill(pid_t pid_arg); bool start(); - bool command(std::string& command_arg); + bool command(libtest::Application& app); protected: - void nap(); bool set_pid_file(); + Options _options; private: bool is_helgrind() const; @@ -179,7 +239,6 @@ private: bool is_debug() const; bool set_log_file(); bool set_socket_file(); - void rebuild_base_command(); void reset_pid(); };