source-engine/engine/downloadthread.cpp
FluorescentCIAAfricanAmerican 3bf9df6b27 1
2020-04-22 12:56:21 -04:00

941 lines
28 KiB
C++

//========= Copyright Valve Corporation, All rights reserved. ============//
//
// Purpose:
//
// $NoKeywords: $
//
//=============================================================================//
//--------------------------------------------------------------------------------------------------------------
// downloadthread.cpp
//
// Implementation file for optional HTTP asset downloading thread
// Author: Matthew D. Campbell (matt@turtlerockstudios.com), 2004
//--------------------------------------------------------------------------------------------------------------
//--------------------------------------------------------------------------------------------------------------
// Includes
//--------------------------------------------------------------------------------------------------------------
#if defined( WIN32 ) && !defined( _X360 )
#include "winlite.h"
#include <WinInet.h>
#endif
#include <assert.h>
#include <sys/stat.h>
#include <stdio.h>
#include "tier0/platform.h"
#include "tier0/dbg.h"
#include "download_internal.h"
#include "tier1/strtools.h"
#include "tier0/threadtools.h"
#if defined( _X360 )
#include "xbox/xbox_win32stubs.h"
#endif
// memdbgon must be the last include file in a .cpp file!!!
#include "tier0/memdbgon.h"
//--------------------------------------------------------------------------------------------------------------
void WriteFileFromRequestContext( const RequestContext_t &rc )
{
struct stat buf;
int rt = stat(rc.absLocalPath, &buf);
if ( rt == -1 && !rc.bSuppressFileWrite )
{
FILE *fp = fopen( rc.absLocalPath, "wb" );
if ( fp )
{
if ( rc.data )
{
fwrite( rc.data, rc.nBytesTotal, 1, fp );
}
fclose( fp );
}
}
}
//--------------------------------------------------------------------------------------------------------------
#ifdef _WIN32
//--------------------------------------------------------------------------------------------------------------
/**
* Formats a string to spit out via OutputDebugString (only in debug). OutputDebugString
* is threadsafe, so this should be fine.
*
* Since I don't want to be playing with the developer cvar in the other thread, I'll use
* the presence of _DEBUG as my developer flag.
*/
void Thread_DPrintf (char *fmt, ...)
{
#ifdef _DEBUG
va_list argptr;
char msg[4096];
va_start( argptr, fmt );
Q_vsnprintf( msg, sizeof(msg), fmt, argptr );
va_end( argptr );
OutputDebugString( msg );
#endif // _DEBUG
}
//--------------------------------------------------------------------------------------------------------------
/**
* Convenience function to name status states for debugging
*/
static const char *StateString( DWORD dwStatus )
{
switch (dwStatus)
{
case INTERNET_STATUS_RESOLVING_NAME:
return "INTERNET_STATUS_RESOLVING_NAME";
case INTERNET_STATUS_NAME_RESOLVED:
return "INTERNET_STATUS_NAME_RESOLVED";
case INTERNET_STATUS_CONNECTING_TO_SERVER:
return "INTERNET_STATUS_CONNECTING_TO_SERVER";
case INTERNET_STATUS_CONNECTED_TO_SERVER:
return "INTERNET_STATUS_CONNECTED_TO_SERVER";
case INTERNET_STATUS_SENDING_REQUEST:
return "INTERNET_STATUS_SENDING_REQUEST";
case INTERNET_STATUS_REQUEST_SENT:
return "INTERNET_STATUS_REQUEST_SENT";
case INTERNET_STATUS_REQUEST_COMPLETE:
return "INTERNET_STATUS_REQUEST_COMPLETE";
case INTERNET_STATUS_CLOSING_CONNECTION:
return "INTERNET_STATUS_CLOSING_CONNECTION";
case INTERNET_STATUS_CONNECTION_CLOSED:
return "INTERNET_STATUS_CONNECTION_CLOSED";
case INTERNET_STATUS_RECEIVING_RESPONSE:
return "INTERNET_STATUS_RECEIVING_RESPONSE";
case INTERNET_STATUS_RESPONSE_RECEIVED:
return "INTERNET_STATUS_RESPONSE_RECEIVED";
case INTERNET_STATUS_HANDLE_CLOSING:
return "INTERNET_STATUS_HANDLE_CLOSING";
case INTERNET_STATUS_HANDLE_CREATED:
return "INTERNET_STATUS_HANDLE_CREATED";
case INTERNET_STATUS_INTERMEDIATE_RESPONSE:
return "INTERNET_STATUS_INTERMEDIATE_RESPONSE";
case INTERNET_STATUS_REDIRECT:
return "INTERNET_STATUS_REDIRECT";
case INTERNET_STATUS_STATE_CHANGE:
return "INTERNET_STATUS_STATE_CHANGE";
}
return "Unknown";
}
//--------------------------------------------------------------------------------------------------------------
/**
* Callback function to update status information for a download (connecting to server, etc)
*/
void __stdcall DownloadStatusCallback( HINTERNET hOpenResource, DWORD dwContext, DWORD dwStatus, LPVOID pStatusInfo, DWORD dwStatusInfoLength )
{
RequestContext_t *rc = (RequestContext_t*)pStatusInfo;
switch (dwStatus)
{
case INTERNET_STATUS_RESOLVING_NAME:
case INTERNET_STATUS_NAME_RESOLVED:
case INTERNET_STATUS_CONNECTING_TO_SERVER:
case INTERNET_STATUS_CONNECTED_TO_SERVER:
case INTERNET_STATUS_SENDING_REQUEST:
case INTERNET_STATUS_REQUEST_SENT:
case INTERNET_STATUS_REQUEST_COMPLETE:
case INTERNET_STATUS_CLOSING_CONNECTION:
case INTERNET_STATUS_CONNECTION_CLOSED:
if ( rc )
{
rc->fetchStatus = dwStatus;
}
else
{
//Thread_DPrintf( "** No RequestContext_t **\n" );
}
//Thread_DPrintf( "DownloadStatusCallback %s\n", StateString(dwStatus) );
break;
}
}
//--------------------------------------------------------------------------------------------------------------
/**
* Reads data from a handle opened by InternetOpenUrl().
*/
void ReadData( RequestContext_t& rc )
{
const int BufferSize = 2048;
unsigned char data[BufferSize];
DWORD dwSize = 0;
if ( !rc.nBytesTotal )
{
rc.status = HTTP_ERROR;
rc.error = HTTP_ERROR_ZERO_LENGTH_FILE;
return;
}
rc.nBytesCurrent = rc.nBytesCached;
rc.status = HTTP_FETCH;
while ( !rc.shouldStop )
{
// InternetReadFile() will block until there is data, or the socket gets closed. This means the
// main thread could request an abort while we're blocked here. This is okay, because the main
// thread will not wait for this thread to finish, but will clean up the RequestContext_t at some
// later point when InternetReadFile() has returned and this thread has finished.
if ( !InternetReadFile( rc.hDataResource, (LPVOID)data, BufferSize, &dwSize ) )
{
// if InternetReadFile() returns 0, there was a socket error (connection closed, etc)
rc.status = HTTP_ERROR;
rc.error = HTTP_ERROR_CONNECTION_CLOSED;
return;
}
if ( !dwSize )
{
// if InternetReadFile() succeeded, but we read 0 bytes, we're at the end of the file.
// if the file doesn't exist, write it out
WriteFileFromRequestContext( rc );
// Let the main thread know we finished reading data, and wait for it to let us exit.
rc.status = HTTP_DONE;
return;
}
else
{
// We've read some data. Make sure we won't walk off the end of our buffer, then
// use memcpy() to save the data off.
DWORD safeSize = (DWORD) min( rc.nBytesTotal - rc.nBytesCurrent, (int)dwSize );
//Thread_DPrintf( "Read %d bytes @ offset %d\n", safeSize, rc.nBytesCurrent );
if ( safeSize != dwSize )
{
//Thread_DPrintf( "Warning - read more data than expected!\n" );
}
if ( rc.data && safeSize > 0 )
{
memcpy( rc.data + rc.nBytesCurrent, data, safeSize );
}
rc.nBytesCurrent += safeSize;
}
}
// If we get here, rc.shouldStop was set early (user hit cancel).
// Let the main thread know we aborted properly.
rc.status = HTTP_ABORTED;
}
//--------------------------------------------------------------------------------------------------------------
const char *StatusString[] =
{
"HTTP_CONNECTING",
"HTTP_FETCH",
"HTTP_DONE",
"HTTP_ABORTED",
"HTTP_ERROR",
};
//--------------------------------------------------------------------------------------------------------------
const char *ErrorString[] =
{
"HTTP_ERROR_NONE",
"HTTP_ERROR_ZERO_LENGTH_FILE",
"HTTP_ERROR_CONNECTION_CLOSED",
"HTTP_ERROR_INVALID_URL",
"HTTP_ERROR_INVALID_PROTOCOL",
"HTTP_ERROR_CANT_BIND_SOCKET",
"HTTP_ERROR_CANT_CONNECT",
"HTTP_ERROR_NO_HEADERS",
"HTTP_ERROR_FILE_NONEXISTENT",
"HTTP_ERROR_MAX",
};
//--------------------------------------------------------------------------------------------------------------
/**
* Closes all open handles, and waits until the main thread has given the OK
* to quit.
*/
void CleanUpDownload( RequestContext_t& rc, HTTPStatus_t status, HTTPError_t error = HTTP_ERROR_NONE )
{
if ( status != HTTP_DONE || error != HTTP_ERROR_NONE )
{
//Thread_DPrintf( "CleanUpDownload() - http status is %s, error state is %s\n",
// StatusString[status], ErrorString[error] );
}
rc.status = status;
rc.error = error;
// Close all HINTERNET handles we have open
if ( rc.hDataResource && !InternetCloseHandle(rc.hDataResource) )
{
//Thread_DPrintf( "Failed to close data resource for %s%s\n", rc.baseURL, rc.gamePath );
}
else if ( rc.hOpenResource && !InternetCloseHandle(rc.hOpenResource) )
{
//Thread_DPrintf( "Failed to close open resource for %s%s\n", rc.baseURL, rc.gamePath );
}
rc.hDataResource = NULL;
rc.hOpenResource = NULL;
// wait until the main thread says we can go away (so it can look at rc.data).
while ( !rc.shouldStop )
{
Sleep( 100 );
}
// Delete rc.data, which was allocated in this thread
if ( rc.data != NULL )
{
delete[] rc.data;
rc.data = NULL;
}
// and tell the main thread we're exiting, so it can delete rc.cachedData and rc itself.
rc.threadDone = true;
}
//--------------------------------------------------------------------------------------------------------------
static void DumpHeaders( RequestContext_t& rc )
{
#ifdef _DEBUG
DWORD dwSize;
// First time we will find out the size of the headers.
HttpQueryInfo ( rc.hDataResource, HTTP_QUERY_RAW_HEADERS_CRLF, NULL, &dwSize, NULL );
char *lpBuffer = new char [dwSize + 2];
// Now we call HttpQueryInfo again to get the headers.
if (!HttpQueryInfo ( rc.hDataResource, HTTP_QUERY_RAW_HEADERS_CRLF, (LPVOID)lpBuffer, &dwSize, NULL))
{
return;
}
*(lpBuffer + dwSize) = '\n';
*(lpBuffer + dwSize + 1) = '\0';
Thread_DPrintf( "------------------------------\n%s%s\n%s------------------------------\n",
rc.baseURL, rc.gamePath, lpBuffer );
#endif
}
//--------------------------------------------------------------------------------------------------------------
/**
* Main download thread function - implements a (partial) synchronous HTTP download.
*/
DWORD __stdcall DownloadThread( void *voidPtr )
{
RequestContext_t& rc = *(RequestContext_t *)voidPtr;
URL_COMPONENTS url;
char urlBuf[6][BufferSize];
url.dwStructSize = sizeof(url);
url.dwSchemeLength = BufferSize;
url.dwHostNameLength = BufferSize;
url.dwUserNameLength = BufferSize;
url.dwPasswordLength = BufferSize;
url.dwUrlPathLength = BufferSize;
url.dwExtraInfoLength = BufferSize;
url.lpszScheme = urlBuf[0];
url.lpszHostName = urlBuf[1];
url.lpszUserName = urlBuf[2];
url.lpszPassword = urlBuf[3];
url.lpszUrlPath = urlBuf[4];
url.lpszExtraInfo = urlBuf[5];
char fullURL[BufferSize*2];
DWORD fullURLLength = BufferSize*2;
Q_snprintf( fullURL, fullURLLength, "%s%s", rc.baseURL, rc.urlPath );
if ( !InternetCrackUrl( fullURL, fullURLLength, 0, &url ) )
{
CleanUpDownload( rc, HTTP_ERROR, HTTP_ERROR_INVALID_URL );
return rc.status;
}
/// @TODO: Add FTP support (including restart of aborted transfers) -MDC 2004/01/08
// We should be able to handle FTP downloads as well, but I don't have a server to check against, so
// I'm gonna disallow it in case something bad would happen.
if ( url.nScheme != INTERNET_SCHEME_HTTP && url.nScheme != INTERNET_SCHEME_HTTPS )
{
CleanUpDownload( rc, HTTP_ERROR, HTTP_ERROR_INVALID_PROTOCOL );
return rc.status;
}
// Open a socket etc for the download.
// The first parameter, "Half-Life", is the User-Agent that gets sent with HTTP requests.
// INTERNET_OPEN_TYPE_PRECONFIG specifies using IE's proxy info from the registry for HTTP downloads.
rc.hOpenResource = InternetOpen( "Half-Life 2", INTERNET_OPEN_TYPE_PRECONFIG ,NULL, NULL, 0);
if ( !rc.hOpenResource )
{
CleanUpDownload( rc, HTTP_ERROR, HTTP_ERROR_CANT_BIND_SOCKET );
return rc.status;
}
InternetSetStatusCallback( rc.hOpenResource, (INTERNET_STATUS_CALLBACK)DownloadStatusCallback );
if ( rc.shouldStop )
{
CleanUpDownload( rc, HTTP_ABORTED );
return rc.status;
}
// Set up some flags
DWORD flags = 0;
flags |= INTERNET_FLAG_RELOAD; // Get from server, not IE's cache
flags |= INTERNET_FLAG_NO_CACHE_WRITE; // Don't write to IE's cache, since we're doing our own caching of partial downloads
flags |= INTERNET_FLAG_KEEP_CONNECTION; // Use keep-alive semantics. Since each file downloaded is a separate connection
// from a separate thread, I don't think this does much. Can't hurt, though.
if ( url.nScheme == INTERNET_SCHEME_HTTPS )
{
// The following flags allow us to use https:// URLs, but don't provide much in the way of authentication.
// In other words, this allows people with only access to https:// servers (?!?) to host files as transparently
// as possible.
flags |= INTERNET_FLAG_SECURE; // Use SSL, etc. Kinda need this for HTTPS URLs.
flags |= INTERNET_FLAG_IGNORE_CERT_CN_INVALID; // Don't check hostname on the SSL cert.
flags |= INTERNET_FLAG_IGNORE_CERT_DATE_INVALID; // Don't check for expired SSL certs.
}
// Request a partial if we have the data
char headers[BufferSize] = "";
DWORD headerLen = 0;
char *headerPtr = NULL;
if ( *rc.cachedTimestamp && rc.nBytesCached )
{
if ( *rc.serverURL )
{
Q_snprintf( headers, BufferSize, "If-Range: %s\nRange: bytes=%d-\nReferer: hl2://%s\n",
rc.cachedTimestamp, rc.nBytesCached, rc.serverURL );
}
else
{
Q_snprintf( headers, BufferSize, "If-Range: %s\nRange: bytes=%d-\n",
rc.cachedTimestamp, rc.nBytesCached );
}
headerPtr = headers;
headerLen = (DWORD)-1L; // the DWORD cast is because we get a signed/unsigned mismatch even with an L on the -1.
//Thread_DPrintf( "Requesting partial download\n%s", headers );
}
else if ( *rc.serverURL )
{
Q_snprintf( headers, BufferSize, "Referer: hl2://%s\n", rc.serverURL );
headerPtr = headers;
headerLen = (DWORD)-1L; // the DWORD cast is because we get a signed/unsigned mismatch even with an L on the -1.
//Thread_DPrintf( "Requesting full download\n%s", headers );
}
rc.hDataResource = InternetOpenUrl(rc.hOpenResource, fullURL, headerPtr, headerLen, flags,(DWORD)(&rc) );
// send the request off
if ( !rc.hDataResource )
{
CleanUpDownload( rc, HTTP_ERROR, HTTP_ERROR_CANT_CONNECT );
return rc.status;
}
if ( rc.shouldStop )
{
CleanUpDownload( rc, HTTP_ABORTED );
return rc.status;
}
//DumpHeaders( rc ); // debug
// check the status (are we gonna get anything?)
DWORD size = sizeof(DWORD);
DWORD code;
if ( !HttpQueryInfo( rc.hDataResource, HTTP_QUERY_STATUS_CODE | HTTP_QUERY_FLAG_NUMBER, &code, &size, NULL ) )
{
CleanUpDownload( rc, HTTP_ERROR, HTTP_ERROR_NO_HEADERS );
return rc.status;
}
// Only status codes we're looking for are HTTP_STATUS_OK (200) and HTTP_STATUS_PARTIAL_CONTENT (206)
if ( code != HTTP_STATUS_OK && code != HTTP_STATUS_PARTIAL_CONTENT )
{
CleanUpDownload( rc, HTTP_ERROR, HTTP_ERROR_FILE_NONEXISTENT );
return rc.status;
}
// get the timestamp, and save it off for future resumes, in case we abort this transfer later.
size = BufferSize;
if ( !HttpQueryInfo( rc.hDataResource, HTTP_QUERY_LAST_MODIFIED, rc.cachedTimestamp, &size, NULL ) )
{
rc.cachedTimestamp[0] = 0;
}
rc.cachedTimestamp[BufferSize-1] = 0;
// If we're not getting a partial download, don't use any cached data, even if we have some.
if ( code != HTTP_STATUS_PARTIAL_CONTENT )
{
if ( rc.nBytesCached )
{
//Thread_DPrintf( "Partial download refused - getting full version\n" );
}
rc.nBytesCached = 0; // start from the beginning
}
// Check the resource size, and allocate a buffer
size = sizeof(code);
if ( HttpQueryInfo( rc.hDataResource, HTTP_QUERY_CONTENT_LENGTH | HTTP_QUERY_FLAG_NUMBER, &code, &size, NULL ) )
{
rc.nBytesTotal = code + rc.nBytesCached;
if ( code > 0 )
{
rc.data = new unsigned char[rc.nBytesTotal + 1]; // Extra byte for NULL terminator
if ( !rc.data )
{
// We're seeing crazy large numbers being returned in code (0x48e1e22), the new fails, and we crash.
// This should probably be a different error?
CleanUpDownload( rc, HTTP_ERROR, HTTP_ERROR_ZERO_LENGTH_FILE );
return rc.status;
}
rc.data[ rc.nBytesTotal ] = 0;
}
}
else
{
CleanUpDownload( rc, HTTP_ERROR, HTTP_ERROR_ZERO_LENGTH_FILE );
return rc.status;
}
// copy cached data into buffer
if ( rc.cacheData && rc.nBytesCached )
{
int len = min( rc.nBytesCached, rc.nBytesTotal );
memcpy( rc.data, rc.cacheData, len );
}
if ( rc.shouldStop )
{
CleanUpDownload( rc, HTTP_ABORTED );
return rc.status;
}
// now download the actual data
ReadData( rc );
// and stick around until the main thread has gotten the data.
CleanUpDownload( rc, rc.status, rc.error );
return rc.status;
}
#elif defined( POSIX )
#include "curl/curl.h"
// curl callback functions
static size_t curlWriteFn( void *ptr, size_t size, size_t nmemb, void *stream)
{
RequestContext_t *pRC = (RequestContext_t *) stream;
if ( pRC->nBytesTotal && pRC->nBytesCurrent + ( size * nmemb ) <= pRC->nBytesTotal )
{
Q_memcpy( pRC->data + pRC->nBytesCurrent, ptr, ( size * nmemb ) );
pRC->nBytesCurrent += size * nmemb;
}
return size * nmemb;
}
int Q_StrTrim( char *pStr )
{
char *pSource = pStr;
char *pDest = pStr;
// skip white space at the beginning
while ( *pSource != 0 && isspace( *pSource ) )
{
pSource++;
}
// copy everything else
char *pLastWhiteBlock = NULL;
char *pStart = pDest;
while ( *pSource != 0 )
{
*pDest = *pSource++;
if ( isspace( *pDest ) )
{
if ( pLastWhiteBlock == NULL )
pLastWhiteBlock = pDest;
}
else
{
pLastWhiteBlock = NULL;
}
pDest++;
}
*pDest = 0;
// did we end in a whitespace block?
if ( pLastWhiteBlock != NULL )
{
// yep; shorten the string
pDest = pLastWhiteBlock;
*pLastWhiteBlock = 0;
}
return pDest - pStart;
}
static size_t curlHeaderFn( void *ptr, size_t size, size_t nmemb, void *stream)
{
char *pszHeader = (char*)ptr;
char *pszValue = NULL;
RequestContext_t *pRC = (RequestContext_t *) stream;
pszHeader[ ( size * nmemb - 1 ) ] = NULL;
pszValue = Q_strstr( pszHeader, ":" );
if ( pszValue )
{
// null terminate the header name, and point pszValue at it's value
*pszValue = NULL;
pszValue++;
Q_StrTrim( pszValue );
}
if ( 0 == Q_stricmp( pszHeader, "Content-Length" ) )
{
size_t len = atol( pszValue );
if ( pRC && len )
{
pRC->nBytesTotal = len;
pRC->data = (byte*)malloc( len );
}
}
return size * nmemb;
}
// we're going to abuse this by using it for proxy pac fetching
// the cacheData field will hold the URL of the PAC, and the data
// field the contents of the pac
RequestContext_t g_pacRequestCtx;
// system specific headers for proxy configuration
#if defined(OSX)
#include <CoreFoundation/CoreFoundation.h>
#include <CoreServices/CoreServices.h>
#include <SystemConfiguration/SystemConfiguration.h>
#endif
void SetProxiesForURL( CURL *hMasterCURL, const char *pszURL )
{
uint32 uProxyPort = 0;
char rgchProxyHost[1024];
char *pszProxyExceptionList = NULL;
rgchProxyHost[0] = '\0';
#if defined(OSX)
// create an urlref around the raw URL
CFURLRef url = CFURLCreateWithBytes( NULL, ( const UInt8 * ) pszURL, strlen( pszURL ), kCFStringEncodingASCII, NULL );
// copy the proxies dictionary
CFDictionaryRef proxyDict = SCDynamicStoreCopyProxies(NULL);
// and ask the system what proxies it thinks I should consider for the given URL
CFArrayRef proxies = CFNetworkCopyProxiesForURL( url, proxyDict );
CFIndex iProxy;
// walk through the returned set, looking for any types we (and lib curl) can handle
// the list is returned in "preference order", but we can only handle http, and pac urls
for( iProxy = 0; iProxy < CFArrayGetCount( proxies ); iProxy++ )
{
CFDictionaryRef proxy = (CFDictionaryRef) CFArrayGetValueAtIndex( proxies, iProxy );
if ( proxy == NULL )
break;
// what type of proxy is this one?
CFStringRef proxyType = (CFStringRef) CFDictionaryGetValue( proxy, kCFProxyTypeKey );
if ( CFEqual( proxyType, kCFProxyTypeNone ) )
{
// no proxy should be used - we're done.
break;
}
else if ( CFEqual( proxyType, kCFProxyTypeHTTP ) )
{
// manually configured HTTP proxy settings.
const void *val = NULL;
// grab the proxy port
val = CFDictionaryGetValue( proxy, kCFProxyPortNumberKey );
if ( val == NULL || !CFNumberGetValue( (CFNumberRef) val, kCFNumberIntType, &uProxyPort ) )
// either we failed, or the port was invalid
continue;
// no port specified - use the default http port
if ( uProxyPort == 0 )
uProxyPort = 80;
int cbOffset = 0;
// see if they've specified authentication (username/password)
val = CFDictionaryGetValue( proxy, kCFProxyUsernameKey );
if ( val != NULL && CFStringGetCString( (CFStringRef) val, rgchProxyHost + cbOffset, sizeof(rgchProxyHost) - cbOffset, kCFStringEncodingASCII ) && rgchProxyHost[cbOffset] != '\0' )
{
// we've got "username" in rgchProxyHost
cbOffset = Q_strlen( rgchProxyHost );
val = CFDictionaryGetValue( proxy, kCFProxyPasswordKey );
if ( val != NULL && CFStringGetLength( (CFStringRef) val ) )
{
// and there's a non-null password value - put a colon after username
rgchProxyHost[cbOffset++] = ':';
CFStringGetCString( (CFStringRef) val, rgchProxyHost + cbOffset, sizeof(rgchProxyHost) - cbOffset, kCFStringEncodingASCII );
// now we've got user:password in rgchProxyHost
cbOffset = Q_strlen( rgchProxyHost );
}
// since we've got at least a username, we need an @
rgchProxyHost[cbOffset++] = '@';
}
val = CFDictionaryGetValue( proxy, kCFProxyHostNameKey );
if ( val == NULL || !CFStringGetCString( (CFStringRef) val, rgchProxyHost + cbOffset, sizeof(rgchProxyHost) - cbOffset, kCFStringEncodingASCII ) || rgchProxyHost[cbOffset] == '\0' )
continue;
break;
}
else if ( CFEqual( proxyType, kCFProxyTypeAutoConfigurationURL ) )
{
// a proxy autoconfig URL has been provided
char rgchPacURL[1024];
// get the url (as an urlref) and turn it into a string
CFURLRef cfUrl = (CFURLRef) CFDictionaryGetValue( proxy, kCFProxyAutoConfigurationURLKey );
CFStringGetCString( (CFStringRef) CFStringCreateWithFormat( NULL, NULL, CFSTR("%@"), cfUrl ),
rgchPacURL, sizeof( rgchPacURL ), kCFStringEncodingASCII );
CURLcode res = CURLE_OK;
// see if we've not yet fetched this pac file
if ( !g_pacRequestCtx.cacheData || Q_strcmp( (const char *)g_pacRequestCtx.cacheData, rgchPacURL ) )
{
if ( g_pacRequestCtx.cacheData )
{
free( g_pacRequestCtx.cacheData );
g_pacRequestCtx.cacheData = NULL;
}
if ( g_pacRequestCtx.data )
{
free( g_pacRequestCtx.data );
g_pacRequestCtx.data = NULL;
}
// grab the data, using the same request context structure (and callbacks) we use for real downloads
CURL *hCURL = curl_easy_init();
if ( !hCURL )
{
AssertMsg( hCURL, "failed to initialize curl handle" );
break;
}
curl_easy_setopt( hCURL, CURLOPT_NOPROGRESS, 1 );
curl_easy_setopt( hCURL, CURLOPT_NOSIGNAL, 1 );
curl_easy_setopt( hCURL, CURLOPT_CONNECTTIMEOUT, 30 );
curl_easy_setopt( hCURL, CURLOPT_FOLLOWLOCATION, 1 ); // follow 30x redirections from the web server
// and setup the callback fns
curl_easy_setopt( hCURL, CURLOPT_HEADERFUNCTION, &curlHeaderFn );
curl_easy_setopt( hCURL, CURLOPT_WRITEFUNCTION, &curlWriteFn );
// setup callback stream pointers
curl_easy_setopt( hCURL, CURLOPT_WRITEHEADER, &g_pacRequestCtx );
curl_easy_setopt( hCURL, CURLOPT_WRITEDATA, &g_pacRequestCtx );
curl_easy_setopt( hCURL, CURLOPT_URL, rgchPacURL );
res = curl_easy_perform( hCURL );
curl_easy_cleanup( hCURL );
}
if ( res == CURLE_OK )
{
// copy the URL into the "pac cache", if necessary
if ( !g_pacRequestCtx.cacheData )
{
g_pacRequestCtx.cacheData = (unsigned char*) malloc( Q_strlen( rgchPacURL ) + 1 );
Q_memcpy( g_pacRequestCtx.cacheData, rgchPacURL, Q_strlen( rgchPacURL ) );
}
if ( !g_pacRequestCtx.data ) // no data in the proxy.pac they have, so just ignore it
return;
// wrap the data (the pac contents) into a cfstring
CFStringRef cfPacStr = CFStringCreateWithCString( kCFAllocatorDefault, (const char *)g_pacRequestCtx.data, kCFStringEncodingASCII );
// and ask the system, given this proxy pac, what (list of) proxies should I consider for this URL?
CFErrorRef err;
CFArrayRef proxiesForUrl = CFNetworkCopyProxiesForAutoConfigurationScript( cfPacStr, cfUrl, &err );
if ( proxiesForUrl )
{
// we're re-assigning the value that the loop is iterating over, the postincrement will fire after we do this,
// hence the -1 (rather than 0) assignment to iProxy
proxies = proxiesForUrl;
iProxy = -1;
}
continue;
}
else
{
if ( g_pacRequestCtx.cacheData )
{
free( g_pacRequestCtx.cacheData );
g_pacRequestCtx.cacheData = NULL;
}
}
}
else
{
Msg( "unsupported proxy type\n" );
break;
}
}
#else
#warning "CHTTPDownloadThread doesn't know how to set proxy config"
#endif
if ( rgchProxyHost[0] == '\0' || uProxyPort <= 0 )
{
if ( pszProxyExceptionList )
free( pszProxyExceptionList );
return;
}
curl_easy_setopt( hMasterCURL, CURLOPT_PROXY, rgchProxyHost );
curl_easy_setopt( hMasterCURL, CURLOPT_PROXYPORT, uProxyPort );
if ( pszProxyExceptionList )
{
curl_easy_setopt( hMasterCURL, (CURLoption) (CURLOPTTYPE_OBJECTPOINT + 177) /*CURLOPT_NOPROXY*/ , pszProxyExceptionList );
free( pszProxyExceptionList );
}
}
void DownloadThread( void *voidPtr )
{
static bool bDoneInit = false;
if ( !bDoneInit )
{
bDoneInit = true;
curl_global_init( CURL_GLOBAL_SSL );
}
RequestContext_t& rc = *(RequestContext_t *)voidPtr;
rc.status = HTTP_FETCH;
CURL *hCURL = curl_easy_init();
if ( !hCURL )
{
rc.error = HTTP_ERROR_INVALID_URL;
rc.status = HTTP_ERROR;
rc.threadDone = true;
return;
}
curl_easy_setopt( hCURL, CURLOPT_NOPROGRESS, 1 );
curl_easy_setopt( hCURL, CURLOPT_NOSIGNAL, 1 );
curl_easy_setopt( hCURL, CURLOPT_CONNECTTIMEOUT, 30 );
curl_easy_setopt( hCURL, CURLOPT_IPRESOLVE, CURL_IPRESOLVE_V4 );
// turn off certificate verification similar to how we set INTERNET_FLAG_IGNORE_CERT_CN_INVALID and INTERNET_FLAG_IGNORE_CERT_DATE_INVALID on Windows
curl_easy_setopt( hCURL, CURLOPT_SSL_VERIFYHOST, 0 );
curl_easy_setopt( hCURL, CURLOPT_SSL_VERIFYPEER, 0 );
// and now the callback fns
curl_easy_setopt( hCURL, CURLOPT_HEADERFUNCTION, &curlHeaderFn );
curl_easy_setopt( hCURL, CURLOPT_WRITEFUNCTION, &curlWriteFn );
uint32 cubURL = Q_strlen( rc.baseURL ) + Q_strlen( rc.urlPath ) + 2 /*one for the /, one for the null*/;
char *pchURL = (char *) malloc( cubURL );
Q_snprintf( pchURL, cubURL, "%s%s", rc.baseURL, rc.urlPath );
uint32 cubReferer = 0;
char *pchReferer = NULL;
if ( *rc.serverURL )
{
cubReferer = Q_strlen( rc.serverURL ) + 8;
pchReferer = (char *) malloc( cubReferer );
Q_snprintf( pchReferer, cubURL, "hl2://%s", rc.serverURL );
}
// setup proxies
SetProxiesForURL( hCURL, pchURL );
// set the url
curl_easy_setopt( hCURL, CURLOPT_URL, pchURL );
// setup callback stream pointers
curl_easy_setopt( hCURL, CURLOPT_WRITEHEADER, &rc );
curl_easy_setopt( hCURL, CURLOPT_WRITEDATA, &rc );
curl_easy_setopt( hCURL, CURLOPT_FOLLOWLOCATION, 1 );
curl_easy_setopt( hCURL, CURLOPT_MAXREDIRS, 1 );
curl_easy_setopt( hCURL, CURLOPT_UNRESTRICTED_AUTH, 1 );
curl_easy_setopt( hCURL, CURLOPT_USERAGENT, "Half-Life 2" );
if ( pchReferer )
{
curl_easy_setopt( hCURL, CURLOPT_REFERER, pchReferer );
}
// g0g0g0
CURLcode res = curl_easy_perform( hCURL );
free( pchURL );
if ( pchReferer )
{
free( pchReferer );
}
if ( res == CURLE_OK )
{
curl_easy_getinfo( hCURL , CURLINFO_RESPONSE_CODE , &rc.status );
if ( rc.status == HTTPStatus_t(200) || rc.status == HTTPStatus_t(206) )
{
// write the file before we change the status to DONE, so that the write
// will finish before the main thread goes on and starts messing with the file
WriteFileFromRequestContext( rc );
rc.status = HTTP_DONE;
rc.error = HTTP_ERROR_NONE;
}
else
{
rc.status = HTTP_ERROR;
rc.error = HTTP_ERROR_FILE_NONEXISTENT;
}
}
else
{
rc.status = HTTP_ERROR;
}
// wait until the main thread says we can go away (so it can look at rc.data).
while ( !rc.shouldStop )
{
ThreadSleep( 100 );
}
// Delete rc.data, which was allocated in this thread
if ( rc.data != NULL )
{
delete[] rc.data;
rc.data = NULL;
}
curl_easy_cleanup( hCURL );
rc.threadDone = true;
}
#else
void DownloadThread( void *voidPtr )
{
#warning "DownloadThread Not Implemented"
Assert( !"Implement me" );
}
#endif