//--------------------------------------------------------------------------------
void Connection::sendrequest(char *cmd, char *local, char *remote)
{
//??SaveConsoleTitle _title;
sendrequestINT(cmd,local,remote);
IdleMessage(NULL,0);
}
int FTP::GetFindData(PluginPanelItem **pPanelItem, int *pItemsNumber, int OpMode)
{
PROC(("FTP::GetFindData",NULL))
DWORD b,e;
char *Data[3];
*pPanelItem = NULL;
*pItemsNumber = 0;
//Hosts
if(ShowHosts)
{
EnumHost Enum(HostsPath);
FP_SizeItemList il(FALSE);
PluginPanelItem tmp;
FTPHost h;
if(!IS_SILENT(OpMode))
{
memset(&tmp, 0, sizeof(tmp));
strcpy(tmp.FindData.cFileName,"..");
tmp.FindData.dwFileAttributes = FILE_ATTRIBUTE_DIRECTORY;
if(!IS_SILENT(OpMode))
{
tmp.Description = (char *)"..";
tmp.CustomColumnNumber = 3;
tmp.CustomColumnData = Data;
tmp.CustomColumnData[0] = (char *)"..";
tmp.CustomColumnData[1] = (char *)"..";
tmp.CustomColumnData[2] = (char *)"..";
}
if(!il.Add(&tmp))
return FALSE;
}
while(true)
{
if(!Enum.GetNextHost(&h))
break;
if(!h.Read(NULL))
continue;
memset(&tmp, 0, sizeof(tmp));
/* Panel item MUST have name the save as file saved to disk
in case you want to copy between panels work.
*/
h.MkINIFile(tmp.FindData.cFileName,NULL,"");
tmp.FindData.ftLastWriteTime = h.LastWrite;
tmp.FindData.dwFileAttributes = h.Folder ? FILE_ATTRIBUTE_DIRECTORY : 0;
tmp.Flags = PPIF_USERDATA;
tmp.PackSizeHigh = FTP_HOSTID;
tmp.UserData = (DWORD_PTR)&h;
if(!IS_SILENT(OpMode))
{
tmp.Description = h.HostDescr;
tmp.CustomColumnNumber = 3;
tmp.CustomColumnData = Data;
tmp.CustomColumnData[0] = h.Host; //C0
tmp.CustomColumnData[1] = h.Home; //C1
tmp.CustomColumnData[2] = h.User; //C2
}
if(!il.Add(&tmp))
return FALSE;
Log(("Item[%d]=[%s] attr=%08X",
il.Count()-1, FTP_FILENAME(il.Item(il.Count()-1)),
il.Item(il.Count()-1)->FindData.dwFileAttributes));
}
*pPanelItem = il.Items();
*pItemsNumber = il.Count();
return TRUE;
}
//FTP
FP_Screen _scr;
FTPFileInfo FileInfo;
if(!hConnect)
{
goto AskConnect;
}
Restart:
if(!FtpFindFirstFile(hConnect, "*", &FileInfo, &ResetCache))
{
if(GetLastError() == ERROR_NO_MORE_FILES)
{
*pItemsNumber = 0;
return TRUE;
}
if(SwitchingToFTP && GetLastError() == ERROR_CANCELLED)
{
;
}
else
{
if(CurrentState == fcsExpandList)
{
FreeFindData(*pPanelItem,*pItemsNumber);
*pPanelItem = NULL;
*pItemsNumber = 0;
return FALSE;
}
//Query reconnect
do
{
if(!hConnect)
break;
if(GetLastError() == ERROR_CANCELLED)
break;
if(!hConnect->ConnectMessageTimeout(MConnectionLost,Host.HostName,-MRestore))
{
Log(("WaitMessage cancelled"));
break;
}
if(FtpCmdLineAlive(hConnect) &&
FtpKeepAlive(hConnect))
goto Restart;
if(SelectFile.Length() && CurrentState != fcsExpandList)
SaveUsedDirNFile();
AskConnect:
if(Connect())
goto Restart;
else
break;
}
while(true);
}
if(!ShowHosts)
BackToHosts();
FreeFindData(*pPanelItem, *pItemsNumber);
return GetFindData(pPanelItem,pItemsNumber,OpMode);
}
GET_TIME(b);
do
{
if(Opt.ShowIdle)
{
char str[ 200 ];
GET_TIME(e);
if(CMP_TIME(e,b) > 0.5)
{
_snprintf(str,ARRAYSIZE(str),"%s%d", FP_GetMsg(MReaded), *pItemsNumber);
SetLastError(ERROR_SUCCESS);
IdleMessage(str,Opt.ProcessColor);
b = e;
if(CheckForEsc(FALSE))
{
SetLastError(ERROR_CANCELLED);
return FALSE;
}
}
}
PluginPanelItem *NewPanelItem=*pPanelItem;
if((*pItemsNumber % 1024) == 0)
{
if(!NewPanelItem)
NewPanelItem = (PluginPanelItem *)malloc((1024+1)*sizeof(PluginPanelItem));
else
NewPanelItem = (PluginPanelItem *)realloc(NewPanelItem,(*pItemsNumber+1024+1)*sizeof(PluginPanelItem));
if(NewPanelItem == NULL)
{
/*-*/Log(("GetFindData(file)::!reallocate plugin panels items %d -> %d",*pItemsNumber,*pItemsNumber+1024+1));
return FALSE;
}
*pPanelItem=NewPanelItem;
}
PluginPanelItem *CurItem = &NewPanelItem[*pItemsNumber];
memset(CurItem, 0, sizeof(PluginPanelItem));
CurItem->FindData = FileInfo.FindData;
if(!IS_SILENT(OpMode))
{
CurItem->CustomColumnNumber = FTP_COL_MAX;
CurItem->Owner = FileInfo.FTPOwner[0] ? strdup(FileInfo.FTPOwner) : NULL;
CurItem->CustomColumnData = (LPSTR*)malloc(sizeof(LPSTR*)*FTP_COL_MAX);
CurItem->CustomColumnData[FTP_COL_MODE] = strdup(FileInfo.UnixMode);
CurItem->CustomColumnData[FTP_COL_LINK] = strdup(FileInfo.Link);
hConnect->ToOEM(CurItem->CustomColumnData[FTP_COL_LINK]);
}
(*pItemsNumber)++;
}
while(FtpFindNextFile(hConnect,&FileInfo));
return TRUE;
}
void Connection::recvrequestINT(char *cmd, char *local, char *remote, const char *mode)
{
int oldtype = 0,
is_retr;
FHandle fout;
SOCKET din = INVALID_SOCKET;
int ocode, oecode;
BOOL oldBrk = FtpSetBreakable(this, -1);
FTPCurrentStates oState = CurrentState;
FTNNotify ni;
if(type == TYPE_A)
restart_point = 0;
ni.Upload = FALSE;
ni.Starting = TRUE;
ni.Success = TRUE;
ni.RestartPoint = restart_point;
ni.Port = ntohs(portnum);
ni.Password[0] = 0; //StrCpy( ni.Password, UserPassword, ARRAYSIZE(ni.Password));
StrCpy(ni.User, UserName, ARRAYSIZE(ni.User));
StrCpy(ni.HostName, hostname, ARRAYSIZE(ni.HostName));
StrCpy(ni.LocalFile, local, ARRAYSIZE(ni.LocalFile));
StrCpy(ni.RemoteFile, remote, ARRAYSIZE(ni.RemoteFile));
if(local[0] == '-' && local[1] == 0)
{
;
}
else
{
fout.Handle = Fopen(local, mode, Opt.SetHiddenOnAbort ? FILE_ATTRIBUTE_HIDDEN : FILE_ATTRIBUTE_NORMAL);
if(!fout.Handle)
{
ErrorCode = GetLastError();
SysError = TRUE;
Log(("!Fopen [%s] %s",mode,__WINError()));
if(!ConnectMessage(MErrorOpenFile,local,-MRetry))
ErrorCode = ERROR_CANCELLED;
//goto abort;
return;
}
Log(("recv file [%d] \"%s\"=%p",Host.IOBuffSize,local,fout.Handle));
if(restart_point != -1)
{
if(!Fmove(fout.Handle,restart_point))
{
ErrorCode = GetLastError();
SysError = TRUE;
if(!ConnectMessage(MErrorPosition,local,-MRetry))
ErrorCode = ERROR_CANCELLED;
return;
}
}
TrafficInfo->Resume(restart_point == -1 ? 0 : restart_point);
}
is_retr = StrCmp(cmd,Opt.cmdRetr) == 0;
if(proxy && is_retr)
{
proxtrans(cmd, local, remote);
return;
}
if(!initconn())
{
Log(("!initconn"));
return;
}
if(!is_retr)
{
if(type != TYPE_A)
{
oldtype = type;
setascii();
}
}
else if(restart_point)
{
if(!ResumeSupport)
{
AddCmdLine(FMSG(MResumeRestart));
restart_point = 0;
}
else if(restart_point != -1)
{
if(command("%s %I64u",Opt.cmdRest,restart_point) != RPL_CONTINUE)
{
Log(("!restart SIZE"));
return;
}
}
}
if(Host.PassiveMode)
{
din = dataconn();
if(din == INVALID_SOCKET)
{
Log(("!dataconn: PASV ent"));
goto abort;
}
}
if(remote)
{
if(command("%s %s", cmd, remote) != RPL_PRELIM)
{
if(oldtype)
SetType(oldtype);
Log(("!command [%s]",cmd));
fout.Close();
if(Fsize(local))
DeleteFile(local);
return;
}
}
else if(command("%s", cmd) != RPL_PRELIM)
{
if(oldtype)
SetType(oldtype);
return;
}
if(!Host.PassiveMode)
{
din = dataconn();
if(din == INVALID_SOCKET)
{
Log(("!dataconn: PASV ret"));
goto abort;
}
}
/**/
switch(type)
{
case TYPE_A:
case TYPE_I:
case TYPE_L:
{
FtpSetBreakable(this, FALSE);
CurrentState = fcsProcessFile;
if(fout.Handle && PluginAvailable(PLUGIN_NOTIFY))
FTPNotify().Notify(&ni);
DWORD b,e,bw;
__int64 totalValue;
int b_done;
DWORD ind;
int b_ost = Host.IOBuffSize, wsz = get_cluster_size(local)*2;
if(!wsz || (wsz > b_ost && (wsz /= 2) > b_ost)) wsz = 512;
ind = Min(1024*1024, Max(4*wsz, 256*1024)); // 256K - 1M
setsockopt(din, SOL_SOCKET, SO_RCVBUF, (char*)&ind, sizeof(ind));
b_done = ind = 0;
totalValue = 0;
bool unalign = false;
GET_TIME(b);
bw = b;
while(true)
{
int c;
if(wsz != 512 && b_done >= wsz) // pseudo ansync io
{
DWORD off = 0, rdy = 0, ost = b_done % wsz, top = b_done - ost;
while(ioctlsocket(din, FIONREAD, &rdy) && !rdy)
{
if(Fwrite(fout.Handle,IOBuff+off,wsz) != wsz) goto write_error;
if((off += wsz) >= top) break;
}
if(off)
{
b_done -= off;
if(b_done) memmove(IOBuff, IOBuff+off, b_done);
b_ost = Host.IOBuffSize - b_done;
}
}
//Recv
c = nb_recv(&din, IOBuff+b_done, b_ost, 0);
if(c <= 0)
{
if(b_done && Fwrite(fout.Handle,IOBuff,b_done) != b_done) goto write_error;
if(c < 0)
{
Log(("gf(%d,%s)=%I64u: !read buff",code,GetSocketErrorSTR(),totalValue));
code = RPL_TRANSFERERROR;
goto NormExit;
}
Log(("gf(%d,%s)=%I64u: read zero",code,GetSocketErrorSTR(),totalValue));
break;
}
totalValue += c;
GET_TIME(e);
if(!fout.Handle)
{
//Add readed to buffer
Log(("AddOutput: +%d bytes", c));
AddOutput((BYTE*)IOBuff,c);
}
else //Write to file
{
b_done += c;
b_ost -= c;
if(b_ost < wsz || CMP_TIME(e,bw) >= 3.0)
{
DWORD ost = 0;
if(wsz == 512 || b_done <= wsz) // timeout or very small buffer
{
if(Fwrite(fout.Handle,IOBuff,b_done) != b_done) goto write_error;
if(b_done < wsz) unalign = true; // flag of timeout witing (optimize)
}
else
{
// scatter-gatter for RAID in win32 is very bad on large buffer
// and when work without RAID synchronous write speed is independ
// if buffer size is >= 2*cluster size
int off = 0;
if(unalign) // was 'timeouted unaligned write'
{
unalign = false;
off = (DWORD)(totalValue % wsz);
if(off)
{
if(Fwrite(fout.Handle,IOBuff,off) != off) goto write_error;
b_done -= off;
if(b_done < wsz)
{
memmove(IOBuff, IOBuff+off, b_done);
goto skip_sg;
}
}
}
ost = b_done % wsz;
b_done -= ost;
do
if(Fwrite(fout.Handle,IOBuff+off,wsz) != wsz) goto write_error;
while((off += wsz) < b_done);
if(ost) memmove(IOBuff, IOBuff+off, ost);
}
b_done = ost;
skip_sg:
b_ost = Host.IOBuffSize - b_done;
GET_TIME(e);
bw = e;
}
}
ind += c;
if(CMP_TIME(e,b) >= 0.5)
{
b = e;
c = ind;
ind = 0;
//Call user CB
if(IOCallback)
{
if(!TrafficInfo->Callback(c))
{
Log(("gf: canceled by CB"));
do_cancel:
ErrorCode = ERROR_CANCELLED;
if(b_done && Fwrite(fout.Handle,IOBuff,b_done) != b_done)
{
write_error:
SysError = TRUE;
ErrorCode = GetLastError();
if(ErrorCode == ERROR_SUCCESS)
ErrorCode = ERROR_WRITE_FAULT; // for non equal counter
Log(("!write local"));
}
goto abort;
}
}
else
//Show Quite progressing
if(Opt.ShowIdle && !remote)
{
char digit[ 20 ];
String str;
str.printf("%s%s ", FP_GetMsg(MReaded), FCps(digit,(double)totalValue));
SetLastError(ERROR_SUCCESS);
IdleMessage(str.c_str(),Opt.ProcessColor);
if(CheckForEsc(FALSE)) goto do_cancel;
}
}
}
if(IOCallback)
TrafficInfo->Callback(0);
break;
}
}
NormExit:
FtpSetBreakable(this, oldBrk);
ocode = code;
oecode = ErrorCode;
CurrentState = oState;
scClose(data_peer,-1);
if(getreply(FALSE) == RPL_ERROR ||
oldtype && !SetType(oldtype))
{
lostpeer();
}
else
{
code = ocode;
ErrorCode = oecode;
}
if(fout.Handle && PluginAvailable(PLUGIN_NOTIFY))
{
ni.Starting = FALSE;
ni.Success = TRUE;
FTPNotify().Notify(&ni);
}
return;
abort:
FtpSetBreakable(this, oldBrk);
if(!cpend)
{
Log(("!!!cpend"));
}
ocode = code;
oecode = ErrorCode;
CurrentState = oState;
if(!SendAbort(din) ||
(oldtype && !SetType(oldtype)))
lostpeer();
else
{
code = ocode;
ErrorCode = oecode;
}
scClose(data_peer,-1);
if(fout.Handle && PluginAvailable(PLUGIN_NOTIFY))
{
ni.Starting = FALSE;
ni.Success = FALSE;
FTPNotify().Notify(&ni);
}
return;
}
int FTP::ProcessEvent(int Event,void *Param)
{
#if defined(__FILELOG__)
static LPCSTR states[] = { "fcsNormal", "fcsExpandList", "fcsClose", "fcsConnecting","fcsFTP" };
PROC(("ProcessEvent","%d (%s),%08X",Event,states[CurrentState],Param))
#endif
PanelInfo pi;
int n;
FTPHost* p;
pi.PanelItems = NULL;
if(Event == FE_BREAK ||
CurrentState == fcsClose ||
CurrentState == fcsConnecting)
{
Log(("Skip event"));
return FALSE;
}
//Close notify
if(Event == FE_CLOSE)
{
Log(("Close notify"));
CurrentState = fcsClose;
if(ShowHosts)
{
if(!pi.PanelItems)
FP_Info->Control(this, FCTL_GETPANELINFO, &pi);
FP_SetRegKey("LastHostsMode",pi.ViewMode);
Log(("Write HostsMode: %d",pi.ViewMode));
}
return FALSE;
}
//Position cursor to added|corrected item on panel
if(Event == FE_REDRAW)
{
if(SelectFile.Length())
{
Log(("PositionItem: [%s]",SelectFile.c_str()));
if(!pi.PanelItems)
FP_Info->Control(this, FCTL_GETPANELINFO, &pi);
for(n = 0; n < pi.ItemsNumber; n++)
{
if(ShowHosts)
{
if((p=FTPHost::Convert(&pi.PanelItems[n])) == NULL ||
!SelectFile.Cmp(PointToName(p->RegKey)))
continue;
}
else
{
if(!SelectFile.Cmp(FTP_FILENAME(&pi.PanelItems[n])))
continue;
}
Log(("PosItem[%d] [%s]", n, FTP_FILENAME(&pi.PanelItems[n])));
PanelRedrawInfo pri;
pri.CurrentItem = n;
pri.TopPanelItem = pi.TopPanelItem;
SelectFile = "";
FP_Info->Control(this,FCTL_REDRAWPANEL,&pri);
break;
}
SelectFile = "";
}
}/*REDRAW*/
//Correct saved hosts mode on change
if(ShowHosts && Event == FE_CHANGEVIEWMODE)
{
if(!pi.PanelItems)
FP_Info->Control(this, FCTL_GETPANELINFO, &pi);
Log(("New ColumnMode %d",pi.ViewMode));
FP_SetRegKey("LastHostsMode",pi.ViewMode);
}
//Set start hosts panel mode
if(Event == FE_REDRAW)
{
if(ShowHosts && !PluginColumnModeSet)
{
if(Opt.PluginColumnMode != -1)
FP_Info->Control(this,FCTL_SETVIEWMODE,&Opt.PluginColumnMode);
else
{
int num = FP_GetRegKey("LastHostsMode",2);
FP_Info->Control(this,FCTL_SETVIEWMODE,&num);
}
PluginColumnModeSet = TRUE;
}
else if(isBackup() && NeedToSetActiveMode)
{
FP_Info->Control(this,FCTL_SETVIEWMODE,&ActiveColumnMode);
NeedToSetActiveMode = FALSE;
}
}
//Keep alive
if(Event == FE_IDLE &&
!ShowHosts &&
FtpCmdLineAlive(hConnect) &&
KeepAlivePeriod &&
FP_PeriodEnd(KeepAlivePeriod))
{
Log(("Keep alive"));
FTPCmdBlock blocked(this,TRUE);
if(Opt.ShowIdle)
{
SetLastError(ERROR_SUCCESS);
IdleMessage(FP_GetMsg(MKeepAliveExec),Opt.ProcessColor);
}
FtpKeepAlive(hConnect);
Log(("end Keep alive"));
}
//CMD line
if(Event == FE_COMMAND)
{
Log(("CMD line"));
return ExecCmdLine((LPCSTR)Param,FALSE);
}
return FALSE;
}
//--------------------------------------------------------------------------------
void Connection::recvrequest(char *cmd, char *local, char *remote, const char *mode)
{
//??SaveConsoleTitle _title;
recvrequestINT(cmd,local,remote,mode);
IdleMessage(NULL,0);
}