1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
|
// Copyright (c) 2006, 2007 Julio M. Merino Vidal
// Copyright (c) 2008 Ilya Sokolov, Boris Schaeling
// Copyright (c) 2009 Boris Schaeling
// Copyright (c) 2010 Felipe Tanus, Boris Schaeling
// Copyright (c) 2011, 2012 Jeff Flinn, Boris Schaeling
//
// 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)
#define BOOST_TEST_MAIN
#define BOOST_TEST_IGNORE_SIGCHLD
#include <boost/test/included/unit_test.hpp>
#include <boost/process.hpp>
#include <boost/system/error_code.hpp>
#include <boost/iostreams/device/file_descriptor.hpp>
#include <boost/iostreams/stream.hpp>
#include <boost/asio.hpp>
#include <boost/algorithm/string/predicate.hpp>
#include <string>
#include <istream>
#include <cstdlib>
#if defined(BOOST_WINDOWS_API)
# include <Windows.h>
typedef boost::asio::windows::stream_handle pipe_end;
#elif defined(BOOST_POSIX_API)
# include <sys/wait.h>
# include <unistd.h>
typedef boost::asio::posix::stream_descriptor pipe_end;
#endif
namespace bp = boost::process;
namespace bpi = boost::process::initializers;
namespace bio = boost::iostreams;
BOOST_AUTO_TEST_CASE(sync_io)
{
using boost::unit_test::framework::master_test_suite;
bp::pipe p = bp::create_pipe();
{
bio::file_descriptor_sink sink(p.sink, bio::close_handle);
boost::system::error_code ec;
bp::execute(
bpi::run_exe(master_test_suite().argv[1]),
bpi::set_cmd_line("test --echo-stdout hello"),
bpi::bind_stdout(sink),
bpi::set_on_error(ec)
);
BOOST_REQUIRE(!ec);
}
bio::file_descriptor_source source(p.source, bio::close_handle);
bio::stream<bio::file_descriptor_source> is(source);
std::string s;
is >> s;
BOOST_CHECK_EQUAL(s, "hello");
}
bp::pipe create_async_pipe()
{
#if defined(BOOST_WINDOWS_API)
std::string name = "\\\\.\\pipe\\boost_process_test_bind_stdout";
HANDLE handle1 = CreateNamedPipeA(name.c_str(), PIPE_ACCESS_INBOUND |
FILE_FLAG_OVERLAPPED, 0, 1, 8192, 8192, 0, NULL);
HANDLE handle2 = CreateFileA(name.c_str(), GENERIC_WRITE, 0, NULL,
OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL);
return bp::make_pipe(handle1, handle2);
#elif defined(BOOST_POSIX_API)
return bp::create_pipe();
#endif
}
struct read_handler
{
boost::asio::streambuf &buffer_;
read_handler(boost::asio::streambuf &buffer) : buffer_(buffer) {}
void operator()(const boost::system::error_code &ec, std::size_t size)
{
BOOST_REQUIRE(!ec);
std::istream is(&buffer_);
std::string line;
std::getline(is, line);
BOOST_CHECK(boost::algorithm::starts_with(line, "abc"));
}
};
BOOST_AUTO_TEST_CASE(async_io)
{
using boost::unit_test::framework::master_test_suite;
bp::pipe p = create_async_pipe();
{
bio::file_descriptor_sink sink(p.sink, bio::close_handle);
boost::system::error_code ec;
bp::execute(
bpi::run_exe(master_test_suite().argv[1]),
bpi::set_cmd_line("test --echo-stdout abc"),
bpi::bind_stdout(sink),
bpi::set_on_error(ec)
);
BOOST_REQUIRE(!ec);
}
boost::asio::io_service io_service;
pipe_end pend(io_service, p.source);
boost::asio::streambuf buffer;
boost::asio::async_read_until(pend, buffer, '\n',
read_handler(buffer));
io_service.run();
}
BOOST_AUTO_TEST_CASE(nul)
{
using boost::unit_test::framework::master_test_suite;
#if defined(BOOST_WINDOWS_API)
bio::file_descriptor_sink sink("NUL");
#elif defined(BOOST_POSIX_API)
bio::file_descriptor_sink sink("/dev/null");
#endif
boost::system::error_code ec;
bp::child c = bp::execute(
bpi::run_exe(master_test_suite().argv[1]),
bpi::set_cmd_line("test --is-nul-stdout"),
bpi::bind_stdout(sink),
bpi::set_on_error(ec)
);
BOOST_REQUIRE(!ec);
int exit_code = bp::wait_for_exit(c);
#if defined(BOOST_WINDOWS_API)
BOOST_CHECK_EQUAL(EXIT_SUCCESS, exit_code);
#elif defined(BOOST_POSIX_API)
BOOST_CHECK_EQUAL(EXIT_SUCCESS, WEXITSTATUS(exit_code));
#endif
}
|
