mirror of
https://github.com/HIllya51/LunaHook.git
synced 2024-11-27 07:44:02 +08:00
78 lines
2.6 KiB
C++
78 lines
2.6 KiB
C++
#include <winhttp.h>
|
|
using InternetHandle = AutoHandle<Functor<WinHttpCloseHandle>>;
|
|
|
|
struct HttpRequest
|
|
{
|
|
HttpRequest(
|
|
const wchar_t *agentName,
|
|
const wchar_t *serverName,
|
|
const wchar_t *action,
|
|
const wchar_t *objectName,
|
|
std::string body = "",
|
|
const wchar_t *headers = NULL,
|
|
DWORD port = INTERNET_DEFAULT_PORT,
|
|
const wchar_t *referrer = NULL,
|
|
DWORD requestFlags = WINHTTP_FLAG_SECURE | WINHTTP_FLAG_ESCAPE_DISABLE,
|
|
const wchar_t *httpVersion = NULL,
|
|
const wchar_t **acceptTypes = NULL);
|
|
operator bool() { return errorCode == ERROR_SUCCESS; }
|
|
|
|
std::wstring response;
|
|
std::wstring headers;
|
|
InternetHandle connection = NULL;
|
|
InternetHandle request = NULL;
|
|
DWORD errorCode = ERROR_SUCCESS;
|
|
};
|
|
|
|
HttpRequest::HttpRequest(
|
|
const wchar_t *agentName,
|
|
const wchar_t *serverName,
|
|
const wchar_t *action,
|
|
const wchar_t *objectName,
|
|
std::string body,
|
|
const wchar_t *headers,
|
|
DWORD port,
|
|
const wchar_t *referrer,
|
|
DWORD requestFlags,
|
|
const wchar_t *httpVersion,
|
|
const wchar_t **acceptTypes)
|
|
{
|
|
static std::atomic<HINTERNET> internet = NULL;
|
|
if (!internet)
|
|
internet = WinHttpOpen(agentName, WINHTTP_ACCESS_TYPE_DEFAULT_PROXY, NULL, NULL, 0);
|
|
if (internet)
|
|
if (InternetHandle connection = WinHttpConnect(internet, serverName, port, 0))
|
|
if (InternetHandle request = WinHttpOpenRequest(connection, action, objectName, httpVersion, referrer, acceptTypes, requestFlags))
|
|
if (WinHttpSendRequest(request, headers, -1UL, body.empty() ? NULL : body.data(), body.size(), body.size(), NULL))
|
|
{
|
|
WinHttpReceiveResponse(request, NULL);
|
|
|
|
// DWORD size = 0;
|
|
// WinHttpQueryHeaders(request, WINHTTP_QUERY_RAW_HEADERS_CRLF, WINHTTP_HEADER_NAME_BY_INDEX, NULL, &size, WINHTTP_NO_HEADER_INDEX);
|
|
// this->headers.resize(size);
|
|
// WinHttpQueryHeaders(request, WINHTTP_QUERY_RAW_HEADERS_CRLF, WINHTTP_HEADER_NAME_BY_INDEX, this->headers.data(), &size, WINHTTP_NO_HEADER_INDEX);
|
|
std::string data;
|
|
DWORD availableSize, downloadedSize;
|
|
do
|
|
{
|
|
availableSize = 0;
|
|
WinHttpQueryDataAvailable(request, &availableSize);
|
|
if (!availableSize)
|
|
break;
|
|
std::vector<char> buffer(availableSize);
|
|
WinHttpReadData(request, buffer.data(), availableSize, &downloadedSize);
|
|
data.append(buffer.data(), downloadedSize);
|
|
} while (availableSize > 0);
|
|
response = StringToWideString(data);
|
|
this->connection = std::move(connection);
|
|
this->request = std::move(request);
|
|
}
|
|
else
|
|
errorCode = GetLastError();
|
|
else
|
|
errorCode = GetLastError();
|
|
else
|
|
errorCode = GetLastError();
|
|
else
|
|
errorCode = GetLastError();
|
|
} |