diff options
Diffstat (limited to 'Src/Plugins/Library/ml_wire/DownloadThread.cpp')
| -rw-r--r-- | Src/Plugins/Library/ml_wire/DownloadThread.cpp | 222 |
1 files changed, 222 insertions, 0 deletions
diff --git a/Src/Plugins/Library/ml_wire/DownloadThread.cpp b/Src/Plugins/Library/ml_wire/DownloadThread.cpp new file mode 100644 index 00000000..67e1aed7 --- /dev/null +++ b/Src/Plugins/Library/ml_wire/DownloadThread.cpp @@ -0,0 +1,222 @@ +#include "Main.h" + +#pragma warning(disable:4786) + +#include "DownloadThread.h" +#include "api__ml_wire.h" +#include "api/service/waServiceFactory.h" +#include "../../..\Components\wac_network\wac_network_http_receiver_api.h" +#include "errors.h" +#include <strsafe.h> + +extern int winampVersion; + +#define USER_AGENT_SIZE (10 /*User-Agent*/ + 2 /*: */ + 6 /*Winamp*/ + 1 /*/*/ + 1 /*5*/ + 3/*.21*/ + 1 /*Null*/) +void SetUserAgent( api_httpreceiver *http ) +{ + char user_agent[ USER_AGENT_SIZE ] = { 0 }; + int bigVer = ( ( winampVersion & 0x0000FF00 ) >> 12 ); + int smallVer = ( ( winampVersion & 0x000000FF ) ); + StringCchPrintfA( user_agent, USER_AGENT_SIZE, "User-Agent: Winamp/%01x.%02x", bigVer, smallVer ); + http->addheader( user_agent ); +} + +#define HTTP_BUFFER_SIZE 32768 + +static int FeedXMLHTTP( api_httpreceiver *http, obj_xml *parser, bool *noData ) +{ + char downloadedData[ HTTP_BUFFER_SIZE ] = { 0 }; + int xmlResult = API_XML_SUCCESS; + int downloadSize = http->get_bytes( downloadedData, HTTP_BUFFER_SIZE ); + if ( downloadSize ) + { + xmlResult = parser->xmlreader_feed( (void *)downloadedData, downloadSize ); + *noData = false; + } + else + *noData = true; + + return xmlResult; +} + + +DownloadThread::DownloadThread() : parser( 0 ), parserFactory( 0 ) +{ + parserFactory = plugin.service->service_getServiceByGuid( obj_xmlGUID ); + if ( parserFactory ) + parser = (obj_xml *)parserFactory->getInterface(); + + if ( parser ) + { + parser->xmlreader_setCaseSensitive(); + parser->xmlreader_registerCallback( L"*", &xmlDOM ); + parser->xmlreader_open(); + } +} + +DownloadThread::~DownloadThread() +{ + if ( parser ) + { + parser->xmlreader_unregisterCallback( &xmlDOM ); + parser->xmlreader_close(); + } + + if ( parserFactory && parser ) + parserFactory->releaseInterface( parser ); + + parserFactory = 0; + parser = 0; +} + +void URLToFileName( wchar_t *url ) +{ + while ( url && *url != 0 ) + { + switch ( *url ) + { + case ':': + case '/': + case '\\': + case '*': + case '?': + case '"': + case '<': + case '>': + case '|': + *url = '_'; + } + url++; + } +} + +#define FILE_BUFFER_SIZE 32768 +void DownloadThread::DownloadFile( const wchar_t *fileName ) +{ + if ( !parser ) + return; // no sense in continuing if there's no parser available + + HANDLE file = CreateFile( fileName, GENERIC_READ, FILE_SHARE_READ, NULL, OPEN_EXISTING, FILE_FLAG_SEQUENTIAL_SCAN, NULL ); + if ( file == INVALID_HANDLE_VALUE ) + return; + + while ( true ) + { + char data[ FILE_BUFFER_SIZE ] = { 0 }; + DWORD bytesRead = 0; + if ( ReadFile( file, data, FILE_BUFFER_SIZE, &bytesRead, NULL ) && bytesRead ) + { + parser->xmlreader_feed( (void *)data, bytesRead ); + } + else + break; + } + + CloseHandle( file ); + parser->xmlreader_feed( 0, 0 ); + ReadNodes( fileName ); +} + +#ifdef _DEBUG +#include <iostream> +void ShowAllHeaders( api_httpreceiver *http ) +{ + std::cout << "--------------------" << std::endl; + const char *blah = http->getallheaders(); + while ( blah && *blah ) + { + std::cout << blah << std::endl; + blah += strlen( blah ) + 1; + } +} +#else +#define ShowAllHeaders(x) +#endif + +int RunXMLDownload( api_httpreceiver *http, obj_xml *parser ) +{ + int ret; + bool noData; + do + { + Sleep( 50 ); + ret = http->run(); + if ( FeedXMLHTTP( http, parser, &noData ) != API_XML_SUCCESS ) + return DOWNLOAD_ERROR_PARSING_XML; + } while ( ret == HTTPRECEIVER_RUN_OK ); + + // finish off the data + do + { + if ( FeedXMLHTTP( http, parser, &noData ) != API_XML_SUCCESS ) + return DOWNLOAD_ERROR_PARSING_XML; + } while ( !noData ); + + parser->xmlreader_feed( 0, 0 ); + if ( ret != HTTPRECEIVER_RUN_ERROR ) + return DOWNLOAD_SUCCESS; + else + return DOWNLOAD_CONNECTIONRESET; +} + + +int DownloadThread::DownloadURL( const wchar_t *url ) +{ + if ( !parser ) + return DOWNLOAD_NOPARSER; // no sense in continuing if there's no parser available + + api_httpreceiver *http = 0; + waServiceFactory *sf = plugin.service->service_getServiceByGuid( httpreceiverGUID ); + if ( sf ) + http = (api_httpreceiver *)sf->getInterface(); + + if ( !http ) + return DOWNLOAD_NOHTTP; + + http->AllowCompression(); + http->open( API_DNS_AUTODNS, HTTP_BUFFER_SIZE, mediaLibrary.GetProxy() ); + + SetUserAgent( http ); + + http->connect( AutoChar( url ) ); + int ret; + + do + { + Sleep( 50 ); + ret = http->run(); + if ( ret == -1 ) // connection failed + break; + + // ---- check our reply code ---- + int status = http->get_status(); + switch ( status ) + { + case HTTPRECEIVER_STATUS_CONNECTING: + case HTTPRECEIVER_STATUS_READING_HEADERS: + break; + + case HTTPRECEIVER_STATUS_READING_CONTENT: + { + ShowAllHeaders( http ); // benski> don't cut, only enabled in debug mode + int downloadError; + downloadError = RunXMLDownload( http, parser ); + if ( downloadError == DOWNLOAD_SUCCESS ) + ReadNodes( url ); + sf->releaseInterface( http ); + return downloadError; + } + break; + case HTTPRECEIVER_STATUS_ERROR: + default: + sf->releaseInterface( http ); + return DOWNLOAD_404; + } + } while ( ret == HTTPRECEIVER_RUN_OK ); + + const char *er = http->geterrorstr(); + + sf->releaseInterface( http ); + + return DOWNLOAD_404; +}
\ No newline at end of file |