/* * WININET - Ftp implementation * * Copyright 1999 Corel Corporation * Copyright 2004 Mike McCormack for CodeWeavers * Copyright 2004 Kevin Koltzau * Copyright 2007 Hans Leidekker * * Ulrich Czekalla * Noureddine Jemmali * * Copyright 2000 Andreas Mohr * Copyright 2002 Jaco Greeff * * This library is free software; you can redistribute it and/or * modify it under the terms of the GNU Lesser General Public * License as published by the Free Software Foundation; either * version 2.1 of the License, or (at your option) any later version. * * This library is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU * Lesser General Public License for more details. * * You should have received a copy of the GNU Lesser General Public * License along with this library; if not, write to the Free Software * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA */ #include "ws2tcpip.h" #include #include #include #include #include #include #include "windef.h" #include "winbase.h" #include "wingdi.h" #include "winuser.h" #include "wininet.h" #include "winnls.h" #include "winerror.h" #include "winreg.h" #include "winternl.h" #include "shlwapi.h" #include "wine/debug.h" #include "internet.h" WINE_DEFAULT_DEBUG_CHANNEL(wininet); #define RESPONSE_TIMEOUT 30 typedef struct _ftp_session_t ftp_session_t; typedef struct { object_header_t hdr; ftp_session_t *lpFtpSession; BOOL session_deleted; int nDataSocket; WCHAR *cache_file; HANDLE cache_file_handle; } ftp_file_t; struct _ftp_session_t { object_header_t hdr; appinfo_t *lpAppInfo; int sndSocket; int lstnSocket; int pasvSocket; /* data socket connected by us in case of passive FTP */ ftp_file_t *download_in_progress; struct sockaddr_in socketAddress; struct sockaddr_in lstnSocketAddress; LPWSTR servername; INTERNET_PORT serverport; LPWSTR lpszPassword; LPWSTR lpszUserName; }; typedef struct { BOOL bIsDirectory; LPWSTR lpszName; DWORD nSize; SYSTEMTIME tmLastModified; unsigned short permissions; } FILEPROPERTIESW, *LPFILEPROPERTIESW; typedef struct { object_header_t hdr; ftp_session_t *lpFtpSession; DWORD index; DWORD size; LPFILEPROPERTIESW lpafp; } WININETFTPFINDNEXTW, *LPWININETFTPFINDNEXTW; #define DATA_PACKET_SIZE 0x2000 #define szCRLF "\r\n" #define MAX_BACKLOG 5 /* Testing shows that Windows only accepts dwFlags where the last * 3 (yes 3) bits define FTP_TRANSFER_TYPE_UNKNOWN, FTP_TRANSFER_TYPE_ASCII or FTP_TRANSFER_TYPE_BINARY. */ #define FTP_CONDITION_MASK 0x0007 typedef enum { /* FTP commands with arguments. */ FTP_CMD_ACCT, FTP_CMD_CWD, FTP_CMD_DELE, FTP_CMD_MKD, FTP_CMD_PASS, FTP_CMD_PORT, FTP_CMD_RETR, FTP_CMD_RMD, FTP_CMD_RNFR, FTP_CMD_RNTO, FTP_CMD_STOR, FTP_CMD_TYPE, FTP_CMD_USER, FTP_CMD_SIZE, /* FTP commands without arguments. */ FTP_CMD_ABOR, FTP_CMD_LIST, FTP_CMD_NLST, FTP_CMD_PASV, FTP_CMD_PWD, FTP_CMD_QUIT, } FTP_COMMAND; static const CHAR *const szFtpCommands[] = { "ACCT", "CWD", "DELE", "MKD", "PASS", "PORT", "RETR", "RMD", "RNFR", "RNTO", "STOR", "TYPE", "USER", "SIZE", "ABOR", "LIST", "NLST", "PASV", "PWD", "QUIT", }; static const CHAR szMonths[] = "JANFEBMARAPRMAYJUNJULAUGSEPOCTNOVDEC"; static const WCHAR szNoAccount[] = {'n','o','a','c','c','o','u','n','t','\0'}; static BOOL FTP_SendCommand(INT nSocket, FTP_COMMAND ftpCmd, LPCWSTR lpszParam, INTERNET_STATUS_CALLBACK lpfnStatusCB, object_header_t *hdr, DWORD_PTR dwContext); static BOOL FTP_SendStore(ftp_session_t*, LPCWSTR lpszRemoteFile, DWORD dwType); static BOOL FTP_GetDataSocket(ftp_session_t*, LPINT nDataSocket); static BOOL FTP_SendData(ftp_session_t*, INT nDataSocket, HANDLE hFile); static INT FTP_ReceiveResponse(ftp_session_t*, DWORD_PTR dwContext); static BOOL FTP_SendRetrieve(ftp_session_t*, LPCWSTR lpszRemoteFile, DWORD dwType); static BOOL FTP_RetrieveFileData(ftp_session_t*, INT nDataSocket, HANDLE hFile); static BOOL FTP_InitListenSocket(ftp_session_t*); static BOOL FTP_ConnectToHost(ftp_session_t*); static BOOL FTP_SendPassword(ftp_session_t*); static BOOL FTP_SendAccount(ftp_session_t*); static BOOL FTP_SendType(ftp_session_t*, DWORD dwType); static BOOL FTP_SendPort(ftp_session_t*); static BOOL FTP_DoPassive(ftp_session_t*); static BOOL FTP_SendPortOrPasv(ftp_session_t*); static BOOL FTP_ParsePermission(LPCSTR lpszPermission, LPFILEPROPERTIESW lpfp); static BOOL FTP_ParseNextFile(INT nSocket, LPCWSTR lpszSearchFile, LPFILEPROPERTIESW fileprop); static BOOL FTP_ParseDirectory(ftp_session_t*, INT nSocket, LPCWSTR lpszSearchFile, LPFILEPROPERTIESW *lpafp, LPDWORD dwfp); static HINTERNET FTP_ReceiveFileList(ftp_session_t*, INT nSocket, LPCWSTR lpszSearchFile, LPWIN32_FIND_DATAW lpFindFileData, DWORD_PTR dwContext); static DWORD FTP_SetResponseError(DWORD dwResponse); static BOOL FTP_ConvertFileProp(LPFILEPROPERTIESW lpafp, LPWIN32_FIND_DATAW lpFindFileData); static BOOL FTP_FtpPutFileW(ftp_session_t*, LPCWSTR lpszLocalFile, LPCWSTR lpszNewRemoteFile, DWORD dwFlags, DWORD_PTR dwContext); static BOOL FTP_FtpSetCurrentDirectoryW(ftp_session_t*, LPCWSTR lpszDirectory); static BOOL FTP_FtpCreateDirectoryW(ftp_session_t*, LPCWSTR lpszDirectory); static HINTERNET FTP_FtpFindFirstFileW(ftp_session_t*, LPCWSTR lpszSearchFile, LPWIN32_FIND_DATAW lpFindFileData, DWORD dwFlags, DWORD_PTR dwContext); static BOOL FTP_FtpGetCurrentDirectoryW(ftp_session_t*, LPWSTR lpszCurrentDirectory, LPDWORD lpdwCurrentDirectory); static BOOL FTP_FtpRenameFileW(ftp_session_t*, LPCWSTR lpszSrc, LPCWSTR lpszDest); static BOOL FTP_FtpRemoveDirectoryW(ftp_session_t*, LPCWSTR lpszDirectory); static BOOL FTP_FtpDeleteFileW(ftp_session_t*, LPCWSTR lpszFileName); static BOOL FTP_FtpGetFileW(ftp_session_t*, LPCWSTR lpszRemoteFile, LPCWSTR lpszNewFile, BOOL fFailIfExists, DWORD dwLocalFlagsAttribute, DWORD dwInternetFlags, DWORD_PTR dwContext); /* A temporary helper until we get rid of INTERNET_GetLastError calls */ static BOOL res_to_le(DWORD res) { if(res != ERROR_SUCCESS) INTERNET_SetLastError(res); return res == ERROR_SUCCESS; } /*********************************************************************** * FtpPutFileA (WININET.@) * * Uploads a file to the FTP server * * RETURNS * TRUE on success * FALSE on failure * */ BOOL WINAPI FtpPutFileA(HINTERNET hConnect, LPCSTR lpszLocalFile, LPCSTR lpszNewRemoteFile, DWORD dwFlags, DWORD_PTR dwContext) { LPWSTR lpwzLocalFile; LPWSTR lpwzNewRemoteFile; BOOL ret; lpwzLocalFile = heap_strdupAtoW(lpszLocalFile); lpwzNewRemoteFile = heap_strdupAtoW(lpszNewRemoteFile); ret = FtpPutFileW(hConnect, lpwzLocalFile, lpwzNewRemoteFile, dwFlags, dwContext); heap_free(lpwzLocalFile); heap_free(lpwzNewRemoteFile); return ret; } typedef struct { task_header_t hdr; WCHAR *local_file; WCHAR *remote_file; DWORD flags; DWORD_PTR context; } put_file_task_t; static void AsyncFtpPutFileProc(task_header_t *hdr) { put_file_task_t *task = (put_file_task_t*)hdr; ftp_session_t *session = (ftp_session_t*)task->hdr.hdr; TRACE("%p\n", session); FTP_FtpPutFileW(session, task->local_file, task->remote_file, task->flags, task->context); heap_free(task->local_file); heap_free(task->remote_file); } /*********************************************************************** * FtpPutFileW (WININET.@) * * Uploads a file to the FTP server * * RETURNS * TRUE on success * FALSE on failure * */ BOOL WINAPI FtpPutFileW(HINTERNET hConnect, LPCWSTR lpszLocalFile, LPCWSTR lpszNewRemoteFile, DWORD dwFlags, DWORD_PTR dwContext) { ftp_session_t *lpwfs; appinfo_t *hIC = NULL; BOOL r = FALSE; if (!lpszLocalFile || !lpszNewRemoteFile) { INTERNET_SetLastError(ERROR_INVALID_PARAMETER); return FALSE; } lpwfs = (ftp_session_t*) get_handle_object( hConnect ); if (!lpwfs) { INTERNET_SetLastError(ERROR_INVALID_HANDLE); return FALSE; } if (WH_HFTPSESSION != lpwfs->hdr.htype) { INTERNET_SetLastError(ERROR_INTERNET_INCORRECT_HANDLE_TYPE); goto lend; } if (lpwfs->download_in_progress != NULL) { INTERNET_SetLastError(ERROR_FTP_TRANSFER_IN_PROGRESS); goto lend; } if ((dwFlags & FTP_CONDITION_MASK) > FTP_TRANSFER_TYPE_BINARY) { INTERNET_SetLastError(ERROR_INVALID_PARAMETER); goto lend; } hIC = lpwfs->lpAppInfo; if (hIC->hdr.dwFlags & INTERNET_FLAG_ASYNC) { put_file_task_t *task = alloc_async_task(&lpwfs->hdr, AsyncFtpPutFileProc, sizeof(*task)); task->local_file = heap_strdupW(lpszLocalFile); task->remote_file = heap_strdupW(lpszNewRemoteFile); task->flags = dwFlags; task->context = dwContext; r = res_to_le(INTERNET_AsyncCall(&task->hdr)); } else { r = FTP_FtpPutFileW(lpwfs, lpszLocalFile, lpszNewRemoteFile, dwFlags, dwContext); } lend: WININET_Release( &lpwfs->hdr ); return r; } /*********************************************************************** * FTP_FtpPutFileW (Internal) * * Uploads a file to the FTP server * * RETURNS * TRUE on success * FALSE on failure * */ static BOOL FTP_FtpPutFileW(ftp_session_t *lpwfs, LPCWSTR lpszLocalFile, LPCWSTR lpszNewRemoteFile, DWORD dwFlags, DWORD_PTR dwContext) { HANDLE hFile; BOOL bSuccess = FALSE; appinfo_t *hIC = NULL; INT nResCode; TRACE(" lpszLocalFile(%s) lpszNewRemoteFile(%s)\n", debugstr_w(lpszLocalFile), debugstr_w(lpszNewRemoteFile)); /* Clear any error information */ INTERNET_SetLastError(0); /* Open file to be uploaded */ if (INVALID_HANDLE_VALUE == (hFile = CreateFileW(lpszLocalFile, GENERIC_READ, 0, 0, OPEN_EXISTING, 0, 0))) /* Let CreateFile set the appropriate error */ return FALSE; hIC = lpwfs->lpAppInfo; INTERNET_SendCallback(&lpwfs->hdr, lpwfs->hdr.dwContext, INTERNET_STATUS_SENDING_REQUEST, NULL, 0); if (FTP_SendStore(lpwfs, lpszNewRemoteFile, dwFlags)) { INT nDataSocket; /* Get data socket to server */ if (FTP_GetDataSocket(lpwfs, &nDataSocket)) { FTP_SendData(lpwfs, nDataSocket, hFile); closesocket(nDataSocket); nResCode = FTP_ReceiveResponse(lpwfs, dwContext); if (nResCode) { if (nResCode == 226) bSuccess = TRUE; else FTP_SetResponseError(nResCode); } } } if (lpwfs->lstnSocket != -1) { closesocket(lpwfs->lstnSocket); lpwfs->lstnSocket = -1; } if (hIC->hdr.dwFlags & INTERNET_FLAG_ASYNC) { INTERNET_ASYNC_RESULT iar; iar.dwResult = (DWORD)bSuccess; iar.dwError = bSuccess ? ERROR_SUCCESS : INTERNET_GetLastError(); INTERNET_SendCallback(&lpwfs->hdr, lpwfs->hdr.dwContext, INTERNET_STATUS_REQUEST_COMPLETE, &iar, sizeof(INTERNET_ASYNC_RESULT)); } CloseHandle(hFile); return bSuccess; } /*********************************************************************** * FtpSetCurrentDirectoryA (WININET.@) * * Change the working directory on the FTP server * * RETURNS * TRUE on success * FALSE on failure * */ BOOL WINAPI FtpSetCurrentDirectoryA(HINTERNET hConnect, LPCSTR lpszDirectory) { LPWSTR lpwzDirectory; BOOL ret; lpwzDirectory = heap_strdupAtoW(lpszDirectory); ret = FtpSetCurrentDirectoryW(hConnect, lpwzDirectory); heap_free(lpwzDirectory); return ret; } typedef struct { task_header_t hdr; WCHAR *directory; } directory_task_t; static void AsyncFtpSetCurrentDirectoryProc(task_header_t *hdr) { directory_task_t *task = (directory_task_t*)hdr; ftp_session_t *session = (ftp_session_t*)task->hdr.hdr; TRACE("%p\n", session); FTP_FtpSetCurrentDirectoryW(session, task->directory); heap_free(task->directory); } /*********************************************************************** * FtpSetCurrentDirectoryW (WININET.@) * * Change the working directory on the FTP server * * RETURNS * TRUE on success * FALSE on failure * */ BOOL WINAPI FtpSetCurrentDirectoryW(HINTERNET hConnect, LPCWSTR lpszDirectory) { ftp_session_t *lpwfs = NULL; appinfo_t *hIC = NULL; BOOL r = FALSE; if (!lpszDirectory) { INTERNET_SetLastError(ERROR_INVALID_PARAMETER); goto lend; } lpwfs = (ftp_session_t*) get_handle_object( hConnect ); if (NULL == lpwfs || WH_HFTPSESSION != lpwfs->hdr.htype) { INTERNET_SetLastError(ERROR_INTERNET_INCORRECT_HANDLE_TYPE); goto lend; } if (lpwfs->download_in_progress != NULL) { INTERNET_SetLastError(ERROR_FTP_TRANSFER_IN_PROGRESS); goto lend; } TRACE("lpszDirectory(%s)\n", debugstr_w(lpszDirectory)); hIC = lpwfs->lpAppInfo; if (hIC->hdr.dwFlags & INTERNET_FLAG_ASYNC) { directory_task_t *task; task = alloc_async_task(&lpwfs->hdr, AsyncFtpSetCurrentDirectoryProc, sizeof(*task)); task->directory = heap_strdupW(lpszDirectory); r = res_to_le(INTERNET_AsyncCall(&task->hdr)); } else { r = FTP_FtpSetCurrentDirectoryW(lpwfs, lpszDirectory); } lend: if( lpwfs ) WININET_Release( &lpwfs->hdr ); return r; } /*********************************************************************** * FTP_FtpSetCurrentDirectoryW (Internal) * * Change the working directory on the FTP server * * RETURNS * TRUE on success * FALSE on failure * */ static BOOL FTP_FtpSetCurrentDirectoryW(ftp_session_t *lpwfs, LPCWSTR lpszDirectory) { INT nResCode; appinfo_t *hIC = NULL; BOOL bSuccess = FALSE; TRACE("lpszDirectory(%s)\n", debugstr_w(lpszDirectory)); /* Clear any error information */ INTERNET_SetLastError(0); hIC = lpwfs->lpAppInfo; if (!FTP_SendCommand(lpwfs->sndSocket, FTP_CMD_CWD, lpszDirectory, lpwfs->hdr.lpfnStatusCB, &lpwfs->hdr, lpwfs->hdr.dwContext)) goto lend; nResCode = FTP_ReceiveResponse(lpwfs, lpwfs->hdr.dwContext); if (nResCode) { if (nResCode == 250) bSuccess = TRUE; else FTP_SetResponseError(nResCode); } lend: if (hIC->hdr.dwFlags & INTERNET_FLAG_ASYNC) { INTERNET_ASYNC_RESULT iar; iar.dwResult = bSuccess; iar.dwError = bSuccess ? ERROR_SUCCESS : ERROR_INTERNET_EXTENDED_ERROR; INTERNET_SendCallback(&lpwfs->hdr, lpwfs->hdr.dwContext, INTERNET_STATUS_REQUEST_COMPLETE, &iar, sizeof(INTERNET_ASYNC_RESULT)); } return bSuccess; } /*********************************************************************** * FtpCreateDirectoryA (WININET.@) * * Create new directory on the FTP server * * RETURNS * TRUE on success * FALSE on failure * */ BOOL WINAPI FtpCreateDirectoryA(HINTERNET hConnect, LPCSTR lpszDirectory) { LPWSTR lpwzDirectory; BOOL ret; lpwzDirectory = heap_strdupAtoW(lpszDirectory); ret = FtpCreateDirectoryW(hConnect, lpwzDirectory); heap_free(lpwzDirectory); return ret; } static void AsyncFtpCreateDirectoryProc(task_header_t *hdr) { directory_task_t *task = (directory_task_t*)hdr; ftp_session_t *session = (ftp_session_t*)task->hdr.hdr; TRACE(" %p\n", session); FTP_FtpCreateDirectoryW(session, task->directory); heap_free(task->directory); } /*********************************************************************** * FtpCreateDirectoryW (WININET.@) * * Create new directory on the FTP server * * RETURNS * TRUE on success * FALSE on failure * */ BOOL WINAPI FtpCreateDirectoryW(HINTERNET hConnect, LPCWSTR lpszDirectory) { ftp_session_t *lpwfs; appinfo_t *hIC = NULL; BOOL r = FALSE; lpwfs = (ftp_session_t*) get_handle_object( hConnect ); if (!lpwfs) { INTERNET_SetLastError(ERROR_INVALID_HANDLE); return FALSE; } if (WH_HFTPSESSION != lpwfs->hdr.htype) { INTERNET_SetLastError(ERROR_INTERNET_INCORRECT_HANDLE_TYPE); goto lend; } if (lpwfs->download_in_progress != NULL) { INTERNET_SetLastError(ERROR_FTP_TRANSFER_IN_PROGRESS); goto lend; } if (!lpszDirectory) { INTERNET_SetLastError(ERROR_INVALID_PARAMETER); goto lend; } hIC = lpwfs->lpAppInfo; if (hIC->hdr.dwFlags & INTERNET_FLAG_ASYNC) { directory_task_t *task; task = alloc_async_task(&lpwfs->hdr, AsyncFtpCreateDirectoryProc, sizeof(*task)); task->directory = heap_strdupW(lpszDirectory); r = res_to_le(INTERNET_AsyncCall(&task->hdr)); } else { r = FTP_FtpCreateDirectoryW(lpwfs, lpszDirectory); } lend: WININET_Release( &lpwfs->hdr ); return r; } /*********************************************************************** * FTP_FtpCreateDirectoryW (Internal) * * Create new directory on the FTP server * * RETURNS * TRUE on success * FALSE on failure * */ static BOOL FTP_FtpCreateDirectoryW(ftp_session_t *lpwfs, LPCWSTR lpszDirectory) { INT nResCode; BOOL bSuccess = FALSE; appinfo_t *hIC = NULL; TRACE("lpszDirectory(%s)\n", debugstr_w(lpszDirectory)); /* Clear any error information */ INTERNET_SetLastError(0); if (!FTP_SendCommand(lpwfs->sndSocket, FTP_CMD_MKD, lpszDirectory, 0, 0, 0)) goto lend; nResCode = FTP_ReceiveResponse(lpwfs, lpwfs->hdr.dwContext); if (nResCode) { if (nResCode == 257) bSuccess = TRUE; else FTP_SetResponseError(nResCode); } lend: hIC = lpwfs->lpAppInfo; if (hIC->hdr.dwFlags & INTERNET_FLAG_ASYNC) { INTERNET_ASYNC_RESULT iar; iar.dwResult = (DWORD)bSuccess; iar.dwError = bSuccess ? ERROR_SUCCESS : INTERNET_GetLastError(); INTERNET_SendCallback(&lpwfs->hdr, lpwfs->hdr.dwContext, INTERNET_STATUS_REQUEST_COMPLETE, &iar, sizeof(INTERNET_ASYNC_RESULT)); } return bSuccess; } /*********************************************************************** * FtpFindFirstFileA (WININET.@) * * Search the specified directory * * RETURNS * HINTERNET on success * NULL on failure * */ HINTERNET WINAPI FtpFindFirstFileA(HINTERNET hConnect, LPCSTR lpszSearchFile, LPWIN32_FIND_DATAA lpFindFileData, DWORD dwFlags, DWORD_PTR dwContext) { LPWSTR lpwzSearchFile; WIN32_FIND_DATAW wfd; LPWIN32_FIND_DATAW lpFindFileDataW; HINTERNET ret; lpwzSearchFile = heap_strdupAtoW(lpszSearchFile); lpFindFileDataW = lpFindFileData?&wfd:NULL; ret = FtpFindFirstFileW(hConnect, lpwzSearchFile, lpFindFileDataW, dwFlags, dwContext); heap_free(lpwzSearchFile); if (ret && lpFindFileData) WININET_find_data_WtoA(lpFindFileDataW, lpFindFileData); return ret; } typedef struct { task_header_t hdr; WCHAR *search_file; WIN32_FIND_DATAW *find_file_data; DWORD flags; DWORD_PTR context; } find_first_file_task_t; static void AsyncFtpFindFirstFileProc(task_header_t *hdr) { find_first_file_task_t *task = (find_first_file_task_t*)hdr; ftp_session_t *session = (ftp_session_t*)task->hdr.hdr; TRACE("%p\n", session); FTP_FtpFindFirstFileW(session, task->search_file, task->find_file_data, task->flags, task->context); heap_free(task->search_file); } /*********************************************************************** * FtpFindFirstFileW (WININET.@) * * Search the specified directory * * RETURNS * HINTERNET on success * NULL on failure * */ HINTERNET WINAPI FtpFindFirstFileW(HINTERNET hConnect, LPCWSTR lpszSearchFile, LPWIN32_FIND_DATAW lpFindFileData, DWORD dwFlags, DWORD_PTR dwContext) { ftp_session_t *lpwfs; appinfo_t *hIC = NULL; HINTERNET r = NULL; lpwfs = (ftp_session_t*) get_handle_object( hConnect ); if (NULL == lpwfs || WH_HFTPSESSION != lpwfs->hdr.htype) { INTERNET_SetLastError(ERROR_INTERNET_INCORRECT_HANDLE_TYPE); goto lend; } if (lpwfs->download_in_progress != NULL) { INTERNET_SetLastError(ERROR_FTP_TRANSFER_IN_PROGRESS); goto lend; } hIC = lpwfs->lpAppInfo; if (hIC->hdr.dwFlags & INTERNET_FLAG_ASYNC) { find_first_file_task_t *task; task = alloc_async_task(&lpwfs->hdr, AsyncFtpFindFirstFileProc, sizeof(*task)); task->search_file = heap_strdupW(lpszSearchFile); task->find_file_data = lpFindFileData; task->flags = dwFlags; task->context = dwContext; INTERNET_AsyncCall(&task->hdr); r = NULL; } else { r = FTP_FtpFindFirstFileW(lpwfs, lpszSearchFile, lpFindFileData, dwFlags, dwContext); } lend: if( lpwfs ) WININET_Release( &lpwfs->hdr ); return r; } /*********************************************************************** * FTP_FtpFindFirstFileW (Internal) * * Search the specified directory * * RETURNS * HINTERNET on success * NULL on failure * */ static HINTERNET FTP_FtpFindFirstFileW(ftp_session_t *lpwfs, LPCWSTR lpszSearchFile, LPWIN32_FIND_DATAW lpFindFileData, DWORD dwFlags, DWORD_PTR dwContext) { INT nResCode; appinfo_t *hIC = NULL; HINTERNET hFindNext = NULL; LPWSTR lpszSearchPath = NULL; TRACE("\n"); /* Clear any error information */ INTERNET_SetLastError(0); if (!FTP_InitListenSocket(lpwfs)) goto lend; if (!FTP_SendType(lpwfs, INTERNET_FLAG_TRANSFER_ASCII)) goto lend; if (!FTP_SendPortOrPasv(lpwfs)) goto lend; /* split search path into file and path */ if (lpszSearchFile) { LPCWSTR name = lpszSearchFile, p; if ((p = wcsrchr( name, '\\' ))) name = p + 1; if ((p = wcsrchr( name, '/' ))) name = p + 1; if (name != lpszSearchFile) { lpszSearchPath = heap_strndupW(lpszSearchFile, name - lpszSearchFile); lpszSearchFile = name; } } if (!FTP_SendCommand(lpwfs->sndSocket, FTP_CMD_LIST, lpszSearchPath, lpwfs->hdr.lpfnStatusCB, &lpwfs->hdr, lpwfs->hdr.dwContext)) goto lend; nResCode = FTP_ReceiveResponse(lpwfs, lpwfs->hdr.dwContext); if (nResCode) { if (nResCode == 125 || nResCode == 150) { INT nDataSocket; /* Get data socket to server */ if (FTP_GetDataSocket(lpwfs, &nDataSocket)) { hFindNext = FTP_ReceiveFileList(lpwfs, nDataSocket, lpszSearchFile, lpFindFileData, dwContext); closesocket(nDataSocket); nResCode = FTP_ReceiveResponse(lpwfs, lpwfs->hdr.dwContext); if (nResCode != 226 && nResCode != 250) INTERNET_SetLastError(ERROR_NO_MORE_FILES); } } else FTP_SetResponseError(nResCode); } lend: heap_free(lpszSearchPath); if (lpwfs->lstnSocket != -1) { closesocket(lpwfs->lstnSocket); lpwfs->lstnSocket = -1; } hIC = lpwfs->lpAppInfo; if (hIC->hdr.dwFlags & INTERNET_FLAG_ASYNC) { INTERNET_ASYNC_RESULT iar; if (hFindNext) { iar.dwResult = (DWORD_PTR)hFindNext; iar.dwError = ERROR_SUCCESS; INTERNET_SendCallback(&lpwfs->hdr, lpwfs->hdr.dwContext, INTERNET_STATUS_HANDLE_CREATED, &iar, sizeof(INTERNET_ASYNC_RESULT)); } iar.dwResult = (DWORD_PTR)hFindNext; iar.dwError = hFindNext ? ERROR_SUCCESS : INTERNET_GetLastError(); INTERNET_SendCallback(&lpwfs->hdr, lpwfs->hdr.dwContext, INTERNET_STATUS_REQUEST_COMPLETE, &iar, sizeof(INTERNET_ASYNC_RESULT)); } return hFindNext; } /*********************************************************************** * FtpGetCurrentDirectoryA (WININET.@) * * Retrieves the current directory * * RETURNS * TRUE on success * FALSE on failure * */ BOOL WINAPI FtpGetCurrentDirectoryA(HINTERNET hFtpSession, LPSTR lpszCurrentDirectory, LPDWORD lpdwCurrentDirectory) { WCHAR *dir = NULL; DWORD len; BOOL ret; if(lpdwCurrentDirectory) { len = *lpdwCurrentDirectory; if(lpszCurrentDirectory) { dir = heap_alloc(len * sizeof(WCHAR)); if (NULL == dir) { INTERNET_SetLastError(ERROR_OUTOFMEMORY); return FALSE; } } } ret = FtpGetCurrentDirectoryW(hFtpSession, lpszCurrentDirectory?dir:NULL, lpdwCurrentDirectory?&len:NULL); if (ret && lpszCurrentDirectory) WideCharToMultiByte(CP_ACP, 0, dir, -1, lpszCurrentDirectory, len, NULL, NULL); if (lpdwCurrentDirectory) *lpdwCurrentDirectory = len; heap_free(dir); return ret; } typedef struct { task_header_t hdr; WCHAR *directory; DWORD *directory_len; } get_current_dir_task_t; static void AsyncFtpGetCurrentDirectoryProc(task_header_t *hdr) { get_current_dir_task_t *task = (get_current_dir_task_t*)hdr; ftp_session_t *session = (ftp_session_t*)task->hdr.hdr; TRACE("%p\n", session); FTP_FtpGetCurrentDirectoryW(session, task->directory, task->directory_len); } /*********************************************************************** * FtpGetCurrentDirectoryW (WININET.@) * * Retrieves the current directory * * RETURNS * TRUE on success * FALSE on failure * */ BOOL WINAPI FtpGetCurrentDirectoryW(HINTERNET hFtpSession, LPWSTR lpszCurrentDirectory, LPDWORD lpdwCurrentDirectory) { ftp_session_t *lpwfs; appinfo_t *hIC = NULL; BOOL r = FALSE; TRACE("%p %p %p\n", hFtpSession, lpszCurrentDirectory, lpdwCurrentDirectory); lpwfs = (ftp_session_t*) get_handle_object( hFtpSession ); if (NULL == lpwfs) { INTERNET_SetLastError(ERROR_INVALID_HANDLE); goto lend; } if (WH_HFTPSESSION != lpwfs->hdr.htype) { INTERNET_SetLastError(ERROR_INTERNET_INCORRECT_HANDLE_TYPE); goto lend; } if (!lpdwCurrentDirectory) { INTERNET_SetLastError(ERROR_INVALID_PARAMETER); goto lend; } if (lpszCurrentDirectory == NULL) { INTERNET_SetLastError(ERROR_INSUFFICIENT_BUFFER); goto lend; } if (lpwfs->download_in_progress != NULL) { INTERNET_SetLastError(ERROR_FTP_TRANSFER_IN_PROGRESS); goto lend; } hIC = lpwfs->lpAppInfo; if (hIC->hdr.dwFlags & INTERNET_FLAG_ASYNC) { get_current_dir_task_t *task; task = alloc_async_task(&lpwfs->hdr, AsyncFtpGetCurrentDirectoryProc, sizeof(*task)); task->directory = lpszCurrentDirectory; task->directory_len = lpdwCurrentDirectory; r = res_to_le(INTERNET_AsyncCall(&task->hdr)); } else { r = FTP_FtpGetCurrentDirectoryW(lpwfs, lpszCurrentDirectory, lpdwCurrentDirectory); } lend: if( lpwfs ) WININET_Release( &lpwfs->hdr ); return r; } /*********************************************************************** * FTP_FtpGetCurrentDirectoryW (Internal) * * Retrieves the current directory * * RETURNS * TRUE on success * FALSE on failure * */ static BOOL FTP_FtpGetCurrentDirectoryW(ftp_session_t *lpwfs, LPWSTR lpszCurrentDirectory, LPDWORD lpdwCurrentDirectory) { INT nResCode; appinfo_t *hIC = NULL; BOOL bSuccess = FALSE; /* Clear any error information */ INTERNET_SetLastError(0); hIC = lpwfs->lpAppInfo; if (!FTP_SendCommand(lpwfs->sndSocket, FTP_CMD_PWD, NULL, lpwfs->hdr.lpfnStatusCB, &lpwfs->hdr, lpwfs->hdr.dwContext)) goto lend; nResCode = FTP_ReceiveResponse(lpwfs, lpwfs->hdr.dwContext); if (nResCode) { if (nResCode == 257) /* Extract directory name */ { DWORD firstpos, lastpos, len; LPWSTR lpszResponseBuffer = heap_strdupAtoW(INTERNET_GetResponseBuffer()); for (firstpos = 0, lastpos = 0; lpszResponseBuffer[lastpos]; lastpos++) { if ('"' == lpszResponseBuffer[lastpos]) { if (!firstpos) firstpos = lastpos; else break; } } len = lastpos - firstpos; if (*lpdwCurrentDirectory >= len) { memcpy(lpszCurrentDirectory, &lpszResponseBuffer[firstpos + 1], len * sizeof(WCHAR)); lpszCurrentDirectory[len - 1] = 0; *lpdwCurrentDirectory = len; bSuccess = TRUE; } else INTERNET_SetLastError(ERROR_INSUFFICIENT_BUFFER); heap_free(lpszResponseBuffer); } else FTP_SetResponseError(nResCode); } lend: if (hIC->hdr.dwFlags & INTERNET_FLAG_ASYNC) { INTERNET_ASYNC_RESULT iar; iar.dwResult = bSuccess; iar.dwError = bSuccess ? ERROR_SUCCESS : ERROR_INTERNET_EXTENDED_ERROR; INTERNET_SendCallback(&lpwfs->hdr, lpwfs->hdr.dwContext, INTERNET_STATUS_REQUEST_COMPLETE, &iar, sizeof(INTERNET_ASYNC_RESULT)); } return bSuccess; } /*********************************************************************** * FTPFILE_Destroy(internal) * * Closes the file transfer handle. This also 'cleans' the data queue of * the 'transfer complete' message (this is a bit of a hack though :-/ ) * */ static void FTPFILE_Destroy(object_header_t *hdr) { ftp_file_t *lpwh = (ftp_file_t*) hdr; ftp_session_t *lpwfs = lpwh->lpFtpSession; INT nResCode; TRACE("\n"); if (lpwh->cache_file_handle != INVALID_HANDLE_VALUE) CloseHandle(lpwh->cache_file_handle); heap_free(lpwh->cache_file); if (!lpwh->session_deleted) lpwfs->download_in_progress = NULL; if (lpwh->nDataSocket != -1) closesocket(lpwh->nDataSocket); nResCode = FTP_ReceiveResponse(lpwfs, lpwfs->hdr.dwContext); if (nResCode > 0 && nResCode != 226) WARN("server reports failed transfer\n"); WININET_Release(&lpwh->lpFtpSession->hdr); } static DWORD FTPFILE_QueryOption(object_header_t *hdr, DWORD option, void *buffer, DWORD *size, BOOL unicode) { switch(option) { case INTERNET_OPTION_HANDLE_TYPE: TRACE("INTERNET_OPTION_HANDLE_TYPE\n"); if (*size < sizeof(ULONG)) return ERROR_INSUFFICIENT_BUFFER; *size = sizeof(DWORD); *(DWORD*)buffer = INTERNET_HANDLE_TYPE_FTP_FILE; return ERROR_SUCCESS; case INTERNET_OPTION_DATAFILE_NAME: { DWORD required; ftp_file_t *file = (ftp_file_t *)hdr; TRACE("INTERNET_OPTION_DATAFILE_NAME\n"); if (!file->cache_file) { *size = 0; return ERROR_INTERNET_ITEM_NOT_FOUND; } if (unicode) { required = (lstrlenW(file->cache_file) + 1) * sizeof(WCHAR); if (*size < required) return ERROR_INSUFFICIENT_BUFFER; *size = required; memcpy(buffer, file->cache_file, *size); return ERROR_SUCCESS; } else { required = WideCharToMultiByte(CP_ACP, 0, file->cache_file, -1, NULL, 0, NULL, NULL); if (required > *size) return ERROR_INSUFFICIENT_BUFFER; *size = WideCharToMultiByte(CP_ACP, 0, file->cache_file, -1, buffer, *size, NULL, NULL); return ERROR_SUCCESS; } } } return INET_QueryOption(hdr, option, buffer, size, unicode); } static DWORD FTPFILE_ReadFile(object_header_t *hdr, void *buffer, DWORD size, DWORD *read, DWORD flags, DWORD_PTR context) { ftp_file_t *file = (ftp_file_t*)hdr; int res; DWORD error; if (file->nDataSocket == -1) return ERROR_INTERNET_DISCONNECTED; /* FIXME: FTP should use NETCON_ stuff */ res = sock_recv(file->nDataSocket, buffer, size, MSG_WAITALL); *read = res>0 ? res : 0; error = res >= 0 ? ERROR_SUCCESS : INTERNET_ERROR_BASE; /* FIXME */ if (error == ERROR_SUCCESS && file->cache_file) { DWORD bytes_written; if (!WriteFile(file->cache_file_handle, buffer, *read, &bytes_written, NULL)) WARN("WriteFile failed: %u\n", GetLastError()); } return error; } static DWORD FTPFILE_WriteFile(object_header_t *hdr, const void *buffer, DWORD size, DWORD *written) { ftp_file_t *lpwh = (ftp_file_t*) hdr; int res; res = sock_send(lpwh->nDataSocket, buffer, size, 0); *written = res>0 ? res : 0; return res >= 0 ? ERROR_SUCCESS : WSAGetLastError(); } static void FTP_ReceiveRequestData(ftp_file_t *file, BOOL first_notif) { INTERNET_ASYNC_RESULT iar; BYTE buffer[4096]; int available; TRACE("%p\n", file); available = sock_recv(file->nDataSocket, buffer, sizeof(buffer), MSG_PEEK); if(available != -1) { iar.dwResult = (DWORD_PTR)file->hdr.hInternet; iar.dwError = first_notif ? 0 : available; }else { iar.dwResult = 0; iar.dwError = INTERNET_GetLastError(); } INTERNET_SendCallback(&file->hdr, file->hdr.dwContext, INTERNET_STATUS_REQUEST_COMPLETE, &iar, sizeof(INTERNET_ASYNC_RESULT)); } static void FTPFILE_AsyncQueryDataAvailableProc(task_header_t *task) { ftp_file_t *file = (ftp_file_t*)task->hdr; FTP_ReceiveRequestData(file, FALSE); } static DWORD FTPFILE_QueryDataAvailable(object_header_t *hdr, DWORD *available, DWORD flags, DWORD_PTR ctx) { ftp_file_t *file = (ftp_file_t*) hdr; ULONG unread = 0; int retval; TRACE("(%p %p %x %lx)\n", file, available, flags, ctx); retval = ioctlsocket(file->nDataSocket, FIONREAD, &unread); if (!retval) TRACE("%d bytes of queued, but unread data\n", unread); *available = unread; if(!unread) { BYTE byte; *available = 0; retval = sock_recv(file->nDataSocket, &byte, 1, MSG_PEEK); if(retval > 0) { task_header_t *task; task = alloc_async_task(&file->hdr, FTPFILE_AsyncQueryDataAvailableProc, sizeof(*task)); INTERNET_AsyncCall(task); return ERROR_IO_PENDING; } } return ERROR_SUCCESS; } static DWORD FTPFILE_LockRequestFile(object_header_t *hdr, req_file_t **ret) { ftp_file_t *file = (ftp_file_t*)hdr; FIXME("%p\n", file); return ERROR_NOT_SUPPORTED; } static const object_vtbl_t FTPFILEVtbl = { FTPFILE_Destroy, NULL, FTPFILE_QueryOption, INET_SetOption, FTPFILE_ReadFile, FTPFILE_WriteFile, FTPFILE_QueryDataAvailable, NULL, FTPFILE_LockRequestFile }; /*********************************************************************** * FTP_FtpOpenFileW (Internal) * * Open a remote file for writing or reading * * RETURNS * HINTERNET handle on success * NULL on failure * */ static HINTERNET FTP_FtpOpenFileW(ftp_session_t *lpwfs, LPCWSTR lpszFileName, DWORD fdwAccess, DWORD dwFlags, DWORD_PTR dwContext) { INT nDataSocket; BOOL bSuccess = FALSE; ftp_file_t *lpwh = NULL; appinfo_t *hIC = NULL; TRACE("\n"); /* Clear any error information */ INTERNET_SetLastError(0); if (GENERIC_READ == fdwAccess) { /* Set up socket to retrieve data */ bSuccess = FTP_SendRetrieve(lpwfs, lpszFileName, dwFlags); } else if (GENERIC_WRITE == fdwAccess) { /* Set up socket to send data */ bSuccess = FTP_SendStore(lpwfs, lpszFileName, dwFlags); } /* Get data socket to server */ if (bSuccess && FTP_GetDataSocket(lpwfs, &nDataSocket)) { lpwh = alloc_object(&lpwfs->hdr, &FTPFILEVtbl, sizeof(ftp_file_t)); lpwh->hdr.htype = WH_HFILE; lpwh->hdr.dwFlags = dwFlags; lpwh->hdr.dwContext = dwContext; lpwh->nDataSocket = nDataSocket; lpwh->cache_file = NULL; lpwh->cache_file_handle = INVALID_HANDLE_VALUE; lpwh->session_deleted = FALSE; WININET_AddRef( &lpwfs->hdr ); lpwh->lpFtpSession = lpwfs; list_add_head( &lpwfs->hdr.children, &lpwh->hdr.entry ); /* Indicate that a download is currently in progress */ lpwfs->download_in_progress = lpwh; } if (lpwfs->lstnSocket != -1) { closesocket(lpwfs->lstnSocket); lpwfs->lstnSocket = -1; } if (bSuccess && fdwAccess == GENERIC_READ) { WCHAR filename[MAX_PATH + 1]; URL_COMPONENTSW uc; DWORD len; memset(&uc, 0, sizeof(uc)); uc.dwStructSize = sizeof(uc); uc.nScheme = INTERNET_SCHEME_FTP; uc.lpszHostName = lpwfs->servername; uc.nPort = lpwfs->serverport; uc.lpszUserName = lpwfs->lpszUserName; uc.lpszUrlPath = heap_strdupW(lpszFileName); if (!InternetCreateUrlW(&uc, 0, NULL, &len) && GetLastError() == ERROR_INSUFFICIENT_BUFFER) { WCHAR *url = heap_alloc(len * sizeof(WCHAR)); if (url && InternetCreateUrlW(&uc, 0, url, &len) && CreateUrlCacheEntryW(url, 0, NULL, filename, 0)) { lpwh->cache_file = heap_strdupW(filename); lpwh->cache_file_handle = CreateFileW(filename, GENERIC_WRITE, FILE_SHARE_READ, NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL); if (lpwh->cache_file_handle == INVALID_HANDLE_VALUE) { WARN("Could not create cache file: %u\n", GetLastError()); heap_free(lpwh->cache_file); lpwh->cache_file = NULL; } } heap_free(url); } heap_free(uc.lpszUrlPath); } hIC = lpwfs->lpAppInfo; if (hIC->hdr.dwFlags & INTERNET_FLAG_ASYNC) { INTERNET_ASYNC_RESULT iar; if (lpwh) { iar.dwResult = (DWORD_PTR)lpwh->hdr.hInternet; iar.dwError = ERROR_SUCCESS; INTERNET_SendCallback(&lpwfs->hdr, lpwfs->hdr.dwContext, INTERNET_STATUS_HANDLE_CREATED, &iar, sizeof(INTERNET_ASYNC_RESULT)); } if(bSuccess) { FTP_ReceiveRequestData(lpwh, TRUE); }else { iar.dwResult = 0; iar.dwError = INTERNET_GetLastError(); INTERNET_SendCallback(&lpwfs->hdr, lpwfs->hdr.dwContext, INTERNET_STATUS_REQUEST_COMPLETE, &iar, sizeof(INTERNET_ASYNC_RESULT)); } } if(!bSuccess) return FALSE; return lpwh->hdr.hInternet; } /*********************************************************************** * FtpOpenFileA (WININET.@) * * Open a remote file for writing or reading * * RETURNS * HINTERNET handle on success * NULL on failure * */ HINTERNET WINAPI FtpOpenFileA(HINTERNET hFtpSession, LPCSTR lpszFileName, DWORD fdwAccess, DWORD dwFlags, DWORD_PTR dwContext) { LPWSTR lpwzFileName; HINTERNET ret; lpwzFileName = heap_strdupAtoW(lpszFileName); ret = FtpOpenFileW(hFtpSession, lpwzFileName, fdwAccess, dwFlags, dwContext); heap_free(lpwzFileName); return ret; } typedef struct { task_header_t hdr; WCHAR *file_name; DWORD access; DWORD flags; DWORD_PTR context; } open_file_task_t; static void AsyncFtpOpenFileProc(task_header_t *hdr) { open_file_task_t *task = (open_file_task_t*)hdr; ftp_session_t *session = (ftp_session_t*)task->hdr.hdr; TRACE("%p\n", session); FTP_FtpOpenFileW(session, task->file_name, task->access, task->flags, task->context); heap_free(task->file_name); } /*********************************************************************** * FtpOpenFileW (WININET.@) * * Open a remote file for writing or reading * * RETURNS * HINTERNET handle on success * NULL on failure * */ HINTERNET WINAPI FtpOpenFileW(HINTERNET hFtpSession, LPCWSTR lpszFileName, DWORD fdwAccess, DWORD dwFlags, DWORD_PTR dwContext) { ftp_session_t *lpwfs; appinfo_t *hIC = NULL; HINTERNET r = NULL; TRACE("(%p,%s,0x%08x,0x%08x,0x%08lx)\n", hFtpSession, debugstr_w(lpszFileName), fdwAccess, dwFlags, dwContext); lpwfs = (ftp_session_t*) get_handle_object( hFtpSession ); if (!lpwfs) { INTERNET_SetLastError(ERROR_INVALID_HANDLE); return FALSE; } if (WH_HFTPSESSION != lpwfs->hdr.htype) { INTERNET_SetLastError(ERROR_INTERNET_INCORRECT_HANDLE_TYPE); goto lend; } if ((!lpszFileName) || ((fdwAccess != GENERIC_READ) && (fdwAccess != GENERIC_WRITE)) || ((dwFlags & FTP_CONDITION_MASK) > FTP_TRANSFER_TYPE_BINARY)) { INTERNET_SetLastError(ERROR_INVALID_PARAMETER); goto lend; } if (lpwfs->download_in_progress != NULL) { INTERNET_SetLastError(ERROR_FTP_TRANSFER_IN_PROGRESS); goto lend; } hIC = lpwfs->lpAppInfo; if (hIC->hdr.dwFlags & INTERNET_FLAG_ASYNC) { open_file_task_t *task; task = alloc_async_task(&lpwfs->hdr, AsyncFtpOpenFileProc, sizeof(*task)); task->file_name = heap_strdupW(lpszFileName); task->access = fdwAccess; task->flags = dwFlags; task->context = dwContext; INTERNET_AsyncCall(&task->hdr); r = NULL; } else { r = FTP_FtpOpenFileW(lpwfs, lpszFileName, fdwAccess, dwFlags, dwContext); } lend: WININET_Release( &lpwfs->hdr ); return r; } /*********************************************************************** * FtpGetFileA (WININET.@) * * Retrieve file from the FTP server * * RETURNS * TRUE on success * FALSE on failure * */ BOOL WINAPI FtpGetFileA(HINTERNET hInternet, LPCSTR lpszRemoteFile, LPCSTR lpszNewFile, BOOL fFailIfExists, DWORD dwLocalFlagsAttribute, DWORD dwInternetFlags, DWORD_PTR dwContext) { LPWSTR lpwzRemoteFile; LPWSTR lpwzNewFile; BOOL ret; lpwzRemoteFile = heap_strdupAtoW(lpszRemoteFile); lpwzNewFile = heap_strdupAtoW(lpszNewFile); ret = FtpGetFileW(hInternet, lpwzRemoteFile, lpwzNewFile, fFailIfExists, dwLocalFlagsAttribute, dwInternetFlags, dwContext); heap_free(lpwzRemoteFile); heap_free(lpwzNewFile); return ret; } typedef struct { task_header_t hdr; WCHAR *remote_file; WCHAR *new_file; BOOL fail_if_exists; DWORD local_attr; DWORD flags; DWORD_PTR context; } get_file_task_t; static void AsyncFtpGetFileProc(task_header_t *hdr) { get_file_task_t *task = (get_file_task_t*)hdr; ftp_session_t *session = (ftp_session_t*)task->hdr.hdr; TRACE("%p\n", session); FTP_FtpGetFileW(session, task->remote_file, task->new_file, task->fail_if_exists, task->local_attr, task->flags, task->context); heap_free(task->remote_file); heap_free(task->new_file); } /*********************************************************************** * FtpGetFileW (WININET.@) * * Retrieve file from the FTP server * * RETURNS * TRUE on success * FALSE on failure * */ BOOL WINAPI FtpGetFileW(HINTERNET hInternet, LPCWSTR lpszRemoteFile, LPCWSTR lpszNewFile, BOOL fFailIfExists, DWORD dwLocalFlagsAttribute, DWORD dwInternetFlags, DWORD_PTR dwContext) { ftp_session_t *lpwfs; appinfo_t *hIC = NULL; BOOL r = FALSE; if (!lpszRemoteFile || !lpszNewFile) { INTERNET_SetLastError(ERROR_INVALID_PARAMETER); return FALSE; } lpwfs = (ftp_session_t*) get_handle_object( hInternet ); if (!lpwfs) { INTERNET_SetLastError(ERROR_INVALID_HANDLE); return FALSE; } if (WH_HFTPSESSION != lpwfs->hdr.htype) { INTERNET_SetLastError(ERROR_INTERNET_INCORRECT_HANDLE_TYPE); goto lend; } if ((dwInternetFlags & FTP_CONDITION_MASK) > FTP_TRANSFER_TYPE_BINARY) { INTERNET_SetLastError(ERROR_INVALID_PARAMETER); goto lend; } if (lpwfs->download_in_progress != NULL) { INTERNET_SetLastError(ERROR_FTP_TRANSFER_IN_PROGRESS); goto lend; } hIC = lpwfs->lpAppInfo; if (hIC->hdr.dwFlags & INTERNET_FLAG_ASYNC) { get_file_task_t *task; task = alloc_async_task(&lpwfs->hdr, AsyncFtpGetFileProc, sizeof(*task)); task->remote_file = heap_strdupW(lpszRemoteFile); task->new_file = heap_strdupW(lpszNewFile); task->local_attr = dwLocalFlagsAttribute; task->fail_if_exists = fFailIfExists; task->flags = dwInternetFlags; task->context = dwContext; r = res_to_le(INTERNET_AsyncCall(&task->hdr)); } else { r = FTP_FtpGetFileW(lpwfs, lpszRemoteFile, lpszNewFile, fFailIfExists, dwLocalFlagsAttribute, dwInternetFlags, dwContext); } lend: WININET_Release( &lpwfs->hdr ); return r; } /*********************************************************************** * FTP_FtpGetFileW (Internal) * * Retrieve file from the FTP server * * RETURNS * TRUE on success * FALSE on failure * */ static BOOL FTP_FtpGetFileW(ftp_session_t *lpwfs, LPCWSTR lpszRemoteFile, LPCWSTR lpszNewFile, BOOL fFailIfExists, DWORD dwLocalFlagsAttribute, DWORD dwInternetFlags, DWORD_PTR dwContext) { BOOL bSuccess = FALSE; HANDLE hFile; appinfo_t *hIC = NULL; TRACE("lpszRemoteFile(%s) lpszNewFile(%s)\n", debugstr_w(lpszRemoteFile), debugstr_w(lpszNewFile)); /* Clear any error information */ INTERNET_SetLastError(0); /* Ensure we can write to lpszNewfile by opening it */ hFile = CreateFileW(lpszNewFile, GENERIC_WRITE, 0, 0, fFailIfExists ? CREATE_NEW : CREATE_ALWAYS, dwLocalFlagsAttribute, 0); if (INVALID_HANDLE_VALUE == hFile) return FALSE; /* Set up socket to retrieve data */ if (FTP_SendRetrieve(lpwfs, lpszRemoteFile, dwInternetFlags)) { INT nDataSocket; /* Get data socket to server */ if (FTP_GetDataSocket(lpwfs, &nDataSocket)) { INT nResCode; /* Receive data */ FTP_RetrieveFileData(lpwfs, nDataSocket, hFile); closesocket(nDataSocket); nResCode = FTP_ReceiveResponse(lpwfs, dwContext); if (nResCode) { if (nResCode == 226) bSuccess = TRUE; else FTP_SetResponseError(nResCode); } } } if (lpwfs->lstnSocket != -1) { closesocket(lpwfs->lstnSocket); lpwfs->lstnSocket = -1; } CloseHandle(hFile); hIC = lpwfs->lpAppInfo; if (hIC->hdr.dwFlags & INTERNET_FLAG_ASYNC) { INTERNET_ASYNC_RESULT iar; iar.dwResult = (DWORD)bSuccess; iar.dwError = bSuccess ? ERROR_SUCCESS : INTERNET_GetLastError(); INTERNET_SendCallback(&lpwfs->hdr, lpwfs->hdr.dwContext, INTERNET_STATUS_REQUEST_COMPLETE, &iar, sizeof(INTERNET_ASYNC_RESULT)); } if (!bSuccess) DeleteFileW(lpszNewFile); return bSuccess; } /*********************************************************************** * FtpGetFileSize (WININET.@) */ DWORD WINAPI FtpGetFileSize( HINTERNET hFile, LPDWORD lpdwFileSizeHigh ) { FIXME("(%p, %p)\n", hFile, lpdwFileSizeHigh); if (lpdwFileSizeHigh) *lpdwFileSizeHigh = 0; return 0; } /*********************************************************************** * FtpDeleteFileA (WININET.@) * * Delete a file on the ftp server * * RETURNS * TRUE on success * FALSE on failure * */ BOOL WINAPI FtpDeleteFileA(HINTERNET hFtpSession, LPCSTR lpszFileName) { LPWSTR lpwzFileName; BOOL ret; lpwzFileName = heap_strdupAtoW(lpszFileName); ret = FtpDeleteFileW(hFtpSession, lpwzFileName); heap_free(lpwzFileName); return ret; } typedef struct { task_header_t hdr; WCHAR *file_name; } delete_file_task_t; static void AsyncFtpDeleteFileProc(task_header_t *hdr) { delete_file_task_t *task = (delete_file_task_t*)hdr; ftp_session_t *session = (ftp_session_t*)task->hdr.hdr; TRACE("%p\n", session); FTP_FtpDeleteFileW(session, task->file_name); heap_free(task->file_name); } /*********************************************************************** * FtpDeleteFileW (WININET.@) * * Delete a file on the ftp server * * RETURNS * TRUE on success * FALSE on failure * */ BOOL WINAPI FtpDeleteFileW(HINTERNET hFtpSession, LPCWSTR lpszFileName) { ftp_session_t *lpwfs; appinfo_t *hIC = NULL; BOOL r = FALSE; lpwfs = (ftp_session_t*) get_handle_object( hFtpSession ); if (!lpwfs) { INTERNET_SetLastError(ERROR_INVALID_HANDLE); return FALSE; } if (WH_HFTPSESSION != lpwfs->hdr.htype) { INTERNET_SetLastError(ERROR_INTERNET_INCORRECT_HANDLE_TYPE); goto lend; } if (lpwfs->download_in_progress != NULL) { INTERNET_SetLastError(ERROR_FTP_TRANSFER_IN_PROGRESS); goto lend; } if (!lpszFileName) { INTERNET_SetLastError(ERROR_INVALID_PARAMETER); goto lend; } hIC = lpwfs->lpAppInfo; if (hIC->hdr.dwFlags & INTERNET_FLAG_ASYNC) { delete_file_task_t *task; task = alloc_async_task(&lpwfs->hdr, AsyncFtpDeleteFileProc, sizeof(*task)); task->file_name = heap_strdupW(lpszFileName); r = res_to_le(INTERNET_AsyncCall(&task->hdr)); } else { r = FTP_FtpDeleteFileW(lpwfs, lpszFileName); } lend: WININET_Release( &lpwfs->hdr ); return r; } /*********************************************************************** * FTP_FtpDeleteFileW (Internal) * * Delete a file on the ftp server * * RETURNS * TRUE on success * FALSE on failure * */ BOOL FTP_FtpDeleteFileW(ftp_session_t *lpwfs, LPCWSTR lpszFileName) { INT nResCode; BOOL bSuccess = FALSE; appinfo_t *hIC = NULL; TRACE("%p\n", lpwfs); /* Clear any error information */ INTERNET_SetLastError(0); if (!FTP_SendCommand(lpwfs->sndSocket, FTP_CMD_DELE, lpszFileName, 0, 0, 0)) goto lend; nResCode = FTP_ReceiveResponse(lpwfs, lpwfs->hdr.dwContext); if (nResCode) { if (nResCode == 250) bSuccess = TRUE; else FTP_SetResponseError(nResCode); } lend: hIC = lpwfs->lpAppInfo; if (hIC->hdr.dwFlags & INTERNET_FLAG_ASYNC) { INTERNET_ASYNC_RESULT iar; iar.dwResult = (DWORD)bSuccess; iar.dwError = bSuccess ? ERROR_SUCCESS : INTERNET_GetLastError(); INTERNET_SendCallback(&lpwfs->hdr, lpwfs->hdr.dwContext, INTERNET_STATUS_REQUEST_COMPLETE, &iar, sizeof(INTERNET_ASYNC_RESULT)); } return bSuccess; } /*********************************************************************** * FtpRemoveDirectoryA (WININET.@) * * Remove a directory on the ftp server * * RETURNS * TRUE on success * FALSE on failure * */ BOOL WINAPI FtpRemoveDirectoryA(HINTERNET hFtpSession, LPCSTR lpszDirectory) { LPWSTR lpwzDirectory; BOOL ret; lpwzDirectory = heap_strdupAtoW(lpszDirectory); ret = FtpRemoveDirectoryW(hFtpSession, lpwzDirectory); heap_free(lpwzDirectory); return ret; } static void AsyncFtpRemoveDirectoryProc(task_header_t *hdr) { directory_task_t *task = (directory_task_t*)hdr; ftp_session_t *session = (ftp_session_t*)task->hdr.hdr; TRACE("%p\n", session); FTP_FtpRemoveDirectoryW(session, task->directory); heap_free(task->directory); } /*********************************************************************** * FtpRemoveDirectoryW (WININET.@) * * Remove a directory on the ftp server * * RETURNS * TRUE on success * FALSE on failure * */ BOOL WINAPI FtpRemoveDirectoryW(HINTERNET hFtpSession, LPCWSTR lpszDirectory) { ftp_session_t *lpwfs; appinfo_t *hIC = NULL; BOOL r = FALSE; lpwfs = (ftp_session_t*) get_handle_object( hFtpSession ); if (!lpwfs) { INTERNET_SetLastError(ERROR_INVALID_HANDLE); return FALSE; } if (WH_HFTPSESSION != lpwfs->hdr.htype) { INTERNET_SetLastError(ERROR_INTERNET_INCORRECT_HANDLE_TYPE); goto lend; } if (lpwfs->download_in_progress != NULL) { INTERNET_SetLastError(ERROR_FTP_TRANSFER_IN_PROGRESS); goto lend; } if (!lpszDirectory) { INTERNET_SetLastError(ERROR_INVALID_PARAMETER); goto lend; } hIC = lpwfs->lpAppInfo; if (hIC->hdr.dwFlags & INTERNET_FLAG_ASYNC) { directory_task_t *task; task = alloc_async_task(&lpwfs->hdr, AsyncFtpRemoveDirectoryProc, sizeof(*task)); task->directory = heap_strdupW(lpszDirectory); r = res_to_le(INTERNET_AsyncCall(&task->hdr)); } else { r = FTP_FtpRemoveDirectoryW(lpwfs, lpszDirectory); } lend: WININET_Release( &lpwfs->hdr ); return r; } /*********************************************************************** * FTP_FtpRemoveDirectoryW (Internal) * * Remove a directory on the ftp server * * RETURNS * TRUE on success * FALSE on failure * */ BOOL FTP_FtpRemoveDirectoryW(ftp_session_t *lpwfs, LPCWSTR lpszDirectory) { INT nResCode; BOOL bSuccess = FALSE; appinfo_t *hIC = NULL; TRACE("\n"); /* Clear any error information */ INTERNET_SetLastError(0); if (!FTP_SendCommand(lpwfs->sndSocket, FTP_CMD_RMD, lpszDirectory, 0, 0, 0)) goto lend; nResCode = FTP_ReceiveResponse(lpwfs, lpwfs->hdr.dwContext); if (nResCode) { if (nResCode == 250) bSuccess = TRUE; else FTP_SetResponseError(nResCode); } lend: hIC = lpwfs->lpAppInfo; if (hIC->hdr.dwFlags & INTERNET_FLAG_ASYNC) { INTERNET_ASYNC_RESULT iar; iar.dwResult = (DWORD)bSuccess; iar.dwError = bSuccess ? ERROR_SUCCESS : INTERNET_GetLastError(); INTERNET_SendCallback(&lpwfs->hdr, lpwfs->hdr.dwContext, INTERNET_STATUS_REQUEST_COMPLETE, &iar, sizeof(INTERNET_ASYNC_RESULT)); } return bSuccess; } /*********************************************************************** * FtpRenameFileA (WININET.@) * * Rename a file on the ftp server * * RETURNS * TRUE on success * FALSE on failure * */ BOOL WINAPI FtpRenameFileA(HINTERNET hFtpSession, LPCSTR lpszSrc, LPCSTR lpszDest) { LPWSTR lpwzSrc; LPWSTR lpwzDest; BOOL ret; lpwzSrc = heap_strdupAtoW(lpszSrc); lpwzDest = heap_strdupAtoW(lpszDest); ret = FtpRenameFileW(hFtpSession, lpwzSrc, lpwzDest); heap_free(lpwzSrc); heap_free(lpwzDest); return ret; } typedef struct { task_header_t hdr; WCHAR *src_file; WCHAR *dst_file; } rename_file_task_t; static void AsyncFtpRenameFileProc(task_header_t *hdr) { rename_file_task_t *task = (rename_file_task_t*)hdr; ftp_session_t *session = (ftp_session_t*)task->hdr.hdr; TRACE("%p\n", session); FTP_FtpRenameFileW(session, task->src_file, task->dst_file); heap_free(task->src_file); heap_free(task->dst_file); } /*********************************************************************** * FtpRenameFileW (WININET.@) * * Rename a file on the ftp server * * RETURNS * TRUE on success * FALSE on failure * */ BOOL WINAPI FtpRenameFileW(HINTERNET hFtpSession, LPCWSTR lpszSrc, LPCWSTR lpszDest) { ftp_session_t *lpwfs; appinfo_t *hIC = NULL; BOOL r = FALSE; lpwfs = (ftp_session_t*) get_handle_object( hFtpSession ); if (!lpwfs) { INTERNET_SetLastError(ERROR_INVALID_HANDLE); return FALSE; } if (WH_HFTPSESSION != lpwfs->hdr.htype) { INTERNET_SetLastError(ERROR_INTERNET_INCORRECT_HANDLE_TYPE); goto lend; } if (lpwfs->download_in_progress != NULL) { INTERNET_SetLastError(ERROR_FTP_TRANSFER_IN_PROGRESS); goto lend; } if (!lpszSrc || !lpszDest) { INTERNET_SetLastError(ERROR_INVALID_PARAMETER); goto lend; } hIC = lpwfs->lpAppInfo; if (hIC->hdr.dwFlags & INTERNET_FLAG_ASYNC) { rename_file_task_t *task; task = alloc_async_task(&lpwfs->hdr, AsyncFtpRenameFileProc, sizeof(*task)); task->src_file = heap_strdupW(lpszSrc); task->dst_file = heap_strdupW(lpszDest); r = res_to_le(INTERNET_AsyncCall(&task->hdr)); } else { r = FTP_FtpRenameFileW(lpwfs, lpszSrc, lpszDest); } lend: WININET_Release( &lpwfs->hdr ); return r; } /*********************************************************************** * FTP_FtpRenameFileW (Internal) * * Rename a file on the ftp server * * RETURNS * TRUE on success * FALSE on failure * */ BOOL FTP_FtpRenameFileW(ftp_session_t *lpwfs, LPCWSTR lpszSrc, LPCWSTR lpszDest) { INT nResCode; BOOL bSuccess = FALSE; appinfo_t *hIC = NULL; TRACE("\n"); /* Clear any error information */ INTERNET_SetLastError(0); if (!FTP_SendCommand(lpwfs->sndSocket, FTP_CMD_RNFR, lpszSrc, 0, 0, 0)) goto lend; nResCode = FTP_ReceiveResponse(lpwfs, lpwfs->hdr.dwContext); if (nResCode == 350) { if (!FTP_SendCommand(lpwfs->sndSocket, FTP_CMD_RNTO, lpszDest, 0, 0, 0)) goto lend; nResCode = FTP_ReceiveResponse(lpwfs, lpwfs->hdr.dwContext); } if (nResCode == 250) bSuccess = TRUE; else FTP_SetResponseError(nResCode); lend: hIC = lpwfs->lpAppInfo; if (hIC->hdr.dwFlags & INTERNET_FLAG_ASYNC) { INTERNET_ASYNC_RESULT iar; iar.dwResult = (DWORD)bSuccess; iar.dwError = bSuccess ? ERROR_SUCCESS : INTERNET_GetLastError(); INTERNET_SendCallback(&lpwfs->hdr, lpwfs->hdr.dwContext, INTERNET_STATUS_REQUEST_COMPLETE, &iar, sizeof(INTERNET_ASYNC_RESULT)); } return bSuccess; } /*********************************************************************** * FtpCommandA (WININET.@) */ BOOL WINAPI FtpCommandA( HINTERNET hConnect, BOOL fExpectResponse, DWORD dwFlags, LPCSTR lpszCommand, DWORD_PTR dwContext, HINTERNET* phFtpCommand ) { BOOL r; WCHAR *cmdW; TRACE("%p %d 0x%08x %s 0x%08lx %p\n", hConnect, fExpectResponse, dwFlags, debugstr_a(lpszCommand), dwContext, phFtpCommand); if (fExpectResponse) { FIXME("data connection not supported\n"); return FALSE; } if (!lpszCommand || !lpszCommand[0]) { INTERNET_SetLastError(ERROR_INVALID_PARAMETER); return FALSE; } if (!(cmdW = heap_strdupAtoW(lpszCommand))) { INTERNET_SetLastError(ERROR_OUTOFMEMORY); return FALSE; } r = FtpCommandW(hConnect, fExpectResponse, dwFlags, cmdW, dwContext, phFtpCommand); heap_free(cmdW); return r; } /*********************************************************************** * FtpCommandW (WININET.@) */ BOOL WINAPI FtpCommandW( HINTERNET hConnect, BOOL fExpectResponse, DWORD dwFlags, LPCWSTR lpszCommand, DWORD_PTR dwContext, HINTERNET* phFtpCommand ) { BOOL r = FALSE; ftp_session_t *lpwfs; LPSTR cmd = NULL; DWORD len, nBytesSent= 0; INT nResCode, nRC = 0; TRACE("%p %d 0x%08x %s 0x%08lx %p\n", hConnect, fExpectResponse, dwFlags, debugstr_w(lpszCommand), dwContext, phFtpCommand); if (!lpszCommand || !lpszCommand[0]) { INTERNET_SetLastError(ERROR_INVALID_PARAMETER); return FALSE; } if (fExpectResponse) { FIXME("data connection not supported\n"); return FALSE; } lpwfs = (ftp_session_t*) get_handle_object( hConnect ); if (!lpwfs) { INTERNET_SetLastError(ERROR_INVALID_HANDLE); return FALSE; } if (WH_HFTPSESSION != lpwfs->hdr.htype) { INTERNET_SetLastError(ERROR_INTERNET_INCORRECT_HANDLE_TYPE); goto lend; } if (lpwfs->download_in_progress != NULL) { INTERNET_SetLastError(ERROR_FTP_TRANSFER_IN_PROGRESS); goto lend; } len = WideCharToMultiByte(CP_ACP, 0, lpszCommand, -1, NULL, 0, NULL, NULL) + strlen(szCRLF); if ((cmd = heap_alloc(len))) WideCharToMultiByte(CP_ACP, 0, lpszCommand, -1, cmd, len, NULL, NULL); else { INTERNET_SetLastError(ERROR_OUTOFMEMORY); goto lend; } strcat(cmd, szCRLF); len--; TRACE("Sending (%s) len(%d)\n", debugstr_a(cmd), len); while ((nBytesSent < len) && (nRC != -1)) { nRC = sock_send(lpwfs->sndSocket, cmd + nBytesSent, len - nBytesSent, 0); if (nRC != -1) { nBytesSent += nRC; TRACE("Sent %d bytes\n", nRC); } } if (nBytesSent) { nResCode = FTP_ReceiveResponse(lpwfs, lpwfs->hdr.dwContext); if (nResCode > 0 && nResCode < 400) r = TRUE; else FTP_SetResponseError(nResCode); } lend: WININET_Release( &lpwfs->hdr ); heap_free( cmd ); return r; } /*********************************************************************** * FTPSESSION_Destroy (internal) * * Deallocate session handle */ static void FTPSESSION_Destroy(object_header_t *hdr) { ftp_session_t *lpwfs = (ftp_session_t*) hdr; TRACE("\n"); WININET_Release(&lpwfs->lpAppInfo->hdr); heap_free(lpwfs->lpszPassword); heap_free(lpwfs->lpszUserName); heap_free(lpwfs->servername); } static void FTPSESSION_CloseConnection(object_header_t *hdr) { ftp_session_t *lpwfs = (ftp_session_t*) hdr; TRACE("\n"); INTERNET_SendCallback(&lpwfs->hdr, lpwfs->hdr.dwContext, INTERNET_STATUS_CLOSING_CONNECTION, 0, 0); if (lpwfs->download_in_progress != NULL) lpwfs->download_in_progress->session_deleted = TRUE; if (lpwfs->sndSocket != -1) closesocket(lpwfs->sndSocket); if (lpwfs->lstnSocket != -1) closesocket(lpwfs->lstnSocket); if (lpwfs->pasvSocket != -1) closesocket(lpwfs->pasvSocket); INTERNET_SendCallback(&lpwfs->hdr, lpwfs->hdr.dwContext, INTERNET_STATUS_CONNECTION_CLOSED, 0, 0); } static DWORD FTPSESSION_QueryOption(object_header_t *hdr, DWORD option, void *buffer, DWORD *size, BOOL unicode) { switch(option) { case INTERNET_OPTION_HANDLE_TYPE: TRACE("INTERNET_OPTION_HANDLE_TYPE\n"); if (*size < sizeof(ULONG)) return ERROR_INSUFFICIENT_BUFFER; *size = sizeof(DWORD); *(DWORD*)buffer = INTERNET_HANDLE_TYPE_CONNECT_FTP; return ERROR_SUCCESS; } return INET_QueryOption(hdr, option, buffer, size, unicode); } static const object_vtbl_t FTPSESSIONVtbl = { FTPSESSION_Destroy, FTPSESSION_CloseConnection, FTPSESSION_QueryOption, INET_SetOption, NULL, NULL, NULL, NULL }; /*********************************************************************** * FTP_Connect (internal) * * Connect to a ftp server * * RETURNS * HINTERNET a session handle on success * NULL on failure * * NOTES: * * Windows uses 'anonymous' as the username, when given a NULL username * and a NULL password. The password is first looked up in: * * HKCU\Software\Microsoft\Windows\CurrentVersion\Internet Settings\EmailName * * If this entry is not present it uses the current username as the password. * */ HINTERNET FTP_Connect(appinfo_t *hIC, LPCWSTR lpszServerName, INTERNET_PORT nServerPort, LPCWSTR lpszUserName, LPCWSTR lpszPassword, DWORD dwFlags, DWORD_PTR dwContext, DWORD dwInternalFlags) { static const WCHAR szKey[] = {'S','o','f','t','w','a','r','e','\\', 'M','i','c','r','o','s','o','f','t','\\', 'W','i','n','d','o','w','s','\\', 'C','u','r','r','e','n','t','V','e','r','s','i','o','n','\\', 'I','n','t','e','r','n','e','t',' ','S','e','t','t','i','n','g','s',0}; static const WCHAR szValue[] = {'E','m','a','i','l','N','a','m','e',0}; static const WCHAR szDefaultUsername[] = {'a','n','o','n','y','m','o','u','s','\0'}; static const WCHAR szEmpty[] = {'\0'}; struct sockaddr_in socketAddr; INT nsocket = -1; socklen_t sock_namelen; BOOL bSuccess = FALSE; ftp_session_t *lpwfs = NULL; char szaddr[INET6_ADDRSTRLEN]; TRACE("%p Server(%s) Port(%d) User(%s) Paswd(%s)\n", hIC, debugstr_w(lpszServerName), nServerPort, debugstr_w(lpszUserName), debugstr_w(lpszPassword)); assert( hIC->hdr.htype == WH_HINIT ); if ((!lpszUserName || !*lpszUserName) && lpszPassword && *lpszPassword) { INTERNET_SetLastError(ERROR_INVALID_PARAMETER); return NULL; } lpwfs = alloc_object(&hIC->hdr, &FTPSESSIONVtbl, sizeof(ftp_session_t)); if (NULL == lpwfs) { INTERNET_SetLastError(ERROR_OUTOFMEMORY); return NULL; } if (nServerPort == INTERNET_INVALID_PORT_NUMBER) lpwfs->serverport = INTERNET_DEFAULT_FTP_PORT; else lpwfs->serverport = nServerPort; lpwfs->hdr.htype = WH_HFTPSESSION; lpwfs->hdr.dwFlags = dwFlags; lpwfs->hdr.dwContext = dwContext; lpwfs->hdr.dwInternalFlags |= dwInternalFlags; lpwfs->download_in_progress = NULL; lpwfs->sndSocket = -1; lpwfs->lstnSocket = -1; lpwfs->pasvSocket = -1; WININET_AddRef( &hIC->hdr ); lpwfs->lpAppInfo = hIC; list_add_head( &hIC->hdr.children, &lpwfs->hdr.entry ); if(hIC->proxy && hIC->accessType == INTERNET_OPEN_TYPE_PROXY) { if(wcschr(hIC->proxy, ' ')) FIXME("Several proxies not implemented.\n"); if(hIC->proxyBypass) FIXME("Proxy bypass is ignored.\n"); } if (!lpszUserName || !lpszUserName[0]) { HKEY key; WCHAR szPassword[MAX_PATH]; DWORD len = sizeof(szPassword); lpwfs->lpszUserName = heap_strdupW(szDefaultUsername); RegOpenKeyW(HKEY_CURRENT_USER, szKey, &key); if (RegQueryValueExW(key, szValue, NULL, NULL, (LPBYTE)szPassword, &len)) { /* Nothing in the registry, get the username and use that as the password */ if (!GetUserNameW(szPassword, &len)) { /* Should never get here, but use an empty password as failsafe */ lstrcpyW(szPassword, szEmpty); } } RegCloseKey(key); TRACE("Password used for anonymous ftp : (%s)\n", debugstr_w(szPassword)); lpwfs->lpszPassword = heap_strdupW(szPassword); } else { lpwfs->lpszUserName = heap_strdupW(lpszUserName); lpwfs->lpszPassword = heap_strdupW(lpszPassword ? lpszPassword : szEmpty); } lpwfs->servername = heap_strdupW(lpszServerName); /* Don't send a handle created callback if this handle was created with InternetOpenUrl */ if (!(lpwfs->hdr.dwInternalFlags & INET_OPENURL)) { INTERNET_ASYNC_RESULT iar; iar.dwResult = (DWORD_PTR)lpwfs->hdr.hInternet; iar.dwError = ERROR_SUCCESS; INTERNET_SendCallback(&hIC->hdr, dwContext, INTERNET_STATUS_HANDLE_CREATED, &iar, sizeof(INTERNET_ASYNC_RESULT)); } INTERNET_SendCallback(&hIC->hdr, dwContext, INTERNET_STATUS_RESOLVING_NAME, (LPWSTR) lpszServerName, (lstrlenW(lpszServerName)+1) * sizeof(WCHAR)); sock_namelen = sizeof(socketAddr); if (!GetAddress(lpszServerName, lpwfs->serverport, (struct sockaddr *)&socketAddr, &sock_namelen, szaddr)) { INTERNET_SetLastError(ERROR_INTERNET_NAME_NOT_RESOLVED); goto lerror; } if (socketAddr.sin_family != AF_INET) { WARN("unsupported address family %d\n", socketAddr.sin_family); INTERNET_SetLastError(ERROR_INTERNET_CANNOT_CONNECT); goto lerror; } INTERNET_SendCallback(&hIC->hdr, dwContext, INTERNET_STATUS_NAME_RESOLVED, szaddr, strlen(szaddr)+1); init_winsock(); nsocket = socket(AF_INET,SOCK_STREAM,0); if (nsocket == -1) { INTERNET_SetLastError(ERROR_INTERNET_CANNOT_CONNECT); goto lerror; } INTERNET_SendCallback(&hIC->hdr, dwContext, INTERNET_STATUS_CONNECTING_TO_SERVER, szaddr, strlen(szaddr)+1); if (connect(nsocket, (struct sockaddr *)&socketAddr, sock_namelen) < 0) { ERR("Unable to connect (%d)\n", WSAGetLastError()); INTERNET_SetLastError(ERROR_INTERNET_CANNOT_CONNECT); closesocket(nsocket); } else { TRACE("Connected to server\n"); lpwfs->sndSocket = nsocket; INTERNET_SendCallback(&hIC->hdr, dwContext, INTERNET_STATUS_CONNECTED_TO_SERVER, szaddr, strlen(szaddr)+1); sock_namelen = sizeof(lpwfs->socketAddress); getsockname(nsocket, (struct sockaddr *) &lpwfs->socketAddress, &sock_namelen); if (FTP_ConnectToHost(lpwfs)) { TRACE("Successfully logged into server\n"); bSuccess = TRUE; } } lerror: if (!bSuccess) { WININET_Release(&lpwfs->hdr); return NULL; } return lpwfs->hdr.hInternet; } /*********************************************************************** * FTP_ConnectToHost (internal) * * Connect to a ftp server * * RETURNS * TRUE on success * NULL on failure * */ static BOOL FTP_ConnectToHost(ftp_session_t *lpwfs) { INT nResCode; BOOL bSuccess = FALSE; TRACE("\n"); FTP_ReceiveResponse(lpwfs, lpwfs->hdr.dwContext); if (!FTP_SendCommand(lpwfs->sndSocket, FTP_CMD_USER, lpwfs->lpszUserName, 0, 0, 0)) goto lend; nResCode = FTP_ReceiveResponse(lpwfs, lpwfs->hdr.dwContext); if (nResCode) { /* Login successful... */ if (nResCode == 230) bSuccess = TRUE; /* User name okay, need password... */ else if (nResCode == 331) bSuccess = FTP_SendPassword(lpwfs); /* Need account for login... */ else if (nResCode == 332) bSuccess = FTP_SendAccount(lpwfs); else FTP_SetResponseError(nResCode); } TRACE("Returning %d\n", bSuccess); lend: return bSuccess; } /*********************************************************************** * FTP_GetNextLine (internal) * * Parse next line in directory string listing * * RETURNS * Pointer to beginning of next line * NULL on failure * */ static LPSTR FTP_GetNextLine(INT nSocket, LPDWORD dwLen) { struct timeval tv = {RESPONSE_TIMEOUT,0}; FD_SET set; INT nRecv = 0; LPSTR lpszBuffer = INTERNET_GetResponseBuffer(); TRACE("\n"); FD_ZERO(&set); FD_SET(nSocket, &set); while (nRecv < MAX_REPLY_LEN) { if (select(nSocket+1, &set, NULL, NULL, &tv) > 0) { if (sock_recv(nSocket, &lpszBuffer[nRecv], 1, 0) <= 0) { INTERNET_SetLastError(ERROR_FTP_TRANSFER_IN_PROGRESS); return NULL; } if (lpszBuffer[nRecv] == '\n') { lpszBuffer[nRecv] = '\0'; *dwLen = nRecv - 1; TRACE(":%d %s\n", nRecv, lpszBuffer); return lpszBuffer; } if (lpszBuffer[nRecv] != '\r') nRecv++; } else { INTERNET_SetLastError(ERROR_INTERNET_TIMEOUT); return NULL; } } return NULL; } /*********************************************************************** * FTP_SendCommandA (internal) * * Send command to server * * RETURNS * TRUE on success * NULL on failure * */ static BOOL FTP_SendCommandA(INT nSocket, FTP_COMMAND ftpCmd, LPCSTR lpszParam, INTERNET_STATUS_CALLBACK lpfnStatusCB, object_header_t *hdr, DWORD_PTR dwContext) { DWORD len; CHAR *buf; DWORD nBytesSent = 0; int nRC = 0; DWORD dwParamLen; TRACE("%d: (%s) %d\n", ftpCmd, debugstr_a(lpszParam), nSocket); if (lpfnStatusCB) { lpfnStatusCB(hdr->hInternet, dwContext, INTERNET_STATUS_SENDING_REQUEST, NULL, 0); } dwParamLen = lpszParam?strlen(lpszParam)+1:0; len = dwParamLen + strlen(szFtpCommands[ftpCmd]) + strlen(szCRLF); if (NULL == (buf = heap_alloc(len+1))) { INTERNET_SetLastError(ERROR_OUTOFMEMORY); return FALSE; } sprintf(buf, "%s%s%s%s", szFtpCommands[ftpCmd], dwParamLen ? " " : "", dwParamLen ? lpszParam : "", szCRLF); TRACE("Sending (%s) len(%d)\n", debugstr_a(buf), len); while((nBytesSent < len) && (nRC != -1)) { nRC = sock_send(nSocket, buf+nBytesSent, len - nBytesSent, 0); nBytesSent += nRC; } heap_free(buf); if (lpfnStatusCB) { lpfnStatusCB(hdr->hInternet, dwContext, INTERNET_STATUS_REQUEST_SENT, &nBytesSent, sizeof(DWORD)); } TRACE("Sent %d bytes\n", nBytesSent); return (nRC != -1); } /*********************************************************************** * FTP_SendCommand (internal) * * Send command to server * * RETURNS * TRUE on success * NULL on failure * */ static BOOL FTP_SendCommand(INT nSocket, FTP_COMMAND ftpCmd, LPCWSTR lpszParam, INTERNET_STATUS_CALLBACK lpfnStatusCB, object_header_t *hdr, DWORD_PTR dwContext) { BOOL ret; LPSTR lpszParamA = heap_strdupWtoA(lpszParam); ret = FTP_SendCommandA(nSocket, ftpCmd, lpszParamA, lpfnStatusCB, hdr, dwContext); heap_free(lpszParamA); return ret; } /*********************************************************************** * FTP_ReceiveResponse (internal) * * Receive response from server * * RETURNS * Reply code on success * 0 on failure * */ INT FTP_ReceiveResponse(ftp_session_t *lpwfs, DWORD_PTR dwContext) { LPSTR lpszResponse = INTERNET_GetResponseBuffer(); DWORD nRecv; INT rc = 0; char firstprefix[5]; BOOL multiline = FALSE; TRACE("socket(%d)\n", lpwfs->sndSocket); INTERNET_SendCallback(&lpwfs->hdr, dwContext, INTERNET_STATUS_RECEIVING_RESPONSE, NULL, 0); while(1) { if (!FTP_GetNextLine(lpwfs->sndSocket, &nRecv)) goto lerror; if (nRecv >= 3) { if(!multiline) { if(lpszResponse[3] != '-') break; else { /* Start of multiline response. Loop until we get "nnn " */ multiline = TRUE; memcpy(firstprefix, lpszResponse, 3); firstprefix[3] = ' '; firstprefix[4] = '\0'; } } else { if(!memcmp(firstprefix, lpszResponse, 4)) break; } } } if (nRecv >= 3) { rc = atoi(lpszResponse); INTERNET_SendCallback(&lpwfs->hdr, dwContext, INTERNET_STATUS_RESPONSE_RECEIVED, &nRecv, sizeof(DWORD)); } lerror: TRACE("return %d\n", rc); return rc; } /*********************************************************************** * FTP_SendPassword (internal) * * Send password to ftp server * * RETURNS * TRUE on success * NULL on failure * */ static BOOL FTP_SendPassword(ftp_session_t *lpwfs) { INT nResCode; BOOL bSuccess = FALSE; TRACE("\n"); if (!FTP_SendCommand(lpwfs->sndSocket, FTP_CMD_PASS, lpwfs->lpszPassword, 0, 0, 0)) goto lend; nResCode = FTP_ReceiveResponse(lpwfs, lpwfs->hdr.dwContext); if (nResCode) { TRACE("Received reply code %d\n", nResCode); /* Login successful... */ if (nResCode == 230) bSuccess = TRUE; /* Command not implemented, superfluous at the server site... */ /* Need account for login... */ else if (nResCode == 332) bSuccess = FTP_SendAccount(lpwfs); else FTP_SetResponseError(nResCode); } lend: TRACE("Returning %d\n", bSuccess); return bSuccess; } /*********************************************************************** * FTP_SendAccount (internal) * * * * RETURNS * TRUE on success * FALSE on failure * */ static BOOL FTP_SendAccount(ftp_session_t *lpwfs) { INT nResCode; BOOL bSuccess = FALSE; TRACE("\n"); if (!FTP_SendCommand(lpwfs->sndSocket, FTP_CMD_ACCT, szNoAccount, 0, 0, 0)) goto lend; nResCode = FTP_ReceiveResponse(lpwfs, lpwfs->hdr.dwContext); if (nResCode) bSuccess = TRUE; else FTP_SetResponseError(nResCode); lend: return bSuccess; } /*********************************************************************** * FTP_SendStore (internal) * * Send request to upload file to ftp server * * RETURNS * TRUE on success * FALSE on failure * */ static BOOL FTP_SendStore(ftp_session_t *lpwfs, LPCWSTR lpszRemoteFile, DWORD dwType) { INT nResCode; BOOL bSuccess = FALSE; TRACE("\n"); if (!FTP_InitListenSocket(lpwfs)) goto lend; if (!FTP_SendType(lpwfs, dwType)) goto lend; if (!FTP_SendPortOrPasv(lpwfs)) goto lend; if (!FTP_SendCommand(lpwfs->sndSocket, FTP_CMD_STOR, lpszRemoteFile, 0, 0, 0)) goto lend; nResCode = FTP_ReceiveResponse(lpwfs, lpwfs->hdr.dwContext); if (nResCode) { if (nResCode == 150 || nResCode == 125) bSuccess = TRUE; else FTP_SetResponseError(nResCode); } lend: if (!bSuccess && lpwfs->lstnSocket != -1) { closesocket(lpwfs->lstnSocket); lpwfs->lstnSocket = -1; } return bSuccess; } /*********************************************************************** * FTP_InitListenSocket (internal) * * Create a socket to listen for server response * * RETURNS * TRUE on success * FALSE on failure * */ static BOOL FTP_InitListenSocket(ftp_session_t *lpwfs) { BOOL bSuccess = FALSE; socklen_t namelen = sizeof(lpwfs->lstnSocketAddress); TRACE("\n"); init_winsock(); lpwfs->lstnSocket = socket(AF_INET, SOCK_STREAM, 0); if (lpwfs->lstnSocket == -1) { TRACE("Unable to create listening socket\n"); goto lend; } /* We obtain our ip addr from the name of the command channel socket */ lpwfs->lstnSocketAddress = lpwfs->socketAddress; /* and get the system to assign us a port */ lpwfs->lstnSocketAddress.sin_port = htons(0); if (bind(lpwfs->lstnSocket,(struct sockaddr *) &lpwfs->lstnSocketAddress, sizeof(lpwfs->lstnSocketAddress)) == -1) { TRACE("Unable to bind socket\n"); goto lend; } if (listen(lpwfs->lstnSocket, MAX_BACKLOG) == -1) { TRACE("listen failed\n"); goto lend; } if (getsockname(lpwfs->lstnSocket, (struct sockaddr *) &lpwfs->lstnSocketAddress, &namelen) != -1) bSuccess = TRUE; lend: if (!bSuccess && lpwfs->lstnSocket != -1) { closesocket(lpwfs->lstnSocket); lpwfs->lstnSocket = -1; } return bSuccess; } /*********************************************************************** * FTP_SendType (internal) * * Tell server type of data being transferred * * RETURNS * TRUE on success * FALSE on failure * * W98SE doesn't cache the type that's currently set * (i.e. it sends it always), * so we probably don't want to do that either. */ static BOOL FTP_SendType(ftp_session_t *lpwfs, DWORD dwType) { INT nResCode; WCHAR type[] = { 'I','\0' }; BOOL bSuccess = FALSE; TRACE("\n"); if (dwType & INTERNET_FLAG_TRANSFER_ASCII) type[0] = 'A'; if (!FTP_SendCommand(lpwfs->sndSocket, FTP_CMD_TYPE, type, 0, 0, 0)) goto lend; nResCode = FTP_ReceiveResponse(lpwfs, lpwfs->hdr.dwContext)/100; if (nResCode) { if (nResCode == 2) bSuccess = TRUE; else FTP_SetResponseError(nResCode); } lend: return bSuccess; } #if 0 /* FIXME: should probably be used for FtpGetFileSize */ /*********************************************************************** * FTP_GetFileSize (internal) * * Retrieves from the server the size of the given file * * RETURNS * TRUE on success * FALSE on failure * */ static BOOL FTP_GetFileSize(ftp_session_t *lpwfs, LPCWSTR lpszRemoteFile, DWORD *dwSize) { INT nResCode; BOOL bSuccess = FALSE; TRACE("\n"); if (!FTP_SendCommand(lpwfs->sndSocket, FTP_CMD_SIZE, lpszRemoteFile, 0, 0, 0)) goto lend; nResCode = FTP_ReceiveResponse(lpwfs, lpwfs->hdr.dwContext); if (nResCode) { if (nResCode == 213) { /* Now parses the output to get the actual file size */ int i; LPSTR lpszResponseBuffer = INTERNET_GetResponseBuffer(); for (i = 0; (lpszResponseBuffer[i] != ' ') && (lpszResponseBuffer[i] != '\0'); i++) ; if (lpszResponseBuffer[i] == '\0') return FALSE; *dwSize = atol(&(lpszResponseBuffer[i + 1])); bSuccess = TRUE; } else { FTP_SetResponseError(nResCode); } } lend: return bSuccess; } #endif /*********************************************************************** * FTP_SendPort (internal) * * Tell server which port to use * * RETURNS * TRUE on success * FALSE on failure * */ static BOOL FTP_SendPort(ftp_session_t *lpwfs) { static const WCHAR szIPFormat[] = {'%','d',',','%','d',',','%','d',',','%','d',',','%','d',',','%','d','\0'}; INT nResCode; WCHAR szIPAddress[64]; BOOL bSuccess = FALSE; TRACE("\n"); swprintf(szIPAddress, ARRAY_SIZE(szIPAddress), szIPFormat, lpwfs->lstnSocketAddress.sin_addr.S_un.S_addr&0x000000FF, (lpwfs->lstnSocketAddress.sin_addr.S_un.S_addr&0x0000FF00)>>8, (lpwfs->lstnSocketAddress.sin_addr.S_un.S_addr&0x00FF0000)>>16, (lpwfs->lstnSocketAddress.sin_addr.S_un.S_addr&0xFF000000)>>24, lpwfs->lstnSocketAddress.sin_port & 0xFF, (lpwfs->lstnSocketAddress.sin_port & 0xFF00)>>8); if (!FTP_SendCommand(lpwfs->sndSocket, FTP_CMD_PORT, szIPAddress, 0, 0, 0)) goto lend; nResCode = FTP_ReceiveResponse(lpwfs, lpwfs->hdr.dwContext); if (nResCode) { if (nResCode == 200) bSuccess = TRUE; else FTP_SetResponseError(nResCode); } lend: return bSuccess; } /*********************************************************************** * FTP_DoPassive (internal) * * Tell server that we want to do passive transfers * and connect data socket * * RETURNS * TRUE on success * FALSE on failure * */ static BOOL FTP_DoPassive(ftp_session_t *lpwfs) { INT nResCode; BOOL bSuccess = FALSE; TRACE("\n"); if (!FTP_SendCommand(lpwfs->sndSocket, FTP_CMD_PASV, NULL, 0, 0, 0)) goto lend; nResCode = FTP_ReceiveResponse(lpwfs, lpwfs->hdr.dwContext); if (nResCode) { if (nResCode == 227) { LPSTR lpszResponseBuffer = INTERNET_GetResponseBuffer(); LPSTR p; int f[6]; int i; char *pAddr, *pPort; INT nsocket = -1; struct sockaddr_in dataSocketAddress; p = lpszResponseBuffer+4; /* skip status code */ while (*p != '\0' && (*p < '0' || *p > '9')) p++; if (*p == '\0') { ERR("no address found in response, aborting\n"); goto lend; } if (sscanf(p, "%d,%d,%d,%d,%d,%d", &f[0], &f[1], &f[2], &f[3], &f[4], &f[5]) != 6) { ERR("unknown response address format '%s', aborting\n", p); goto lend; } for (i=0; i < 6; i++) f[i] = f[i] & 0xff; dataSocketAddress = lpwfs->socketAddress; pAddr = (char *)&(dataSocketAddress.sin_addr.S_un.S_addr); pPort = (char *)&(dataSocketAddress.sin_port); pAddr[0] = f[0]; pAddr[1] = f[1]; pAddr[2] = f[2]; pAddr[3] = f[3]; pPort[0] = f[4]; pPort[1] = f[5]; nsocket = socket(AF_INET,SOCK_STREAM,0); if (nsocket == -1) goto lend; if (connect(nsocket, (struct sockaddr *)&dataSocketAddress, sizeof(dataSocketAddress))) { ERR("can't connect passive FTP data port.\n"); closesocket(nsocket); goto lend; } lpwfs->pasvSocket = nsocket; bSuccess = TRUE; } else FTP_SetResponseError(nResCode); } lend: return bSuccess; } static BOOL FTP_SendPortOrPasv(ftp_session_t *lpwfs) { if (lpwfs->hdr.dwFlags & INTERNET_FLAG_PASSIVE) { if (!FTP_DoPassive(lpwfs)) return FALSE; } else { if (!FTP_SendPort(lpwfs)) return FALSE; } return TRUE; } /*********************************************************************** * FTP_GetDataSocket (internal) * * Either accepts an incoming data socket connection from the server * or just returns the already opened socket after a PASV command * in case of passive FTP. * * * RETURNS * TRUE on success * FALSE on failure * */ static BOOL FTP_GetDataSocket(ftp_session_t *lpwfs, LPINT nDataSocket) { struct sockaddr_in saddr; socklen_t addrlen = sizeof(saddr); TRACE("\n"); if (lpwfs->hdr.dwFlags & INTERNET_FLAG_PASSIVE) { *nDataSocket = lpwfs->pasvSocket; lpwfs->pasvSocket = -1; } else { *nDataSocket = accept(lpwfs->lstnSocket, (struct sockaddr *) &saddr, &addrlen); closesocket(lpwfs->lstnSocket); lpwfs->lstnSocket = -1; } return *nDataSocket != -1; } /*********************************************************************** * FTP_SendData (internal) * * Send data to the server * * RETURNS * TRUE on success * FALSE on failure * */ static BOOL FTP_SendData(ftp_session_t *lpwfs, INT nDataSocket, HANDLE hFile) { BY_HANDLE_FILE_INFORMATION fi; DWORD nBytesRead = 0; DWORD nBytesSent = 0; DWORD nTotalSent = 0; DWORD nBytesToSend, nLen; int nRC = 1; time_t s_long_time, e_long_time; LONG nSeconds; CHAR *lpszBuffer; TRACE("\n"); lpszBuffer = heap_alloc_zero(sizeof(CHAR)*DATA_PACKET_SIZE); /* Get the size of the file. */ GetFileInformationByHandle(hFile, &fi); time(&s_long_time); do { nBytesToSend = nBytesRead - nBytesSent; if (nBytesToSend <= 0) { /* Read data from file. */ nBytesSent = 0; if (!ReadFile(hFile, lpszBuffer, DATA_PACKET_SIZE, &nBytesRead, 0)) ERR("Failed reading from file\n"); if (nBytesRead > 0) nBytesToSend = nBytesRead; else break; } nLen = DATA_PACKET_SIZE < nBytesToSend ? DATA_PACKET_SIZE : nBytesToSend; nRC = sock_send(nDataSocket, lpszBuffer, nLen, 0); if (nRC != -1) { nBytesSent += nRC; nTotalSent += nRC; } /* Do some computation to display the status. */ time(&e_long_time); nSeconds = e_long_time - s_long_time; if( nSeconds / 60 > 0 ) { TRACE( "%d bytes of %d bytes (%d%%) in %d min %d sec estimated remaining time %d sec\n", nTotalSent, fi.nFileSizeLow, nTotalSent*100/fi.nFileSizeLow, nSeconds / 60, nSeconds % 60, (fi.nFileSizeLow - nTotalSent) * nSeconds / nTotalSent ); } else { TRACE( "%d bytes of %d bytes (%d%%) in %d sec estimated remaining time %d sec\n", nTotalSent, fi.nFileSizeLow, nTotalSent*100/fi.nFileSizeLow, nSeconds, (fi.nFileSizeLow - nTotalSent) * nSeconds / nTotalSent); } } while (nRC != -1); TRACE("file transfer complete!\n"); heap_free(lpszBuffer); return nTotalSent; } /*********************************************************************** * FTP_SendRetrieve (internal) * * Send request to retrieve a file * * RETURNS * Number of bytes to be received on success * 0 on failure * */ static BOOL FTP_SendRetrieve(ftp_session_t *lpwfs, LPCWSTR lpszRemoteFile, DWORD dwType) { INT nResCode; BOOL ret; TRACE("\n"); if (!(ret = FTP_InitListenSocket(lpwfs))) goto lend; if (!(ret = FTP_SendType(lpwfs, dwType))) goto lend; if (!(ret = FTP_SendPortOrPasv(lpwfs))) goto lend; if (!(ret = FTP_SendCommand(lpwfs->sndSocket, FTP_CMD_RETR, lpszRemoteFile, 0, 0, 0))) goto lend; nResCode = FTP_ReceiveResponse(lpwfs, lpwfs->hdr.dwContext); if ((nResCode != 125) && (nResCode != 150)) { /* That means that we got an error getting the file. */ FTP_SetResponseError(nResCode); ret = FALSE; } lend: if (!ret && lpwfs->lstnSocket != -1) { closesocket(lpwfs->lstnSocket); lpwfs->lstnSocket = -1; } return ret; } /*********************************************************************** * FTP_RetrieveData (internal) * * Retrieve data from server * * RETURNS * TRUE on success * FALSE on failure * */ static BOOL FTP_RetrieveFileData(ftp_session_t *lpwfs, INT nDataSocket, HANDLE hFile) { DWORD nBytesWritten; INT nRC = 0; CHAR *lpszBuffer; TRACE("\n"); lpszBuffer = heap_alloc_zero(sizeof(CHAR)*DATA_PACKET_SIZE); if (NULL == lpszBuffer) { INTERNET_SetLastError(ERROR_OUTOFMEMORY); return FALSE; } while (nRC != -1) { nRC = sock_recv(nDataSocket, lpszBuffer, DATA_PACKET_SIZE, 0); if (nRC != -1) { /* other side closed socket. */ if (nRC == 0) goto recv_end; WriteFile(hFile, lpszBuffer, nRC, &nBytesWritten, NULL); } } TRACE("Data transfer complete\n"); recv_end: heap_free(lpszBuffer); return (nRC != -1); } /*********************************************************************** * FTPFINDNEXT_Destroy (internal) * * Deallocate session handle */ static void FTPFINDNEXT_Destroy(object_header_t *hdr) { LPWININETFTPFINDNEXTW lpwfn = (LPWININETFTPFINDNEXTW) hdr; DWORD i; TRACE("\n"); WININET_Release(&lpwfn->lpFtpSession->hdr); for (i = 0; i < lpwfn->size; i++) { heap_free(lpwfn->lpafp[i].lpszName); } heap_free(lpwfn->lpafp); } static DWORD FTPFINDNEXT_FindNextFileProc(WININETFTPFINDNEXTW *find, LPVOID data) { WIN32_FIND_DATAW *find_data = data; DWORD res = ERROR_SUCCESS; TRACE("index(%d) size(%d)\n", find->index, find->size); ZeroMemory(find_data, sizeof(WIN32_FIND_DATAW)); if (find->index < find->size) { FTP_ConvertFileProp(&find->lpafp[find->index], find_data); find->index++; TRACE("Name: %s\nSize: %d\n", debugstr_w(find_data->cFileName), find_data->nFileSizeLow); }else { res = ERROR_NO_MORE_FILES; } if (find->hdr.dwFlags & INTERNET_FLAG_ASYNC) { INTERNET_ASYNC_RESULT iar; iar.dwResult = (res == ERROR_SUCCESS); iar.dwError = res; INTERNET_SendCallback(&find->hdr, find->hdr.dwContext, INTERNET_STATUS_REQUEST_COMPLETE, &iar, sizeof(INTERNET_ASYNC_RESULT)); } return res; } typedef struct { task_header_t hdr; WIN32_FIND_DATAW *find_data; } find_next_task_t; static void FTPFINDNEXT_AsyncFindNextFileProc(task_header_t *hdr) { find_next_task_t *task = (find_next_task_t*)hdr; FTPFINDNEXT_FindNextFileProc((WININETFTPFINDNEXTW*)task->hdr.hdr, task->find_data); } static DWORD FTPFINDNEXT_QueryOption(object_header_t *hdr, DWORD option, void *buffer, DWORD *size, BOOL unicode) { switch(option) { case INTERNET_OPTION_HANDLE_TYPE: TRACE("INTERNET_OPTION_HANDLE_TYPE\n"); if (*size < sizeof(ULONG)) return ERROR_INSUFFICIENT_BUFFER; *size = sizeof(DWORD); *(DWORD*)buffer = INTERNET_HANDLE_TYPE_FTP_FIND; return ERROR_SUCCESS; } return INET_QueryOption(hdr, option, buffer, size, unicode); } static DWORD FTPFINDNEXT_FindNextFileW(object_header_t *hdr, void *data) { WININETFTPFINDNEXTW *find = (WININETFTPFINDNEXTW*)hdr; if (find->lpFtpSession->lpAppInfo->hdr.dwFlags & INTERNET_FLAG_ASYNC) { find_next_task_t *task; task = alloc_async_task(&find->hdr, FTPFINDNEXT_AsyncFindNextFileProc, sizeof(*task)); task->find_data = data; INTERNET_AsyncCall(&task->hdr); return ERROR_SUCCESS; } return FTPFINDNEXT_FindNextFileProc(find, data); } static const object_vtbl_t FTPFINDNEXTVtbl = { FTPFINDNEXT_Destroy, NULL, FTPFINDNEXT_QueryOption, INET_SetOption, NULL, NULL, NULL, FTPFINDNEXT_FindNextFileW }; /*********************************************************************** * FTP_ReceiveFileList (internal) * * Read file list from server * * RETURNS * Handle to file list on success * NULL on failure * */ static HINTERNET FTP_ReceiveFileList(ftp_session_t *lpwfs, INT nSocket, LPCWSTR lpszSearchFile, LPWIN32_FIND_DATAW lpFindFileData, DWORD_PTR dwContext) { DWORD dwSize = 0; LPFILEPROPERTIESW lpafp = NULL; LPWININETFTPFINDNEXTW lpwfn = NULL; TRACE("(%p,%d,%s,%p,%08lx)\n", lpwfs, nSocket, debugstr_w(lpszSearchFile), lpFindFileData, dwContext); if (FTP_ParseDirectory(lpwfs, nSocket, lpszSearchFile, &lpafp, &dwSize)) { if(lpFindFileData) FTP_ConvertFileProp(lpafp, lpFindFileData); lpwfn = alloc_object(&lpwfs->hdr, &FTPFINDNEXTVtbl, sizeof(WININETFTPFINDNEXTW)); if (lpwfn) { lpwfn->hdr.htype = WH_HFTPFINDNEXT; lpwfn->hdr.dwContext = dwContext; lpwfn->index = 1; /* Next index is 1 since we return index 0 */ lpwfn->size = dwSize; lpwfn->lpafp = lpafp; WININET_AddRef( &lpwfs->hdr ); lpwfn->lpFtpSession = lpwfs; list_add_head( &lpwfs->hdr.children, &lpwfn->hdr.entry ); } } TRACE("Matched %d files\n", dwSize); return lpwfn ? lpwfn->hdr.hInternet : NULL; } /*********************************************************************** * FTP_ConvertFileProp (internal) * * Converts FILEPROPERTIESW struct to WIN32_FIND_DATAA * * RETURNS * TRUE on success * FALSE on failure * */ static BOOL FTP_ConvertFileProp(LPFILEPROPERTIESW lpafp, LPWIN32_FIND_DATAW lpFindFileData) { BOOL bSuccess = FALSE; ZeroMemory(lpFindFileData, sizeof(WIN32_FIND_DATAW)); if (lpafp) { SystemTimeToFileTime( &lpafp->tmLastModified, &lpFindFileData->ftLastAccessTime ); lpFindFileData->ftLastWriteTime = lpFindFileData->ftLastAccessTime; lpFindFileData->ftCreationTime = lpFindFileData->ftLastAccessTime; /* Not all fields are filled in */ lpFindFileData->nFileSizeHigh = 0; /* We do not handle files bigger than 0xFFFFFFFF bytes yet :-) */ lpFindFileData->nFileSizeLow = lpafp->nSize; if (lpafp->bIsDirectory) lpFindFileData->dwFileAttributes |= FILE_ATTRIBUTE_DIRECTORY; if (lpafp->lpszName) lstrcpynW(lpFindFileData->cFileName, lpafp->lpszName, MAX_PATH); bSuccess = TRUE; } return bSuccess; } /*********************************************************************** * FTP_ParseNextFile (internal) * * Parse the next line in file listing * * RETURNS * TRUE on success * FALSE on failure */ static BOOL FTP_ParseNextFile(INT nSocket, LPCWSTR lpszSearchFile, LPFILEPROPERTIESW lpfp) { static const char szSpace[] = " \t"; DWORD nBufLen; char *pszLine; char *pszToken; char *pszTmp; BOOL found = FALSE; int i; lpfp->lpszName = NULL; do { if(!(pszLine = FTP_GetNextLine(nSocket, &nBufLen))) return FALSE; pszToken = strtok(pszLine, szSpace); /* ls format *