void CUpdater::ProcessData(CNotification* notification)
{
if( state_ != checking ) {
return;
}
CDataNotification* pData = reinterpret_cast<CDataNotification*>(notification);
int len;
char* data = pData->Detach(len);
if( raw_version_information_.size() + len > 131072 ) {
engine_->Command(CCancelCommand());
SetState(failed);
}
else {
for (int i = 0; i < len; i++) {
if (data[i] < 10 || (unsigned char)data[i] > 127) {
SetState(failed);
engine_->Command(CCancelCommand());
break;
}
}
}
if( state_ == checking ) {
raw_version_information_ += wxString(data, wxConvUTF8, len);
}
delete [] data;
}
bool CCommandQueue::Cancel()
{
if (m_exclusiveEngineLock)
return false;
if (m_CommandList.empty())
return true;
std::list<CCommand *>::iterator iter = m_CommandList.begin();
CCommand *pCommand = *(iter++);
for (; iter != m_CommandList.end(); iter++)
delete *iter;
m_CommandList.clear();
m_CommandList.push_back(pCommand);
if (!m_pEngine)
{
delete pCommand;
m_CommandList.clear();
m_pState->NotifyHandlers(STATECHANGE_REMOTE_IDLE);
return true;
}
int res = m_pEngine->Command(CCancelCommand());
if (res == FZ_REPLY_WOULDBLOCK)
return false;
else
{
delete pCommand;
m_CommandList.clear();
m_pState->NotifyHandlers(STATECHANGE_REMOTE_IDLE);
return true;
}
}
void CUpdateWizard::OnEngineEvent(wxEvent& event)
{
if (!m_pEngine)
return;
if (m_currentPage >= 3)
{
CNotification *pNotification = m_pEngine->GetNextNotification();
while (pNotification)
{
delete pNotification;
pNotification = m_pEngine->GetNextNotification();
}
return;
}
CNotification *pNotification = m_pEngine->GetNextNotification();
while (pNotification)
{
switch (pNotification->GetID())
{
case nId_logmsg:
{
if (!m_loaded)
break;
CLogmsgNotification* pLogMsg = reinterpret_cast<CLogmsgNotification *>(pNotification);
if (pLogMsg->msgType == Status || pLogMsg->msgType == Command)
{
if (!m_currentPage)
{
wxStaticText *pText = XRCCTRL(*this, "ID_CHECKINGTEXTPROGRESS", wxStaticText);
wxString text = pLogMsg->msg;
text.Replace(_T("&"), _T("&&"));
WrapText(pText, text, m_pages[0]->GetClientSize().x);
pText->SetLabel(text);
m_pages[0]->GetSizer()->Layout();
wxGauge* pProgress = XRCCTRL(*this, "ID_CHECKINGPROGRESS", wxGauge);
int value = pProgress->GetValue();
#ifdef __WXDEBUG__
wxASSERT(value < MAXCHECKPROGRESS);
#endif
if (value < MAXCHECKPROGRESS)
pProgress->SetValue(value + 1);
}
else if (m_currentPage == 2)
{
wxStaticText *pText = XRCCTRL(*this, "ID_DOWNLOADPROGRESSTEXT", wxStaticText);
wxString text = pLogMsg->msg;
text.Replace(_T("&"), _T("&&"));
WrapText(pText, text, m_pages[2]->GetClientSize().x);
pText->SetLabel(text);
m_pages[2]->GetSizer()->Layout();
}
}
wxString label;
switch (pLogMsg->msgType)
{
case Error:
label = _("Error:");
break;
case Status:
label = _("Status:");
break;
case Command:
label = _("Command:");
break;
case Response:
label = _("Response:");
break;
default:
break;
}
if (label != _T(""))
m_update_log += label + _T(" ") + pLogMsg->msg + _T("\n");
}
break;
case nId_operation:
{
COperationNotification* pOpMsg = reinterpret_cast<COperationNotification*>(pNotification);
if (pOpMsg->nReplyCode != FZ_REPLY_OK)
{
while (pNotification)
{
delete pNotification;
pNotification = m_pEngine->GetNextNotification();
}
FailedTransfer();
return;
}
if (!m_inTransfer)
{
if (m_loaded && !m_currentPage)
{
wxGauge* pProgress = XRCCTRL(*this, "ID_CHECKINGPROGRESS", wxGauge);
pProgress->SetValue(pProgress->GetValue() + 1);
}
int res = SendTransferCommand();
if (res == FZ_REPLY_WOULDBLOCK)
break;
else if (res != FZ_REPLY_OK)
{
FailedTransfer();
break;
}
}
if (!m_loaded || !m_currentPage)
{
m_pEngine->Command(CDisconnectCommand());
ParseData();
}
else if (m_currentPage == 2)
{
if (!VerifyChecksum())
break;
int pos = m_localFile.Find('.', true);
wxASSERT(pos > 0);
wxRenameFile(m_localFile, m_localFile.Left(pos));
m_localFile = m_localFile.Left(pos);
wxStaticText* pText = XRCCTRL(*this, "ID_DOWNLOADCOMPLETE", wxStaticText);
wxASSERT(pText);
wxButton* pNext = wxDynamicCast(FindWindow(wxID_FORWARD), wxButton);
pNext->Enable();
XRCCTRL(*this, "ID_DOWNLOADPROGRESS", wxGauge)->SetValue(100);
#ifdef __WXMSW__
pText->SetLabel(_("The most recent version has been downloaded. Click on Finish to close FileZilla and to start the installation."));
#else
pText->SetLabel(_("The most recent version has been downloaded. Please install it the same way you installed this version."));
#endif
pText->Show();
RewrapPage(2);
m_successfully_downloaded = true;
}
}
break;
case nId_data:
{
if (!m_inTransfer)
break;
wxASSERT(!m_currentPage);
CDataNotification* pOpMsg = reinterpret_cast<CDataNotification*>(pNotification);
int len;
char* data = pOpMsg->Detach(len);
if (m_data.Len() + len > 131072)
{
delete [] data;
m_pEngine->Command(CCancelCommand());
FailedTransfer();
break;
}
for (int i = 0; i < len; i++)
{
if (!data[i] || (unsigned char)data[i] > 127)
{
delete [] data;
data = 0;
m_pEngine->Command(CCancelCommand());
FailedTransfer();
break;
}
}
if (data)
{
m_data += wxString(data, wxConvUTF8, len);
delete [] data;
}
break;
}
case nId_asyncrequest:
{
CAsyncRequestNotification* pData = reinterpret_cast<CAsyncRequestNotification *>(pNotification);
if (pData->GetRequestID() == reqId_fileexists)
{
reinterpret_cast<CFileExistsNotification *>(pData)->overwriteAction = CFileExistsNotification::overwrite;
}
else if (pData->GetRequestID() == reqId_certificate)
{
CCertificateNotification* pCertNotification = (CCertificateNotification*)pData;
pCertNotification->m_trusted = true;
}
m_pEngine->SetAsyncRequestReply(pData);
}
break;
case nId_transferstatus:
if (!m_loaded)
break;
if (m_currentPage == 2)
{
CTransferStatusNotification *pTransferStatusNotification = reinterpret_cast<CTransferStatusNotification *>(pNotification);
const CTransferStatus *pStatus = pTransferStatusNotification->GetStatus();
SetTransferStatus(pStatus);
}
break;
default:
break;
}
delete pNotification;
pNotification = m_pEngine->GetNextNotification();
}
}