void FileTransfer::CreateFileTransferDialog()
{
m_hwndFileTransfer = CreateDialog(m_pApp->m_instance,
MAKEINTRESOURCE(IDD_FILETRANSFER_DLG),
NULL, (DLGPROC)FileTransferDlgProc);
VNCviewerApp32 *pApp = (VNCviewerApp32 *)(m_clientconn->m_pApp);
pApp->AddModelessDialog(m_hwndFileTransfer);
ShowWindow(m_hwndFileTransfer, SW_SHOW);
UpdateWindow(m_hwndFileTransfer);
SetWindowLongPtr(m_hwndFileTransfer, GWLP_USERDATA, (LONG_PTR) this);
m_hwndFTProgress = GetDlgItem(m_hwndFileTransfer, IDC_FTPROGRESS);
m_hwndFTClientList = GetDlgItem(m_hwndFileTransfer, IDC_FTCLIENTLIST);
m_hwndFTServerList = GetDlgItem(m_hwndFileTransfer, IDC_FTSERVERLIST);
m_hwndFTClientPath = GetDlgItem(m_hwndFileTransfer, IDC_CLIENTPATH);
m_hwndFTServerPath = GetDlgItem(m_hwndFileTransfer, IDC_SERVERPATH);
m_hwndFTStatus = GetDlgItem(m_hwndFileTransfer, IDC_FTSTATUS);
ListView_SetExtendedListViewStyleEx(m_hwndFTClientList, LVS_EX_FULLROWSELECT,
LVS_EX_FULLROWSELECT);
ListView_SetExtendedListViewStyleEx(m_hwndFTServerList, LVS_EX_FULLROWSELECT,
LVS_EX_FULLROWSELECT);
HANDLE hIcon = LoadImage(m_pApp->m_instance, MAKEINTRESOURCE(IDI_FILEUP),
IMAGE_ICON, 16, 16, LR_SHARED);
SendMessage(GetDlgItem(m_hwndFileTransfer, IDC_CLIENTUP), BM_SETIMAGE,
(WPARAM)IMAGE_ICON, (LPARAM)hIcon);
SendMessage(GetDlgItem(m_hwndFileTransfer, IDC_SERVERUP), BM_SETIMAGE,
(WPARAM)IMAGE_ICON, (LPARAM)hIcon);
DestroyIcon((HICON)hIcon);
hIcon = LoadImage(m_pApp->m_instance, MAKEINTRESOURCE(IDI_FILERELOAD),
IMAGE_ICON, 16, 16, LR_SHARED);
SendMessage(GetDlgItem(m_hwndFileTransfer, IDC_CLIENTRELOAD), BM_SETIMAGE,
(WPARAM)IMAGE_ICON, (LPARAM)hIcon);
SendMessage(GetDlgItem(m_hwndFileTransfer, IDC_SERVERRELOAD), BM_SETIMAGE,
(WPARAM)IMAGE_ICON, (LPARAM)hIcon);
DestroyIcon((HICON)hIcon);
RECT Rect;
GetClientRect(m_hwndFTClientList, &Rect);
int xwidth = (int)(0.7 * Rect.right);
int xwidth_ = (int)(0.25 * Rect.right);
FTInsertColumn(m_hwndFTClientList, "Name", 0, xwidth);
FTInsertColumn(m_hwndFTClientList, "Size", 1, xwidth_);
FTInsertColumn(m_hwndFTServerList, "Name", 0, xwidth);
FTInsertColumn(m_hwndFTServerList, "Size", 1, xwidth_);
ShowClientItems(m_ClientPathTmp);
SendFileListRequestMessage(m_ServerPathTmp, 0);
}
void
FileTransfer::ProcessListViewDBLCLK(HWND hwnd, char *Path, char *PathTmp, int iItem)
{
SendMessage(m_hwndFTProgress, PBM_SETPOS, 0, 0);
SetWindowText(m_hwndFTStatus, "");
strcpy(PathTmp, Path);
char buffer[rfbMAX_PATH];
char buffer_tmp[16];
ListView_GetItemText(hwnd, iItem, 0, buffer, rfbMAX_PATH);
ListView_GetItemText(hwnd, iItem, 1, buffer_tmp, 16);
if (strcmp(buffer_tmp, m_FTClientItemInfo.folderText) == 0) {
BlockingFileTransferDialog(FALSE);
if (strlen(PathTmp) >= 2) strcat(PathTmp, "\\");
strcat(PathTmp, buffer);
if (hwnd == m_hwndFTClientList) ShowClientItems(PathTmp);
if (hwnd == m_hwndFTServerList) SendFileListRequestMessage(PathTmp, 0);
}
}
void
FileTransfer::FileTransferDownload()
{
rfbFileDownloadDataMsg fdd;
m_clientconn->ReadExact((char *)&fdd, sz_rfbFileDownloadDataMsg);
fdd.realSize = Swap16IfLE(fdd.realSize);
fdd.compressedSize = Swap16IfLE(fdd.compressedSize);
char path[rfbMAX_PATH + rfbMAX_PATH + 3];
if (m_bFirstFileDownloadMsg) {
m_dwDownloadBlockSize = fdd.compressedSize;
sprintf(path, "%s\\%s", m_ClientPath, m_ServerFilename);
strcpy(m_DownloadFilename, path);
m_hFiletoWrite = CreateFile(path, GENERIC_WRITE, FILE_SHARE_READ | FILE_SHARE_WRITE,
NULL, CREATE_ALWAYS, FILE_FLAG_SEQUENTIAL_SCAN, NULL);
int amount = m_sizeDownloadFile / ((m_dwDownloadBlockSize + 1) * 10);
InitProgressBar(0, 0, amount, 1);
m_bFirstFileDownloadMsg = FALSE;
m_bDownloadStarted = TRUE;
}
if ((fdd.realSize == 0) && (fdd.compressedSize == 0)) {
unsigned int mTime;
m_clientconn->ReadExact((char *) &mTime, sizeof(unsigned int));
if (m_hFiletoWrite == INVALID_HANDLE_VALUE) {
CancelDownload("Could not create file");
MessageBox(m_hwndFileTransfer, "Download failed: could not create local file",
"Download Failed", MB_ICONEXCLAMATION | MB_OK);
SendMessage(m_hwndFTProgress, PBM_SETPOS, 0, 0);
// Send message to start download for next selected file.
PostMessage(m_hwndFileTransfer, WM_COMMAND, IDC_FTCOPY, 0);
return;
}
FILETIME Filetime;
Time70ToFiletime(mTime, &Filetime);
SetFileTime(m_hFiletoWrite, &Filetime, &Filetime, &Filetime);
SendMessage(m_hwndFTProgress, PBM_SETPOS, 0, 0);
SetWindowText(m_hwndFTStatus, "");
CloseHandle(m_hFiletoWrite);
ShowClientItems(m_ClientPath);
m_bFirstFileDownloadMsg = TRUE;
m_bDownloadStarted = FALSE;
// Send message to start download for next selected file.
PostMessage(m_hwndFileTransfer, WM_COMMAND, IDC_FTCOPY, 0);
return;
}
char * pBuff = new char [fdd.compressedSize];
DWORD dwNumberOfBytesWritten;
m_clientconn->ReadExact(pBuff, fdd.compressedSize);
ProcessDlgMessage(m_hwndFileTransfer);
if (!m_bTransferEnable) {
CancelDownload("Download cancelled by user");
delete [] pBuff;
return;
}
if (m_hFiletoWrite == INVALID_HANDLE_VALUE) {
CancelDownload("Could not create file");
MessageBox(m_hwndFileTransfer, "Download failed: could not create local file",
"Download Failed", MB_ICONEXCLAMATION | MB_OK);
delete [] pBuff;
return;
}
WriteFile(m_hFiletoWrite, pBuff, fdd.compressedSize, &dwNumberOfBytesWritten, NULL);
m_dwDownloadRead += dwNumberOfBytesWritten;
if (m_dwDownloadRead >= (10 * m_dwDownloadBlockSize)) {
m_dwDownloadRead = 0;
SendMessage(m_hwndFTProgress, PBM_STEPIT, 0, 0);
}
delete [] pBuff;
}
