[Support/FileSystem] Introduce llvm::sys::fs::create_symbolic_link().

git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@203136 91177308-0d34-0410-b5e6-96231b3b80d8
Svengali · Mar 6, 2014 · bd3f9c0 · bd3f9c0
1 parent 161e3a8
commit bd3f9c0
Show file tree

Hide file tree

Showing 3 changed files with 40 additions and 0 deletions.
diff --git a/include/llvm/Support/FileSystem.h b/include/llvm/Support/FileSystem.h
@@ -293,6 +293,14 @@ error_code create_directory(const Twine &path, bool IgnoreExisting = true);
 ///          , otherwise a platform specific error_code.
 error_code create_hard_link(const Twine &to, const Twine &from);
 
+/// @brief Create a symbolic link from \a from to \a to.
+///
+/// @param to The path to link to.
+/// @param from The path to link from. This is created.
+/// @returns errc::success if successful
+///          , otherwise a platform specific error_code.
+error_code create_symbolic_link(const Twine &to, const Twine &from);
+
 /// @brief Get the current path.
 ///
 /// @param result Holds the current path on return.

diff --git a/lib/Support/Unix/Path.inc b/lib/Support/Unix/Path.inc
@@ -285,6 +285,19 @@ error_code create_hard_link(const Twine &to, const Twine &from) {
   return error_code::success();
 }
 
+error_code create_symbolic_link(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 error_code::success();
+}
+
 error_code remove(const Twine &path, bool IgnoreNonExisting) {
   SmallString<128> path_storage;
   StringRef p = path.toNullTerminatedStringRef(path_storage);

diff --git a/lib/Support/Windows/Path.inc b/lib/Support/Windows/Path.inc
@@ -177,6 +177,25 @@ error_code create_hard_link(const Twine &to, const Twine &from) {
   return error_code::success();
 }
 
+error_code create_symbolic_link(const Twine &to, const Twine &from) {
+  // 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 (!::CreateSymbolicLinkW(wide_from.begin(), wide_to.begin(), NULL))
+    return windows_error(::GetLastError());
+
+  return error_code::success();
+}
+
 error_code remove(const Twine &path, bool IgnoreNonExisting) {
   SmallString<128> path_storage;
   SmallVector<wchar_t, 128> path_utf16;