void CKnownFileList::AddCancelledFileID(const uchar* hash){
if (thePrefs.IsRememberingCancelledFiles()){
if (m_dwCancelledFilesSeed == 0) {
m_dwCancelledFilesSeed = (GetRandomUInt32() % 0xFFFFFFFE) + 1;
}
uchar pachSeedHash[20];
PokeUInt32(pachSeedHash, m_dwCancelledFilesSeed);
md4cpy(pachSeedHash + 4, hash);
MD5Sum md5(pachSeedHash, sizeof(pachSeedHash));
md4cpy(pachSeedHash, md5.GetRawHash());
m_mapCancelledFiles.SetAt(CSKey(pachSeedHash), 1);
}
}
bool CUpDownClient::SendPeerCacheFileRequest()
{
if (GetDownloadState() == DS_ONQUEUE){
ASSERT( m_ePeerCacheDownState == PCDS_NONE );
ASSERT( m_pPCDownSocket == NULL );
}
else if (GetDownloadState() == DS_DOWNLOADING){
ASSERT( m_ePeerCacheDownState == PCDS_NONE );
ASSERT( m_pPCDownSocket == NULL );
}
else{
ASSERT(0);
}
if (!SupportPeerCache() || socket == NULL){
ASSERT(0);
return false;
}
m_uPeerCacheDownloadPushId = GetRandomUInt32();
CSafeMemFile data(128);
data.WriteUInt8(PCPCK_VERSION);
data.WriteUInt8(PCOP_REQ);
data.WriteUInt8(5);
CTag tagCacheIP(PCTAG_CACHEIP, theApp.m_pPeerCache->GetCacheIP());
tagCacheIP.WriteNewEd2kTag(&data);
CTag tagPushId(PCTAG_PUSHID, m_uPeerCacheDownloadPushId);
tagPushId.WriteNewEd2kTag(&data);
CTag tagFileId(PCTAG_FILEID, (uchar*)reqfile->GetFileHash());
tagFileId.WriteNewEd2kTag(&data);
CTag tagPublicIP(PCTAG_PUBLICIP, theApp.GetPublicIP());
tagPublicIP.WriteNewEd2kTag(&data);
CTag tagCachePort(PCTAG_CACHEPORT, theApp.m_pPeerCache->GetCachePort());
tagCachePort.WriteNewEd2kTag(&data);
if (thePrefs.GetDebugClientTCPLevel() > 0){
DebugSend("OP__PeerCacheQuery", this, reqfile->GetFileHash());
Debug(_T(" CacheIP=%s PushId=%u PublicIP=%s FileId=%s\n"), ipstr(tagCacheIP.GetInt()), tagPushId.GetInt(), ipstr(tagPublicIP.GetInt()), md4str(tagFileId.GetHash()));
}
Packet* pEd2kPacket = new Packet(&data, OP_EMULEPROT, OP_PEERCACHE_QUERY);
theStats.AddUpDataOverheadFileRequest(pEd2kPacket->size);
socket->SendPacket(pEd2kPacket);
SetDownloadState(DS_DOWNLOADING);
m_dwLastBlockReceived = ::GetTickCount();
SetPeerCacheDownState(PCDS_WAIT_CLIENT_REPLY);
return true;
}
bool CKnownFileList::LoadCancelledFiles(){
// cancelled.met Format: <Header 1 = CANCELLED_HEADER><Version 1 = CANCELLED_VERSION><Seed 4><Count 4>[<HashHash 16><TagCount 1>[Tags TagCount] Count]
if (!thePrefs.IsRememberingCancelledFiles())
return true;
CString fullpath = thePrefs.GetMuleDirectory(EMULE_CONFIGDIR);
fullpath.Append(CANCELLED_MET_FILENAME);
CSafeBufferedFile file;
CFileException fexp;
if (!file.Open(fullpath,CFile::modeRead|CFile::osSequentialScan|CFile::typeBinary|CFile::shareDenyWrite, &fexp)){
if (fexp.m_cause != CFileException::fileNotFound){
CString strError(_T("Failed to load ") CANCELLED_MET_FILENAME _T(" file"));
TCHAR szError[MAX_CFEXP_ERRORMSG];
if (fexp.GetErrorMessage(szError, ARRSIZE(szError))){
strError += _T(" - ");
strError += szError;
}
LogError(LOG_STATUSBAR, _T("%s"), strError);
}
return false;
}
setvbuf(file.m_pStream, NULL, _IOFBF, 16384);
uchar ucHash[16];
try {
bool bOldVersion = false;
uint8 header = file.ReadUInt8();
if (header != CANCELLED_HEADER){
if (header == CANCELLED_HEADER_OLD){
bOldVersion = true;
DebugLog(_T("Deprecated version of cancelled.met found, converting to new version"));
}
else{
file.Close();
return false;
}
}
uint8 byVersion = 0;
if (!bOldVersion){
byVersion = file.ReadUInt8();
if (byVersion > CANCELLED_VERSION){
file.Close();
return false;
}
m_dwCancelledFilesSeed = file.ReadUInt32();
}
if (m_dwCancelledFilesSeed == 0) {
ASSERT( bOldVersion || file.GetLength() <= 10 );
m_dwCancelledFilesSeed = (GetRandomUInt32() % 0xFFFFFFFE) + 1;
}
UINT RecordsNumber = file.ReadUInt32();
for (UINT i = 0; i < RecordsNumber; i++) {
file.ReadHash16(ucHash);
uint8 nCount = file.ReadUInt8();
// for compatibility with future versions which may add more data than just the hash
for (UINT j = 0; j < nCount; j++) {
CTag tag(&file, false);
}
if (bOldVersion){
// convert old real hash to new hashash
uchar pachSeedHash[20];
PokeUInt32(pachSeedHash, m_dwCancelledFilesSeed);
md4cpy(pachSeedHash + 4, ucHash);
MD5Sum md5(pachSeedHash, sizeof(pachSeedHash));
md4cpy(ucHash, md5.GetRawHash());
}
m_mapCancelledFiles.SetAt(CSKey(ucHash), 1);
}
file.Close();
}
catch(CFileException* error){
if (error->m_cause == CFileException::endOfFile)
LogError(LOG_STATUSBAR, GetResString(IDS_ERR_CONFIGCORRUPT), CANCELLED_MET_FILENAME);
else{
TCHAR buffer[MAX_CFEXP_ERRORMSG];
error->GetErrorMessage(buffer, ARRSIZE(buffer));
LogError(LOG_STATUSBAR, GetResString(IDS_ERR_FAILEDTOLOAD), CANCELLED_MET_FILENAME, buffer);
}
error->Delete();
return false;
}
return true;
}
