Browse Source

Merge pull request #210 from mlt/fixes

Few fixes
pull/213/head
orignal 9 years ago
parent
commit
d07c68bd9a
  1. 3
      DaemonLinux.cpp
  2. 2
      HTTPProxy.cpp
  3. 42
      NetDb.cpp
  4. 4
      NetDb.h
  5. 9
      RouterInfo.cpp
  6. 6
      SAM.cpp

3
DaemonLinux.cpp

@ -62,7 +62,8 @@ namespace i2p @@ -62,7 +62,8 @@ namespace i2p
LogPrint("Error, could not create process group.");
return false;
}
chdir(i2p::util::filesystem::GetDataDir().string().c_str());
std::string d(i2p::util::filesystem::GetDataDir()); // make a copy
chdir(d.c_str());
// close stdin/stdout/stderr descriptors
::close (0);

2
HTTPProxy.cpp

@ -90,7 +90,7 @@ namespace proxy @@ -90,7 +90,7 @@ namespace proxy
//TODO: handle this apropriately
void HTTPProxyHandler::HTTPRequestFailed(/*HTTPProxyHandler::errTypes error*/)
{
std::string response = "HTTP/1.0 500 Internal Server Error\r\nContent-type: text/html\r\nContent-length: 0\r\n";
static std::string response = "HTTP/1.0 500 Internal Server Error\r\nContent-type: text/html\r\nContent-length: 0\r\n";
boost::asio::async_write(*m_sock, boost::asio::buffer(response,response.size()),
std::bind(&HTTPProxyHandler::SentHTTPFailed, shared_from_this(), std::placeholders::_1));
}

42
NetDb.cpp

@ -21,11 +21,7 @@ namespace i2p @@ -21,11 +21,7 @@ namespace i2p
{
namespace data
{
#ifndef _WIN32
const char NetDb::m_NetDbPath[] = "/netDb";
#else
const char NetDb::m_NetDbPath[] = "\\netDb";
#endif
const char NetDb::m_NetDbPath[] = "netDb";
NetDb netdb;
NetDb::NetDb (): m_IsRunning (false), m_Thread (nullptr), m_Reseeder (nullptr)
@ -40,7 +36,7 @@ namespace data @@ -40,7 +36,7 @@ namespace data
void NetDb::Start ()
{
Load (m_NetDbPath);
Load ();
if (m_RouterInfos.size () < 25) // reseed if # of router less than 50
{
// try SU3 first
@ -55,7 +51,7 @@ namespace data @@ -55,7 +51,7 @@ namespace data
{
m_Reseeder->reseedNow();
reseedRetries++;
Load (m_NetDbPath);
Load ();
}
}
}
@ -133,7 +129,7 @@ namespace data @@ -133,7 +129,7 @@ namespace data
{
if (lastSave)
{
SaveUpdated (m_NetDbPath);
SaveUpdated ();
ManageLeaseSets ();
}
lastSave = ts;
@ -295,10 +291,9 @@ namespace data @@ -295,10 +291,9 @@ namespace data
LogPrint (eLogWarning, "Failed to reseed after 10 attempts");
}
void NetDb::Load (const char * directory)
void NetDb::Load ()
{
boost::filesystem::path p (i2p::util::filesystem::GetDataDir());
p /= (directory);
boost::filesystem::path p(i2p::util::filesystem::GetDataDir() / m_NetDbPath);
if (!boost::filesystem::exists (p))
{
// seems netDb doesn't exist yet
@ -345,27 +340,15 @@ namespace data @@ -345,27 +340,15 @@ namespace data
LogPrint (m_Floodfills.size (), " floodfills loaded");
}
void NetDb::SaveUpdated (const char * directory)
void NetDb::SaveUpdated ()
{
auto GetFilePath = [](const char * directory, const RouterInfo * routerInfo)
auto GetFilePath = [](const boost::filesystem::path& directory, const RouterInfo * routerInfo)
{
#ifndef _WIN32
return std::string (directory) + "/r" +
routerInfo->GetIdentHashBase64 ()[0] + "/routerInfo-" +
#else
return std::string (directory) + "\\r" +
routerInfo->GetIdentHashBase64 ()[0] + "\\routerInfo-" +
#endif
routerInfo->GetIdentHashBase64 () + ".dat";
std::string s(routerInfo->GetIdentHashBase64());
return directory / (std::string("r") + s[0]) / ("routerInfo-" + s + ".dat");
};
boost::filesystem::path p (i2p::util::filesystem::GetDataDir());
p /= (directory);
#if BOOST_VERSION > 10500
const char * fullDirectory = p.string().c_str ();
#else
const char * fullDirectory = p.c_str ();
#endif
boost::filesystem::path fullDirectory (i2p::util::filesystem::GetDataDir() / m_NetDbPath);
int count = 0, deletedCount = 0;
auto total = m_RouterInfos.size ();
uint64_t ts = i2p::util::GetMillisecondsSinceEpoch ();
@ -373,7 +356,8 @@ namespace data @@ -373,7 +356,8 @@ namespace data
{
if (it.second->IsUpdated ())
{
it.second->SaveToFile (GetFilePath(fullDirectory, it.second.get ()));
std::string f = GetFilePath(fullDirectory, it.second.get()).string();
it.second->SaveToFile (f);
it.second->SetUpdated (false);
it.second->SetUnreachable (false);
it.second->DeleteBuffer ();

4
NetDb.h

@ -68,8 +68,8 @@ namespace data @@ -68,8 +68,8 @@ namespace data
private:
bool CreateNetDb(boost::filesystem::path directory);
void Load (const char * directory);
void SaveUpdated (const char * directory);
void Load ();
void SaveUpdated ();
void Run (); // exploratory thread
void Explore (int numDestinations);
void Publish ();

9
RouterInfo.cpp

@ -447,10 +447,13 @@ namespace data @@ -447,10 +447,13 @@ namespace data
if (m_Buffer)
{
std::ofstream f (fullPath, std::ofstream::binary | std::ofstream::out);
f.write ((char *)m_Buffer, m_BufferLen);
}
if (f.is_open ())
f.write ((char *)m_Buffer, m_BufferLen);
else
LogPrint(eLogError, "Can't save RouterInfo to ", fullPath);
}
else
LogPrint (eLogError, "Can't save to file");
LogPrint (eLogError, "Can't save RouterInfo m_Buffer==NULL");
}
size_t RouterInfo::ReadString (char * str, std::istream& s)

6
SAM.cpp

@ -85,6 +85,9 @@ namespace client @@ -85,6 +85,9 @@ namespace client
else
{
m_Buffer[bytes_transferred] = 0;
char * eol = (char *)memchr (m_Buffer, '\n', bytes_transferred);
if (eol)
*eol = 0;
LogPrint ("SAM handshake ", m_Buffer);
char * separator = strchr (m_Buffer, ' ');
if (separator)
@ -101,9 +104,6 @@ namespace client @@ -101,9 +104,6 @@ namespace client
if (separator)
{
separator++;
char *eol = strchr (separator, '\n');
if (eol)
*eol = 0;
std::map<std::string, std::string> params;
ExtractParams (separator, params);
auto it = params.find (SAM_PARAM_MAX);

Loading…
Cancel
Save