bool CServerList::SaveServermetToFile()
{
if (thePrefs.GetLogFileSaving())
AddDebugLogLine(false, _T("Saving servers list file \"%s\""), SERVER_MET_FILENAME);
m_nLastSaved = ::GetTickCount();
CString newservermet(thePrefs.GetMuleDirectory(EMULE_CONFIGDIR));
newservermet += SERVER_MET_FILENAME _T(".new");
CSafeBufferedFile servermet;
CFileException fexp;
if (!servermet.Open(newservermet, CFile::modeWrite|CFile::modeCreate|CFile::typeBinary|CFile::shareDenyWrite, &fexp)){
CString strError(GetResString(IDS_ERR_SAVESERVERMET));
TCHAR szError[MAX_CFEXP_ERRORMSG];
if (fexp.GetErrorMessage(szError, ARRSIZE(szError))){
strError += _T(" - ");
strError += szError;
}
LogError(LOG_STATUSBAR, _T("%s"), strError);
return false;
}
setvbuf(servermet.m_pStream, NULL, _IOFBF, 16384);
try{
servermet.WriteUInt8(0xE0);
UINT fservercount = list.GetCount();
servermet.WriteUInt32(fservercount);
for (UINT j = 0; j < fservercount; j++)
{
const CServer* nextserver = GetServerAt(j);
// don't write potential out-dated IPs of dynIP-servers
servermet.WriteUInt32(nextserver->HasDynIP() ? 0 : nextserver->GetIP());
servermet.WriteUInt16(nextserver->GetPort());
UINT uTagCount = 0;
ULONG uTagCountFilePos = (ULONG)servermet.GetPosition();
servermet.WriteUInt32(uTagCount);
if (!nextserver->GetListName().IsEmpty()){
CTag servername(ST_SERVERNAME, nextserver->GetListName());
servername.WriteTagToFile(&servermet, utf8strOptBOM);
uTagCount++;
}
if (!nextserver->GetDynIP().IsEmpty()){
CTag serverdynip(ST_DYNIP, nextserver->GetDynIP());
serverdynip.WriteTagToFile(&servermet, utf8strOptBOM);
uTagCount++;
}
if (!nextserver->GetDescription().IsEmpty()){
CTag serverdesc(ST_DESCRIPTION, nextserver->GetDescription());
serverdesc.WriteTagToFile(&servermet, utf8strOptBOM);
uTagCount++;
}
if (nextserver->GetFailedCount()){
CTag serverfail(ST_FAIL, nextserver->GetFailedCount());
serverfail.WriteTagToFile(&servermet);
uTagCount++;
}
if (nextserver->GetPreference() != SRV_PR_NORMAL){
CTag serverpref(ST_PREFERENCE, nextserver->GetPreference());
serverpref.WriteTagToFile(&servermet);
uTagCount++;
}
if (nextserver->GetUsers()){
CTag serveruser("users", nextserver->GetUsers());
serveruser.WriteTagToFile(&servermet);
uTagCount++;
}
if (nextserver->GetFiles()){
CTag serverfiles("files", nextserver->GetFiles());
serverfiles.WriteTagToFile(&servermet);
uTagCount++;
}
if (nextserver->GetPing()){
CTag serverping(ST_PING, nextserver->GetPing());
serverping.WriteTagToFile(&servermet);
uTagCount++;
}
if (nextserver->GetLastPingedTime()){
CTag serverlastp(ST_LASTPING, nextserver->GetLastPingedTime());
serverlastp.WriteTagToFile(&servermet);
uTagCount++;
}
if (nextserver->GetMaxUsers()){
CTag servermaxusers(ST_MAXUSERS, nextserver->GetMaxUsers());
servermaxusers.WriteTagToFile(&servermet);
uTagCount++;
}
if (nextserver->GetSoftFiles()){
CTag softfiles(ST_SOFTFILES, nextserver->GetSoftFiles());
softfiles.WriteTagToFile(&servermet);
uTagCount++;
}
if (nextserver->GetHardFiles()){
CTag hardfiles(ST_HARDFILES, nextserver->GetHardFiles());
hardfiles.WriteTagToFile(&servermet);
uTagCount++;
}
if (!nextserver->GetVersion().IsEmpty()){
// as long as we don't receive an integer version tag from the local server (TCP) we store it as string
CTag version(ST_VERSION, nextserver->GetVersion());
version.WriteTagToFile(&servermet, utf8strOptBOM);
uTagCount++;
}
if (nextserver->GetUDPFlags()){
CTag tagUDPFlags(ST_UDPFLAGS, nextserver->GetUDPFlags());
tagUDPFlags.WriteTagToFile(&servermet);
uTagCount++;
}
if (nextserver->GetLowIDUsers()){
CTag tagLowIDUsers(ST_LOWIDUSERS, nextserver->GetLowIDUsers());
tagLowIDUsers.WriteTagToFile(&servermet);
uTagCount++;
}
if (nextserver->GetServerKeyUDP(true)){
CTag tagServerKeyUDP(ST_UDPKEY, nextserver->GetServerKeyUDP(true));
tagServerKeyUDP.WriteTagToFile(&servermet);
uTagCount++;
}
if (nextserver->GetServerKeyUDPIP()){
CTag tagServerKeyUDPIP(ST_UDPKEYIP, nextserver->GetServerKeyUDPIP());
tagServerKeyUDPIP.WriteTagToFile(&servermet);
uTagCount++;
}
if (nextserver->GetObfuscationPortTCP()){
CTag tagObfuscationPortTCP(ST_TCPPORTOBFUSCATION, nextserver->GetObfuscationPortTCP());
tagObfuscationPortTCP.WriteTagToFile(&servermet);
uTagCount++;
}
if (nextserver->GetObfuscationPortUDP()){
CTag tagObfuscationPortUDP(ST_UDPPORTOBFUSCATION, nextserver->GetObfuscationPortUDP());
tagObfuscationPortUDP.WriteTagToFile(&servermet);
uTagCount++;
}
servermet.Seek(uTagCountFilePos, CFile::begin);
servermet.WriteUInt32(uTagCount);
servermet.SeekToEnd();
}
if (thePrefs.GetCommitFiles() >= 2 || (thePrefs.GetCommitFiles() >= 1 && !theApp.emuledlg->IsRunning())){
servermet.Flush(); // flush file stream buffers to disk buffers
if (_commit(_fileno(servermet.m_pStream)) != 0) // commit disk buffers to disk
AfxThrowFileException(CFileException::hardIO, GetLastError(), servermet.GetFileName());
}
servermet.Close();
CString curservermet(thePrefs.GetMuleDirectory(EMULE_CONFIGDIR));
CString oldservermet(thePrefs.GetMuleDirectory(EMULE_CONFIGDIR));
curservermet += SERVER_MET_FILENAME;
oldservermet += _T("server_met.old");
if (_taccess(oldservermet, 0) == 0)
CFile::Remove(oldservermet);
if (_taccess(curservermet, 0) == 0)
CFile::Rename(curservermet,oldservermet);
CFile::Rename(newservermet,curservermet);
}
catch(CFileException* error) {
CString strError(GetResString(IDS_ERR_SAVESERVERMET2));
TCHAR szError[MAX_CFEXP_ERRORMSG];
if (error->GetErrorMessage(szError, ARRSIZE(szError))){
strError += _T(" - ");
strError += szError;
}
LogError(LOG_STATUSBAR, _T("%s"), strError);
error->Delete();
return false;
}
return true;
}
void CRoutingZone::ReadFile(CString strSpecialNodesdate)
{
if (m_pSuperZone != NULL || (m_sFilename.IsEmpty() && strSpecialNodesdate.IsEmpty())){
ASSERT( false );
return;
}
bool bDoHaveVerifiedContacts = false;
// Read in the saved contact list.
try
{
CSafeBufferedFile file;
CFileException fexp;
if (file.Open(strSpecialNodesdate.IsEmpty() ? m_sFilename : strSpecialNodesdate, CFile::modeRead | CFile::osSequentialScan|CFile::typeBinary|CFile::shareDenyWrite, &fexp))
{
setvbuf(file.m_pStream, NULL, _IOFBF, 32768);
// Get how many contacts in the saved list.
// NOTE: Older clients put the number of contacts here..
// Newer clients always have 0 here to prevent older clients from reading it.
uint32_t uNumContacts = file.ReadUInt32();
uint32_t uVersion = 0;
if (uNumContacts == 0)
{
if (file.GetLength() >= 8){
uVersion = file.ReadUInt32();
if (uVersion == 3){
uint32_t nBoostrapEdition = file.ReadUInt32();
if (nBoostrapEdition == 1){
// this is a special bootstrap-only nodes.dat, handle it in a seperate reading function
ReadBootstrapNodesDat(file);
file.Close();
return;
}
}
if(uVersion >= 1 && uVersion <= 3) // those version we know, others we ignore
uNumContacts = file.ReadUInt32();
}
else
AddDebugLogLine( false, GetResString(IDS_ERR_KADCONTACTS));
}
if (uNumContacts != 0 && uNumContacts * 25 <= (file.GetLength() - file.GetPosition()))
{
// Hide contact list in the GUI
theApp.emuledlg->kademliawnd->HideContacts();
uint32_t uValidContacts = 0;
CUInt128 uID;
while ( uNumContacts )
{
file.ReadUInt128(&uID);
uint32_t uIP = file.ReadUInt32();
uint16_t uUDPPort = file.ReadUInt16();
uint16_t uTCPPort = file.ReadUInt16();
byte byType = 0;
uint8_t uContactVersion = 0;
if(uVersion >= 1)
uContactVersion = file.ReadUInt8();
else
byType = file.ReadUInt8();
CKadUDPKey kadUDPKey;
bool bVerified = false;
if(uVersion >= 2){
kadUDPKey.ReadFromFile(file);
bVerified = file.ReadUInt8() != 0;
if (bVerified)
bDoHaveVerifiedContacts = true;
}
// IP Appears valid
if( byType < 4)
{
uint32_t uhostIP = ntohl(uIP);
if (::IsGoodIPPort(uhostIP, uUDPPort))
{
if (::theApp.ipfilter->IsFiltered(uhostIP))
{
if (::thePrefs.GetLogFilteredIPs())
AddDebugLogLine(false, _T("Ignored kad contact (IP=%s:%u)--read known.dat -- - IP filter (%s)") , ipstr(uhostIP), uUDPPort, ::theApp.ipfilter->GetLastHit());
}
else if (uUDPPort == 53 && uContactVersion <= KADEMLIA_VERSION5_48a) /*No DNS Port without encryption*/
{
if (::thePrefs.GetLogFilteredIPs())
AddDebugLogLine(false, _T("Ignored kad contact (IP=%s:%u)--read known.dat") , ipstr(uhostIP), uUDPPort);
}
else
{
// This was not a dead contact, Inc counter if add was successful
if (AddUnfiltered(uID, uIP, uUDPPort, uTCPPort, uContactVersion, kadUDPKey, bVerified, false, true, false))
uValidContacts++;
}
}
}
uNumContacts--;
}
AddLogLine( false, GetResString(IDS_KADCONTACTSREAD), uValidContacts);
if (!bDoHaveVerifiedContacts){
DebugLogWarning(_T("No verified contacts found in nodes.dat - might be an old file version. Setting all contacts verified for this time to speed up Kad bootstrapping"));
SetAllContactsVerified();
}
}
file.Close();
}
else
DebugLogWarning(_T("Unable to read Kad file: %s"), m_sFilename);
}
catch (CFileException* e)
{
e->Delete();
DebugLogError(_T("CFileException in CRoutingZone::readFile"));
}
// Show contact list in GUI
theApp.emuledlg->kademliawnd->ShowContacts();
}