void GotJob(struct connection *c, char *job, char *d) {
jobs--;
SendNextCommand();
TryQuit();
}
void GotLoginSuccess(struct connection *c, char *d) {
logged_in = 1;
SendNextCommand();
TryQuit();
}
void CControlSocket::OnObtainLock()
{
if (ObtainLockFromEvent() == lock_unknown)
return;
SendNextCommand();
UnlockCache();
}
void GotAttach(struct connection *c, int status, char *d) {
if (status == 0) Done(102);
logged_in = 1;
SendNextCommand();
TryQuit();
}
void CControlSocket::OnObtainLock(wxCommandEvent& event)
{
if (!ObtainLockFromEvent())
return;
SendNextCommand();
UnlockCache();
}
void GotPromptEnd(struct connection *c, char type, char *job, char *d) {
Prompt = type;
switch (type) {
case 'I':
case '<': SendNextCommand(); break;
case 'U': SendUsername(Username); break;
case 'P': SendPassword(Password); break;
}
TryQuit();
}
bool CControlSocket::SetFileExistsAction(CFileExistsNotification *pFileExistsNotification)
{
wxASSERT(pFileExistsNotification);
if (!m_pCurOpData || m_pCurOpData->opId != Command::transfer)
{
LogMessage(__TFILE__, __LINE__, this, MessageType::Debug_Info, _T("No or invalid operation in progress, ignoring request reply %f"), pFileExistsNotification->GetRequestID());
return false;
}
CFileTransferOpData *pData = static_cast<CFileTransferOpData *>(m_pCurOpData);
switch (pFileExistsNotification->overwriteAction)
{
case CFileExistsNotification::overwrite:
SendNextCommand();
break;
case CFileExistsNotification::overwriteNewer:
if (!pFileExistsNotification->localTime.empty() || !pFileExistsNotification->remoteTime.empty())
SendNextCommand();
else if (pFileExistsNotification->download && pFileExistsNotification->localTime.earlier_than(pFileExistsNotification->remoteTime))
SendNextCommand();
else if (!pFileExistsNotification->download && pFileExistsNotification->localTime.later_than(pFileExistsNotification->remoteTime))
SendNextCommand();
else
{
if (pData->download)
{
wxString filename = pData->remotePath.FormatFilename(pData->remoteFile);
LogMessage(MessageType::Status, _("Skipping download of %s"), filename);
}
else
{
LogMessage(MessageType::Status, _("Skipping upload of %s"), pData->localFile);
}
ResetOperation(FZ_REPLY_OK);
}
break;
case CFileExistsNotification::overwriteSize:
/* First compare flags both size known but different, one size known and the other not (obviously they are different).
Second compare flags the remaining case in which we need to send command : both size unknown */
if ((pFileExistsNotification->localSize != pFileExistsNotification->remoteSize) || (pFileExistsNotification->localSize < 0))
SendNextCommand();
else {
if (pData->download) {
wxString filename = pData->remotePath.FormatFilename(pData->remoteFile);
LogMessage(MessageType::Status, _("Skipping download of %s"), filename);
}
else {
LogMessage(MessageType::Status, _("Skipping upload of %s"), pData->localFile);
}
ResetOperation(FZ_REPLY_OK);
}
break;
case CFileExistsNotification::overwriteSizeOrNewer:
if (!pFileExistsNotification->localTime.empty() || !pFileExistsNotification->remoteTime.empty())
SendNextCommand();
/* First compare flags both size known but different, one size known and the other not (obviously they are different).
Second compare flags the remaining case in which we need to send command : both size unknown */
else if ((pFileExistsNotification->localSize != pFileExistsNotification->remoteSize) || (pFileExistsNotification->localSize < 0))
SendNextCommand();
else if (pFileExistsNotification->download && pFileExistsNotification->localTime.earlier_than(pFileExistsNotification->remoteTime))
SendNextCommand();
else if (!pFileExistsNotification->download && pFileExistsNotification->localTime.later_than(pFileExistsNotification->remoteTime))
SendNextCommand();
else
{
if (pData->download)
{
wxString filename = pData->remotePath.FormatFilename(pData->remoteFile);
LogMessage(MessageType::Status, _("Skipping download of %s"), filename);
}
else
{
LogMessage(MessageType::Status, _("Skipping upload of %s"), pData->localFile);
}
ResetOperation(FZ_REPLY_OK);
}
break;
case CFileExistsNotification::resume:
if (pData->download && pData->localFileSize >= 0)
pData->resume = true;
else if (!pData->download && pData->remoteFileSize >= 0)
pData->resume = true;
SendNextCommand();
break;
case CFileExistsNotification::rename:
if (pData->download) {
wxFileName fn = pData->localFile;
fn.SetFullName(pFileExistsNotification->newName);
pData->localFile = fn.GetFullPath();
int64_t size;
bool isLink;
if (fz::local_filesys::get_file_info(fz::to_native(pData->localFile), isLink, &size, 0, 0) == fz::local_filesys::file)
pData->localFileSize = size;
else
pData->localFileSize = -1;
if (CheckOverwriteFile() == FZ_REPLY_OK)
SendNextCommand();
}
else {
pData->remoteFile = pFileExistsNotification->newName;
CDirentry entry;
bool dir_did_exist;
bool matched_case;
if (engine_.GetDirectoryCache().LookupFile(entry, *m_pCurrentServer, pData->tryAbsolutePath ? pData->remotePath : m_CurrentPath, pData->remoteFile, dir_did_exist, matched_case) &&
matched_case)
{
pData->remoteFileSize = entry.size;
if (entry.has_date())
pData->fileTime = entry.time;
if (CheckOverwriteFile() != FZ_REPLY_OK)
break;
}
else {
pData->fileTime = fz::datetime();
pData->remoteFileSize = -1;
}
SendNextCommand();
}
break;
case CFileExistsNotification::skip:
if (pData->download)
{
wxString filename = pData->remotePath.FormatFilename(pData->remoteFile);
LogMessage(MessageType::Status, _("Skipping download of %s"), filename);
}
else
{
LogMessage(MessageType::Status, _("Skipping upload of %s"), pData->localFile);
}
ResetOperation(FZ_REPLY_OK);
break;
default:
LogMessage(__TFILE__, __LINE__, this, MessageType::Debug_Warning, _T("Unknown file exists action: %d"), pFileExistsNotification->overwriteAction);
ResetOperation(FZ_REPLY_INTERNALERROR);
return false;
}
return true;
}
void MainProc() {
for (;;) {
int MaxFd;
fd_set ReadQ;
fd_set WriteQ;
if (Reconnect) {
close(connection->Fd);
FreeConnection(connection);
// To prevent floods. I know, this should be rather handled by ewrecv, but that would be nontrivial.
sleep(2);
AttachConnection();
Reconnect = 0;
}
MaxFd = 0;
FD_ZERO(&ReadQ);
FD_ZERO(&WriteQ);
// stdin
if (read_from_stdin
&& !want_quit
&& strlen(Commands) < COMMANDS_MAXLEN) {
FD_SET(0, &ReadQ);
}
if (connection) {
FD_SET(connection->Fd, &ReadQ);
if (connection->Fd > MaxFd) MaxFd = connection->Fd;
if (connection->WriteBuffer) FD_SET(connection->Fd, &WriteQ);
}
struct timeval to;
to.tv_sec = timeout_denominator;
to.tv_usec = 0;
struct timeval *top = &to;
if (timeout_denominator == 0) top = NULL;
int s = select(MaxFd + 1, &ReadQ, &WriteQ, 0, top);
if (s < 0) {
if (errno == EINTR) continue;
Done(1);
} else if (s == 0) {
// timeout
} else {
// stdin
if (read_from_stdin
&& strlen(Commands) < COMMANDS_MAXLEN
&& FD_ISSET(0, &ReadQ)) {
char buf[256+1] = "";
int to_read = COMMANDS_MAXLEN - strlen(Commands);
if (to_read > 256) to_read = 256;
int r = read(0, buf, to_read);
buf[r] = 0;
if (r == 0) {
want_quit = 1;
TryQuit();
} else {
strncat(Commands, buf, r);
SendNextCommand();
}
}
// Exchange input
if (connection && FD_ISSET(connection->Fd, &ReadQ)) {
errno = 0;
if (DoRead(connection) <= 0) {
if (detaching) {
Done(0);
} else {
Done(1);
}
} else {
int Chr;
while (Read(connection, &Chr, 1)) {
TestIProtoChar(connection, Chr);
}
}
}
// Exchange output
if (connection && FD_ISSET(connection->Fd, &WriteQ)) {
if (DoWrite(connection) < 0) Done(1);
}
}
if (login_timeout && login_start && !logged_in) {
if (time(NULL) - login_start > login_timeout) Done(103);
}
if (command_timeout && command_start && jobs) {
if (time(NULL) - command_start > command_timeout) Done(104);
}
}
}
bool CControlSocket::SetFileExistsAction(CFileExistsNotification *pFileExistsNotification)
{
wxASSERT(pFileExistsNotification);
if (!m_pCurOpData || m_pCurOpData->opId != cmd_transfer)
{
LogMessage(__TFILE__, __LINE__, this, Debug_Info, _T("No or invalid operation in progress, ignoring request reply %f"), pFileExistsNotification->GetRequestID());
return false;
}
CFileTransferOpData *pData = static_cast<CFileTransferOpData *>(m_pCurOpData);
switch (pFileExistsNotification->overwriteAction)
{
case CFileExistsNotification::overwrite:
SendNextCommand();
break;
case CFileExistsNotification::overwriteNewer:
if (!pFileExistsNotification->localTime.IsValid() || !pFileExistsNotification->remoteTime.IsValid())
SendNextCommand();
else if (pFileExistsNotification->download && pFileExistsNotification->localTime.IsEarlierThan(pFileExistsNotification->remoteTime))
SendNextCommand();
else if (!pFileExistsNotification->download && pFileExistsNotification->localTime.IsLaterThan(pFileExistsNotification->remoteTime))
SendNextCommand();
else
{
if (pData->download)
{
wxString filename = pData->remotePath.FormatFilename(pData->remoteFile);
LogMessage(Status, _("Skipping download of %s"), filename.c_str());
}
else
{
LogMessage(Status, _("Skipping upload of %s"), pData->localFile.c_str());
}
ResetOperation(FZ_REPLY_OK);
}
break;
case CFileExistsNotification::overwriteSize:
/* First compare flags both size known but different, one size known and the other not (obviously they are different).
Second compare flags the remaining case in which we need to send command : both size unknown */
if ((pFileExistsNotification->localSize != pFileExistsNotification->remoteSize) || (pFileExistsNotification->localSize == -1))
SendNextCommand();
else
{
if (pData->download)
{
wxString filename = pData->remotePath.FormatFilename(pData->remoteFile);
LogMessage(Status, _("Skipping download of %s"), filename.c_str());
}
else
{
LogMessage(Status, _("Skipping upload of %s"), pData->localFile.c_str());
}
ResetOperation(FZ_REPLY_OK);
}
break;
case CFileExistsNotification::overwriteSizeOrNewer:
if (!pFileExistsNotification->localTime.IsValid() || !pFileExistsNotification->remoteTime.IsValid())
SendNextCommand();
/* First compare flags both size known but different, one size known and the other not (obviously they are different).
Second compare flags the remaining case in which we need to send command : both size unknown */
else if ((pFileExistsNotification->localSize != pFileExistsNotification->remoteSize) || (pFileExistsNotification->localSize == -1))
SendNextCommand();
else if (pFileExistsNotification->download && pFileExistsNotification->localTime.IsEarlierThan(pFileExistsNotification->remoteTime))
SendNextCommand();
else if (!pFileExistsNotification->download && pFileExistsNotification->localTime.IsLaterThan(pFileExistsNotification->remoteTime))
SendNextCommand();
else
{
if (pData->download)
{
wxString filename = pData->remotePath.FormatFilename(pData->remoteFile);
LogMessage(Status, _("Skipping download of %s"), filename.c_str());
}
else
{
LogMessage(Status, _("Skipping upload of %s"), pData->localFile.c_str());
}
ResetOperation(FZ_REPLY_OK);
}
break;
case CFileExistsNotification::resume:
if (pData->download && pData->localFileSize != -1)
pData->resume = true;
else if (!pData->download && pData->remoteFileSize != -1)
pData->resume = true;
SendNextCommand();
break;
case CFileExistsNotification::rename:
if (pData->download)
{
wxFileName fn = pData->localFile;
fn.SetFullName(pFileExistsNotification->newName);
pData->localFile = fn.GetFullPath();
wxLongLong size;
bool isLink;
if (CLocalFileSystem::GetFileInfo(pData->localFile, isLink, &size, 0, 0) == CLocalFileSystem::file)
pData->localFileSize = size.GetValue();
else
pData->localFileSize = -1;
if (CheckOverwriteFile() == FZ_REPLY_OK)
SendNextCommand();
}
else
{
pData->remoteFile = pFileExistsNotification->newName;
CDirectoryCache cache;
CDirentry entry;
bool dir_did_exist;
bool matched_case;
if (cache.LookupFile(entry, *m_pCurrentServer, pData->tryAbsolutePath ? pData->remotePath : m_CurrentPath, pData->remoteFile, dir_did_exist, matched_case) &&
matched_case)
{
wxLongLong size = entry.size;
pData->remoteFileSize = size.GetLo() + ((wxFileOffset)size.GetHi() << 32);
if (CheckOverwriteFile() != FZ_REPLY_OK)
break;
}
SendNextCommand();
}
break;
case CFileExistsNotification::skip:
if (pData->download)
{
wxString filename = pData->remotePath.FormatFilename(pData->remoteFile);
LogMessage(Status, _("Skipping download of %s"), filename.c_str());
}
else
{
LogMessage(Status, _("Skipping upload of %s"), pData->localFile.c_str());
}
ResetOperation(FZ_REPLY_OK);
break;
default:
LogMessage(__TFILE__, __LINE__, this, Debug_Warning, _T("Unknown file exists action: %d"), pFileExistsNotification->overwriteAction);
ResetOperation(FZ_REPLY_INTERNALERROR);
return false;
}
return true;
}
