1 //===----------------------------------------------------------------------===//
2 //
3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4 // See https://llvm.org/LICENSE.txt for license information.
5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6 //
7 //===----------------------------------------------------------------------===//
8
9 #include "std_stream.h"
10 #include <__locale>
11 #include <new>
12 #include <string>
13
14 #ifdef _LIBCPP_MSVCRT_LIKE
15 # include <__locale_dir/locale_base_api/locale_guard.h>
16 #endif
17
18 #define _str(s) #s
19 #define str(s) _str(s)
20 #define _LIBCPP_ABI_NAMESPACE_STR str(_LIBCPP_ABI_NAMESPACE)
21
22 _LIBCPP_BEGIN_NAMESPACE_STD
23
24 alignas(istream) _LIBCPP_EXPORTED_FROM_ABI char cin[sizeof(istream)]
25 #if defined(_LIBCPP_ABI_MICROSOFT) && defined(__clang__)
26 __asm__("?cin@" _LIBCPP_ABI_NAMESPACE_STR "@std@@3V?$basic_istream@DU?$char_traits@D@" _LIBCPP_ABI_NAMESPACE_STR
27 "@std@@@12@A")
28 #endif
29 ;
30 alignas(__stdinbuf<char>) static char __cin[sizeof(__stdinbuf<char>)];
31 static mbstate_t mb_cin;
32
33 #ifndef _LIBCPP_HAS_NO_WIDE_CHARACTERS
34 alignas(wistream) _LIBCPP_EXPORTED_FROM_ABI char wcin[sizeof(wistream)]
35 # if defined(_LIBCPP_ABI_MICROSOFT) && defined(__clang__)
36 __asm__("?wcin@" _LIBCPP_ABI_NAMESPACE_STR "@std@@3V?$basic_istream@_WU?$char_traits@_W@" _LIBCPP_ABI_NAMESPACE_STR
37 "@std@@@12@A")
38 # endif
39 ;
40 alignas(__stdinbuf<wchar_t>) static char __wcin[sizeof(__stdinbuf<wchar_t>)];
41 static mbstate_t mb_wcin;
42 #endif // _LIBCPP_HAS_NO_WIDE_CHARACTERS
43
44 alignas(ostream) _LIBCPP_EXPORTED_FROM_ABI char cout[sizeof(ostream)]
45 #if defined(_LIBCPP_ABI_MICROSOFT) && defined(__clang__)
46 __asm__("?cout@" _LIBCPP_ABI_NAMESPACE_STR "@std@@3V?$basic_ostream@DU?$char_traits@D@" _LIBCPP_ABI_NAMESPACE_STR
47 "@std@@@12@A")
48 #endif
49 ;
50 alignas(__stdoutbuf<char>) static char __cout[sizeof(__stdoutbuf<char>)];
51 static mbstate_t mb_cout;
52
53 #ifndef _LIBCPP_HAS_NO_WIDE_CHARACTERS
54 alignas(wostream) _LIBCPP_EXPORTED_FROM_ABI char wcout[sizeof(wostream)]
55 # if defined(_LIBCPP_ABI_MICROSOFT) && defined(__clang__)
56 __asm__("?wcout@" _LIBCPP_ABI_NAMESPACE_STR "@std@@3V?$basic_ostream@_WU?$char_traits@_W@" _LIBCPP_ABI_NAMESPACE_STR
57 "@std@@@12@A")
58 # endif
59 ;
60 alignas(__stdoutbuf<wchar_t>) static char __wcout[sizeof(__stdoutbuf<wchar_t>)];
61 static mbstate_t mb_wcout;
62 #endif // _LIBCPP_HAS_NO_WIDE_CHARACTERS
63
64 alignas(ostream) _LIBCPP_EXPORTED_FROM_ABI char cerr[sizeof(ostream)]
65 #if defined(_LIBCPP_ABI_MICROSOFT) && defined(__clang__)
66 __asm__("?cerr@" _LIBCPP_ABI_NAMESPACE_STR "@std@@3V?$basic_ostream@DU?$char_traits@D@" _LIBCPP_ABI_NAMESPACE_STR
67 "@std@@@12@A")
68 #endif
69 ;
70 alignas(__stdoutbuf<char>) static char __cerr[sizeof(__stdoutbuf<char>)];
71 static mbstate_t mb_cerr;
72
73 #ifndef _LIBCPP_HAS_NO_WIDE_CHARACTERS
74 alignas(wostream) _LIBCPP_EXPORTED_FROM_ABI char wcerr[sizeof(wostream)]
75 # if defined(_LIBCPP_ABI_MICROSOFT) && defined(__clang__)
76 __asm__("?wcerr@" _LIBCPP_ABI_NAMESPACE_STR "@std@@3V?$basic_ostream@_WU?$char_traits@_W@" _LIBCPP_ABI_NAMESPACE_STR
77 "@std@@@12@A")
78 # endif
79 ;
80 alignas(__stdoutbuf<wchar_t>) static char __wcerr[sizeof(__stdoutbuf<wchar_t>)];
81 static mbstate_t mb_wcerr;
82 #endif // _LIBCPP_HAS_NO_WIDE_CHARACTERS
83
84 alignas(ostream) _LIBCPP_EXPORTED_FROM_ABI char clog[sizeof(ostream)]
85 #if defined(_LIBCPP_ABI_MICROSOFT) && defined(__clang__)
86 __asm__("?clog@" _LIBCPP_ABI_NAMESPACE_STR "@std@@3V?$basic_ostream@DU?$char_traits@D@" _LIBCPP_ABI_NAMESPACE_STR
87 "@std@@@12@A")
88 #endif
89 ;
90
91 #ifndef _LIBCPP_HAS_NO_WIDE_CHARACTERS
92 alignas(wostream) _LIBCPP_EXPORTED_FROM_ABI char wclog[sizeof(wostream)]
93 # if defined(_LIBCPP_ABI_MICROSOFT) && defined(__clang__)
94 __asm__("?wclog@" _LIBCPP_ABI_NAMESPACE_STR "@std@@3V?$basic_ostream@_WU?$char_traits@_W@" _LIBCPP_ABI_NAMESPACE_STR
95 "@std@@@12@A")
96 # endif
97 ;
98 #endif // _LIBCPP_HAS_NO_WIDE_CHARACTERS
99
100 // Pretend we're inside a system header so the compiler doesn't flag the use of the init_priority
101 // attribute with a value that's reserved for the implementation (we're the implementation).
102 #include "iostream_init.h"
103
104 // On Windows the TLS storage for locales needs to be initialized before we create
105 // the standard streams, otherwise it may not be alive during program termination
106 // when we flush the streams.
force_locale_initialization()107 static void force_locale_initialization() {
108 #if defined(_LIBCPP_MSVCRT_LIKE)
109 static bool once = []() {
110 auto loc = newlocale(LC_ALL_MASK, "C", 0);
111 {
112 __libcpp_locale_guard g(loc); // forces initialization of locale TLS
113 ((void)g);
114 }
115 freelocale(loc);
116 return true;
117 }();
118 ((void)once);
119 #endif
120 }
121
122 class DoIOSInit {
123 public:
124 DoIOSInit();
125 ~DoIOSInit();
126 };
127
DoIOSInit()128 DoIOSInit::DoIOSInit() {
129 force_locale_initialization();
130
131 istream* cin_ptr = ::new (cin) istream(::new (__cin) __stdinbuf<char>(stdin, &mb_cin));
132 ostream* cout_ptr = ::new (cout) ostream(::new (__cout) __stdoutbuf<char>(stdout, &mb_cout));
133 ostream* cerr_ptr = ::new (cerr) ostream(::new (__cerr) __stdoutbuf<char>(stderr, &mb_cerr));
134 ::new (clog) ostream(cerr_ptr->rdbuf());
135 cin_ptr->tie(cout_ptr);
136 std::unitbuf(*cerr_ptr);
137 cerr_ptr->tie(cout_ptr);
138
139 #ifndef _LIBCPP_HAS_NO_WIDE_CHARACTERS
140 wistream* wcin_ptr = ::new (wcin) wistream(::new (__wcin) __stdinbuf<wchar_t>(stdin, &mb_wcin));
141 wostream* wcout_ptr = ::new (wcout) wostream(::new (__wcout) __stdoutbuf<wchar_t>(stdout, &mb_wcout));
142 wostream* wcerr_ptr = ::new (wcerr) wostream(::new (__wcerr) __stdoutbuf<wchar_t>(stderr, &mb_wcerr));
143 ::new (wclog) wostream(wcerr_ptr->rdbuf());
144
145 wcin_ptr->tie(wcout_ptr);
146 std::unitbuf(*wcerr_ptr);
147 wcerr_ptr->tie(wcout_ptr);
148 #endif
149 }
150
~DoIOSInit()151 DoIOSInit::~DoIOSInit() {
152 ostream* cout_ptr = reinterpret_cast<ostream*>(cout);
153 cout_ptr->flush();
154 ostream* clog_ptr = reinterpret_cast<ostream*>(clog);
155 clog_ptr->flush();
156
157 #ifndef _LIBCPP_HAS_NO_WIDE_CHARACTERS
158 wostream* wcout_ptr = reinterpret_cast<wostream*>(wcout);
159 wcout_ptr->flush();
160 wostream* wclog_ptr = reinterpret_cast<wostream*>(wclog);
161 wclog_ptr->flush();
162 #endif
163 }
164
Init()165 ios_base::Init::Init() {
166 static DoIOSInit init_the_streams; // gets initialized once
167 }
168
~Init()169 ios_base::Init::~Init() {}
170
171 _LIBCPP_END_NAMESPACE_STD
172