Drop curl in favor of WinHTTP

Abstract out HTTP comunication and replace libcurl with WinHTTP.

1 files changed, 481 insertions, 0 deletions

diff --git a/src/http.c b/src/http.c
new file mode 100644
index 0000000..2c86b44
--- /dev/null
+++ b/src/http.c
@@ -0,0 +1,481 @@
+/*

+Copyright (c) 2015

+	Michał Cichoń <thedmd@interia.pl>

+

+Permission is hereby granted, free of charge, to any person obtaining a copy

+of this software and associated documentation files (the "Software"), to deal

+in the Software without restriction, including without limitation the rights

+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell

+copies of the Software, and to permit persons to whom the Software is

+furnished to do so, subject to the following conditions:

+

+The above copyright notice and this permission notice shall be included in

+all copies or substantial portions of the Software.

+

+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR

+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,

+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE

+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER

+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,

+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN

+THE SOFTWARE.

+*/

+

+#include "config.h"

+#include "http.h"

+#include <Windows.h>

+#include <winhttp.h>

+#pragma comment(lib, "winhttp.lib")

+

+struct _http_t {

+	HINTERNET		session;

+	HINTERNET		connection;

+	wchar_t*		endpoint;

+	wchar_t*		securePort;

+	wchar_t*		autoProxy;

+	wchar_t*		proxy;

+	wchar_t*		proxyUsername;

+	wchar_t*		proxyPassword;

+	char*			error;

+};

+

+static char* HttpToString(const wchar_t* wideString, int size);

+static wchar_t* HttpToWideString(const char* string, int size);

+static bool HttpCreateConnection (http_t http);

+static void HttpCloseConnection (http_t http);

+static void HttpSetLastError (http_t http, const char* message);

+static void HttpSetLastErrorFromWinHttp (http_t http);

+static char* HttpFormatWinApiError (DWORD errorCode, HINSTANCE module);

+static char* HttpFormatWinHttpError (DWORD errorCode);

+static void HttpClearProxy (http_t http);

+

+# define WINHTTP_SAFE(condition) do { if (condition) break; HttpSetLastErrorFromWinHttp (http); return false; } while (false)

+# define WINHTTP_SAFE_DONE(condition) do { if (condition) break; HttpSetLastErrorFromWinHttp (http); goto done; } while (false)

+

+static char* HttpToString(const wchar_t* wideString, int size) {

+	int utfSize = WideCharToMultiByte(CP_UTF8, 0, wideString, size, NULL, 0, NULL, NULL);

+	char* utfMessage = malloc(utfSize + 1);

+	if (utfMessage)	{

+		utfMessage[utfSize] = 0;

+		WideCharToMultiByte(CP_UTF8, 0, wideString, size, utfMessage, utfSize, NULL, NULL);

+	}

+	return utfMessage;

+}

+

+static wchar_t* HttpToWideString(const char* string, int size) {

+	int wideSize = MultiByteToWideChar(CP_UTF8, 0, string, size, NULL, 0);

+	int wideBytes = (wideSize + 1) * sizeof(wchar_t);

+	wchar_t* wideMessage = malloc(wideBytes);

+	if (wideMessage) {

+		wideMessage[wideSize] = 0;

+		MultiByteToWideChar(CP_UTF8, 0, string, size, wideMessage, wideSize);

+	}

+	return wideMessage;

+}

+

+

+static bool HttpCreateConnection (http_t http) {

+	INTERNET_PORT defaultPort = INTERNET_DEFAULT_PORT;

+

+	HttpCloseConnection (http);

+

+	http->session = WinHttpOpen(

+		L"WinHTTP/1.0",

+		WINHTTP_ACCESS_TYPE_NO_PROXY,

+		WINHTTP_NO_PROXY_NAME,

+		WINHTTP_NO_PROXY_BYPASS,

+		0);

+	WINHTTP_SAFE(http->session != NULL);

+

+	WinHttpSetTimeouts(http->session,

+		60 * 1000,  // DNS time-out

+		60 * 1000,  // connect time-out

+		30 * 1000,  // send time-out

+		30 * 1000); // receive time-out

+

+	http->connection = WinHttpConnect(

+		http->session,

+		http->endpoint,

+		defaultPort,

+		0);

+	WINHTTP_SAFE(http->connection != NULL);

+

+	return true;

+}

+

+static void HttpCloseConnection (http_t http) {

+	if (http->connection) {

+		WinHttpCloseHandle(http->connection);

+		http->connection = NULL;

+	}

+

+	if (http->session) {

+		WinHttpCloseHandle(http->session);

+		http->session = NULL;

+	}

+}

+

+static void HttpSetLastError (http_t http, const char* message) {

+	free(http->error);

+	http->error = NULL;

+	

+	if (message)

+		http->error = strdup(message);

+}

+

+static void HttpSetLastErrorFromWinHttp (http_t http) {

+	free(http->error);

+	http->error = NULL;

+

+	DWORD error = GetLastError();

+	if (error)

+		http->error = HttpFormatWinHttpError(error);

+}

+

+static char* HttpFormatWinApiError (DWORD errorCode, HINSTANCE module) {

+	const int source_flag = module ? FORMAT_MESSAGE_FROM_HMODULE : FORMAT_MESSAGE_FROM_SYSTEM;

+

+	HLOCAL buffer = NULL;

+	int bufferLength = FormatMessageW(

+		FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_IGNORE_INSERTS | source_flag,

+		(void*)module,

+		errorCode,

+		MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT),

+		(LPWSTR)&buffer,

+		0,

+		NULL);

+

+	if (bufferLength > 0) {

+		char* message;

+

+		wchar_t* wideMessage = (wchar_t*)buffer;

+

+		/* Drop new line from the end. */

+		wchar_t* wideMessageBack = wideMessage + bufferLength - 1;

+		while (wideMessageBack > wideMessage && (*wideMessageBack == '\r' || *wideMessageBack == '\n'))

+			--wideMessageBack;

+

+		message = HttpToString (wideMessage, wideMessageBack - wideMessage + 1);

+

+		LocalFree(buffer);

+

+		return message;

+	}

+	else

+		return NULL;

+}

+

+static char* HttpFormatWinHttpError (DWORD errorCode) {

+	if (errorCode >= WINHTTP_ERROR_BASE && errorCode <= WINHTTP_ERROR_LAST) {

+		HMODULE module = GetModuleHandleW(L"WinHTTP.dll");

+		if (module) {

+			char* message = HttpFormatWinApiError(errorCode, module);

+			if (message)

+				return message;

+		}

+	}

+

+	return HttpFormatWinApiError(errorCode, NULL);

+}

+

+bool HttpInit(http_t* http, const char* endpoint, const char* securePort) {

+	http_t out = malloc(sizeof(struct _http_t));

+	if (!out)

+		return false;

+	memset(out, 0, sizeof(struct _http_t));

+

+	out->endpoint   = HttpToWideString(endpoint, -1);

+	out->securePort = HttpToWideString(securePort, -1);

+

+	if (!HttpCreateConnection (out)) {

+		HttpDestroy (out);

+		return false;

+	}

+

+	*http = out;

+	return true;

+}

+

+void HttpDestroy(http_t http) {

+	if (http) {

+		free(http->endpoint);

+		free(http->securePort);

+		http->endpoint = NULL;

+		http->securePort = NULL;

+		HttpCloseConnection (http);

+		HttpClearProxy (http);

+	}

+	free(http);

+}

+

+static void HttpClearProxy (http_t http) {

+	if (http->autoProxy) {

+		free(http->autoProxy);

+		http->autoProxy = NULL;

+	}

+

+	if (http->proxy) {

+		free(http->proxy);

+		http->proxy = NULL;

+	}

+

+	if (http->proxyUsername) {

+		free(http->proxyUsername);

+		http->proxyUsername = NULL;

+	}

+

+	if (http->proxyPassword) {

+		free(http->proxyPassword);

+		http->proxyPassword = NULL;

+	}

+}

+

+bool HttpSetAutoProxy (http_t http, const char* url) {

+	HttpClearProxy (http);

+	if (HttpSetProxy (http, url)) {

+		http->autoProxy = http->proxy;

+		http->proxy     = NULL;

+		return true;

+	}

+	else

+		return false;

+}

+

+bool HttpSetProxy (http_t http, const char* url) {

+	URL_COMPONENTS urlComponents;

+	wchar_t* wideUrl = NULL;

+	wchar_t* wideUrl2 = NULL;

+	wchar_t* wideUsername = NULL;

+	wchar_t* widePassword = NULL;

+

+	ZeroMemory(&urlComponents, sizeof(urlComponents));

+	urlComponents.dwStructSize      = sizeof(urlComponents);

+	urlComponents.dwUserNameLength = -1;

+	urlComponents.dwPasswordLength = -1;

+

+	wideUrl = HttpToWideString(url, -1);

+	if (WinHttpCrackUrl(wideUrl, wcslen(wideUrl), 0, &urlComponents)) {

+		if (urlComponents.lpszUserName && urlComponents.dwUserNameLength > 0) {

+			wideUsername = wcsdup(urlComponents.lpszUserName);

+			wideUsername[urlComponents.dwUserNameLength] = 0;

+		}

+		if (urlComponents.lpszPassword && urlComponents.dwPasswordLength > 0) {

+			widePassword = wcsdup(urlComponents.lpszPassword);

+			widePassword[urlComponents.dwPasswordLength] = 0;

+		}

+	}

+

+	ZeroMemory(&urlComponents, sizeof(urlComponents));

+	urlComponents.dwStructSize = sizeof(urlComponents);

+	urlComponents.dwHostNameLength  = -1;

+	urlComponents.dwUrlPathLength   = -1;

+

+	if (!WinHttpCrackUrl(wideUrl, wcslen(wideUrl), 0, &urlComponents)) {

+		free(wideUsername);

+		free(widePassword);

+		return false;

+	}

+

+	if (urlComponents.lpszHostName && urlComponents.dwHostNameLength > 0) {

+		wideUrl2 = wcsdup(urlComponents.lpszHostName);

+		wideUrl2[urlComponents.lpszUrlPath - urlComponents.lpszHostName] = 0;

+	}

+

+	free(wideUrl);

+

+	HttpClearProxy(http);

+	http->proxy         = wideUrl2;

+	http->proxyUsername = wideUsername;

+	http->proxyPassword = widePassword;

+	return true;

+}

+

+bool HttpRequest(http_t http, PianoRequest_t * const request) {

+	HINTERNET handle = NULL;

+	wchar_t* wideQuery = NULL;

+	bool requestSent = false;

+	bool complete = false;

+	int retryLimit = 3;

+	size_t responseDataSize;

+

+	wideQuery = HttpToWideString(request->urlPath, -1);

+	WINHTTP_SAFE_DONE(wideQuery != NULL);

+

+	handle = WinHttpOpenRequest(

+		http->connection,

+		L"POST",

+		wideQuery,

+		L"HTTP/1.1",

+		WINHTTP_NO_REFERER,

+		WINHTTP_DEFAULT_ACCEPT_TYPES,

+		request->secure ? WINHTTP_FLAG_SECURE : 0);

+	WINHTTP_SAFE_DONE(handle != NULL);

+

+	if (http->proxy || http->autoProxy) {

+		wchar_t* fullUrl;

+		DWORD fullUrlSize = 0;

+		WINHTTP_PROXY_INFO proxyInfo;

+		bool success;

+

+		if (http->autoProxy) {

+			WINHTTP_AUTOPROXY_OPTIONS proxyOptions = { 0 };

+

+			success = WinHttpQueryOption(request, WINHTTP_OPTION_URL, NULL, &fullUrlSize);

+			WINHTTP_SAFE(!success && GetLastError() == ERROR_INSUFFICIENT_BUFFER);

+			fullUrl = calloc(1, fullUrlSize + 1);

+			success = WinHttpQueryOption(request, WINHTTP_OPTION_URL, fullUrl, &fullUrlSize);

+			if (!success) {

+				free(fullUrl);

+				WINHTTP_SAFE(success);

+			}

+

+			proxyOptions.lpszAutoConfigUrl = http->autoProxy;

+			proxyOptions.dwFlags = WINHTTP_AUTOPROXY_CONFIG_URL;

+

+			if (!(success = WinHttpGetProxyForUrl(http->session, fullUrl, &proxyOptions, &proxyInfo))) {

+				proxyOptions.fAutoLogonIfChallenged = true;

+				success = WinHttpGetProxyForUrl(http->session, fullUrl, &proxyOptions, &proxyInfo);

+			}

+

+			if (!success) {

+				free(fullUrl);

+				WINHTTP_SAFE(success);

+			}

+		}

+		else {

+			proxyInfo.dwAccessType    = WINHTTP_ACCESS_TYPE_NAMED_PROXY;

+			proxyInfo.lpszProxy       = http->proxy;

+			proxyInfo.lpszProxyBypass = NULL;

+		}

+

+		WINHTTP_SAFE(WinHttpSetOption(handle,

+			WINHTTP_OPTION_PROXY,

+			&proxyInfo, sizeof(proxyInfo)));

+

+		if (http->proxyUsername && http->proxyPassword) {

+			WINHTTP_SAFE(WinHttpSetCredentials(handle,

+				WINHTTP_AUTH_TARGET_PROXY,

+				WINHTTP_AUTH_SCHEME_BASIC,

+				http->proxyUsername,

+				http->proxyPassword,

+				NULL));

+		}

+	}

+

+	while (retryLimit > 0) {

+		DWORD errorCode, statusCode, statusCodeSize;

+		bool succeeded = false;

+		bool retry = false;

+

+		if (!requestSent) {

+			size_t postDataSize = strlen(request->postData);

+			succeeded = WinHttpSendRequest(handle,

+				WINHTTP_NO_ADDITIONAL_HEADERS,

+				0,

+				request->postData,

+				postDataSize,

+				postDataSize,

+				0);

+

+			if (succeeded)

+				requestSent = true;

+		}

+

+		if (requestSent)

+			succeeded = WinHttpReceiveResponse(handle, NULL);

+

+		errorCode = GetLastError();

+

+		statusCode = 0;

+		statusCodeSize = sizeof(statusCode);

+		if (!WinHttpQueryHeaders(handle,

+				WINHTTP_QUERY_STATUS_CODE | WINHTTP_QUERY_FLAG_NUMBER,

+				WINHTTP_HEADER_NAME_BY_INDEX,

+				&statusCode, &statusCodeSize, WINHTTP_NO_HEADER_INDEX)) {

+			statusCode = 0;

+		}

+

+		if (succeeded && statusCode == 407) {

+			requestSent = false;

+			retry       = true;

+		}

+		else {

+			if (errorCode == ERROR_SUCCESS)

+				break;

+

+			switch (errorCode) {

+				case ERROR_WINHTTP_RESEND_REQUEST:

+					requestSent = false;

+					/* pass trough */

+

+				case ERROR_WINHTTP_NAME_NOT_RESOLVED:

+				case ERROR_WINHTTP_CANNOT_CONNECT:

+				case ERROR_WINHTTP_TIMEOUT:

+					retry = true;

+					break;

+

+				default:

+					HttpSetLastErrorFromWinHttp (http);

+					goto done;

+			}

+		}

+

+		if (retry)

+			--retryLimit;

+	}

+

+	responseDataSize = 0;

+	while (retryLimit > 0)

+	{

+		DWORD bytesLeft;

+		char* writePtr;

+

+		DWORD bytesAvailable = 0;

+		if (!WinHttpQueryDataAvailable(handle, &bytesAvailable)) {

+			WINHTTP_SAFE(GetLastError() == ERROR_WINHTTP_TIMEOUT);

+			--retryLimit;

+			continue;

+		}

+

+		if (0 == bytesAvailable)

+			break;

+

+		responseDataSize += bytesAvailable;

+		request->responseData = realloc(request->responseData, responseDataSize + 1);

+

+		writePtr = request->responseData + responseDataSize - bytesAvailable;

+		writePtr[bytesAvailable] = 0;

+

+		bytesLeft = bytesAvailable;

+		while (bytesLeft > 0)

+		{

+			DWORD bytesRead = 0;

+			if (!WinHttpReadData(handle, writePtr, bytesLeft, &bytesRead))

+			{

+				WINHTTP_SAFE(GetLastError() == ERROR_WINHTTP_TIMEOUT);

+				if (--retryLimit == 0)

+					break;

+

+				continue;

+			}

+

+			bytesLeft -= bytesRead;

+			writePtr  += bytesRead;

+		}

+

+		if (bytesLeft > 0)

+			HttpSetLastError (http, "Maximum retries count exceeded");

+	}

+

+	complete = true;

+

+	HttpSetLastError (http, NULL);

+

+done:

+	free(wideQuery);

+	return complete;

+}

+

+const char* HttpGetError(http_t http) {

+	return http->error;

+}