// SuperTux
-// Copyright (C) 2014 Ingo Ruhnke <grumbel@gmail.com>
+// Copyright (C) 2007 Christoph Sommer <christoph.sommer@2007.expires.deltadevelopment.de>
+// 2014 Ingo Ruhnke <grumbel@gmail.com>
//
// 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
#include "addon/downloader.hpp"
-#include <curl/curl.h>
-#include <curl/easy.h>
-#include <physfs.h>
+#include <algorithm>
#include <memory>
+#include <physfs.h>
#include <stdexcept>
#include "util/log.hpp"
} // namespace
-Downloader::Downloader()
+class Transfer
+{
+private:
+ Downloader& m_downloader;
+ TransferId m_id;
+
+ std::string m_url;
+ CURL* m_handle;
+ std::array<char, CURL_ERROR_SIZE> m_error_buffer;
+
+ TransferStatusPtr m_status;
+
+public:
+ Transfer(Downloader& downloader, TransferId id, const std::string& url) :
+ m_downloader(downloader),
+ m_id(id),
+ m_url(url),
+ m_handle(curl_easy_init()),
+ m_error_buffer(),
+ m_status(new TransferStatus(id))
+ {
+ if (!m_handle)
+ {
+ throw std::runtime_error("curl_easy_init() failed");
+ }
+ else
+ {
+ curl_easy_setopt(m_handle, CURLOPT_URL, url.c_str());
+ curl_easy_setopt(m_handle, CURLOPT_USERAGENT, "SuperTux/" PACKAGE_VERSION " libcURL");
+
+ curl_easy_setopt(m_handle, CURLOPT_WRITEDATA, this);
+ curl_easy_setopt(m_handle, CURLOPT_WRITEFUNCTION,
+ [](void* ptr, size_t size, size_t nmemb, void* userdata) -> size_t
+ {
+ return static_cast<Transfer*>(userdata)
+ ->on_data(ptr, size, nmemb);
+ });
+
+ curl_easy_setopt(m_handle, CURLOPT_ERRORBUFFER, m_error_buffer.data());
+ curl_easy_setopt(m_handle, CURLOPT_NOSIGNAL, 1);
+ curl_easy_setopt(m_handle, CURLOPT_FAILONERROR, 1);
+ curl_easy_setopt(m_handle, CURLOPT_FOLLOWLOCATION, 1);
+
+ curl_easy_setopt(m_handle, CURLOPT_NOPROGRESS, 0);
+ curl_easy_setopt(m_handle, CURLOPT_XFERINFODATA, this);
+ curl_easy_setopt(m_handle, CURLOPT_XFERINFOFUNCTION,
+ [](void* userdata,
+ curl_off_t dltotal, curl_off_t dlnow,
+ curl_off_t ultotal, curl_off_t ulnow)
+ {
+ return static_cast<Transfer*>(userdata)
+ ->on_progress(dltotal, dlnow,
+ ultotal, ulnow);
+ });
+ }
+ }
+
+ ~Transfer()
+ {
+ curl_easy_cleanup(m_handle);
+ }
+
+ TransferStatusPtr get_status() const
+ {
+ return m_status;
+ }
+
+ TransferId get_id() const
+ {
+ return m_id;
+ }
+
+ CURL* get_curl_handle() const
+ {
+ return m_handle;
+ }
+
+ std::string get_url() const
+ {
+ return m_url;
+ }
+
+ size_t on_data(void* ptr, size_t size, size_t nmemb)
+ {
+ return size * nmemb;;
+ }
+
+ void on_progress(curl_off_t dltotal, curl_off_t dlnow,
+ curl_off_t ultotal, curl_off_t ulnow)
+ {
+ m_status->dltotal = dltotal;
+ m_status->dlnow = dlnow;
+
+ m_status->ultotal = ultotal;
+ m_status->ulnow = ulnow;
+ }
+
+private:
+ Transfer(const Transfer&) = delete;
+ Transfer& operator=(const Transfer&) = delete;
+};
+
+Downloader::Downloader() :
+ m_multi_handle(),
+ m_transfers(),
+ m_next_transfer_id(1)
{
curl_global_init(CURL_GLOBAL_ALL);
+ m_multi_handle = curl_multi_init();
+ if (!m_multi_handle)
+ {
+ throw std::runtime_error("curl_multi_init() failed");
+ }
}
Downloader::~Downloader()
{
+ for(auto& transfer : m_transfers)
+ {
+ curl_multi_remove_handle(m_multi_handle, transfer->get_curl_handle());
+ }
+ m_transfers.clear();
+
+ curl_multi_cleanup(m_multi_handle);
curl_global_cleanup();
}
size_t (*write_func)(void* ptr, size_t size, size_t nmemb, void* userdata),
void* userdata)
{
+ log_info << "Downloading " << url << std::endl;
+
char error_buffer[CURL_ERROR_SIZE+1];
CURL* curl_handle = curl_easy_init();
download(url, my_curl_physfs_write, fout.get());
}
+void
+Downloader::abort(TransferId id)
+{
+ auto it = std::find_if(m_transfers.begin(), m_transfers.end(),
+ [&id](const std::unique_ptr<Transfer>& rhs)
+ {
+ return id == rhs->get_id();
+ });
+ if (it == m_transfers.end())
+ {
+ log_warning << "transfer not found: " << id << std::endl;
+ }
+ else
+ {
+ curl_multi_remove_handle(m_multi_handle, (*it)->get_curl_handle());
+ m_transfers.erase(it);
+ }
+}
+
+void
+Downloader::update()
+{
+ // read data from the network
+ CURLMcode ret;
+ int running_handles;
+ while((ret = curl_multi_perform(m_multi_handle, &running_handles)) == CURLM_CALL_MULTI_PERFORM)
+ {
+ log_debug << "updating" << std::endl;
+ }
+
+ // check if any downloads got finished
+ int msgs_in_queue;
+ CURLMsg* msg;
+ while ((msg = curl_multi_info_read(m_multi_handle, &msgs_in_queue)))
+ {
+ switch(msg->msg)
+ {
+ case CURLMSG_DONE:
+ curl_multi_remove_handle(m_multi_handle, msg->easy_handle);
+ //FIXME: finish_transfer(msg->easy_handle);
+ break;
+
+ default:
+ log_warning << "unhandled cURL message: " << msg->msg << std::endl;
+ break;
+ }
+ }
+}
+
+TransferStatusPtr
+Downloader::request_download(const std::string& url, const std::string& filename)
+{
+ std::unique_ptr<Transfer> transfer(new Transfer(*this, m_next_transfer_id++, url));
+ curl_multi_add_handle(m_multi_handle, transfer->get_curl_handle());
+ m_transfers.push_back(std::move(transfer));
+ return m_transfers.back()->get_status();
+}
+
/* EOF */