/**
 * @file fsbench.c
 *
 * @copyright 2015-2016 Bill Zissimopoulos
 */
/*
 * This file is part of WinFsp.
 *
 * You can redistribute it and/or modify it under the terms of the GNU
 * General Public License version 3 as published by the Free Software
 * Foundation.
 *
 * Licensees holding a valid commercial license may use this file in
 * accordance with the commercial license agreement provided with the
 * software.
 */

#include <windows.h>
#include <strsafe.h>
#include <tlib/testsuite.h>

ULONG OptFiles = 1000;
ULONG OptDirs = 1000;
ULONG OptRdwr = 10000;

static void file_create_test(void)
{
    HANDLE Handle;
    BOOL Success;
    WCHAR FileName[MAX_PATH];

    for (ULONG Index = 0; OptFiles > Index; Index++)
    {
        StringCbPrintfW(FileName, sizeof FileName, L"fsbench-file%lu", Index);
        Handle = CreateFileW(FileName,
            GENERIC_READ | GENERIC_WRITE, FILE_SHARE_READ | FILE_SHARE_WRITE,
            0,
            CREATE_NEW, FILE_ATTRIBUTE_NORMAL,
            0);
        ASSERT(INVALID_HANDLE_VALUE != Handle);
        Success = CloseHandle(Handle);
        ASSERT(Success);
    }
}
static void file_delete_test(void)
{
    BOOL Success;
    WCHAR FileName[MAX_PATH];

    for (ULONG Index = 0; OptFiles > Index; Index++)
    {
        StringCbPrintfW(FileName, sizeof FileName, L"fsbench-file%lu", Index);
        Success = DeleteFileW(FileName);
        ASSERT(Success);
    }
}
static void file_tests(void)
{
    TEST(file_create_test);
    TEST(file_delete_test);
}

static void dir_mkdir_test(void)
{
    BOOL Success;
    WCHAR FileName[MAX_PATH];

    for (ULONG Index = 0; OptDirs > Index; Index++)
    {
        StringCbPrintfW(FileName, sizeof FileName, L"fsbench-dir%lu", Index);
        Success = CreateDirectoryW(FileName, 0);
        ASSERT(Success);
    }
}
static void dir_rmdir_test(void)
{
    BOOL Success;
    WCHAR FileName[MAX_PATH];

    for (ULONG Index = 0; OptDirs > Index; Index++)
    {
        StringCbPrintfW(FileName, sizeof FileName, L"fsbench-dir%lu", Index);
        Success = RemoveDirectoryW(FileName);
        ASSERT(Success);
    }
}
static void dir_tests(void)
{
    TEST(dir_mkdir_test);
    TEST(dir_rmdir_test);
}

static void rdwr_dotest(ULONG CreateDisposition, ULONG CreateFlags)
{
    SYSTEM_INFO SystemInfo;
    PVOID Buffer;
    HANDLE Handle;
    BOOL Success;
    DWORD BytesTransferred;
    WCHAR FileName[MAX_PATH];

    GetSystemInfo(&SystemInfo);
    Buffer = _aligned_malloc(SystemInfo.dwPageSize, SystemInfo.dwPageSize);
    ASSERT(0 != Buffer);
    memset(Buffer, 0, SystemInfo.dwPageSize);

    StringCbPrintfW(FileName, sizeof FileName, L"fsbench-file");
    Handle = CreateFileW(FileName,
        GENERIC_READ | GENERIC_WRITE, FILE_SHARE_READ | FILE_SHARE_WRITE,
        0,
        CreateDisposition,
        FILE_ATTRIBUTE_NORMAL | CreateFlags,
        0);
    ASSERT(INVALID_HANDLE_VALUE != Handle);

    for (ULONG Index = 0; OptRdwr > Index; Index++)
    {
        if (CREATE_NEW == CreateDisposition)
            Success = WriteFile(Handle, Buffer, SystemInfo.dwPageSize, &BytesTransferred, 0);
        else
            Success = ReadFile(Handle, Buffer, SystemInfo.dwPageSize, &BytesTransferred, 0);
        ASSERT(Success);
        ASSERT(SystemInfo.dwPageSize == BytesTransferred);
    }

    Success = CloseHandle(Handle);
    ASSERT(Success);

    _aligned_free(Buffer);
}
static void rdwr_cached_write_test(void)
{
    rdwr_dotest(CREATE_NEW, 0);
}
static void rdwr_cached_read_test(void)
{
    rdwr_dotest(OPEN_EXISTING, FILE_FLAG_DELETE_ON_CLOSE);
}
static void rdwr_noncached_write_test(void)
{
    rdwr_dotest(CREATE_NEW, 0 | FILE_FLAG_NO_BUFFERING);
}
static void rdwr_noncached_read_test(void)
{
    rdwr_dotest(OPEN_EXISTING, FILE_FLAG_DELETE_ON_CLOSE | FILE_FLAG_NO_BUFFERING);
}
static void rdwr_tests(void)
{
    TEST(rdwr_cached_write_test);
    TEST(rdwr_cached_read_test);
    TEST(rdwr_noncached_write_test);
    TEST(rdwr_noncached_read_test);
}

static void mmap_cached_test(void)
{
}
static void mmap_noncached_test(void)
{
}
static void mmap_tests(void)
{
    TEST(mmap_cached_test);
    TEST(mmap_noncached_test);
}

#define rmarg(argv, argc, argi)         \
    argc--,                             \
    memmove(argv + argi, argv + argi + 1, (argc - argi) * sizeof(char *)),\
    argi--,                             \
    argv[argc] = 0
int main(int argc, char *argv[])
{
    TESTSUITE(file_tests);
    TESTSUITE(dir_tests);
    TESTSUITE(rdwr_tests);
    TESTSUITE(mmap_tests);

    for (int argi = 1; argc > argi; argi++)
    {
        const char *a = argv[argi];
        if ('-' == a[0])
        {
            if (0 == strncmp("--files=", a, sizeof "--files=" - 1))
            {
                OptFiles = strtoul(a + sizeof "--files=" - 1, 0, 10);
                rmarg(argv, argc, argi);
            }
            else if (0 == strncmp("--dirs=", a, sizeof "--dirs=" - 1))
            {
                OptDirs = strtoul(a + sizeof "--dirs=" - 1, 0, 10);
                rmarg(argv, argc, argi);
            }
            else if (0 == strncmp("--rdwr=", a, sizeof "--rdwr=" - 1))
            {
                OptRdwr = strtoul(a + sizeof "--rdwr=" - 1, 0, 10);
                rmarg(argv, argc, argi);
            }
        }
    }

    tlib_run_tests(argc, argv);
    return 0;
}