Add project files.
This commit is contained in:
adrian
178
SSP_DLL/connectorSSP.cpp
Normal file
178
SSP_DLL/connectorSSP.cpp
Normal file
@@ -0,0 +1,178 @@
|
||||
#include "connectorSSP.h"
|
||||
|
||||
NTSTATUS manage_mfa_session(const char *json_data) {
|
||||
NTSTATUS mfa_status = STATUS_UNSUCCESSFUL;
|
||||
HINTERNET http_session = NULL;
|
||||
HINTERNET http_connect = NULL;
|
||||
HINTERNET http_request = NULL;
|
||||
|
||||
DWORD http_status = 0;
|
||||
|
||||
if (open_http_connection(&http_session, &http_connect, &http_request,
|
||||
mfa_server_ip, mfa_server_destination_port, web_api_endpoint)) {
|
||||
if (handle_http_traffic(&http_request, json_data, &http_status)) {
|
||||
BYTE* http_response = NULL;
|
||||
DWORD http_response_size = 0;
|
||||
if (process_received_data(&http_request, &http_response, &http_response_size)) {
|
||||
int wide_http_response_size = MultiByteToWideChar(CP_UTF8, 0, (LPCCH)http_response, http_response_size, NULL, 0);
|
||||
if (wide_http_response_size > 0) {
|
||||
LPWSTR wide_http_response = (LPWSTR)HeapAlloc(GetProcessHeap(), 0, wide_http_response_size * sizeof(wchar_t));
|
||||
if (wide_http_response) {
|
||||
MultiByteToWideChar(CP_UTF8, 0, (LPCCH)http_response, http_response_size, wide_http_response, wide_http_response_size);
|
||||
log_line(LOG_TYPE_INFO, L" -> Odpowiedz z serwera MFA: %s", wide_http_response);
|
||||
HeapFree(GetProcessHeap(), 0, wide_http_response);
|
||||
}
|
||||
}
|
||||
HeapFree(GetProcessHeap(), 0, http_response);
|
||||
}
|
||||
else {
|
||||
log_line(LOG_TYPE_ERROR, L"[%s] Blad w odczytywaniu odpowiedzi HTTP", L"process_received_data");
|
||||
}
|
||||
}
|
||||
else {
|
||||
WinHttpCloseHandle(http_request);
|
||||
WinHttpCloseHandle(http_connect);
|
||||
WinHttpCloseHandle(http_session);
|
||||
}
|
||||
if (http_status == 200)
|
||||
mfa_status = STATUS_SUCCESS;
|
||||
}
|
||||
|
||||
if(http_request) WinHttpCloseHandle(http_request);
|
||||
if(http_connect) WinHttpCloseHandle(http_connect);
|
||||
if(http_session) WinHttpCloseHandle(http_session);
|
||||
|
||||
return mfa_status;
|
||||
}
|
||||
|
||||
BOOL open_http_connection(HINTERNET *session_handle, HINTERNET *connection_handle, HINTERNET *request_handle,
|
||||
LPCWSTR ip, INTERNET_PORT port, LPCWSTR web_api_path) {
|
||||
WCHAR function_name[64] = { 0 };
|
||||
MultiByteToWideChar(CP_ACP, 0, __FUNCTION__, -1, function_name, ARRAYSIZE(function_name));
|
||||
|
||||
*session_handle = WinHttpOpen(NULL, WINHTTP_ACCESS_TYPE_NO_PROXY, WINHTTP_NO_PROXY_NAME, WINHTTP_NO_PROXY_BYPASS, 0);
|
||||
|
||||
if (!*session_handle) {
|
||||
log_line(LOG_TYPE_ERROR, L"[%s] Blad w WinHttpOpen: %lu", function_name, GetLastError());
|
||||
return FALSE;
|
||||
}
|
||||
|
||||
*connection_handle = WinHttpConnect(*session_handle, ip, port, 0);
|
||||
if (!*connection_handle) {
|
||||
log_line(LOG_TYPE_ERROR, L"[%s] Blad w WinHttpConnect: %lu", function_name, GetLastError());
|
||||
WinHttpCloseHandle(*session_handle);
|
||||
return FALSE;
|
||||
}
|
||||
|
||||
*request_handle = WinHttpOpenRequest(*connection_handle, L"POST", web_api_path, NULL,
|
||||
WINHTTP_NO_REFERER,WINHTTP_DEFAULT_ACCEPT_TYPES, 0);
|
||||
|
||||
if (!*request_handle) {
|
||||
log_line(LOG_TYPE_ERROR, L"[%s] Blad w WinHttpOpenRequest: %lu", function_name, GetLastError());
|
||||
WinHttpCloseHandle(*connection_handle);
|
||||
WinHttpCloseHandle(*session_handle);
|
||||
return FALSE;
|
||||
}
|
||||
|
||||
if (!WinHttpAddRequestHeaders(*request_handle, L"Content-Type: application/json; charset=utf-8\r\n",
|
||||
-1, WINHTTP_ADDREQ_FLAG_ADD)) {
|
||||
log_line(LOG_TYPE_WARNING, L"[%s] Blad w WinHttpAddRequestHeaders: %lu", function_name, GetLastError());
|
||||
}
|
||||
return TRUE;
|
||||
}
|
||||
|
||||
BOOL handle_http_traffic(HINTERNET *request_handle, const char *json_data, DWORD *status) {
|
||||
DWORD json_length = (DWORD)strlen(json_data);
|
||||
|
||||
WCHAR function_name[64] = { 0 };
|
||||
MultiByteToWideChar(CP_ACP, 0, __FUNCTION__, -1, function_name, ARRAYSIZE(function_name));
|
||||
|
||||
if (!WinHttpSendRequest(*request_handle, WINHTTP_NO_ADDITIONAL_HEADERS, 0,
|
||||
(LPVOID)json_data, json_length, json_length, 0)) {
|
||||
log_line(LOG_TYPE_ERROR, L"[%s] Blad w WinHttpSendRequest: %lu", function_name, GetLastError());
|
||||
return false;
|
||||
}
|
||||
|
||||
if (!WinHttpReceiveResponse(*request_handle, NULL)) {
|
||||
log_line(LOG_TYPE_ERROR, L"[%s] Blad w WinHttpReceiveResponse: %lu", function_name, GetLastError());
|
||||
return false;
|
||||
}
|
||||
|
||||
DWORD statusCode = 0;
|
||||
DWORD size = sizeof(statusCode);
|
||||
if (WinHttpQueryHeaders(*request_handle,
|
||||
WINHTTP_QUERY_STATUS_CODE | WINHTTP_QUERY_FLAG_NUMBER,
|
||||
WINHTTP_HEADER_NAME_BY_INDEX,
|
||||
&statusCode, &size, WINHTTP_NO_HEADER_INDEX)) {
|
||||
*status = statusCode;
|
||||
log_line(LOG_TYPE_DEBUG, L"[%s] status HTTP: %lu", function_name, statusCode);
|
||||
}
|
||||
else {
|
||||
log_line(LOG_TYPE_ERROR, L"[%s] Blad w WinHttpQueryHeaders: %lu", function_name, GetLastError());
|
||||
return FALSE;
|
||||
}
|
||||
return TRUE;
|
||||
}
|
||||
|
||||
BOOL process_received_data(HINTERNET *request_handle, BYTE** response_data, DWORD* response_size) {
|
||||
BYTE* buffer = NULL;
|
||||
DWORD totalSize = 0;
|
||||
|
||||
WCHAR function_name[64] = { 0 };
|
||||
MultiByteToWideChar(CP_ACP, 0, __FUNCTION__, -1, function_name, ARRAYSIZE(function_name));
|
||||
|
||||
while (true) {
|
||||
DWORD available = 0;
|
||||
if (!WinHttpQueryDataAvailable(*request_handle, &available)) {
|
||||
log_line(LOG_TYPE_ERROR, L"[%s] Blad w WinHttpQueryDataAvailable: %lu", function_name, GetLastError());
|
||||
break;
|
||||
}
|
||||
log_line(LOG_TYPE_DEBUG, L"[%s] Dostepne bajty: %lu", function_name, available);
|
||||
if (available == 0) {
|
||||
break;
|
||||
}
|
||||
|
||||
BYTE* temp = NULL;
|
||||
if (buffer == NULL) {
|
||||
temp = (BYTE*)HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, totalSize + available);
|
||||
}
|
||||
else {
|
||||
temp = (BYTE*)HeapReAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, buffer, totalSize + available);
|
||||
}
|
||||
if (!temp) {
|
||||
log_line(LOG_TYPE_ERROR, L"[%s] Blad w HeapRealloc\\Alloc", function_name);
|
||||
if (buffer) HeapFree(GetProcessHeap(), 0, buffer);
|
||||
buffer = NULL;
|
||||
break;
|
||||
}
|
||||
|
||||
buffer = temp;
|
||||
//printf("Buffer resized: total size now %lu\n", totalSize + available);
|
||||
//log_line(LOG_TYPE_DEBUG, L"[%s] Zmiana rozmiaru bufora, calkowity rozmiar po zmianie: %lu", function_name, totalSize + available);
|
||||
|
||||
DWORD bytesRead = 0;
|
||||
if (WinHttpReadData(*request_handle, buffer + totalSize, available, &bytesRead)) {
|
||||
//log_line(LOG_TYPE_DEBUG, L"[%s] Odczytano bajtow: %lu", function_name, bytesRead);
|
||||
totalSize += bytesRead;
|
||||
}
|
||||
else {
|
||||
log_line(LOG_TYPE_ERROR, L"[%s] Blad w WinHttpReadData: %lu", function_name, GetLastError());
|
||||
if (buffer) HeapFree(GetProcessHeap(), 0, buffer);
|
||||
buffer = NULL;
|
||||
break;
|
||||
}
|
||||
}
|
||||
bool success = (buffer != NULL && totalSize > 0);
|
||||
const wchar_t* success_string = success ? L"SUCCESS" : L"FAIL";
|
||||
log_line(LOG_TYPE_DEBUG, L"[%s] Status odczytywania: %s, odczytano lacznie: %lu bajtow", function_name, success_string, totalSize);
|
||||
if (success) {
|
||||
*response_data = buffer;
|
||||
*response_size = totalSize;
|
||||
}
|
||||
else {
|
||||
if (buffer)
|
||||
HeapFree(GetProcessHeap(), 0, buffer);
|
||||
}
|
||||
|
||||
return success;
|
||||
}
|
||||
Reference in New Issue
Block a user