dlgdonkeyservers.cpp

著名的下载软件核心Shareaza

//
// DlgDonkeyServers.cpp
//
// Copyright (c) Shareaza Development Team, 2002-2004.
// This file is part of SHAREAZA (www.shareaza.com)
//
// Shareaza is free software; you can redistribute it
// and/or modify it under the terms of the GNU General Public License
// as published by the Free Software Foundation; either version 2 of
// the License, or (at your option) any later version.
//
// Shareaza is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with Shareaza; if not, write to the Free Software
// Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
//

#include "StdAfx.h"
#include "Shareaza.h"
#include "Settings.h"
#include "HostCache.h"
#include "DlgDonkeyServers.h"

#ifdef _DEBUG
#define new DEBUG_NEW
#undef THIS_FILE
static char THIS_FILE[] = __FILE__;
#endif

BEGIN_MESSAGE_MAP(CDonkeyServersDlg, CSkinDialog)
	//{{AFX_MSG_MAP(CDonkeyServersDlg)
	ON_EN_CHANGE(IDC_URL, OnChangeURL)
	ON_WM_TIMER()
	//}}AFX_MSG_MAP
END_MESSAGE_MAP()


/////////////////////////////////////////////////////////////////////////////
// CDonkeyServersDlg dialog

CDonkeyServersDlg::CDonkeyServersDlg(CWnd* pParent) : CSkinDialog(CDonkeyServersDlg::IDD, pParent)
{
	//{{AFX_DATA_INIT(CDonkeyServersDlg)
	m_sURL = _T("");
	//}}AFX_DATA_INIT
	m_hInternet = NULL;
	m_hThread = NULL;
}

CDonkeyServersDlg::~CDonkeyServersDlg()
{
	ASSERT( m_hThread == NULL );
	ASSERT( m_hInternet == NULL );
}

void CDonkeyServersDlg::DoDataExchange(CDataExchange* pDX)
{
	CSkinDialog::DoDataExchange(pDX);
	//{{AFX_DATA_MAP(CDonkeyServersDlg)
	DDX_Control(pDX, IDC_URL, m_wndURL);
	DDX_Control(pDX, IDOK, m_wndOK);
	DDX_Control(pDX, IDC_PROGRESS, m_wndProgress);
	DDX_Text(pDX, IDC_URL, m_sURL);
	//}}AFX_DATA_MAP
}

/////////////////////////////////////////////////////////////////////////////
// CDonkeyServersDlg message handlers

BOOL CDonkeyServersDlg::OnInitDialog() 
{
	CSkinDialog::OnInitDialog();
	
	SkinMe( _T("CDonkeyServersDlg") );
	
	m_sURL = Settings.eDonkey.ServerListURL;
	UpdateData( FALSE );
	
	m_wndOK.EnableWindow( m_sURL.Find( _T("http://") ) == 0 );
	m_wndProgress.SetRange( 0, 100 );
	m_wndProgress.SetPos( 0 );
	
	return TRUE;
}

void CDonkeyServersDlg::OnChangeURL() 
{
	UpdateData();
	m_wndOK.EnableWindow( m_sURL.Find( _T("http://") ) == 0 );
}

void CDonkeyServersDlg::OnOK() 
{
	UpdateData();
	
	if ( m_sURL.Find( _T("http://") ) != 0 ) return;
	
	Settings.eDonkey.ServerListURL = m_sURL;
	
	CString strAgent = Settings.SmartAgent( Settings.General.UserAgent );
	m_hInternet = InternetOpen( strAgent, INTERNET_OPEN_TYPE_PRECONFIG, NULL, NULL, 0 );
	if ( m_hInternet == NULL ) return;
	
	CWinThread* pThread = AfxBeginThread( ThreadStart, this, THREAD_PRIORITY_NORMAL );
	m_hThread = pThread->m_hThread;
	
	m_wndOK.EnableWindow( FALSE );
	m_wndURL.EnableWindow( FALSE );
}

void CDonkeyServersDlg::OnCancel() 
{
	OnTimer( 2 );
	CSkinDialog::OnCancel();
}

void CDonkeyServersDlg::OnTimer(UINT nIDEvent) 
{
	if ( m_hInternet != NULL )
	{
		InternetCloseHandle( m_hInternet );
		m_hInternet = NULL;
	}
	
	if ( m_hThread != NULL )
	{
		for ( int nAttempt = 5 ; nAttempt > 0 ; nAttempt-- )
		{
			DWORD nCode;

			if ( ! GetExitCodeThread( m_hThread, &nCode ) ) break;
			if ( nCode != STILL_ACTIVE ) break;
			Sleep( 100 );
		}
		
		if ( nAttempt == 0 )
		{
			TerminateThread( m_hThread, 0 );
			theApp.Message( MSG_DEBUG, _T("WARNING: Terminating CDonkeyServersDlg thread.") );
			Sleep( 100 );
		}
		
		m_hThread = NULL;
	}
	
	if ( nIDEvent == 1 ) EndDialog( IDOK );
}

/////////////////////////////////////////////////////////////////////////////
// CDonkeyServersDlg thread

UINT CDonkeyServersDlg::ThreadStart(LPVOID pParam)
{
	CDonkeyServersDlg* pDlg = (CDonkeyServersDlg*)pParam;
	pDlg->OnRun();
	return 0;
}

void CDonkeyServersDlg::OnRun()
{
	if ( m_hInternet == NULL ) return;
	
	HINTERNET hRequest = InternetOpenUrl( m_hInternet, m_sURL, NULL, 0,
		INTERNET_FLAG_RELOAD|INTERNET_FLAG_DONT_CACHE, 0 );
	
	if ( hRequest == NULL )
	{
		InternetCloseHandle( m_hInternet );
		m_hInternet = NULL;
		PostMessage( WM_TIMER, FALSE );
		return;
	}
	
	DWORD nLength, nlLength = 4;
	DWORD nRemaining = 0;
	BYTE pBuffer[1024];
	CMemFile pFile;
	
	if ( HttpQueryInfo( hRequest, HTTP_QUERY_CONTENT_LENGTH|HTTP_QUERY_FLAG_NUMBER,
		&nLength, &nlLength, NULL ) )
	{
		m_wndProgress.PostMessage( PBM_SETRANGE32, 0, nLength );
	}
	
	nLength = 0;
	
	while ( InternetQueryDataAvailable( hRequest, &nRemaining, 0, 0 ) && nRemaining > 0 )
	{
		nLength += nRemaining;
		m_wndProgress.PostMessage( PBM_SETPOS, nLength );
		
		while ( nRemaining > 0 )
		{
			DWORD nBuffer = min( nRemaining, 1024 );
			InternetReadFile( hRequest, pBuffer, nBuffer, &nBuffer );
			pFile.Write( pBuffer, nBuffer );
			nRemaining -= nBuffer;
		}
	}
	
	pFile.Seek( 0, CFile::begin );
	
	BOOL bSuccess = HostCache.eDonkey.ImportMET( &pFile );
	if ( bSuccess ) HostCache.Save();
	
	InternetCloseHandle( m_hInternet );
	m_hInternet = NULL;

	PostMessage( WM_TIMER, bSuccess ? 1 : 0 );
}