LCL history request and response. (#59)

Detect and request missing lcl history from another random node.
Sending lcl history response to a asked node.
Getting lcl history response and applying it.
Delete lcl that exceeds max ledger sequence.
This commit is contained in:
Asanka Indrajith
2019-11-19 07:50:33 -05:00
committed by GitHub
parent 0439ec93e2
commit 95683035b9
18 changed files with 1006 additions and 98 deletions

View File

@@ -4,60 +4,156 @@
#include "../crypto.hpp"
#include "../p2p/p2p.hpp"
#include "../fbschema/ledger_helpers.hpp"
#include "../fbschema/p2pmsg_helpers.hpp"
#include "ledger_handler.hpp"
#include "cons.hpp"
namespace cons
{
std::string save_ledger(const p2p::proposal &proposal, const uint64_t led_seq_no)
namespace p2pmsg = fbschema::p2pmsg;
std::string last_requested_lcl;
/**
* Create and save ledger from the given proposal message.
* @param proposal consensus reached Satge 3 proposal.
* @return tuple of current lcl sequence number and file name of the saved lcl.
*/
const std::tuple<const uint64_t, std::string> save_ledger(const p2p::proposal &proposal)
{
const size_t pos = proposal.lcl.find("-");
uint64_t led_seq_no = 0;
if (pos != std::string::npos)
{
led_seq_no = std::stoull(proposal.lcl.substr(0, pos)); //get lcl sequence number.
led_seq_no++; //current lcl sequence number.
}
else
{
//lcl records should follow [ledger sequnce numer]-lcl[lcl hex] format.
LOG_ERR << "Invalid lcl name: " << proposal.lcl << " when saving ledger.";
}
//Serialize lcl using flatbuffer ledger schema.
flatbuffers::FlatBufferBuilder builder(1024);
const std::string_view ledger_str = fbschema::ledger::create_ledger_from_proposal(builder, proposal);
const std::string_view ledger_str = fbschema::ledger::create_ledger_from_proposal(builder, proposal, led_seq_no);
//Get binary hash of the the serialized lcl.
const std::string lcl = crypto::get_hash(ledger_str);
//Get hex from binary hash
//Get hex from binary hash.
std::string lcl_hash;
util::bin2hex(lcl_hash,
reinterpret_cast<const unsigned char *>(lcl.data()),
lcl.size());
//create file path to save lcl.
//file name -> [ledger sequnce numer]-[lcl hex]
std::string path;
const std::string seq_no = std::to_string(led_seq_no);
path.reserve(conf::ctx.histdir.size() + lcl_hash.size() + seq_no.size() + 6);
path.append(conf::ctx.histdir);
path.append("/");
path.append(seq_no);
path.append("-");
path.append(lcl_hash);
path.append(".lcl");
//construct lcl file name.
//lcl file name should follow [ledger sequnce numer]-lcl[lcl hex] format.
const std::string seq_no_str = std::to_string(led_seq_no);
std::string file_name;
file_name.reserve(lcl_hash.size() + seq_no_str.size() + 1);
file_name.append(seq_no_str)
.append("-")
.append(lcl_hash);
//write lcl to file system
std::ofstream ofs(std::move(path));
ofs.write(ledger_str.data(), ledger_str.size());
ofs.close();
write_ledger(file_name, ledger_str.data(), ledger_str.size());
return lcl_hash;
cons::ctx.lcl_list.emplace(led_seq_no, file_name);
//Remove old ledgers that exceeds max sequence range.
if (led_seq_no > MAX_LEDGER_SEQUENCE)
{
remove_old_ledgers(led_seq_no - MAX_LEDGER_SEQUENCE);
}
return std::make_tuple(led_seq_no, std::move(file_name));
}
ledger_history load_ledger()
/**
* Remove old ledgers that exceeds max sequence range from file system and ledger history cache.
* @param led_seq_no minimum sequence number to be in history.
*/
void remove_old_ledgers(const uint64_t led_seq_no)
{
std::map<uint64_t, std::string>::iterator itr;
std::string dir_path;
dir_path.reserve(conf::ctx.histdir.size() + 1);
dir_path.append(conf::ctx.histdir)
.append("/");
for (itr = cons::ctx.lcl_list.begin();
itr != cons::ctx.lcl_list.lower_bound(led_seq_no + 1);
itr++)
{
const std::string file_name = itr->second;
std::string file_path;
file_path.reserve(dir_path.size() + itr->second.size() + 4);
file_path.append(dir_path)
.append(file_name)
.append(".lcl");
if (boost::filesystem::exists(file_path))
boost::filesystem::remove(file_path);
}
cons::ctx.lcl_list.erase(cons::ctx.lcl_list.begin(), cons::ctx.lcl_list.lower_bound(led_seq_no + 1));
}
/**
* Write ledger to file system.
* @param file_name current ledger sequence number.
* @param ledger_raw raw lcl data.
* @param ledger_size size of the raw lcl data.
*/
void write_ledger(const std::string &file_name, const char *ledger_raw, size_t ledger_size)
{
//create file path to save ledger.
//file name -> [ledger sequnce numer]-[lcl hex]
std::string path;
path.reserve(file_name.size() + conf::ctx.histdir.size() + 5);
path.append(conf::ctx.histdir)
.append("/")
.append(file_name)
.append(".lcl");
//write ledger to file system
std::ofstream ofs(std::move(path));
ofs.write(ledger_raw, ledger_size);
ofs.close();
}
/**
* Delete ledger from file system.
* @param file_name name of ledger to be deleted.
*/
void remove_ledger(const std::string &file_name)
{
std::string file_path;
file_path.reserve(conf::ctx.histdir.size() + file_name.size() + 5);
file_path.append(conf::ctx.histdir)
.append("/")
.append(file_name)
.append(".lcl");
boost::filesystem::remove(file_path);
}
/**
* Retrieve lcl(last closed ledger) information from ledger history.
* @return A ledger_history struct representing the lcl.
*/
const ledger_history load_ledger()
{
ledger_history ldg_hist;
ldg_hist.led_seq_no = 0;
// might need to load history in order to request response lcl history
//std::unordered_map<std::string, std::string_view> lcl_history_files;
//Get all records at lcl history direcory and find the last closed ledger.
std::string latest_file_name;
size_t latest_pos = 0;
for (const auto &entry : boost::filesystem::directory_iterator(conf::ctx.histdir))
{
const boost::filesystem::path file_path = entry.path();
const std::string file_name = entry.path().filename().string();
const std::string file_name = entry.path().stem().string();
if (boost::filesystem::is_directory(file_path))
{
@@ -70,36 +166,253 @@ ledger_history load_ledger()
else
{
const size_t pos = file_name.find("-");
uint64_t seq_no;
uint64_t seq_no = 0;
if (pos != std::string::npos)
{
seq_no = std::stoull(file_name.substr(0, pos));
ldg_hist.lcl_list.emplace(seq_no, file_name); //lcl -> [seq_no-hash]
}
else
{
//lcl records should follow [ledger sequnce numer]-lcl[lcl hex] format.
LOG_ERR << "Invalid file name: " << file_name;
}
if (seq_no > ldg_hist.led_seq_no)
{
ldg_hist.led_seq_no = seq_no;
latest_pos = pos;
latest_file_name = file_name;
LOG_ERR << "Invalid lcl file name: " << file_name << " in " << conf::ctx.histdir;
}
}
}
//check if there is a saved lcl file -> if no send genesis lcl.
if (latest_file_name.empty())
ldg_hist.lcl = "genesis";
else if ((latest_file_name.size() - 6) > latest_pos) //validation to check position is not the end of the file name.
ldg_hist.lcl = latest_file_name.substr(latest_pos + 1, (latest_file_name.size() - 6));
if (ldg_hist.lcl_list.empty())
{
ldg_hist.led_seq_no = 0;
ldg_hist.lcl = "0-genesis";
}
else
LOG_ERR << "Invalid latest file name: " << latest_file_name;
{
ldg_hist.led_seq_no = ldg_hist.lcl_list.rbegin()->first;
ldg_hist.lcl = ldg_hist.lcl_list.rbegin()->second;
//Remove old ledgers that exceeds max sequence range.
if (ldg_hist.led_seq_no > MAX_LEDGER_SEQUENCE)
{
remove_old_ledgers(ldg_hist.led_seq_no - MAX_LEDGER_SEQUENCE);
}
}
return ldg_hist;
}
/**
* Create and send ledger history request to random node from unl list.
* @param minimum_lcl hash of the minimum lcl from which node need lcl history.
* @param required_lcl hash of the required lcl.
*/
void send_ledger_history_request(const std::string &minimum_lcl, const std::string &required_lcl)
{
p2p::history_request hr;
hr.required_lcl = required_lcl;
hr.minimum_lcl = minimum_lcl;
p2p::peer_outbound_message msg(std::make_unique<flatbuffers::FlatBufferBuilder>(1024));
p2pmsg::create_msg_from_history_request(msg.builder(), hr);
p2p::send_message_to_random_peer(msg);
last_requested_lcl = required_lcl;
LOG_DBG << "Ledger history request sent."
<< " lcl:" << required_lcl;
}
/**
* Check requested lcl is in node's lcl history cache.
* @param hr lcl history request information.
* @return true if requested lcl is in lcl history cache.
*/
bool check_required_lcl_availability(const p2p::history_request &hr)
{
size_t pos = hr.required_lcl.find("-");
uint64_t req_seq_no = 0;
//get sequence number of required lcl
if (pos != std::string::npos)
{
req_seq_no = std::stoull(hr.required_lcl.substr(0, pos)); //get required lcl sequence number
}
if (req_seq_no > 0)
{
const auto itr = cons::ctx.lcl_list.find(req_seq_no);
if (itr == cons::ctx.lcl_list.end())
{
LOG_DBG << "Required lcl peer asked for is not in our lcl cache.";
//either this node is also not in consesnsus ledger or other node requesting a lcl that is older than maximum ledger range.
return false;
}
else if (itr->second != hr.required_lcl)
{
LOG_DBG << "Required lcl peer asked for is not in our lcl cache.";
//either this node or requesting node is in a fork condition.
return false;
}
}
return true;
}
/**
* Retrieve lcl(last closed ledger) information from ledger history.
* @param hr lcl history request information.
* @return A ledger history response containing requested ledger details.
*/
const p2p::history_response retrieve_ledger_history(const p2p::history_request &hr)
{
p2p::history_response history_response;
size_t pos = hr.minimum_lcl.find("-");
uint64_t min_seq_no = 0;
//get sequence number of minimum lcl required
if (pos != std::string::npos)
{
min_seq_no = std::stoull(hr.minimum_lcl.substr(0, pos)); //get required lcl sequence number
}
const auto itr = cons::ctx.lcl_list.find(min_seq_no);
if (itr != cons::ctx.lcl_list.end()) //requested minimum lcl is not in our lcl history cache
{
LOG_DBG << "Minimum lcl peer asked for is not in our lcl cache. Therefore sending from node minimum lcl";
min_seq_no = itr->first;
}
else
{
min_seq_no = cons::ctx.lcl_list.begin()->first;
}
//copy current history cache.
std::map<uint64_t, std::string> lcl_list = cons::ctx.lcl_list;
//filter out cache and get raw files here.
lcl_list.erase(
lcl_list.begin(),
lcl_list.lower_bound(min_seq_no));
for (auto &[seq_no, lcl_hash] : lcl_list)
{
p2p::history_ledger ledger;
ledger.lcl = lcl_hash;
std::string path;
path.reserve(conf::ctx.histdir.size() + lcl_hash.size() + 5);
path.append(conf::ctx.histdir)
.append("/")
.append(lcl_hash)
.append(".lcl");
//read lcl file
std::ifstream file(path, std::ios::binary | std::ios::ate);
std::streamsize size = file.tellg();
file.seekg(0, std::ios::beg);
std::vector<char> buffer(size);
if (file.read(buffer.data(), size))
{
ledger.raw_ledger = reinterpret_cast<std::vector<uint8_t> &>(buffer);
history_response.hist_ledgers.emplace(seq_no, ledger);
}
}
return history_response;
}
/**
* Send ledger history response for history request.
* @param hr lcl history request information.
* @return peer outbound message object with ledger history response.
*/
p2p::peer_outbound_message send_ledger_history(const p2p::history_request &hr)
{
p2p::peer_outbound_message msg(std::make_unique<flatbuffers::FlatBufferBuilder>(1024));
p2pmsg::create_msg_from_history_response(msg.builder(), retrieve_ledger_history(hr));
return msg;
}
/**
* Handle recieved ledger history response.
* @param hr lcl history request information.
* @return peer outbound message object with ledger history response.
*/
void handle_ledger_history_response(const p2p::history_response &hr)
{
//check response object contains
if (last_requested_lcl.empty())
{
LOG_DBG << "Peer sent us a history response but we never asked for one!";
return;
}
//check whether recieved lcl history contains the current lcl node required.
bool have_requested_lcl = false;
for (auto &[seq_no, ledger] : hr.hist_ledgers)
{
if (last_requested_lcl == ledger.lcl)
{
have_requested_lcl = true;
break;
}
}
if (!have_requested_lcl)
{
LOG_DBG << "Peer sent us a history response but not containing the lcl we asked for!";
return;
}
//Check integrity of recieved lcl list.
//By checking recieved lcl hashes matches lcl content by applying hashing for each raw content.
for (auto &[seq_no, ledger] : hr.hist_ledgers)
{
const size_t pos = ledger.lcl.find("-");
std::string rec_lcl_hash = ledger.lcl.substr((pos + 1), (ledger.lcl.size() - 1));
//Get binary hash of the the serialized lcl.
const std::string lcl = crypto::get_hash(&ledger.raw_ledger[0], ledger.raw_ledger.size());
//Get hex from binary hash
std::string lcl_hash;
util::bin2hex(lcl_hash,
reinterpret_cast<const unsigned char *>(lcl.data()),
lcl.size());
//LOG_DBG << "passed lcl: " << ledger.lcl << " gen lcl: " << lcl_hash;
//recieved lcl hash and hash generated from recieved lcl content doesn't match -> abandon applying it
if (lcl_hash != rec_lcl_hash)
{
LOG_WARN << "peer sent us a history response we asked for but the ledger data does not match the ledger hashes";
//todo: we should penalize peer who send this?
return;
}
}
//Execution to here means the history data sent checks out
//Save recieved lcl in file system and update lcl history cache
for (auto &[seq_no, ledger] : hr.hist_ledgers)
{
auto prev_dup_itr = cons::ctx.lcl_list.find(seq_no);
if (prev_dup_itr != cons::ctx.lcl_list.end())
{
remove_ledger(prev_dup_itr->second);
cons::ctx.lcl_list.erase(prev_dup_itr);
}
write_ledger(ledger.lcl, reinterpret_cast<const char *>(&ledger.raw_ledger[0]), ledger.raw_ledger.size());
cons::ctx.lcl_list.emplace(seq_no, ledger.lcl);
}
last_requested_lcl = "";
const auto latest_lcl_itr = cons::ctx.lcl_list.rbegin();
cons::ctx.lcl = latest_lcl_itr->second;
cons::ctx.led_seq_no = latest_lcl_itr->first;
}
} // namespace cons