emule电驴源码（VC）

//this file is part of eMule //Copyright (C)2002 Merkur ( merkur-@users.sourceforge.net / 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 <math.h> #include <sys/stat.h> #include <io.h> #include <winioctl.h> #ifdef _DEBUG #include "DebugHelpers.h" #endif #include "emule.h" #include "PartFile.h" #include "UpDownClient.h" #include "ED2KLink.h" #include "Preview.h" #include "ArchiveRecovery.h" #include "SearchList.h" #include "Kademlia/Kademlia/Kademlia.h" #include "kademlia/kademlia/search.h" #include "kademlia/kademlia/SearchManager.h" #include "kademlia/utils/MiscUtils.h" #include "kademlia/kademlia/prefs.h" #include "DownloadQueue.h" #include "IPFilter.h" #include "MMServer.h" #include "UploadQueue.h" #include "OtherFunctions.h" #include "Packets.h" #include "Preferences.h" #include "SafeFile.h" #include "SharedFileList.h" #include "ListenSocket.h" #include "Sockets.h" #include "Server.h" #include "KnownFileList.h" #ifndef _CONSOLE #include "emuledlg.h" #include "TransferWnd.h" #include "TaskbarNotifier.h" #endif #ifdef _DEBUG #undef THIS_FILE static char THIS_FILE[]=__FILE__; #define new DEBUG_NEW #endif // Barry - use this constant for both places #define PROGRESS_HEIGHT 3 CBarShader CPartFile::s_LoadBar(PROGRESS_HEIGHT); // Barry - was 5 CBarShader CPartFile::s_ChunkBar(16); IMPLEMENT_DYNAMIC(CPartFile, CKnownFile) CPartFile::CPartFile() { Init(); } CPartFile::CPartFile(CSearchFile* searchresult) { Init(); md4cpy(m_abyFileHash, searchresult->GetFileHash()); for (int i = 0; i < searchresult->taglist.GetCount();i++){ const CTag* pTag = searchresult->taglist[i]; switch (pTag->tag.specialtag){ case FT_FILENAME:{ if (pTag->IsStr()) SetFileName(pTag->tag.stringvalue); break; } case FT_FILESIZE:{ if (pTag->IsInt()) SetFileSize(pTag->tag.intvalue); break; } default:{ bool bTagAdded = false; if (pTag->tag.specialtag != 0 && pTag->tag.tagname == NULL && (pTag->IsStr() || pTag->IsInt())) { static const struct { uint8 nName; uint8 nType; } _aMetaTags[] = { { FT_MEDIA_ARTIST, 2 }, { FT_MEDIA_ALBUM, 2 }, { FT_MEDIA_TITLE, 2 }, { FT_MEDIA_LENGTH, 3 }, { FT_MEDIA_BITRATE, 3 }, { FT_MEDIA_CODEC, 2 }, { FT_FILETYPE, 2 }, { FT_FILEFORMAT, 2 } }; for (int t = 0; t < ARRSIZE(_aMetaTags); t++) { if (pTag->tag.type == _aMetaTags[t].nType && pTag->tag.specialtag == _aMetaTags[t].nName) { // skip string tags with empty string values if (pTag->IsStr() && (pTag->tag.stringvalue == NULL || pTag->tag.stringvalue[0] == '\0')) break; // skip integer tags with '0' values if (pTag->IsInt() && pTag->tag.intvalue == 0) break; TRACE("CPartFile::CPartFile(CSearchFile*): added tag %s\n", pTag->GetFullInfo()); CTag* newtag = new CTag(pTag->tag); taglist.Add(newtag); bTagAdded = true; break; } } } if (!bTagAdded) TRACE("CPartFile::CPartFile(CSearchFile*): ignored tag %s\n", pTag->GetFullInfo()); } } } CreatePartFile(); } CPartFile::CPartFile(CString edonkeylink) { CED2KLink* pLink = 0; try { pLink = CED2KLink::CreateLinkFromUrl(edonkeylink); _ASSERT( pLink != 0 ); CED2KFileLink* pFileLink = pLink->GetFileLink(); if (pFileLink==0) throw GetResString(IDS_ERR_NOTAFILELINK); InitializeFromLink(pFileLink); } catch (CString error) { char buffer[200]; sprintf(buffer,GetResString(IDS_ERR_INVALIDLINK),error.GetBuffer()); AddLogLine(true, GetResString(IDS_ERR_LINKERROR), buffer); SetStatus(PS_ERROR); } delete pLink; } void CPartFile::InitializeFromLink(CED2KFileLink* fileLink) { Init(); try{ SetFileName(fileLink->GetName()); SetFileSize(fileLink->GetSize()); md4cpy(m_abyFileHash, fileLink->GetHashKey()); if (!theApp.downloadqueue->IsFileExisting(m_abyFileHash)) CreatePartFile(); else SetStatus(PS_ERROR); } catch(CString error){ char buffer[200]; sprintf(buffer, GetResString(IDS_ERR_INVALIDLINK), error.GetBuffer()); AddLogLine(true, GetResString(IDS_ERR_LINKERROR), buffer); SetStatus(PS_ERROR); } } CPartFile::CPartFile(CED2KFileLink* fileLink) { InitializeFromLink(fileLink); } void CPartFile::Init(){ newdate = true; lastsearchtime = 0; lastsearchtimeKad = 0; lastpurgetime = ::GetTickCount(); paused = false; stopped= false; status = PS_EMPTY; insufficient = false; m_bCompletionError = false; transfered = 0; m_iLastPausePurge = time(NULL); m_AllocateThread=NULL; if(thePrefs.GetNewAutoDown()){ m_iDownPriority = PR_HIGH; m_bAutoDownPriority = true; } else{ m_iDownPriority = PR_NORMAL; m_bAutoDownPriority = false; } srcarevisible = false; memset(m_anStates,0,sizeof(m_anStates)); datarate = 0; hashsetneeded = true; count = 0; percentcompleted = 0; completedsize=0; m_bPreviewing = false; lastseencomplete = NULL; availablePartsCount=0; m_ClientSrcAnswered = 0; m_LastNoNeededCheck = 0; m_iRate = 0; m_strComment = ""; m_nTotalBufferData = 0; m_nLastBufferFlushTime = 0; m_bRecoveringArchive = false; m_iGainDueToCompression = 0; m_iLostDueToCorruption = 0; m_iTotalPacketsSavedDueToICH = 0; hasRating = false; hasComment = false; hasBadRating= false; m_category=0; m_lastRefreshedDLDisplay = 0; m_is_A4AF_auto=false; m_bLocalSrcReqQueued = false; memset(src_stats,0,sizeof(src_stats)); m_nCompleteSourcesTime = time(NULL); m_nCompleteSourcesCount = 0; m_nCompleteSourcesCountLo = 0; m_nCompleteSourcesCountHi = 0; m_dwFileAttributes = 0; m_bDeleteAfterAlloc=false; m_tActivated = 0; m_nDlActiveTime = 0; m_tLastModified = 0; m_tCreated = 0; } CPartFile::~CPartFile() { // Barry - Ensure all buffered data is written try{ if (m_AllocateThread != NULL){ HANDLE hThread = m_AllocateThread->m_hThread; // 2 minutes to let the thread finish if (WaitForSingleObject(hThread, 120000) == WAIT_TIMEOUT) TerminateThread(hThread, 100); } if (m_hpartfile.m_hFile != INVALID_HANDLE_VALUE) FlushBuffer(true); } catch(CFileException* e){ e->Delete(); } if (m_hpartfile.m_hFile != INVALID_HANDLE_VALUE){ // commit file and directory entry m_hpartfile.Close(); // Update met file (with current directory entry) SavePartFile(); } for (POSITION pos = gaplist.GetHeadPosition();pos != 0;) delete gaplist.GetNext(pos); pos = m_BufferedData_list.GetHeadPosition(); while (pos){ PartFileBufferedData *item = m_BufferedData_list.GetNext(pos); delete[] item->data; delete item; } } #ifdef _DEBUG void CPartFile::AssertValid() const { CKnownFile::AssertValid(); (void)lastsearchtime; (void)lastsearchtimeKad; srclist.AssertValid(); A4AFsrclist.AssertValid(); (void)lastseencomplete; m_hpartfile.AssertValid(); m_FileCompleteMutex.AssertValid(); (void)src_stats; CHECK_BOOL(m_bPreviewing); CHECK_BOOL(m_bRecoveringArchive); CHECK_BOOL(m_bLocalSrcReqQueued); CHECK_BOOL(srcarevisible); CHECK_BOOL(hashse