X-Git-Url: https://git.m6w6.name/?a=blobdiff_plain;f=libtest%2Fcmdline.h;h=a3d9cb25333fb6b33da74febfa544e44d846160b;hb=7933fa8f9217d7e744876773a4033b3cda4c3c16;hp=dcb4b0ab2b78b0ccea3fdb96e2c84d82ddb3e5af;hpb=b77f874c7d7ff386d01eeedb44c14d3003354bae;p=m6w6%2Flibmemcached diff --git a/libtest/cmdline.h b/libtest/cmdline.h index dcb4b0ab..a3d9cb25 100644 --- a/libtest/cmdline.h +++ b/libtest/cmdline.h @@ -21,10 +21,175 @@ #pragma once +#include + namespace libtest { -bool exec_cmdline(const std::string& executable, const char *args[]); +class Application { +private: + typedef std::vector< std::pair > Options; + +public: + + enum error_t { + SUCCESS= EXIT_SUCCESS, + FAILURE= EXIT_FAILURE, + INVALID= 127 + }; + + class Pipe { + public: + Pipe(int); + ~Pipe(); + + int fd(); + + enum close_t { + READ= 0, + WRITE= 1 + }; + + void reset(); + void close(const close_t& arg); + void dup_for_spawn(const close_t& arg, + posix_spawn_file_actions_t& file_actions); + + void nonblock(); + void cloexec(); + bool read(libtest::vchar_t&); + + private: + const int _std_fd; + int _pipe_fd[2]; + bool _open[2]; + }; + +public: + Application(const std::string& arg, const bool _use_libtool_arg= false); + + virtual ~Application(); + + void add_option(const std::string&); + void add_option(const std::string&, const std::string&); + void add_long_option(const std::string& option_name, const std::string& option_value); + error_t run(const char *args[]= NULL); + error_t wait(bool nohang= true); + + libtest::vchar_t stdout_result() const + { + return _stdout_buffer; + } + + size_t stdout_result_length() const + { + return _stdout_buffer.size(); + } + + libtest::vchar_t stderr_result() const + { + return _stderr_buffer; + } + + const char* stderr_c_str() const + { + return &_stderr_buffer[0]; + } + + size_t stderr_result_length() const + { + return _stderr_buffer.size(); + } + + std::string print(); + + void use_valgrind(bool arg= true) + { + _use_valgrind= arg; + } + + bool check() const; + + bool slurp(); + void murder(); + + void use_gdb(bool arg= true) + { + _use_gdb= arg; + } + + void use_ptrcheck(bool arg= true) + { + _use_ptrcheck= arg; + } + + std::string arguments(); + + std::string gdb_filename() + { + return _gdb_filename; + } + + pid_t pid() const + { + return _pid; + } + + void will_fail() + { + _will_fail= true; + } + +private: + void create_argv(const char *args[]); + void delete_argv(); + +private: + const bool _use_libtool; + bool _use_valgrind; + bool _use_gdb; + bool _use_ptrcheck; + bool _will_fail; + size_t _argc; + std::string _exectuble_name; + std::string _exectuble; + std::string _exectuble_with_path; + std::string _gdb_filename; + Options _options; + Pipe stdin_fd; + Pipe stdout_fd; + Pipe stderr_fd; + char * * built_argv; + pid_t _pid; + libtest::vchar_t _stdout_buffer; + libtest::vchar_t _stderr_buffer; +}; + +static inline std::ostream& operator<<(std::ostream& output, const enum Application::error_t &arg) +{ + switch (arg) + { + case Application::SUCCESS: + output << "EXIT_SUCCESS"; + break; + + case Application::FAILURE: + output << "EXIT_FAILURE"; + break; + + case Application::INVALID: + output << "127"; + break; + + default: + output << "EXIT_UNKNOWN"; + } + + return output; +} + +int exec_cmdline(const std::string& executable, const char *args[], bool use_libtool= false); const char *gearmand_binary(); +const char *drizzled_binary(); }