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
|
/* Distributed under the OSI-approved BSD 3-Clause License. See accompanying
file LICENSE.rst or https://cmake.org/licensing for details. */
#include "cmStdIoInit.h"
#include <cerrno>
#include <cstdio>
#include <cstdlib>
#include <iostream>
#include <fcntl.h>
#ifdef _WIN32
# include <windows.h>
# include <io.h> // for _close, _dup2, _get_osfhandle
# include "cm_fileno.hxx"
#else
# include <unistd.h>
#endif
#include "cmStdIoStream.h"
namespace cm {
namespace StdIo {
namespace {
#ifdef _WIN32
void InitStdPipe(int stdFd, DWORD nStdHandle, FILE* stream,
wchar_t const* mode)
{
if (cm_fileno(stream) >= 0) {
return;
}
_close(stdFd);
_wfreopen(L"NUL", mode, stream);
int fd = cm_fileno(stream);
if (fd < 0) {
perror("failed to open NUL for missing stdio pipe");
abort();
}
if (fd != stdFd) {
_dup2(fd, stdFd);
}
SetStdHandle(nStdHandle, reinterpret_cast<HANDLE>(_get_osfhandle(fd)));
}
#else
void InitStdPipe(int fd)
{
if (fcntl(fd, F_GETFD) != -1 || errno != EBADF) {
return;
}
int f = open("/dev/null", fd == STDIN_FILENO ? O_RDONLY : O_WRONLY);
if (f == -1) {
perror("failed to open /dev/null for missing stdio pipe");
abort();
}
if (f != fd) {
dup2(f, fd);
close(f);
}
}
#endif
struct InitStdPipes
{
InitStdPipes()
{
#ifdef _WIN32
InitStdPipe(0, STD_INPUT_HANDLE, stdin, L"rb");
InitStdPipe(1, STD_OUTPUT_HANDLE, stdout, L"wb");
InitStdPipe(2, STD_ERROR_HANDLE, stderr, L"wb");
#else
InitStdPipe(STDIN_FILENO);
InitStdPipe(STDOUT_FILENO);
InitStdPipe(STDERR_FILENO);
#endif
}
};
} // anonymous namespace
class Globals
{
public:
std::ios::Init InitIos;
InitStdPipes InitPipes;
IStream StdIn{ std::cin, stdin };
OStream StdOut{ std::cout, stdout };
OStream StdErr{ std::cerr, stderr };
static Globals& Get();
};
Globals& Globals::Get()
{
static Globals globals;
return globals;
}
Init::Init()
{
Globals::Get();
}
IStream& In()
{
return Globals::Get().StdIn;
}
OStream& Out()
{
return Globals::Get().StdOut;
}
OStream& Err()
{
return Globals::Get().StdErr;
}
}
}
|
