void CBaseMatchWnd::OnSearchDownload()
{
CSingleLock pSingleLock( &m_pMatches->m_pSection, TRUE );
CPtrList pFiles, pHits;
POSITION pos;
for ( pos = m_pMatches->m_pSelectedFiles.GetHeadPosition() ; pos ; )
{
CMatchFile* pFile = (CMatchFile*)m_pMatches->m_pSelectedFiles.GetNext( pos );
pSingleLock.Unlock();
switch ( CheckExisting( pFile->m_bSHA1, &pFile->m_pSHA1, pFile->m_bTiger, &pFile->m_pTiger, pFile->m_bED2K, &pFile->m_pED2K ) )
{
case 1:
pFiles.AddTail( pFile );
break;
case 3:
return;
}
pSingleLock.Lock();
}
for ( pos = m_pMatches->m_pSelectedHits.GetHeadPosition() ; pos ; )
{
CQueryHit* pHit = (CQueryHit*)m_pMatches->m_pSelectedHits.GetNext( pos );
pSingleLock.Unlock();
switch ( CheckExisting( pHit->m_bSHA1, &pHit->m_pSHA1, pHit->m_bTiger, &pHit->m_pTiger, pHit->m_bED2K, &pHit->m_pED2K ) )
{
case 1:
pHits.AddTail( pHit );
break;
case 3:
return;
}
pSingleLock.Lock();
}
pSingleLock.Unlock();
if ( pFiles.IsEmpty() && pHits.IsEmpty() ) return;
CSyncObject* pSync[2] = { &Network.m_pSection, &Transfers.m_pSection };
CMultiLock pMultiLock( pSync, 2, TRUE );
for ( pos = pFiles.GetHeadPosition() ; pos ; )
{
CMatchFile* pFile = (CMatchFile*)pFiles.GetNext( pos );
if ( m_pMatches->m_pSelectedFiles.Find( pFile ) != NULL ) Downloads.Add( pFile );
}
for ( pos = pHits.GetHeadPosition() ; pos ; )
{
CQueryHit* pHit = (CQueryHit*)pHits.GetNext( pos );
if ( m_pMatches->m_pSelectedHits.Find( pHit ) != NULL )
{
CDownload *pDownload = Downloads.Add( pHit );
// Send any reviews to the download, so they can be viewed later
if ( pDownload && ( pHit->m_nRating || ! pHit->m_sComments.IsEmpty() ) )
{
pDownload->AddReview( &pHit->m_pAddress, 2, pHit->m_nRating, pHit->m_sNick, pHit->m_sComments );
}
}
}
pMultiLock.Unlock();
m_wndList.Invalidate();
if ( Settings.Search.SwitchToTransfers && ! m_bContextMenu && GetTickCount() - m_tContextMenu > 5000 )
{
GetManager()->Open( RUNTIME_CLASS(CDownloadsWnd) );
}
}
CDownload* CDownloads::Add(CMatchFile* pFile, BOOL bAddToHead)
{
CSingleLock pLock( &Transfers.m_pSection, TRUE );
CDownload* pDownload = NULL;
if ( pDownload == NULL && pFile->m_bSHA1 )
pDownload = FindBySHA1( &pFile->m_pSHA1 );
if ( pDownload == NULL && pFile->m_bTiger )
pDownload = FindByTiger( &pFile->m_pTiger );
if ( pDownload == NULL && pFile->m_bED2K )
pDownload = FindByED2K( &pFile->m_pED2K );
if ( pDownload != NULL )
{
theApp.Message( MSG_DOWNLOAD, IDS_DOWNLOAD_ALREADY, (LPCTSTR)pFile->m_pHits->m_sName );
for ( CQueryHit* pHit = pFile->m_pHits ; pHit ; pHit = pHit->m_pNext )
{
pDownload->AddSourceHit( pHit );
// Send any reviews to the download, so they can be viewed later
if ( pHit->m_nRating || ! pHit->m_sComments.IsEmpty() )
{
pDownload->AddReview( &pHit->m_pAddress, 2, pHit->m_nRating, pHit->m_sNick, pHit->m_sComments );
}
}
pDownload->Resume();
}
else
{
pDownload = new CDownload();
if ( bAddToHead ) m_pList.AddHead( pDownload );
else m_pList.AddTail( pDownload );
if ( pFile->m_pBest != NULL )
{
pDownload->AddSourceHit( pFile->m_pBest, TRUE );
}
for ( CQueryHit* pHit = pFile->m_pHits ; pHit ; pHit = pHit->m_pNext )
{
if ( pHit != pFile->m_pBest )
{
pDownload->AddSourceHit( pHit, TRUE );
}
// Send any reviews to the download, so they can be viewed later
if ( pHit->m_nRating || ! pHit->m_sComments.IsEmpty() )
{
pDownload->AddReview( &pHit->m_pAddress, 2, pHit->m_nRating, pHit->m_sNick, pHit->m_sComments );
}
}
theApp.Message( MSG_DOWNLOAD, IDS_DOWNLOAD_ADDED,
(LPCTSTR)pDownload->GetDisplayName(), pDownload->GetSourceCount() );
if( pDownload->m_bSHA1 ) pDownload->m_bSHA1Trusted = TRUE;
else if( pDownload->m_bED2K ) pDownload->m_bED2KTrusted = TRUE;
}
pFile->m_bDownload = TRUE;
DownloadGroups.Link( pDownload );
Transfers.StartThread();
if ( ( (pDownload->GetSourceCount() == 0 ) ||
( pDownload->m_bED2K && ! pDownload->m_bSHA1 )) &&
(GetTryingCount() < Settings.Downloads.MaxFiles ) )
{
pDownload->FindMoreSources();
pDownload->SetStartTimer();
}
return pDownload;
}
