123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175 |
- // Copyright (c) 2016 Klemens D. Morgenstern
- //
- // Distributed under the Boost Software License, Version 1.0. (See accompanying
- // file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
- #ifndef BOOST_PROCESS_DETAIL_POSIX_BASIC_CMD_HPP_
- #define BOOST_PROCESS_DETAIL_POSIX_BASIC_CMD_HPP_
- #include <boost/process/detail/posix/handler.hpp>
- #include <boost/process/detail/posix/cmd.hpp>
- #include <boost/algorithm/string/replace.hpp>
- #include <boost/process/shell.hpp>
- #include <boost/algorithm/string/trim.hpp>
- #include <boost/algorithm/string/join.hpp>
- #include <string>
- #include <vector>
- namespace boost
- {
- namespace process
- {
- namespace detail
- {
- namespace posix
- {
- inline std::string build_cmd_shell(const std::string & exe, std::vector<std::string> && data)
- {
- std::string st = exe;
- for (auto & arg : data)
- {
- boost::replace_all(arg, "\"", "\\\"");
- auto it = std::find(arg.begin(), arg.end(), ' ');//contains space?
- if (it != arg.end())//ok, contains spaces.
- {
- //the first one is put directly onto the output,
- //because then I don't have to copy the whole string
- arg.insert(arg.begin(), '"' );
- arg += '"'; //thats the post one.
- }
- if (!st.empty())//first one does not need a preceeding space
- st += ' ';
- st += arg;
- }
- return st ;
- }
- inline std::vector<std::string> build_args(const std::string & data)
- {
- std::vector<std::string> st;
-
- typedef std::string::const_iterator itr_t;
- //normal quotes outside can be stripped, inside ones marked as \" will be replaced.
- auto make_entry = [](const itr_t & begin, const itr_t & end)
- {
- std::string data;
- if ((*begin == '"') && (*(end-1) == '"'))
- data.assign(begin+1, end-1);
- else
- data.assign(begin, end);
- boost::replace_all(data, "\\\"", "\"");
- return data;
- };
- bool in_quote = false;
- auto part_beg = data.cbegin();
- auto itr = data.cbegin();
- for (; itr != data.cend(); itr++)
- {
- if (*itr == '"')
- in_quote ^= true;
- if (!in_quote && (*itr == ' '))
- {
- //alright, got a space
- if ((itr != data.cbegin()) && (*(itr -1) != ' ' ))
- st.push_back(make_entry(part_beg, itr));
- part_beg = itr+1;
- }
- }
- if (part_beg != itr)
- st.emplace_back(make_entry(part_beg, itr));
- return st;
- }
- template<typename Char>
- struct exe_cmd_init;
- template<>
- struct exe_cmd_init<char> : boost::process::detail::api::handler_base_ext
- {
- exe_cmd_init(const exe_cmd_init & ) = delete;
- exe_cmd_init(exe_cmd_init && ) = default;
- exe_cmd_init(std::string && exe, std::vector<std::string> && args)
- : exe(std::move(exe)), args(std::move(args)) {};
- template <class Executor>
- void on_setup(Executor& exec)
- {
- if (exe.empty()) //cmd style
- {
- exec.exe = args.front().c_str();
- exec.cmd_style = true;
- }
- else
- exec.exe = &exe.front();
- cmd_impl = make_cmd();
- exec.cmd_line = cmd_impl.data();
- }
- static exe_cmd_init exe_args(std::string && exe, std::vector<std::string> && args) {return exe_cmd_init(std::move(exe), std::move(args));}
- static exe_cmd_init cmd (std::string && cmd)
- {
- auto args = build_args(cmd);
- return exe_cmd_init({}, std::move(args));
- }
- static exe_cmd_init exe_args_shell(std::string&& exe, std::vector<std::string> && args)
- {
- auto cmd = build_cmd_shell(std::move(exe), std::move(args));
- std::vector<std::string> args_ = {"-c", std::move(cmd)};
- std::string sh = shell().string();
- return exe_cmd_init(std::move(sh), std::move(args_));
- }
- static exe_cmd_init cmd_shell(std::string&& cmd)
- {
- std::vector<std::string> args = {"-c", "\"" + cmd + "\""};
- std::string sh = shell().string();
- return exe_cmd_init(
- std::move(sh),
- {std::move(args)});
- }
- private:
- inline std::vector<char*> make_cmd();
- std::string exe;
- std::vector<std::string> args;
- std::vector<char*> cmd_impl;
- };
- std::vector<char*> exe_cmd_init<char>::make_cmd()
- {
- std::vector<char*> vec;
- if (!exe.empty())
- vec.push_back(&exe.front());
- if (!args.empty()) {
- for (auto & v : args)
- vec.push_back(&v.front());
- }
- vec.push_back(nullptr);
- return vec;
- }
- }}}}
- #endif
|