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
|
//===----------------------------------------------------------------------===//
//
// 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
//
//===----------------------------------------------------------------------===//
// <iostream>
// wistream wcin;
// UNSUPPORTED: no-wide-characters
// UNSUPPORTED: executor-has-no-bash
// FILE_DEPENDENCIES: ../send-stdin.sh
// RUN: %{build}
// RUN: %{exec} bash send-stdin.sh "%t.exe" "1234"
#include <iostream>
#include <cassert>
struct custom_codecvt : std::codecvt<wchar_t, char, std::mbstate_t> {
using base = std::codecvt<wchar_t, char, std::mbstate_t>;
protected:
result do_in(std::mbstate_t&, const char *from, const char *from_end,
const char *&from_next, wchar_t *to, wchar_t *to_end, wchar_t *&to_next) const {
while (from != from_end && to != to_end) {
++from;
*to++ = L'z';
}
from_next = from;
to_next = to;
return ok;
}
};
int main(int, char**) {
std::locale loc(std::locale::classic(), new custom_codecvt);
std::wcin.imbue(loc);
std::wstring str;
std::wcin >> str;
assert(str == L"zzzz");
return 0;
}
|
