mirror of
https://github.com/vczh-libraries/Release.git
synced 2026-08-17 17:31:44 +08:00
4527 lines
116 KiB
C++
4527 lines
116 KiB
C++
/***********************************************************************
|
|
THIS FILE IS AUTOMATICALLY GENERATED. DO NOT MODIFY
|
|
DEVELOPER: Zihan Chen(vczh)
|
|
***********************************************************************/
|
|
#include "VlppOS.Windows.h"
|
|
|
|
/***********************************************************************
|
|
.\FILESYSTEM.WINDOWS.CPP
|
|
***********************************************************************/
|
|
/***********************************************************************
|
|
Author: Zihan Chen (vczh)
|
|
Licensed under https://github.com/vczh-libraries/License
|
|
***********************************************************************/
|
|
|
|
#define _WINSOCKAPI_
|
|
#include <Windows.h>
|
|
#include <Shlwapi.h>
|
|
|
|
#ifndef VCZH_MSVC
|
|
static_assert(false, "Do not build this file for non-Windows applications.");
|
|
#endif
|
|
|
|
#pragma comment(lib, "Shlwapi.lib")
|
|
|
|
namespace vl
|
|
{
|
|
namespace stream
|
|
{
|
|
extern Ptr<IFileStreamImpl> CreateOSFileStreamImpl(const WString& fileName, FileStream::AccessRight accessRight);
|
|
}
|
|
|
|
namespace filesystem
|
|
{
|
|
using namespace collections;
|
|
using namespace stream;
|
|
|
|
/***********************************************************************
|
|
WindowsFileSystemImpl
|
|
***********************************************************************/
|
|
|
|
class WindowsFileSystemImpl : public feature_injection::FeatureImpl<IFileSystemImpl>
|
|
{
|
|
public:
|
|
wchar_t GetPathDelimiter() const override
|
|
{
|
|
return L'\\';
|
|
}
|
|
|
|
const wchar_t* GetCompatibleDelimiters() const override
|
|
{
|
|
return L"/";
|
|
}
|
|
|
|
WString ConcatPath(const WString& fullPath, const WString& relativePath) const override
|
|
{
|
|
if (IsRoot(fullPath))
|
|
{
|
|
return relativePath;
|
|
}
|
|
|
|
return fullPath + WString::FromChar(GetPathDelimiter()) + relativePath;
|
|
}
|
|
|
|
void Initialize(WString& fullPath) const override
|
|
{
|
|
{
|
|
Array<wchar_t> buffer(fullPath.Length() + 1);
|
|
wcscpy_s(&buffer[0], fullPath.Length() + 1, fullPath.Buffer());
|
|
FilePath::NormalizeDelimiters(buffer);
|
|
fullPath = &buffer[0];
|
|
}
|
|
|
|
if (fullPath != L"")
|
|
{
|
|
if (fullPath.Length() < 2 || fullPath[1] != L':')
|
|
{
|
|
wchar_t buffer[MAX_PATH + 1] = { 0 };
|
|
auto result = GetCurrentDirectory(sizeof(buffer) / sizeof(*buffer), buffer);
|
|
if (result > MAX_PATH + 1 || result == 0)
|
|
{
|
|
throw ArgumentException(L"Failed to call GetCurrentDirectory.", L"vl::filesystem::FilePath::Initialize", L"");
|
|
}
|
|
fullPath = WString(buffer) + L"\\" + fullPath;
|
|
}
|
|
{
|
|
wchar_t buffer[MAX_PATH + 1] = { 0 };
|
|
if (fullPath.Length() == 2 && fullPath[1] == L':')
|
|
{
|
|
fullPath += L"\\";
|
|
}
|
|
auto result = GetFullPathName(fullPath.Buffer(), sizeof(buffer) / sizeof(*buffer), buffer, NULL);
|
|
if (result > MAX_PATH + 1 || result == 0)
|
|
{
|
|
throw ArgumentException(L"The path is illegal.", L"vl::filesystem::FilePath::FilePath", L"_filePath");
|
|
}
|
|
|
|
{
|
|
wchar_t shortPath[MAX_PATH + 1];
|
|
wchar_t longPath[MAX_PATH + 1];
|
|
if (GetShortPathName(buffer, shortPath, MAX_PATH) > 0)
|
|
{
|
|
if (GetLongPathName(shortPath, longPath, MAX_PATH) > 0)
|
|
{
|
|
memcpy(buffer, longPath, sizeof(buffer));
|
|
}
|
|
}
|
|
}
|
|
fullPath = buffer;
|
|
}
|
|
}
|
|
|
|
FilePath::TrimLastDelimiter(fullPath);
|
|
}
|
|
|
|
bool IsFile(const WString& fullPath) const override
|
|
{
|
|
WIN32_FILE_ATTRIBUTE_DATA info;
|
|
BOOL result = GetFileAttributesEx(fullPath.Buffer(), GetFileExInfoStandard, &info);
|
|
if (!result) return false;
|
|
return (info.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) == 0;
|
|
}
|
|
|
|
bool IsFolder(const WString& fullPath) const override
|
|
{
|
|
WIN32_FILE_ATTRIBUTE_DATA info;
|
|
BOOL result = GetFileAttributesEx(fullPath.Buffer(), GetFileExInfoStandard, &info);
|
|
if (!result) return false;
|
|
return (info.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) != 0;
|
|
}
|
|
|
|
bool IsRoot(const WString& fullPath) const override
|
|
{
|
|
return fullPath == L"";
|
|
}
|
|
|
|
WString GetRelativePathFor(const WString& fromPath, const WString& toPath) const override
|
|
{
|
|
if (fromPath.Length() == 0 || toPath.Length() == 0 || fromPath[0] != toPath[0])
|
|
{
|
|
return toPath;
|
|
}
|
|
|
|
wchar_t buffer[MAX_PATH + 1] = { 0 };
|
|
PathRelativePathTo(
|
|
buffer,
|
|
fromPath.Buffer(),
|
|
(IsFolder(fromPath) ? FILE_ATTRIBUTE_DIRECTORY : 0),
|
|
toPath.Buffer(),
|
|
(IsFolder(toPath) ? FILE_ATTRIBUTE_DIRECTORY : 0)
|
|
);
|
|
return buffer;
|
|
}
|
|
|
|
bool FileDelete(const FilePath& filePath) const override
|
|
{
|
|
return DeleteFile(filePath.GetFullPath().Buffer()) != 0;
|
|
}
|
|
|
|
bool FileRename(const FilePath& filePath, const WString& newName) const override
|
|
{
|
|
WString oldFileName = filePath.GetFullPath();
|
|
WString newFileName = (filePath.GetFolder() / newName).GetFullPath();
|
|
return MoveFile(oldFileName.Buffer(), newFileName.Buffer()) != 0;
|
|
}
|
|
|
|
bool GetFolders(const FilePath& folderPath, collections::List<Folder>& folders) const override
|
|
{
|
|
if (folderPath.IsRoot())
|
|
{
|
|
auto bufferSize = GetLogicalDriveStrings(0, nullptr);
|
|
if (bufferSize > 0)
|
|
{
|
|
Array<wchar_t> buffer(bufferSize);
|
|
if (GetLogicalDriveStrings((DWORD)buffer.Count(), &buffer[0]) > 0)
|
|
{
|
|
auto begin = &buffer[0];
|
|
auto end = begin + buffer.Count();
|
|
while (begin < end && *begin)
|
|
{
|
|
WString driveString = begin;
|
|
begin += driveString.Length() + 1;
|
|
folders.Add(Folder(FilePath(driveString)));
|
|
}
|
|
return true;
|
|
}
|
|
}
|
|
return false;
|
|
}
|
|
else
|
|
{
|
|
if (!IsFolder(folderPath.GetFullPath())) return false;
|
|
WIN32_FIND_DATA findData;
|
|
HANDLE findHandle = INVALID_HANDLE_VALUE;
|
|
|
|
while (true)
|
|
{
|
|
if (findHandle == INVALID_HANDLE_VALUE)
|
|
{
|
|
WString searchPath = (folderPath / L"*").GetFullPath();
|
|
findHandle = FindFirstFile(searchPath.Buffer(), &findData);
|
|
if (findHandle == INVALID_HANDLE_VALUE)
|
|
{
|
|
break;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
BOOL result = FindNextFile(findHandle, &findData);
|
|
if (result == 0)
|
|
{
|
|
FindClose(findHandle);
|
|
break;
|
|
}
|
|
}
|
|
|
|
if (findData.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY)
|
|
{
|
|
if (wcscmp(findData.cFileName, L".") != 0 && wcscmp(findData.cFileName, L"..") != 0)
|
|
{
|
|
folders.Add(Folder(folderPath / findData.cFileName));
|
|
}
|
|
}
|
|
}
|
|
return true;
|
|
}
|
|
}
|
|
|
|
bool GetFiles(const FilePath& folderPath, collections::List<File>& files) const override
|
|
{
|
|
if (IsRoot(folderPath.GetFullPath()))
|
|
{
|
|
return true;
|
|
}
|
|
if (!IsFolder(folderPath.GetFullPath())) return false;
|
|
WIN32_FIND_DATA findData;
|
|
HANDLE findHandle = INVALID_HANDLE_VALUE;
|
|
|
|
while (true)
|
|
{
|
|
if (findHandle == INVALID_HANDLE_VALUE)
|
|
{
|
|
WString searchPath = (folderPath / L"*").GetFullPath();
|
|
findHandle = FindFirstFile(searchPath.Buffer(), &findData);
|
|
if (findHandle == INVALID_HANDLE_VALUE)
|
|
{
|
|
break;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
BOOL result = FindNextFile(findHandle, &findData);
|
|
if (result == 0)
|
|
{
|
|
FindClose(findHandle);
|
|
break;
|
|
}
|
|
}
|
|
|
|
if (!(findData.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY))
|
|
{
|
|
files.Add(File(folderPath / findData.cFileName));
|
|
}
|
|
}
|
|
return true;
|
|
}
|
|
|
|
bool CreateFolder(const FilePath& folderPath) const override
|
|
{
|
|
return CreateDirectory(folderPath.GetFullPath().Buffer(), NULL) != 0;
|
|
}
|
|
|
|
bool DeleteFolder(const FilePath& folderPath) const override
|
|
{
|
|
return RemoveDirectory(folderPath.GetFullPath().Buffer()) != 0;
|
|
}
|
|
|
|
bool FolderRename(const FilePath& folderPath, const WString& newName) const override
|
|
{
|
|
WString oldFileName = folderPath.GetFullPath();
|
|
WString newFileName = (folderPath.GetFolder() / newName).GetFullPath();
|
|
return MoveFile(oldFileName.Buffer(), newFileName.Buffer()) != 0;
|
|
}
|
|
|
|
Ptr<stream::IFileStreamImpl> GetFileStreamImpl(const WString& fileName, stream::FileStream::AccessRight accessRight) const override
|
|
{
|
|
return stream::CreateOSFileStreamImpl(fileName, accessRight);
|
|
}
|
|
};
|
|
|
|
IFileSystemImpl* GetOSFileSystemImpl()
|
|
{
|
|
static WindowsFileSystemImpl osFileSystemImpl;
|
|
return &osFileSystemImpl;
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
/***********************************************************************
|
|
.\LOCALE.WINDOWS.CPP
|
|
***********************************************************************/
|
|
/***********************************************************************
|
|
Author: Zihan Chen (vczh)
|
|
Licensed under https://github.com/vczh-libraries/License
|
|
***********************************************************************/
|
|
|
|
#define _WINSOCKAPI_
|
|
|
|
#ifndef VCZH_MSVC
|
|
static_assert(false, "Do not build this file for non-Windows applications.");
|
|
#endif
|
|
|
|
namespace vl
|
|
{
|
|
using namespace collections;
|
|
|
|
/***********************************************************************
|
|
Locale Helper Functions
|
|
***********************************************************************/
|
|
|
|
SYSTEMTIME DateTimeToSystemTime(const DateTime& dateTime)
|
|
{
|
|
SYSTEMTIME systemTime;
|
|
systemTime.wYear = (WORD)dateTime.year;
|
|
systemTime.wMonth = (WORD)dateTime.month;
|
|
systemTime.wDayOfWeek = (WORD)dateTime.dayOfWeek;
|
|
systemTime.wDay = (WORD)dateTime.day;
|
|
systemTime.wHour = (WORD)dateTime.hour;
|
|
systemTime.wMinute = (WORD)dateTime.minute;
|
|
systemTime.wSecond = (WORD)dateTime.second;
|
|
systemTime.wMilliseconds = (WORD)dateTime.milliseconds;
|
|
return systemTime;
|
|
}
|
|
|
|
BOOL CALLBACK Locale_EnumLocalesProcEx(
|
|
_In_ LPWSTR lpLocaleString,
|
|
_In_ DWORD dwFlags,
|
|
_In_ LPARAM lParam
|
|
)
|
|
{
|
|
((List<Locale>*)lParam)->Add(Locale(lpLocaleString));
|
|
return TRUE;
|
|
}
|
|
|
|
BOOL CALLBACK Locale_EnumDateFormatsProcExEx(
|
|
_In_ LPWSTR lpDateFormatString,
|
|
_In_ CALID CalendarID,
|
|
_In_ LPARAM lParam
|
|
)
|
|
{
|
|
((List<WString>*)lParam)->Add(lpDateFormatString);
|
|
return TRUE;
|
|
}
|
|
|
|
BOOL CALLBACK EnumTimeFormatsProcEx(
|
|
_In_ LPWSTR lpTimeFormatString,
|
|
_In_ LPARAM lParam
|
|
)
|
|
{
|
|
((List<WString>*)lParam)->Add(lpTimeFormatString);
|
|
return TRUE;
|
|
}
|
|
|
|
WString Transform(const WString& localeName, const WString& input, DWORD flag)
|
|
{
|
|
int length = LCMapStringEx(localeName.Buffer(), flag, input.Buffer(), (int)input.Length() + 1, NULL, 0, NULL, NULL, NULL);
|
|
Array<wchar_t> buffer(length);
|
|
LCMapStringEx(localeName.Buffer(), flag, input.Buffer(), (int)input.Length() + 1, &buffer[0], (int)buffer.Count(), NULL, NULL, NULL);
|
|
return &buffer[0];
|
|
}
|
|
|
|
DWORD TranslateNormalization(Locale::Normalization normalization)
|
|
{
|
|
DWORD result = 0;
|
|
if (normalization & Locale::IgnoreCase) result |= NORM_IGNORECASE;
|
|
if (normalization & Locale::IgnoreCaseLinguistic) result |= NORM_IGNORECASE | NORM_LINGUISTIC_CASING;
|
|
if (normalization & Locale::IgnoreKanaType) result |= NORM_IGNOREKANATYPE;
|
|
if (normalization & Locale::IgnoreNonSpace) result |= NORM_IGNORENONSPACE;
|
|
if (normalization & Locale::IgnoreSymbol) result |= NORM_IGNORESYMBOLS;
|
|
if (normalization & Locale::IgnoreWidth) result |= NORM_IGNOREWIDTH;
|
|
if (normalization & Locale::DigitsAsNumbers) result |= SORT_DIGITSASNUMBERS;
|
|
if (normalization & Locale::StringSoft) result |= SORT_STRINGSORT;
|
|
return result;
|
|
}
|
|
|
|
/***********************************************************************
|
|
WindowsLocaleImpl
|
|
***********************************************************************/
|
|
|
|
class WindowsLocaleImpl : public feature_injection::FeatureImpl<ILocaleImpl>
|
|
{
|
|
public:
|
|
Locale Invariant() const override
|
|
{
|
|
return Locale(LOCALE_NAME_INVARIANT);
|
|
}
|
|
|
|
Locale SystemDefault() const override
|
|
{
|
|
wchar_t buffer[LOCALE_NAME_MAX_LENGTH + 1] = { 0 };
|
|
GetSystemDefaultLocaleName(buffer, LOCALE_NAME_MAX_LENGTH);
|
|
return Locale(buffer);
|
|
}
|
|
|
|
Locale UserDefault() const override
|
|
{
|
|
wchar_t buffer[LOCALE_NAME_MAX_LENGTH + 1] = { 0 };
|
|
GetUserDefaultLocaleName(buffer, LOCALE_NAME_MAX_LENGTH);
|
|
return Locale(buffer);
|
|
}
|
|
|
|
void Enumerate(collections::List<Locale>& locales) const override
|
|
{
|
|
EnumSystemLocalesEx(&Locale_EnumLocalesProcEx, LOCALE_ALL, (LPARAM)&locales, NULL);
|
|
}
|
|
|
|
void GetShortDateFormats(const WString& localeName, collections::List<WString>& formats) const override
|
|
{
|
|
EnumDateFormatsExEx(&Locale_EnumDateFormatsProcExEx, localeName.Buffer(), DATE_SHORTDATE, (LPARAM)&formats);
|
|
}
|
|
|
|
void GetLongDateFormats(const WString& localeName, collections::List<WString>& formats) const override
|
|
{
|
|
EnumDateFormatsExEx(&Locale_EnumDateFormatsProcExEx, localeName.Buffer(), DATE_LONGDATE, (LPARAM)&formats);
|
|
}
|
|
|
|
void GetYearMonthDateFormats(const WString& localeName, collections::List<WString>& formats) const override
|
|
{
|
|
EnumDateFormatsExEx(&Locale_EnumDateFormatsProcExEx, localeName.Buffer(), DATE_YEARMONTH, (LPARAM)&formats);
|
|
}
|
|
|
|
void GetLongTimeFormats(const WString& localeName, collections::List<WString>& formats) const override
|
|
{
|
|
EnumTimeFormatsEx(&EnumTimeFormatsProcEx, localeName.Buffer(), 0, (LPARAM)&formats);
|
|
}
|
|
|
|
void GetShortTimeFormats(const WString& localeName, collections::List<WString>& formats) const override
|
|
{
|
|
EnumTimeFormatsEx(&EnumTimeFormatsProcEx, localeName.Buffer(), TIME_NOSECONDS, (LPARAM)&formats);
|
|
}
|
|
|
|
WString FormatDate(const WString& localeName, const WString& format, DateTime date) const override
|
|
{
|
|
SYSTEMTIME st = DateTimeToSystemTime(date);
|
|
int length = GetDateFormatEx(localeName.Buffer(), 0, &st, format.Buffer(), NULL, 0, NULL);
|
|
if (length == 0) return L"";
|
|
Array<wchar_t> buffer(length);
|
|
GetDateFormatEx(localeName.Buffer(), 0, &st, format.Buffer(), &buffer[0], (int)buffer.Count(), NULL);
|
|
return &buffer[0];
|
|
}
|
|
|
|
WString FormatTime(const WString& localeName, const WString& format, DateTime time) const override
|
|
{
|
|
SYSTEMTIME st = DateTimeToSystemTime(time);
|
|
int length = GetTimeFormatEx(localeName.Buffer(), 0, &st, format.Buffer(), NULL, 0);
|
|
if (length == 0) return L"";
|
|
Array<wchar_t> buffer(length);
|
|
GetTimeFormatEx(localeName.Buffer(), 0, &st, format.Buffer(), &buffer[0], (int)buffer.Count());
|
|
return &buffer[0];
|
|
}
|
|
|
|
WString FormatNumber(const WString& localeName, const WString& number) const override
|
|
{
|
|
int length = GetNumberFormatEx(localeName.Buffer(), 0, number.Buffer(), NULL, NULL, 0);
|
|
if (length == 0) return L"";
|
|
Array<wchar_t> buffer(length);
|
|
GetNumberFormatEx(localeName.Buffer(), 0, number.Buffer(), NULL, &buffer[0], (int)buffer.Count());
|
|
return &buffer[0];
|
|
}
|
|
|
|
WString FormatCurrency(const WString& localeName, const WString& currency) const override
|
|
{
|
|
int length = GetCurrencyFormatEx(localeName.Buffer(), 0, currency.Buffer(), NULL, NULL, 0);
|
|
if (length == 0) return L"";
|
|
Array<wchar_t> buffer(length);
|
|
GetCurrencyFormatEx(localeName.Buffer(), 0, currency.Buffer(), NULL, &buffer[0], (int)buffer.Count());
|
|
return &buffer[0];
|
|
}
|
|
|
|
WString GetShortDayOfWeekName(const WString& localeName, vint dayOfWeek) const override
|
|
{
|
|
return FormatDate(localeName, L"ddd", DateTime::FromDateTime(2000, 1, 2 + dayOfWeek));
|
|
}
|
|
|
|
WString GetLongDayOfWeekName(const WString& localeName, vint dayOfWeek) const override
|
|
{
|
|
return FormatDate(localeName, L"dddd", DateTime::FromDateTime(2000, 1, 2 + dayOfWeek));
|
|
}
|
|
|
|
WString GetShortMonthName(const WString& localeName, vint month) const override
|
|
{
|
|
return FormatDate(localeName, L"MMM", DateTime::FromDateTime(2000, month, 1));
|
|
}
|
|
|
|
WString GetLongMonthName(const WString& localeName, vint month) const override
|
|
{
|
|
return FormatDate(localeName, L"MMMM", DateTime::FromDateTime(2000, month, 1));
|
|
}
|
|
|
|
WString ToLower(const WString& localeName, const WString& str) const override
|
|
{
|
|
return Transform(localeName, str, LCMAP_LOWERCASE);
|
|
}
|
|
|
|
WString ToUpper(const WString& localeName, const WString& str) const override
|
|
{
|
|
return Transform(localeName, str, LCMAP_UPPERCASE);
|
|
}
|
|
|
|
WString ToLinguisticLower(const WString& localeName, const WString& str) const override
|
|
{
|
|
return Transform(localeName, str, LCMAP_LOWERCASE | LCMAP_LINGUISTIC_CASING);
|
|
}
|
|
|
|
WString ToLinguisticUpper(const WString& localeName, const WString& str) const override
|
|
{
|
|
return Transform(localeName, str, LCMAP_UPPERCASE | LCMAP_LINGUISTIC_CASING);
|
|
}
|
|
|
|
vint Compare(const WString& localeName, const WString& s1, const WString& s2, Locale::Normalization normalization) const override
|
|
{
|
|
switch (CompareStringEx(localeName.Buffer(), TranslateNormalization(normalization), s1.Buffer(), (int)s1.Length(), s2.Buffer(), (int)s2.Length(), NULL, NULL, NULL))
|
|
{
|
|
case CSTR_LESS_THAN: return -1;
|
|
case CSTR_GREATER_THAN: return 1;
|
|
default: return 0;
|
|
}
|
|
}
|
|
|
|
vint CompareOrdinal(const WString& s1, const WString& s2) const override
|
|
{
|
|
switch (CompareStringOrdinal(s1.Buffer(), (int)s1.Length(), s2.Buffer(), (int)s2.Length(), FALSE))
|
|
{
|
|
case CSTR_LESS_THAN: return -1;
|
|
case CSTR_GREATER_THAN: return 1;
|
|
default: return 0;
|
|
}
|
|
}
|
|
|
|
vint CompareOrdinalIgnoreCase(const WString& s1, const WString& s2) const override
|
|
{
|
|
switch (CompareStringOrdinal(s1.Buffer(), (int)s1.Length(), s2.Buffer(), (int)s2.Length(), TRUE))
|
|
{
|
|
case CSTR_LESS_THAN: return -1;
|
|
case CSTR_GREATER_THAN: return 1;
|
|
default: return 0;
|
|
}
|
|
}
|
|
|
|
collections::Pair<vint, vint> FindFirst(const WString& localeName, const WString& text, const WString& find, Locale::Normalization normalization) const override
|
|
{
|
|
int length = 0;
|
|
int result = FindNLSStringEx(localeName.Buffer(), FIND_FROMSTART | TranslateNormalization(normalization), text.Buffer(), (int)text.Length(), find.Buffer(), (int)find.Length(), &length, NULL, NULL, NULL);
|
|
return result == -1 ? collections::Pair<vint, vint>(-1, 0) : collections::Pair<vint, vint>(result, length);
|
|
}
|
|
|
|
collections::Pair<vint, vint> FindLast(const WString& localeName, const WString& text, const WString& find, Locale::Normalization normalization) const override
|
|
{
|
|
int length = 0;
|
|
int result = FindNLSStringEx(localeName.Buffer(), FIND_FROMEND | TranslateNormalization(normalization), text.Buffer(), (int)text.Length(), find.Buffer(), (int)find.Length(), &length, NULL, NULL, NULL);
|
|
return result == -1 ? collections::Pair<vint, vint>(-1, 0) : collections::Pair<vint, vint>(result, length);
|
|
}
|
|
|
|
bool StartsWith(const WString& localeName, const WString& text, const WString& find, Locale::Normalization normalization) const override
|
|
{
|
|
int result = FindNLSStringEx(localeName.Buffer(), FIND_STARTSWITH | TranslateNormalization(normalization), text.Buffer(), (int)text.Length(), find.Buffer(), (int)find.Length(), NULL, NULL, NULL, NULL);
|
|
return result != -1;
|
|
}
|
|
|
|
bool EndsWith(const WString& localeName, const WString& text, const WString& find, Locale::Normalization normalization) const override
|
|
{
|
|
int result = FindNLSStringEx(localeName.Buffer(), FIND_ENDSWITH | TranslateNormalization(normalization), text.Buffer(), (int)text.Length(), find.Buffer(), (int)find.Length(), NULL, NULL, NULL, NULL);
|
|
return result != -1;
|
|
}
|
|
};
|
|
|
|
ILocaleImpl* GetOSLocaleImpl()
|
|
{
|
|
static WindowsLocaleImpl windowsLocaleImpl;
|
|
return &windowsLocaleImpl;
|
|
}
|
|
|
|
/***********************************************************************
|
|
Locale (Windows Specific)
|
|
***********************************************************************/
|
|
|
|
WString Locale::ToFullWidth(const WString& str)const
|
|
{
|
|
return Transform(localeName, str, LCMAP_FULLWIDTH);
|
|
}
|
|
|
|
WString Locale::ToHalfWidth(const WString& str)const
|
|
{
|
|
return Transform(localeName, str, LCMAP_HALFWIDTH);
|
|
}
|
|
|
|
WString Locale::ToHiragana(const WString& str)const
|
|
{
|
|
return Transform(localeName, str, LCMAP_HIRAGANA);
|
|
}
|
|
|
|
WString Locale::ToKatagana(const WString& str)const
|
|
{
|
|
return Transform(localeName, str, LCMAP_KATAKANA);
|
|
}
|
|
|
|
WString Locale::ToSimplifiedChinese(const WString& str)const
|
|
{
|
|
return Transform(localeName, str, LCMAP_SIMPLIFIED_CHINESE);
|
|
}
|
|
|
|
WString Locale::ToTraditionalChinese(const WString& str)const
|
|
{
|
|
return Transform(localeName, str, LCMAP_TRADITIONAL_CHINESE);
|
|
}
|
|
|
|
WString Locale::ToTileCase(const WString& str)const
|
|
{
|
|
return Transform(localeName, str, LCMAP_TITLECASE);
|
|
}
|
|
}
|
|
|
|
|
|
/***********************************************************************
|
|
.\THREADING.WINDOWS.CPP
|
|
***********************************************************************/
|
|
/***********************************************************************
|
|
Author: Zihan Chen (vczh)
|
|
Licensed under https://github.com/vczh-libraries/License
|
|
***********************************************************************/
|
|
|
|
#define _WINSOCKAPI_
|
|
|
|
#ifndef VCZH_MSVC
|
|
static_assert(false, "Do not build this file for non-Windows applications.");
|
|
#endif
|
|
|
|
namespace vl
|
|
{
|
|
using namespace threading_internal;
|
|
using namespace collections;
|
|
|
|
/***********************************************************************
|
|
WaitableObject
|
|
***********************************************************************/
|
|
|
|
namespace threading_internal
|
|
{
|
|
struct WaitableData
|
|
{
|
|
HANDLE handle;
|
|
|
|
WaitableData(HANDLE _handle)
|
|
:handle(_handle)
|
|
{
|
|
}
|
|
};
|
|
}
|
|
|
|
WaitableObject::WaitableObject()
|
|
:waitableData(0)
|
|
{
|
|
}
|
|
|
|
void WaitableObject::SetData(threading_internal::WaitableData* data)
|
|
{
|
|
waitableData=data;
|
|
}
|
|
|
|
bool WaitableObject::IsCreated()
|
|
{
|
|
return waitableData!=0;
|
|
}
|
|
|
|
bool WaitableObject::Wait()
|
|
{
|
|
return WaitForTime(INFINITE);
|
|
}
|
|
|
|
bool WaitableObject::WaitForTime(vint ms)
|
|
{
|
|
if(IsCreated())
|
|
{
|
|
if(WaitForSingleObject(waitableData->handle, (DWORD)ms)==WAIT_OBJECT_0)
|
|
{
|
|
return true;
|
|
}
|
|
}
|
|
return false;
|
|
}
|
|
|
|
bool WaitableObject::WaitAll(WaitableObject** objects, vint count)
|
|
{
|
|
Array<HANDLE> handles(count);
|
|
for(vint i=0;i<count;i++)
|
|
{
|
|
handles[i]=objects[i]->waitableData->handle;
|
|
}
|
|
DWORD result=WaitForMultipleObjects((DWORD)count, &handles[0], TRUE, INFINITE);
|
|
return result==WAIT_OBJECT_0 || result==WAIT_ABANDONED_0;
|
|
|
|
}
|
|
|
|
bool WaitableObject::WaitAllForTime(WaitableObject** objects, vint count, vint ms)
|
|
{
|
|
Array<HANDLE> handles(count);
|
|
for(vint i=0;i<count;i++)
|
|
{
|
|
handles[i]=objects[i]->waitableData->handle;
|
|
}
|
|
DWORD result=WaitForMultipleObjects((DWORD)count, &handles[0], TRUE, (DWORD)ms);
|
|
return result==WAIT_OBJECT_0 || result==WAIT_ABANDONED_0;
|
|
}
|
|
|
|
vint WaitableObject::WaitAny(WaitableObject** objects, vint count, bool* abandoned)
|
|
{
|
|
Array<HANDLE> handles(count);
|
|
for(vint i=0;i<count;i++)
|
|
{
|
|
handles[i]=objects[i]->waitableData->handle;
|
|
}
|
|
DWORD result=WaitForMultipleObjects((DWORD)count, &handles[0], FALSE, INFINITE);
|
|
if(WAIT_OBJECT_0 <= result && result<WAIT_OBJECT_0+count)
|
|
{
|
|
*abandoned=false;
|
|
return result-WAIT_OBJECT_0;
|
|
}
|
|
else if(WAIT_ABANDONED_0 <= result && result<WAIT_ABANDONED_0+count)
|
|
{
|
|
*abandoned=true;
|
|
return result-WAIT_ABANDONED_0;
|
|
}
|
|
else
|
|
{
|
|
return -1;
|
|
}
|
|
}
|
|
|
|
vint WaitableObject::WaitAnyForTime(WaitableObject** objects, vint count, vint ms, bool* abandoned)
|
|
{
|
|
Array<HANDLE> handles(count);
|
|
for(vint i=0;i<count;i++)
|
|
{
|
|
handles[i]=objects[i]->waitableData->handle;
|
|
}
|
|
DWORD result=WaitForMultipleObjects((DWORD)count, &handles[0], FALSE, (DWORD)ms);
|
|
if(WAIT_OBJECT_0 <= result && result<WAIT_OBJECT_0+count)
|
|
{
|
|
*abandoned=false;
|
|
return result-WAIT_OBJECT_0;
|
|
}
|
|
else if(WAIT_ABANDONED_0 <= result && result<WAIT_ABANDONED_0+count)
|
|
{
|
|
*abandoned=true;
|
|
return result-WAIT_ABANDONED_0;
|
|
}
|
|
else
|
|
{
|
|
return -1;
|
|
}
|
|
}
|
|
|
|
/***********************************************************************
|
|
Thread
|
|
***********************************************************************/
|
|
|
|
namespace threading_internal
|
|
{
|
|
struct ThreadData : public WaitableData
|
|
{
|
|
DWORD id;
|
|
|
|
ThreadData()
|
|
:WaitableData(NULL)
|
|
{
|
|
id=-1;
|
|
}
|
|
};
|
|
|
|
class ProceduredThread : public Thread
|
|
{
|
|
private:
|
|
Thread::ThreadProcedure procedure;
|
|
void* argument;
|
|
bool deleteAfterStopped;
|
|
|
|
protected:
|
|
void Run()
|
|
{
|
|
bool deleteAfterStopped = this->deleteAfterStopped;
|
|
ThreadLocalStorage::FixStorages();
|
|
try
|
|
{
|
|
procedure(this, argument);
|
|
threadState=Thread::Stopped;
|
|
ThreadLocalStorage::ClearStorages();
|
|
}
|
|
catch (...)
|
|
{
|
|
ThreadLocalStorage::ClearStorages();
|
|
throw;
|
|
}
|
|
if(deleteAfterStopped)
|
|
{
|
|
delete this;
|
|
}
|
|
}
|
|
public:
|
|
ProceduredThread(Thread::ThreadProcedure _procedure, void* _argument, bool _deleteAfterStopped)
|
|
:procedure(_procedure)
|
|
,argument(_argument)
|
|
,deleteAfterStopped(_deleteAfterStopped)
|
|
{
|
|
}
|
|
};
|
|
|
|
class LambdaThread : public Thread
|
|
{
|
|
private:
|
|
Func<void()> procedure;
|
|
bool deleteAfterStopped;
|
|
|
|
protected:
|
|
void Run()
|
|
{
|
|
bool deleteAfterStopped = this->deleteAfterStopped;
|
|
ThreadLocalStorage::FixStorages();
|
|
try
|
|
{
|
|
procedure();
|
|
threadState=Thread::Stopped;
|
|
ThreadLocalStorage::ClearStorages();
|
|
}
|
|
catch (...)
|
|
{
|
|
ThreadLocalStorage::ClearStorages();
|
|
throw;
|
|
}
|
|
if(deleteAfterStopped)
|
|
{
|
|
delete this;
|
|
}
|
|
}
|
|
public:
|
|
LambdaThread(const Func<void()>& _procedure, bool _deleteAfterStopped)
|
|
:procedure(_procedure)
|
|
,deleteAfterStopped(_deleteAfterStopped)
|
|
{
|
|
}
|
|
};
|
|
}
|
|
|
|
void InternalThreadProc(Thread* thread)
|
|
{
|
|
thread->Run();
|
|
}
|
|
|
|
DWORD WINAPI InternalThreadProcWrapper(LPVOID lpParameter)
|
|
{
|
|
InternalThreadProc((Thread*)lpParameter);
|
|
return 0;
|
|
}
|
|
|
|
Thread::Thread()
|
|
{
|
|
internalData=new ThreadData;
|
|
internalData->handle=CreateThread(NULL, 0, InternalThreadProcWrapper, this, CREATE_SUSPENDED, &internalData->id);
|
|
threadState=Thread::NotStarted;
|
|
SetData(internalData);
|
|
}
|
|
|
|
Thread::~Thread()
|
|
{
|
|
if (internalData)
|
|
{
|
|
Stop();
|
|
CloseHandle(internalData->handle);
|
|
delete internalData;
|
|
}
|
|
}
|
|
|
|
Thread* Thread::CreateAndStart(ThreadProcedure procedure, void* argument, bool deleteAfterStopped)
|
|
{
|
|
if(procedure)
|
|
{
|
|
Thread* thread=new ProceduredThread(procedure, argument, deleteAfterStopped);
|
|
if(thread->Start())
|
|
{
|
|
return thread;
|
|
}
|
|
else
|
|
{
|
|
delete thread;
|
|
}
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
Thread* Thread::CreateAndStart(const Func<void()>& procedure, bool deleteAfterStopped)
|
|
{
|
|
Thread* thread=new LambdaThread(procedure, deleteAfterStopped);
|
|
if(thread->Start())
|
|
{
|
|
return thread;
|
|
}
|
|
else
|
|
{
|
|
delete thread;
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
void Thread::Sleep(vint ms)
|
|
{
|
|
::Sleep((DWORD)ms);
|
|
}
|
|
|
|
|
|
vint Thread::GetCPUCount()
|
|
{
|
|
SYSTEM_INFO info;
|
|
GetSystemInfo(&info);
|
|
return info.dwNumberOfProcessors;
|
|
}
|
|
|
|
vint Thread::GetCurrentThreadId()
|
|
{
|
|
return (vint)::GetCurrentThreadId();
|
|
}
|
|
|
|
bool Thread::Start()
|
|
{
|
|
if(threadState==Thread::NotStarted && internalData->handle!=NULL)
|
|
{
|
|
if(ResumeThread(internalData->handle)!=-1)
|
|
{
|
|
threadState=Thread::Running;
|
|
return true;
|
|
}
|
|
}
|
|
return false;
|
|
}
|
|
|
|
bool Thread::Stop()
|
|
{
|
|
if(internalData->handle!=NULL)
|
|
{
|
|
if (SuspendThread(internalData->handle) != -1)
|
|
{
|
|
threadState=Thread::Stopped;
|
|
return true;
|
|
}
|
|
}
|
|
return false;
|
|
}
|
|
|
|
Thread::ThreadState Thread::GetState()
|
|
{
|
|
return threadState;
|
|
}
|
|
|
|
void Thread::SetCPU(vint index)
|
|
{
|
|
SetThreadAffinityMask(internalData->handle, ((vint)1 << index));
|
|
}
|
|
|
|
/***********************************************************************
|
|
Mutex
|
|
***********************************************************************/
|
|
|
|
namespace threading_internal
|
|
{
|
|
struct MutexData : public WaitableData
|
|
{
|
|
MutexData(HANDLE _handle)
|
|
:WaitableData(_handle)
|
|
{
|
|
}
|
|
};
|
|
}
|
|
|
|
Mutex::Mutex()
|
|
:internalData(0)
|
|
{
|
|
}
|
|
|
|
Mutex::~Mutex()
|
|
{
|
|
if(internalData)
|
|
{
|
|
CloseHandle(internalData->handle);
|
|
delete internalData;
|
|
}
|
|
}
|
|
|
|
bool Mutex::Create(bool owned, const WString& name)
|
|
{
|
|
if(IsCreated())return false;
|
|
BOOL aOwned=owned?TRUE:FALSE;
|
|
LPCTSTR aName=name==L""?NULL:name.Buffer();
|
|
HANDLE handle=CreateMutex(NULL, aOwned, aName);
|
|
if(handle)
|
|
{
|
|
internalData=new MutexData(handle);
|
|
SetData(internalData);
|
|
}
|
|
return IsCreated();
|
|
}
|
|
|
|
bool Mutex::Open(bool inheritable, const WString& name)
|
|
{
|
|
if(IsCreated())return false;
|
|
BOOL aInteritable=inheritable?TRUE:FALSE;
|
|
HANDLE handle=OpenMutex(SYNCHRONIZE, aInteritable, name.Buffer());
|
|
if(handle)
|
|
{
|
|
internalData=new MutexData(handle);
|
|
SetData(internalData);
|
|
}
|
|
return IsCreated();
|
|
}
|
|
|
|
bool Mutex::Release()
|
|
{
|
|
if(IsCreated())
|
|
{
|
|
return ReleaseMutex(internalData->handle)!=0;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
/***********************************************************************
|
|
Semaphore
|
|
***********************************************************************/
|
|
|
|
namespace threading_internal
|
|
{
|
|
struct SemaphoreData : public WaitableData
|
|
{
|
|
SemaphoreData(HANDLE _handle)
|
|
:WaitableData(_handle)
|
|
{
|
|
}
|
|
};
|
|
}
|
|
|
|
Semaphore::Semaphore()
|
|
:internalData(0)
|
|
{
|
|
}
|
|
|
|
Semaphore::~Semaphore()
|
|
{
|
|
if(internalData)
|
|
{
|
|
CloseHandle(internalData->handle);
|
|
delete internalData;
|
|
}
|
|
}
|
|
|
|
bool Semaphore::Create(vint initialCount, vint maxCount, const WString& name)
|
|
{
|
|
if(IsCreated())return false;
|
|
LONG aInitial=(LONG)initialCount;
|
|
LONG aMax=(LONG)maxCount;
|
|
LPCTSTR aName=name==L""?NULL:name.Buffer();
|
|
HANDLE handle=CreateSemaphore(NULL, aInitial, aMax, aName);
|
|
if(handle)
|
|
{
|
|
internalData=new SemaphoreData(handle);
|
|
SetData(internalData);
|
|
}
|
|
return IsCreated();
|
|
}
|
|
|
|
bool Semaphore::Open(bool inheritable, const WString& name)
|
|
{
|
|
if(IsCreated())return false;
|
|
BOOL aInteritable=inheritable?TRUE:FALSE;
|
|
HANDLE handle=OpenSemaphore(SYNCHRONIZE, aInteritable, name.Buffer());
|
|
if(handle)
|
|
{
|
|
internalData=new SemaphoreData(handle);
|
|
SetData(internalData);
|
|
}
|
|
return IsCreated();
|
|
}
|
|
|
|
bool Semaphore::Release()
|
|
{
|
|
if(IsCreated())
|
|
{
|
|
return Release(1)!=-1;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
vint Semaphore::Release(vint count)
|
|
{
|
|
if(IsCreated())
|
|
{
|
|
LONG previous=-1;
|
|
if(ReleaseSemaphore(internalData->handle, (LONG)count, &previous)!=0)
|
|
{
|
|
return (vint)previous;
|
|
}
|
|
}
|
|
return -1;
|
|
}
|
|
|
|
/***********************************************************************
|
|
EventObject
|
|
***********************************************************************/
|
|
|
|
namespace threading_internal
|
|
{
|
|
struct EventData : public WaitableData
|
|
{
|
|
EventData(HANDLE _handle)
|
|
:WaitableData(_handle)
|
|
{
|
|
}
|
|
};
|
|
}
|
|
|
|
EventObject::EventObject()
|
|
:internalData(0)
|
|
{
|
|
}
|
|
|
|
EventObject::~EventObject()
|
|
{
|
|
if(internalData)
|
|
{
|
|
CloseHandle(internalData->handle);
|
|
delete internalData;
|
|
}
|
|
}
|
|
|
|
bool EventObject::CreateAutoUnsignal(bool signaled, const WString& name)
|
|
{
|
|
if(IsCreated())return false;
|
|
BOOL aSignaled=signaled?TRUE:FALSE;
|
|
LPCTSTR aName=name==L""?NULL:name.Buffer();
|
|
HANDLE handle=CreateEvent(NULL, FALSE, aSignaled, aName);
|
|
if(handle)
|
|
{
|
|
internalData=new EventData(handle);
|
|
SetData(internalData);
|
|
}
|
|
return IsCreated();
|
|
}
|
|
|
|
bool EventObject::CreateManualUnsignal(bool signaled, const WString& name)
|
|
{
|
|
if(IsCreated())return false;
|
|
BOOL aSignaled=signaled?TRUE:FALSE;
|
|
LPCTSTR aName=name==L""?NULL:name.Buffer();
|
|
HANDLE handle=CreateEvent(NULL, TRUE, aSignaled, aName);
|
|
if(handle)
|
|
{
|
|
internalData=new EventData(handle);
|
|
SetData(internalData);
|
|
}
|
|
return IsCreated();
|
|
}
|
|
|
|
bool EventObject::Open(bool inheritable, const WString& name)
|
|
{
|
|
if(IsCreated())return false;
|
|
BOOL aInteritable=inheritable?TRUE:FALSE;
|
|
HANDLE handle=OpenEvent(SYNCHRONIZE, aInteritable, name.Buffer());
|
|
if(handle)
|
|
{
|
|
internalData=new EventData(handle);
|
|
SetData(internalData);
|
|
}
|
|
return IsCreated();
|
|
}
|
|
|
|
bool EventObject::Signal()
|
|
{
|
|
if(IsCreated())
|
|
{
|
|
return SetEvent(internalData->handle)!=0;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
bool EventObject::Unsignal()
|
|
{
|
|
if(IsCreated())
|
|
{
|
|
return ResetEvent(internalData->handle)!=0;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
/***********************************************************************
|
|
ThreadPoolLite
|
|
***********************************************************************/
|
|
|
|
DWORD WINAPI ThreadPoolQueueFunc(void* argument)
|
|
{
|
|
auto proc=Ptr((Func<void()>*)argument);
|
|
ThreadLocalStorage::FixStorages();
|
|
try
|
|
{
|
|
(*proc.Obj())();
|
|
ThreadLocalStorage::ClearStorages();
|
|
}
|
|
catch (...)
|
|
{
|
|
ThreadLocalStorage::ClearStorages();
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
ThreadPoolLite::ThreadPoolLite()
|
|
{
|
|
}
|
|
|
|
ThreadPoolLite::~ThreadPoolLite()
|
|
{
|
|
}
|
|
|
|
bool ThreadPoolLite::Queue(void(*proc)(void*), void* argument)
|
|
{
|
|
return Queue([=]() {proc(argument); });
|
|
}
|
|
|
|
bool ThreadPoolLite::Queue(const Func<void()>& proc)
|
|
{
|
|
Func<void()>* p=new Func<void()>(proc);
|
|
if(QueueUserWorkItem(&ThreadPoolQueueFunc, p, WT_EXECUTEDEFAULT))
|
|
{
|
|
return true;
|
|
}
|
|
else
|
|
{
|
|
delete p;
|
|
return false;
|
|
}
|
|
}
|
|
|
|
/***********************************************************************
|
|
CriticalSection
|
|
***********************************************************************/
|
|
|
|
namespace threading_internal
|
|
{
|
|
struct CriticalSectionData
|
|
{
|
|
CRITICAL_SECTION criticalSection;
|
|
};
|
|
}
|
|
|
|
CriticalSection::Scope::Scope(CriticalSection& _criticalSection)
|
|
:criticalSection(&_criticalSection)
|
|
{
|
|
criticalSection->Enter();
|
|
}
|
|
|
|
CriticalSection::Scope::~Scope()
|
|
{
|
|
criticalSection->Leave();
|
|
}
|
|
|
|
CriticalSection::CriticalSection()
|
|
{
|
|
internalData=new CriticalSectionData;
|
|
InitializeCriticalSection(&internalData->criticalSection);
|
|
}
|
|
|
|
CriticalSection::~CriticalSection()
|
|
{
|
|
DeleteCriticalSection(&internalData->criticalSection);
|
|
delete internalData;
|
|
}
|
|
|
|
bool CriticalSection::TryEnter()
|
|
{
|
|
return TryEnterCriticalSection(&internalData->criticalSection)!=0;
|
|
}
|
|
|
|
void CriticalSection::Enter()
|
|
{
|
|
EnterCriticalSection(&internalData->criticalSection);
|
|
}
|
|
|
|
void CriticalSection::Leave()
|
|
{
|
|
LeaveCriticalSection(&internalData->criticalSection);
|
|
}
|
|
|
|
/***********************************************************************
|
|
ReaderWriterLock
|
|
***********************************************************************/
|
|
|
|
namespace threading_internal
|
|
{
|
|
struct ReaderWriterLockData
|
|
{
|
|
SRWLOCK lock;
|
|
};
|
|
}
|
|
|
|
ReaderWriterLock::ReaderScope::ReaderScope(ReaderWriterLock& _lock)
|
|
:lock(&_lock)
|
|
{
|
|
lock->EnterReader();
|
|
}
|
|
|
|
ReaderWriterLock::ReaderScope::~ReaderScope()
|
|
{
|
|
lock->LeaveReader();
|
|
}
|
|
|
|
ReaderWriterLock::WriterScope::WriterScope(ReaderWriterLock& _lock)
|
|
:lock(&_lock)
|
|
{
|
|
lock->EnterWriter();
|
|
}
|
|
|
|
ReaderWriterLock::WriterScope::~WriterScope()
|
|
{
|
|
lock->LeaveWriter();
|
|
}
|
|
|
|
ReaderWriterLock::ReaderWriterLock()
|
|
:internalData(new threading_internal::ReaderWriterLockData)
|
|
{
|
|
InitializeSRWLock(&internalData->lock);
|
|
}
|
|
|
|
ReaderWriterLock::~ReaderWriterLock()
|
|
{
|
|
delete internalData;
|
|
}
|
|
|
|
bool ReaderWriterLock::TryEnterReader()
|
|
{
|
|
return TryAcquireSRWLockShared(&internalData->lock)!=0;
|
|
}
|
|
|
|
void ReaderWriterLock::EnterReader()
|
|
{
|
|
AcquireSRWLockShared(&internalData->lock);
|
|
}
|
|
|
|
void ReaderWriterLock::LeaveReader()
|
|
{
|
|
ReleaseSRWLockShared(&internalData->lock);
|
|
}
|
|
|
|
bool ReaderWriterLock::TryEnterWriter()
|
|
{
|
|
return TryAcquireSRWLockExclusive(&internalData->lock)!=0;
|
|
}
|
|
|
|
void ReaderWriterLock::EnterWriter()
|
|
{
|
|
AcquireSRWLockExclusive(&internalData->lock);
|
|
}
|
|
|
|
void ReaderWriterLock::LeaveWriter()
|
|
{
|
|
ReleaseSRWLockExclusive(&internalData->lock);
|
|
}
|
|
|
|
/***********************************************************************
|
|
ConditionVariable
|
|
***********************************************************************/
|
|
|
|
namespace threading_internal
|
|
{
|
|
struct ConditionVariableData
|
|
{
|
|
CONDITION_VARIABLE variable;
|
|
};
|
|
}
|
|
|
|
ConditionVariable::ConditionVariable()
|
|
:internalData(new threading_internal::ConditionVariableData)
|
|
{
|
|
InitializeConditionVariable(&internalData->variable);
|
|
}
|
|
|
|
ConditionVariable::~ConditionVariable()
|
|
{
|
|
delete internalData;
|
|
}
|
|
|
|
bool ConditionVariable::SleepWith(CriticalSection& cs)
|
|
{
|
|
return SleepConditionVariableCS(&internalData->variable, &cs.internalData->criticalSection, INFINITE)!=0;
|
|
}
|
|
|
|
bool ConditionVariable::SleepWithForTime(CriticalSection& cs, vint ms)
|
|
{
|
|
return SleepConditionVariableCS(&internalData->variable, &cs.internalData->criticalSection, (DWORD)ms)!=0;
|
|
}
|
|
|
|
bool ConditionVariable::SleepWithReader(ReaderWriterLock& lock)
|
|
{
|
|
return SleepConditionVariableSRW(&internalData->variable, &lock.internalData->lock, INFINITE, CONDITION_VARIABLE_LOCKMODE_SHARED)!=0;
|
|
}
|
|
|
|
bool ConditionVariable::SleepWithReaderForTime(ReaderWriterLock& lock, vint ms)
|
|
{
|
|
return SleepConditionVariableSRW(&internalData->variable, &lock.internalData->lock, (DWORD)ms, CONDITION_VARIABLE_LOCKMODE_SHARED)!=0;
|
|
}
|
|
|
|
bool ConditionVariable::SleepWithWriter(ReaderWriterLock& lock)
|
|
{
|
|
return SleepConditionVariableSRW(&internalData->variable, &lock.internalData->lock, INFINITE, 0)!=0;
|
|
}
|
|
|
|
bool ConditionVariable::SleepWithWriterForTime(ReaderWriterLock& lock, vint ms)
|
|
{
|
|
return SleepConditionVariableSRW(&internalData->variable, &lock.internalData->lock, (DWORD)ms, 0)!=0;
|
|
}
|
|
|
|
void ConditionVariable::WakeOnePending()
|
|
{
|
|
WakeConditionVariable(&internalData->variable);
|
|
}
|
|
|
|
void ConditionVariable::WakeAllPendings()
|
|
{
|
|
WakeAllConditionVariable(&internalData->variable);
|
|
}
|
|
|
|
/***********************************************************************
|
|
ThreadLocalStorage
|
|
***********************************************************************/
|
|
|
|
#define KEY ((DWORD&)key)
|
|
|
|
ThreadLocalStorage::ThreadLocalStorage(Destructor _destructor)
|
|
:destructor(_destructor)
|
|
{
|
|
static_assert(sizeof(key) >= sizeof(DWORD), "ThreadLocalStorage's key storage is not large enouth.");
|
|
PushStorage(this);
|
|
KEY = TlsAlloc();
|
|
CHECK_ERROR(KEY != TLS_OUT_OF_INDEXES, L"vl::ThreadLocalStorage::ThreadLocalStorage()#Failed to alloc new thread local storage index.");
|
|
}
|
|
|
|
ThreadLocalStorage::~ThreadLocalStorage()
|
|
{
|
|
TlsFree(KEY);
|
|
}
|
|
|
|
void* ThreadLocalStorage::Get()
|
|
{
|
|
CHECK_ERROR(!disposed, L"vl::ThreadLocalStorage::Get()#Cannot access a disposed ThreadLocalStorage.");
|
|
return TlsGetValue(KEY);
|
|
}
|
|
|
|
void ThreadLocalStorage::Set(void* data)
|
|
{
|
|
CHECK_ERROR(!disposed, L"vl::ThreadLocalStorage::Set()#Cannot access a disposed ThreadLocalStorage.");
|
|
TlsSetValue(KEY, data);
|
|
}
|
|
|
|
#undef KEY
|
|
}
|
|
|
|
|
|
/***********************************************************************
|
|
.\ENCODING\CHARFORMAT\CHARFORMAT.WINDOWS.CPP
|
|
***********************************************************************/
|
|
/***********************************************************************
|
|
Author: Zihan Chen (vczh)
|
|
Licensed under https://github.com/vczh-libraries/License
|
|
***********************************************************************/
|
|
|
|
#define _WINSOCKAPI_
|
|
|
|
#ifndef VCZH_MSVC
|
|
static_assert(false, "Do not build this file for non-Windows applications.");
|
|
#endif
|
|
|
|
namespace vl
|
|
{
|
|
namespace stream
|
|
{
|
|
bool IsMbcsLeadByte(char c)
|
|
{
|
|
return IsDBCSLeadByte(c);
|
|
}
|
|
|
|
void MbcsToWChar(wchar_t* wideBuffer, vint wideChars, vint wideReaded, char* mbcsBuffer, vint mbcsChars)
|
|
{
|
|
MultiByteToWideChar(CP_THREAD_ACP, 0, mbcsBuffer, (int)mbcsChars, wideBuffer, (int)wideChars);
|
|
}
|
|
|
|
/***********************************************************************
|
|
MbcsEncoder
|
|
***********************************************************************/
|
|
|
|
vint MbcsEncoder::WriteString(wchar_t* _buffer, vint chars)
|
|
{
|
|
vint length = WideCharToMultiByte(CP_THREAD_ACP, 0, _buffer, (int)chars, NULL, NULL, NULL, NULL);
|
|
char* mbcs = new char[length];
|
|
WideCharToMultiByte(CP_THREAD_ACP, 0, _buffer, (int)chars, mbcs, (int)length, NULL, NULL);
|
|
vint result = stream->Write(mbcs, length);
|
|
delete[] mbcs;
|
|
|
|
if (result != length)
|
|
{
|
|
Close();
|
|
return 0;
|
|
}
|
|
return chars;
|
|
}
|
|
|
|
/***********************************************************************
|
|
Helper Functions
|
|
***********************************************************************/
|
|
|
|
extern bool CanBeMbcs(unsigned char* buffer, vint size);
|
|
extern bool CanBeUtf8(unsigned char* buffer, vint size);
|
|
extern bool CanBeUtf16(unsigned char* buffer, vint size, bool& hitSurrogatePairs);
|
|
extern bool CanBeUtf16BE(unsigned char* buffer, vint size, bool& hitSurrogatePairs);
|
|
|
|
template<vint Count>
|
|
bool GetEncodingResult(int(&tests)[Count], bool(&results)[Count], int test)
|
|
{
|
|
for (vint i = 0; i < Count; i++)
|
|
{
|
|
if (tests[i] & test)
|
|
{
|
|
if (results[i]) return true;
|
|
}
|
|
}
|
|
return false;
|
|
}
|
|
|
|
/***********************************************************************
|
|
TestEncoding
|
|
***********************************************************************/
|
|
|
|
extern void TestEncodingInternal(
|
|
unsigned char* buffer,
|
|
vint size,
|
|
BomEncoder::Encoding& encoding,
|
|
bool containsBom,
|
|
bool utf16HitSurrogatePairs,
|
|
bool utf16BEHitSurrogatePairs,
|
|
bool roughMbcs,
|
|
bool roughUtf8,
|
|
bool roughUtf16,
|
|
bool roughUtf16BE
|
|
)
|
|
{
|
|
int tests[] =
|
|
{
|
|
IS_TEXT_UNICODE_REVERSE_ASCII16,
|
|
IS_TEXT_UNICODE_REVERSE_STATISTICS,
|
|
IS_TEXT_UNICODE_REVERSE_CONTROLS,
|
|
|
|
IS_TEXT_UNICODE_ASCII16,
|
|
IS_TEXT_UNICODE_STATISTICS,
|
|
IS_TEXT_UNICODE_CONTROLS,
|
|
|
|
IS_TEXT_UNICODE_ILLEGAL_CHARS,
|
|
IS_TEXT_UNICODE_ODD_LENGTH,
|
|
IS_TEXT_UNICODE_NULL_BYTES,
|
|
};
|
|
|
|
const vint TestCount = sizeof(tests) / sizeof(*tests);
|
|
bool results[TestCount];
|
|
for (vint i = 0; i < TestCount; i++)
|
|
{
|
|
int test = tests[i];
|
|
results[i] = IsTextUnicode(buffer, (int)size, &test) != 0;
|
|
}
|
|
|
|
if (size % 2 == 0
|
|
&& !GetEncodingResult(tests, results, IS_TEXT_UNICODE_REVERSE_ASCII16)
|
|
&& !GetEncodingResult(tests, results, IS_TEXT_UNICODE_REVERSE_STATISTICS)
|
|
&& !GetEncodingResult(tests, results, IS_TEXT_UNICODE_REVERSE_CONTROLS)
|
|
)
|
|
{
|
|
for (vint i = 0; i < size; i += 2)
|
|
{
|
|
unsigned char c = buffer[i];
|
|
buffer[i] = buffer[i + 1];
|
|
buffer[i + 1] = c;
|
|
}
|
|
// 3 = (count of reverse group) = (count of unicode group)
|
|
for (vint i = 0; i < 3; i++)
|
|
{
|
|
int test = tests[i + 3];
|
|
results[i] = IsTextUnicode(buffer, (int)size, &test) != 0;
|
|
}
|
|
for (vint i = 0; i < size; i += 2)
|
|
{
|
|
unsigned char c = buffer[i];
|
|
buffer[i] = buffer[i + 1];
|
|
buffer[i + 1] = c;
|
|
}
|
|
}
|
|
|
|
if (GetEncodingResult(tests, results, IS_TEXT_UNICODE_NOT_UNICODE_MASK))
|
|
{
|
|
if (GetEncodingResult(tests, results, IS_TEXT_UNICODE_NOT_ASCII_MASK))
|
|
{
|
|
encoding = BomEncoder::Utf8;
|
|
}
|
|
else if (roughUtf8 || !roughMbcs)
|
|
{
|
|
encoding = BomEncoder::Utf8;
|
|
}
|
|
}
|
|
else if (GetEncodingResult(tests, results, IS_TEXT_UNICODE_ASCII16))
|
|
{
|
|
encoding = BomEncoder::Utf16;
|
|
}
|
|
else if (GetEncodingResult(tests, results, IS_TEXT_UNICODE_REVERSE_ASCII16))
|
|
{
|
|
encoding = BomEncoder::Utf16BE;
|
|
}
|
|
else if (GetEncodingResult(tests, results, IS_TEXT_UNICODE_CONTROLS))
|
|
{
|
|
encoding = BomEncoder::Utf16;
|
|
}
|
|
else if (GetEncodingResult(tests, results, IS_TEXT_UNICODE_REVERSE_CONTROLS))
|
|
{
|
|
encoding = BomEncoder::Utf16BE;
|
|
}
|
|
else
|
|
{
|
|
if (!roughUtf8)
|
|
{
|
|
if (GetEncodingResult(tests, results, IS_TEXT_UNICODE_STATISTICS))
|
|
{
|
|
encoding = BomEncoder::Utf16;
|
|
}
|
|
else if (GetEncodingResult(tests, results, IS_TEXT_UNICODE_STATISTICS))
|
|
{
|
|
encoding = BomEncoder::Utf16BE;
|
|
}
|
|
}
|
|
else if (GetEncodingResult(tests, results, IS_TEXT_UNICODE_NOT_UNICODE_MASK))
|
|
{
|
|
encoding = BomEncoder::Utf8;
|
|
}
|
|
else if (roughUtf8 || !roughMbcs)
|
|
{
|
|
encoding = BomEncoder::Utf8;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
/***********************************************************************
|
|
.\INTERPROCESS\WINDOWS\HTTPCLIENT.WINDOWS.CPP
|
|
***********************************************************************/
|
|
|
|
namespace vl::inter_process
|
|
{
|
|
|
|
/***********************************************************************
|
|
HttpClient (Reading)
|
|
***********************************************************************/
|
|
|
|
void HttpClient::RaiseLocalError(WString errorMessage, bool fatal)
|
|
{
|
|
if (callback)
|
|
{
|
|
callback->OnLocalError(errorMessage, fatal);
|
|
}
|
|
if (fatal)
|
|
{
|
|
Stop();
|
|
}
|
|
}
|
|
|
|
bool HttpClient::IsStopping()
|
|
{
|
|
bool result = false;
|
|
SPIN_LOCK(lockState)
|
|
{
|
|
result = state == State::Stopping;
|
|
}
|
|
return result;
|
|
}
|
|
|
|
void HttpClient::BeginReadingLoopUnsafe()
|
|
{
|
|
SendHttpRequest(HttpRequestType::Request, L"POST", urlRequest, WString::Empty);
|
|
}
|
|
|
|
/***********************************************************************
|
|
HttpClient (WaitForServer)
|
|
***********************************************************************/
|
|
|
|
INetworkProtocolConnection* HttpClient::GetConnection()
|
|
{
|
|
return this;
|
|
}
|
|
|
|
void HttpClient::CompleteConnectRequest(const WString& response, const WString& error)
|
|
{
|
|
SPIN_LOCK(lockConnectResult)
|
|
{
|
|
connectResponse = response;
|
|
connectError = error;
|
|
connectCompleted = true;
|
|
}
|
|
eventWaitForServer.Signal();
|
|
}
|
|
|
|
void HttpClient::WaitForServer()
|
|
{
|
|
{
|
|
SPIN_LOCK(lockState)
|
|
{
|
|
if (state == State::Stopping) return;
|
|
CHECK_ERROR(state == State::Ready, L"WaitForServer can only be called once.");
|
|
state = State::WaitForServerConnection;
|
|
}
|
|
}
|
|
|
|
{
|
|
SPIN_LOCK(lockConnectResult)
|
|
{
|
|
connectCompleted = false;
|
|
connectResponse = WString::Empty;
|
|
connectError = WString::Empty;
|
|
}
|
|
}
|
|
|
|
eventWaitForServer.Unsignal();
|
|
if (!SendHttpRequest(HttpRequestType::Connect, L"GET", urlConnect, WString::Empty))
|
|
{
|
|
return;
|
|
}
|
|
|
|
eventWaitForServer.Wait();
|
|
if (IsStopping()) return;
|
|
|
|
WString body;
|
|
WString error;
|
|
bool completed = false;
|
|
{
|
|
SPIN_LOCK(lockConnectResult)
|
|
{
|
|
body = connectResponse;
|
|
error = connectError;
|
|
completed = connectCompleted;
|
|
}
|
|
}
|
|
|
|
CHECK_ERROR(completed, L"/Connect did not complete.");
|
|
CHECK_ERROR(error == WString::Empty, L"/Connect failed.");
|
|
|
|
vint separatorIndex = body.IndexOf(L';');
|
|
CHECK_ERROR(separatorIndex != -1, L"/Connect response body is not in the correct format: requestUrl;responseUrl.");
|
|
urlRequest = baseUrl + body.Left(separatorIndex);
|
|
urlResponse = baseUrl + body.Right(body.Length() - separatorIndex - 1);
|
|
{
|
|
SPIN_LOCK(lockState)
|
|
{
|
|
if (state == State::Stopping) return;
|
|
state = State::Running;
|
|
}
|
|
}
|
|
|
|
if (callback)
|
|
{
|
|
callback->OnConnected();
|
|
}
|
|
}
|
|
|
|
ClientStatus HttpClient::GetStatus()
|
|
{
|
|
ClientStatus result = ClientStatus::Disconnected;
|
|
SPIN_LOCK(lockState)
|
|
{
|
|
switch (state)
|
|
{
|
|
case State::Ready:
|
|
result = ClientStatus::Ready;
|
|
break;
|
|
case State::WaitForServerConnection:
|
|
result = ClientStatus::WaitingForServer;
|
|
break;
|
|
case State::Running:
|
|
result = ClientStatus::Connected;
|
|
break;
|
|
default:
|
|
result = ClientStatus::Disconnected;
|
|
break;
|
|
}
|
|
}
|
|
return result;
|
|
}
|
|
|
|
/***********************************************************************
|
|
HttpClient (Writing)
|
|
***********************************************************************/
|
|
|
|
bool HttpClient::SendHttpRequest(HttpRequestType requestType, const wchar_t* method, const WString& url, const WString& body, vint attempt)
|
|
{
|
|
Ptr<HttpClientApi> api;
|
|
{
|
|
SPIN_LOCK(lockState)
|
|
{
|
|
if (state == State::Stopping) return false;
|
|
switch (requestType)
|
|
{
|
|
case HttpRequestType::Connect:
|
|
CHECK_ERROR(state == State::WaitForServerConnection, L"/Connect can only be called when client is waiting for the server.");
|
|
break;
|
|
case HttpRequestType::Request:
|
|
CHECK_ERROR(state == State::Running, L"/Request can only be called when client is running.");
|
|
break;
|
|
case HttpRequestType::Response:
|
|
CHECK_ERROR(state == State::Running, L"/Response can only be called when client is running.");
|
|
break;
|
|
}
|
|
api = httpClientApi;
|
|
}
|
|
}
|
|
|
|
if (!api) return false;
|
|
|
|
HttpRequest request;
|
|
request.method = method;
|
|
request.query = url;
|
|
request.acceptTypes.Add(JsonContentType);
|
|
if (requestType == HttpRequestType::Response)
|
|
{
|
|
request.contentType = JsonContentType;
|
|
request.keepAliveOnStop = true;
|
|
}
|
|
else if (requestType == HttpRequestType::Request)
|
|
{
|
|
request.receiveTimeout = 0;
|
|
}
|
|
if (body.Length() > 0)
|
|
{
|
|
request.contentType = JsonContentType;
|
|
request.SetBodyUtf8(body);
|
|
}
|
|
|
|
api->HttpQuery(request, [this, requestType, body, attempt](Variant<HttpResponse, HttpError> result)
|
|
{
|
|
OnHttpRequestCompleted(requestType, body, attempt, std::move(result));
|
|
});
|
|
return true;
|
|
}
|
|
|
|
void HttpClient::OnHttpRequestFailed(HttpRequestType requestType, const WString& body, vint attempt, const WString& errorMessage)
|
|
{
|
|
if (IsStopping()) return;
|
|
|
|
switch (requestType)
|
|
{
|
|
case HttpRequestType::Connect:
|
|
{
|
|
bool fatal = attempt >= HttpRequestMaxAttempts;
|
|
RaiseLocalError(errorMessage, fatal);
|
|
if (!fatal && !IsStopping())
|
|
{
|
|
SendHttpRequest(HttpRequestType::Connect, L"GET", urlConnect, WString::Empty, attempt + 1);
|
|
}
|
|
}
|
|
break;
|
|
case HttpRequestType::Request:
|
|
SendHttpRequest(HttpRequestType::Request, L"POST", urlRequest, WString::Empty, attempt + 1);
|
|
break;
|
|
case HttpRequestType::Response:
|
|
{
|
|
bool fatal = attempt >= HttpRequestMaxAttempts;
|
|
RaiseLocalError(errorMessage, fatal);
|
|
if (!fatal && !IsStopping())
|
|
{
|
|
SendHttpRequest(HttpRequestType::Response, L"POST", urlResponse, body, attempt + 1);
|
|
}
|
|
}
|
|
break;
|
|
}
|
|
}
|
|
|
|
void HttpClient::OnHttpRequestCompleted(HttpRequestType requestType, WString body, vint attempt, Variant<HttpResponse, HttpError> result)
|
|
{
|
|
if (auto error = result.TryGet<HttpError>())
|
|
{
|
|
switch (requestType)
|
|
{
|
|
case HttpRequestType::Connect:
|
|
OnHttpRequestFailed(requestType, body, attempt, L"/Connect failed: " + error->message);
|
|
break;
|
|
case HttpRequestType::Request:
|
|
OnHttpRequestFailed(requestType, body, attempt, L"/Request failed: " + error->message);
|
|
break;
|
|
case HttpRequestType::Response:
|
|
OnHttpRequestFailed(requestType, body, attempt, L"/Response failed: " + error->message);
|
|
break;
|
|
}
|
|
return;
|
|
}
|
|
|
|
auto&& response = result.Get<HttpResponse>();
|
|
if (response.statusCode != 200)
|
|
{
|
|
switch (requestType)
|
|
{
|
|
case HttpRequestType::Connect:
|
|
OnHttpRequestFailed(requestType, body, attempt, WString::Unmanaged(L"/Connect returned status code: ") + itow(response.statusCode) + L".");
|
|
break;
|
|
case HttpRequestType::Request:
|
|
OnHttpRequestFailed(requestType, body, attempt, WString::Unmanaged(L"/Request returned status code: ") + itow(response.statusCode) + L", another renderer may have connected to the core.");
|
|
break;
|
|
case HttpRequestType::Response:
|
|
OnHttpRequestFailed(requestType, body, attempt, WString::Unmanaged(L"/Response returned status code: ") + itow(response.statusCode) + L", another renderer may have connected to the core.");
|
|
break;
|
|
}
|
|
return;
|
|
}
|
|
|
|
if (response.contentType != JsonContentType)
|
|
{
|
|
switch (requestType)
|
|
{
|
|
case HttpRequestType::Connect:
|
|
OnHttpRequestFailed(requestType, body, attempt, L"/Connect response did not return content type: application/json; charset=utf8.");
|
|
break;
|
|
case HttpRequestType::Request:
|
|
OnHttpRequestFailed(requestType, body, attempt, L"/Request response did not return content type: application/json; charset=utf8.");
|
|
break;
|
|
case HttpRequestType::Response:
|
|
OnHttpRequestFailed(requestType, body, attempt, L"/Response response did not return content type: application/json; charset=utf8.");
|
|
break;
|
|
}
|
|
return;
|
|
}
|
|
|
|
auto responseBody = response.GetBodyUtf8();
|
|
switch (requestType)
|
|
{
|
|
case HttpRequestType::Connect:
|
|
CompleteConnectRequest(responseBody, WString::Empty);
|
|
break;
|
|
case HttpRequestType::Request:
|
|
if (!IsStopping())
|
|
{
|
|
BeginReadingLoopUnsafe();
|
|
if (responseBody.Length() > 0 && callback)
|
|
{
|
|
callback->OnReadString(responseBody);
|
|
}
|
|
}
|
|
break;
|
|
case HttpRequestType::Response:
|
|
if (!IsStopping() && responseBody.Length() > 0 && callback)
|
|
{
|
|
callback->OnReadString(responseBody);
|
|
}
|
|
break;
|
|
}
|
|
}
|
|
|
|
void HttpClient::SendString(const WString& str)
|
|
{
|
|
SendHttpRequest(HttpRequestType::Response, L"POST", urlResponse, str);
|
|
}
|
|
|
|
/***********************************************************************
|
|
HttpClient
|
|
***********************************************************************/
|
|
|
|
HttpClient::HttpClient(const WString _baseUrl, vint port)
|
|
: baseUrl(_baseUrl)
|
|
{
|
|
CHECK_ERROR(eventWaitForServer.CreateAutoUnsignal(false), L"HttpClient initialization failed on eventWaitForServer.CreateAutoUnsignal.");
|
|
|
|
httpClientApi = Ptr(new HttpClientApi(L"localhost", port));
|
|
urlConnect = baseUrl + HttpServerUrl_Connect;
|
|
}
|
|
|
|
HttpClient::~HttpClient()
|
|
{
|
|
Stop();
|
|
}
|
|
|
|
void HttpClient::InstallCallback(INetworkProtocolCallback* _callback)
|
|
{
|
|
callback = _callback;
|
|
CHECK_ERROR(callback, L"HttpClient::InstallCallback needs a valid INetworkProtocolCallback.");
|
|
callback->OnInstalled(this);
|
|
}
|
|
|
|
void HttpClient::Stop()
|
|
{
|
|
Ptr<HttpClientApi> stoppingApi;
|
|
bool notifyDisconnected = false;
|
|
{
|
|
SPIN_LOCK(lockState)
|
|
{
|
|
if (httpClientApi)
|
|
{
|
|
state = State::Stopping;
|
|
stoppingApi = httpClientApi;
|
|
httpClientApi = nullptr;
|
|
notifyDisconnected = true;
|
|
}
|
|
else
|
|
{
|
|
state = State::Stopping;
|
|
}
|
|
}
|
|
}
|
|
|
|
eventWaitForServer.Signal();
|
|
if (stoppingApi)
|
|
{
|
|
stoppingApi->Stop();
|
|
}
|
|
|
|
if (notifyDisconnected && callback)
|
|
{
|
|
callback->OnDisconnected();
|
|
}
|
|
}
|
|
|
|
}
|
|
|
|
|
|
/***********************************************************************
|
|
.\INTERPROCESS\WINDOWS\HTTPCLIENTAPI.WINDOWS.CPP
|
|
***********************************************************************/
|
|
|
|
#ifndef VCZH_MSVC
|
|
static_assert(false, "Do not build this file for non-Windows applications.");
|
|
#endif
|
|
|
|
#pragma comment(lib, "WinHttp.lib")
|
|
|
|
namespace vl::inter_process
|
|
{
|
|
|
|
using namespace vl::collections;
|
|
|
|
/***********************************************************************
|
|
HttpRequest
|
|
***********************************************************************/
|
|
|
|
void HttpRequest::SetBodyUtf8(const WString& bodyString)
|
|
{
|
|
vint utf8Size = WideCharToMultiByte(CP_UTF8, 0, bodyString.Buffer(), (int)bodyString.Length(), NULL, 0, NULL, NULL);
|
|
body.Resize(utf8Size);
|
|
if (utf8Size > 0)
|
|
{
|
|
WideCharToMultiByte(CP_UTF8, 0, bodyString.Buffer(), (int)bodyString.Length(), &body[0], (int)utf8Size, NULL, NULL);
|
|
}
|
|
}
|
|
|
|
/***********************************************************************
|
|
HttpResponse
|
|
***********************************************************************/
|
|
|
|
WString HttpResponse::GetBodyUtf8() const
|
|
{
|
|
if (body.Count() == 0)
|
|
{
|
|
return WString::Empty;
|
|
}
|
|
|
|
vint utf16Size = MultiByteToWideChar(CP_UTF8, 0, &body[0], (int)body.Count(), NULL, 0);
|
|
Array<wchar_t> utf16(utf16Size + 1);
|
|
ZeroMemory(&utf16[0], utf16.Count() * sizeof(wchar_t));
|
|
if (utf16Size > 0)
|
|
{
|
|
MultiByteToWideChar(CP_UTF8, 0, &body[0], (int)body.Count(), &utf16[0], (int)utf16Size);
|
|
}
|
|
return &utf16[0];
|
|
}
|
|
|
|
/***********************************************************************
|
|
HttpClientApi
|
|
***********************************************************************/
|
|
|
|
HttpError HttpClientApi::MakeError(const WString& operation, DWORD errorCode)
|
|
{
|
|
HttpError error;
|
|
error.operation = operation;
|
|
error.errorCode = errorCode;
|
|
error.message = operation + L" failed with Windows error " + itow((vint)errorCode) + L".";
|
|
return error;
|
|
}
|
|
|
|
vint HttpClientApi::HexValue(wchar_t c)
|
|
{
|
|
if (L'0' <= c && c <= L'9') return c - L'0';
|
|
if (L'a' <= c && c <= L'f') return c - L'a' + 10;
|
|
if (L'A' <= c && c <= L'F') return c - L'A' + 10;
|
|
return -1;
|
|
}
|
|
|
|
bool HttpClientApi::IsStopping()
|
|
{
|
|
bool result = false;
|
|
SPIN_LOCK(lockActiveRequests)
|
|
{
|
|
result = stopping;
|
|
}
|
|
return result;
|
|
}
|
|
|
|
void HttpClientApi::BeginPendingCallback()
|
|
{
|
|
if (pendingCallbacks++ == 0)
|
|
{
|
|
eventPendingCallbacks.Unsignal();
|
|
}
|
|
}
|
|
|
|
void HttpClientApi::EndPendingCallback()
|
|
{
|
|
if (--pendingCallbacks == 0)
|
|
{
|
|
eventPendingCallbacks.Signal();
|
|
}
|
|
}
|
|
|
|
void HttpClientApi::AttachRequestUnsafe(Ptr<HttpRequestContext> context)
|
|
{
|
|
activeRequests.Add(context);
|
|
}
|
|
|
|
void HttpClientApi::RemoveRequestUnsafe(Ptr<HttpRequestContext> context)
|
|
{
|
|
for (vint index = 0; index < activeRequests.Count(); index++)
|
|
{
|
|
if (activeRequests[index] == context)
|
|
{
|
|
activeRequests.RemoveAt(index);
|
|
return;
|
|
}
|
|
}
|
|
}
|
|
|
|
void HttpClientApi::CloseRequest(Ptr<HttpRequestContext> context)
|
|
{
|
|
HINTERNET httpRequest = NULL;
|
|
SPIN_LOCK(context->lockContext)
|
|
{
|
|
if (!context->closing)
|
|
{
|
|
context->closing = true;
|
|
httpRequest = context->httpRequest;
|
|
}
|
|
}
|
|
if (httpRequest)
|
|
{
|
|
WinHttpCloseHandle(httpRequest);
|
|
}
|
|
}
|
|
|
|
void HttpClientApi::OnRequestHandleClosing(Ptr<HttpRequestContext> context)
|
|
{
|
|
SPIN_LOCK(lockActiveRequests)
|
|
{
|
|
RemoveRequestUnsafe(context);
|
|
}
|
|
EndPendingCallback();
|
|
}
|
|
|
|
void HttpClientApi::CompleteRequest(Ptr<HttpRequestContext> context, HttpResponse&& response)
|
|
{
|
|
Func<void(Variant<HttpResponse, HttpError>)> callback;
|
|
bool stoppingNow = IsStopping();
|
|
bool invokeCallback = false;
|
|
SPIN_LOCK(context->lockContext)
|
|
{
|
|
if (!context->completed)
|
|
{
|
|
context->completed = true;
|
|
callback = context->callback;
|
|
invokeCallback = !stoppingNow;
|
|
}
|
|
}
|
|
|
|
if (invokeCallback && callback)
|
|
{
|
|
CloseRequest(context);
|
|
callback(Variant<HttpResponse, HttpError>(std::move(response)));
|
|
}
|
|
else
|
|
{
|
|
CloseRequest(context);
|
|
}
|
|
}
|
|
|
|
void HttpClientApi::CompleteRequest(Ptr<HttpRequestContext> context, HttpError&& error)
|
|
{
|
|
Func<void(Variant<HttpResponse, HttpError>)> callback;
|
|
bool stoppingNow = IsStopping();
|
|
bool invokeCallback = false;
|
|
SPIN_LOCK(context->lockContext)
|
|
{
|
|
if (!context->completed)
|
|
{
|
|
context->completed = true;
|
|
callback = context->callback;
|
|
invokeCallback = !stoppingNow;
|
|
}
|
|
}
|
|
|
|
if (invokeCallback && callback)
|
|
{
|
|
CloseRequest(context);
|
|
callback(Variant<HttpResponse, HttpError>(std::move(error)));
|
|
}
|
|
else
|
|
{
|
|
CloseRequest(context);
|
|
}
|
|
}
|
|
|
|
void HttpClientApi::CompleteRequestWithLastError(Ptr<HttpRequestContext> context, const WString& operation, DWORD errorCode)
|
|
{
|
|
if (errorCode == ERROR_INVALID_HANDLE && IsStopping() && !context->keepAliveOnStop)
|
|
{
|
|
return;
|
|
}
|
|
CompleteRequest(context, MakeError(operation, errorCode));
|
|
}
|
|
|
|
void CALLBACK HttpClientApi::HttpStatusCallback(HINTERNET httpRequest, DWORD_PTR contextValue, DWORD status, LPVOID statusInformation, DWORD statusInformationLength)
|
|
{
|
|
if (!contextValue) return;
|
|
auto contextPtr = reinterpret_cast<Ptr<HttpClientApi::HttpRequestContext>*>(contextValue);
|
|
auto context = *contextPtr;
|
|
auto self = context->api;
|
|
|
|
switch (status)
|
|
{
|
|
case WINHTTP_CALLBACK_STATUS_SENDREQUEST_COMPLETE:
|
|
{
|
|
if (self->IsStopping() && !context->keepAliveOnStop) return;
|
|
|
|
BOOL httpResult = WinHttpReceiveResponse(httpRequest, NULL);
|
|
DWORD lastError = GetLastError();
|
|
if (httpResult == FALSE)
|
|
{
|
|
self->CompleteRequestWithLastError(context, L"WinHttpReceiveResponse", lastError);
|
|
}
|
|
}
|
|
break;
|
|
case WINHTTP_CALLBACK_STATUS_HEADERS_AVAILABLE:
|
|
{
|
|
if (self->IsStopping() && !context->keepAliveOnStop) return;
|
|
|
|
DWORD statusCode = 0;
|
|
DWORD dwordLength = sizeof(DWORD);
|
|
BOOL httpResult = WinHttpQueryHeaders(
|
|
httpRequest,
|
|
WINHTTP_QUERY_STATUS_CODE | WINHTTP_QUERY_FLAG_NUMBER,
|
|
WINHTTP_HEADER_NAME_BY_INDEX,
|
|
&statusCode,
|
|
&dwordLength,
|
|
WINHTTP_NO_HEADER_INDEX);
|
|
DWORD lastError = GetLastError();
|
|
if (httpResult == FALSE)
|
|
{
|
|
self->CompleteRequestWithLastError(context, L"WinHttpQueryHeaders(status)", lastError);
|
|
return;
|
|
}
|
|
context->response.statusCode = statusCode;
|
|
|
|
DWORD headerLength = 0;
|
|
httpResult = WinHttpQueryHeaders(
|
|
httpRequest,
|
|
WINHTTP_QUERY_CONTENT_TYPE,
|
|
WINHTTP_HEADER_NAME_BY_INDEX,
|
|
NULL,
|
|
&headerLength,
|
|
WINHTTP_NO_HEADER_INDEX);
|
|
lastError = GetLastError();
|
|
if (httpResult == FALSE && lastError == ERROR_INSUFFICIENT_BUFFER)
|
|
{
|
|
Array<wchar_t> headerBuffer(headerLength / sizeof(wchar_t) + 1);
|
|
ZeroMemory(&headerBuffer[0], headerBuffer.Count() * sizeof(wchar_t));
|
|
httpResult = WinHttpQueryHeaders(
|
|
httpRequest,
|
|
WINHTTP_QUERY_CONTENT_TYPE,
|
|
WINHTTP_HEADER_NAME_BY_INDEX,
|
|
&headerBuffer[0],
|
|
&headerLength,
|
|
WINHTTP_NO_HEADER_INDEX);
|
|
lastError = GetLastError();
|
|
if (httpResult == FALSE)
|
|
{
|
|
self->CompleteRequestWithLastError(context, L"WinHttpQueryHeaders(content-type)", lastError);
|
|
return;
|
|
}
|
|
context->response.contentType = &headerBuffer[0];
|
|
}
|
|
else if (httpResult == FALSE && lastError != ERROR_WINHTTP_HEADER_NOT_FOUND)
|
|
{
|
|
self->CompleteRequestWithLastError(context, L"WinHttpQueryHeaders(content-type)", lastError);
|
|
return;
|
|
}
|
|
|
|
headerLength = 0;
|
|
httpResult = WinHttpQueryHeaders(
|
|
httpRequest,
|
|
WINHTTP_QUERY_SET_COOKIE,
|
|
WINHTTP_HEADER_NAME_BY_INDEX,
|
|
NULL,
|
|
&headerLength,
|
|
WINHTTP_NO_HEADER_INDEX);
|
|
lastError = GetLastError();
|
|
if (httpResult == FALSE && lastError == ERROR_INSUFFICIENT_BUFFER)
|
|
{
|
|
Array<wchar_t> headerBuffer(headerLength / sizeof(wchar_t) + 1);
|
|
ZeroMemory(&headerBuffer[0], headerBuffer.Count() * sizeof(wchar_t));
|
|
httpResult = WinHttpQueryHeaders(
|
|
httpRequest,
|
|
WINHTTP_QUERY_SET_COOKIE,
|
|
WINHTTP_HEADER_NAME_BY_INDEX,
|
|
&headerBuffer[0],
|
|
&headerLength,
|
|
WINHTTP_NO_HEADER_INDEX);
|
|
lastError = GetLastError();
|
|
if (httpResult == FALSE)
|
|
{
|
|
self->CompleteRequestWithLastError(context, L"WinHttpQueryHeaders(cookie)", lastError);
|
|
return;
|
|
}
|
|
context->response.cookie = &headerBuffer[0];
|
|
}
|
|
else if (httpResult == FALSE && lastError != ERROR_WINHTTP_HEADER_NOT_FOUND)
|
|
{
|
|
self->CompleteRequestWithLastError(context, L"WinHttpQueryHeaders(cookie)", lastError);
|
|
return;
|
|
}
|
|
|
|
context->bodyBufferWriting = 0;
|
|
httpResult = WinHttpQueryDataAvailable(httpRequest, NULL);
|
|
lastError = GetLastError();
|
|
if (httpResult == FALSE)
|
|
{
|
|
self->CompleteRequestWithLastError(context, L"WinHttpQueryDataAvailable", lastError);
|
|
}
|
|
}
|
|
break;
|
|
case WINHTTP_CALLBACK_STATUS_DATA_AVAILABLE:
|
|
{
|
|
if (self->IsStopping() && !context->keepAliveOnStop) return;
|
|
CHECK_ERROR(statusInformationLength == sizeof(DWORD), L"WinHttpQueryDataAvailable returned an unexpected payload.");
|
|
DWORD dataAvailable = *(PDWORD)statusInformation;
|
|
|
|
if (dataAvailable == 0)
|
|
{
|
|
context->response.body.Resize(context->bodyBufferWriting);
|
|
self->CompleteRequest(context, std::move(context->response));
|
|
return;
|
|
}
|
|
|
|
context->bodyBufferWritingAvailable = dataAvailable;
|
|
DWORD bufferSize = context->bodyBufferWriting + dataAvailable + 1;
|
|
if (context->response.body.Count() < (vint)bufferSize)
|
|
{
|
|
context->response.body.Resize((bufferSize + HttpRespondBodyStep - 1) / HttpRespondBodyStep * HttpRespondBodyStep);
|
|
}
|
|
|
|
BOOL httpResult = WinHttpReadData(
|
|
httpRequest,
|
|
&context->response.body[context->bodyBufferWriting],
|
|
dataAvailable,
|
|
NULL);
|
|
DWORD lastError = GetLastError();
|
|
if (httpResult == FALSE)
|
|
{
|
|
self->CompleteRequestWithLastError(context, L"WinHttpReadData", lastError);
|
|
}
|
|
}
|
|
break;
|
|
case WINHTTP_CALLBACK_STATUS_READ_COMPLETE:
|
|
{
|
|
if (self->IsStopping() && !context->keepAliveOnStop) return;
|
|
if (context->bodyBufferWritingAvailable != statusInformationLength)
|
|
{
|
|
self->CompleteRequest(context, MakeError(L"WinHttpReadData", ERROR_INVALID_DATA));
|
|
return;
|
|
}
|
|
context->bodyBufferWriting += context->bodyBufferWritingAvailable;
|
|
|
|
BOOL httpResult = WinHttpQueryDataAvailable(httpRequest, NULL);
|
|
DWORD lastError = GetLastError();
|
|
if (httpResult == FALSE)
|
|
{
|
|
self->CompleteRequestWithLastError(context, L"WinHttpQueryDataAvailable", lastError);
|
|
}
|
|
}
|
|
break;
|
|
case WINHTTP_CALLBACK_STATUS_REQUEST_ERROR:
|
|
{
|
|
if (self->IsStopping() && !context->keepAliveOnStop) return;
|
|
|
|
auto asyncResult = reinterpret_cast<WINHTTP_ASYNC_RESULT*>(statusInformation);
|
|
DWORD errorCode = asyncResult ? asyncResult->dwError : ERROR_WINHTTP_INTERNAL_ERROR;
|
|
HttpError error = MakeError(L"WinHTTP async request", errorCode);
|
|
if (asyncResult)
|
|
{
|
|
error.message += L" Operation code: " + itow((vint)asyncResult->dwResult) + L".";
|
|
}
|
|
self->CompleteRequest(context, std::move(error));
|
|
}
|
|
break;
|
|
case WINHTTP_CALLBACK_STATUS_HANDLE_CLOSING:
|
|
self->OnRequestHandleClosing(context);
|
|
delete contextPtr;
|
|
break;
|
|
}
|
|
}
|
|
|
|
HttpClientApi::HttpClientApi(const WString& _server, vint _port)
|
|
: server(_server)
|
|
, port(_port)
|
|
{
|
|
CHECK_ERROR(eventPendingCallbacks.CreateManualUnsignal(true), L"HttpClientApi initialization failed on eventPendingCallbacks.CreateManualUnsignal.");
|
|
|
|
httpSession = WinHttpOpen(
|
|
L"vl::inter_process::HttpClientApi",
|
|
WINHTTP_ACCESS_TYPE_DEFAULT_PROXY,
|
|
WINHTTP_NO_PROXY_NAME,
|
|
WINHTTP_NO_PROXY_BYPASS,
|
|
WINHTTP_FLAG_ASYNC);
|
|
CHECK_ERROR(httpSession != NULL, L"WinHttpOpen failed.");
|
|
|
|
httpConnection = WinHttpConnect(
|
|
httpSession,
|
|
server.Buffer(),
|
|
(INTERNET_PORT)port,
|
|
0);
|
|
CHECK_ERROR(httpConnection != NULL, L"WinHttpConnect failed.");
|
|
}
|
|
|
|
HttpClientApi::~HttpClientApi()
|
|
{
|
|
Stop();
|
|
}
|
|
|
|
void HttpClientApi::HttpQuery(const HttpRequest& request, Func<void(Variant<HttpResponse, HttpError>)> callback)
|
|
{
|
|
bool rejected = false;
|
|
{
|
|
SPIN_LOCK(lockActiveRequests)
|
|
{
|
|
if (stopping)
|
|
{
|
|
rejected = true;
|
|
}
|
|
else
|
|
{
|
|
BeginPendingCallback();
|
|
}
|
|
}
|
|
}
|
|
if (rejected)
|
|
{
|
|
if (callback)
|
|
{
|
|
callback(Variant<HttpResponse, HttpError>(MakeError(L"HttpClientApi::HttpQuery", ERROR_OPERATION_ABORTED)));
|
|
}
|
|
return;
|
|
}
|
|
|
|
BOOL httpResult = FALSE;
|
|
DWORD lastError = 0;
|
|
List<LPCWSTR> acceptTypes;
|
|
for (vint i = 0; i < request.acceptTypes.Count(); i++)
|
|
{
|
|
acceptTypes.Add(request.acceptTypes.Get(i).Buffer());
|
|
}
|
|
acceptTypes.Add(nullptr);
|
|
|
|
auto method = request.method == WString::Empty ? WString::Unmanaged(L"GET") : request.method;
|
|
auto httpRequest = WinHttpOpenRequest(
|
|
httpConnection,
|
|
method.Buffer(),
|
|
request.query.Buffer(),
|
|
NULL,
|
|
WINHTTP_NO_REFERER,
|
|
&acceptTypes[0],
|
|
(request.secure ? WINHTTP_FLAG_SECURE : 0) | WINHTTP_FLAG_REFRESH);
|
|
lastError = GetLastError();
|
|
if (httpRequest == NULL)
|
|
{
|
|
EndPendingCallback();
|
|
if (callback)
|
|
{
|
|
callback(Variant<HttpResponse, HttpError>(MakeError(L"WinHttpOpenRequest", lastError)));
|
|
}
|
|
return;
|
|
}
|
|
|
|
httpResult = WinHttpSetTimeouts(
|
|
httpRequest,
|
|
(int)request.resolveTimeout,
|
|
(int)request.connectTimeout,
|
|
(int)request.sendTimeout,
|
|
(int)request.receiveTimeout);
|
|
lastError = GetLastError();
|
|
if (httpResult == FALSE)
|
|
{
|
|
EndPendingCallback();
|
|
WinHttpCloseHandle(httpRequest);
|
|
if (callback)
|
|
{
|
|
callback(Variant<HttpResponse, HttpError>(MakeError(L"WinHttpSetTimeouts", lastError)));
|
|
}
|
|
return;
|
|
}
|
|
|
|
auto contextPtr = new Ptr<HttpRequestContext>(new HttpRequestContext);
|
|
auto context = *contextPtr;
|
|
context->api = this;
|
|
context->httpRequest = httpRequest;
|
|
context->callback = callback;
|
|
context->keepAliveOnStop = request.keepAliveOnStop;
|
|
|
|
if (request.body.Count() > 0)
|
|
{
|
|
context->requestBody.Resize(request.body.Count());
|
|
memcpy(&context->requestBody[0], &request.body.Get(0), request.body.Count());
|
|
}
|
|
|
|
auto failBeforeCallbackInstalled = [&](const WString& operation, DWORD errorCode)
|
|
{
|
|
WinHttpCloseHandle(httpRequest);
|
|
delete contextPtr;
|
|
EndPendingCallback();
|
|
if (callback)
|
|
{
|
|
callback(Variant<HttpResponse, HttpError>(MakeError(operation, errorCode)));
|
|
}
|
|
};
|
|
|
|
DWORD_PTR contextValue = reinterpret_cast<DWORD_PTR>(contextPtr);
|
|
httpResult = WinHttpSetOption(
|
|
httpRequest,
|
|
WINHTTP_OPTION_CONTEXT_VALUE,
|
|
&contextValue,
|
|
sizeof(contextValue));
|
|
lastError = GetLastError();
|
|
if (httpResult == FALSE)
|
|
{
|
|
failBeforeCallbackInstalled(L"WinHttpSetOption(WINHTTP_OPTION_CONTEXT_VALUE)", lastError);
|
|
return;
|
|
}
|
|
|
|
if (request.username != WString::Empty && request.password != WString::Empty)
|
|
{
|
|
httpResult = WinHttpSetCredentials(
|
|
httpRequest,
|
|
WINHTTP_AUTH_TARGET_SERVER,
|
|
WINHTTP_AUTH_SCHEME_BASIC,
|
|
request.username.Buffer(),
|
|
request.password.Buffer(),
|
|
NULL);
|
|
lastError = GetLastError();
|
|
if (httpResult == FALSE)
|
|
{
|
|
failBeforeCallbackInstalled(L"WinHttpSetCredentials", lastError);
|
|
return;
|
|
}
|
|
}
|
|
if (request.contentType != WString::Empty)
|
|
{
|
|
httpResult = WinHttpAddRequestHeaders(
|
|
httpRequest,
|
|
(L"Content-Type: " + request.contentType).Buffer(),
|
|
-1,
|
|
WINHTTP_ADDREQ_FLAG_REPLACE | WINHTTP_ADDREQ_FLAG_ADD);
|
|
lastError = GetLastError();
|
|
if (httpResult == FALSE)
|
|
{
|
|
failBeforeCallbackInstalled(L"WinHttpAddRequestHeaders(content-type)", lastError);
|
|
return;
|
|
}
|
|
}
|
|
if (request.cookie != WString::Empty)
|
|
{
|
|
httpResult = WinHttpAddRequestHeaders(
|
|
httpRequest,
|
|
(L"Cookie: " + request.cookie).Buffer(),
|
|
-1,
|
|
WINHTTP_ADDREQ_FLAG_REPLACE | WINHTTP_ADDREQ_FLAG_ADD);
|
|
lastError = GetLastError();
|
|
if (httpResult == FALSE)
|
|
{
|
|
failBeforeCallbackInstalled(L"WinHttpAddRequestHeaders(cookie)", lastError);
|
|
return;
|
|
}
|
|
}
|
|
|
|
for (vint i = 0; i < request.extraHeaders.Count(); i++)
|
|
{
|
|
WString key = request.extraHeaders.Keys()[i];
|
|
WString value = request.extraHeaders.Values().Get(i);
|
|
httpResult = WinHttpAddRequestHeaders(
|
|
httpRequest,
|
|
(key + L": " + value).Buffer(),
|
|
-1,
|
|
WINHTTP_ADDREQ_FLAG_REPLACE | WINHTTP_ADDREQ_FLAG_ADD);
|
|
lastError = GetLastError();
|
|
if (httpResult == FALSE)
|
|
{
|
|
failBeforeCallbackInstalled(L"WinHttpAddRequestHeaders(extra)", lastError);
|
|
return;
|
|
}
|
|
}
|
|
|
|
bool failedBeforeSend = false;
|
|
WString failedOperation;
|
|
DWORD failedError = 0;
|
|
SPIN_LOCK(lockActiveRequests)
|
|
{
|
|
if (stopping)
|
|
{
|
|
failedBeforeSend = true;
|
|
failedOperation = L"HttpClientApi::HttpQuery";
|
|
failedError = ERROR_OPERATION_ABORTED;
|
|
}
|
|
else
|
|
{
|
|
auto previousCallback = WinHttpSetStatusCallback(
|
|
httpRequest,
|
|
HttpStatusCallback,
|
|
WINHTTP_CALLBACK_FLAG_ALL_NOTIFICATIONS,
|
|
NULL);
|
|
lastError = GetLastError();
|
|
if (previousCallback == WINHTTP_INVALID_STATUS_CALLBACK)
|
|
{
|
|
failedBeforeSend = true;
|
|
failedOperation = L"WinHttpSetStatusCallback";
|
|
failedError = lastError;
|
|
}
|
|
else
|
|
{
|
|
AttachRequestUnsafe(context);
|
|
}
|
|
}
|
|
}
|
|
if (failedBeforeSend)
|
|
{
|
|
failBeforeCallbackInstalled(failedOperation, failedError);
|
|
return;
|
|
}
|
|
|
|
DWORD requestBodyLength = (DWORD)context->requestBody.Count();
|
|
LPVOID requestBodyBuffer = requestBodyLength == 0 ? WINHTTP_NO_REQUEST_DATA : (LPVOID)&context->requestBody[0];
|
|
httpResult = WinHttpSendRequest(
|
|
httpRequest,
|
|
WINHTTP_NO_ADDITIONAL_HEADERS,
|
|
0,
|
|
requestBodyBuffer,
|
|
requestBodyLength,
|
|
requestBodyLength,
|
|
contextValue);
|
|
lastError = GetLastError();
|
|
if (httpResult == FALSE)
|
|
{
|
|
CompleteRequestWithLastError(context, L"WinHttpSendRequest", lastError);
|
|
}
|
|
}
|
|
|
|
void HttpClientApi::Stop()
|
|
{
|
|
if (httpSession == NULL) return;
|
|
|
|
List<HINTERNET> stoppingRequests;
|
|
SPIN_LOCK(lockActiveRequests)
|
|
{
|
|
stopping = true;
|
|
for (auto&& context : activeRequests)
|
|
{
|
|
HINTERNET httpRequest = NULL;
|
|
SPIN_LOCK(context->lockContext)
|
|
{
|
|
if (!context->keepAliveOnStop && !context->closing)
|
|
{
|
|
context->closing = true;
|
|
httpRequest = context->httpRequest;
|
|
}
|
|
}
|
|
if (httpRequest)
|
|
{
|
|
stoppingRequests.Add(httpRequest);
|
|
}
|
|
}
|
|
}
|
|
for (auto httpRequest : stoppingRequests)
|
|
{
|
|
WinHttpCloseHandle(httpRequest);
|
|
}
|
|
|
|
eventPendingCallbacks.Wait();
|
|
|
|
WinHttpCloseHandle(httpConnection);
|
|
WinHttpSetStatusCallback(
|
|
httpSession,
|
|
NULL,
|
|
WINHTTP_CALLBACK_FLAG_ALL_NOTIFICATIONS,
|
|
NULL);
|
|
WinHttpCloseHandle(httpSession);
|
|
|
|
SPIN_LOCK(lockActiveRequests)
|
|
{
|
|
activeRequests.Clear();
|
|
}
|
|
|
|
httpConnection = NULL;
|
|
httpSession = NULL;
|
|
}
|
|
|
|
WString HttpClientApi::UrlEncodeQuery(const WString& query)
|
|
{
|
|
vint utf8Size = WideCharToMultiByte(CP_UTF8, 0, query.Buffer(), (int)query.Length(), NULL, 0, NULL, NULL);
|
|
Array<char> utf8(utf8Size);
|
|
if (utf8Size > 0)
|
|
{
|
|
WideCharToMultiByte(CP_UTF8, 0, query.Buffer(), (int)query.Length(), &utf8[0], (int)utf8Size, NULL, NULL);
|
|
}
|
|
|
|
Array<wchar_t> encoded(utf8Size * 3 + 1);
|
|
ZeroMemory(&encoded[0], encoded.Count() * sizeof(wchar_t));
|
|
wchar_t* writing = &encoded[0];
|
|
for (vint i = 0; i < utf8Size; i++)
|
|
{
|
|
unsigned char x = (unsigned char)utf8[i];
|
|
if ((L'a' <= x && x <= L'z') || (L'A' <= x && x <= L'Z') || (L'0' <= x && x <= L'9'))
|
|
{
|
|
writing[0] = x;
|
|
writing += 1;
|
|
}
|
|
else
|
|
{
|
|
writing[0] = L'%';
|
|
writing[1] = L"0123456789ABCDEF"[x / 16];
|
|
writing[2] = L"0123456789ABCDEF"[x % 16];
|
|
writing += 3;
|
|
}
|
|
}
|
|
|
|
return &encoded[0];
|
|
}
|
|
|
|
WString HttpClientApi::UrlDecodeQuery(const WString& query)
|
|
{
|
|
List<char> utf8;
|
|
for (vint i = 0; i < query.Length(); i++)
|
|
{
|
|
wchar_t c = query[i];
|
|
if (c == L'%' && i + 2 < query.Length())
|
|
{
|
|
vint high = HexValue(query[i + 1]);
|
|
vint low = HexValue(query[i + 2]);
|
|
if (high != -1 && low != -1)
|
|
{
|
|
utf8.Add((char)(high * 16 + low));
|
|
i += 2;
|
|
continue;
|
|
}
|
|
}
|
|
|
|
if (c == L'+')
|
|
{
|
|
utf8.Add(' ');
|
|
}
|
|
else if (c <= 0x7F)
|
|
{
|
|
utf8.Add((char)c);
|
|
}
|
|
else
|
|
{
|
|
wchar_t single[] = { c, 0 };
|
|
vint utf8Size = WideCharToMultiByte(CP_UTF8, 0, single, 1, NULL, 0, NULL, NULL);
|
|
if (utf8Size > 0)
|
|
{
|
|
Array<char> singleUtf8(utf8Size);
|
|
WideCharToMultiByte(CP_UTF8, 0, single, 1, &singleUtf8[0], (int)utf8Size, NULL, NULL);
|
|
for (vint j = 0; j < utf8Size; j++)
|
|
{
|
|
utf8.Add(singleUtf8[j]);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
if (utf8.Count() == 0)
|
|
{
|
|
return WString::Empty;
|
|
}
|
|
|
|
vint utf16Size = MultiByteToWideChar(CP_UTF8, 0, &utf8[0], (int)utf8.Count(), NULL, 0);
|
|
Array<wchar_t> utf16(utf16Size + 1);
|
|
ZeroMemory(&utf16[0], utf16.Count() * sizeof(wchar_t));
|
|
if (utf16Size > 0)
|
|
{
|
|
MultiByteToWideChar(CP_UTF8, 0, &utf8[0], (int)utf8.Count(), &utf16[0], (int)utf16Size);
|
|
}
|
|
return &utf16[0];
|
|
}
|
|
|
|
}
|
|
|
|
|
|
/***********************************************************************
|
|
.\INTERPROCESS\WINDOWS\HTTPSERVER.WINDOWS.CPP
|
|
***********************************************************************/
|
|
|
|
namespace vl::inter_process
|
|
{
|
|
|
|
using namespace vl::collections;
|
|
|
|
/***********************************************************************
|
|
HttpServerConnection
|
|
***********************************************************************/
|
|
|
|
void HttpServerConnection::OnCancelCurrentHttpRequestForPendingRequest()
|
|
{
|
|
if (httpPendingRequestId != HTTP_NULL_ID)
|
|
{
|
|
if (!server)
|
|
{
|
|
httpPendingRequestId = HTTP_NULL_ID;
|
|
return;
|
|
}
|
|
ULONG result = HttpCancelHttpRequest(
|
|
server->GetHttpRequestQueue(),
|
|
httpPendingRequestId,
|
|
NULL);
|
|
CHECK_ERROR(
|
|
result == NO_ERROR || result == ERROR_CONNECTION_INVALID || result == ERROR_OPERATION_ABORTED,
|
|
L"HttpCancelHttpRequest failed for canceling outdated /Request.");
|
|
httpPendingRequestId = HTTP_NULL_ID;
|
|
}
|
|
}
|
|
|
|
void HttpServerConnection::OnNewHttpRequestForPendingRequest(HTTP_REQUEST_ID httpRequestId)
|
|
{
|
|
OnCancelCurrentHttpRequestForPendingRequest();
|
|
httpPendingRequestId = httpRequestId;
|
|
if (pendingRequestsToSend.Count() > 0)
|
|
{
|
|
auto pendingRequest = pendingRequestsToSend[0];
|
|
pendingRequestsToSend.RemoveAt(0);
|
|
HttpServerApi::SendResponseUtf8(server->GetHttpRequestQueue(), httpPendingRequestId, pendingRequest);
|
|
httpPendingRequestId = HTTP_NULL_ID;
|
|
}
|
|
}
|
|
|
|
WString HttpServerConnection::SubmitResponse(PHTTP_REQUEST pRequest)
|
|
{
|
|
auto body = server->GetUtf8Body(pRequest).Value();
|
|
|
|
SPIN_LOCK(pendingRequestLock)
|
|
{
|
|
submittingResponse = true;
|
|
}
|
|
|
|
try
|
|
{
|
|
SPIN_LOCK(lockQueuedStrings)
|
|
{
|
|
if (callback)
|
|
{
|
|
callback->OnReadString(body);
|
|
}
|
|
else
|
|
{
|
|
queuedStrings.Add(body);
|
|
}
|
|
}
|
|
}
|
|
catch (...)
|
|
{
|
|
SPIN_LOCK(pendingRequestLock)
|
|
{
|
|
submittingResponse = false;
|
|
responsesToSubmit.Clear();
|
|
}
|
|
throw;
|
|
}
|
|
|
|
WString responseToClient;
|
|
SPIN_LOCK(pendingRequestLock)
|
|
{
|
|
submittingResponse = false;
|
|
if (responsesToSubmit.Count() > 0)
|
|
{
|
|
responseToClient = responsesToSubmit[0];
|
|
responsesToSubmit.RemoveAt(0);
|
|
while (responsesToSubmit.Count() > 0)
|
|
{
|
|
pendingRequestsToSend.Add(responsesToSubmit[0]);
|
|
responsesToSubmit.RemoveAt(0);
|
|
}
|
|
}
|
|
else if (pendingRequestsToSend.Count() > 0)
|
|
{
|
|
responseToClient = pendingRequestsToSend[0];
|
|
pendingRequestsToSend.RemoveAt(0);
|
|
}
|
|
}
|
|
return responseToClient;
|
|
}
|
|
|
|
void HttpServerConnection::InstallCallback(INetworkProtocolCallback* _callback)
|
|
{
|
|
CHECK_ERROR(_callback, L"HttpServerConnection::InstallCallback needs a valid INetworkProtocolCallback.");
|
|
_callback->OnInstalled(this);
|
|
|
|
List<WString> strings;
|
|
SPIN_LOCK(lockQueuedStrings)
|
|
{
|
|
callback = _callback;
|
|
strings = std::move(queuedStrings);
|
|
}
|
|
for (const auto& str : strings)
|
|
{
|
|
_callback->OnReadString(str);
|
|
}
|
|
}
|
|
|
|
void HttpServerConnection::BeginReadingLoopUnsafe()
|
|
{
|
|
// Do nothing, HttpServer automatically handles this.
|
|
}
|
|
|
|
void HttpServerConnection::SendString(const WString& str)
|
|
{
|
|
SPIN_LOCK(pendingRequestLock)
|
|
{
|
|
if (submittingResponse)
|
|
{
|
|
responsesToSubmit.Add(str);
|
|
}
|
|
else if (httpPendingRequestId != HTTP_NULL_ID)
|
|
{
|
|
ULONG result = HttpServerApi::SendResponse(server->GetHttpRequestQueue(), httpPendingRequestId, { 200, L"OK", str, L"application/json; charset=utf8" });
|
|
if (result == NO_ERROR)
|
|
{
|
|
httpPendingRequestId = HTTP_NULL_ID;
|
|
}
|
|
else if (result == ERROR_CONNECTION_INVALID || result == ERROR_OPERATION_ABORTED)
|
|
{
|
|
httpPendingRequestId = HTTP_NULL_ID;
|
|
pendingRequestsToSend.Add(str);
|
|
}
|
|
else
|
|
{
|
|
CHECK_FAIL(L"HttpSendHttpResponse failed for responding /Request.");
|
|
}
|
|
}
|
|
else
|
|
{
|
|
pendingRequestsToSend.Add(str);
|
|
}
|
|
}
|
|
}
|
|
|
|
void HttpServerConnection::Stop()
|
|
{
|
|
auto holding = Ptr(this);
|
|
if (server)
|
|
{
|
|
SPIN_LOCK(server->lockConnections)
|
|
{
|
|
server->connections.Remove(guid);
|
|
}
|
|
SPIN_LOCK(pendingRequestLock)
|
|
{
|
|
OnCancelCurrentHttpRequestForPendingRequest();
|
|
}
|
|
if (callback)
|
|
{
|
|
callback->OnDisconnected();
|
|
}
|
|
}
|
|
}
|
|
|
|
WString HttpServerConnection::GenerateNewGuid()
|
|
{
|
|
RPC_STATUS status = -1;
|
|
UUID guid;
|
|
status = UuidCreate(&guid);
|
|
CHECK_ERROR(status == RPC_S_OK, L"UuidCreate failed.");
|
|
|
|
RPC_WSTR guidString = nullptr;
|
|
status = UuidToString(&guid, &guidString);
|
|
CHECK_ERROR(status == RPC_S_OK, L"UuidToString failed.");
|
|
|
|
WString result = guidString;
|
|
status = RpcStringFree(&guidString);
|
|
CHECK_ERROR(status == RPC_S_OK, L"RpcStringFree failed.");
|
|
return result;
|
|
}
|
|
|
|
/***********************************************************************
|
|
HttpServer (HttpServerApi)
|
|
***********************************************************************/
|
|
|
|
void HttpServer::OnHttpRequestReceived(PHTTP_REQUEST pRequest)
|
|
{
|
|
bool isValidRequest = wcsncmp(pRequest->CookedUrl.pAbsPath, urlRequestPrefix.Buffer(), urlRequestPrefix.Length()) == 0;
|
|
bool isValidResponse = wcsncmp(pRequest->CookedUrl.pAbsPath, urlResponsePrefix.Buffer(), urlResponsePrefix.Length()) == 0;
|
|
|
|
auto FindExistingConnection = [=, this](const WString& guid)->Ptr<HttpServerConnection>
|
|
{
|
|
SPIN_LOCK(lockConnections)
|
|
{
|
|
vint index = connections.Keys().IndexOf(guid);
|
|
if (index == -1)
|
|
{
|
|
HttpServerApi::SendResponse(GetHttpRequestQueue(), pRequest->RequestId, { 404, L"Unknown connection guid" });
|
|
}
|
|
else
|
|
{
|
|
return connections.Values()[index];
|
|
}
|
|
}
|
|
return {};
|
|
};
|
|
|
|
if (pRequest->Verb == HttpVerbGET && pRequest->CookedUrl.pAbsPath == urlConnect)
|
|
{
|
|
auto newGuid = HttpServerConnection::GenerateNewGuid();
|
|
auto connection = Ptr(new HttpServerConnection);
|
|
connection->server = this;
|
|
connection->guid = newGuid;
|
|
SPIN_LOCK(lockConnections)
|
|
{
|
|
connections.Add(newGuid, connection);
|
|
}
|
|
auto result = OnClientConnected(connection.Obj());
|
|
if (result == WaitForClientResult::Reject)
|
|
{
|
|
SPIN_LOCK(lockConnections)
|
|
{
|
|
connections.Remove(newGuid);
|
|
}
|
|
connection->server = nullptr;
|
|
HttpServerApi::SendResponse(GetHttpRequestQueue(), pRequest->RequestId, { 404, L"Connection rejected" });
|
|
}
|
|
else
|
|
{
|
|
auto completeUrlRequest = WString::Unmanaged(HttpServerUrl_Request) + L"/" + newGuid;
|
|
auto completeUrlResponse = WString::Unmanaged(HttpServerUrl_Response) + L"/" + newGuid;
|
|
HttpServerApi::SendResponseUtf8(GetHttpRequestQueue(), pRequest->RequestId, completeUrlRequest + L";" + completeUrlResponse);
|
|
}
|
|
}
|
|
else if (pRequest->Verb == HttpVerbPOST && isValidRequest)
|
|
{
|
|
auto guid = WString::Unmanaged(pRequest->CookedUrl.pAbsPath + urlRequestPrefix.Length());
|
|
if (auto connection = FindExistingConnection(guid))
|
|
{
|
|
SPIN_LOCK(connection->pendingRequestLock)
|
|
{
|
|
connection->OnNewHttpRequestForPendingRequest(pRequest->RequestId);
|
|
}
|
|
}
|
|
}
|
|
else if (pRequest->Verb == HttpVerbPOST && isValidResponse)
|
|
{
|
|
auto guid = WString::Unmanaged(pRequest->CookedUrl.pAbsPath + urlResponsePrefix.Length());
|
|
if (auto connection = FindExistingConnection(guid))
|
|
{
|
|
auto responseToClient = connection->SubmitResponse(pRequest);
|
|
auto result = HttpServerApi::SendResponse(GetHttpRequestQueue(), pRequest->RequestId, { 200, L"OK", responseToClient, L"application/json; charset=utf8" });
|
|
CHECK_ERROR(
|
|
result == NO_ERROR || result == ERROR_CONNECTION_INVALID || result == ERROR_OPERATION_ABORTED,
|
|
L"HttpSendHttpResponse failed for responding /Response."
|
|
);
|
|
}
|
|
}
|
|
else
|
|
{
|
|
HttpServerApi::SendResponse(GetHttpRequestQueue(), pRequest->RequestId, { 404, L"Unknown URL" });
|
|
}
|
|
}
|
|
|
|
void HttpServer::OnHttpServerStopping()
|
|
{
|
|
List<Ptr<HttpServerConnection>> stoppingConnections;
|
|
SPIN_LOCK(lockConnections)
|
|
{
|
|
for (auto connection : connections.Values())
|
|
{
|
|
stoppingConnections.Add(connection);
|
|
}
|
|
connections.Clear();
|
|
}
|
|
for (auto connection : stoppingConnections)
|
|
{
|
|
SPIN_LOCK(connection->pendingRequestLock)
|
|
{
|
|
connection->OnCancelCurrentHttpRequestForPendingRequest();
|
|
}
|
|
connection->server = nullptr;
|
|
}
|
|
for (auto connection : stoppingConnections)
|
|
{
|
|
if (connection->callback)
|
|
{
|
|
connection->callback->OnDisconnected();
|
|
}
|
|
}
|
|
}
|
|
|
|
/***********************************************************************
|
|
HttpServer
|
|
***********************************************************************/
|
|
|
|
HttpServer::HttpServer(const WString _baseUrl, vint port)
|
|
: HttpServerApi(WString::Unmanaged(L"http://localhost:") + itow(port) + _baseUrl + L"/", true)
|
|
, baseUrl(_baseUrl)
|
|
{
|
|
urlConnect = baseUrl + HttpServerUrl_Connect;
|
|
urlRequestPrefix = baseUrl + HttpServerUrl_Request + L"/";
|
|
urlResponsePrefix = baseUrl + HttpServerUrl_Response + L"/";
|
|
}
|
|
|
|
HttpServer::~HttpServer()
|
|
{
|
|
Stop();
|
|
}
|
|
|
|
WaitForClientResult HttpServer::OnClientConnected(INetworkProtocolConnection* connection)
|
|
{
|
|
return WaitForClientResult::Accept;
|
|
}
|
|
|
|
void HttpServer::Start()
|
|
{
|
|
HttpServerApi::Start();
|
|
}
|
|
|
|
void HttpServer::Stop()
|
|
{
|
|
HttpServerApi::Stop();
|
|
}
|
|
|
|
bool HttpServer::IsStopped()
|
|
{
|
|
return HttpServerApi::IsStopped();
|
|
}
|
|
|
|
}
|
|
|
|
|
|
/***********************************************************************
|
|
.\INTERPROCESS\WINDOWS\HTTPSERVERAPI.WINDOWS.CPP
|
|
***********************************************************************/
|
|
|
|
#ifndef VCZH_MSVC
|
|
static_assert(false, "Do not build this file for non-Windows applications.");
|
|
#endif
|
|
|
|
#pragma comment(lib, "Httpapi.lib")
|
|
|
|
namespace vl::inter_process
|
|
{
|
|
|
|
using namespace vl::collections;
|
|
|
|
/***********************************************************************
|
|
HttpServerApi (ListenToHttpRequest)
|
|
***********************************************************************/
|
|
|
|
void HttpServerApi::OnHttpConnectionBrokenUnsafe()
|
|
{
|
|
if (state == State::Running)
|
|
{
|
|
state = State::Stopping;
|
|
OnHttpServerStopping();
|
|
}
|
|
}
|
|
|
|
void HttpServerApi::OnHttpRequestReceivedUnsafe(PHTTP_REQUEST pRequest)
|
|
{
|
|
if (state == State::Stopping)
|
|
{
|
|
SendResponse(httpRequestQueue, pRequest->RequestId, { 404, L"Server is stopping" });
|
|
return;
|
|
}
|
|
|
|
if (respondToOptions && pRequest->Verb == HttpVerbOPTIONS)
|
|
{
|
|
SendOptionsResponse(httpRequestQueue, pRequest->RequestId);
|
|
}
|
|
else
|
|
{
|
|
OnHttpRequestReceived(pRequest);
|
|
}
|
|
ListenToHttpRequest();
|
|
}
|
|
|
|
ULONG HttpServerApi::ListenToHttpRequest_Init(OVERLAPPED* overlapped)
|
|
{
|
|
ZeroMemory(&bufferRequest[0], bufferRequest.Count());
|
|
|
|
ULONG result = HttpReceiveHttpRequest(
|
|
httpRequestQueue,
|
|
HTTP_NULL_ID,
|
|
0,
|
|
(PHTTP_REQUEST)&bufferRequest[0],
|
|
(ULONG)bufferRequest.Count(),
|
|
NULL,
|
|
overlapped);
|
|
|
|
return result;
|
|
}
|
|
|
|
ULONG HttpServerApi::ListenToHttpRequest_InitMoreData(ULONG* bytesReturned)
|
|
{
|
|
HTTP_REQUEST_ID httpRequestIdReading = ((PHTTP_REQUEST)&bufferRequest[0])->RequestId;
|
|
ZeroMemory(&bufferRequest[0], bufferRequest.Count());
|
|
|
|
ULONG result = HttpReceiveHttpRequest(
|
|
httpRequestQueue,
|
|
httpRequestIdReading,
|
|
0,
|
|
(PHTTP_REQUEST)&bufferRequest[0],
|
|
(ULONG)bufferRequest.Count(),
|
|
bytesReturned,
|
|
NULL);
|
|
|
|
return result;
|
|
}
|
|
|
|
ULONG HttpServerApi::ListenToHttpRequest_OverlappedMoreData(vint expectedBufferSize)
|
|
{
|
|
HTTP_REQUEST_ID httpRequestIdReading = ((PHTTP_REQUEST)&bufferRequest[0])->RequestId;
|
|
bufferRequest.Resize(expectedBufferSize);
|
|
ZeroMemory(&bufferRequest[0], bufferRequest.Count());
|
|
|
|
ULONG bytesReturned = 0;
|
|
ULONG result = HttpReceiveHttpRequest(
|
|
httpRequestQueue,
|
|
httpRequestIdReading,
|
|
0,
|
|
(PHTTP_REQUEST)&bufferRequest[0],
|
|
(ULONG)bufferRequest.Count(),
|
|
&bytesReturned,
|
|
NULL);
|
|
|
|
return result;
|
|
}
|
|
|
|
void HttpServerApi::ListenToHttpRequest()
|
|
{
|
|
if (state == State::Stopping) return;
|
|
|
|
ResetEvent(hEventRequest);
|
|
ZeroMemory(&overlappedRequest, sizeof(overlappedRequest));
|
|
overlappedRequest.hEvent = hEventRequest;
|
|
|
|
ZeroMemory(&bufferRequest[0], sizeof(HTTP_REQUEST));
|
|
PHTTP_REQUEST pRequest = (PHTTP_REQUEST)&bufferRequest[0];
|
|
|
|
ULONG result = ListenToHttpRequest_Init(&overlappedRequest);
|
|
if (result == ERROR_CONNECTION_INVALID || result == ERROR_OPERATION_ABORTED)
|
|
{
|
|
OnHttpConnectionBrokenUnsafe();
|
|
return;
|
|
}
|
|
|
|
if (result == NO_ERROR)
|
|
{
|
|
OnHttpRequestReceivedUnsafe(pRequest);
|
|
return;
|
|
}
|
|
|
|
if (result == ERROR_MORE_DATA)
|
|
{
|
|
ULONG bytesReturned = 0;
|
|
result = ListenToHttpRequest_InitMoreData(&bytesReturned);
|
|
if (result == ERROR_CONNECTION_INVALID || result == ERROR_OPERATION_ABORTED)
|
|
{
|
|
OnHttpConnectionBrokenUnsafe();
|
|
return;
|
|
}
|
|
CHECK_ERROR(result == ERROR_MORE_DATA, L"HttpReceiveHttpRequest(#1) failed on unexpected result.");
|
|
|
|
result = ListenToHttpRequest_OverlappedMoreData((vint)bytesReturned);
|
|
if (result == ERROR_CONNECTION_INVALID || result == ERROR_OPERATION_ABORTED)
|
|
{
|
|
OnHttpConnectionBrokenUnsafe();
|
|
return;
|
|
}
|
|
CHECK_ERROR(result == NO_ERROR, L"HttpReceiveHttpRequest(#2) failed on unexpected result.");
|
|
|
|
PHTTP_REQUEST pRequest = (PHTTP_REQUEST)&bufferRequest[0];
|
|
OnHttpRequestReceivedUnsafe(pRequest);
|
|
return;
|
|
}
|
|
|
|
CHECK_ERROR(result == ERROR_IO_PENDING, L"HttpReceiveHttpRequest(#3) failed on unexpected result.");
|
|
|
|
BOOL waitResult = RegisterWaitForSingleObject(
|
|
&hWaitHandleRequest,
|
|
hEventRequest,
|
|
[](PVOID lpParameter, BOOLEAN TimerOrWaitFired)
|
|
{
|
|
auto self = (HttpServerApi*)lpParameter;
|
|
struct PendingCallbackScope
|
|
{
|
|
HttpServerApi* server;
|
|
|
|
PendingCallbackScope(HttpServerApi* _server)
|
|
: server(_server)
|
|
{
|
|
server->BeginPendingCallback();
|
|
}
|
|
|
|
~PendingCallbackScope()
|
|
{
|
|
server->EndPendingCallback();
|
|
}
|
|
} pendingCallbackScope(self);
|
|
|
|
auto waitHandle = std::atomic_ref<HANDLE>(self->hWaitHandleRequest).exchange(INVALID_HANDLE_VALUE);
|
|
if (waitHandle != INVALID_HANDLE_VALUE)
|
|
{
|
|
UnregisterWait(waitHandle);
|
|
}
|
|
|
|
DWORD read = 0;
|
|
BOOL result = GetOverlappedResult(self->httpRequestQueue, &self->overlappedRequest, &read, FALSE);
|
|
if (result == TRUE)
|
|
{
|
|
PHTTP_REQUEST pRequest = (PHTTP_REQUEST)&self->bufferRequest[0];
|
|
self->OnHttpRequestReceivedUnsafe(pRequest);
|
|
}
|
|
else
|
|
{
|
|
DWORD error = GetLastError();
|
|
if (error == ERROR_CONNECTION_INVALID || error == ERROR_OPERATION_ABORTED)
|
|
{
|
|
self->OnHttpConnectionBrokenUnsafe();
|
|
return;
|
|
}
|
|
CHECK_ERROR(error == ERROR_MORE_DATA, L"GetOverlappedResult(#4) failed on unexpected GetLastError.");
|
|
CHECK_ERROR(self->bufferRequest.Count() < (vint)read, L"GetOverlappedResult(#5) failed on unexpected read size.");
|
|
|
|
ULONG result = self->ListenToHttpRequest_OverlappedMoreData((vint)read);
|
|
if (result == ERROR_CONNECTION_INVALID || result == ERROR_OPERATION_ABORTED)
|
|
{
|
|
self->OnHttpConnectionBrokenUnsafe();
|
|
return;
|
|
}
|
|
CHECK_ERROR(result == NO_ERROR, L"HttpReceiveHttpRequest(#6) failed on unexpected result.");
|
|
|
|
PHTTP_REQUEST pRequest = (PHTTP_REQUEST)&self->bufferRequest[0];
|
|
self->OnHttpRequestReceivedUnsafe(pRequest);
|
|
}
|
|
},
|
|
this,
|
|
INFINITE,
|
|
WT_EXECUTEONLYONCE);
|
|
CHECK_ERROR(waitResult, L"RegisterWaitForSingleObject failed for HttpReceiveHttpRequest.");
|
|
if (state == State::Stopping)
|
|
{
|
|
auto waitHandle = std::atomic_ref<HANDLE>(hWaitHandleRequest).exchange(INVALID_HANDLE_VALUE);
|
|
if (waitHandle != INVALID_HANDLE_VALUE)
|
|
{
|
|
UnregisterWaitEx(waitHandle, INVALID_HANDLE_VALUE);
|
|
}
|
|
}
|
|
}
|
|
|
|
void HttpServerApi::BeginPendingCallback()
|
|
{
|
|
if (pendingCallbacks++ == 0)
|
|
{
|
|
eventPendingCallbacks.Unsignal();
|
|
}
|
|
}
|
|
|
|
void HttpServerApi::EndPendingCallback()
|
|
{
|
|
if (--pendingCallbacks == 0)
|
|
{
|
|
eventPendingCallbacks.Signal();
|
|
}
|
|
}
|
|
|
|
void HttpServerApi::OnHttpServerStopping()
|
|
{
|
|
}
|
|
|
|
HANDLE HttpServerApi::GetHttpRequestQueue() const
|
|
{
|
|
return httpRequestQueue;
|
|
}
|
|
|
|
Nullable<WString> HttpServerApi::GetUtf8Body(PHTTP_REQUEST pRequest)
|
|
{
|
|
ULONG bodyLength = 0;
|
|
ULONG bodyReceived = 0;
|
|
{
|
|
auto& headerContentType = pRequest->Headers.KnownHeaders[HttpHeaderContentType];
|
|
CHECK_ERROR(headerContentType.pRawValue != NULL, L"/Response missing Content-Type header.");
|
|
CHECK_ERROR(
|
|
strncmp((const char*)headerContentType.pRawValue, "application/json; charset=utf8", headerContentType.RawValueLength) == 0,
|
|
L"/Response Content-Type header must be \"application/json; charset=utf8\".");
|
|
}
|
|
{
|
|
auto& headerContentLength = pRequest->Headers.KnownHeaders[HttpHeaderContentLength];
|
|
CHECK_ERROR(headerContentLength.pRawValue != NULL, L"/Response missing Content-Type header.");
|
|
bodyLength = (ULONG)atoi(headerContentLength.pRawValue);
|
|
}
|
|
CHECK_ERROR(bodyLength > 0, L"/Response must contain body data.");
|
|
|
|
Array<char8_t> bodyBuffer(bodyLength + 1);
|
|
ZeroMemory(&bodyBuffer[0], bodyBuffer.Count() * sizeof(char8_t));
|
|
ULONG bodyWritten = 0;
|
|
for (USHORT i = 0; i < pRequest->EntityChunkCount; i++)
|
|
{
|
|
auto& chunk = pRequest->pEntityChunks[i];
|
|
CHECK_ERROR(chunk.DataChunkType == HttpDataChunkFromMemory, L"/Response contains an unsupported body chunk.");
|
|
auto chunkLength = chunk.FromMemory.BufferLength;
|
|
CHECK_ERROR(bodyWritten + chunkLength <= bodyLength, L"/Response body is longer than Content-Length.");
|
|
memcpy(&bodyBuffer[bodyWritten], chunk.FromMemory.pBuffer, chunkLength);
|
|
bodyWritten += chunkLength;
|
|
}
|
|
if (pRequest->Flags & HTTP_REQUEST_FLAG_MORE_ENTITY_BODY_EXISTS)
|
|
{
|
|
ULONG result = NO_ERROR;
|
|
result = HttpReceiveRequestEntityBody(
|
|
httpRequestQueue,
|
|
pRequest->RequestId,
|
|
HTTP_RECEIVE_REQUEST_ENTITY_BODY_FLAG_FILL_BUFFER,
|
|
&bodyBuffer[bodyWritten],
|
|
bodyLength - bodyWritten,
|
|
&bodyReceived,
|
|
NULL);
|
|
CHECK_ERROR(result == NO_ERROR || result == ERROR_HANDLE_EOF, L"HttpReceiveRequestEntityBody.");
|
|
bodyWritten += bodyReceived;
|
|
}
|
|
CHECK_ERROR(bodyWritten == bodyLength, L"/Response body is shorter than Content-Length.");
|
|
|
|
U8String bodyUtf8 = U8String::Unmanaged(&bodyBuffer[0]);
|
|
return u8tow(bodyUtf8);
|
|
}
|
|
|
|
/***********************************************************************
|
|
HttpServerApi (Writing)
|
|
***********************************************************************/
|
|
|
|
void HttpServerApi::SendOptionsResponse(HANDLE httpRequestQueue, HTTP_REQUEST_ID requestId)
|
|
{
|
|
ULONG bytesSent = 0;
|
|
HTTP_RESPONSE httpResponse;
|
|
ZeroMemory(&httpResponse, sizeof(httpResponse));
|
|
|
|
httpResponse.StatusCode = 200;
|
|
|
|
static const char headerACAOName[] = "Access-Control-Allow-Origin";
|
|
static const char headerACAMName[] = "Access-Control-Allow-Methods";
|
|
static const char headerACAMValue[] = "GET, POST, OPTIONS";
|
|
static const char headerACAHName[] = "Access-Control-Allow-Headers";
|
|
static const char headerACAHValue[] = "Content-Type";
|
|
static HTTP_UNKNOWN_HEADER unknownHeaders[] = { {
|
|
sizeof(headerACAOName) - 1,
|
|
1,
|
|
headerACAOName,
|
|
"*"
|
|
},{
|
|
sizeof(headerACAMName) - 1,
|
|
sizeof(headerACAMValue) - 1,
|
|
headerACAMName,
|
|
headerACAMValue
|
|
},{
|
|
sizeof(headerACAHName) - 1,
|
|
sizeof(headerACAHValue) - 1,
|
|
headerACAHName,
|
|
headerACAHValue
|
|
} };
|
|
httpResponse.Headers.UnknownHeaderCount = sizeof(unknownHeaders) / sizeof(HTTP_UNKNOWN_HEADER);
|
|
httpResponse.Headers.pUnknownHeaders = unknownHeaders;
|
|
|
|
ULONG result = NO_ERROR;
|
|
result = HttpSendHttpResponse(
|
|
httpRequestQueue,
|
|
requestId,
|
|
0,
|
|
&httpResponse,
|
|
NULL,
|
|
&bytesSent,
|
|
NULL,
|
|
0,
|
|
NULL,
|
|
NULL);
|
|
CHECK_ERROR(result == NO_ERROR, L"HttpSendHttpResponse failed (OPTIONS).");
|
|
}
|
|
|
|
ULONG HttpServerApi::SendResponse(HANDLE httpRequestQueue, HTTP_REQUEST_ID requestId, const HttpServerResponse& response)
|
|
{
|
|
ULONG bytesSent = 0;
|
|
HTTP_RESPONSE httpResponse;
|
|
HTTP_DATA_CHUNK httpResponseBody;
|
|
ZeroMemory(&httpResponse, sizeof(httpResponse));
|
|
ZeroMemory(&httpResponseBody, sizeof(httpResponseBody));
|
|
|
|
httpResponse.StatusCode = (USHORT)response.statusCode;
|
|
|
|
U8String reasonUtf8;
|
|
if (response.reason != WString::Empty)
|
|
{
|
|
reasonUtf8 = wtou8(response.reason);
|
|
httpResponse.pReason = (PCSTR)reasonUtf8.Buffer();
|
|
httpResponse.ReasonLength = (USHORT)reasonUtf8.Length();
|
|
}
|
|
|
|
U8String bodyUtf8;
|
|
if (response.body != WString::Empty)
|
|
{
|
|
bodyUtf8 = wtou8(response.body);
|
|
httpResponse.EntityChunkCount = 1;
|
|
httpResponse.pEntityChunks = &httpResponseBody;
|
|
httpResponseBody.DataChunkType = HttpDataChunkFromMemory;
|
|
httpResponseBody.FromMemory.pBuffer = (PVOID)bodyUtf8.Buffer();
|
|
httpResponseBody.FromMemory.BufferLength = (ULONG)bodyUtf8.Length();
|
|
}
|
|
|
|
U8String contentTypeUtf8;
|
|
if (response.contentType != WString::Empty)
|
|
{
|
|
contentTypeUtf8 = wtou8(response.contentType);
|
|
httpResponse.Headers.KnownHeaders[HttpHeaderContentType].pRawValue = (PCSTR)contentTypeUtf8.Buffer();
|
|
httpResponse.Headers.KnownHeaders[HttpHeaderContentType].RawValueLength = (USHORT)contentTypeUtf8.Length();
|
|
}
|
|
|
|
static const char headerACAOName[] = "Access-Control-Allow-Origin";
|
|
static HTTP_UNKNOWN_HEADER unknownHeaders[] = { {
|
|
sizeof(headerACAOName) - 1,
|
|
1,
|
|
headerACAOName,
|
|
"*"
|
|
} };
|
|
httpResponse.Headers.UnknownHeaderCount = sizeof(unknownHeaders) / sizeof(HTTP_UNKNOWN_HEADER);
|
|
httpResponse.Headers.pUnknownHeaders = unknownHeaders;
|
|
|
|
ULONG result = NO_ERROR;
|
|
result = HttpSendHttpResponse(
|
|
httpRequestQueue,
|
|
requestId,
|
|
0,
|
|
&httpResponse,
|
|
NULL,
|
|
&bytesSent,
|
|
NULL,
|
|
0,
|
|
NULL,
|
|
NULL);
|
|
return result;
|
|
}
|
|
|
|
void HttpServerApi::SendResponseUtf8(HANDLE httpRequestQueue, HTTP_REQUEST_ID requestId, WString body)
|
|
{
|
|
auto result = SendResponse(httpRequestQueue, requestId, { 200, WString::Unmanaged(L"OK"), body, L"application/json; charset=utf8" });
|
|
CHECK_ERROR(result == NO_ERROR, L"HttpSendHttpResponse failed for responding UTF-8 body.");
|
|
}
|
|
|
|
/***********************************************************************
|
|
HttpServerApi
|
|
***********************************************************************/
|
|
|
|
HttpServerApi::HttpServerApi(const WString& _urlPrefix, bool _respondToOptions)
|
|
: bufferRequest(HttpRequestBufferInitSize)
|
|
, urlPrefix(_urlPrefix)
|
|
, respondToOptions(_respondToOptions)
|
|
{
|
|
hEventRequest = CreateEvent(NULL, TRUE, TRUE, NULL);
|
|
CHECK_ERROR(hEventRequest != NULL, L"HttpServerApi initialization failed on CreateEvent(hEventRequest).");
|
|
CHECK_ERROR(eventPendingCallbacks.CreateManualUnsignal(true), L"HttpServerApi initialization failed on eventPendingCallbacks.CreateManualUnsignal.");
|
|
|
|
ULONG result = NO_ERROR;
|
|
|
|
result = HttpInitialize(
|
|
HTTPAPI_VERSION_2,
|
|
HTTP_INITIALIZE_SERVER,
|
|
NULL);
|
|
CHECK_ERROR(result == NO_ERROR, L"HttpInitialize failed.");
|
|
|
|
result = HttpCreateRequestQueue(
|
|
HTTPAPI_VERSION_2,
|
|
NULL,
|
|
NULL,
|
|
0,
|
|
&httpRequestQueue);
|
|
CHECK_ERROR(result == NO_ERROR, L"HttpCreateRequestQueue failed.");
|
|
|
|
result = HttpCreateServerSession(
|
|
HTTPAPI_VERSION_2,
|
|
&httpSessionId,
|
|
0);
|
|
CHECK_ERROR(result == NO_ERROR, L"HttpCreateServerSession failed.");
|
|
|
|
result = HttpCreateUrlGroup(
|
|
httpSessionId,
|
|
&httpUrlGroupId,
|
|
0);
|
|
CHECK_ERROR(result == NO_ERROR, L"HttpCreateUrlGroup failed.");
|
|
|
|
result = HttpAddUrlToUrlGroup(
|
|
httpUrlGroupId,
|
|
urlPrefix.Buffer(),
|
|
0,
|
|
0);
|
|
CHECK_ERROR(result == NO_ERROR, L"HttpAddUrlToUrlGroup failed.");
|
|
|
|
HTTP_BINDING_INFO bindingInfo;
|
|
ZeroMemory(&bindingInfo, sizeof(bindingInfo));
|
|
bindingInfo.Flags.Present = 1;
|
|
bindingInfo.RequestQueueHandle = httpRequestQueue;
|
|
|
|
result = HttpSetUrlGroupProperty(
|
|
httpUrlGroupId,
|
|
HttpServerBindingProperty,
|
|
&bindingInfo,
|
|
sizeof(bindingInfo));
|
|
CHECK_ERROR(result == NO_ERROR, L"HttpSetUrlGroupProperty failed (HttpServerBindingProperty).");
|
|
}
|
|
|
|
HttpServerApi::~HttpServerApi()
|
|
{
|
|
Stop();
|
|
CloseHandle(hEventRequest);
|
|
}
|
|
|
|
void HttpServerApi::Start()
|
|
{
|
|
CHECK_ERROR(state == State::Ready, L"HttpServerApi can only be started once.");
|
|
state = State::Running;
|
|
ListenToHttpRequest();
|
|
}
|
|
|
|
void HttpServerApi::Stop()
|
|
{
|
|
state = State::Stopping;
|
|
auto waitHandle = std::atomic_ref<HANDLE>(hWaitHandleRequest).exchange(INVALID_HANDLE_VALUE);
|
|
if (waitHandle != INVALID_HANDLE_VALUE)
|
|
{
|
|
UnregisterWaitEx(waitHandle, INVALID_HANDLE_VALUE);
|
|
}
|
|
eventPendingCallbacks.Wait();
|
|
|
|
OnHttpServerStopping();
|
|
|
|
if (httpRequestQueue != INVALID_HANDLE_VALUE)
|
|
{
|
|
HttpCloseUrlGroup(httpUrlGroupId);
|
|
HttpCloseServerSession(httpSessionId);
|
|
HttpCloseRequestQueue(httpRequestQueue);
|
|
httpRequestQueue = INVALID_HANDLE_VALUE;
|
|
httpUrlGroupId = HTTP_NULL_ID;
|
|
httpSessionId = HTTP_NULL_ID;
|
|
|
|
HttpTerminate(
|
|
HTTP_INITIALIZE_SERVER,
|
|
NULL);
|
|
}
|
|
}
|
|
|
|
bool HttpServerApi::IsStopped()
|
|
{
|
|
return state == State::Stopping;
|
|
}
|
|
|
|
}
|
|
|
|
|
|
/***********************************************************************
|
|
.\INTERPROCESS\WINDOWS\NAMEDPIPE.WINDOWS.CPP
|
|
***********************************************************************/
|
|
|
|
namespace vl::inter_process
|
|
{
|
|
|
|
using namespace vl::console;
|
|
using namespace vl::collections;
|
|
|
|
class NamedPipeConnection::ReadWaitContext
|
|
{
|
|
public:
|
|
NamedPipeConnection* connection = nullptr;
|
|
HANDLE hWaitHandle = INVALID_HANDLE_VALUE;
|
|
EventObject eventRegistrationFinished;
|
|
atomic_vint callbackStarted = 0;
|
|
|
|
ReadWaitContext()
|
|
{
|
|
CHECK_ERROR(eventRegistrationFinished.CreateManualUnsignal(false), L"ReadFile failed on eventRegistrationFinished.CreateManualUnsignal.");
|
|
}
|
|
};
|
|
|
|
class NamedPipeServer::PendingConnection::ConnectWaitContext
|
|
{
|
|
public:
|
|
Ptr<PendingConnection> pendingConnection;
|
|
HANDLE hWaitHandle = INVALID_HANDLE_VALUE;
|
|
EventObject eventRegistrationFinished;
|
|
atomic_vint callbackStarted = 0;
|
|
|
|
ConnectWaitContext(Ptr<PendingConnection> _pendingConnection)
|
|
: pendingConnection(_pendingConnection)
|
|
{
|
|
CHECK_ERROR(eventRegistrationFinished.CreateManualUnsignal(false), L"ConnectNamedPipe failed on eventRegistrationFinished.CreateManualUnsignal.");
|
|
}
|
|
};
|
|
|
|
/***********************************************************************
|
|
NamedPipeConnection (Reading)
|
|
***********************************************************************/
|
|
|
|
void NamedPipeConnection::BeginReadingUnsafe()
|
|
{
|
|
if (firstRead)
|
|
{
|
|
streamReadFile.SeekFromBegin(0);
|
|
}
|
|
}
|
|
|
|
void NamedPipeConnection::SubmitReadBufferUnsafe(vint bytes)
|
|
{
|
|
streamReadFile.Write(&bufferReadFile[0], bytes);
|
|
}
|
|
|
|
void NamedPipeConnection::EndReadingUnsafe()
|
|
{
|
|
vint32_t position = (vint32_t)streamReadFile.Position();
|
|
streamReadFile.SeekFromBegin(0);
|
|
CHECK_ERROR(position >= 2 * sizeof(vint32_t), L"ReadFile failed on incomplete message.");
|
|
|
|
vint32_t bytes = 0;
|
|
vint consumed = 0;
|
|
consumed = streamReadFile.Read(&bytes, sizeof(bytes));
|
|
CHECK_ERROR(consumed == sizeof(bytes), L"ReadFile failed on incomplete message.");
|
|
|
|
if (bytes > position - (vint32_t)sizeof(bytes))
|
|
{
|
|
firstRead = false;
|
|
streamReadFile.SeekFromBegin(position);
|
|
return;
|
|
}
|
|
CHECK_ERROR(bytes == position - sizeof(bytes), L"ReadFile failed on corrupted message.");
|
|
firstRead = true;
|
|
|
|
Array<wchar_t> strBuffer;
|
|
auto ReadSingleString = [&]()
|
|
{
|
|
vint32_t length = 0;
|
|
consumed = streamReadFile.Read(&length, sizeof(length));
|
|
CHECK_ERROR(consumed == sizeof(length) && streamReadFile.Position() <= position, L"ReadFile failed on incomplete message.");
|
|
|
|
if (length == 0)
|
|
{
|
|
return WString::Empty;
|
|
}
|
|
else
|
|
{
|
|
strBuffer.Resize(length);
|
|
consumed = streamReadFile.Read(&strBuffer[0], length * sizeof(wchar_t));
|
|
CHECK_ERROR(consumed == length * sizeof(wchar_t) && streamReadFile.Position() <= position, L"ReadFile failed on incomplete message.");
|
|
return WString::CopyFrom(&strBuffer[0], length);
|
|
}
|
|
};
|
|
|
|
callback->OnReadString(ReadSingleString());
|
|
|
|
CHECK_ERROR(streamReadFile.Position() == position, L"ReadFile failed on incomplete message.");
|
|
}
|
|
|
|
void NamedPipeConnection::BeginReadingLoopUnsafe()
|
|
{
|
|
if (stopped) return;
|
|
RESTART_LOOP:
|
|
{
|
|
BeginReadingUnsafe();
|
|
ResetEvent(hEventReadFile);
|
|
ZeroMemory(&overlappedReadFile, sizeof(overlappedReadFile));
|
|
overlappedReadFile.hEvent = hEventReadFile;
|
|
DWORD read = 0;
|
|
BOOL result = ReadFile(hPipe, &bufferReadFile[0], sizeof(BYTE) * MaxMessageSize, &read, &overlappedReadFile);
|
|
|
|
if (result == TRUE)
|
|
{
|
|
SubmitReadBufferUnsafe((vint)read);
|
|
EndReadingUnsafe();
|
|
goto RESTART_LOOP;
|
|
}
|
|
|
|
DWORD error = GetLastError();
|
|
if (error == ERROR_BROKEN_PIPE || error == ERROR_INVALID_HANDLE)
|
|
{
|
|
if (!stopped)
|
|
{
|
|
OnLocalError(L"ReadFile failed because the named pipe was closed.");
|
|
OnDisconnected();
|
|
}
|
|
return;
|
|
}
|
|
CHECK_ERROR(error == ERROR_MORE_DATA || error == ERROR_IO_PENDING, L"ReadFile failed on unexpected GetLastError.");
|
|
|
|
auto context = new ReadWaitContext;
|
|
context->connection = this;
|
|
BeginPendingCallback();
|
|
|
|
ReadWaitContext* expectedContext = nullptr;
|
|
if (stopped || !readWaitContext.compare_exchange_strong(expectedContext, context))
|
|
{
|
|
EndPendingCallback();
|
|
delete context;
|
|
return;
|
|
}
|
|
|
|
BOOL waitResult = RegisterWaitForSingleObject(
|
|
&context->hWaitHandle,
|
|
hEventReadFile,
|
|
[](PVOID lpParameter, BOOLEAN TimerOrWaitFired)
|
|
{
|
|
auto context = (ReadWaitContext*)lpParameter;
|
|
context->callbackStarted = 1;
|
|
|
|
auto self = context->connection;
|
|
ReadWaitContext* expectedContext = context;
|
|
bool ownsContext = self->readWaitContext.compare_exchange_strong(expectedContext, nullptr);
|
|
|
|
auto finalize = [=]()
|
|
{
|
|
if (ownsContext)
|
|
{
|
|
context->eventRegistrationFinished.Wait();
|
|
UnregisterWait(context->hWaitHandle);
|
|
}
|
|
self->EndPendingCallback();
|
|
if (ownsContext)
|
|
{
|
|
delete context;
|
|
}
|
|
};
|
|
|
|
DWORD read = 0;
|
|
BOOL result = GetOverlappedResult(self->hPipe, &self->overlappedReadFile, &read, FALSE);
|
|
if (result == TRUE)
|
|
{
|
|
self->SubmitReadBufferUnsafe((vint)read);
|
|
self->EndReadingUnsafe();
|
|
}
|
|
else
|
|
{
|
|
DWORD error = GetLastError();
|
|
if (error == ERROR_OPERATION_ABORTED || error == ERROR_INVALID_HANDLE || error == ERROR_BROKEN_PIPE || error == ERROR_NO_DATA)
|
|
{
|
|
if (!self->stopped)
|
|
{
|
|
self->OnLocalError(L"GetOverlappedResult(ReadFile) failed because the named pipe was closed.");
|
|
self->OnDisconnected();
|
|
}
|
|
finalize();
|
|
return;
|
|
}
|
|
CHECK_ERROR(error == ERROR_MORE_DATA, L"GetOverlappedResult(ReadFile) failed on unexpected GetLastError.");
|
|
self->SubmitReadBufferUnsafe((vint)read);
|
|
}
|
|
if (!self->stopped)
|
|
{
|
|
self->BeginReadingLoopUnsafe();
|
|
}
|
|
finalize();
|
|
},
|
|
context,
|
|
INFINITE,
|
|
WT_EXECUTEONLYONCE);
|
|
|
|
context->eventRegistrationFinished.Signal();
|
|
if (!waitResult)
|
|
{
|
|
expectedContext = context;
|
|
if (readWaitContext.compare_exchange_strong(expectedContext, nullptr))
|
|
{
|
|
EndPendingCallback();
|
|
delete context;
|
|
CHECK_FAIL(L"RegisterWaitForSingleObject failed for ReadFile.");
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
/***********************************************************************
|
|
NamedPipeConnection (Writing)
|
|
***********************************************************************/
|
|
|
|
vint32_t NamedPipeConnection::WriteInt32ToStream(vint32_t number)
|
|
{
|
|
return (vint32_t)streamWriteFile.Write(&number, sizeof(number));
|
|
}
|
|
|
|
vint32_t NamedPipeConnection::WriteStringToStream(const WString& str)
|
|
{
|
|
vint32_t bytes = 0;
|
|
vint32_t count = (vint32_t)str.Length();
|
|
bytes += (vint32_t)streamWriteFile.Write(&count, sizeof(count));
|
|
if (count > 0)
|
|
{
|
|
bytes += (vint32_t)streamWriteFile.Write((void*)str.Buffer(), sizeof(wchar_t) * str.Length());
|
|
}
|
|
return bytes;
|
|
}
|
|
|
|
void NamedPipeConnection::BeginSendStream()
|
|
{
|
|
vint32_t bytes = 0;
|
|
streamWriteFile.SeekFromBegin(0);
|
|
streamWriteFile.Write(&bytes, sizeof(bytes));
|
|
}
|
|
|
|
void NamedPipeConnection::EndSendStream(vint32_t bytes)
|
|
{
|
|
streamWriteFile.SeekFromBegin(0);
|
|
WriteInt32ToStream(bytes);
|
|
|
|
vint32_t length = bytes + sizeof(bytes);
|
|
for (vint i = 0; i < (length + MaxMessageSize - 1) / MaxMessageSize; i++)
|
|
{
|
|
vint offset = i * MaxMessageSize;
|
|
vint bytesToSend = length >= (i + 1) * MaxMessageSize ? MaxMessageSize : length % MaxMessageSize;
|
|
auto buffer = (LPCVOID)((char*)streamWriteFile.GetInternalBuffer() + offset);
|
|
|
|
ResetEvent(hEventWriteFile);
|
|
ZeroMemory(&overlappedWriteFile, sizeof(overlappedWriteFile));
|
|
overlappedWriteFile.hEvent = hEventWriteFile;
|
|
DWORD written = 0;
|
|
BOOL result = WriteFile(hPipe, buffer, (DWORD)bytesToSend, NULL, &overlappedWriteFile);
|
|
if (result == FALSE)
|
|
{
|
|
auto error = GetLastError();
|
|
if (error == ERROR_BROKEN_PIPE || error == ERROR_INVALID_HANDLE)
|
|
{
|
|
if (!stopped)
|
|
{
|
|
OnLocalError(L"WriteFile failed because the named pipe was closed.");
|
|
}
|
|
OnDisconnected();
|
|
return;
|
|
}
|
|
CHECK_ERROR(error == ERROR_IO_PENDING, L"WriteFile failed on unexpected GetLastError.");
|
|
WaitForSingleObject(hEventWriteFile, INFINITE);
|
|
result = GetOverlappedResult(hPipe, &overlappedWriteFile, &written, FALSE);
|
|
if (result == FALSE)
|
|
{
|
|
error = GetLastError();
|
|
if (error == ERROR_BROKEN_PIPE || error == ERROR_INVALID_HANDLE || error == ERROR_OPERATION_ABORTED)
|
|
{
|
|
if (!stopped)
|
|
{
|
|
OnLocalError(L"GetOverlappedResult(WriteFile) failed because the named pipe was closed.");
|
|
}
|
|
OnDisconnected();
|
|
return;
|
|
}
|
|
CHECK_FAIL(L"GetOverlappedResult(WriteFile) failed on unexpected GetLastError.");
|
|
}
|
|
}
|
|
else
|
|
{
|
|
written = (DWORD)bytesToSend;
|
|
}
|
|
CHECK_ERROR(written == (DWORD)bytesToSend, L"WriteFile failed to write all data.");
|
|
}
|
|
}
|
|
|
|
void NamedPipeConnection::SendString(const WString& str)
|
|
{
|
|
if (stopped) return;
|
|
SPIN_LOCK(lockWrite)
|
|
{
|
|
vint32_t bytes = 0;
|
|
BeginSendStream();
|
|
bytes += WriteStringToStream(str);
|
|
EndSendStream(bytes);
|
|
}
|
|
}
|
|
|
|
/***********************************************************************
|
|
NamedPipeConnection
|
|
***********************************************************************/
|
|
|
|
void NamedPipeConnection::OnLocalError(const WString& errorMessage)
|
|
{
|
|
if (callback)
|
|
{
|
|
callback->OnLocalError(errorMessage, true);
|
|
}
|
|
}
|
|
|
|
void NamedPipeConnection::OnDisconnected()
|
|
{
|
|
if (callback)
|
|
{
|
|
callback->OnDisconnected();
|
|
}
|
|
auto owningServer = server;
|
|
if (owningServer && pendingCallbacks == 0)
|
|
{
|
|
SPIN_LOCK(owningServer->lockConnections)
|
|
{
|
|
if (server == owningServer)
|
|
{
|
|
owningServer->connections.Remove(this);
|
|
server = nullptr;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
NamedPipeConnection::NamedPipeConnection(HANDLE _hPipe)
|
|
: bufferReadFile(MaxMessageSize)
|
|
{
|
|
hPipe = _hPipe;
|
|
|
|
hEventReadFile = CreateEvent(NULL, TRUE, TRUE, NULL);
|
|
CHECK_ERROR(hEventReadFile != NULL, L"NamedPipeConnection initialization failed on CreateEvent(hEventReadFile).");
|
|
|
|
hEventWriteFile = CreateEvent(NULL, TRUE, TRUE, NULL);
|
|
CHECK_ERROR(hEventWriteFile != NULL, L"NamedPipeConnection initialization failed on CreateEvent(hEventWriteFile).");
|
|
|
|
CHECK_ERROR(eventPendingCallbacks.CreateManualUnsignal(true), L"NamedPipeConnection initialization failed on eventPendingCallbacks.CreateManualUnsignal.");
|
|
}
|
|
|
|
NamedPipeConnection::~NamedPipeConnection()
|
|
{
|
|
Stop();
|
|
CloseHandle(hEventReadFile);
|
|
CloseHandle(hEventWriteFile);
|
|
}
|
|
|
|
void NamedPipeConnection::InstallCallback(INetworkProtocolCallback* _callback)
|
|
{
|
|
callback = _callback;
|
|
CHECK_ERROR(callback, L"NamedPipeConnection::InstallCallback needs a valid INetworkProtocolCallback.");
|
|
callback->OnInstalled(this);
|
|
}
|
|
|
|
void NamedPipeConnection::Stop()
|
|
{
|
|
stopped = 1;
|
|
ReadWaitContext* context = readWaitContext.exchange(nullptr);
|
|
if (context)
|
|
{
|
|
context->eventRegistrationFinished.Wait();
|
|
if (context->hWaitHandle != INVALID_HANDLE_VALUE)
|
|
{
|
|
UnregisterWaitEx(context->hWaitHandle, INVALID_HANDLE_VALUE);
|
|
}
|
|
if (context->callbackStarted == 0)
|
|
{
|
|
EndPendingCallback();
|
|
}
|
|
delete context;
|
|
}
|
|
eventPendingCallbacks.Wait();
|
|
|
|
SPIN_LOCK(lockWrite)
|
|
{
|
|
if (hPipe != INVALID_HANDLE_VALUE)
|
|
{
|
|
CancelIoEx(hPipe, NULL);
|
|
CloseHandle(hPipe);
|
|
hPipe = INVALID_HANDLE_VALUE;
|
|
}
|
|
}
|
|
}
|
|
|
|
void NamedPipeConnection::BeginPendingCallback()
|
|
{
|
|
if (pendingCallbacks++ == 0)
|
|
{
|
|
eventPendingCallbacks.Unsignal();
|
|
}
|
|
}
|
|
|
|
void NamedPipeConnection::EndPendingCallback()
|
|
{
|
|
if (--pendingCallbacks == 0)
|
|
{
|
|
eventPendingCallbacks.Signal();
|
|
}
|
|
}
|
|
|
|
/***********************************************************************
|
|
NamedPipeServer::PendingConnection
|
|
***********************************************************************/
|
|
|
|
NamedPipeServer::PendingConnection::PendingConnection(NamedPipeServer* _server, Ptr<NamedPipeConnection> _connection)
|
|
: server(_server)
|
|
, connection(_connection)
|
|
{
|
|
ZeroMemory(&overlappedConnect, sizeof(overlappedConnect));
|
|
hEventConnect = CreateEvent(NULL, TRUE, FALSE, NULL);
|
|
CHECK_ERROR(hEventConnect != NULL, L"ConnectNamedPipe failed on CreateEvent.");
|
|
overlappedConnect.hEvent = hEventConnect;
|
|
CHECK_ERROR(eventPendingCallbacks.CreateManualUnsignal(true), L"ConnectNamedPipe failed on eventPendingCallbacks.CreateManualUnsignal.");
|
|
}
|
|
|
|
NamedPipeServer::PendingConnection::~PendingConnection()
|
|
{
|
|
Stop();
|
|
CloseHandle(hEventConnect);
|
|
}
|
|
|
|
void NamedPipeServer::PendingConnection::Stop()
|
|
{
|
|
server = nullptr;
|
|
ConnectWaitContext* context = connectWaitContext.exchange(nullptr);
|
|
if (context)
|
|
{
|
|
context->eventRegistrationFinished.Wait();
|
|
if (context->hWaitHandle != INVALID_HANDLE_VALUE)
|
|
{
|
|
UnregisterWaitEx(context->hWaitHandle, INVALID_HANDLE_VALUE);
|
|
}
|
|
if (context->callbackStarted == 0)
|
|
{
|
|
EndPendingCallback();
|
|
}
|
|
delete context;
|
|
}
|
|
eventPendingCallbacks.Wait();
|
|
if (connection)
|
|
{
|
|
connection->Stop();
|
|
connection = nullptr;
|
|
}
|
|
}
|
|
|
|
void NamedPipeServer::PendingConnection::BeginPendingCallback()
|
|
{
|
|
if (pendingCallbacks++ == 0)
|
|
{
|
|
eventPendingCallbacks.Unsignal();
|
|
}
|
|
}
|
|
|
|
void NamedPipeServer::PendingConnection::EndPendingCallback()
|
|
{
|
|
if (--pendingCallbacks == 0)
|
|
{
|
|
eventPendingCallbacks.Signal();
|
|
}
|
|
}
|
|
|
|
/***********************************************************************
|
|
NamedPipeServer
|
|
***********************************************************************/
|
|
|
|
HANDLE NamedPipeServer::ServerCreatePipe(const WString& pipeName)
|
|
{
|
|
HANDLE hPipe = CreateNamedPipe(
|
|
(L"\\\\.\\pipe\\" + pipeName).Buffer(),
|
|
PIPE_ACCESS_DUPLEX | FILE_FLAG_OVERLAPPED,
|
|
PIPE_TYPE_MESSAGE | PIPE_READMODE_MESSAGE | PIPE_REJECT_REMOTE_CLIENTS,
|
|
PIPE_UNLIMITED_INSTANCES,
|
|
65536,
|
|
65536,
|
|
6000,
|
|
NULL);
|
|
CHECK_ERROR(hPipe != INVALID_HANDLE_VALUE, L"CreateNamedPipe failed.");
|
|
return hPipe;
|
|
}
|
|
|
|
NamedPipeServer::NamedPipeServer(const WString& _pipeName)
|
|
: pipeName(_pipeName)
|
|
{
|
|
}
|
|
|
|
NamedPipeServer::~NamedPipeServer()
|
|
{
|
|
Stop();
|
|
}
|
|
|
|
void NamedPipeServer::BeginListening()
|
|
{
|
|
Ptr<PendingConnection> pendingConnection;
|
|
SPIN_LOCK(lockConnections)
|
|
{
|
|
if (!started || stopped)
|
|
{
|
|
return;
|
|
}
|
|
}
|
|
|
|
auto connection = Ptr(new NamedPipeConnection(ServerCreatePipe(pipeName)));
|
|
pendingConnection = Ptr(new PendingConnection(this, connection));
|
|
|
|
BOOL result = ConnectNamedPipe(connection->hPipe, &pendingConnection->overlappedConnect);
|
|
DWORD error = result ? ERROR_SUCCESS : GetLastError();
|
|
switch (error)
|
|
{
|
|
case ERROR_SUCCESS:
|
|
case ERROR_PIPE_CONNECTED:
|
|
CompletePendingConnection(pendingConnection, true);
|
|
break;
|
|
case ERROR_IO_PENDING:
|
|
SPIN_LOCK(lockConnections)
|
|
{
|
|
if (stopped)
|
|
{
|
|
pendingConnection->Stop();
|
|
return;
|
|
}
|
|
pendingConnections.Add(pendingConnection);
|
|
auto context = new PendingConnection::ConnectWaitContext(pendingConnection);
|
|
pendingConnection->BeginPendingCallback();
|
|
PendingConnection::ConnectWaitContext* expectedContext = nullptr;
|
|
CHECK_ERROR(pendingConnection->connectWaitContext.compare_exchange_strong(expectedContext, context), L"ConnectNamedPipe found an existing wait context.");
|
|
BOOL waitResult = RegisterWaitForSingleObject(
|
|
&context->hWaitHandle,
|
|
pendingConnection->hEventConnect,
|
|
[](PVOID lpParameter, BOOLEAN TimerOrWaitFired)
|
|
{
|
|
auto context = (PendingConnection::ConnectWaitContext*)lpParameter;
|
|
context->callbackStarted = 1;
|
|
|
|
auto pendingConnection = context->pendingConnection;
|
|
PendingConnection::ConnectWaitContext* expectedContext = context;
|
|
bool ownsContext = pendingConnection->connectWaitContext.compare_exchange_strong(expectedContext, nullptr);
|
|
|
|
auto finalize = [=]()
|
|
{
|
|
if (ownsContext)
|
|
{
|
|
context->eventRegistrationFinished.Wait();
|
|
UnregisterWait(context->hWaitHandle);
|
|
}
|
|
pendingConnection->EndPendingCallback();
|
|
if (ownsContext)
|
|
{
|
|
delete context;
|
|
}
|
|
};
|
|
|
|
DWORD transferred = 0;
|
|
BOOL result = GetOverlappedResult(pendingConnection->connection->hPipe, &pendingConnection->overlappedConnect, &transferred, FALSE);
|
|
if (result == TRUE)
|
|
{
|
|
if (pendingConnection->server)
|
|
{
|
|
pendingConnection->server->CompletePendingConnection(pendingConnection.Obj(), true);
|
|
}
|
|
}
|
|
else
|
|
{
|
|
auto error = GetLastError();
|
|
if (error == ERROR_OPERATION_ABORTED || error == ERROR_INVALID_HANDLE || error == ERROR_BROKEN_PIPE || error == ERROR_NO_DATA)
|
|
{
|
|
if (pendingConnection->server)
|
|
{
|
|
pendingConnection->server->CompletePendingConnection(pendingConnection.Obj(), false);
|
|
}
|
|
finalize();
|
|
return;
|
|
}
|
|
CHECK_FAIL(L"GetOverlappedResult(ConnectNamedPipe) failed on unexpected GetLastError.");
|
|
}
|
|
finalize();
|
|
},
|
|
context,
|
|
INFINITE,
|
|
WT_EXECUTEONLYONCE);
|
|
|
|
context->eventRegistrationFinished.Signal();
|
|
if (!waitResult)
|
|
{
|
|
expectedContext = context;
|
|
if (pendingConnection->connectWaitContext.compare_exchange_strong(expectedContext, nullptr))
|
|
{
|
|
pendingConnection->EndPendingCallback();
|
|
delete context;
|
|
CHECK_FAIL(L"RegisterWaitForSingleObject failed for ConnectNamedPipe.");
|
|
}
|
|
}
|
|
}
|
|
break;
|
|
default:
|
|
CHECK_FAIL(L"ConnectNamedPipe failed on unexpected GetLastError.");
|
|
}
|
|
}
|
|
|
|
void NamedPipeServer::CompletePendingConnection(Ptr<PendingConnection> pendingConnection, bool connected)
|
|
{
|
|
auto connection = pendingConnection->connection;
|
|
bool shouldListen = false;
|
|
bool shouldNotify = false;
|
|
|
|
{
|
|
SPIN_LOCK(lockConnections)
|
|
{
|
|
pendingConnections.Remove(pendingConnection.Obj());
|
|
if (started && !stopped)
|
|
{
|
|
shouldListen = true;
|
|
if (connected)
|
|
{
|
|
connection->server = this;
|
|
connections.Add(connection);
|
|
shouldNotify = true;
|
|
pendingConnection->connection = nullptr;
|
|
pendingConnection->server = nullptr;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
if (shouldListen)
|
|
{
|
|
BeginListening();
|
|
}
|
|
|
|
if (shouldNotify)
|
|
{
|
|
auto result = OnClientConnected(connection.Obj());
|
|
if (result == WaitForClientResult::Reject)
|
|
{
|
|
SPIN_LOCK(lockConnections)
|
|
{
|
|
connections.Remove(connection.Obj());
|
|
}
|
|
connection->server = nullptr;
|
|
connection->Stop();
|
|
}
|
|
}
|
|
else if (connection)
|
|
{
|
|
connection->Stop();
|
|
}
|
|
}
|
|
|
|
void NamedPipeServer::CompletePendingConnection(PendingConnection* pendingConnection, bool connected)
|
|
{
|
|
Ptr<PendingConnection> holding;
|
|
{
|
|
SPIN_LOCK(lockConnections)
|
|
{
|
|
for (vint i = 0; i < pendingConnections.Count(); i++)
|
|
{
|
|
if (pendingConnections[i].Obj() == pendingConnection)
|
|
{
|
|
holding = pendingConnections[i];
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
if (holding)
|
|
{
|
|
CompletePendingConnection(holding, connected);
|
|
}
|
|
}
|
|
|
|
WaitForClientResult NamedPipeServer::OnClientConnected(INetworkProtocolConnection* connection)
|
|
{
|
|
return WaitForClientResult::Accept;
|
|
}
|
|
|
|
void NamedPipeServer::Start()
|
|
{
|
|
{
|
|
SPIN_LOCK(lockConnections)
|
|
{
|
|
CHECK_ERROR(!started, L"NamedPipeServer has already started.");
|
|
CHECK_ERROR(!stopped, L"NamedPipeServer has stopped.");
|
|
started = true;
|
|
}
|
|
}
|
|
BeginListening();
|
|
}
|
|
|
|
void NamedPipeServer::Stop()
|
|
{
|
|
List<Ptr<NamedPipeConnection>> stoppingConnections;
|
|
List<Ptr<PendingConnection>> stoppingPendingConnections;
|
|
SPIN_LOCK(lockConnections)
|
|
{
|
|
if (!stopped)
|
|
{
|
|
started = false;
|
|
stopped = true;
|
|
for (auto connection : connections)
|
|
{
|
|
connection->server = nullptr;
|
|
stoppingConnections.Add(connection);
|
|
}
|
|
for (auto pendingConnection : pendingConnections)
|
|
{
|
|
stoppingPendingConnections.Add(pendingConnection);
|
|
}
|
|
connections.Clear();
|
|
pendingConnections.Clear();
|
|
}
|
|
}
|
|
|
|
for (auto pendingConnection : stoppingPendingConnections)
|
|
{
|
|
pendingConnection->Stop();
|
|
}
|
|
for (auto connection : stoppingConnections)
|
|
{
|
|
connection->Stop();
|
|
}
|
|
}
|
|
|
|
bool NamedPipeServer::IsStopped()
|
|
{
|
|
bool result = false;
|
|
SPIN_LOCK(lockConnections)
|
|
{
|
|
result = stopped;
|
|
}
|
|
return result;
|
|
}
|
|
|
|
/***********************************************************************
|
|
NamedPipeClient
|
|
***********************************************************************/
|
|
|
|
HANDLE NamedPipeClient::ClientCreatePipe(const WString& pipeName)
|
|
{
|
|
auto fullPipeName = L"\\\\.\\pipe\\" + pipeName;
|
|
auto start = GetTickCount64();
|
|
while (true)
|
|
{
|
|
HANDLE hPipe = CreateFile(
|
|
fullPipeName.Buffer(),
|
|
GENERIC_READ | GENERIC_WRITE,
|
|
0,
|
|
NULL,
|
|
OPEN_EXISTING,
|
|
FILE_FLAG_OVERLAPPED,
|
|
NULL);
|
|
if (hPipe != INVALID_HANDLE_VALUE)
|
|
{
|
|
return hPipe;
|
|
}
|
|
|
|
auto error = GetLastError();
|
|
CHECK_ERROR(error == ERROR_FILE_NOT_FOUND || error == ERROR_PIPE_BUSY, L"CreateFile failed.");
|
|
CHECK_ERROR(GetTickCount64() - start < 6000, L"CreateFile failed because the pipe server did not become available.");
|
|
if (error == ERROR_PIPE_BUSY)
|
|
{
|
|
WaitNamedPipe(fullPipeName.Buffer(), 100);
|
|
}
|
|
else
|
|
{
|
|
Thread::Sleep(1);
|
|
}
|
|
}
|
|
}
|
|
|
|
NamedPipeClient::NamedPipeClient(const WString& _pipeName)
|
|
: NamedPipeConnection(ClientCreatePipe(_pipeName))
|
|
{
|
|
}
|
|
|
|
NamedPipeClient::~NamedPipeClient()
|
|
{
|
|
}
|
|
|
|
INetworkProtocolConnection* NamedPipeClient::GetConnection()
|
|
{
|
|
return this;
|
|
}
|
|
|
|
void NamedPipeClient::WaitForServer()
|
|
{
|
|
status = ClientStatus::WaitingForServer;
|
|
DWORD dwPipeMode = PIPE_READMODE_MESSAGE;
|
|
BOOL bSucceeded = SetNamedPipeHandleState(
|
|
hPipe,
|
|
&dwPipeMode,
|
|
NULL,
|
|
NULL);
|
|
CHECK_ERROR(bSucceeded, L"SetNamedPipeHandleState failed.");
|
|
status = ClientStatus::Connected;
|
|
if (callback)
|
|
{
|
|
callback->OnConnected();
|
|
}
|
|
}
|
|
|
|
ClientStatus NamedPipeClient::GetStatus()
|
|
{
|
|
return status;
|
|
}
|
|
|
|
void NamedPipeClient::Stop()
|
|
{
|
|
NamedPipeConnection::Stop();
|
|
status = ClientStatus::Disconnected;
|
|
}
|
|
|
|
}
|
|
|
|
|
|
/***********************************************************************
|
|
.\INTERPROCESS\WINDOWS\NETWORKPROTOCOL.WINDOWS.CPP
|
|
***********************************************************************/
|
|
|
|
#pragma comment(lib, "httpapi.lib")
|
|
#pragma comment(lib, "rpcrt4.lib")
|