From a7cd16c1590683995751d891360dbc3dbceb8fbe Mon Sep 17 00:00:00 2001 From: orignal Date: Wed, 17 Jun 2015 12:25:02 -0400 Subject: [PATCH] use shared_ptr for direct DatabaseLookup message --- NetDbRequests.cpp | 4 ++-- NetDbRequests.h | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/NetDbRequests.cpp b/NetDbRequests.cpp index 5e71e3f7..3509bcc7 100644 --- a/NetDbRequests.cpp +++ b/NetDbRequests.cpp @@ -19,13 +19,13 @@ namespace data return msg; } - I2NPMessage * RequestedDestination::CreateRequestMessage (const IdentHash& floodfill) + std::shared_ptr RequestedDestination::CreateRequestMessage (const IdentHash& floodfill) { I2NPMessage * msg = i2p::CreateRouterInfoDatabaseLookupMsg (m_Destination, i2p::context.GetRouterInfo ().GetIdentHash () , 0, false, &m_ExcludedPeers); m_ExcludedPeers.insert (floodfill); m_CreationTime = i2p::util::GetSecondsSinceEpoch (); - return msg; + return ToSharedI2NPMessage (msg); } void RequestedDestination::ClearExcludedPeers () diff --git a/NetDbRequests.h b/NetDbRequests.h index 4f7555d4..9dc49b26 100644 --- a/NetDbRequests.h +++ b/NetDbRequests.h @@ -29,7 +29,7 @@ namespace data bool IsExcluded (const IdentHash& ident) const { return m_ExcludedPeers.count (ident); }; uint64_t GetCreationTime () const { return m_CreationTime; }; I2NPMessage * CreateRequestMessage (std::shared_ptr, std::shared_ptr replyTunnel); - I2NPMessage * CreateRequestMessage (const IdentHash& floodfill); + std::shared_ptr CreateRequestMessage (const IdentHash& floodfill); void SetRequestComplete (const RequestComplete& requestComplete) { m_RequestComplete = requestComplete; }; bool IsRequestComplete () const { return m_RequestComplete != nullptr; };