CNewTaskDlg::CNewTaskDlg( MainFrame* parent, wxString defUrl, wxString refer, wxString defSavePath )
: wxDialog( parent, -1, _MGSTR( _S_NEW_TASK ), wxDefaultPosition, wxDefaultSize, wxDEFAULT_DIALOG_STYLE | wxRESIZE_BORDER )
{
m_pParent = parent;
m_sUrl = defUrl;
m_sRefer = refer;
//对这个url分离用户名和密码
if ( !m_sUrl.IsEmpty() )
{
CUrlParser par;
std::string m_swap;
#ifndef _UNICODE
if ( par.SetUrl( m_sUrl.mb_str( wxConvLocal ) ) )
#else
if ( par.SetUrl( m_sUrl.mb_str( wxConvLocal ).data() ) )
#endif
{
if ( par.GetUrlType() == FTP_PROTOCOL )
{
m_sUser = wxString( par.GetUser().c_str(), wxConvLocal );
m_sPass = wxString( par.GetPass().c_str(), wxConvLocal );
m_sUrl = wxString( par.GetRawUrl().c_str(), wxConvLocal );
}
else
{
m_sUrl = wxString( par.GetRawUrl().c_str(), wxConvLocal );
}
if ( par.GetFileName().empty() )
{
m_sRename = wxT( "index.html" );
}
else
{
//m_sRename = par.GetFileName();
m_sRename=wxString(par.GetFileName().c_str(),wxConvLocal);
}
}
}
m_sSavePath = defSavePath;
if ( m_sSavePath.IsEmpty() )
{
m_sSavePath = wxString( gDefSavePath.c_str(), wxConvLocal );
if ( m_sSavePath.IsEmpty() )
{
std::string home;
GetUserHome( home );
m_sSavePath = wxString( home.c_str(), wxConvLocal );
}
}
m_nThreadNum = gTaskDefThread;
m_nSocksProxy = 0;
m_nFtpProxy = 0;
m_nHttpProxy = 0;
m_nRunNow = 0; //run
m_nFileSpin = 1;
m_nRetry = gRetryTime;
m_nRetryWait = gRetryWait;
m_sSavePath = wxString( gDefSavePath.c_str(), wxConvLocal );
m_pMirrorBox = NULL;
m_ThSpin = NULL;
m_ThRetry = NULL;
m_ThRetryWait = NULL;
m_FileSpin = NULL;
m_pTextCtrl = NULL;
m_RenameCtrl=NULL;
m_bOpenUrlKeyEvent = false;
//这里准备一下代理列表
InitProxyList();
Init();
}
CPropertyDlg::CPropertyDlg( MainFrame* parent, int taskid )
: wxDialog( parent, -1, _("Task properties"), wxDefaultPosition, wxDefaultSize, wxDEFAULT_DIALOG_STYLE| wxRESIZE_BORDER )
{
m_pParent = parent;
_TaskAttr *tsk = m_pParent->GetTask( taskid );
if ( tsk == NULL )
return ;
//m_sUrl = wxString( tsk->sURL.c_str() );
m_sUrl=wxString( tsk->sURL.c_str(), wxConvLocal );//unicode patch
//m_sRefer = wxString( tsk->sRefer.c_str() );
m_sRefer=wxString( tsk->sRefer.c_str(), wxConvLocal ); //unicode patch
//对这个url分离用户名和密码
if ( !m_sUrl.IsEmpty() )
{
CUrlParser par;
std::string m_swap;
#ifndef _UNICODE
if ( par.SetUrl( m_sUrl.mb_str( wxConvLocal ) ) )
#else
if( par.SetUrl( m_sUrl.mb_str(wxConvLocal).data() ) ) //unicode patch
#endif
{
if ( par.GetUrlType() == FTP_PROTOCOL )
{
m_sUser = wxString( par.GetUser().c_str(), wxConvLocal );
m_sPass = wxString( par.GetPass().c_str(), wxConvLocal );
m_sUrl = wxString( par.GetRawUrl().c_str(), wxConvLocal );
}
else
{
m_sUrl = wxString( par.GetRawUrl().c_str(), wxConvLocal );
}
if ( par.GetFileName().empty() )
{
m_sRename = _( "index.html" );
}
else
{
//m_sRename = par.GetFileName();
m_sRename=wxString(par.GetFileName().c_str(),wxConvLocal); //unicode patch
}
}
}
//m_sSavePath = wxString( tsk->sSavePath.c_str() );
m_sSavePath=wxString( tsk->sSavePath.c_str(), wxConvLocal ); //unicode patch
m_nThreadNum = tsk->nThread;
m_nRetry = tsk->nRetry;
m_nRetryWait = tsk->nRetryWait;
m_nSocksProxy = 0;
m_nFtpProxy = 0;
m_nHttpProxy = 0;
m_nRunNow = 0;
m_nFileSpin = 0;
m_pMirrorBox = NULL;
m_ThSpin = NULL;
m_ThRetry = NULL;
m_ThRetryWait = NULL;
m_FileSpin = NULL;
m_pTextCtrl = NULL;
m_bOpenUrlKeyEvent = false;
InitProxyList( tsk->sSockProxy, tsk->sFtpProxy, tsk->sHttpProxy );
Init( tsk->nStatus );
//add mirror string
for ( int i = 0;i < int( tsk->sMirrorURL.size() );i++ )
{
AddMirrorUrl( wxString( tsk->sMirrorURL[ i ].c_str(), wxConvLocal ) );
}
}
void CMgSingleTask::InitThread()
{
OutMsg( -1, _("Job running..."), MSG_INFO );
pthread_setcanceltype ( PTHREAD_CANCEL_ASYNCHRONOUS, NULL );
CUrlParser upar;
if ( !m_Refer.empty() )
upar.SetRefer( m_Refer );
int oldstate;
vector<string> cookie; //for http
again:
OutMsg( -1, _("Analysis URL..."), MSG_INFO );
if ( !upar.SetUrl( m_sUrl ) )
{
OutMsg( -1, c_str(_("Not a support URL type.")) + m_sUrl, MSG_ERROR );
FinishTask( _TASK_ERROR );
pthread_exit( 0 );
}
OutMsg( -1, _("URL OK."), MSG_INFO );
//任务参数
m_Server = upar.GetServer();
m_ServerPort = upar.GetPort();
m_Username = upar.GetUser();
m_Password = upar.GetPass();
m_EscFilePathName = upar.GetEscFilePathName(); //文件名
if ( upar.GetUrlType() == FTP_PROTOCOL ) //ftp
{
OutMsg( -1, _("Main URL is ftp type."), MSG_INFO );
m_pFtpInfo = new CMgFtpInfo(
this,
m_sUrl,
m_nRetry,
m_nRetryWait,
-1
);
//set proxy here
if ( m_bUseProxy )
{
OutMsg( -1, _("Task will use socks proxy."), MSG_INFO );
m_pFtpInfo->SetProxy( m_Proxy, m_ProxyPort, m_ProxyUser, m_ProxyPass, m_ProxyVersion );
}
if ( m_bUseFtpProxy )
{
OutMsg( -1, _("Task will use ftp proxy."), MSG_INFO );
m_pFtpInfo->SetFtpProxy( m_FtpProxy, m_FtpProxyPort );
}
//go
if ( !m_pFtpInfo->GetInfo() )
{ //error when get info. 2006/09/21
OutMsg( -1, _("Get file info error."), MSG_ERROR );
pthread_setcancelstate ( PTHREAD_CANCEL_DISABLE, &oldstate );
delete m_pFtpInfo;
m_pFtpInfo = NULL;
pthread_setcancelstate ( oldstate, NULL );
FinishTask( _TASK_ERROR );
pthread_exit( 0 );
}
m_nFileLen = m_pFtpInfo->FileSize();
bool bresume = m_pFtpInfo->IsResume();
pthread_setcancelstate ( PTHREAD_CANCEL_DISABLE, &oldstate );
delete m_pFtpInfo;
m_pFtpInfo = NULL;
pthread_setcancelstate ( oldstate, NULL );
if ( m_nFileLen == -1 || !bresume )
{
if ( m_nFileLen == -1 )
{
OutMsg( -1, _("No file length got."), MSG_WARNNING );
}
if ( !bresume )
{
OutMsg( -1, _("Server cannot support resume method."), MSG_WARNNING );
}
m_nAnts = 1;
//FinishTask(_TASK_ERROR);
//InitThreadQuit();
}
} //main url ftp
else if ( upar.GetUrlType() == HTTP_PROTOCOL )
{ //http
OutMsg( -1, _("Main URL is http type."), MSG_INFO );
m_pHttpInfo =
new CMgHttpInfo(
this,
m_sUrl,
m_nRetry,
m_nRetryWait,
upar.GetRefer(),
-1
);
//if need proxy, make here
if ( m_bUseProxy )
{
OutMsg( -1, _("Task will use socks proxy."), MSG_INFO );
m_pHttpInfo->SetProxy( m_Proxy, m_ProxyPort, m_ProxyUser, m_ProxyPass, m_ProxyVersion );
}
if ( m_bUseHttpProxy )
{
OutMsg( -1, _("Task will use http proxy."), MSG_INFO );
m_pHttpInfo->SetHttpProxy( m_HttpProxy, m_HttpProxyPort );
}
if ( !m_pHttpInfo->GetInfo() )
{ //error when get info.
OutMsg( -1, _("Get file info error."), MSG_ERROR );
pthread_setcancelstate ( PTHREAD_CANCEL_DISABLE, &oldstate );
delete m_pHttpInfo;
m_pHttpInfo = NULL;
pthread_setcancelstate ( oldstate, NULL );
FinishTask( _TASK_ERROR );
pthread_exit( 0 );
}
if ( m_pHttpInfo->IsRedirect() )
{
upar.SetRefer( m_sUrl );
m_sUrl = m_pHttpInfo->GetRedirect();
pthread_setcancelstate ( PTHREAD_CANCEL_DISABLE, &oldstate );
delete m_pHttpInfo;
m_pHttpInfo = NULL;
pthread_setcancelstate ( oldstate, NULL );
//跳转后就不再用原来的名字了。
if ( upar.SetUrl( m_sUrl ) )
{
if ( upar.GetFileName() != string( "" ) )
m_sFilename = upar.GetFileName();
}
goto again;
}
// for( int coo=0; coo < m_pHttpInfo->GetCookieNum(); coo++)
// {
//
// cookie.push_back( m_pHttpInfo->GetCookie( coo ) );
// }
m_nFileLen = m_pHttpInfo->GetFileSize();
pthread_setcancelstate ( PTHREAD_CANCEL_DISABLE, &oldstate );
delete m_pHttpInfo;
m_pHttpInfo = NULL;
pthread_setcancelstate ( oldstate, NULL );
if ( m_nFileLen == -1 )
{
OutMsg( -1, _("No file length got."), MSG_WARNNING );
m_nAnts = 1;
}
} //http
else
{
OutMsg( -1, c_str(_("Not a support URL type.")) + m_sUrl, MSG_ERROR );
FinishTask( _TASK_ERROR );
pthread_exit( 0 );
}
//create file manager
pthread_setcancelstate ( PTHREAD_CANCEL_DISABLE, &oldstate );
m_pFM = new CMgFileManager ( this, m_sUrl, m_sFilename, m_sSavePath, m_nFileLen, upar.GetRefer() );
pthread_setcancelstate ( oldstate, NULL );
//check file
if ( !m_pFM->CheckFile() )
{
pthread_setcancelstate ( PTHREAD_CANCEL_DISABLE, &oldstate );
delete m_pFM;
m_pFM = NULL;
pthread_setcancelstate ( oldstate, NULL );
FinishTask( _TASK_ERROR );
pthread_exit( 0 );
}
//input mirror url
std::vector<std::string>::const_iterator it;
for ( it = m_sMirrorUrl.begin();it != m_sMirrorUrl.end();it++ )
{
m_pFM->AddMirrorUrl( *it );
}
//start work thread
pthread_setcancelstate ( PTHREAD_CANCEL_DISABLE, &oldstate );
pthread_mutex_lock( &m_AntsMutex );
for ( int i = 0;i < m_nAnts;i++ )
{
//not run when construct 2006/09/03
m_pAnts[ i ] = new CProAnts(
this, //用于线程消息回送
m_pFM,
i + 1, //线程自然数编号，回送消息时带上，用于分辨消息来源
m_nFileLen,
m_nRetry,
m_nRetryWait
);
//if need proxy, make here
if ( m_bUseProxy )
{
m_pAnts[ i ] ->SetProxy( m_Proxy, m_ProxyPort, m_ProxyUser, m_ProxyPass, m_ProxyVersion );
}
if ( m_bUseFtpProxy )
{
m_pAnts[ i ] ->SetFtpProxy( m_FtpProxy, m_FtpProxyPort );
}
if ( m_bUseHttpProxy )
{
m_pAnts[ i ] ->SetHttpProxy( m_HttpProxy, m_HttpProxyPort );
}
m_pAnts[ i ] ->Go();
}
m_nRunningAnts = m_nAnts;
m_bInitAnts = true;
pthread_mutex_unlock( &m_AntsMutex );
pthread_setcancelstate ( oldstate, NULL );
pthread_exit( 0 );
}
