X-Git-Url: https://git.m6w6.name/?a=blobdiff_plain;f=libtest%2Fcmdline.h;h=a3d9cb25333fb6b33da74febfa544e44d846160b;hb=0792d5bdd0b2a352cce1a4cb084eca19b15acc56;hp=582cb4a506369f5b4677e149a3d4c2e270231299;hpb=04b8554c3724eae57fbd75dc6b3e69dca8b58187;p=m6w6%2Flibmemcached diff --git a/libtest/cmdline.h b/libtest/cmdline.h index 582cb4a5..a3d9cb25 100644 --- a/libtest/cmdline.h +++ b/libtest/cmdline.h @@ -1,39 +1,195 @@ /* vim:expandtab:shiftwidth=2:tabstop=2:smarttab: * - * uTest + * libtest * * Copyright (C) 2011 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: + * 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. * - * * 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. + * 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 -bool exec_cmdline(const std::string& executable, const char *args[]); +#include + +namespace libtest { + +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(); + +}