Initial commit

1 files changed, 54 insertions, 0 deletions

diff --git a/src/dutil/uncutil.cpp b/src/dutil/uncutil.cpp
new file mode 100644
index 00000000..6deb43bd
--- /dev/null
+++ b/src/dutil/uncutil.cpp
@@ -0,0 +1,54 @@
+// Copyright (c) .NET Foundation and contributors. All rights reserved. Licensed under the Microsoft Reciprocal License. See LICENSE.TXT file in the project root for full license information.
+
+#include "precomp.h"
+
+DAPI_(HRESULT) UncConvertFromMountedDrive(
+    __inout LPWSTR *psczUNCPath,
+    __in LPCWSTR sczMountedDrivePath
+    )
+{
+    HRESULT hr = S_OK;
+    DWORD dwLength = 0;
+    DWORD er = ERROR_SUCCESS;
+    LPWSTR sczDrive = NULL;
+
+    // Only copy drive letter and colon
+    hr = StrAllocString(&sczDrive, sczMountedDrivePath, 2);
+    ExitOnFailure(hr, "Failed to copy drive");
+
+    // ERROR_NOT_CONNECTED means it's not a mapped drive
+    er = ::WNetGetConnectionW(sczDrive, NULL, &dwLength);
+    if (ERROR_MORE_DATA == er)
+    {
+        er = ERROR_SUCCESS;
+
+        hr = StrAlloc(psczUNCPath, dwLength);
+        ExitOnFailure(hr, "Failed to allocate string to get raw UNC path of length %u", dwLength);
+
+        er = ::WNetGetConnectionW(sczDrive, *psczUNCPath, &dwLength);
+        if (ERROR_CONNECTION_UNAVAIL == er)
+        {
+            // This means the drive is remembered but not currently connected, this can mean the location is accessible via UNC path but not via mounted drive path
+            er = ERROR_SUCCESS;
+        }
+        ExitOnWin32Error(er, hr, "::WNetGetConnectionW() failed with buffer provided on drive %ls", sczDrive);
+
+        // Skip drive letter and colon
+        hr = StrAllocConcat(psczUNCPath, sczMountedDrivePath + 2, 0);
+        ExitOnFailure(hr, "Failed to copy rest of database path");
+    }
+    else
+    {
+        if (ERROR_SUCCESS == er)
+        {
+            er = ERROR_NO_DATA;
+        }
+
+        ExitOnWin32Error(er, hr, "::WNetGetConnectionW() failed on drive %ls", sczDrive);
+    }
+
+LExit:
+    ReleaseStr(sczDrive);
+
+    return hr;
+}