summaryrefslogtreecommitdiff
path: root/src/net/download.cpp
diff options
context:
space:
mode:
authorAndrei Karas <akaras@inbox.ru>2011-01-02 01:48:38 +0200
committerAndrei Karas <akaras@inbox.ru>2011-01-02 02:41:24 +0200
commit3eeae12c498d1a4dbe969462d2ba841f77ee3ccb (patch)
treeff8eab35e732bc0749fc11677c8873a7b3a58704 /src/net/download.cpp
downloadplus-3eeae12c498d1a4dbe969462d2ba841f77ee3ccb.tar.gz
plus-3eeae12c498d1a4dbe969462d2ba841f77ee3ccb.tar.bz2
plus-3eeae12c498d1a4dbe969462d2ba841f77ee3ccb.tar.xz
plus-3eeae12c498d1a4dbe969462d2ba841f77ee3ccb.zip
Initial commit.
This code based on mana client http://www.gitorious.org/mana/mana and my private repository.
Diffstat (limited to 'src/net/download.cpp')
-rw-r--r--src/net/download.cpp355
1 files changed, 355 insertions, 0 deletions
diff --git a/src/net/download.cpp b/src/net/download.cpp
new file mode 100644
index 000000000..2d391b783
--- /dev/null
+++ b/src/net/download.cpp
@@ -0,0 +1,355 @@
+/*
+ * The Mana Client
+ * Copyright (C) 2009-2010 The Mana Developers
+ *
+ * This file is part of The Mana Client.
+ *
+ * This program is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation; either version 2 of the License, or
+ * any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program. If not, see <http://www.gnu.org/licenses/>.
+ */
+
+#include "net/download.h"
+
+#include "configuration.h"
+#include "log.h"
+#include "main.h"
+
+#include "utils/stringutils.h"
+
+#include <curl/curl.h>
+
+#include <SDL.h>
+#include <SDL_thread.h>
+
+#include <zlib.h>
+
+const char *DOWNLOAD_ERROR_MESSAGE_THREAD
+ = "Could not create download thread!";
+
+/**
+ * Calculates the Alder-32 checksum for the given file.
+ */
+static unsigned long fadler32(FILE *file)
+{
+ // Obtain file size
+ fseek(file, 0, SEEK_END);
+ long fileSize = ftell(file);
+ rewind(file);
+
+ // Calculate Adler-32 checksum
+ char *buffer = static_cast<char*>(malloc(fileSize));
+ const size_t read = fread(buffer, 1, fileSize, file);
+ unsigned long adler = adler32(0L, Z_NULL, 0);
+ adler = adler32(static_cast<uInt>(adler), (Bytef*)buffer, read);
+ free(buffer);
+
+ return adler;
+}
+
+enum
+{
+ OPTIONS_NONE = 0,
+ OPTIONS_MEMORY = 1
+};
+
+namespace Net
+{
+
+Download::Download(void *ptr, const std::string &url,
+ DownloadUpdate updateFunction, bool ignoreError):
+ mPtr(ptr),
+ mUrl(url),
+ mFileName(""),
+ mWriteFunction(NULL),
+ mUpdateFunction(updateFunction),
+ mThread(NULL),
+ mCurl(NULL),
+ mHeaders(NULL),
+ mIgnoreError(ignoreError)
+{
+ mError = static_cast<char*>(malloc(CURL_ERROR_SIZE + 1));
+ mError[0] = 0;
+
+ mOptions.cancel = false;
+}
+
+Download::~Download()
+{
+ if (mHeaders)
+ curl_slist_free_all(mHeaders);
+
+ int status;
+ if (mThread && SDL_GetThreadID(mThread))
+ SDL_WaitThread(mThread, &status);
+ mThread = 0;
+ free(mError);
+}
+
+void Download::addHeader(const std::string &header)
+{
+ mHeaders = curl_slist_append(mHeaders, header.c_str());
+}
+
+void Download::noCache()
+{
+ addHeader("pragma: no-cache");
+ addHeader("Cache-Control: no-cache");
+}
+
+void Download::setFile(const std::string &filename, Sint64 adler32)
+{
+ mOptions.memoryWrite = false;
+ mFileName = filename;
+
+ if (adler32 > -1)
+ {
+ mAdler = static_cast<unsigned long>(adler32);
+ mOptions.checkAdler = true;
+ }
+ else
+ {
+ mOptions.checkAdler = false;
+ }
+}
+
+void Download::setWriteFunction(WriteFunction write)
+{
+ mOptions.memoryWrite = true;
+ mWriteFunction = write;
+}
+
+bool Download::start()
+{
+ logger->log("Starting download: %s", mUrl.c_str());
+
+ mThread = SDL_CreateThread(downloadThread, this);
+
+ if (!mThread)
+ {
+ logger->log1(DOWNLOAD_ERROR_MESSAGE_THREAD);
+ strcpy(mError, DOWNLOAD_ERROR_MESSAGE_THREAD);
+ mUpdateFunction(mPtr, DOWNLOAD_STATUS_THREAD_ERROR, 0, 0);
+ if (!mIgnoreError)
+ return false;
+ }
+
+ return true;
+}
+
+void Download::cancel()
+{
+ logger->log("Canceling download: %s", mUrl.c_str());
+
+ mOptions.cancel = true;
+ if (mThread && SDL_GetThreadID(mThread))
+ SDL_WaitThread(mThread, NULL);
+
+ mThread = NULL;
+}
+
+char *Download::getError()
+{
+ return mError;
+}
+
+int Download::downloadProgress(void *clientp, double dltotal, double dlnow,
+ double ultotal _UNUSED_, double ulnow _UNUSED_)
+{
+ Download *d = reinterpret_cast<Download*>(clientp);
+ if (!d)
+ return -5;
+
+ if (d->mOptions.cancel)
+ {
+ return d->mUpdateFunction(d->mPtr, DOWNLOAD_STATUS_CANCELLED,
+ static_cast<size_t>(dltotal),
+ static_cast<size_t>(dlnow));
+ return -5;
+ }
+
+ return d->mUpdateFunction(d->mPtr, DOWNLOAD_STATUS_IDLE,
+ static_cast<size_t>(dltotal),
+ static_cast<size_t>(dlnow));
+}
+
+int Download::downloadThread(void *ptr)
+{
+ int attempts = 0;
+ bool complete = false;
+ Download *d = reinterpret_cast<Download*>(ptr);
+ CURLcode res;
+ std::string outFilename;
+
+ if (!d)
+ return 0;
+
+ if (!d->mOptions.memoryWrite)
+ outFilename = d->mFileName + ".part";
+
+ while (attempts < 3 && !complete && !d->mOptions.cancel)
+ {
+ FILE *file = NULL;
+
+ d->mUpdateFunction(d->mPtr, DOWNLOAD_STATUS_STARTING, 0, 0);
+
+ if (d->mOptions.cancel)
+ {
+ //need terminate thread?
+ d->mThread = NULL;
+ return 0;
+ }
+
+ d->mCurl = curl_easy_init();
+
+ if (d->mCurl && !d->mOptions.cancel)
+ {
+ logger->log("Downloading: %s", d->mUrl.c_str());
+
+ curl_easy_setopt(d->mCurl, CURLOPT_FOLLOWLOCATION, 1);
+ curl_easy_setopt(d->mCurl, CURLOPT_HTTPHEADER, d->mHeaders);
+// curl_easy_setopt(d->mCurl, CURLOPT_HTTP_VERSION, CURL_HTTP_VERSION_1_1);
+
+ if (d->mOptions.memoryWrite)
+ {
+ curl_easy_setopt(d->mCurl, CURLOPT_FAILONERROR, 1);
+ curl_easy_setopt(d->mCurl, CURLOPT_WRITEFUNCTION,
+ d->mWriteFunction);
+ curl_easy_setopt(d->mCurl, CURLOPT_WRITEDATA, d->mPtr);
+ }
+ else
+ {
+ file = fopen(outFilename.c_str(), "w+b");
+ curl_easy_setopt(d->mCurl, CURLOPT_WRITEDATA, file);
+ }
+
+ curl_easy_setopt(d->mCurl, CURLOPT_USERAGENT,
+ strprintf(PACKAGE_EXTENDED_VERSION,
+ branding.getStringValue("appShort").c_str()).c_str());
+ curl_easy_setopt(d->mCurl, CURLOPT_ERRORBUFFER, d->mError);
+ curl_easy_setopt(d->mCurl, CURLOPT_URL, d->mUrl.c_str());
+ curl_easy_setopt(d->mCurl, CURLOPT_NOPROGRESS, 0);
+ curl_easy_setopt(d->mCurl, CURLOPT_PROGRESSFUNCTION,
+ downloadProgress);
+ curl_easy_setopt(d->mCurl, CURLOPT_PROGRESSDATA, ptr);
+ curl_easy_setopt(d->mCurl, CURLOPT_NOSIGNAL, 1);
+ curl_easy_setopt(d->mCurl, CURLOPT_CONNECTTIMEOUT, 30);
+ curl_easy_setopt(d->mCurl, CURLOPT_TIMEOUT, 1800);
+
+ if ((res = curl_easy_perform(d->mCurl)) != 0
+ && !d->mOptions.cancel)
+ {
+ switch (res)
+ {
+ case CURLE_ABORTED_BY_CALLBACK:
+ d->mOptions.cancel = true;
+ break;
+ case CURLE_COULDNT_CONNECT:
+ default:
+ logger->log("curl error %d: %s host: %s",
+ res, d->mError, d->mUrl.c_str());
+ break;
+ }
+
+ if (d->mOptions.cancel)
+ break;
+
+ d->mUpdateFunction(d->mPtr, DOWNLOAD_STATUS_ERROR, 0, 0);
+
+ if (!d->mOptions.memoryWrite)
+ {
+ fclose(file);
+ ::remove(outFilename.c_str());
+ }
+ attempts++;
+ continue;
+ }
+
+ curl_easy_cleanup(d->mCurl);
+ d->mCurl = 0;
+
+ if (!d->mOptions.memoryWrite)
+ {
+ // Don't check resources.xml checksum
+ if (d->mOptions.checkAdler)
+ {
+ unsigned long adler = fadler32(file);
+
+ if (d->mAdler != adler)
+ {
+ fclose(file);
+
+ // Remove the corrupted file
+ ::remove(d->mFileName.c_str());
+ logger->log("Checksum for file %s failed: (%lx/%lx)",
+ d->mFileName.c_str(),
+ adler, d->mAdler);
+ attempts++;
+ continue; // Bail out here to avoid the renaming
+ }
+ }
+ fclose(file);
+
+ // Any existing file with this name is deleted first, otherwise
+ // the rename will fail on Windows.
+ ::remove(d->mFileName.c_str());
+ ::rename(outFilename.c_str(), d->mFileName.c_str());
+
+ // Check if we can open it and no errors were encountered
+ // during renaming
+ file = fopen(d->mFileName.c_str(), "rb");
+ if (file)
+ {
+ fclose(file);
+ complete = true;
+ }
+ }
+ else
+ {
+ // It's stored in memory, we're done
+ complete = true;
+ }
+ }
+
+ if (d->mCurl)
+ {
+ curl_easy_cleanup(d->mCurl);
+ d->mCurl = 0;
+ }
+
+ if (d->mOptions.cancel)
+ {
+ //need ternibate thread?
+ d->mThread = NULL;
+ return 0;
+ }
+ attempts++;
+ }
+
+ if (d->mOptions.cancel)
+ {
+ // Nothing to do...
+ }
+ else if (!complete || attempts >= 3)
+ {
+ d->mUpdateFunction(d->mPtr, DOWNLOAD_STATUS_ERROR, 0, 0);
+ }
+ else
+ {
+ d->mUpdateFunction(d->mPtr, DOWNLOAD_STATUS_COMPLETE, 0, 0);
+ }
+
+ d->mThread = NULL;
+ return 0;
+}
+
+} // namespace Net