/*
 * Copyright (c) 2017-2022 The Khronos Group Inc.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 *
 * OpenCL is a trademark of Apple Inc. used under license by Khronos.
 */

#include <icd.h>
#include "icd_windows_apppackage.h"

typedef _Check_return_ LONG(WINAPI *PFN_GetPackagesByPackageFamily)(
    _In_ PCWSTR packageFamilyName,
    _Inout_ UINT32* count,
    _Out_writes_opt_(*count) PWSTR* packageFullNames,
    _Inout_ UINT32* bufferLength,
    _Out_writes_opt_(*bufferLength) WCHAR* buffer
    );

typedef LONG (WINAPI *PFN_GetPackagePathByFullName)(
    _In_ PCWSTR packageFullName,
    _Inout_ UINT32* pathLength,
    _Out_writes_opt_(*pathLength) PWSTR path
    );

bool khrIcdOsVendorsEnumerateAppPackage(void)
{
    bool ret = false;
    WCHAR *buffer = NULL;
    PWSTR *packages = NULL;

    HMODULE h = LoadLibraryA("kernel32.dll");
    if (h == NULL)
        return ret;

    PFN_GetPackagesByPackageFamily pGetPackagesByPackageFamily =
        (PFN_GetPackagesByPackageFamily)GetProcAddress(h, "GetPackagesByPackageFamily");
    if (!pGetPackagesByPackageFamily)
    {
        KHR_ICD_TRACE("GetProcAddress failed for GetPackagesByPackageFamily\n");
        goto cleanup;
    }

    PFN_GetPackagePathByFullName pGetPackagePathByFullName =
        (PFN_GetPackagePathByFullName)GetProcAddress(h, "GetPackagePathByFullName");
    if (!pGetPackagePathByFullName)
    {
        KHR_ICD_TRACE("GetProcAddress failed for GetPackagePathByFullName\n");
        goto cleanup;
    }

    UINT32 numPackages = 0, bufferLength = 0;
    PCWSTR familyName = L"Microsoft.D3DMappingLayers_8wekyb3d8bbwe";
    if (ERROR_INSUFFICIENT_BUFFER != pGetPackagesByPackageFamily(familyName,
                                                                 &numPackages, NULL,
                                                                 &bufferLength, NULL) ||
        numPackages == 0 || bufferLength == 0)
    {
        KHR_ICD_TRACE("Failed to find mapping layers packages by family name\n");
        goto cleanup;
    }

    buffer = malloc(sizeof(WCHAR) * bufferLength);
    packages = malloc(sizeof(PWSTR) * numPackages);
    if (!buffer || !packages)
    {
        KHR_ICD_TRACE("Failed to allocate memory for package names\n");
        goto cleanup;
    }

    if (ERROR_SUCCESS != pGetPackagesByPackageFamily(familyName,
                                                     &numPackages, packages,
                                                     &bufferLength, buffer))
    {
        KHR_ICD_TRACE("Failed to get mapping layers package full names\n");
        goto cleanup;
    }

    UINT32 pathLength = 0;
    WCHAR path[MAX_PATH];
    if (ERROR_INSUFFICIENT_BUFFER != pGetPackagePathByFullName(packages[0], &pathLength, NULL) ||
        pathLength > MAX_PATH ||
        ERROR_SUCCESS != pGetPackagePathByFullName(packages[0], &pathLength, path))
    {
        KHR_ICD_TRACE("Failed to get mapping layers package path length\n");
        goto cleanup;
    }

#if defined(_M_AMD64)
#define PLATFORM_PATH L"x64"
#elif defined(_M_ARM)
#define PLATFORM_PATH L"arm"
#elif defined(_M_ARM64)
#define PLATFORM_PATH L"arm64"
#elif defined(_M_IX86)
#define PLATFORM_PATH L"x86"
#endif

    wchar_t dllPath[MAX_PATH];
    wcscpy_s(dllPath, MAX_PATH, path);
    wcscat_s(dllPath, MAX_PATH, L"\\" PLATFORM_PATH L"\\OpenCLOn12.dll");

    char narrowDllPath[MAX_PATH];
    WideCharToMultiByte(CP_UTF8, 0, dllPath, -1, narrowDllPath, MAX_PATH, NULL, NULL);

    ret = adapterAdd(narrowDllPath, ZeroLuid);

cleanup:
    FreeLibrary(h);
    free(buffer);
    free(packages);
    return ret;
}