trustbridge/trustbridge: ui/downloader

comparison ui/downloader_win.cpp @ 23:7d07d261b003

Wrap HINTERNET handles in a class to take advantage of scoping

author	Andre Heinecke <aheinecke@intevation.de>
date	Thu, 20 Feb 2014 15:44:40 +0000
parents	fec1a610d43f
children	62cd56cea09b

comparison

equal deleted inserted replaced

-:bc302bbceaf5
+:7d07d261b003
 #define LIST_RESOURCE "/incoming/aheinecke/test"
 #define SW_RESOURCE "/incoming/aheinecke/test"
+/** @brief A wrapper around a HINTERNET structure that handles closing
+*
+* Holds a HINTERNET structure and closes it if necessary on destruction.
+*
+*/
+class SmartHINTERNET {
+public:
+SmartHINTERNET() : handle(NULL) {}
+~SmartHINTERNET() {
+if (handle) {
+WinHttpCloseHandle(handle);
+}
+}
+HINTERNET handle;
+};
 /** @brief Qt wrapper around FormatMessage
 *
 * @returns The error message of the error that occurred
 */
 }
 QDateTime Downloader::getLastModifiedHeader(HINTERNET hSession,
 HINTERNET hConnect, LPCWSTR resource)
 {
-HINTERNET hRequest = NULL;
+SmartHINTERNET sHRequest;
 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",
+if (!createRequest(hSession, hConnect, &sHRequest.handle, L"HEAD",
 resource)) {
-err = GetLastError();
+return retval;
-goto cleanup;
+}
-}
+if (!WinHttpSendRequest(sHRequest.handle,
-if (!WinHttpSendRequest(hRequest,
 WINHTTP_NO_ADDITIONAL_HEADERS,
 0, WINHTTP_NO_REQUEST_DATA, 0,
 0, 0)) {
-err = GetLastError();
+return retval;
-goto cleanup;
+}
-}
+if (!WinHttpReceiveResponse(sHRequest.handle, NULL)) {
-if (!WinHttpReceiveResponse(hRequest, NULL)) {
+return retval;
-err = GetLastError();
+}
-goto cleanup;
-}
+if (!verifyCertificate(sHRequest.handle)) {
-if (!verifyCertificate(hRequest)) {
 DEBUG << "Certificate verification failed";
-goto cleanup;
+return retval;
 }
-if (!(WinHttpQueryHeaders(hRequest,
+if (!(WinHttpQueryHeaders(sHRequest.handle,
 WINHTTP_QUERY_LAST_MODIFIED |
 WINHTTP_QUERY_FLAG_SYSTEMTIME,
 NULL,
 &lMod,
 &sizeOfSystemtime,
 WINHTTP_NO_HEADER_INDEX))) {
-err = GetLastError();
+return retval;
-goto cleanup;
 }
 retval = QDateTime(QDate(lMod.wYear, lMod.wMonth, lMod.wDay),
 QTime(lMod.wHour, lMod.wMinute, lMod.wSecond,
 lMod.wMilliseconds),
 Qt::UTC);
-// TODO remove and replace with a smart handle
-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;
+SmartHINTERNET sHRequest;
 bool retval = false;
 DWORD bytesAvailable = 0,
-err = 0,
 bytesRead = 0,
 totalDownloaded = 0,
 contentLength = 0,
 sizeOfDWORD = sizeof (DWORD);
 if (!hSession || !hConnect || !resource) {
 SetLastError(ERROR_INVALID_PARAMETER);
 return retval;
 }
-if (!createRequest(hSession, hConnect, &hRequest, L"GET",
+if (!createRequest(hSession, hConnect, &sHRequest.handle, L"GET",
 resource)) {
-err = GetLastError();
+return retval;
-goto cleanup;
+}
-}
+if (!WinHttpSendRequest(sHRequest.handle,
-if (!WinHttpSendRequest(hRequest,
 WINHTTP_NO_ADDITIONAL_HEADERS,
 0, WINHTTP_NO_REQUEST_DATA, 0,
 0, 0)) {
-err = GetLastError();
+return retval;
-goto cleanup;
+}
-}
+if (!WinHttpReceiveResponse(sHRequest.handle, NULL)) {
-if (!WinHttpReceiveResponse(hRequest, NULL)) {
+return retval;
-err = GetLastError();
+}
-goto cleanup;
-}
+if (!verifyCertificate(sHRequest.handle)) {
-if (!verifyCertificate(hRequest)) {
 DEBUG << "Certificate verification failed";
-goto cleanup;
+return retval;
 }
-if (!(WinHttpQueryHeaders(hRequest,
+if (!(WinHttpQueryHeaders(sHRequest.handle,
 WINHTTP_QUERY_CONTENT_LENGTH |
 WINHTTP_QUERY_FLAG_NUMBER,
 NULL,
 &contentLength,
 &sizeOfDWORD,
 // much we download.
 DEBUG << "No content-length";
 }
 if (contentLength > maxSize) {
-err = ERROR_INVALID_DATA;
+return retval;
-goto cleanup;
 }
 if (contentLength) {
 QFileInfo finf(fileName);
 if (finf.exists() && finf.isReadable() &&
 finf.size() == contentLength) {
 // We already have data of the same size
 // No need to waste bandwidth.
 DEBUG << "Skipping download because file exists";
 retval = true;
-goto cleanup;
+return retval;
 }
 }
 // Open / Create the file to write to.
 if (!outputFile.open(QIODevice::WriteOnly)) {
 DEBUG << "Failed to open file";
-err = GetLastError();
+return retval;
-goto cleanup;
 }
 DEBUG << "output file size: " << outputFile.size();
 do
 {
 char outBuf[8192]; // 8KB is the internal buffer size of winhttp
 memset(outBuf, 0, sizeof(outBuf));
 bytesRead = 0;
-if (!WinHttpQueryDataAvailable(hRequest, &bytesAvailable)) {
+if (!WinHttpQueryDataAvailable(sHRequest.handle, &bytesAvailable)) {
 DEBUG << "Querying for available data failed";
 retval = false;
-err = GetLastError();
 break;
 }
 if (!bytesAvailable) {
 // Might indicate that we are done.
 }
 if (bytesAvailable > maxSize) {
 DEBUG << "File to large";
 retval = false;
-err = ERROR_INVALID_DATA;
+SetLastError(ERROR_INVALID_DATA);
 break;
 }
-if (!WinHttpReadData(hRequest, (LPVOID)outBuf,
+if (!WinHttpReadData(sHRequest.handle, (LPVOID)outBuf,
 sizeof(outBuf), &bytesRead)) {
 DEBUG << "Error reading data";
-err = GetLastError();
 break;
 } else {
 if (bytesRead) {
 // 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 = false;
 break;
 }
 } while (bytesAvailable > 0);
-// TODO remove and replace with a smart handle
-cleanup:
 if (retval && outputFile.isOpen()) {
 // 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 results = false;
-HINTERNET hSession = NULL,
+SmartHINTERNET sHSession;
-hConnect = NULL;
+SmartHINTERNET sHConnect;
 wchar_t wUrl[mUrl.size() + 1];
 QDateTime lastModifiedSoftware;
 QDateTime lastModifiedList;
 int rc = 0;
 // Should not be necessary because we initialized the memory
 wUrl[rc] = '\0';
 // Initialize connection
-if (!openSession(&hSession)) {
+if (!openSession(&sHConnect.handle)) {
 DEBUG << "Failed to open session: " << getLastErrorMsg();
 return;
 }
-if (!initializeConnection(hSession, &hConnect, wUrl)) {
+if (!initializeConnection(sHConnect.handle, &sHConnect.handle, wUrl)) {
 DEBUG << "Failed to initialize connection: " << getLastErrorMsg();
-goto cleanup;
+return;
 }
-lastModifiedSoftware = getLastModifiedHeader(hSession, hConnect,
+lastModifiedSoftware = getLastModifiedHeader(sHConnect.handle, sHConnect.handle,
 L""SW_RESOURCE);
-lastModifiedList = getLastModifiedHeader(hSession, hConnect,
+lastModifiedList = getLastModifiedHeader(sHConnect.handle, sHConnect.handle,
 L""LIST_RESOURCE);
 if (!lastModifiedList.isValid() || !lastModifiedSoftware.isValid()) {
 DEBUG << "Could not read headers: " << getLastErrorMsg();
-goto cleanup;
+return;
 }
 if (!mLastModSW.isValid() || lastModifiedSoftware > mLastModSW) {
 QString dataDirectory = getDataDirectory();
 if (dataDirectory.isEmpty()) {
 DEBUG << "Failed to get data directory";
-goto cleanup;
+return;
 }
 QString fileName = dataDirectory.append("/SW-")
 .append(lastModifiedSoftware.toString("yyyymmddHHmmss"))
 .append(".exe");
 DEBUG << "fileName: " << fileName;
-if (!downloadFile(hSession, hConnect, L""SW_RESOURCE,
+if (!downloadFile(sHConnect.handle, sHConnect.handle, L""SW_RESOURCE,
 fileName, MAX_SW_SIZE)) {
 DEBUG << "Error downloading File: " << getLastErrorMsg();
-goto cleanup;
+return;
 }
 emit newSoftwareAvailable(fileName, lastModifiedSoftware);
 } else if (!mLastModList.isValid() || lastModifiedList > mLastModList) {
 QString dataDirectory = getDataDirectory();
 if (dataDirectory.isEmpty()) {
 DEBUG << "Failed to get data directory";
-goto cleanup;
+return;
 }
 QString fileName = dataDirectory.append("/list-")
 .append(lastModifiedSoftware.toString("yyyymmddHHmmss"))
 .append(".txt");
 DEBUG << "fileName: " << fileName;
-if (!downloadFile(hSession, hConnect, L""LIST_RESOURCE,
+if (!downloadFile(sHConnect.handle, sHConnect.handle, L""LIST_RESOURCE,
 fileName, MAX_LIST_SIZE)) {
 DEBUG << "Error downloading File: " << getLastErrorMsg();
-goto cleanup;
+return;
 }
 emit newListAvailable(fileName, lastModifiedList);
 }
 DEBUG << "SW date: " << lastModifiedSoftware;
 DEBUG << "List date: " << lastModifiedList;
-/*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"), ErrUnknown);
 }
-// TODO remove and replace with a smart handle
-cleanup:
-if (hConnect) {
-WinHttpCloseHandle(hConnect);
-}
-if (hSession) {
-WinHttpCloseHandle(hSession);
-}
 return;
 }
 #endif

Mercurial > trustbridge > trustbridge

comparison ui/downloader_win.cpp @ 23:7d07d261b003