/* * Wininet * * Copyright 1999 Corel Corporation * * Ulrich Czekalla * */ #include "config.h" #include #include #ifdef HAVE_SYS_SOCKET_H # include #endif #include #include #include "windows.h" #include "wininet.h" #include "debugtools.h" #include "winerror.h" #include "winsock.h" #include "tchar.h" #include "heap.h" #include "internet.h" DEFAULT_DEBUG_CHANNEL(wininet); #define MAX_IDLE_WORKER 1000*60*1 #define MAX_WORKER_THREADS 10 #define RESPONSE_TIMEOUT 30 #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(LPCSTR lpszScheme, INT nMaxCmp); 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) { LPVOID lpwite = TlsGetValue(g_dwTlsErrIndex); if (lpwite) HeapFree(GetProcessHeap(), 0, lpwite); } 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 = HEAP_strdupA(GetProcessHeap(),0,lpszAgent); if (NULL != lpszProxy) lpwai->lpszProxy = HEAP_strdupA(GetProcessHeap(),0,lpszProxy); if (NULL != lpszProxyBypass) lpwai->lpszProxyBypass = HEAP_strdupA(GetProcessHeap(),0,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: rc = HTTP_Connect(hInternet, lpszServerName, nServerPort, lpszUserName, lpszPassword, dwFlags, dwContext); 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; } /*********************************************************************** * INTERNET_CloseHandle (internal) * * Close internet handle * * RETURNS * Void * */ VOID INTERNET_CloseHandle(LPWININETAPPINFOA lpwai) { if (lpwai->lpszAgent) HeapFree(GetProcessHeap(), 0, lpwai->lpszAgent); if (lpwai->lpszProxy) HeapFree(GetProcessHeap(), 0, lpwai->lpszProxy); if (lpwai->lpszProxyBypass) HeapFree(GetProcessHeap(), 0, lpwai->lpszProxyBypass); HeapFree(GetProcessHeap(), 0, lpwai); } /*********************************************************************** * InternetCloseHandle (WININET.89) * * Generic close handle function * * 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: INTERNET_CloseHandle((LPWININETAPPINFOA) lpwh); break; case WH_HHTTPSESSION: HTTP_CloseHTTPSessionHandle((LPWININETHTTPSESSIONA) lpwh); break; case WH_HHTTPREQ: HTTP_CloseHTTPRequestHandle((LPWININETHTTPREQA) lpwh); break; case WH_HFTPSESSION: retval = FTP_CloseSessionHandle((LPWININETFTPSESSIONA) lpwh); break; case WH_HFINDNEXT: retval = FTP_CloseFindNextHandle((LPWININETFINDNEXTA) lpwh); break; default: break; } return retval; } /*********************************************************************** * SetUrlComponentValue (Internal) * * Helper function for InternetCrackUrlA * * RETURNS * TRUE on success * FALSE on failure * */ BOOL SetUrlComponentValue(LPSTR* lppszComponent, LPDWORD dwComponentLen, LPCSTR lpszStart, INT len) { TRACE("%s (%d)\n", lpszStart, len); if (*dwComponentLen != 0) { if (*lppszComponent == NULL) { *lppszComponent = lpszStart; *dwComponentLen = len; } else { INT ncpylen = min((*dwComponentLen)-1, len); strncpy(*lppszComponent, lpszStart, ncpylen); (*lppszComponent)[ncpylen] = '\0'; *dwComponentLen = ncpylen; } } return TRUE; } /*********************************************************************** * InternetCrackUrlA (WININET.95) * * Break up URL into its components * * TODO: Hadnle dwFlags * * RETURNS * TRUE on success * FALSE on failure * */ BOOL WINAPI InternetCrackUrlA(LPCSTR lpszUrl, DWORD dwUrlLength, DWORD dwFlags, LPURL_COMPONENTSA lpUrlComponents) { /* * RFC 1808 * :[//][/path][;][?][#] * */ LPSTR lpszParam = NULL; BOOL bIsAbsolute = FALSE; LPSTR lpszap = (char*)lpszUrl; LPSTR lpszcp = NULL; TRACE("\n"); /* Determine if the URI is absolute. */ while (*lpszap != '\0') { if (isalnum(*lpszap)) { lpszap++; continue; } if ((*lpszap == ':') && (lpszap - lpszUrl >= 2)) { bIsAbsolute = TRUE; lpszcp = lpszap; } else { lpszcp = lpszUrl; /* Relative url */ } break; } /* Parse */ lpszParam = strpbrk(lpszap, ";?"); if (lpszParam != NULL) { if (!SetUrlComponentValue(&lpUrlComponents->lpszExtraInfo, &lpUrlComponents->dwExtraInfoLength, lpszParam+1, strlen(lpszParam+1))) { return FALSE; } } if (bIsAbsolute) /* Parse :[//] */ { LPSTR lpszNetLoc; /* Get scheme first. */ lpUrlComponents->nScheme = GetInternetScheme(lpszUrl, lpszcp - lpszUrl); if (!SetUrlComponentValue(&lpUrlComponents->lpszScheme, &lpUrlComponents->dwSchemeLength, lpszUrl, lpszcp - lpszUrl)) return FALSE; /* Eat ':' in protocol. */ lpszcp++; /* Skip over slashes. */ if (*lpszcp == '/') { lpszcp++; if (*lpszcp == '/') { lpszcp++; if (*lpszcp == '/') lpszcp++; } } lpszNetLoc = strpbrk(lpszcp, "/"); if (lpszParam) { if (lpszNetLoc) lpszNetLoc = min(lpszNetLoc, lpszParam); else lpszNetLoc = lpszParam; } else if (!lpszNetLoc) lpszNetLoc = lpszcp + strlen(lpszcp); /* Parse net-loc */ if (lpszNetLoc) { LPSTR lpszHost; LPSTR lpszPort; /* [[<:password>]@][:] */ /* First find the user and password if they exist */ lpszHost = strchr(lpszcp, '@'); if (lpszHost == NULL || lpszHost > lpszNetLoc) { /* username and password not specified. */ SetUrlComponentValue(&lpUrlComponents->lpszUserName, &lpUrlComponents->dwUserNameLength, NULL, 0); SetUrlComponentValue(&lpUrlComponents->lpszPassword, &lpUrlComponents->dwPasswordLength, NULL, 0); } else /* Parse out username and password */ { LPSTR lpszUser = lpszcp; LPSTR lpszPasswd = lpszHost; while (lpszcp < lpszHost) { if (*lpszcp == ':') lpszPasswd = lpszcp; lpszcp++; } SetUrlComponentValue(&lpUrlComponents->lpszUserName, &lpUrlComponents->dwUserNameLength, lpszUser, lpszPasswd - lpszUser); SetUrlComponentValue(&lpUrlComponents->lpszPassword, &lpUrlComponents->dwPasswordLength, lpszPasswd == lpszHost ? NULL : ++lpszPasswd, lpszHost - lpszPasswd); lpszcp++; /* Advance to beginning of host */ } /* Parse <:port> */ lpszHost = lpszcp; lpszPort = lpszNetLoc; while (lpszcp < lpszNetLoc) { if (*lpszcp == ':') lpszPort = lpszcp; lpszcp++; } SetUrlComponentValue(&lpUrlComponents->lpszHostName, &lpUrlComponents->dwHostNameLength, lpszHost, lpszPort - lpszHost); if (lpszPort != lpszNetLoc) lpUrlComponents->nPort = atoi(++lpszPort); } } /* Here lpszcp points to: * * :[//][/path][;][?][#] * ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ */ if (lpszcp != 0 && *lpszcp != '\0' && (!lpszParam || lpszcp < lpszParam)) { if (!SetUrlComponentValue(&lpUrlComponents->lpszUrlPath, &lpUrlComponents->dwUrlPathLength, lpszcp, lpszParam ? lpszParam - lpszcp : strlen(lpszcp))) return FALSE; } else { lpUrlComponents->dwUrlPathLength = 0; } TRACE("%s: host(%s) path(%s)\n", lpszUrl, lpUrlComponents->lpszHostName, lpUrlComponents->lpszUrlPath); 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; } /*********************************************************************** * InternetQueryOptionA * * Queries an options on the specified handle * * RETURNS * TRUE on success * FALSE on failure * */ BOOL WINAPI InternetQueryOptionA(HINTERNET hInternet, DWORD dwOption, LPVOID lpBuffer, LPDWORD lpdwBufferLength) { LPWININETHANDLEHEADER lpwhh; BOOL bSuccess = FALSE; TRACE("0x%08lx\n", dwOption); if (NULL == hInternet) { INTERNET_SetLastError(ERROR_INTERNET_INCORRECT_HANDLE_TYPE); return FALSE; } lpwhh = (LPWININETHANDLEHEADER) hInternet; switch (dwOption) { case INTERNET_OPTION_HANDLE_TYPE: { ULONG type = lpwhh->htype; TRACE("INTERNET_OPTION_HANDLE_TYPE: %ld\n", type); if (*lpdwBufferLength < sizeof(ULONG)) INTERNET_SetLastError(ERROR_INSUFFICIENT_BUFFER); else { memcpy(lpBuffer, &type, sizeof(ULONG)); *lpdwBufferLength = sizeof(ULONG); bSuccess = TRUE; } break; } default: FIXME("Stub!"); break; } return bSuccess; } /*********************************************************************** * GetInternetScheme (internal) * * Get scheme of url * * RETURNS * scheme on success * INTERNET_SCHEME_UNKNOWN on failure * */ INTERNET_SCHEME GetInternetScheme(LPCSTR lpszScheme, INT nMaxCmp) { if(lpszScheme==NULL) return INTERNET_SCHEME_UNKNOWN; if (!_strnicmp("ftp", lpszScheme, nMaxCmp)) return INTERNET_SCHEME_FTP; else if (!_strnicmp("gopher", lpszScheme, nMaxCmp)) return INTERNET_SCHEME_GOPHER; else if (!_strnicmp("http", lpszScheme, nMaxCmp)) return INTERNET_SCHEME_HTTP; else if (!_strnicmp("https", lpszScheme, nMaxCmp)) return INTERNET_SCHEME_HTTPS; else if (!_strnicmp("file", lpszScheme, nMaxCmp)) return INTERNET_SCHEME_FILE; else if (!_strnicmp("news", lpszScheme, nMaxCmp)) return INTERNET_SCHEME_NEWS; else if (!_strnicmp("mailto", lpszScheme, nMaxCmp)) 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; case HTTPSENDREQUESTA: HTTP_HttpSendRequestA((HINTERNET)workRequest.HFTPSESSION, (LPCSTR)workRequest.LPSZHEADER, workRequest.DWHEADERLENGTH, (LPVOID)workRequest.LPOPTIONAL, workRequest.DWOPTIONALLENGTH); HeapFree(GetProcessHeap(), 0, (LPVOID)workRequest.LPSZHEADER); break; case HTTPOPENREQUESTA: HTTP_HttpOpenRequestA((HINTERNET)workRequest.HFTPSESSION, (LPCSTR)workRequest.LPSZVERB, (LPCSTR)workRequest.LPSZOBJECTNAME, (LPCSTR)workRequest.LPSZVERSION, (LPCSTR)workRequest.LPSZREFERRER, (LPCSTR*)workRequest.LPSZACCEPTTYPES, workRequest.DWFLAGS, workRequest.DWCONTEXT); HeapFree(GetProcessHeap(), 0, (LPVOID)workRequest.LPSZVERB); HeapFree(GetProcessHeap(), 0, (LPVOID)workRequest.LPSZOBJECTNAME); HeapFree(GetProcessHeap(), 0, (LPVOID)workRequest.LPSZVERSION); HeapFree(GetProcessHeap(), 0, (LPVOID)workRequest.LPSZREFERRER); break; } } } /*********************************************************************** * INTERNET_GetResponseBuffer * * RETURNS * */ LPSTR INTERNET_GetResponseBuffer() { LPWITHREADERROR lpwite = (LPWITHREADERROR)TlsGetValue(g_dwTlsErrIndex); return lpwite->response; } /*********************************************************************** * INTERNET_GetNextLine (internal) * * Parse next line in directory string listing * * RETURNS * Pointer to begining of next line * NULL on failure * */ LPSTR INTERNET_GetNextLine(INT nSocket, LPSTR lpszBuffer, LPDWORD dwBuffer) { struct timeval tv; fd_set infd; BOOL bSuccess = FALSE; INT nRecv = 0; TRACE("\n"); FD_ZERO(&infd); FD_SET(nSocket, &infd); tv.tv_sec=RESPONSE_TIMEOUT; tv.tv_usec=0; while (nRecv < *dwBuffer) { if (select(nSocket+1,&infd,NULL,NULL,&tv) > 0) { if (recv(nSocket, &lpszBuffer[nRecv], 1, 0) <= 0) { INTERNET_SetLastError(ERROR_FTP_TRANSFER_IN_PROGRESS); goto lend; } if (lpszBuffer[nRecv] == '\n') { bSuccess = TRUE; break; } if (lpszBuffer[nRecv] != '\r') nRecv++; } else { INTERNET_SetLastError(ERROR_INTERNET_TIMEOUT); goto lend; } } lend: if (bSuccess) { lpszBuffer[nRecv] = '\0'; *dwBuffer = nRecv - 1; TRACE(":%d %s\n", nRecv, lpszBuffer); return lpszBuffer; } else { return NULL; } }