f8d2d87fbb
The MSVC CRT uses TLS storage to implement per-thread locales. This storage gets freed during program termination, and if we attempt to do any io operations (like flushing the std streams) after this occurs the program may abort. This patch is a speculative fix for that issue. The fix tries forcing the initialization of the locale TLS before initializing the std streams. This should mean that the TLS is freed after we destroy the streams. llvm-svn: 361348
145 lines
5.5 KiB
C++
145 lines
5.5 KiB
C++
//===------------------------ iostream.cpp --------------------------------===//
|
|
//
|
|
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
|
|
// See https://llvm.org/LICENSE.txt for license information.
|
|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "__std_stream"
|
|
#include "__locale"
|
|
#include "string"
|
|
#include "new"
|
|
|
|
#define _str(s) #s
|
|
#define str(s) _str(s)
|
|
#define _LIBCPP_ABI_NAMESPACE_STR str(_LIBCPP_ABI_NAMESPACE)
|
|
|
|
_LIBCPP_BEGIN_NAMESPACE_STD
|
|
|
|
#ifndef _LIBCPP_HAS_NO_STDIN
|
|
_ALIGNAS_TYPE (istream) _LIBCPP_FUNC_VIS char cin[sizeof(istream)]
|
|
#if defined(_LIBCPP_ABI_MICROSOFT) && defined(__clang__)
|
|
__asm__("?cin@" _LIBCPP_ABI_NAMESPACE_STR "@std@@3V?$basic_istream@DU?$char_traits@D@" _LIBCPP_ABI_NAMESPACE_STR "@std@@@12@A")
|
|
#endif
|
|
;
|
|
_ALIGNAS_TYPE (__stdinbuf<char> ) static char __cin[sizeof(__stdinbuf <char>)];
|
|
static mbstate_t mb_cin;
|
|
_ALIGNAS_TYPE (wistream) _LIBCPP_FUNC_VIS char wcin[sizeof(wistream)]
|
|
#if defined(_LIBCPP_ABI_MICROSOFT) && defined(__clang__)
|
|
__asm__("?wcin@" _LIBCPP_ABI_NAMESPACE_STR "@std@@3V?$basic_istream@_WU?$char_traits@_W@" _LIBCPP_ABI_NAMESPACE_STR "@std@@@12@A")
|
|
#endif
|
|
;
|
|
_ALIGNAS_TYPE (__stdinbuf<wchar_t> ) static char __wcin[sizeof(__stdinbuf <wchar_t>)];
|
|
static mbstate_t mb_wcin;
|
|
#endif
|
|
|
|
#ifndef _LIBCPP_HAS_NO_STDOUT
|
|
_ALIGNAS_TYPE (ostream) _LIBCPP_FUNC_VIS char cout[sizeof(ostream)]
|
|
#if defined(_LIBCPP_ABI_MICROSOFT) && defined(__clang__)
|
|
__asm__("?cout@" _LIBCPP_ABI_NAMESPACE_STR "@std@@3V?$basic_ostream@DU?$char_traits@D@" _LIBCPP_ABI_NAMESPACE_STR "@std@@@12@A")
|
|
#endif
|
|
;
|
|
_ALIGNAS_TYPE (__stdoutbuf<char>) static char __cout[sizeof(__stdoutbuf<char>)];
|
|
static mbstate_t mb_cout;
|
|
_ALIGNAS_TYPE (wostream) _LIBCPP_FUNC_VIS char wcout[sizeof(wostream)]
|
|
#if defined(_LIBCPP_ABI_MICROSOFT) && defined(__clang__)
|
|
__asm__("?wcout@" _LIBCPP_ABI_NAMESPACE_STR "@std@@3V?$basic_ostream@_WU?$char_traits@_W@" _LIBCPP_ABI_NAMESPACE_STR "@std@@@12@A")
|
|
#endif
|
|
;
|
|
_ALIGNAS_TYPE (__stdoutbuf<wchar_t>) static char __wcout[sizeof(__stdoutbuf<wchar_t>)];
|
|
static mbstate_t mb_wcout;
|
|
#endif
|
|
|
|
_ALIGNAS_TYPE (ostream) _LIBCPP_FUNC_VIS char cerr[sizeof(ostream)]
|
|
#if defined(_LIBCPP_ABI_MICROSOFT) && defined(__clang__)
|
|
__asm__("?cerr@" _LIBCPP_ABI_NAMESPACE_STR "@std@@3V?$basic_ostream@DU?$char_traits@D@" _LIBCPP_ABI_NAMESPACE_STR "@std@@@12@A")
|
|
#endif
|
|
;
|
|
_ALIGNAS_TYPE (__stdoutbuf<char>) static char __cerr[sizeof(__stdoutbuf<char>)];
|
|
static mbstate_t mb_cerr;
|
|
_ALIGNAS_TYPE (wostream) _LIBCPP_FUNC_VIS char wcerr[sizeof(wostream)]
|
|
#if defined(_LIBCPP_ABI_MICROSOFT) && defined(__clang__)
|
|
__asm__("?wcerr@" _LIBCPP_ABI_NAMESPACE_STR "@std@@3V?$basic_ostream@_WU?$char_traits@_W@" _LIBCPP_ABI_NAMESPACE_STR "@std@@@12@A")
|
|
#endif
|
|
;
|
|
_ALIGNAS_TYPE (__stdoutbuf<wchar_t>) static char __wcerr[sizeof(__stdoutbuf<wchar_t>)];
|
|
static mbstate_t mb_wcerr;
|
|
|
|
_ALIGNAS_TYPE (ostream) _LIBCPP_FUNC_VIS char clog[sizeof(ostream)]
|
|
#if defined(_LIBCPP_ABI_MICROSOFT) && defined(__clang__)
|
|
__asm__("?clog@" _LIBCPP_ABI_NAMESPACE_STR "@std@@3V?$basic_ostream@DU?$char_traits@D@" _LIBCPP_ABI_NAMESPACE_STR "@std@@@12@A")
|
|
#endif
|
|
;
|
|
_ALIGNAS_TYPE (wostream) _LIBCPP_FUNC_VIS char wclog[sizeof(wostream)]
|
|
#if defined(_LIBCPP_ABI_MICROSOFT) && defined(__clang__)
|
|
__asm__("?wclog@" _LIBCPP_ABI_NAMESPACE_STR "@std@@3V?$basic_ostream@_WU?$char_traits@_W@" _LIBCPP_ABI_NAMESPACE_STR "@std@@@12@A")
|
|
#endif
|
|
;
|
|
|
|
_LIBCPP_HIDDEN ios_base::Init __start_std_streams;
|
|
|
|
// On Windows the TLS storage for locales needs to be initialized before we create
|
|
// the standard streams, otherwise it may not be alive during program termination
|
|
// when we flush the streams.
|
|
static void force_locale_initialization() {
|
|
#if defined(_LIBCPP_MSVCRT_LIKE)
|
|
static bool once = []() {
|
|
auto loc = newlocale(LC_ALL_MASK, "C", 0);
|
|
{
|
|
__libcpp_locale_guard g(loc); // forces initialization of locale TLS
|
|
((void)g);
|
|
}
|
|
freelocale(loc);
|
|
return true;
|
|
}();
|
|
((void)once);
|
|
#endif
|
|
}
|
|
|
|
ios_base::Init::Init()
|
|
{
|
|
force_locale_initialization();
|
|
|
|
#ifndef _LIBCPP_HAS_NO_STDIN
|
|
istream* cin_ptr = ::new(cin) istream(::new(__cin) __stdinbuf <char>(stdin, &mb_cin));
|
|
wistream* wcin_ptr = ::new(wcin) wistream(::new(__wcin) __stdinbuf <wchar_t>(stdin, &mb_wcin));
|
|
#endif
|
|
#ifndef _LIBCPP_HAS_NO_STDOUT
|
|
ostream* cout_ptr = ::new(cout) ostream(::new(__cout) __stdoutbuf<char>(stdout, &mb_cout));
|
|
wostream* wcout_ptr = ::new(wcout) wostream(::new(__wcout) __stdoutbuf<wchar_t>(stdout, &mb_wcout));
|
|
#endif
|
|
ostream* cerr_ptr = ::new(cerr) ostream(::new(__cerr) __stdoutbuf<char>(stderr, &mb_cerr));
|
|
::new(clog) ostream(cerr_ptr->rdbuf());
|
|
wostream* wcerr_ptr = ::new(wcerr) wostream(::new(__wcerr) __stdoutbuf<wchar_t>(stderr, &mb_wcerr));
|
|
::new(wclog) wostream(wcerr_ptr->rdbuf());
|
|
|
|
#if !defined(_LIBCPP_HAS_NO_STDIN) && !defined(_LIBCPP_HAS_NO_STDOUT)
|
|
cin_ptr->tie(cout_ptr);
|
|
wcin_ptr->tie(wcout_ptr);
|
|
#endif
|
|
_VSTD::unitbuf(*cerr_ptr);
|
|
_VSTD::unitbuf(*wcerr_ptr);
|
|
#ifndef _LIBCPP_HAS_NO_STDOUT
|
|
cerr_ptr->tie(cout_ptr);
|
|
wcerr_ptr->tie(wcout_ptr);
|
|
#endif
|
|
}
|
|
|
|
ios_base::Init::~Init()
|
|
{
|
|
#ifndef _LIBCPP_HAS_NO_STDOUT
|
|
ostream* cout_ptr = reinterpret_cast<ostream*>(cout);
|
|
wostream* wcout_ptr = reinterpret_cast<wostream*>(wcout);
|
|
cout_ptr->flush();
|
|
wcout_ptr->flush();
|
|
#endif
|
|
|
|
ostream* clog_ptr = reinterpret_cast<ostream*>(clog);
|
|
wostream* wclog_ptr = reinterpret_cast<wostream*>(wclog);
|
|
clog_ptr->flush();
|
|
wclog_ptr->flush();
|
|
}
|
|
|
|
_LIBCPP_END_NAMESPACE_STD
|