ReceiveTempSMS
/
Sweden-Number
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity
Sweden-Number/dlls/wininet/internet.c

1130 lines
29 KiB
C
Raw Blame History

/*
* Wininet
*
* Copyright 1999 Corel Corporation
*
* Ulrich Czekalla
*
*/
#include "config.h"
#include <string.h>
#include <sys/types.h>
#ifdef HAVE_SYS_SOCKET_H
# include <sys/socket.h>
#endif
#include <stdlib.h>
#include "windows.h"
#include "wininet.h"
#include "debugtools.h"
#include "winerror.h"
#include "winsock.h"
#include "internet.h"
DEFAULT_DEBUG_CHANNEL(wininet);
#define MAX_IDLE_WORKER 1000*60*1
#define MAX_WORKER_THREADS 10
#define GET_HWININET_FROM_LPWININETFINDNEXT(lpwh) \
(LPWININETAPPINFOA)(((LPWININETFTPSESSIONA)(lpwh->hdr.lpwhparent))->hdr.lpwhparent)
typedef struct
{
DWORD dwError;
CHAR response[MAX_REPLY_LEN];
} WITHREADERROR, *LPWITHREADERROR;
INTERNET_SCHEME GetInternetScheme(LPSTR lpszScheme);
BOOL WINAPI INTERNET_FindNextFileA(HINTERNET hFind, LPVOID lpvFindData);
VOID INTERNET_ExecuteWork();
DWORD g_dwTlsErrIndex = TLS_OUT_OF_INDEXES;
DWORD dwNumThreads;
DWORD dwNumIdleThreads;
HANDLE hEventArray[2];
#define hQuitEvent hEventArray[0]
#define hWorkEvent hEventArray[1]
CRITICAL_SECTION csQueue;
LPWORKREQUEST lpHeadWorkQueue;
LPWORKREQUEST lpWorkQueueTail;
/***********************************************************************
* WININET_LibMain [Internal] Initializes the internal 'WININET.DLL'.
*
* PARAMS
* hinstDLL [I] handle to the 'dlls' instance
* fdwReason [I]
* lpvReserved [I] reserverd, must be NULL
*
* RETURNS
* Success: TRUE
* Failure: FALSE
*/
BOOL WINAPI
WININET_LibMain (HINSTANCE hinstDLL, DWORD fdwReason, LPVOID lpvReserved)
{
TRACE("%x,%lx,%p\n", hinstDLL, fdwReason, lpvReserved);
switch (fdwReason) {
case DLL_PROCESS_ATTACH:
g_dwTlsErrIndex = TlsAlloc();
if (g_dwTlsErrIndex == TLS_OUT_OF_INDEXES)
return FALSE;
hQuitEvent = CreateEventA(0, TRUE, FALSE, NULL);
hWorkEvent = CreateEventA(0, FALSE, FALSE, NULL);
InitializeCriticalSection(&csQueue);
dwNumThreads = 0;
dwNumIdleThreads = 0;
case DLL_THREAD_ATTACH:
{
LPWITHREADERROR lpwite = HeapAlloc(GetProcessHeap(), 0, sizeof(WITHREADERROR));
if (NULL == lpwite)
return FALSE;
TlsSetValue(g_dwTlsErrIndex, (LPVOID)lpwite);
}
break;
case DLL_THREAD_DETACH:
if (g_dwTlsErrIndex != TLS_OUT_OF_INDEXES)
HeapFree(GetProcessHeap(), 0, TlsGetValue(g_dwTlsErrIndex));
break;
case DLL_PROCESS_DETACH:
if (g_dwTlsErrIndex != TLS_OUT_OF_INDEXES)
{
HeapFree(GetProcessHeap(), 0, TlsGetValue(g_dwTlsErrIndex));
TlsFree(g_dwTlsErrIndex);
}
SetEvent(hQuitEvent);
CloseHandle(hQuitEvent);
CloseHandle(hWorkEvent);
DeleteCriticalSection(&csQueue);
break;
}
return TRUE;
}
/***********************************************************************
* InternetOpenA (WININET.113)
*
* Per-application initialization of wininet
*
* RETURNS
* HINTERNET on success
* NULL on failure
*
*/
INTERNETAPI HINTERNET WINAPI InternetOpenA(LPCSTR lpszAgent,
DWORD dwAccessType, LPCSTR lpszProxy,
LPCSTR lpszProxyBypass, DWORD dwFlags)
{
LPWININETAPPINFOA lpwai = NULL;
TRACE("\n");
/* Clear any error information */
INTERNET_SetLastError(0);
lpwai = HeapAlloc(GetProcessHeap(), 0, sizeof(WININETAPPINFOA));
if (NULL == lpwai)
INTERNET_SetLastError(ERROR_OUTOFMEMORY);
else
{
memset(lpwai, 0, sizeof(WININETAPPINFOA));
lpwai->hdr.htype = WH_HINIT;
lpwai->hdr.lpwhparent = NULL;
lpwai->hdr.dwFlags = dwFlags;
if (NULL != lpszAgent)
lpwai->lpszAgent = strdup(lpszAgent);
if (NULL != lpszProxy)
lpwai->lpszProxy = strdup(lpszProxy);
if (NULL != lpszProxyBypass)
lpwai->lpszProxyBypass = strdup(lpszProxyBypass);
lpwai->dwAccessType = dwAccessType;
}
return (HINTERNET)lpwai;
}
/***********************************************************************
* InternetGetLastResponseInfoA (WININET.108)
*
* Return last wininet error description on the calling thread
*
* RETURNS
* TRUE on success of writting to buffer
* FALSE on failure
*
*/
BOOL WINAPI InternetGetLastResponseInfoA(LPDWORD lpdwError,
LPSTR lpszBuffer, LPDWORD lpdwBufferLength)
{
LPWITHREADERROR lpwite = (LPWITHREADERROR)TlsGetValue(g_dwTlsErrIndex);
TRACE("\n");
*lpdwError = lpwite->dwError;
if (lpwite->dwError)
{
strncpy(lpszBuffer, lpwite->response, *lpdwBufferLength);
*lpdwBufferLength = strlen(lpszBuffer);
}
else
*lpdwBufferLength = 0;
return TRUE;
}
/***********************************************************************
* InternetGetConnectedState (WININET.103)
*
* Return connected state
*
* RETURNS
* TRUE if connected
* if lpdwStatus is not null, return the status (off line,
* modem, lan...) in it.
* FALSE if not connected
*/
BOOL WINAPI InternetGetConnectedState(LPDWORD lpdwStatus, DWORD dwReserved)
{
FIXME("Stub\n");
return FALSE;
}
/***********************************************************************
* InternetConnectA (WININET.93)
*
* Open a ftp, gopher or http session
*
* RETURNS
* HINTERNET a session handle on success
* NULL on failure
*
*/
INTERNETAPI HINTERNET WINAPI InternetConnectA(HINTERNET hInternet,
LPCSTR lpszServerName, INTERNET_PORT nServerPort,
LPCSTR lpszUserName, LPCSTR lpszPassword,
DWORD dwService, DWORD dwFlags, DWORD dwContext)
{
HINTERNET rc = (HINTERNET) NULL;
TRACE("\n");
/* Clear any error information */
INTERNET_SetLastError(0);
switch (dwService)
{
case INTERNET_SERVICE_FTP:
rc = FTP_Connect(hInternet, lpszServerName, nServerPort,
lpszUserName, lpszPassword, dwFlags, dwContext);
break;
case INTERNET_SERVICE_HTTP:
break;
case INTERNET_SERVICE_GOPHER:
default:
break;
}
return rc;
}
/***********************************************************************
* InternetFindNextFileA (WININET.102)
*
* Continues a file search from a previous call to FindFirstFile
*
* RETURNS
* TRUE on success
* FALSE on failure
*
*/
BOOL WINAPI InternetFindNextFileA(HINTERNET hFind, LPVOID lpvFindData)
{
LPWININETAPPINFOA hIC = NULL;
LPWININETFINDNEXTA lpwh = (LPWININETFINDNEXTA) hFind;
TRACE("\n");
if (NULL == lpwh || lpwh->hdr.htype != WH_HFINDNEXT)
{
INTERNET_SetLastError(ERROR_INTERNET_INCORRECT_HANDLE_TYPE);
return FALSE;
}
hIC = GET_HWININET_FROM_LPWININETFINDNEXT(lpwh);
if (hIC->hdr.dwFlags & INTERNET_FLAG_ASYNC)
{
WORKREQUEST workRequest;
workRequest.asyncall = INTERNETFINDNEXTA;
workRequest.HFTPSESSION = (DWORD)hFind;
workRequest.LPFINDFILEDATA = (DWORD)lpvFindData;
return INTERNET_AsyncCall(&workRequest);
}
else
{
return INTERNET_FindNextFileA(hFind, lpvFindData);
}
}
/***********************************************************************
* INTERNET_FindNextFileA (Internal)
*
* Continues a file search from a previous call to FindFirstFile
*
* RETURNS
* TRUE on success
* FALSE on failure
*
*/
BOOL WINAPI INTERNET_FindNextFileA(HINTERNET hFind, LPVOID lpvFindData)
{
BOOL bSuccess = TRUE;
LPWININETAPPINFOA hIC = NULL;
LPWIN32_FIND_DATAA lpFindFileData;
LPWININETFINDNEXTA lpwh = (LPWININETFINDNEXTA) hFind;
TRACE("\n");
if (NULL == lpwh || lpwh->hdr.htype != WH_HFINDNEXT)
{
INTERNET_SetLastError(ERROR_INTERNET_INCORRECT_HANDLE_TYPE);
return FALSE;
}
/* Clear any error information */
INTERNET_SetLastError(0);
if (lpwh->hdr.lpwhparent->htype != WH_HFTPSESSION)
{
FIXME("Only FTP find next supported\n");
INTERNET_SetLastError(ERROR_INTERNET_INCORRECT_HANDLE_TYPE);
return FALSE;
}
TRACE("index(%d) size(%ld)\n", lpwh->index, lpwh->size);
lpFindFileData = (LPWIN32_FIND_DATAA) lpvFindData;
ZeroMemory(lpFindFileData, sizeof(WIN32_FIND_DATAA));
if (lpwh->index >= lpwh->size)
{
INTERNET_SetLastError(ERROR_NO_MORE_FILES);
bSuccess = FALSE;
goto lend;
}
FTP_ConvertFileProp(&lpwh->lpafp[lpwh->index], lpFindFileData);
lpwh->index++;
TRACE("\nName: %s\nSize: %ld\n", lpFindFileData->cFileName, lpFindFileData->nFileSizeLow);
lend:
hIC = GET_HWININET_FROM_LPWININETFINDNEXT(lpwh);
if (hIC->hdr.dwFlags & INTERNET_FLAG_ASYNC && hIC->lpfnStatusCB)
{
INTERNET_ASYNC_RESULT iar;
iar.dwResult = (DWORD)bSuccess;
iar.dwError = bSuccess ? ERROR_SUCCESS : INTERNET_GetLastError();
hIC->lpfnStatusCB(hFind, lpwh->hdr.dwContext, INTERNET_STATUS_REQUEST_COMPLETE,
&iar, sizeof(INTERNET_ASYNC_RESULT));
}
return bSuccess;
}
/***********************************************************************
* InternetCloseHandle (WININET.89)
*
* Continues a file search from a previous call to FindFirstFile
*
* RETURNS
* TRUE on success
* FALSE on failure
*
*/
BOOL WINAPI InternetCloseHandle(HINTERNET hInternet)
{
BOOL retval = FALSE;
LPWININETHANDLEHEADER lpwh = (LPWININETHANDLEHEADER) hInternet;
TRACE("\n");
if (NULL == lpwh)
return FALSE;
/* Clear any error information */
INTERNET_SetLastError(0);
switch (lpwh->htype)
{
case WH_HINIT:
case WH_HHTTPSESSION:
case WH_HHTTPREQ:
break;
case WH_HFTPSESSION:
retval = FTP_CloseSessionHandle((LPWININETFTPSESSIONA) lpwh);
break;
case WH_HFINDNEXT:
retval = FTP_CloseFindNextHandle((LPWININETFINDNEXTA) lpwh);
break;
default:
break;
}
return retval;
}
/***********************************************************************
* InternetCrackUrlA (WININET.95)
*
* Break up URL into its components
*
* RETURNS
* TRUE on success
* FALSE on failure
*
*/
BOOL WINAPI InternetCrackUrlA(LPCSTR lpszUrl, DWORD dwUrlLength, DWORD dwFlags,
LPURL_COMPONENTSA lpUrlComponents)
{
/*
* RFC 1808
* <protocol>:[//<net_loc>][/path][;<params>][?<query>][#<fragment>]
*
*/
char* szScheme = NULL;
char* szUser = NULL;
char* szPass = NULL;
char* szHost = NULL;
char* szUrlPath = NULL;
char* szParam = NULL;
char* szNetLoc = NULL;
int nPort = 80;
int nSchemeLen = 0;
int nUserLen = 0;
int nPassLen = 0;
int nHostLen = 0;
int nUrlLen = 0;
/* Find out if the URI is absolute... */
BOOL bIsAbsolute = FALSE;
char cAlphanum;
char* ap = (char*)lpszUrl;
char* cp = NULL;
TRACE("\n");
while( (cAlphanum = *ap) != '\0' )
{
if( ((cAlphanum >= 'a') && (cAlphanum <= 'z')) ||
((cAlphanum >= 'A') && (cAlphanum <= 'Z')) ||
((cAlphanum >= '0') && (cAlphanum <= '9')) )
{
ap++;
continue;
}
if( (cAlphanum == ':') && (ap - lpszUrl >= 2) )
{
bIsAbsolute = TRUE;
cp = ap;
break;
}
break;
}
/* Absolute URI...
FIXME!!!! This should work on relative urls too!*/
if( bIsAbsolute )
{
/* Get scheme first... */
nSchemeLen = cp - lpszUrl;
szScheme = strdup( lpszUrl );
szScheme[ nSchemeLen ] = '\0';
/* Eat ':' in protocol... */
cp++;
/* Parse <params>... */
szParam = strpbrk( lpszUrl, ";" );
if( szParam != NULL )
{
char* sParam;
/* Eat ';' in Params... */
szParam++;
sParam = strdup( szParam );
*szParam = '\0';
}
/* Skip over slashes...*/
if( *cp == '/' )
{
cp++;
if( *cp == '/' )
{
cp++;
if( *cp == '/' )
cp++;
}
}
/* Parse the <net-loc>...*/
if( GetInternetScheme( szScheme ) == INTERNET_SCHEME_FILE )
{
szUrlPath = strdup( cp );
nUrlLen = strlen( szUrlPath );
if( nUrlLen >= 2 && szUrlPath[ 1 ] == '|' )
szUrlPath[ 1 ] = ':';
}
else
{
size_t nNetLocLen;
szUrlPath = strpbrk(cp, "/");
if( szUrlPath != NULL )
nUrlLen = strlen( szUrlPath );
/* Find the end of our net-loc... */
nNetLocLen = strcspn( cp, "/" );
szNetLoc = strdup( cp );
szNetLoc[ nNetLocLen ] = '\0';
if( szNetLoc != NULL )
{
char* lpszPort;
int nPortLen;
/* [<user>[<:password>]@]<host>[:<port>] */
/* First find the user and password if they exist...*/
szHost = strchr( szNetLoc, '@' );
if( szHost == NULL )
{
/* username and password not specified... */
szHost = szNetLoc;
nHostLen = nNetLocLen;
}
else
{
int nUserPassLen = nNetLocLen - nHostLen - 1;
char* szUserPass = strdup( szNetLoc );
/* Get username and/or password... */
/* Eat '@' in domain... */
++szHost;
nHostLen = strlen( szHost );
szUserPass[ nUserPassLen ] = '\0';
if( szUserPass != NULL )
{
szPass = strpbrk( szUserPass, ":" );
if( szPass != NULL )
{
/* Eat ':' in UserPass... */
++szPass;
nPassLen = strlen( szPass );
nUserLen = nUserPassLen - nPassLen - 1;
szUser = strdup( szUserPass );
szUser[ nUserLen ] = '\0';
}
else
{
/* password not specified... */
szUser = strdup( szUserPass );
nUserLen = strlen( szUser );
}
}
}
/* <host><:port>...*/
/* Then get the port if it exists... */
lpszPort = strpbrk( szHost, ":" );
nPortLen = 0;
if( lpszPort != NULL )
{
char* szPort = lpszPort + 1;
if( szPort != NULL )
{
nPortLen = strlen( szPort );
nPort = atoi( szPort );
}
*lpszPort = '\0';
nHostLen = strlen(szHost);
}
}
}
}
/* Relative URI... */
else
return FALSE;
return TRUE;
}
/***********************************************************************
* InternetAttemptConnect (WININET.81)
*
* Attempt to make a connection to the internet
*
* RETURNS
* ERROR_SUCCESS on success
* Error value on failure
*
*/
INTERNETAPI DWORD WINAPI InternetAttemptConnect(DWORD dwReserved)
{
FIXME("Stub\n");
return ERROR_SUCCESS;
}
/***********************************************************************
* InternetCanonicalizeUrlA (WININET.85)
*
* Escape unsafe characters and spaces
*
* RETURNS
* TRUE on success
* FALSE on failure
*
*/
BOOL WINAPI InternetCanonicalizeUrlA(LPCSTR lpszUrl, LPSTR lpszBuffer,
LPDWORD lpdwBufferLength, DWORD dwFlags)
{
BOOL bSuccess = FALSE;
FIXME("Stub!\n");
if (lpszUrl)
{
strncpy(lpszBuffer, lpszUrl, *lpdwBufferLength);
*lpdwBufferLength = strlen(lpszBuffer);
bSuccess = TRUE;
}
return bSuccess;
}
/***********************************************************************
* InternetSetStatusCallback (WININET.133)
*
* Sets up a callback function which is called as progress is made
* during an operation.
*
* RETURNS
* Previous callback or NULL on success
* INTERNET_INVALID_STATUS_CALLBACK on failure
*
*/
INTERNETAPI INTERNET_STATUS_CALLBACK WINAPI InternetSetStatusCallback(
HINTERNET hInternet ,INTERNET_STATUS_CALLBACK lpfnIntCB)
{
INTERNET_STATUS_CALLBACK retVal;
LPWININETAPPINFOA lpwai = (LPWININETAPPINFOA)hInternet;
TRACE("0x%08lx\n", (ULONG)hInternet);
if (lpwai->hdr.htype != WH_HINIT)
return INTERNET_INVALID_STATUS_CALLBACK;
retVal = lpwai->lpfnStatusCB;
lpwai->lpfnStatusCB = lpfnIntCB;
return retVal;
}
/***********************************************************************
* InternetWriteFile (WININET.138)
*
* Write data to an open internet file
*
* RETURNS
* TRUE on success
* FALSE on failure
*
*/
BOOL WINAPI InternetWriteFile(HINTERNET hFile, LPCVOID lpBuffer ,
DWORD dwNumOfBytesToWrite, LPDWORD lpdwNumOfBytesWritten)
{
BOOL retval = FALSE;
int nSocket = INVALID_SOCKET;
LPWININETHANDLEHEADER lpwh = (LPWININETHANDLEHEADER) hFile;
TRACE("\n");
if (NULL == lpwh)
return FALSE;
switch (lpwh->htype)
{
case WH_HHTTPREQ:
nSocket = ((LPWININETHTTPREQA)hFile)->nSocketFD;
break;
case WH_HFILE:
nSocket = ((LPWININETFILE)hFile)->nDataSocket;
break;
default:
break;
}
if (INVALID_SOCKET != nSocket)
{
*lpdwNumOfBytesWritten = INTERNET_WriteDataToStream(nSocket, lpBuffer, dwNumOfBytesToWrite);
if (*lpdwNumOfBytesWritten < 0)
*lpdwNumOfBytesWritten = 0;
else
retval = TRUE;
}
return retval;
}
/***********************************************************************
* InternetReadFile (WININET.121)
*
* Read data from an open internet file
*
* RETURNS
* TRUE on success
* FALSE on failure
*
*/
BOOL WINAPI InternetReadFile(HINTERNET hFile, LPVOID lpBuffer,
DWORD dwNumOfBytesToRead, LPDWORD dwNumOfBytesRead)
{
BOOL retval = FALSE;
int nSocket = INVALID_SOCKET;
LPWININETHANDLEHEADER lpwh = (LPWININETHANDLEHEADER) hFile;
TRACE("\n");
if (NULL == lpwh)
return FALSE;
switch (lpwh->htype)
{
case WH_HHTTPREQ:
nSocket = ((LPWININETHTTPREQA)hFile)->nSocketFD;
break;
case WH_HFILE:
nSocket = ((LPWININETFILE)hFile)->nDataSocket;
break;
default:
break;
}
if (INVALID_SOCKET != nSocket)
{
*dwNumOfBytesRead = INTERNET_ReadDataFromStream(nSocket, lpBuffer, dwNumOfBytesToRead);
if (*dwNumOfBytesRead < 0)
*dwNumOfBytesRead = 0;
else
retval = TRUE;
}
return retval;
}
/***********************************************************************
* GetInternetScheme (internal)
*
* Get scheme of url
*
* RETURNS
* scheme on success
* INTERNET_SCHEME_UNKNOWN on failure
*
*/
INTERNET_SCHEME GetInternetScheme(LPSTR lpszScheme)
{
if(lpszScheme==NULL)
return INTERNET_SCHEME_UNKNOWN;
if( (strcmp("ftp", lpszScheme) == 0) ||
(strcmp("FTP", lpszScheme) == 0) )
return INTERNET_SCHEME_FTP;
else if( (strcmp("gopher", lpszScheme) == 0) ||
(strcmp("GOPHER", lpszScheme) == 0) )
return INTERNET_SCHEME_GOPHER;
else if( (strcmp("http", lpszScheme) == 0) ||
(strcmp("HTTP", lpszScheme) == 0) )
return INTERNET_SCHEME_HTTP;
else if( (strcmp("https", lpszScheme) == 0) ||
(strcmp("HTTPS", lpszScheme) == 0) )
return INTERNET_SCHEME_HTTPS;
else if( (strcmp("file", lpszScheme) == 0) ||
(strcmp("FILE", lpszScheme) == 0) )
return INTERNET_SCHEME_FILE;
else if( (strcmp("news", lpszScheme) == 0) ||
(strcmp("NEWS", lpszScheme) == 0) )
return INTERNET_SCHEME_NEWS;
else if( (strcmp("mailto", lpszScheme) == 0) ||
(strcmp("MAILTO", lpszScheme) == 0) )
return INTERNET_SCHEME_MAILTO;
else
return INTERNET_SCHEME_UNKNOWN;
}
/***********************************************************************
* INTERNET_WriteDataToStream (internal)
*
* Send data to server
*
* RETURNS
*
* number of characters sent on success
* -1 on error
*/
int INTERNET_WriteDataToStream(int nDataSocket, LPCVOID Buffer, DWORD BytesToWrite)
{
if (INVALID_SOCKET == nDataSocket)
return SOCKET_ERROR;
return send(nDataSocket, Buffer, BytesToWrite, 0);
}
/***********************************************************************
* INTERNET_ReadDataFromStream (internal)
*
* Read data from http server
*
* RETURNS
*
* number of characters sent on success
* -1 on error
*/
int INTERNET_ReadDataFromStream(int nDataSocket, LPVOID Buffer, DWORD BytesToRead)
{
if (INVALID_SOCKET == nDataSocket)
return SOCKET_ERROR;
return recv(nDataSocket, Buffer, BytesToRead, 0);
}
/***********************************************************************
* INTERNET_SetLastError (internal)
*
* Set last thread specific error
*
* RETURNS
*
*/
void INTERNET_SetLastError(DWORD dwError)
{
LPWITHREADERROR lpwite = (LPWITHREADERROR)TlsGetValue(g_dwTlsErrIndex);
SetLastError(dwError);
lpwite->dwError = dwError;
}
/***********************************************************************
* INTERNET_GetLastError (internal)
*
* Get last thread specific error
*
* RETURNS
*
*/
DWORD INTERNET_GetLastError()
{
LPWITHREADERROR lpwite = (LPWITHREADERROR)TlsGetValue(g_dwTlsErrIndex);
return lpwite->dwError;
}
/***********************************************************************
* INTERNET_WorkerThreadFunc (internal)
*
* Worker thread execution function
*
* RETURNS
*
*/
DWORD INTERNET_WorkerThreadFunc(LPVOID *lpvParam)
{
DWORD dwWaitRes;
while (1)
{
dwWaitRes = WaitForMultipleObjects(2, hEventArray, FALSE, MAX_IDLE_WORKER);
if (dwWaitRes == WAIT_OBJECT_0 + 1)
INTERNET_ExecuteWork();
else
break;
InterlockedIncrement(&dwNumIdleThreads);
}
InterlockedDecrement(&dwNumIdleThreads);
InterlockedDecrement(&dwNumThreads);
TRACE("Worker thread exiting\n");
return TRUE;
}
/***********************************************************************
* INTERNET_InsertWorkRequest (internal)
*
* Insert work request into queue
*
* RETURNS
*
*/
BOOL INTERNET_InsertWorkRequest(LPWORKREQUEST lpWorkRequest)
{
BOOL bSuccess = FALSE;
LPWORKREQUEST lpNewRequest;
TRACE("\n");
lpNewRequest = HeapAlloc(GetProcessHeap(), 0, sizeof(WORKREQUEST));
if (lpNewRequest)
{
memcpy(lpNewRequest, lpWorkRequest, sizeof(WORKREQUEST));
lpNewRequest->prev = NULL;
EnterCriticalSection(&csQueue);
lpNewRequest->next = lpWorkQueueTail;
if (lpWorkQueueTail)
lpWorkQueueTail->prev = lpNewRequest;
lpWorkQueueTail = lpNewRequest;
if (!lpHeadWorkQueue)
lpHeadWorkQueue = lpWorkQueueTail;
LeaveCriticalSection(&csQueue);
bSuccess = TRUE;
}
return bSuccess;
}
/***********************************************************************
* INTERNET_GetWorkRequest (internal)
*
* Retrieves work request from queue
*
* RETURNS
*
*/
BOOL INTERNET_GetWorkRequest(LPWORKREQUEST lpWorkRequest)
{
BOOL bSuccess = FALSE;
LPWORKREQUEST lpRequest = NULL;
TRACE("\n");
EnterCriticalSection(&csQueue);
if (lpHeadWorkQueue)
{
lpRequest = lpHeadWorkQueue;
lpHeadWorkQueue = lpHeadWorkQueue->prev;
if (lpRequest == lpWorkQueueTail)
lpWorkQueueTail = lpHeadWorkQueue;
}
LeaveCriticalSection(&csQueue);
if (lpRequest)
{
memcpy(lpWorkRequest, lpRequest, sizeof(WORKREQUEST));
HeapFree(GetProcessHeap(), 0, lpRequest);
bSuccess = TRUE;
}
return bSuccess;
}
/***********************************************************************
* INTERNET_AsyncCall (internal)
*
* Retrieves work request from queue
*
* RETURNS
*
*/
BOOL INTERNET_AsyncCall(LPWORKREQUEST lpWorkRequest)
{
HANDLE hThread;
DWORD dwTID;
BOOL bSuccess = FALSE;
TRACE("\n");
if (InterlockedDecrement(&dwNumIdleThreads) < 0)
{
InterlockedIncrement(&dwNumIdleThreads);
if (InterlockedIncrement(&dwNumThreads) > MAX_WORKER_THREADS ||
!(hThread = CreateThread(NULL, 0,
(LPTHREAD_START_ROUTINE)INTERNET_WorkerThreadFunc, NULL, 0, &dwTID)))
{
InterlockedDecrement(&dwNumThreads);
INTERNET_SetLastError(ERROR_INTERNET_ASYNC_THREAD_FAILED);
goto lerror;
}
TRACE("Created new thread\n");
}
bSuccess = TRUE;
INTERNET_InsertWorkRequest(lpWorkRequest);
SetEvent(hWorkEvent);
lerror:
return bSuccess;
}
/***********************************************************************
* INTERNET_ExecuteWork (internal)
*
* RETURNS
*
*/
VOID INTERNET_ExecuteWork()
{
WORKREQUEST workRequest;
TRACE("\n");
if (INTERNET_GetWorkRequest(&workRequest))
{
switch (workRequest.asyncall)
{
case FTPPUTFILEA:
FTP_FtpPutFileA((HINTERNET)workRequest.HFTPSESSION, (LPCSTR)workRequest.LPSZLOCALFILE,
(LPCSTR)workRequest.LPSZNEWREMOTEFILE, workRequest.DWFLAGS, workRequest.DWCONTEXT);
HeapFree(GetProcessHeap(), 0, (LPVOID)workRequest.LPSZLOCALFILE);
HeapFree(GetProcessHeap(), 0, (LPVOID)workRequest.LPSZNEWREMOTEFILE);
break;
case FTPSETCURRENTDIRECTORYA:
FTP_FtpSetCurrentDirectoryA((HINTERNET)workRequest.HFTPSESSION,
(LPCSTR)workRequest.LPSZDIRECTORY);
HeapFree(GetProcessHeap(), 0, (LPVOID)workRequest.LPSZDIRECTORY);
break;
case FTPCREATEDIRECTORYA:
FTP_FtpCreateDirectoryA((HINTERNET)workRequest.HFTPSESSION,
(LPCSTR)workRequest.LPSZDIRECTORY);
HeapFree(GetProcessHeap(), 0, (LPVOID)workRequest.LPSZDIRECTORY);
break;
case FTPFINDFIRSTFILEA:
FTP_FtpFindFirstFileA((HINTERNET)workRequest.HFTPSESSION,
(LPCSTR)workRequest.LPSZSEARCHFILE,
(LPWIN32_FIND_DATAA)workRequest.LPFINDFILEDATA, workRequest.DWFLAGS,
workRequest.DWCONTEXT);
HeapFree(GetProcessHeap(), 0, (LPVOID)workRequest.LPSZSEARCHFILE);
break;
case FTPGETCURRENTDIRECTORYA:
FTP_FtpGetCurrentDirectoryA((HINTERNET)workRequest.HFTPSESSION,
(LPSTR)workRequest.LPSZDIRECTORY, (LPDWORD)workRequest.LPDWDIRECTORY);
break;
case FTPOPENFILEA:
FTP_FtpOpenFileA((HINTERNET)workRequest.HFTPSESSION,
(LPCSTR)workRequest.LPSZFILENAME,
workRequest.FDWACCESS,
workRequest.DWFLAGS,
workRequest.DWCONTEXT);
HeapFree(GetProcessHeap(), 0, (LPVOID)workRequest.LPSZFILENAME);
break;
case FTPGETFILEA:
FTP_FtpGetFileA((HINTERNET)workRequest.HFTPSESSION,
(LPCSTR)workRequest.LPSZREMOTEFILE,
(LPCSTR)workRequest.LPSZNEWFILE,
(BOOL)workRequest.FFAILIFEXISTS,
workRequest.DWLOCALFLAGSATTRIBUTE,
workRequest.DWFLAGS,
workRequest.DWCONTEXT);
HeapFree(GetProcessHeap(), 0, (LPVOID)workRequest.LPSZREMOTEFILE);
HeapFree(GetProcessHeap(), 0, (LPVOID)workRequest.LPSZNEWFILE);
break;
case FTPDELETEFILEA:
FTP_FtpDeleteFileA((HINTERNET)workRequest.HFTPSESSION,
(LPCSTR)workRequest.LPSZFILENAME);
HeapFree(GetProcessHeap(), 0, (LPVOID)workRequest.LPSZFILENAME);
break;
case FTPREMOVEDIRECTORYA:
FTP_FtpRemoveDirectoryA((HINTERNET)workRequest.HFTPSESSION,
(LPCSTR)workRequest.LPSZDIRECTORY);
HeapFree(GetProcessHeap(), 0, (LPVOID)workRequest.LPSZDIRECTORY);
break;
case FTPRENAMEFILEA:
FTP_FtpRenameFileA((HINTERNET)workRequest.HFTPSESSION,
(LPCSTR)workRequest.LPSZSRCFILE,
(LPCSTR)workRequest.LPSZDESTFILE);
HeapFree(GetProcessHeap(), 0, (LPVOID)workRequest.LPSZSRCFILE);
HeapFree(GetProcessHeap(), 0, (LPVOID)workRequest.LPSZDESTFILE);
break;
case INTERNETFINDNEXTA:
INTERNET_FindNextFileA((HINTERNET)workRequest.HFTPSESSION,
(LPWIN32_FIND_DATAA)workRequest.LPFINDFILEDATA);
break;
}
}
}
/***********************************************************************
* INTERNET_GetResponseBuffer
*
* RETURNS
*
*/
LPSTR INTERNET_GetResponseBuffer()
{
LPWITHREADERROR lpwite = (LPWITHREADERROR)TlsGetValue(g_dwTlsErrIndex);
return lpwite->response;
}
Reference in New Issue View Git Blame Copy Permalink