void CUpdateWizard::OnPageChanged(wxWizardEvent& event)
{
if (event.GetPage() == m_pages[0])
{
if (m_start_check)
{
m_start_check = false;
StartUpdateCheck();
}
return;
}
if (event.GetPage() != m_pages[2])
return;
wxButton* pNext = wxDynamicCast(FindWindow(wxID_FORWARD), wxButton);
pNext->Disable();
m_currentPage = 2;
wxStaticText *pText = XRCCTRL(*this, "ID_DOWNLOADTEXT", wxStaticText);
wxString text = wxString::Format(_("Downloading %s"), (CServer::GetPrefixFromProtocol(m_urlProtocol) + _T("://") + m_urlServer + m_urlFile).c_str());
text.Replace(_T("&"), _T("&&"));
pText->SetLabel(text);
m_inTransfer = false;
if (m_update_options)
m_update_options->m_use_internal_rootcert = false;
int res = m_pEngine->Command(CConnectCommand(CServer(m_urlProtocol, DEFAULT, m_urlServer, (m_urlProtocol == HTTPS) ? 443 : 80)));
if (res == FZ_REPLY_OK)
{
XRCCTRL(*this, "ID_DOWNLOADPROGRESSTEXT", wxStaticText)->SetLabel(_("Connecting to server"));
res = SendTransferCommand();
XRCCTRL(*this, "ID_DOWNLOADPROGRESS", wxGauge)->SetRange(100);
}
if (res == FZ_REPLY_OK)
ShowPage(m_pages[1]);
else if (res != FZ_REPLY_WOULDBLOCK)
FailedTransfer();
else
{
RewrapPage(2);
}
}
void CUpdateWizard::StartUpdateCheck()
{
m_inTransfer = false;
if (COptions::Get()->GetOptionVal(OPTION_UPDATECHECK_CHECKBETA) != 0)
m_urlFile += _T("&beta=1");
m_update_options->m_use_internal_rootcert = true;
int res = m_pEngine->Command(CConnectCommand(CServer(m_urlProtocol, DEFAULT, m_urlServer, (m_urlProtocol == HTTPS) ? 443 : 80)));
if (res == FZ_REPLY_OK)
{
if (m_loaded)
{
XRCCTRL(*this, "ID_CHECKINGTEXTPROGRESS", wxStaticText)->SetLabel(_("Connecting to server"));
wxGauge* pProgress = XRCCTRL(*this, "ID_CHECKINGPROGRESS", wxGauge);
pProgress->SetValue(pProgress->GetValue() + 1);
}
res = SendTransferCommand();
}
wxASSERT(res != FZ_REPLY_OK);
if (res != FZ_REPLY_WOULDBLOCK)
FailedTransfer();
}
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();
}
}
void CfEncryptGetFile(ServerFileGetState *args)
/* Because the stream doesn't end for each file, we need to know the
exact number of bytes transmitted, which might change during
encryption, hence we need to handle this with transactions */
{
int fd, n_read, cipherlen, finlen;
off_t total = 0, count = 0;
char sendbuffer[CF_BUFSIZE + 256], out[CF_BUFSIZE], filename[CF_BUFSIZE];
unsigned char iv[32] =
{ 1, 2, 3, 4, 5, 6, 7, 8, 1, 2, 3, 4, 5, 6, 7, 8, 1, 2, 3, 4, 5, 6, 7, 8, 1, 2, 3, 4, 5, 6, 7, 8 };
int blocksize = CF_BUFSIZE - 4 * CF_INBAND_OFFSET;
EVP_CIPHER_CTX ctx;
char *key, enctype;
struct stat sb;
ConnectionInfo *conn_info = &args->connect->conn_info;
key = (args->connect)->session_key;
enctype = (args->connect)->encryption_type;
TranslatePath(filename, args->replyfile);
stat(filename, &sb);
Log(LOG_LEVEL_DEBUG, "CfEncryptGetFile('%s'), size = %" PRIdMAX, filename, (intmax_t) sb.st_size);
/* Now check to see if we have remote permission */
if (!TransferRights(filename, args, &sb))
{
RefuseAccess(args->connect, args->buf_size, "");
FailedTransfer(conn_info);
}
EVP_CIPHER_CTX_init(&ctx);
if ((fd = open(filename, O_RDONLY)) == -1)
{
Log(LOG_LEVEL_ERR, "Open error of file '%s'. (open: %s)", filename, GetErrorStr());
FailedTransfer(conn_info);
}
else
{
int div = 3;
if (sb.st_size > 10485760L) /* File larger than 10 MB, checks every 64kB */
{
div = 32;
}
while (true)
{
memset(sendbuffer, 0, CF_BUFSIZE);
if ((n_read = read(fd, sendbuffer, blocksize)) == -1)
{
Log(LOG_LEVEL_ERR, "Read failed in EncryptGetFile. (read: %s)", GetErrorStr());
break;
}
off_t savedlen = sb.st_size;
if (count++ % div == 0) /* Don't do this too often */
{
Log(LOG_LEVEL_DEBUG, "Restatting '%s' - size %d", filename, n_read);
if (stat(filename, &sb))
{
Log(LOG_LEVEL_ERR, "Cannot stat file '%s' (stat: %s)",
filename, GetErrorStr());
break;
}
}
if (sb.st_size != savedlen)
{
AbortTransfer(conn_info, filename);
break;
}
total += n_read;
if (n_read > 0)
{
EVP_EncryptInit_ex(&ctx, CfengineCipher(enctype), NULL, key, iv);
if (!EVP_EncryptUpdate(&ctx, out, &cipherlen, sendbuffer, n_read))
{
FailedTransfer(conn_info);
EVP_CIPHER_CTX_cleanup(&ctx);
close(fd);
return;
}
if (!EVP_EncryptFinal_ex(&ctx, out + cipherlen, &finlen))
{
FailedTransfer(conn_info);
EVP_CIPHER_CTX_cleanup(&ctx);
close(fd);
return;
}
}
if (total >= savedlen)
{
if (SendTransaction(conn_info, out, cipherlen + finlen, CF_DONE) == -1)
{
Log(LOG_LEVEL_VERBOSE, "Send failed in GetFile. (send: %s)", GetErrorStr());
EVP_CIPHER_CTX_cleanup(&ctx);
close(fd);
return;
}
break;
}
else
{
if (SendTransaction(conn_info, out, cipherlen + finlen, CF_MORE) == -1)
{
Log(LOG_LEVEL_VERBOSE, "Send failed in GetFile. (send: %s)", GetErrorStr());
close(fd);
EVP_CIPHER_CTX_cleanup(&ctx);
return;
}
}
}
}
EVP_CIPHER_CTX_cleanup(&ctx);
close(fd);
}
