X512
b3de8bea44
Other request types exists such as BFileRequest. Fixes #15675. Change-Id: Ib2e07fad4dd9f682d2b9fc0cdbf0ca60ecd3adfb Reviewed-on: https://review.haiku-os.org/c/haiku/+/2200 Reviewed-by: Adrien Destugues <[email protected]> Reviewed-by: François Revol <[email protected]>
267 lines
4.4 KiB
C++
267 lines
4.4 KiB
C++
/*
|
|
* Copyright 2011-2015, Haiku, Inc. All Rights Reserved.
|
|
* Distributed under the terms of the MIT License.
|
|
*
|
|
* Authors:
|
|
* Axel Dörfler <[email protected]>
|
|
* Rene Gollent <[email protected]>
|
|
* Oliver Tappe <[email protected]>
|
|
*/
|
|
|
|
|
|
#include "FetchFileJob.h"
|
|
|
|
#include <stdio.h>
|
|
#include <sys/wait.h>
|
|
|
|
#include <Path.h>
|
|
|
|
#ifdef HAIKU_TARGET_PLATFORM_HAIKU
|
|
# include <HttpRequest.h>
|
|
# include <UrlRequest.h>
|
|
# include <UrlProtocolRoster.h>
|
|
#endif
|
|
|
|
|
|
namespace BPackageKit {
|
|
|
|
namespace BPrivate {
|
|
|
|
|
|
#ifdef HAIKU_TARGET_PLATFORM_HAIKU
|
|
|
|
FetchFileJob::FetchFileJob(const BContext& context, const BString& title,
|
|
const BString& fileURL, const BEntry& targetEntry)
|
|
:
|
|
inherited(context, title),
|
|
fFileURL(fileURL),
|
|
fTargetEntry(targetEntry),
|
|
fTargetFile(&targetEntry, B_CREATE_FILE | B_ERASE_FILE | B_WRITE_ONLY),
|
|
fError(B_ERROR),
|
|
fDownloadProgress(0.0)
|
|
{
|
|
}
|
|
|
|
|
|
FetchFileJob::~FetchFileJob()
|
|
{
|
|
}
|
|
|
|
|
|
float
|
|
FetchFileJob::DownloadProgress() const
|
|
{
|
|
return fDownloadProgress;
|
|
}
|
|
|
|
|
|
const char*
|
|
FetchFileJob::DownloadURL() const
|
|
{
|
|
return fFileURL.String();
|
|
}
|
|
|
|
|
|
const char*
|
|
FetchFileJob::DownloadFileName() const
|
|
{
|
|
return fTargetEntry.Name();
|
|
}
|
|
|
|
|
|
off_t
|
|
FetchFileJob::DownloadBytes() const
|
|
{
|
|
return fBytes;
|
|
}
|
|
|
|
|
|
off_t
|
|
FetchFileJob::DownloadTotalBytes() const
|
|
{
|
|
return fTotalBytes;
|
|
}
|
|
|
|
|
|
status_t
|
|
FetchFileJob::Execute()
|
|
{
|
|
status_t result = fTargetFile.InitCheck();
|
|
if (result != B_OK)
|
|
return result;
|
|
|
|
BUrlRequest* request = BUrlProtocolRoster::MakeRequest(fFileURL.String(),
|
|
this);
|
|
if (request == NULL)
|
|
return B_BAD_VALUE;
|
|
|
|
thread_id thread = request->Run();
|
|
wait_for_thread(thread, NULL);
|
|
|
|
return fError;
|
|
}
|
|
|
|
|
|
void
|
|
FetchFileJob::DataReceived(BUrlRequest*, const char* data, off_t position,
|
|
ssize_t size)
|
|
{
|
|
fTargetFile.WriteAt(position, data, size);
|
|
}
|
|
|
|
|
|
void
|
|
FetchFileJob::DownloadProgress(BUrlRequest*, ssize_t bytesReceived,
|
|
ssize_t bytesTotal)
|
|
{
|
|
if (bytesTotal != 0) {
|
|
fBytes = bytesReceived;
|
|
fTotalBytes = bytesTotal;
|
|
fDownloadProgress = (float)bytesReceived/bytesTotal;
|
|
NotifyStateListeners();
|
|
}
|
|
}
|
|
|
|
|
|
void
|
|
FetchFileJob::RequestCompleted(BUrlRequest* request, bool success)
|
|
{
|
|
fError = request->Status();
|
|
|
|
if (success) {
|
|
const BHttpResult* httpResult = dynamic_cast<const BHttpResult*>
|
|
(&request->Result());
|
|
if (httpResult != NULL) {
|
|
uint16 code = httpResult->StatusCode();
|
|
uint16 codeClass = BHttpRequest::StatusCodeClass(code);
|
|
|
|
switch (codeClass) {
|
|
case B_HTTP_STATUS_CLASS_CLIENT_ERROR:
|
|
case B_HTTP_STATUS_CLASS_SERVER_ERROR:
|
|
fError = B_IO_ERROR;
|
|
break;
|
|
default:
|
|
fError = B_OK;
|
|
break;
|
|
}
|
|
switch (code) {
|
|
case B_HTTP_STATUS_OK:
|
|
fError = B_OK;
|
|
break;
|
|
case B_HTTP_STATUS_PARTIAL_CONTENT:
|
|
fError = B_PARTIAL_READ;
|
|
break;
|
|
case B_HTTP_STATUS_REQUEST_TIMEOUT:
|
|
case B_HTTP_STATUS_GATEWAY_TIMEOUT:
|
|
fError = B_DEV_TIMEOUT;
|
|
break;
|
|
case B_HTTP_STATUS_NOT_IMPLEMENTED:
|
|
case B_HTTP_STATUS_UNSUPPORTED_MEDIA_TYPE:
|
|
fError = B_NOT_SUPPORTED;
|
|
break;
|
|
case B_HTTP_STATUS_UNAUTHORIZED:
|
|
fError = B_PERMISSION_DENIED;
|
|
break;
|
|
case B_HTTP_STATUS_FORBIDDEN:
|
|
case B_HTTP_STATUS_METHOD_NOT_ALLOWED:
|
|
case B_HTTP_STATUS_NOT_ACCEPTABLE:
|
|
fError = B_NOT_ALLOWED;
|
|
break;
|
|
case B_HTTP_STATUS_NOT_FOUND:
|
|
fError = B_NAME_NOT_FOUND;
|
|
break;
|
|
case B_HTTP_STATUS_BAD_GATEWAY:
|
|
fError = B_BAD_DATA;
|
|
break;
|
|
default:
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
void
|
|
FetchFileJob::Cleanup(status_t jobResult)
|
|
{
|
|
if (jobResult != B_OK)
|
|
fTargetEntry.Remove();
|
|
}
|
|
|
|
|
|
#else // HAIKU_TARGET_PLATFORM_HAIKU
|
|
|
|
|
|
FetchFileJob::FetchFileJob(const BContext& context, const BString& title,
|
|
const BString& fileURL, const BEntry& targetEntry)
|
|
:
|
|
inherited(context, title),
|
|
fFileURL(fileURL),
|
|
fTargetEntry(targetEntry),
|
|
fTargetFile(&targetEntry, B_CREATE_FILE | B_ERASE_FILE | B_WRITE_ONLY),
|
|
fDownloadProgress(0.0)
|
|
{
|
|
}
|
|
|
|
|
|
FetchFileJob::~FetchFileJob()
|
|
{
|
|
}
|
|
|
|
|
|
float
|
|
FetchFileJob::DownloadProgress() const
|
|
{
|
|
return fDownloadProgress;
|
|
}
|
|
|
|
|
|
const char*
|
|
FetchFileJob::DownloadURL() const
|
|
{
|
|
return fFileURL.String();
|
|
}
|
|
|
|
|
|
const char*
|
|
FetchFileJob::DownloadFileName() const
|
|
{
|
|
return fTargetEntry.Name();
|
|
}
|
|
|
|
|
|
off_t
|
|
FetchFileJob::DownloadBytes() const
|
|
{
|
|
return fBytes;
|
|
}
|
|
|
|
|
|
off_t
|
|
FetchFileJob::DownloadTotalBytes() const
|
|
{
|
|
return fTotalBytes;
|
|
}
|
|
|
|
|
|
status_t
|
|
FetchFileJob::Execute()
|
|
{
|
|
return B_UNSUPPORTED;
|
|
}
|
|
|
|
|
|
void
|
|
FetchFileJob::Cleanup(status_t jobResult)
|
|
{
|
|
if (jobResult != B_OK)
|
|
fTargetEntry.Remove();
|
|
}
|
|
|
|
|
|
#endif // HAIKU_TARGET_PLATFORM_HAIKU
|
|
|
|
} // namespace BPrivate
|
|
|
|
} // namespace BPackageKit
|