1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132
|
/* Copyright (C) 2001-2021 Artifex Software, Inc.
All Rights Reserved.
This software is provided AS-IS with no warranty, either express or
implied.
This software is distributed under license and may not be copied,
modified or distributed except as expressly authorized under the terms
of the license contained in the file LICENSE in this distribution.
Refer to licensing information at http://www.artifex.com or contact
Artifex Software, Inc., 1305 Grant Avenue - Suite 200, Novato,
CA 94945, U.S.A., +1(415)492-9861, for further information.
*/
#include "windows_.h"
#include <string>
#include <sstream>
#include <windows.storage.h>
#include <wrl/client.h>
#include <wrl/wrappers/corewrappers.h>
extern "C" DWORD GetTempPathWRT(DWORD nBufferLength, LPWSTR lpBuffer)
{
try
{
HRESULT hr;
Microsoft::WRL::ComPtr<ABI::Windows::Storage::IApplicationDataStatics> applicationDataStatics;
hr = Windows::Foundation::GetActivationFactory(Microsoft::WRL::Wrappers::HStringReference(
RuntimeClass_Windows_Storage_ApplicationData).Get(), &applicationDataStatics);
if (FAILED(hr))
return 0;
Microsoft::WRL::ComPtr<ABI::Windows::Storage::IApplicationData> applicationData;
hr = applicationDataStatics->get_Current(&applicationData);
if (FAILED(hr))
return 0;
Microsoft::WRL::ComPtr<ABI::Windows::Storage::IStorageFolder> storageFolder;
hr = applicationData->get_TemporaryFolder(&storageFolder);
if (FAILED(hr))
return 0;
Microsoft::WRL::ComPtr<ABI::Windows::Storage::IStorageItem> storageItem;
hr = storageFolder.As(&storageItem);
if (FAILED(hr))
return 0;
HSTRING folderName;
hr = storageItem->get_Path(&folderName);
if (FAILED(hr))
return 0;
UINT32 length;
PCWSTR value = WindowsGetStringRawBuffer(folderName, &length);
std::wstring ws(value);
wcsncpy(lpBuffer, ws.c_str(), nBufferLength);
lpBuffer[nBufferLength-1] = 0;
WindowsDeleteString(folderName);
return ws.length();
}
catch(...)
{
return 0;
}
}
extern "C" UINT GetTempFileNameWRT(LPCWSTR lpPathName, LPCWSTR lpPrefixString, LPWSTR lpTempFileName)
{
try
{
std::wstring path(lpPathName);
std::wstring prefix(lpPrefixString);
FILETIME systemTimeAsFileTime;
if (!path.empty() && path.back() != '\\')
path.push_back('\\');
if (path.length() > _MAX_PATH - 14)
return ERROR_BUFFER_OVERFLOW;
GetSystemTimeAsFileTime(&systemTimeAsFileTime);
DWORD time = systemTimeAsFileTime.dwLowDateTime;
while(true)
{
// Create file name of at most 13 characters, using at most 10
// digits of time, and as many of the prefix characters as can fit
std::wstring num(std::to_wstring((UINT)time));
if (num.length() > 10)
num = num.substr(num.length() - 10);
std::wstring pf(prefix);
if (num.length() + pf.length() > 13)
pf.resize(13 - num.length());
std::wstring fullpath = path+pf+num;
// Test that the file doesn't already exist
LPWIN32_FIND_DATA find_data;
HANDLE h = FindFirstFileExW(fullpath.c_str(), FindExInfoStandard, &find_data, FindExSearchNameMatch, NULL, 0);
if (h == INVALID_HANDLE_VALUE)
{
// File doesn't exist
wcscpy(lpTempFileName, fullpath.c_str());
return fullpath.length();
}
// File exists. Increment time and try again
FindClose(h);
time++;
}
return ERROR_BUFFER_OVERFLOW;
}
catch(...)
{
return ERROR_BUFFER_OVERFLOW;
}
}
extern "C" void OutputDebugStringWRT(LPCSTR str, DWORD len)
{
try
{
std::string s(str, len);
std::wstring w(s.begin(), s.end());
OutputDebugStringW(w.c_str());
}
catch(...)
{
}
}
|