// 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"


// Exit macros
#define PathExitOnLastError(x, s, ...) ExitOnLastErrorSource(DUTIL_SOURCE_PATHUTIL, x, s, __VA_ARGS__)
#define PathExitOnLastErrorDebugTrace(x, s, ...) ExitOnLastErrorDebugTraceSource(DUTIL_SOURCE_PATHUTIL, x, s, __VA_ARGS__)
#define PathExitWithLastError(x, s, ...) ExitWithLastErrorSource(DUTIL_SOURCE_PATHUTIL, x, s, __VA_ARGS__)
#define PathExitOnFailure(x, s, ...) ExitOnFailureSource(DUTIL_SOURCE_PATHUTIL, x, s, __VA_ARGS__)
#define PathExitOnRootFailure(x, s, ...) ExitOnRootFailureSource(DUTIL_SOURCE_PATHUTIL, x, s, __VA_ARGS__)
#define PathExitWithRootFailure(x, e, s, ...) ExitWithRootFailureSource(DUTIL_SOURCE_PATHUTIL, x, e, s, __VA_ARGS__)
#define PathExitOnFailureDebugTrace(x, s, ...) ExitOnFailureDebugTraceSource(DUTIL_SOURCE_PATHUTIL, x, s, __VA_ARGS__)
#define PathExitOnNull(p, x, e, s, ...) ExitOnNullSource(DUTIL_SOURCE_PATHUTIL, p, x, e, s, __VA_ARGS__)
#define PathExitOnNullWithLastError(p, x, s, ...) ExitOnNullWithLastErrorSource(DUTIL_SOURCE_PATHUTIL, p, x, s, __VA_ARGS__)
#define PathExitOnNullDebugTrace(p, x, e, s, ...)  ExitOnNullDebugTraceSource(DUTIL_SOURCE_PATHUTIL, p, x, e, s, __VA_ARGS__)
#define PathExitOnInvalidHandleWithLastError(p, x, s, ...) ExitOnInvalidHandleWithLastErrorSource(DUTIL_SOURCE_PATHUTIL, p, x, s, __VA_ARGS__)
#define PathExitOnWin32Error(e, x, s, ...) ExitOnWin32ErrorSource(DUTIL_SOURCE_PATHUTIL, e, x, s, __VA_ARGS__)
#define PathExitOnGdipFailure(g, x, s, ...) ExitOnGdipFailureSource(DUTIL_SOURCE_PATHUTIL, g, x, s, __VA_ARGS__)


typedef HRESULT(WINAPI* PFN_PATH_ALLOC_CANONICALIZE)(
    __in LPCWSTR wzSource,
    __in DWORD dwFlags,
    __out_z LPWSTR* psczPathOut
    );

static BOOL vfInitialized = FALSE;
static HMODULE vhPathApiSet_1_1_0 = NULL;
static PFN_PATH_ALLOC_CANONICALIZE vpfnPathAllocCanonicalize = NULL;
static BOOL vfForceFallback = FALSE;

// from PathCch.h
#ifndef PATHCCH_ALLOW_LONG_PATHS
#define PATHCCH_ALLOW_LONG_PATHS       0x01
#endif

static HRESULT Initialize()
{
    HRESULT hr = S_OK;
    DWORD er = ERROR_SUCCESS;

    if (vfInitialized)
    {
        ExitFunction();
    }

    hr = LoadSystemApiSet(L"api-ms-win-core-path-l1-1-0.dll", &vhPathApiSet_1_1_0);
    if (E_MODNOTFOUND == hr)
    {
        hr = E_NOTIMPL;
    }
    PathExitOnFailure(hr, "Failed to load api-ms-win-core-path-l1-1-0.dll");

    vpfnPathAllocCanonicalize = reinterpret_cast<PFN_PATH_ALLOC_CANONICALIZE>(::GetProcAddress(vhPathApiSet_1_1_0, "PathAllocCanonicalize"));
    if (!vpfnPathAllocCanonicalize)
    {
        er = ::GetLastError();
        PathExitWithRootFailure(hr, ERROR_PROC_NOT_FOUND == er ? E_NOTIMPL : HRESULT_FROM_WIN32(er), "Failed to get address of PathAllocCanonicalize.");
    }

    vfInitialized = TRUE;

LExit:
    return hr;
}


DAPI_(void) Path2FunctionAllowFallback()
{
    vfForceFallback = FALSE;
}


DAPI_(void) Path2FunctionForceFallback()
{
    vfForceFallback = TRUE;
}


DAPI_(HRESULT) PathCanonicalizePath(
    __in_z LPCWSTR wzPath,
    __deref_out_z LPWSTR* psczCanonicalized
    )
{
    HRESULT hr = S_OK;
    int cch = MAX_PATH + 1;

    hr = StrAlloc(psczCanonicalized, cch);
    PathExitOnFailure(hr, "Failed to allocate string for the canonicalized path.");

    if (::PathCanonicalizeW(*psczCanonicalized, wzPath))
    {
        hr = S_OK;
    }
    else
    {
        ExitFunctionWithLastError(hr);
    }

LExit:
    return hr;
}

DAPI_(HRESULT) PathAllocCanonicalizePath(
    __in_z LPCWSTR wzPath,
    __in DWORD dwFlags,
    __deref_out_z LPWSTR* psczCanonicalized
    )
{
    HRESULT hr = S_OK;
    LPWSTR sczCanonicalizedPath = NULL;

    hr = Initialize();
    if (E_NOTIMPL == hr || SUCCEEDED(hr) && !vpfnPathAllocCanonicalize)
    {
        ExitFunction1(hr = E_NOTIMPL);
    }
    PathExitOnFailure(hr, "Failed to initialize path2utl.");

    hr = vpfnPathAllocCanonicalize(wzPath, dwFlags, &sczCanonicalizedPath);
    PathExitOnFailure(hr, "Failed to canonicalize: %ls", wzPath);

    hr = StrAllocString(psczCanonicalized, sczCanonicalizedPath, 0);
    PathExitOnFailure(hr, "Failed to copy the canonicalized path.");

LExit:
    if (sczCanonicalizedPath)
    {
        ::LocalFree(sczCanonicalizedPath);
    }

    return hr;
}

DAPI_(HRESULT) PathCanonicalizeForComparison(
    __in_z LPCWSTR wzPath,
    __in DWORD dwCanonicalizeFlags,
    __deref_out_z LPWSTR* psczCanonicalized
    )
{
    HRESULT hr = S_OK;
    LPWSTR sczNormalizedPath = NULL;
    LPCWSTR wzNormalizedPath = NULL;
    SIZE_T cchUncRootLength = 0;
    BOOL fHasPrefix = FALSE;

    hr = StrAllocString(&sczNormalizedPath, wzPath, 0);
    PathExitOnFailure(hr, "Failed to allocate string for the normalized path.");

    PathFixedNormalizeSlashes(sczNormalizedPath);

    wzNormalizedPath = sczNormalizedPath;

    if (PATH_CANONICALIZE_KEEP_UNC_ROOT & dwCanonicalizeFlags)
    {
        BOOL fUNC = FALSE;
        LPCWSTR wzPastRoot = PathSkipPastRoot(sczNormalizedPath, NULL, NULL, &fUNC);
        if (fUNC)
        {
            wzNormalizedPath = wzPastRoot;
            cchUncRootLength = wzPastRoot - sczNormalizedPath;
        }
    }

    if (*wzNormalizedPath)
    {
        if (!vfForceFallback)
        {
            hr = PathAllocCanonicalizePath(wzNormalizedPath, PATHCCH_ALLOW_LONG_PATHS, psczCanonicalized);
        }
        else
        {
            hr = E_NOTIMPL;
        }

        if (E_NOTIMPL == hr)
        {
            hr = PathCanonicalizePath(wzNormalizedPath, psczCanonicalized);
        }
        PathExitOnFailure(hr, "Failed to canonicalize: %ls", wzNormalizedPath);
    }
    else
    {
        Assert(cchUncRootLength);
        ReleaseStr(*psczCanonicalized);
        *psczCanonicalized = sczNormalizedPath;
        sczNormalizedPath = NULL;
        cchUncRootLength = 0;
    }

    if (cchUncRootLength)
    {
        hr = StrAllocPrefix(psczCanonicalized, sczNormalizedPath, cchUncRootLength);
        PathExitOnFailure(hr, "Failed to prefix the UNC root to the canonicalized path.");
    }

    if (PATH_CANONICALIZE_BACKSLASH_TERMINATE & dwCanonicalizeFlags)
    {
        hr = PathBackslashTerminate(psczCanonicalized);
        PathExitOnFailure(hr, "Failed to backslash terminate the canonicalized path.");
    }

    if (PATH_CANONICALIZE_APPEND_EXTENDED_PATH_PREFIX & dwCanonicalizeFlags)
    {
        hr = PathPrefix(psczCanonicalized, 0, PATH_PREFIX_SHORT_PATHS);
        PathExitOnFailure(hr, "Failed to ensure the extended path prefix on the canonicalized path.");
    }

    PathSkipPastRoot(*psczCanonicalized, &fHasPrefix, NULL, NULL);

    if (fHasPrefix)
    {
        // Canonicalize prefix into \\?\.
        (*psczCanonicalized)[0] = L'\\';
        (*psczCanonicalized)[1] = L'\\';
        (*psczCanonicalized)[2] = L'?';
        (*psczCanonicalized)[3] = L'\\';
    }

LExit:
    ReleaseStr(sczNormalizedPath);

    return hr;
}

DAPI_(HRESULT) PathConcatRelativeToBase(
    __in LPCWSTR wzBase,
    __in_opt LPCWSTR wzRelative,
    __deref_out_z LPWSTR* psczCombined
    )
{
    HRESULT hr = S_OK;
    LPWSTR sczCanonicalizedRelative = NULL;

    if (!wzBase || !*wzBase)
    {
        PathExitWithRootFailure(hr, E_INVALIDARG, "wzBase is required.");
    }

    if (PathIsRooted(wzRelative))
    {
        PathExitWithRootFailure(hr, E_INVALIDARG, "wzRelative cannot be rooted.");
    }

    hr = StrAllocString(psczCombined, wzBase, 0);
    PathExitOnFailure(hr, "Failed to copy base to output.");

    if (wzRelative && *wzRelative)
    {
        hr = PathBackslashTerminate(psczCombined);
        PathExitOnFailure(hr, "Failed to backslashify.");

        hr = PathCanonicalizeForComparison(wzRelative, 0, &sczCanonicalizedRelative);
        PathExitOnFailure(hr, "Failed to canonicalize wzRelative.");

        hr = StrAllocConcat(psczCombined, sczCanonicalizedRelative, 0);
        PathExitOnFailure(hr, "Failed to append relative to output.");
    }

LExit:
    ReleaseStr(sczCanonicalizedRelative);

    return hr;
}

DAPI_(HRESULT) PathConcatRelativeToFullyQualifiedBase(
    __in LPCWSTR wzBase,
    __in_opt LPCWSTR wzRelative,
    __deref_out_z LPWSTR* psczCombined
    )
{
    HRESULT hr = S_OK;

    if (!PathIsFullyQualified(wzBase))
    {
        PathExitWithRootFailure(hr, E_INVALIDARG, "wzBase must be fully qualified: %ls.", wzBase);
    }

    hr = PathConcatRelativeToBase(wzBase, wzRelative, psczCombined);

LExit:
    return hr;
}

DAPI_(HRESULT) PathCompareCanonicalized(
    __in_z LPCWSTR wzPath1,
    __in_z LPCWSTR wzPath2,
    __out BOOL* pfEqual
    )
{
    HRESULT hr = S_OK;
    LPWSTR sczCanonicalized1 = NULL;
    LPWSTR sczCanonicalized2 = NULL;
    DWORD dwDefaultFlags = PATH_CANONICALIZE_APPEND_EXTENDED_PATH_PREFIX | PATH_CANONICALIZE_KEEP_UNC_ROOT;
    int nResult = 0;

    if (!wzPath1 || !wzPath2)
    {
        PathExitWithRootFailure(hr, E_INVALIDARG, "Both paths are required.");
    }

    hr = PathCanonicalizeForComparison(wzPath1, dwDefaultFlags, &sczCanonicalized1);
    PathExitOnFailure(hr, "Failed to canonicalize wzPath1.");

    hr = PathCanonicalizeForComparison(wzPath2, dwDefaultFlags, &sczCanonicalized2);
    PathExitOnFailure(hr, "Failed to canonicalize wzPath2.");

    nResult = ::CompareStringW(LOCALE_NEUTRAL, NORM_IGNORECASE, sczCanonicalized1, -1, sczCanonicalized2, -1);
    PathExitOnNullWithLastError(nResult, hr, "Failed to compare canonicalized paths.");

    *pfEqual = CSTR_EQUAL == nResult;

LExit:
    ReleaseStr(sczCanonicalized1);
    ReleaseStr(sczCanonicalized2);
    return hr;
}

DAPI_(HRESULT) PathDirectoryContainsPath(
    __in_z LPCWSTR wzDirectory,
    __in_z LPCWSTR wzPath
    )
{
    HRESULT hr = S_OK;
    LPWSTR sczCanonicalizedDirectory = NULL;
    LPWSTR sczCanonicalizedPath = NULL;
    DWORD dwDefaultFlags = PATH_CANONICALIZE_APPEND_EXTENDED_PATH_PREFIX | PATH_CANONICALIZE_KEEP_UNC_ROOT;
    size_t cchDirectory = 0;
    size_t cchPath = 0;

    if (!wzDirectory || !*wzDirectory)
    {
        PathExitWithRootFailure(hr, E_INVALIDARG, "wzDirectory is required.");
    }
    if (!wzPath || !*wzPath)
    {
        PathExitWithRootFailure(hr, E_INVALIDARG, "wzPath is required.");
    }

    hr = PathCanonicalizeForComparison(wzDirectory, dwDefaultFlags | PATH_CANONICALIZE_BACKSLASH_TERMINATE, &sczCanonicalizedDirectory);
    PathExitOnFailure(hr, "Failed to canonicalize the directory.");

    hr = PathCanonicalizeForComparison(wzPath, dwDefaultFlags, &sczCanonicalizedPath);
    PathExitOnFailure(hr, "Failed to canonicalize the path.");

    if (!PathIsFullyQualified(sczCanonicalizedDirectory))
    {
        PathExitWithRootFailure(hr, E_INVALIDARG, "wzDirectory must be a fully qualified path.");
    }
    if (!sczCanonicalizedPath || !*sczCanonicalizedPath)
    {
        ExitFunction1(hr = S_FALSE);
    }

    hr = ::StringCchLengthW(sczCanonicalizedDirectory, STRSAFE_MAX_CCH, &cchDirectory);
    PathExitOnFailure(hr, "Failed to get length of canonicalized directory.");

    hr = ::StringCchLengthW(sczCanonicalizedPath, STRSAFE_MAX_CCH, &cchPath);
    PathExitOnFailure(hr, "Failed to get length of canonicalized path.");

    if (cchPath <= cchDirectory)
    {
        ExitFunction1(hr = S_FALSE);
    }

    if (CSTR_EQUAL != ::CompareStringW(LOCALE_NEUTRAL, NORM_IGNORECASE, sczCanonicalizedDirectory, (DWORD)cchDirectory, sczCanonicalizedPath, (DWORD)cchDirectory))
    {
        ExitFunction1(hr = S_FALSE);
    }

    hr = S_OK;

LExit:
    ReleaseStr(sczCanonicalizedPath);
    ReleaseStr(sczCanonicalizedDirectory);
    return hr;
}


DAPI_(HRESULT) PathSystemWindowsSubdirectory(
    __in_z_opt LPCWSTR wzSubdirectory,
    __out_z LPWSTR* psczFullPath
    )
{
    HRESULT hr = S_OK;
    LPWSTR sczWindowsPath = NULL;
    DWORD cchBuffer = MAX_PATH + 1;
    DWORD cch = 0;

    hr = StrAlloc(&sczWindowsPath, cchBuffer);
    PathExitOnFailure(hr, "Failed to alloc Windows directory path.");

    cch = ::GetSystemWindowsDirectoryW(sczWindowsPath, cchBuffer);
    PathExitOnNullWithLastError(cch, hr, "Failed to get Windows directory path with default size.");

    cch += 1; // add 1 for null terminator.

    if (cch > cchBuffer)
    {
        hr = StrAlloc(&sczWindowsPath, cch);
        PathExitOnFailure(hr, "Failed to realloc Windows directory path.");

        cchBuffer = cch;

        cch = ::GetSystemWindowsDirectoryW(sczWindowsPath, cchBuffer);
        PathExitOnNullWithLastError(cch, hr, "Failed to get Windows directory path with returned size.");

        cch += 1; // add 1 for null terminator.

        if (cch > cchBuffer)
        {
            PathExitWithRootFailure(hr, E_INSUFFICIENT_BUFFER, "Failed to get Windows directory path with returned size.");
        }
    }

    if (wzSubdirectory)
    {
        hr = PathConcatRelativeToBase(sczWindowsPath, wzSubdirectory, psczFullPath);
        PathExitOnFailure(hr, "Failed to concat subdirectory on Windows directory path.");
    }
    else
    {
        *psczFullPath = sczWindowsPath;
        sczWindowsPath = NULL;
    }

    hr = PathBackslashTerminate(psczFullPath);
    PathExitOnFailure(hr, "Failed to terminate Windows directory path with backslash.");

LExit:
    ReleaseStr(sczWindowsPath);

    return hr;
}