83 lines
2.8 KiB
C
83 lines
2.8 KiB
C
/* SPDX-License-Identifier: GPL-2.0
|
|
*
|
|
* Copyright (C) 2018-2020 WireGuard LLC. All Rights Reserved.
|
|
*/
|
|
|
|
#include "pch.h"
|
|
|
|
/**
|
|
* Locates RT_RCDATA resource memory address and size.
|
|
*
|
|
* ResourceName Name of the RT_RCDATA resource. Use MAKEINTRESOURCEW to locate resource by ID.
|
|
*
|
|
* Address Pointer to a pointer variable to receive resource address.
|
|
*
|
|
* Size Pointer to a variable to receive resource size.
|
|
*
|
|
* @return ERROR_SUCCESS on success; Win32 error code otherwise.
|
|
*/
|
|
WINTUN_STATUS
|
|
ResourceGetAddress(_In_z_ const WCHAR *ResourceName, _Out_ const VOID **Address, _Out_ DWORD *Size)
|
|
{
|
|
HRSRC FoundResource = FindResourceW(ResourceModule, ResourceName, RT_RCDATA);
|
|
if (!FoundResource)
|
|
return WINTUN_LOGGER_LAST_ERROR(L"Failed to find resource");
|
|
*Size = SizeofResource(ResourceModule, FoundResource);
|
|
if (!*Size)
|
|
return WINTUN_LOGGER_LAST_ERROR(L"Failed to size resource");
|
|
HGLOBAL LoadedResource = LoadResource(ResourceModule, FoundResource);
|
|
if (!LoadedResource)
|
|
return WINTUN_LOGGER_LAST_ERROR(L"Failed to load resource");
|
|
*Address = LockResource(LoadedResource);
|
|
if (!*Address)
|
|
{
|
|
WINTUN_LOGGER(WINTUN_LOG_ERR, L"Failed to lock resource");
|
|
return ERROR_LOCK_FAILED;
|
|
}
|
|
return ERROR_SUCCESS;
|
|
}
|
|
|
|
/**
|
|
* Copies resource to a file.
|
|
*
|
|
* DestinationPath File path
|
|
*
|
|
* SecurityAttributes File security attributes. May be NULL for detault.
|
|
*
|
|
* ResourceName Name of the RT_RCDATA resource. Use MAKEINTRESOURCEW to locate resource by ID.
|
|
*
|
|
* @return ERROR_SUCCESS on success; Win32 error code otherwise.
|
|
*/
|
|
WINTUN_STATUS
|
|
ResourceCopyToFile(
|
|
_In_z_ const WCHAR *DestinationPath,
|
|
_In_opt_ SECURITY_ATTRIBUTES *SecurityAttributes,
|
|
_In_z_ const WCHAR *ResourceName)
|
|
{
|
|
const VOID *LockedResource;
|
|
DWORD SizeResource;
|
|
DWORD Result = ResourceGetAddress(ResourceName, &LockedResource, &SizeResource);
|
|
if (Result != ERROR_SUCCESS)
|
|
return WINTUN_LOGGER_ERROR("Failed to locate resource", Result);
|
|
HANDLE DestinationHandle = CreateFileW(
|
|
DestinationPath,
|
|
GENERIC_WRITE,
|
|
0,
|
|
SecurityAttributes,
|
|
CREATE_NEW,
|
|
FILE_ATTRIBUTE_NORMAL | FILE_ATTRIBUTE_TEMPORARY,
|
|
NULL);
|
|
if (DestinationHandle == INVALID_HANDLE_VALUE)
|
|
return WINTUN_LOGGER_LAST_ERROR(L"Failed to create file");
|
|
DWORD BytesWritten;
|
|
if (!WriteFile(DestinationHandle, LockedResource, SizeResource, &BytesWritten, NULL))
|
|
Result = WINTUN_LOGGER_LAST_ERROR(L"Failed to write file");
|
|
if (BytesWritten != SizeResource)
|
|
{
|
|
WINTUN_LOGGER(WINTUN_LOG_ERR, L"Incomplete write");
|
|
Result = Result != ERROR_SUCCESS ? Result : ERROR_WRITE_FAULT;
|
|
}
|
|
CloseHandle(DestinationHandle);
|
|
return Result;
|
|
}
|