The-Powder-Toy/src/client/Download.cpp

#include <cstdlib>
#include "Download.h"
#include "DownloadManager.h"
#include "HTTP.h"

Download::Download(std::string uri_, bool keepAlive):
	http(NULL),
	keepAlive(keepAlive),
	downloadData(NULL),
	downloadSize(0),
	downloadStatus(0),
	postData(""),
	postDataBoundary(""),
	userID(""),
	userSession(""),
	downloadFinished(false),
	downloadCanceled(false),
	downloadStarted(false)
{
	uri = std::string(uri_);
	DownloadManager::Ref().AddDownload(this);
}

// called by download thread itself if download was canceled
Download::~Download()
{
	if (http && (keepAlive || downloadCanceled))
		http_async_req_close(http);
	if (downloadData)
		free(downloadData);
}

// add post data to a request
void Download::AddPostData(std::map<std::string, std::string> data)
{
	postDataBoundary = FindBoundary(data, "");
	postData = GetMultipartMessage(data, postDataBoundary);
}
void Download::AddPostData(std::pair<std::string, std::string> data)
{
	std::map<std::string, std::string> postData;
	postData.insert(data);
	AddPostData(postData);
}

// add userID and sessionID headers to the download. Must be done after download starts for some reason
void Download::AuthHeaders(std::string ID, std::string session)
{
	if (ID != "0")
		userID = ID;
	userSession = session;
}

// start the download thread
void Download::Start()
{
	if (CheckStarted() || CheckDone())
		return;
	http = http_async_req_start(http, uri.c_str(), postData.c_str(), postData.length(), keepAlive ? 1 : 0);
	// add the necessary headers
	if (userID.length() || userSession.length())
		http_auth_headers(http, userID.c_str(), NULL, userSession.c_str());
	if (postDataBoundary.length())
		http_add_multipart_header(http, postDataBoundary);
	DownloadManager::Ref().Lock();
	downloadStarted = true;
	DownloadManager::Ref().Unlock();
}

// for persistent connections (keepAlive = true), reuse the open connection to make another request
bool Download::Reuse(std::string newuri)
{
	if (!keepAlive || !CheckDone() || CheckCanceled())
	{
		return false;
	}
	uri = std::string(newuri);
	DownloadManager::Ref().Lock();
	downloadFinished = false;
	DownloadManager::Ref().Unlock();
	Start();
	DownloadManager::Ref().EnsureRunning();
	return true;
}

// finish the download (if called before the download is done, this will block)
char* Download::Finish(int *length, int *status)
{
	if (CheckCanceled())
		return NULL; // shouldn't happen but just in case
	while (!CheckDone()); // block
	DownloadManager::Ref().Lock();
	downloadStarted = false;
	if (length)
		*length = downloadSize;
	if (status)
		*status = downloadStatus;
	char *ret = downloadData;
	downloadData = NULL;
	if (!keepAlive)
		downloadCanceled = true;
	DownloadManager::Ref().Unlock();
	return ret;
}

// returns the download size and progress (if the download has the correct length headers)
void Download::CheckProgress(int *total, int *done)
{
	DownloadManager::Ref().Lock();
	if (!downloadFinished && http)
		http_async_get_length(http, total, done);
	else
		*total = *done = 0;
	DownloadManager::Ref().Unlock();
}

// returns true if the download has finished
bool Download::CheckDone()
{
	DownloadManager::Ref().Lock();
	bool ret = downloadFinished;
	DownloadManager::Ref().Unlock();
	return ret;
}

// returns true if the download was canceled
bool Download::CheckCanceled()
{
	DownloadManager::Ref().Lock();
	bool ret = downloadCanceled;
	DownloadManager::Ref().Unlock();
	return ret;
}

// returns true if the download is running
bool Download::CheckStarted()
{
	DownloadManager::Ref().Lock();
	bool ret = downloadStarted;
	DownloadManager::Ref().Unlock();
	return ret;

}

// cancels the download, the download thread will delete the Download* when it finishes (do not use Download in any way after canceling)
void Download::Cancel()
{
	DownloadManager::Ref().Lock();
	downloadCanceled = true;
	DownloadManager::Ref().Unlock();
}