#include "config.h" #include "setup.h" #include <fstream>#include <unistd.h>#include <torrent/exceptions.h>#include <torrent/utils/log.h>#include <torrent/utils/option_strings.h> #include "control.h"#include "globals.h"#include "option_parser.h"#include "core/download_factory.h"#include "rpc/parse_commands.h"#include "session/download_storer.h"#include "utils/directory.h" void do_panic(int signum);void print_help(); //// Parse command line args and config files:// intparse_main_options(int argc, char** argv) {  try {    OptionParser optionParser;     // Converted.    optionParser.insert_flag('h', [](auto&) { print_help(); });    optionParser.insert_flag('n', [](auto&) { });    optionParser.insert_flag('D', [](auto&) { });    optionParser.insert_flag('I', [](auto&) { });    optionParser.insert_flag('K', [](auto&) { });     optionParser.insert_option('b', [](auto& arg) { rpc::call_command_set_string("network.bind_address.set", arg); });    optionParser.insert_option('d', [](auto& arg) { rpc::call_command_set_string("directory.default.set", arg); });    optionParser.insert_option('i', [](auto& arg) { rpc::call_command_set_string("network.local_address.set", arg); });    optionParser.insert_option('p', [](auto& arg) { rpc::call_command_set_string("network.listen.port.range.set", arg); });    optionParser.insert_option('s', [](auto& arg) { rpc::call_command_set_string("session", arg); });     optionParser.insert_option('O',      [](auto& arg) { rpc::parse_command_single_std(arg); });    optionParser.insert_option_list('o', [](auto& arg1, auto& arg2) { rpc::call_command_set_std_string(arg1, arg2); });     return optionParser.process(argc, argv);   } catch (torrent::input_error& e) {    throw torrent::input_error("Failed to parse command line option: " + std::string(e.what()));  }} voidparse_config_file(int argc, char** argv, std::function<void (const std::string&)> parse_fn) {  if (OptionParser::has_flag('n', argc, argv))    return parse_fn("");   char* config_dir = std::getenv("XDG_CONFIG_HOME");  char* home_dir = std::getenv("HOME");   if (config_dir != nullptr && config_dir[0] == '/' && access((std::string(config_dir) + "/rtorrent/rtorrent.rc").c_str(), F_OK) != -1)    return parse_fn(std::string(config_dir) + "/rtorrent/rtorrent.rc");   if (home_dir == nullptr)    throw torrent::input_error("Could not find home directory, HOME environment variable not set.");   if (access((std::string(home_dir) + "/.config/rtorrent/rtorrent.rc").c_str(), F_OK) != -1)    return parse_fn(std::string(home_dir) + "/.config/rtorrent/rtorrent.rc");   return parse_fn(std::string(home_dir) + "/.rtorrent.rc");} voidconfig_comment_log(const std::string& command, const std::string& raw_args) {  std::vector<std::string> args;  size_t                   pos{};   while (pos < raw_args.size()) {    size_t next_pos = raw_args.find(',', pos);     if (next_pos == std::string::npos)      next_pos = raw_args.size();     args.push_back(raw_args.substr(pos, next_pos - pos));    pos = next_pos + 1;  }   if (command == "log.add_output") {    if (args.size() != 2)      throw torrent::input_error("Invalid number of arguments.");     log_add_group_output_str(args[0], args[1]);   } else if (command == "log.open_file")    apply_log_open_str(0, args);  else if (command == "log.open_file.flush")    apply_log_open_str(log_flag_flush, args);  else if (command == "log.open_gz_file")    apply_log_open_str(log_flag_use_gz, args);  else if (command == "log.open_file_pid")    apply_log_open_str(log_flag_append_pid, args);  else if (command == "log.open_gz_file_pid")    apply_log_open_str(log_flag_append_pid | log_flag_use_gz, args);  else if (command == "log.append_file")    apply_log_open_str(log_flag_append_file, args);  else if (command == "log.append_file.flush")    apply_log_open_str(log_flag_append_file | log_flag_flush, args);  else if (command == "log.append_gz_file")    apply_log_open_str(log_flag_append_file | log_flag_use_gz, args);  else    throw torrent::input_error("Unknown log command: " + command);} // Call special commands in the format "# do:command=args" in the config file.voidparse_config_file_comments(const std::string& path) {  std::fstream file(path, std::ios::in);   if (!file.is_open())    return;   std::string line;   while (std::getline(file, line)) {    if (line.size() <= 5 || line.compare(0, 5, "# do:") != 0)      continue;     auto equal_pos = line.find('=');     if (equal_pos == std::string::npos)      throw torrent::input_error("Invalid command in config file comment: " + line);     std::string command = line.substr(5, equal_pos - 5);    std::string args    = line.substr(equal_pos + 1);     if (command.empty())      throw torrent::input_error("Invalid command in config file comment: " + line);     if (command.compare(0, 4, "log.") == 0)      config_comment_log(command, args);    else      throw torrent::input_error("Unknown command in config file comment: " + line);     lt_log_print(torrent::LOG_NOTICE, "Pre-config command: %s=%s", command.c_str(), args.c_str());  }} //// Torrents:// voidload_session_torrents(const std::string& path) {  auto entries = session::DownloadStorer::get_formated_entries(path);   for (const auto& entry : entries) {    // We don't really support session torrents that are links. These    // would be overwritten anyway on exit, and thus not really be    // useful.    if (!entry.is_file())      continue;     auto* f = new core::DownloadFactory(control->core());     f->set_session(true);    f->set_init_load(true);    f->slot_finished([f](){ delete f; });    f->load(entries.path() + entry.s_name);    f->commit();  }} voidload_arg_torrents(char** first, char** last) {  for (; first != last; ++first) {    auto* f = new core::DownloadFactory(control->core());     f->set_start(true);    f->set_init_load(true);    f->slot_finished([f](){ delete f; });    f->load(*first);    f->commit();  }} //// Logging:// voidlog_add_group_output_str(const std::string& group_name, const std::string& output_id) {  int log_group = torrent::option_find_string(torrent::OPTION_LOG_GROUP, group_name.c_str());   torrent::log_add_group_output(log_group, output_id.c_str());} voidapply_log_open_str(int output_flags, const std::vector<std::string>& args) {  if (args.size() < 2)    throw torrent::input_error("Invalid number of arguments.");   auto output_id = args[0];  auto file_name = expand_path(args[1]);   if ((output_flags & log_flag_append_pid)) {    char buffer[32];    snprintf(buffer, 32, ".%li", (long)getpid());     file_name += buffer;  }   bool append = (output_flags & log_flag_append_file);  bool flush = (output_flags & log_flag_flush);   if ((output_flags & log_flag_use_gz))    torrent::log_open_gz_file_output(output_id.c_str(), file_name.c_str(), append);  else    torrent::log_open_file_output(output_id.c_str(), file_name.c_str(), append, flush);   for (size_t i = 2; i < args.size(); i++)    log_add_group_output_str(args[i], output_id);}