/*
 * PROJECT:         ReactOS api tests
 * LICENSE:         GPLv2+ - See COPYING in the top level directory
 * PURPOSE:         Test for RtlDoesFileExists_U*
 * PROGRAMMER:      Thomas Faber <thfabba@gmx.de>
 */

#define WIN32_NO_STATUS
#define UNICODE
#include <stdio.h>
#include <wine/test.h>
#include <pseh/pseh2.h>
#include <ndk/rtlfuncs.h>

#define StartSeh()              ExceptionStatus = STATUS_SUCCESS; _SEH2_TRY {
#define EndSeh(ExpectedStatus)  } _SEH2_EXCEPT(EXCEPTION_EXECUTE_HANDLER) { ExceptionStatus = _SEH2_GetExceptionCode(); } _SEH2_END; ok(ExceptionStatus == ExpectedStatus, "Exception %lx, expected %lx\n", ExceptionStatus, ExpectedStatus)

#define ok_bool_file(value, expected, file) do {                                \
        if (expected)                                                           \
            ok(value == TRUE, "File '%ls' should exist, but does not\n", file); \
        else                                                                    \
            ok(value == FALSE, "File '%ls' should not exist, but does\n", file);\
    } while (0)

#define ok_eq_ustr(str1, str2) do {                                                     \
        ok((str1)->Buffer        == (str2)->Buffer,        "Buffer modified\n");        \
        ok((str1)->Length        == (str2)->Length,        "Length modified\n");        \
        ok((str1)->MaximumLength == (str2)->MaximumLength, "MaximumLength modified\n"); \
    } while (0)

/*
BOOLEAN
NTAPI
RtlDoesFileExists_U(
    IN PCWSTR FileName
);

BOOLEAN
NTAPI
RtlDoesFileExists_UEx(
    IN PCWSTR FileName,
    IN BOOLEAN SucceedIfBusy
);

BOOLEAN
NTAPI
RtlDoesFileExists_UStr(
    IN PUNICODE_STRING FileName
);

BOOLEAN
NTAPI
RtlDoesFileExists_UstrEx(
        IN PCUNICODE_STRING FileName,
        IN BOOLEAN SucceedIfBusy
);
*/

static
BOOLEAN
(NTAPI
*RtlDoesFileExists_UEx)(
    IN PCWSTR FileName,
    IN BOOLEAN SucceedIfBusy
)
//= (PVOID)0x7c8187d0 // 2003 sp1 x86
//= (PVOID)0x7769aeb2 // win7 sp1 wow64
;

static
BOOLEAN
(NTAPI
*RtlDoesFileExists_UStr)(
    IN PUNICODE_STRING FileName
)
//= (PVOID)0x7c8474e5 // 2003 sp1 x86
//= (PVOID)0x776ff304 // win7 sp1 wow64
;

static
BOOLEAN
(NTAPI
*RtlDoesFileExists_UstrEx)(
        IN PCUNICODE_STRING FileName,
        IN BOOLEAN SucceedIfBusy
)
//= (PVOID)0x7c830f89 // 2003 sp1 x86
//= (PVOID)0x7769addb // win7 sp1 wow64
;

START_TEST(RtlDoesFileExists)
{
    NTSTATUS ExceptionStatus;
    BOOLEAN Ret;
    struct
    {
        PCWSTR FileName;
        BOOLEAN Exists;
    } Tests[] =
    {
        { L"",                                          FALSE },
        { L"?",                                         FALSE },
        { L"*",                                         FALSE },
        { L":",                                         FALSE },
        { L";",                                         FALSE },
        { L"\"",                                        FALSE },
        { L".",                                         TRUE },
        { L"..",                                        TRUE },
        { L"/",                                         TRUE },
        { L"//",                                        FALSE },
        { L"///",                                       FALSE },
        { L"\\/",                                       FALSE },
        { L"\\//",                                      FALSE },
        { L"\\\\/",                                     FALSE },
        { L"\\/\\",                                     FALSE },
        { L"\\/\\\\",                                   FALSE },
        { L"\\/\\/\\",                                  FALSE },
        { L"\\",                                        TRUE },
        { L"\\\\",                                      FALSE },
        { L"\\\\\\",                                    FALSE },
        { L"\\\\.",                                     FALSE },
        { L"\\\\.\\",                                   FALSE },
        { L"\\\\.\\GLOBAL??",                           FALSE },
        { L"\\\\.\\GLOBAL??\\",                         FALSE },
        { L"\\\\?",                                     FALSE },
        { L"\\\\??",                                    FALSE },
        { L"\\\\??\\",                                  FALSE },
        { L"\\\\??\\C:\\",                              FALSE },
        { L"\\\\.",                                     FALSE },
        { L"\\\\.\\",                                   FALSE },
        { L"C:",                                        TRUE },
        { L"C:/",                                       TRUE },
        { L"C:/\\",                                     TRUE },
        { L"C:\\/",                                     TRUE },
        { L"C:\\/\\",                                   TRUE },
        { L"C://",                                      TRUE },
        { L"C:\\",                                      TRUE },
        { L"C:\\\\",                                    TRUE },
        { L"C:\\%ls",                                   TRUE },
        { L"C:/%ls",                                    TRUE },
        { L"C://%ls",                                   TRUE },
        { L"C:\\/%ls",                                  TRUE },
        { L"C:/\\%ls",                                  TRUE },
        { L"C:\\/\\%ls",                                TRUE },
        { L"C:\\%ls\\",                                 TRUE },
        { L"C:\\%ls\\ThisFolderExists",                 TRUE },
        { L"C:\\%ls\\ThisDoesntExist",                  FALSE },
        { L"C:\\\\%ls\\\\ThisFolderExists",             TRUE },
        { L"C:\\%ls\\ThisFolderExists\\ThisFileExists", TRUE },
        { L"c:\\%ls\\thisfolderexists\\thisfileexists", TRUE },
        { L"C:\\%ls\\THISFOLDEREXISTS\\THISFILEEXISTS", TRUE },
        { L"C:\\%ls;C:\\",                              FALSE },
        { L"%%SystemRoot%%",                            FALSE },
        { L"%%SystemRoot%%\\",                          FALSE },
        { L"%%SystemRoot%%\\system32",                  FALSE },
        { L"NUL",                                       FALSE },
        { L"CON",                                       FALSE },
        { L"COM1",                                      FALSE },
    };
    ULONG i;
    WCHAR FileName[MAX_PATH];
    WCHAR CustomPath[MAX_PATH] = L"RtlDoesFileExists_U_TestPath";
    BOOL Success;
    HANDLE Handle;

    StartSeh()
        Ret = RtlDoesFileExists_U(NULL);
        ok(Ret == FALSE, "NULL file exists?!\n");
    EndSeh(STATUS_SUCCESS);

    swprintf(FileName, L"C:\\%ls", CustomPath);
    /* Make sure this directory doesn't exist */
    while (GetFileAttributes(FileName) != INVALID_FILE_ATTRIBUTES)
    {
        wcscat(CustomPath, L"X");
        swprintf(FileName, L"C:\\%ls", CustomPath);
    }
    Success = CreateDirectory(FileName, NULL);
    ok(Success, "CreateDirectory failed, results might not be accurate\n");
    swprintf(FileName, L"C:\\%ls\\ThisFolderExists", CustomPath);
    Success = CreateDirectory(FileName, NULL);
    ok(Success, "CreateDirectory failed, results might not be accurate\n");
    swprintf(FileName, L"C:\\%ls\\ThisFolderExists\\ThisFileExists", CustomPath);
    Handle = CreateFile(FileName, 0, 0, NULL, CREATE_NEW, 0, NULL);
    ok(Handle != INVALID_HANDLE_VALUE, "CreateFile failed, results might not be accurate\n");
    if (Handle != INVALID_HANDLE_VALUE)
    {
        /* Check SucceedIfBusy behavior */
        if (RtlDoesFileExists_UEx)
        {
            Ret = RtlDoesFileExists_UEx(FileName, TRUE);
            ok_bool_file(Ret, TRUE, FileName);
            /* TODO: apparently we have to do something worse to make this fail */
            Ret = RtlDoesFileExists_UEx(FileName, FALSE);
            ok_bool_file(Ret, TRUE, FileName);
        }
        if (RtlDoesFileExists_UstrEx)
        {
            UNICODE_STRING FileNameString;
            UNICODE_STRING TempString;
            RtlInitUnicodeString(&FileNameString, FileName);
            TempString = FileNameString;
            Ret = RtlDoesFileExists_UstrEx(&FileNameString, TRUE);
            ok_eq_ustr(&FileNameString, &TempString);
            ok_bool_file(Ret, TRUE, FileName);
            /* TODO: apparently we have to do something worse to make this fail */
            Ret = RtlDoesFileExists_UstrEx(&FileNameString, FALSE);
            ok_eq_ustr(&FileNameString, &TempString);
            ok_bool_file(Ret, TRUE, FileName);
        }
        CloseHandle(Handle);
    }

    for (i = 0; i < sizeof(Tests) / sizeof(Tests[0]); i++)
    {
        swprintf(FileName, Tests[i].FileName, CustomPath);
        StartSeh()
            Ret = RtlDoesFileExists_U(FileName);
            ok_bool_file(Ret, Tests[i].Exists, FileName);
        EndSeh(STATUS_SUCCESS);
        if (RtlDoesFileExists_UEx)
        {
            StartSeh()
                Ret = RtlDoesFileExists_UEx(FileName, TRUE);
                ok_bool_file(Ret, Tests[i].Exists, FileName);
            EndSeh(STATUS_SUCCESS);
            StartSeh()
                Ret = RtlDoesFileExists_UEx(FileName, FALSE);
                ok_bool_file(Ret, Tests[i].Exists, FileName);
            EndSeh(STATUS_SUCCESS);
        }
        /* TODO: use guarded memory to make sure these don't touch the null terminator */
        if (RtlDoesFileExists_UStr)
        {
            UNICODE_STRING FileNameString;
            UNICODE_STRING TempString;
            RtlInitUnicodeString(&FileNameString, FileName);
            TempString = FileNameString;
            StartSeh()
                Ret = RtlDoesFileExists_UStr(&FileNameString);
                ok_bool_file(Ret, Tests[i].Exists, FileName);
            EndSeh(STATUS_SUCCESS);
            ok_eq_ustr(&FileNameString, &TempString);
        }
        if (RtlDoesFileExists_UstrEx)
        {
            UNICODE_STRING FileNameString;
            UNICODE_STRING TempString;
            RtlInitUnicodeString(&FileNameString, FileName);
            TempString = FileNameString;
            StartSeh()
                Ret = RtlDoesFileExists_UstrEx(&FileNameString, TRUE);
                ok_bool_file(Ret, Tests[i].Exists, FileName);
            EndSeh(STATUS_SUCCESS);
            ok_eq_ustr(&FileNameString, &TempString);
            StartSeh()
                Ret = RtlDoesFileExists_UstrEx(&FileNameString, FALSE);
                ok_bool_file(Ret, Tests[i].Exists, FileName);
            EndSeh(STATUS_SUCCESS);
            ok_eq_ustr(&FileNameString, &TempString);
        }
    }

    swprintf(FileName, L"C:\\%ls\\ThisFolderExists\\ThisFileExists", CustomPath);
    Success = DeleteFile(FileName);
    ok(Success, "DeleteFile failed, test might leave stale file\n");
    swprintf(FileName, L"C:\\%ls\\ThisFolderExists", CustomPath);
    Success = RemoveDirectory(FileName);
    ok(Success, "RemoveDirectory failed, test might leave stale directory\n");
    swprintf(FileName, L"C:\\%ls", CustomPath);
    Success = RemoveDirectory(FileName);
    ok(Success, "RemoveDirectory failed, test might leave stale directory\n");
}