#include #include #include #include #include #include #include #ifdef _WIN32 // add the unholy windows headers for iphlpapi #include #include #include #include #else #include #endif /// do a oxenmq request on an lmq instance blocking style /// returns a json object parsed from the result std::optional LMQ_Request( oxenmq::OxenMQ& lmq, const oxenmq::ConnectionID& id, std::string_view method, std::optional args = std::nullopt) { std::promise> result_promise; auto handleRequest = [&result_promise](bool success, std::vector result) { if ((not success) or result.empty()) { result_promise.set_value(std::nullopt); return; } result_promise.set_value(result[0]); }; if (args.has_value()) { lmq.request(id, method, handleRequest, args->dump()); } else { lmq.request(id, method, handleRequest); } auto ftr = result_promise.get_future(); const auto str = ftr.get(); if (str.has_value()) return nlohmann::json::parse(*str); return std::nullopt; } int main(int argc, char* argv[]) { cxxopts::Options opts("lokinet-vpn", "LokiNET vpn control utility"); // clang-format off opts.add_options() ("v,verbose", "Verbose", cxxopts::value()) ("h,help", "help", cxxopts::value()) ("kill", "kill the daemon", cxxopts::value()) ("up", "put vpn up", cxxopts::value()) ("down", "put vpn down", cxxopts::value()) ("exit", "specify exit node address", cxxopts::value()) ("rpc", "rpc url for lokinet", cxxopts::value()) ("endpoint", "endpoint to use", cxxopts::value()) ("token", "exit auth token to use", cxxopts::value()) ("auth", "exit auth token to use", cxxopts::value()) ("status", "print status and exit", cxxopts::value()) ("range", "ip range to map", cxxopts::value()) ; // clang-format on oxenmq::address rpcURL("tcp://127.0.0.1:1190"); std::string exitAddress; std::string endpoint = "default"; std::optional token; std::string range = "::/0"; oxenmq::LogLevel logLevel = oxenmq::LogLevel::warn; bool goUp = false; bool goDown = false; bool printStatus = false; bool killDaemon = false; try { const auto result = opts.parse(argc, argv); if (result.count("help") > 0) { std::cout << opts.help() << std::endl; return 0; } if (result.count("verbose") > 0) { logLevel = oxenmq::LogLevel::debug; } if (result.count("rpc") > 0) { rpcURL = oxenmq::address(result["rpc"].as()); } if (result.count("exit") > 0) { exitAddress = result["exit"].as(); } goUp = result.count("up") > 0; goDown = result.count("down") > 0; printStatus = result.count("status") > 0; killDaemon = result.count("kill") > 0; if (result.count("endpoint") > 0) { endpoint = result["endpoint"].as(); } if (result.count("token") > 0) { token = result["token"].as(); } if (result.count("auth") > 0) { token = result["auth"].as(); } if (result.count("range") > 0) { range = result["range"].as(); } } catch (const cxxopts::option_not_exists_exception& ex) { std::cerr << ex.what(); std::cout << opts.help() << std::endl; return 1; } catch (std::exception& ex) { std::cout << ex.what() << std::endl; return 1; } if ((not goUp) and (not goDown) and (not printStatus) and (not killDaemon)) { std::cout << opts.help() << std::endl; return 1; } if (goUp and exitAddress.empty()) { std::cout << "no exit address provided" << std::endl; return 1; } oxenmq::OxenMQ lmq{ [](oxenmq::LogLevel lvl, const char* file, int line, std::string msg) { std::cout << lvl << " [" << file << ":" << line << "] " << msg << std::endl; }, logLevel}; lmq.start(); std::promise connectPromise; const auto connID = lmq.connect_remote( rpcURL, [&connectPromise](auto) { connectPromise.set_value(true); }, [&connectPromise](auto, std::string_view msg) { std::cout << "failed to connect to lokinet RPC: " << msg << std::endl; connectPromise.set_value(false); }); auto ftr = connectPromise.get_future(); if (not ftr.get()) { return 1; } if (killDaemon) { const auto maybe = LMQ_Request(lmq, connID, "llarp.halt"); if (not maybe.has_value()) { std::cout << "call to llarp.admin.die failed" << std::endl; return 1; } return 0; } if (printStatus) { const auto maybe_status = LMQ_Request(lmq, connID, "llarp.status"); if (not maybe_status.has_value()) { std::cout << "call to llarp.status failed" << std::endl; return 1; } try { const auto& ep = maybe_status->at("result").at("services").at(endpoint); const auto exitMap = ep.at("exitMap"); if (exitMap.empty()) { std::cout << "no exits" << std::endl; } else { for (const auto& [range, exit] : exitMap.items()) { std::cout << range << " via " << exit.get() << std::endl; } } } catch (std::exception& ex) { std::cout << "failed to parse result: " << ex.what() << std::endl; return 1; } return 0; } if (goUp) { std::optional maybe_result; if (token.has_value()) { maybe_result = LMQ_Request( lmq, connID, "llarp.exit", nlohmann::json{{"exit", exitAddress}, {"range", range}, {"token", *token}}); } else { maybe_result = LMQ_Request( lmq, connID, "llarp.exit", nlohmann::json{{"exit", exitAddress}, {"range", range}}); } if (not maybe_result.has_value()) { std::cout << "could not add exit" << std::endl; return 1; } if (maybe_result->contains("error") and maybe_result->at("error").is_string()) { std::cout << maybe_result->at("error").get() << std::endl; return 1; } } if (goDown) { LMQ_Request(lmq, connID, "llarp.exit", nlohmann::json{{"range", range}, {"unmap", true}}); } return 0; }