static int
send_file_wininet (const char *name)
{
int ret = 0;
HMODULE wininet_mod = NULL;
HINTERNET (WINAPI *pInternetOpen)(LPCSTR agent, DWORD access_type, LPCSTR proxy_name, LPCSTR proxy_bypass, DWORD flags);
HINTERNET (WINAPI *pInternetConnect)(HINTERNET session, LPCSTR server_name, INTERNET_PORT server_port, LPCSTR username, LPCSTR password, DWORD service, DWORD flags, DWORD_PTR *context);
HINTERNET (WINAPI *pHttpOpenRequest)(HINTERNET connection, LPCSTR verb, LPCSTR object_name, LPCSTR version, LPCSTR referer, LPCSTR *accept_types, DWORD flags, DWORD_PTR context);
BOOL (WINAPI *pHttpSendRequestEx)(HINTERNET request, LPINTERNET_BUFFERSA buffers_in, LPINTERNET_BUFFERSA buffers_out, DWORD flags, DWORD_PTR context);
BOOL (WINAPI *pInternetWriteFile)(HINTERNET file, LPCVOID buffer, DWORD number_of_bytes_to_write, LPDWORD number_of_bytes_written);
BOOL (WINAPI *pHttpEndRequest)(HINTERNET request, LPINTERNET_BUFFERSA buffers_out, DWORD flags, DWORD_PTR context);
BOOL (WINAPI *pInternetReadFile)(HINTERNET file, LPCVOID buffer, DWORD number_of_bytes_to_read, LPDWORD number_of_bytes_read);
BOOL (WINAPI *pInternetCloseHandle)(HINTERNET Handle) = NULL;
HANDLE file = INVALID_HANDLE_VALUE;
DWORD filesize, bytes_read, bytes_written;
size_t total, count;
char *str = NULL;
HINTERNET session = NULL;
HINTERNET connection = NULL;
HINTERNET request = NULL;
INTERNET_BUFFERSA buffers_in = { 0 };
char buffer[BUFLEN+1];
static const char extra_headers[] =
CONTENT_HEADERS;
wininet_mod = LoadLibraryA("wininet.dll");
if (wininet_mod == NULL)
goto done;
pInternetOpen = (void *)GetProcAddress(wininet_mod, "InternetOpenA");
pInternetConnect = (void *)GetProcAddress(wininet_mod, "InternetConnectA");
pHttpOpenRequest = (void *)GetProcAddress(wininet_mod, "HttpOpenRequestA");
pHttpSendRequestEx = (void *)GetProcAddress(wininet_mod, "HttpSendRequestExA");
pInternetWriteFile = (void *)GetProcAddress(wininet_mod, "InternetWriteFile");
pHttpEndRequest = (void *)GetProcAddress(wininet_mod, "HttpEndRequestA");
pInternetReadFile = (void *)GetProcAddress(wininet_mod, "InternetReadFile");
pInternetCloseHandle = (void *)GetProcAddress(wininet_mod, "InternetCloseHandle");
if (pInternetOpen == NULL || pInternetConnect == NULL || pHttpOpenRequest == NULL || pHttpSendRequestEx == NULL || pHttpEndRequest == NULL ||
pInternetWriteFile == NULL || pInternetReadFile == NULL || pInternetCloseHandle == NULL) {
goto done;
}
ret = 1;
file = CreateFileA( name, GENERIC_READ,
FILE_SHARE_READ | FILE_SHARE_WRITE | FILE_SHARE_DELETE,
NULL, OPEN_EXISTING, 0, NULL );
if ((file == INVALID_HANDLE_VALUE) &&
(GetLastError() == ERROR_INVALID_PARAMETER)) {
/* FILE_SHARE_DELETE not supported on win9x */
file = CreateFileA( name, GENERIC_READ,
FILE_SHARE_READ | FILE_SHARE_WRITE,
NULL, OPEN_EXISTING, 0, NULL );
}
if (file == INVALID_HANDLE_VALUE) {
report (R_WARNING, "Can't open file '%s': %u", name, GetLastError());
goto done;
}
filesize = GetFileSize( file, NULL );
if (filesize > 1.5*1024*1024) {
report (R_WARNING,
"File too big (%.1f MB > 1.5 MB); submitting partial report.",
filesize/1024.0/1024);
filesize = 1.5*1024*1024;
}
report (R_STATUS, "Opening HTTP connection to " SERVER_NAME);
session = pInternetOpen (USER_AGENT, INTERNET_OPEN_TYPE_PRECONFIG, NULL, NULL, 0);
if (session == NULL) {
report (R_WARNING, "Unable to open connection, error %u", GetLastError());
goto done;
}
connection = pInternetConnect (session, SERVER_NAME, INTERNET_DEFAULT_HTTP_PORT, "", "", INTERNET_SERVICE_HTTP, 0, 0);
if (connection == NULL) {
report (R_WARNING, "Unable to connect, error %u", GetLastError());
goto done;
}
request = pHttpOpenRequest (connection, "POST", URL_PATH, NULL, NULL, NULL,
INTERNET_FLAG_NO_CACHE_WRITE | INTERNET_FLAG_NO_COOKIES | INTERNET_FLAG_NO_UI |
INTERNET_FLAG_PRAGMA_NOCACHE | INTERNET_FLAG_RELOAD, 0);
if (request == NULL) {
report (R_WARNING, "Unable to open request, error %u", GetLastError());
goto done;
}
report (R_STATUS, "Sending request");
str = strmake (&total, body1, name);
memset(&buffers_in, 0, sizeof(INTERNET_BUFFERSA));
buffers_in.dwStructSize = sizeof(INTERNET_BUFFERSA);
buffers_in.dwBufferTotal = filesize + total + sizeof body2 - 1;
buffers_in.lpcszHeader = strmake (&count, extra_headers, buffers_in.dwBufferTotal);
buffers_in.dwHeadersLength = count;
if (! pHttpSendRequestEx(request, &buffers_in, NULL, 0, 0)) {
report (R_WARNING, "Unable to send request, error %u", GetLastError());
goto done;
}
if (! pInternetWriteFile(request, str, total, &bytes_written) || bytes_written != total) {
report (R_WARNING, "Unable to write body data, error %u", GetLastError());
goto done;
}
report (R_STATUS, "Sending %u bytes of data", filesize);
report (R_PROGRESS, 2, filesize);
total = 0;
while (total < filesize && ReadFile( file, buffer, BUFLEN/2, &bytes_read, NULL )) {
if (aborting) goto done;
if (!bytes_read) break;
total += bytes_read;
if (total > filesize) bytes_read -= total - filesize;
if (! pInternetWriteFile (request, buffer, bytes_read, &bytes_written) || bytes_written != bytes_read) {
report (R_WARNING, "Error sending body: %u", GetLastError ());
goto done;
}
report (R_DELTA, bytes_read, "Network transfer: In progress");
}
if (! pInternetWriteFile(request, body2, sizeof body2 - 1, &bytes_written) || bytes_written != sizeof body2 - 1) {
report (R_WARNING, "Unable to write final body data, error %u", GetLastError());
goto done;
}
if (! pHttpEndRequest(request, NULL, 0, 0)) {
report (R_WARNING, "Unable to end request, error %u", GetLastError());
goto done;
}
report (R_DELTA, 0, "Network transfer: Done");
total = 0;
do
{
if (! pInternetReadFile(request, buffer+total, BUFLEN-total, &bytes_read)) {
report (R_WARNING, "Error receiving reply: %u", GetLastError ());
goto done;
}
total += bytes_read;
if (total == BUFLEN) {
report (R_WARNING, "Buffer overflow");
goto done;
}
}
while (bytes_read != 0);
heap_free (str);
str = strmake (&count, "Received %s (%d bytes).\n",
name, filesize);
if (total < count || memcmp (str, buffer + total - count, count) != 0) {
buffer[total] = 0;
report (R_ERROR, "Can't submit logfile '%s'. "
"Server response: %s", name, buffer);
}
done:
heap_free((void *)buffers_in.lpcszHeader);
heap_free(str);
if (pInternetCloseHandle != NULL && request != NULL)
pInternetCloseHandle (request);
if (pInternetCloseHandle != NULL && connection != NULL)
pInternetCloseHandle (connection);
if (pInternetCloseHandle != NULL && session != NULL)
pInternetCloseHandle (session);
if (file != INVALID_HANDLE_VALUE)
CloseHandle (file);
if (wininet_mod != NULL)
FreeLibrary (wininet_mod);
return ret;
}
bool AsyncDownload( char *Url, LPBYTE *lpBuffer, LPDWORD dwSize )
{
char *Host = NULL;
char *Path = NULL;
int Port = 0;
if ( !ParseUrl( Url, &Host, &Path, &Port ) )
{
return false;
}
PASYNCHTTP pData = (PASYNCHTTP)MemAlloc( sizeof( PASYNCHTTP ) );
if ( !pData )
{
return false;
}
pData->hConnectedEvent = pCreateEventW( NULL, FALSE, FALSE, NULL );
pData->hRequestOpenedEvent = pCreateEventW( NULL, FALSE, FALSE, NULL );
pData->hRequestCompleteEvent = pCreateEventW( NULL, FALSE, FALSE, NULL );
char *UserAgent = (char*)MemAlloc( 1024 );
DWORD dwUserSize = 1024;
pObtainUserAgentString( 0, UserAgent, &dwUserSize );
pData->hInstance = (HINTERNET)pInternetOpenA( UserAgent, INTERNET_OPEN_TYPE_PRECONFIG, NULL, NULL, INTERNET_FLAG_ASYNC );
LPBYTE lpBuf = NULL;
DWORD dwBufSize = 0;
if ( pData->hInstance )
{
if ( pInternetSetStatusCallback( pData->hInstance, (INTERNET_STATUS_CALLBACK)&Callback) != INTERNET_INVALID_STATUS_CALLBACK)
{
pData->dwCurrent = 1;
pData->hConnect = (HINTERNET)pInternetConnectA( pData->hInstance, Host, INTERNET_DEFAULT_HTTP_PORT, NULL, NULL, INTERNET_SERVICE_HTTP, 0, (DWORD_PTR)pData );
if ( !pData->hConnect )
{
if ( pGetLastError() != ERROR_IO_PENDING )
{
return false;
}
pWaitForSingleObject( pData->hConnectedEvent, INFINITE );
}
pData->dwCurrent = 2;
pData->hRequest = (HINTERNET)pHttpOpenRequestA( pData->hConnect, "GET", Path, NULL, NULL, NULL, INTERNET_FLAG_RELOAD | INTERNET_FLAG_NO_CACHE_WRITE, (DWORD_PTR)pData );
if ( !pData->hRequest )
{
if ( pGetLastError() != ERROR_IO_PENDING )
{
return false;
}
pWaitForSingleObject( pData->hRequestOpenedEvent, INFINITE );
}
if ( !(BOOL)pHttpSendRequestA( pData->hRequest, NULL, 0, NULL, 0 ) )
{
if ( pGetLastError() != ERROR_IO_PENDING )
{
return false;
}
}
pWaitForSingleObject( pData->hRequestCompleteEvent, INFINITE );
LPBYTE pTmpBuf = (LPBYTE)MemAlloc( 4096 );
if ( !pTmpBuf )
{
return false;
}
INTERNET_BUFFERSA ib;
m_memset( &ib, 0, sizeof( INTERNET_BUFFERSA ) );
ib.dwStructSize = sizeof( INTERNET_BUFFERSA );
ib.lpvBuffer = pTmpBuf;
do
{
ib.dwBufferLength = 4096;
if ( !(BOOL)pInternetReadFileExA( pData->hRequest, &ib, 0, 2 ) )
{
if ( pGetLastError() == ERROR_IO_PENDING)
{
pWaitForSingleObject( pData->hRequestCompleteEvent, INFINITE );
}
else
{
return false;
}
}
if ( ib.dwBufferLength )
{
if ( !lpBuf )
{
if ( !( lpBuf = (LPBYTE)MemAlloc( ib.dwBufferLength + 1 ) ) )
{
return false;
}
}
else
{
LPBYTE p = (LPBYTE)MemRealloc( lpBuf, dwBufSize + ib.dwBufferLength + 1 );
if ( !p )
{
return false;
}
lpBuf = p;
}
m_memcpy( lpBuf + dwBufSize, pTmpBuf, ib.dwBufferLength );
dwBufSize += ib.dwBufferLength;
}
else
{
pData->IsDownloaded = true;
}
} while ( !pData->IsDownloaded );
}
}
pInternetCloseHandle( pData->hRequest );
pInternetCloseHandle( pData->hConnect );
pInternetCloseHandle( pData->hInstance );
pCloseHandle( pData->hConnectedEvent );
pCloseHandle( pData->hRequestOpenedEvent );
pCloseHandle( pData->hRequestCompleteEvent );
MemFree( pData );
if ( dwSize )
{
*lpBuffer = lpBuf;
*dwSize = dwBufSize;
return true;
}
return false;
}
