oxen-io
/
lokinet
mirror of https://github.com/oxen-io/lokinet
1
1
Fork 0
Code Issues Projects Releases Wiki Activity

update lokinet-bootstrap 

* compat with old shell script curl wrapper's arguments
* do lazy validation of fetched data
This commit is contained in:
Jeff Becker 2021-04-16 14:16:16 -04:00
parent 77de60af91
commit a80d8b95b3
No known key found for this signature in database
GPG Key ID: F357B3B42F6F9B05
1 changed files with 36 additions and 22 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

58
daemon/lokinet-bootstrap.cpp
View File

@ -1,11 +1,13 @@
#include <cpr/cpr.h>
#include <llarp/constants/files.hpp>
#include <llarp/constants/version.hpp>
#include <llarp/util/buffer.hpp>
#include <llarp/util/fs.hpp>
#include <llarp/router_contact.hpp>
#include <fstream>
#include <sstream>
#include <iostream>
#include <unordered_map>
#ifndef _WIN32
#include <openssl/x509.h>
@ -14,21 +16,41 @@
namespace
{
int
exit_with_message(std::string msg, int exitcode)
fail(std::string msg)
{
std::cout << msg << std::endl;
return exitcode;
return 1;
}
} // namespace
int
main(int argc, char* argv[])
{
std::string bootstrap_url{"https://seed.lokinet.org/lokinet.signed"};
const std::unordered_map<std::string, std::string> bootstrap_urls = {
{"mainnet", "https://seed.lokinet.org/lokinet.signed"},
{"lokinet", "https://seed.lokinet.org/lokinet.signed"},
{"testnet", "https://seed.lokinet.org/testnet.signed"},
{"gamma", "https://seed.lokinet.org/testnet.signed"}};
std::string bootstrap_url = bootstrap_urls.at("lokinet");
fs::path outputfile{llarp::GetDefaultBootstrap()};
if (argc > 1)
{
bootstrap_url = argv[1];
if (auto itr = bootstrap_urls.find(argv[1]); itr != bootstrap_urls.end())
{
bootstrap_url = itr->second;
}
else
{
bootstrap_url = argv[1];
}
}
if (argc > 2)
{
outputfile = fs::path{argv[2]};
}
cpr::Response resp =
#ifdef _WIN32
cpr::Get(
@ -41,24 +63,16 @@ main(int argc, char* argv[])
#endif
if (resp.status_code != 200)
{
return exit_with_message(
"failed to fetch '" + bootstrap_url + "' HTTP " + std::to_string(resp.status_code), 1);
return fail("failed to fetch '" + bootstrap_url + "' HTTP " + std::to_string(resp.status_code));
}
std::stringstream ss;
ss << resp.text;
std::string data{ss.str()};
llarp_buffer_t buf{&data[0], data.size()};
llarp::RouterContact rc;
if (not rc.BDecode(&buf))
return exit_with_message("invalid bootstrap data was fetched", 1);
const auto path = llarp::GetDefaultBootstrap();
if (not rc.Write(path))
return exit_with_message("failed to write bootstrap file to " + path.string(), 1);
const llarp::RouterID router{rc.pubkey};
return exit_with_message("fetched bootstrap file for " + router.ToString(), 0);
if (data[0] == 'l' or data[0] == 'd')
{
std::ofstream ofs{outputfile};
ofs << data;
return 0;
}
return fail("got invalid bootstrap file content");
}