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
125
126
127
128
|
//===- llvm/Support/Win32/PathV2.cpp - Windows Path Impl --------*- C++ -*-===//
//
// The LLVM Compiler Infrastructure
//
// This file is distributed under the University of Illinois Open Source
// License. See LICENSE.TXT for details.
//
//===----------------------------------------------------------------------===//
//
// This file implements the Windows specific implementation of the PathV2 API.
//
//===----------------------------------------------------------------------===//
//===----------------------------------------------------------------------===//
//=== WARNING: Implementation here must contain only generic Windows code that
//=== is guaranteed to work on *all* Windows variants.
//===----------------------------------------------------------------------===//
#include "Windows.h"
using namespace llvm;
namespace {
error_code UTF8ToUTF16(const StringRef &utf8,
SmallVectorImpl<wchar_t> &utf16) {
int len = ::MultiByteToWideChar(CP_UTF8, MB_ERR_INVALID_CHARS,
utf8.begin(), utf8.size(),
utf16.begin(), 0);
if (len == 0)
return make_error_code(windows_error(::GetLastError()));
utf16.reserve(len + 1);
utf16.set_size(len);
len = ::MultiByteToWideChar(CP_UTF8, MB_ERR_INVALID_CHARS,
utf8.begin(), utf8.size(),
utf16.begin(), utf16.size());
if (len == 0)
return make_error_code(windows_error(::GetLastError()));
// Make utf16 null terminated.
utf16.push_back(0);
utf16.pop_back();
return make_error_code(errc::success);
}
}
namespace llvm {
namespace sys {
namespace path {
error_code current_path(SmallVectorImpl<char> &result) {
SmallVector<wchar_t, 128> cur_path;
cur_path.reserve(128);
retry_cur_dir:
DWORD len = ::GetCurrentDirectoryW(cur_path.capacity(), cur_path.data());
// A zero return value indicates a failure other than insufficient space.
if (len == 0)
return make_error_code(windows_error(::GetLastError()));
// If there's insufficient space, the len returned is larger than the len
// given.
if (len > cur_path.capacity()) {
cur_path.reserve(len);
goto retry_cur_dir;
}
cur_path.set_size(len);
// cur_path now holds the current directory in utf-16. Convert to utf-8.
// Find out how much space we need. Sadly, this function doesn't return the
// size needed unless you tell it the result size is 0, which means you
// _always_ have to call it twice.
len = ::WideCharToMultiByte(CP_UTF8, NULL,
cur_path.data(), cur_path.size(),
result.data(), 0,
NULL, NULL);
if (len == 0)
return make_error_code(windows_error(::GetLastError()));
result.reserve(len);
result.set_size(len);
// Now do the actual conversion.
len = ::WideCharToMultiByte(CP_UTF8, NULL,
cur_path.data(), cur_path.size(),
result.data(), result.size(),
NULL, NULL);
if (len == 0)
return make_error_code(windows_error(::GetLastError()));
return make_error_code(errc::success);
}
} // end namespace path
namespace fs {
error_code copy_file(const Twine &from, const Twine &to, copy_option copt) {
// Get arguments.
SmallString<128> from_storage;
SmallString<128> to_storage;
StringRef f = from.toNullTerminatedStringRef(from_storage);
StringRef t = to.toNullTerminatedStringRef(to_storage);
// Convert to utf-16.
SmallVector<wchar_t, 128> wide_from;
SmallVector<wchar_t, 128> wide_to;
if (error_code ec = UTF8ToUTF16(f, wide_from)) return ec;
if (error_code ec = UTF8ToUTF16(t, wide_to)) return ec;
// Copy the file.
BOOL res = ::CopyFileW(wide_from.begin(), wide_to.begin(),
copt != copy_option::overwrite_if_exists);
if (res == 0)
return make_error_code(windows_error(::GetLastError()));
return make_error_code(errc::success);
}
} // end namespace fs
} // end namespace sys
} // end namespace llvm
|
