129 lines
3.1 KiB
C++
129 lines
3.1 KiB
C++
#include "Main.h"
|
|
#include "BackgroundDownloader.h"
|
|
#include "..\..\..\Components\wac_network\wac_network_http_receiver_api.h"
|
|
#include "api__in_flv.h"
|
|
#include "api/service/waservicefactory.h"
|
|
#include "../nu/AutoChar.h"
|
|
|
|
#include <strsafe.h>
|
|
|
|
#define HTTP_BUFFER_SIZE 65536
|
|
|
|
// {C0A565DC-0CFE-405a-A27C-468B0C8A3A5C}
|
|
static const GUID internetConfigGroupGUID =
|
|
{
|
|
0xc0a565dc, 0xcfe, 0x405a, { 0xa2, 0x7c, 0x46, 0x8b, 0xc, 0x8a, 0x3a, 0x5c }
|
|
};
|
|
|
|
|
|
static void SetUserAgent( api_httpreceiver *http )
|
|
{
|
|
char agent[ 256 ] = { 0 };
|
|
StringCchPrintfA( agent, 256, "User-Agent: %S/%S", WASABI_API_APP->main_getAppName(), WASABI_API_APP->main_getVersionNumString() );
|
|
http->addheader( agent );
|
|
}
|
|
|
|
static int FeedHTTP( api_httpreceiver *http, Downloader::DownloadCallback *callback, int *downloaded )
|
|
{
|
|
char downloadedData[ HTTP_BUFFER_SIZE ] = { 0 };
|
|
int result = 0;
|
|
int downloadSize = http->get_bytes( downloadedData, HTTP_BUFFER_SIZE );
|
|
|
|
*downloaded = downloadSize;
|
|
if ( downloadSize )
|
|
{
|
|
result = callback->OnData( downloadedData, downloadSize );
|
|
}
|
|
|
|
return result;
|
|
}
|
|
|
|
static void RunDownload( api_httpreceiver *http, Downloader::DownloadCallback *callback )
|
|
{
|
|
int ret;
|
|
int downloaded = 0;
|
|
do
|
|
{
|
|
ret = http->run();
|
|
Sleep( 55 );
|
|
do
|
|
{
|
|
if ( FeedHTTP( http, callback, &downloaded ) != 0 )
|
|
return;
|
|
} while ( downloaded == HTTP_BUFFER_SIZE );
|
|
} while ( ret == HTTPRECEIVER_RUN_OK );
|
|
|
|
// finish off the data
|
|
do
|
|
{
|
|
if ( FeedHTTP( http, callback, &downloaded ) != 0 )
|
|
return;
|
|
} while ( downloaded );
|
|
}
|
|
|
|
|
|
bool Downloader::Download(const char *url, Downloader::DownloadCallback *callback, uint64_t startPosition)
|
|
{
|
|
api_httpreceiver *http = 0;
|
|
waServiceFactory *sf = plugin.service->service_getServiceByGuid(httpreceiverGUID);
|
|
if (sf) http = (api_httpreceiver *)sf->getInterface();
|
|
|
|
if (!http)
|
|
return false;
|
|
|
|
int use_proxy = 1;
|
|
bool proxy80 = AGAVE_API_CONFIG->GetBool(internetConfigGroupGUID, L"proxy80", false);
|
|
if (proxy80 && strstr(url, ":") && (!strstr(url, ":80/") && strstr(url, ":80") != (url + strlen(url) - 3)))
|
|
use_proxy = 0;
|
|
|
|
const wchar_t *proxy = use_proxy?AGAVE_API_CONFIG->GetString(internetConfigGroupGUID, L"proxy", 0):0;
|
|
|
|
http->AllowCompression();
|
|
http->open(API_DNS_AUTODNS, HTTP_BUFFER_SIZE, (proxy && proxy[0]) ? (const char *)AutoChar(proxy) : NULL);
|
|
if (startPosition > 0)
|
|
{
|
|
char temp[128] = {0};
|
|
StringCchPrintfA(temp, 128, "Range: bytes=%d-", startPosition);
|
|
http->addheader(temp);
|
|
}
|
|
SetUserAgent(http);
|
|
http->connect(url);
|
|
int ret;
|
|
|
|
do
|
|
{
|
|
Sleep(55);
|
|
ret = http->run();
|
|
if (ret == -1) // connection failed
|
|
break;
|
|
|
|
// ---- check our reply code ----
|
|
int replycode = http->getreplycode();
|
|
switch (replycode)
|
|
{
|
|
case 0:
|
|
case 100:
|
|
break;
|
|
case 200:
|
|
{
|
|
if (callback->OnConnect(http) != 0)
|
|
{
|
|
sf->releaseInterface(http);
|
|
return false;
|
|
}
|
|
|
|
RunDownload(http, callback);
|
|
sf->releaseInterface(http);
|
|
return true;
|
|
}
|
|
break;
|
|
default:
|
|
sf->releaseInterface(http);
|
|
return false;
|
|
}
|
|
}
|
|
while (ret == HTTPRECEIVER_RUN_OK);
|
|
//const char *er = http->geterrorstr();
|
|
sf->releaseInterface(http);
|
|
return false;
|
|
} |