trustbridge/trustbridge: ui/downloader

comparison ui/downloader_win.cpp @ 15:95e1b6edf2fc

Implement more downloader functionality for Windows

author	Andre Heinecke <aheinecke@intevation.de>
date	Wed, 19 Feb 2014 10:45:06 +0000
parents	7e2f14c7aba2
children	c12825a651ed

comparison

equal deleted inserted replaced

-:58d51a91d448
+:95e1b6edf2fc
 #include <windows.h>
 #include <winhttp.h>
 #include <QDebug>
+#include <QDateTime>
+#include <QSaveFile>
 #define DEBUG if (1) qDebug() << __PRETTY_FUNCTION__
+#define MAX_SW_SIZE 10485760
+#define MAX_LIST_SIZE 1048576
+/** @brief Qt wrapper around FormatMessage
+*
+* @returns The error message of the error that occurred
+*/
 const QString getLastErrorMsg() {
 LPWSTR bufPtr = NULL;
 DWORD err = GetLastError();
 FormatMessageW(FORMAT_MESSAGE_ALLOCATE_BUFFER |
 FORMAT_MESSAGE_FROM_SYSTEM |
 FORMAT_MESSAGE_IGNORE_INSERTS,
 NULL, err, 0, (LPWSTR)&bufPtr, 0, NULL);
+if (!bufPtr) {
+HMODULE hWinhttp = GetModuleHandleW(L"winhttp");
+if (hWinhttp) {
+FormatMessageW(FORMAT_MESSAGE_ALLOCATE_BUFFER |
+FORMAT_MESSAGE_FROM_HMODULE |
+FORMAT_MESSAGE_IGNORE_INSERTS,
+hWinhttp, HRESULT_CODE(err), 0,
+(LPWSTR)&bufPtr, 0, NULL);
+}
+}
 const QString result =
 (bufPtr) ? QString::fromUtf16((const ushort*)bufPtr).trimmed() :
 QString("Unknown Error %1").arg(err);
 LocalFree(bufPtr);
 return result;
 }
-/** @brief open a session with appropiate proxy settings
+/** @brief open a session with appropriate proxy settings
 *
 * @param[inout] *pHSession pointer to a HInternet structure
 *
 * On error call getLastError to get extended error information.
 *
 if (!pHSession) {
 SetLastError(ERROR_INVALID_PARAMETER);
 return false;
 }
-qDebug() << "2";
 memset(&proxyConfig, 0, sizeof (WINHTTP_CURRENT_USER_IE_PROXY_CONFIG));
 if (WinHttpGetIEProxyConfigForCurrentUser(&proxyConfig)) {
 if (proxyConfig.fAutoDetect) {
 // TODO Handle this
 */
 bool createRequest(HINTERNET hSession, HINTERNET hConnect,
 HINTERNET *pHRequest, LPCWSTR requestType, LPCWSTR resource)
 {
+DWORD dwSSLFlag;
 DEBUG;
 if (!hSession || !hConnect || !pHRequest) {
 SetLastError(ERROR_INVALID_PARAMETER);
 return false;
 }
 *pHRequest = WinHttpOpenRequest(hConnect, requestType, resource,
 NULL, WINHTTP_NO_REFERER,
 WINHTTP_DEFAULT_ACCEPT_TYPES,
 WINHTTP_FLAG_SECURE);
+dwSSLFlag = SECURITY_FLAG_IGNORE_UNKNOWN_CA;
+dwSSLFlag |= SECURITY_FLAG_IGNORE_CERT_DATE_INVALID;
+dwSSLFlag |= SECURITY_FLAG_IGNORE_CERT_CN_INVALID;
+dwSSLFlag |= SECURITY_FLAG_IGNORE_CERT_WRONG_USAGE;
+WinHttpSetOption(*pHRequest, WINHTTP_OPTION_SECURITY_FLAGS,
+&dwSSLFlag, sizeof(dwSSLFlag));
 return *pHRequest;
 }
+bool Downloader::verifyCertificate(HINTERNET hRequest)
+{
+CERT_CONTEXT *certContext = NULL;
+DWORD certContextLen = sizeof(CERT_CONTEXT);
+bool retval = false;
+if (!WinHttpQueryOption(hRequest,
+WINHTTP_OPTION_SERVER_CERT_CONTEXT,
+&certContext,
+&certContextLen)) {
+DEBUG << "Unable to get server certificate";
+return false;
+}
+QByteArray serverCert ((const char *) certContext->pbCertEncoded,
+certContext->cbCertEncoded);
+retval = (serverCert == mCert);
+if (!retval) {
+DEBUG << "Certificate is not the same as the pinned one!"
+<< "Base64 cert: " << serverCert.toBase64();
+}
+CertFreeCertificateContext(certContext);
+return retval;
+}
+QDateTime Downloader::getLastModifiedHeader(HINTERNET hSession,
+HINTERNET hConnect, LPCWSTR resource)
+{
+HINTERNET hRequest = NULL;
+SYSTEMTIME lMod;
+DWORD sizeOfSystemtime = sizeof (SYSTEMTIME);
+QDateTime retval;
+DWORD err = 0;
+memset(&lMod, 0, sizeof (SYSTEMTIME));
+if (!hSession || !hConnect || !resource) {
+SetLastError(ERROR_INVALID_PARAMETER);
+return retval;
+}
+if (!createRequest(hSession, hConnect, &hRequest, L"HEAD",
+resource)) {
+err = GetLastError();
+goto cleanup;
+}
+if (!WinHttpSendRequest(hRequest,
+WINHTTP_NO_ADDITIONAL_HEADERS,
+0, WINHTTP_NO_REQUEST_DATA, 0,
+0, 0)) {
+err = GetLastError();
+goto cleanup;
+}
+if (!WinHttpReceiveResponse(hRequest, NULL)) {
+err = GetLastError();
+goto cleanup;
+}
+if (!verifyCertificate(hRequest)) {
+DEBUG << "Certificate verification failed";
+// TODO error out
+}
+if (!(WinHttpQueryHeaders(hRequest,
+WINHTTP_QUERY_LAST_MODIFIED |
+WINHTTP_QUERY_FLAG_SYSTEMTIME,
+NULL,
+&lMod,
+&sizeOfSystemtime,
+WINHTTP_NO_HEADER_INDEX))) {
+err = GetLastError();
+goto cleanup;
+}
+retval = QDateTime(QDate(lMod.wYear, lMod.wMonth, lMod.wDay),
+QTime(lMod.wHour, lMod.wMinute, lMod.wSecond,
+lMod.wMilliseconds),
+Qt::UTC);
+cleanup:
+if (hRequest) {
+WinHttpCloseHandle(hRequest);
+}
+// Close handle might overwrite the last error.
+SetLastError(err);
+return retval;
+}
+bool Downloader::downloadFile(HINTERNET hSession, HINTERNET hConnect,
+LPCWSTR resource, const QString &filename, DWORD maxSize)
+{
+HINTERNET hRequest = NULL;
+bool retval = false;
+DWORD bytesAvailable = 0,
+err = 0,
+bytesRead = 0,
+totalDownloaded = 0;
+QSaveFile outputFile(filename);
+if (!hSession || !hConnect || !resource) {
+SetLastError(ERROR_INVALID_PARAMETER);
+return retval;
+}
+if (!createRequest(hSession, hConnect, &hRequest, L"GET",
+resource)) {
+err = GetLastError();
+goto cleanup;
+}
+if (!WinHttpSendRequest(hRequest,
+WINHTTP_NO_ADDITIONAL_HEADERS,
+0, WINHTTP_NO_REQUEST_DATA, 0,
+0, 0)) {
+err = GetLastError();
+goto cleanup;
+}
+if (!WinHttpReceiveResponse(hRequest, NULL)) {
+err = GetLastError();
+goto cleanup;
+}
+if (!verifyCertificate(hRequest)) {
+DEBUG << "Certificate verification failed";
+// TODO error out
+}
+// Open / Create the file to write to.
+if (!outputFile.open(QIODevice::WriteOnly)) {
+DEBUG << "Failed to open file";
+err = GetLastError();
+goto cleanup;
+}
+do
+{
+char outBuf[8192]; // 8KB is the internal buffer size of winhttp
+memset(outBuf, 0, sizeof(outBuf));
+bytesRead = 0;
+if (!WinHttpQueryDataAvailable(hRequest, &bytesAvailable)) {
+DEBUG << "Querying for available data failed";
+retval = false;
+err = GetLastError();
+break;
+}
+if (!bytesAvailable) {
+// Might indicate that we are done.
+break;
+}
+if (bytesAvailable > maxSize) {
+DEBUG << "File to large";
+retval = false;
+break;
+}
+if (!WinHttpReadData(hRequest, (LPVOID)outBuf,
+sizeof(outBuf), &bytesRead)) {
+DEBUG << "Error reading data";
+err = GetLastError();
+break;
+} else {
+if (bytesRead) {
+DEBUG << "Downloaded: " << bytesRead << "B";
+// Write data to file.
+if (outputFile.write(outBuf, bytesRead) !=
+bytesRead) {
+err = GetLastError();
+DEBUG << "Error writing to file.";
+retval = false;
+}
+// Completed a read / write cycle. If not error follows
+// the download was successful.
+retval = true;
+} else {
+// Should not happen as we queried for available
+// bytes before and the function did not return an
+// error.
+DEBUG << "Unable to read available data";
+retval = false;
+break;
+}
+}
+totalDownloaded += bytesRead;
+if (totalDownloaded > maxSize) {
+DEBUG << "Downloaded too much data. Breaking.";
+retval = false;
+break;
+}
+} while (bytesAvailable > 0);
+cleanup:
+if (retval) {
+// Actually save the file to disk / move to homedir
+retval = outputFile.commit();
+}
+if (hRequest) {
+WinHttpCloseHandle(hRequest);
+}
+// Close handle might overwrite the last error.
+SetLastError(err);
+return retval;
+}
 void Downloader::run() {
-BOOL bResults = FALSE;
+bool results = false;
 HINTERNET hSession = NULL,
-hConnect = NULL,
+hConnect = NULL;
-hRequest = NULL;
+wchar_t wUrl[mUrl.size() + 1];
+QDateTime lastModifiedSoftware;
-SYSTEMTIME lastModified;
+QDateTime lastModifiedList;
-DWORD sizeOfSystemtime = sizeof (SYSTEMTIME);
+int rc = 0;
-memset(&lastModified, 0, sizeof (SYSTEMTIME));
+memset(wUrl, 0, sizeof (wchar_t) * (mUrl.size() + 1));
+rc = mUrl.toWCharArray(wUrl);
+if (rc != mUrl.size()) {
+DEBUG << "Problem converting to wchar array";
+return;
+}
+// Should not be necessary because we initialized the memory
+wUrl[rc] = '\0';
+// Initialize connection
 if (!openSession(&hSession)) {
 DEBUG << "Failed to open session: " << getLastErrorMsg();
 return;
 }
+if (!initializeConnection(hSession, &hConnect, wUrl)) {
-if (!initializeConnection(hSession, &hConnect, L"www.intevation.de")) {
 DEBUG << "Failed to initialize connection: " << getLastErrorMsg();
 goto cleanup;
 }
-if (!createRequest(hSession, hConnect, &hRequest, L"GET", L"/")) {
-DEBUG << "Failed to create the request: " << getLastErrorMsg();
+lastModifiedSoftware = getLastModifiedHeader(hSession, hConnect,
-goto cleanup;
+L"/incoming/aheinecke/test");
-}
+lastModifiedList = getLastModifiedHeader(hSession, hConnect,
-if (hRequest) {
+L"/incoming/aheinecke/test");
-DEBUG << "Doing Request";
-bResults = WinHttpSendRequest(hRequest,
+if (!lastModifiedList.isValid() || !lastModifiedSoftware.isValid()) {
-WINHTTP_NO_ADDITIONAL_HEADERS,
+DEBUG << "Could not read headers: " << getLastErrorMsg();
-0, WINHTTP_NO_REQUEST_DATA, 0,
+goto cleanup;
-0, 0);
+}
-} else {
-DEBUG << "Error: " << GetLastError();
+if (!mLastModSW.isValid() || lastModifiedSoftware > mLastModSW) {
-}
+QString dataDirectory = getDataDirectory();
+if (dataDirectory.isEmpty()) {
-if (bResults) {
+DEBUG << "Failed to get data directory";
-DEBUG << "Recieving Response";
+goto cleanup;
-bResults = WinHttpReceiveResponse(hRequest, NULL);
+}
-} else {
-DEBUG << "Error: " << GetLastError();
+QString fileName = dataDirectory.append("/SW-")
-}
+.append(lastModifiedSoftware.toString("yyyymmddHHmmss"))
+.append(".exe");
-if (bResults) {
-DEBUG << "Querying Headers";
+DEBUG << "Filename: " << fileName;
-bResults = WinHttpQueryHeaders(hRequest,
-WINHTTP_QUERY_LAST_MODIFIED |
+if (!downloadFile(hSession, hConnect, L"/incoming/aheinecke/test",
-WINHTTP_QUERY_FLAG_SYSTEMTIME,
+fileName, MAX_SW_SIZE)) {
-NULL,
+DEBUG << "Error downloading File: " << getLastErrorMsg();
-&lastModified,
+goto cleanup;
-&sizeOfSystemtime,
+}
-WINHTTP_NO_HEADER_INDEX);
-} else {
+emit newSoftwareAvailable(fileName, lastModifiedSoftware);
-DWORD errCode = GetLastError();
+} else if (!mLastModList.isValid() || lastModifiedList > mLastModList) {
-switch (errCode) {
+QString dataDirectory = getDataDirectory();
-case ERROR_WINHTTP_HEADER_NOT_FOUND:
-DEBUG << "Header not found";
+if (dataDirectory.isEmpty()) {
-break;
+DEBUG << "Failed to get data directory";
-case ERROR_WINHTTP_INCORRECT_HANDLE_STATE:
+goto cleanup;
-DEBUG << "Incorrect handle state";
+}
-break;
-case ERROR_WINHTTP_INCORRECT_HANDLE_TYPE:
+QString fileName = dataDirectory.append("/list-")
-DEBUG << "Incorrect handle type";
+.append(lastModifiedSoftware.toString("yyyymmddHHmmss"))
-break;
+.append(".txt");
-case ERROR_WINHTTP_INTERNAL_ERROR:
-DEBUG << "Internal error";
+DEBUG << "Filename: " << fileName;
-break;
-case ERROR_NOT_ENOUGH_MEMORY:
+if (!downloadFile(hSession, hConnect, L"/incoming/aheinecke/test",
-DEBUG << "OOM";
+fileName, MAX_LIST_SIZE)) {
-break;
+DEBUG << "Error downloading File: " << getLastErrorMsg();
-default:
+goto cleanup;
-DEBUG << "Error: " << getLastErrorMsg();
+}
-}
-}
+emit newListAvailable(fileName, lastModifiedList);
+}
-DEBUG << "Last modified year: " << lastModified.wYear;
+DEBUG << "SW date: " << lastModifiedSoftware;
+DEBUG << "List date: " << lastModifiedList;
-if (!bResults) {
+/*if (!WinHttpQueryDataAvailable(hRequest, &dataAvaiable)) {
+DEBUG << "Failed to query data Available: " << getLastErrorMsg();
+goto cleanup;
+}*/
+if (!results) {
 // Report any errors.
 DEBUG << "Error" << GetLastError();
-emit error(tr("Unknown Problem when connecting"), Unknown);
+emit error(tr("Unknown Problem when connecting"), ErrUnknown);
 }
 cleanup:
-if (hRequest) {
-WinHttpCloseHandle(hRequest);
-}
 if (hConnect) {
 WinHttpCloseHandle(hConnect);
 }
 if (hSession) {
 WinHttpCloseHandle(hSession);
 }

Mercurial > trustbridge > trustbridge

comparison ui/downloader_win.cpp @ 15:95e1b6edf2fc