Support/FileSystem: Add create_symlink implementation.

git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@120800 91177308-0d34-0410-b5e6-96231b3b80d8
This commit is contained in:
Michael J. Spencer 2010-12-03 07:41:25 +00:00
parent d7b305f13e
commit 9b391c513e
2 changed files with 45 additions and 0 deletions

View File

@ -173,6 +173,19 @@ error_code create_hard_link(const Twine &to, const Twine &from) {
return make_error_code(errc::success);
}
error_code create_symlink(const Twine &to, const Twine &from) {
// Get arguments.
SmallString<128> from_storage;
SmallString<128> to_storage;
StringRef f = from.toNullTerminatedStringRef(from_storage);
StringRef t = to.toNullTerminatedStringRef(to_storage);
if (::symlink(t.begin(), f.begin()) == -1)
return error_code(errno, system_category());
return make_error_code(errc::success);
}
error_code exists(const Twine &path, bool &result) {
SmallString<128> path_storage;
StringRef p = path.toNullTerminatedStringRef(path_storage);

View File

@ -23,6 +23,15 @@
using namespace llvm;
namespace {
typedef BOOLEAN (WINAPI *PtrCreateSymbolicLinkW)(
/*__in*/ LPCWSTR lpSymlinkFileName,
/*__in*/ LPCWSTR lpTargetFileName,
/*__in*/ DWORD dwFlags);
PtrCreateSymbolicLinkW create_symbolic_link_api = PtrCreateSymbolicLinkW(
::GetProcAddress(::GetModuleHandleA("kernel32.dll"),
"CreateSymbolicLinkW"));
error_code UTF8ToUTF16(const StringRef &utf8,
SmallVectorImpl<wchar_t> &utf16) {
int len = ::MultiByteToWideChar(CP_UTF8, MB_ERR_INVALID_CHARS,
@ -220,6 +229,29 @@ error_code create_hard_link(const Twine &to, const Twine &from) {
return make_error_code(errc::success);
}
error_code create_symlink(const Twine &to, const Twine &from) {
// Only do it if the function is available at runtime.
if (!create_symbolic_link_api)
return make_error_code(errc::function_not_supported);
// Get arguments.
SmallString<128> from_storage;
SmallString<128> to_storage;
StringRef f = from.toStringRef(from_storage);
StringRef t = to.toStringRef(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;
if (!create_symbolic_link_api(wide_from.begin(), wide_to.begin(), NULL))
return make_error_code(windows_error(::GetLastError()));
return make_error_code(errc::success);
}
error_code exists(const Twine &path, bool &result) {
SmallString<128> path_storage;
SmallVector<wchar_t, 128> path_utf16;