libpackage: Use libcurl directly for downloads.

- Initial step towards allowing listening for download progress.
This commit is contained in:
Rene Gollent
2013-09-29 17:25:29 -04:00
parent 9549402d30
commit cd76e0903f
4 changed files with 87 additions and 15 deletions
+70 -13
View File
@@ -1,15 +1,17 @@
/*
* Copyright 2011, Haiku, Inc. All Rights Reserved.
* Copyright 2011-2013, Haiku, Inc. All Rights Reserved.
* Distributed under the terms of the MIT License.
*
* Authors:
* Oliver Tappe <[email protected]>
* Rene Gollent <[email protected]>
*/
#include <package/FetchFileJob.h>
#include <stdlib.h>
#include <stdio.h>
#include <curl/curl.h>
#include <sys/wait.h>
#include <Path.h>
@@ -25,7 +27,8 @@ FetchFileJob::FetchFileJob(const BContext& context, const BString& title,
:
inherited(context, title),
fFileURL(fileURL),
fTargetEntry(targetEntry)
fTargetEntry(targetEntry),
fTargetFile(&targetEntry, B_CREATE_FILE | B_ERASE_FILE | B_WRITE_ONLY)
{
}
@@ -38,22 +41,76 @@ FetchFileJob::~FetchFileJob()
status_t
FetchFileJob::Execute()
{
BPath targetPath;
status_t result = fTargetEntry.GetPath(&targetPath);
status_t result = fTargetFile.InitCheck();
if (result != B_OK)
return result;
// TODO: implement for real, maybe using the "service kit"-GSOC HTTP-stuff?
BString cmd = BString("curl -# -o ") << targetPath.Path() << " "
<< fFileURL.String();
CURL* handle = curl_easy_init();
int cmdResult = system(cmd.String());
if (WIFSIGNALED(cmdResult)
&& (WTERMSIG(cmdResult) == SIGINT || WTERMSIG(cmdResult) == SIGQUIT)) {
return B_CANCELED;
if (handle == NULL)
return B_NO_MEMORY;
result = curl_easy_setopt(handle, CURLOPT_NOPROGRESS, 0);
result = curl_easy_setopt(handle, CURLOPT_PROGRESSFUNCTION,
&_ProgressCallback);
if (result != CURLE_OK)
return B_BAD_VALUE;
result = curl_easy_setopt(handle, CURLOPT_PROGRESSDATA, this);
if (result != CURLE_OK)
return B_ERROR;
result = curl_easy_setopt(handle, CURLOPT_WRITEFUNCTION,
&_WriteCallback);
if (result != CURLE_OK)
return B_ERROR;
result = curl_easy_setopt(handle, CURLOPT_WRITEDATA, this);
if (result != CURLE_OK)
return B_ERROR;
result = curl_easy_setopt(handle, CURLOPT_PROGRESSFUNCTION,
&_ProgressCallback);
if (result != CURLE_OK)
return B_ERROR;
result = curl_easy_setopt(handle, CURLOPT_URL, fFileURL.String());
if (result != CURLE_OK)
return B_ERROR;
result = curl_easy_perform(handle);
curl_easy_cleanup(handle);
if (result != CURLE_OK) {
// TODO: map more curl error codes to ours for more
// precise error reporting
return B_ERROR;
}
return cmdResult == 0 ? B_OK : B_ERROR;
/* if (WIFSIGNALED(cmdResult)
&& (WTERMSIG(cmdResult) == SIGINT || WTERMSIG(cmdResult) == SIGQUIT)) {
return B_CANCELED;
} */
return B_OK;
}
int
FetchFileJob::_ProgressCallback(void *clientp, double dltotal, double dlnow,
double ultotal, double ulnow)
{
return 0;
}
size_t
FetchFileJob::_WriteCallback(void *buffer, size_t size, size_t nmemb,
void *userp)
{
FetchFileJob* job = reinterpret_cast<FetchFileJob*>(userp);
ssize_t dataWritten = job->fTargetFile.Write(buffer, size * nmemb);
return size_t(dataWritten);
}