|
|
|
@ -3,11 +3,16 @@
@@ -3,11 +3,16 @@
|
|
|
|
|
#include <string> |
|
|
|
|
#include <map> |
|
|
|
|
#include <fstream> |
|
|
|
|
#include <chrono> |
|
|
|
|
#include <condition_variable> |
|
|
|
|
#include <boost/filesystem.hpp> |
|
|
|
|
#include <boost/lexical_cast.hpp> |
|
|
|
|
#include "base64.h" |
|
|
|
|
#include "util.h" |
|
|
|
|
#include "Identity.h" |
|
|
|
|
#include "Log.h" |
|
|
|
|
#include "NetDb.h" |
|
|
|
|
#include "ClientContext.h" |
|
|
|
|
#include "AddressBook.h" |
|
|
|
|
|
|
|
|
|
namespace i2p |
|
|
|
@ -309,6 +314,68 @@ namespace client
@@ -309,6 +314,68 @@ namespace client
|
|
|
|
|
m_IsLoaded = true; |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
AddressBookSubscription::AddressBookSubscription (AddressBook& book, const std::string& link): |
|
|
|
|
m_Book (book), m_Link (link) |
|
|
|
|
{ |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
void AddressBookSubscription::CheckSubscription () |
|
|
|
|
{ |
|
|
|
|
std::thread load_hosts(&AddressBookSubscription::Request, this); |
|
|
|
|
load_hosts.detach(); |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
void AddressBookSubscription::Request () |
|
|
|
|
{ |
|
|
|
|
// must be run in separate thread
|
|
|
|
|
i2p::util::http::url u (m_Link); |
|
|
|
|
i2p::data::IdentHash ident; |
|
|
|
|
if (m_Book.GetIdentHash (u.host_, ident)) |
|
|
|
|
{ |
|
|
|
|
auto leaseSet = i2p::data::netdb.FindLeaseSet (ident); |
|
|
|
|
if (!leaseSet) |
|
|
|
|
{ |
|
|
|
|
i2p::data::netdb.RequestDestination (ident, true, i2p::client::context.GetSharedLocalDestination ()->GetTunnelPool ()); |
|
|
|
|
std::this_thread::sleep_for (std::chrono::seconds (5)); // wait for 5 seconds
|
|
|
|
|
leaseSet = i2p::data::netdb.FindLeaseSet (ident); |
|
|
|
|
} |
|
|
|
|
if (leaseSet) |
|
|
|
|
{ |
|
|
|
|
std::stringstream request, response; |
|
|
|
|
request << "GET " << u.path_ << " HTTP/1.0\r\nHost: " << u.host_ |
|
|
|
|
<< "\r\nAccept: */*\r\n" << "User-Agent: Wget/1.11.4\r\n" << "Connection: close\r\n\r\n"; |
|
|
|
|
|
|
|
|
|
auto stream = i2p::client::context.GetSharedLocalDestination ()->CreateStream (*leaseSet, u.port_); |
|
|
|
|
stream->Send ((uint8_t *)request.str ().c_str (), request.str ().length ()); |
|
|
|
|
|
|
|
|
|
uint8_t buf[4095]; |
|
|
|
|
bool end = false; |
|
|
|
|
while (!end) |
|
|
|
|
{ |
|
|
|
|
std::condition_variable newDataReceived; |
|
|
|
|
std::mutex newDataReceivedMutex; |
|
|
|
|
stream->AsyncReceive (boost::asio::buffer (buf, 4096), |
|
|
|
|
[&](const boost::system::error_code& ecode, std::size_t bytes_transferred) |
|
|
|
|
{ |
|
|
|
|
if (!ecode) |
|
|
|
|
response.write ((char *)buf, bytes_transferred); |
|
|
|
|
else |
|
|
|
|
end = true; |
|
|
|
|
newDataReceived.notify_one (); |
|
|
|
|
}, |
|
|
|
|
30); // wait for 30 seconds
|
|
|
|
|
std::unique_lock<std::mutex> l(newDataReceivedMutex); |
|
|
|
|
newDataReceived.wait (l); |
|
|
|
|
if (!end) |
|
|
|
|
end = !stream->IsOpen (); |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
else |
|
|
|
|
LogPrint (eLogError, "Address ", u.host_, " not found"); |
|
|
|
|
} |
|
|
|
|
else |
|
|
|
|
LogPrint (eLogError, "Can't resolve ", u.host_); |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|