linkcreatordlg.cpp

Pipe类 Pipe类 Pipe类 Pipe类 Pipe类

//this file is part of eMule
//Copyright (C)2004 Merkur ( [e-mail] / http://www.emule-project.net )
//
//This program 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.
//
//This program 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 this program; if not, write to the Free Software
//Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.

#include "stdafx.h"
#include <winsock2.h>

#include "LinkCreator.h"
#include "LinkCreatorDlg.h"
#include "filehasher.h"
#include "functions.h"
#include "SHAHashSet.h"

#ifdef _DEBUG
#define new DEBUG_NEW
#endif


// CAboutDlg-Dialogfeld f黵 Anwendungsbefehl 'Info'

class CAboutDlg : public CDialog
{
public:
	CAboutDlg();
	void Localize();

// Dialogfelddaten
	enum { IDD = IDD_ABOUTBOX };

	protected:
	virtual void DoDataExchange(CDataExchange* pDX);    // DDX/DDV-Unterst黷zung
	virtual BOOL OnInitDialog();

// Implementierung
protected:
	DECLARE_MESSAGE_MAP()
};

CAboutDlg::CAboutDlg() : CDialog(CAboutDlg::IDD)
{
}

void CAboutDlg::DoDataExchange(CDataExchange* pDX)
{
	CDialog::DoDataExchange(pDX);
}

BEGIN_MESSAGE_MAP(CAboutDlg, CDialog)
END_MESSAGE_MAP()

BOOL CAboutDlg::OnInitDialog()
{
	CDialog::OnInitDialog();

	Localize();

	return TRUE; 
}

void CAboutDlg::Localize()
{
	if (!m_hWnd)
		return;

	static const UINT ids[] = {
		IDC_STATIC1, IDC_STATIC2
	};
	ModifyStyleEx(WNDEXSTYLE,GetWndExStyle());
	for (int i=0; i<ARRSIZE(ids); i++)
		GetDlgItem(ids[i])->ModifyStyleEx(WNDEXSTYLE,GetWndExStyle());

	CString buffer;
	buffer.Format(_T("%s v%s"),GetResString(IDS_LINKCREATOR),theApp.GetVersion());
	GetDlgItem(IDC_STATIC1)->SetWindowText(buffer);

	buffer.Format(GetResString(IDS_COPYRIGHTS),"2004-2005");
	GetDlgItem(IDC_STATIC2)->SetWindowText(buffer);

	GetDlgItem(IDC_STATIC3)->SetWindowText(GetResString(IDS_TRANSLATION));
	GetDlgItem(IDOK)->SetWindowText(GetResString(IDS_OK));
	SetWindowText(GetResString(IDS_ABOUT));
}


// CLinkCreatorDlg Dialogfeld



CLinkCreatorDlg::CLinkCreatorDlg(CWnd* pParent /*=NULL*/)
	: CDialog(CLinkCreatorDlg::IDD, pParent)
{
	m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
	m_pKnownFile = NULL;
}

void CLinkCreatorDlg::DoDataExchange(CDataExchange* pDX)
{
	CDialog::DoDataExchange(pDX);
	DDX_Control(pDX, IDC_FILEPATH, m_ctrFilePath);
	DDX_Control(pDX, IDC_WITHHASHSET, m_ctrlWithHashset);
	DDX_Control(pDX, IDC_RESULT, m_ctrlResults);
	DDX_Control(pDX, IDC_PROGRESS1, m_ctrlProgress);
	DDX_Control(pDX, IDC_HTTPLIST, m_ctrlHttpSources);
	DDX_Control(pDX, IDC_CLIENTSOURCESLIST, m_ctrlClientSources);
	DDX_Control(pDX, IDC_WITHAICH, m_ctrlAICH);
	DDX_Control(pDX, IDC_LANGCOMBO, m_ctrlLang);
}

BEGIN_MESSAGE_MAP(CLinkCreatorDlg, CDialog)
	ON_WM_SYSCOMMAND()
	ON_WM_PAINT()
	ON_WM_QUERYDRAGICON()
	ON_WM_CLOSE()
	//}}AFX_MSG_MAP
	ON_BN_CLICKED(IDC_START, OnBnClickedStart)
	ON_BN_CLICKED(IDC_CANCEL, OnBnClickedCancel)
	ON_BN_CLICKED(IDC_EXIT, OnClose)
	ON_BN_CLICKED(IDC_ADDHTTP, OnBnClickedAddhttp)
	ON_BN_CLICKED(IDC_HTTPCLEAR, OnBnClickedHttpclear)
	ON_BN_CLICKED(IDC_HTTPREMOVE, OnBnClickedHttpremove)
	ON_BN_CLICKED(IDC_BROWSE, OnBnClickedBrowse)
	ON_BN_CLICKED(IDC_COPY2CB, OnBnCopy2Clipboard)
	ON_BN_CLICKED(IDC_WITHHASHSET, OnBnClickedWithHashset)
	ON_BN_CLICKED(IDC_WITHAICH, OnBnClickedWithAICH)
	ON_CBN_SELCHANGE(IDC_LANGCOMBO, OnSelchangeLang)
	ON_BN_CLICKED(IDC_ADDCLIENT, OnBnClickedAddClientSource)
	ON_BN_CLICKED(IDC_CHECKSOURCE, SetButtonStates)
//	ON_EN_CHANGE(IDC_FILEPATH, SetStartButtonState)
	ON_EN_CHANGE(IDC_HTTPPATH, SetButtonStates)
	ON_EN_CHANGE(IDC_CLIENTSOURCE, SetButtonStates)
//	ON_EN_KILLFOCUS(IDC_FILEPATH, SetStartButtonState)

	///////////////////////////////////////////////////////////////////////////
	// WM_APP messages
	//
	ON_MESSAGE(TM_FINISHEDHASHING, OnFileHashed)
	ON_MESSAGE(TM_HASHFAILED, OnHashFailed)
	ON_MESSAGE(TM_FILEOPPROGRESS, OnProgress)
	ON_BN_CLICKED(IDC_CLIENTREMOVE, OnBnClickedClientremove)
	ON_BN_CLICKED(IDC_CLIENTSOURCECLEAR, OnBnClickedClientsourceclear)
	ON_LBN_SELCHANGE(IDC_HTTPLIST, OnLbnSelchangeHttplist)
	ON_LBN_SELCHANGE(IDC_CLIENTSOURCESLIST, OnLbnSelchangeClientsourceslist)
	ON_EN_CHANGE(IDC_FILEPATH, OnEnChangeFilepath)
END_MESSAGE_MAP()


// CLinkCreatorDlg Meldungshandler

BOOL CLinkCreatorDlg::OnInitDialog()
{
	CDialog::OnInitDialog();

	RESULTWND=GetDlgItem(IDC_RESULT)->m_hWnd;
	resultbig=false;

	if (!theApp.GetLang())
		theApp.SetLang( (LANGID)::GetThreadLocale() );

	ASSERT((IDM_ABOUTBOX & 0xFFF0) == IDM_ABOUTBOX);
	ASSERT(IDM_ABOUTBOX < 0xF000);

	CMenu* pSysMenu = GetSystemMenu(FALSE);
	if (pSysMenu != NULL)
	{
		pSysMenu->AppendMenu(MF_SEPARATOR);
		pSysMenu->AppendMenu(MF_STRING, IDM_ABOUTBOX, GetResString(IDS_ABOUT));
	}

	SetIcon(m_hIcon, TRUE);	// Gro遝s Symbol verwenden
	SetIcon(m_hIcon, FALSE);	// Kleines Symbol verwenden

	m_ctrlAICH.SetCheck(BST_CHECKED);
	m_ctrlProgress.SetRange(0, 100);
	m_ctrlProgress.SetPos(0);

	m_ctrlLang.SetDroppedWidth(150);
	Localize();

	if (m_strFileToHash.GetLength())
	{
		m_ctrFilePath.SetWindowText(m_strFileToHash);
		m_strFileToHash.Empty();
		OnBnClickedStart();
	}

	CheckDlgButton(IDC_CHECKSOURCE,BST_CHECKED);
	CheckDlgButton(IDC_LENGTHCHECK,BST_CHECKED);
	SetButtonStates();

	return TRUE; 
}

void CLinkCreatorDlg::Localize()
{
	if (!m_hWnd)
		return;

	static const UINT ids[] = {
		IDC_STATIC1, IDC_STATIC2, IDC_STATIC3, IDC_STATIC4,
		IDC_WITHHASHSET, IDC_STATIC5, IDC_STATIC6, IDC_WITHAICH,
		IDC_BROWSE
	};
	ModifyStyleEx(WNDEXSTYLE,GetWndExStyle());
	for (int i=0; i<ARRSIZE(ids); i++)
		GetDlgItem(ids[i])->ModifyStyleEx(WNDEXSTYLE,GetWndExStyle());

	GetDlgItem(IDC_STATIC1)->SetWindowText(GetResString(IDS_FILELOCATION));
	GetDlgItem(IDC_BROWSE)->SetWindowText(GetResString(IDS_BROWSE));
	
	GetDlgItem(IDC_ADDHTTP)->SetWindowText(GetResString(IDS_ADDHTTP));
	GetDlgItem(IDC_HTTPREMOVE)->SetWindowText(GetResString(IDS_HTTPREMOVE));
	GetDlgItem(IDC_HTTPCLEAR)->SetWindowText(GetResString(IDS_HTTPCLEAR));

	GetDlgItem(IDC_ADDCLIENT)->SetWindowText(GetResString(IDS_ADDHTTP));
	GetDlgItem(IDC_CLIENTREMOVE)->SetWindowText(GetResString(IDS_HTTPREMOVE));
	GetDlgItem(IDC_CLIENTSOURCECLEAR)->SetWindowText(GetResString(IDS_HTTPCLEAR));

	GetDlgItem(IDC_STATIC2)->SetWindowText(GetResString(IDS_HTTPADDRESS));
	GetDlgItem(IDC_STATIC3)->SetWindowText(GetResString(IDS_ADDEDHTTPSOURCES));
	GetDlgItem(IDC_STATIC4)->SetWindowText(GetResString(IDS_ED2KLINK));
	GetDlgItem(IDC_START)->SetWindowText(GetResString(IDS_START));
	GetDlgItem(IDC_CANCEL)->SetWindowText(GetResString(IDS_CANCEL));
	GetDlgItem(IDC_EXIT)->SetWindowText(GetResString(IDS_EXIT));
	GetDlgItem(IDC_WITHHASHSET)->SetWindowText(GetResString(IDS_WITHHASHSET));
	GetDlgItem(IDC_STATIC5)->SetWindowText(GetResString(IDS_EX1));
	GetDlgItem(IDC_STATIC6)->SetWindowText(GetResString(IDS_EX2));
	GetDlgItem(IDC_COPY2CB)->SetWindowText(GetResString(IDS_COPYTOCLIPBOARD));
	GetDlgItem(IDC_WITHAICH)->SetWindowText(GetResString(IDS_WITHAICH));

	SetDlgItemText(IDC_SOURCESFRAME,GetResString(IDS_SOURCESMANAGEMENT));
	SetDlgItemText(IDC_STATICCLIENTSOURCES,GetResString(IDS_CLIENTSCOURCE));
	SetDlgItemText(IDC_STATIC8,GetResString(IDS_CLIENTSOURCESLIST));

	SetDlgItemText(IDC_CHECKSOURCE,GetResString(IDS_CHECKSOURCE));
	SetDlgItemText(IDC_LENGTHCHECK,GetResString(IDS_LENGTHCHECK));
    		

	CString buffer;
	buffer.Format(_T("%s v%s"),GetResString(IDS_LINKCREATOR),theApp.GetVersion());
	SetWindowText(buffer);
	MakeLangSelection(m_ctrlLang);

	CMenu* pSysMenu = GetSystemMenu(FALSE);
	if (pSysMenu)
		pSysMenu->ModifyMenu(IDM_ABOUTBOX, MF_BYCOMMAND | MF_STRING, IDM_ABOUTBOX, GetResString(IDS_ABOUT));
}


void CLinkCreatorDlg::OnSysCommand(UINT nID, LPARAM lParam)
{
	if ((nID & 0xFFF0) == IDM_ABOUTBOX)
	{
		CAboutDlg dlgAbout;
		dlgAbout.DoModal();
	}
	else
	{
		CDialog::OnSysCommand(nID, lParam);
	}
}


void CLinkCreatorDlg::OnPaint() 
{
	if (IsIconic())
	{
		CPaintDC dc(this); // Ger鋞ekontext zum Zeichnen

		SendMessage(WM_ICONERASEBKGND, reinterpret_cast<WPARAM>(dc.GetSafeHdc()), 0);

		// Symbol in Clientrechteck zentrieren
		int cxIcon = GetSystemMetrics(SM_CXICON);
		int cyIcon = GetSystemMetrics(SM_CYICON);
		CRect rect;
		GetClientRect(&rect);
		int x = (rect.Width() - cxIcon + 1) / 2;
		int y = (rect.Height() - cyIcon + 1) / 2;

		// Symbol zeichnen
		dc.DrawIcon(x, y, m_hIcon);
	}
	else
	{
		CDialog::OnPaint();
	}
}

HCURSOR CLinkCreatorDlg::OnQueryDragIcon()
{
	return static_cast<HCURSOR>(m_hIcon);
}

void CLinkCreatorDlg::StartOnFolder(CString folder){

	if (PathIsDirectory(folder) && 
		folder.Right(1)!=_T("\\") && folder.Right(1)!=_T("/"))
		folder+="\\";

   if (folder.Right(1)==_T("\\") || folder.Right(1)==_T("/"))
	   folder+=_T("*.*");

   CFileFind ff;
   BOOL bFound = ff.FindFile(folder);
   bool bAich = (m_ctrlAICH.GetCheck() == BST_CHECKED);
   theApp.bFolderJob=true;

   while (bFound)
   {
		bFound = ff.FindNextFile();
		if (ff.IsDirectory())
			continue;
		
		if (!IsValidFile(ff.GetFilePath()))
			continue;

		theApp.m_saHashList.Add(ff.GetFilePath());
   }

   if (!theApp.m_saHashList.IsEmpty())
	   StartNextFile();
}

void CLinkCreatorDlg::OnBnClickedStart()
{
	theApp.bCancelHashing = false;
	CString strFullPath;
	m_ctrFilePath.GetWindowText(strFullPath);

	m_ctrlResults.SetWindowText(_T(""));

	// folder job?
	if (PathIsDirectory(strFullPath) ||
		strFullPath.Find(_T("*"))>=0    || strFullPath.Find(_T("?"))>=0 ) {
		
		StartOnFolder(strFullPath);
		return;
	}

	theApp.bFolderJob=false;
	uint32 Pos;
	if( ((Pos = strFullPath.ReverseFind('\\')) == (-1) && (Pos = strFullPath.ReverseFind('/')) == (-1) )
		|| !IsValidFile(strFullPath) ){
		AfxMessageBox(GetResString(IDS_INVALIDFILEPATH), MB_ICONSTOP, MB_OK);
		return;
	}
	CString strPath = strFullPath.Left(Pos+1);
	CString strFile = strFullPath.Mid(Pos+1);

	// already have the hashed file in m_pKnownFile?
	if (m_pKnownFile)
	{
		bool bAich = (m_ctrlAICH.GetCheck() == BST_CHECKED);
		EAICHStatus aichStatus = (m_pKnownFile->m_pAICHHashSet) ? m_pKnownFile->m_pAICHHashSet->GetStatus() : AICH_EMPTY;

		if (!strFullPath.CompareNoCase(m_pKnownFile->GetFilePath()) && (!bAich || (aichStatus == AICH_HASHSETCOMPLETE && bAich)))
		{
			RenewFileLink();
			return;
		}

		delete m_pKnownFile; // delete previous known file
		m_pKnownFile = NULL;
	}
	StartHashingFile(strPath, strFile);
}


void CLinkCreatorDlg::StartHashingFile(CString strPath, CString strFile)
{
	GetDlgItem(IDC_START)->EnableWindow(FALSE);
	GetDlgItem(IDC_CANCEL)->EnableWindow(TRUE);
	GetDlgItem(IDC_WITHAICH)->EnableWindow(FALSE);
	GetDlgItem(IDC_WITHHASHSET)->EnableWindow(FALSE);

	if (theApp.m_saHashList.IsEmpty()) {
		GetDlgItem(IDC_STATUS)->SetWindowText(GetResString(IDS_HASHING) + strFile );
	} else {
		CString info;
		info.Format(_T("%s (+%i): %s"),GetResString(IDS_HASHING) , theApp.m_saHashList.GetCount() ,strFile);

		GetDlgItem(IDC_STATUS)->SetWindowText(info);
	}

	CAddFileThread* addfilethread = (CAddFileThread*) AfxBeginThread(RUNTIME_CLASS(CAddFileThread), THREAD_PRIORITY_BELOW_NORMAL,0, CREATE_SUSPENDED);
	addfilethread->SetValues(strPath, strFile, m_ctrlAICH.GetCheck() == BST_CHECKED);
	addfilethread->ResumeThread();
}

void CLinkCreatorDlg::OnBnClickedCancel()
{
	GetDlgItem(IDC_CANCEL)->EnableWindow(FALSE);
	theApp.bCancelHashing = true;
}

bool CLinkCreatorDlg::StartNextFile() {
	if (m_pKnownFile) {
		delete m_pKnownFile;
		m_pKnownFile = NULL;
	}

	if (theApp.bCancelHashing) {
		theApp.m_saHashList.RemoveAll();
		return false;
	}

	CString nextf= theApp.m_saHashList.GetAt(0);
	theApp.m_saHashList.RemoveAt(0);
	int pos=nextf.ReverseFind(_T('\\'));
	StartHashingFile(nextf.Left(pos),nextf.Mid(pos+1));

	return true;
}