xref: /freebsd/contrib/llvm-project/libcxx/src/print.cpp (revision a90b9d0159070121c221b966469c3e36d912bf82)
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 <__config>
10 
11 #include <cstdlib>
12 #include <print>
13 
14 #include <__system_error/system_error.h>
15 
16 #include "filesystem/error.h"
17 
18 #if defined(_LIBCPP_WIN32API)
19 #  define WIN32_LEAN_AND_MEAN
20 #  define NOMINMAX
21 #  include <io.h>
22 #  include <windows.h>
23 #elif __has_include(<unistd.h>)
24 #  include <unistd.h>
25 #endif
26 
27 _LIBCPP_BEGIN_NAMESPACE_STD
28 
29 #if defined(_LIBCPP_WIN32API)
30 
31 _LIBCPP_EXPORTED_FROM_ABI bool __is_windows_terminal(FILE* __stream) {
32   // Note the Standard does this in one call, but it's unclear whether
33   // an invalid handle is allowed when calling GetConsoleMode.
34   //
35   // https://learn.microsoft.com/en-us/cpp/c-runtime-library/reference/get-osfhandle?view=msvc-170
36   // https://learn.microsoft.com/en-us/windows/console/getconsolemode
37   intptr_t __handle = _get_osfhandle(fileno(__stream));
38   if (__handle == -1)
39     return false;
40 
41   unsigned long __mode;
42   return GetConsoleMode(reinterpret_cast<void*>(__handle), &__mode);
43 }
44 
45 #  ifndef _LIBCPP_HAS_NO_WIDE_CHARACTERS
46 _LIBCPP_EXPORTED_FROM_ABI void
47 __write_to_windows_console([[maybe_unused]] FILE* __stream, [[maybe_unused]] wstring_view __view) {
48   // https://learn.microsoft.com/en-us/windows/console/writeconsole
49   if (WriteConsoleW(reinterpret_cast<void*>(_get_osfhandle(fileno(__stream))), // clang-format aid
50                     __view.data(),
51                     __view.size(),
52                     nullptr,
53                     nullptr) == 0) {
54     __throw_system_error(filesystem::detail::make_windows_error(GetLastError()), "failed to write formatted output");
55   }
56 }
57 #  endif // _LIBCPP_HAS_NO_WIDE_CHARACTERS
58 
59 #elif __has_include(<unistd.h>) // !_LIBCPP_WIN32API
60 
61 _LIBCPP_EXPORTED_FROM_ABI bool __is_posix_terminal(FILE* __stream) { return isatty(fileno(__stream)); }
62 #endif
63 
64 _LIBCPP_END_NAMESPACE_STD
65