mirror of
https://github.com/rakshasa/rtorrent.git
synced 2026-08-06 18:22:32 +00:00
rakshasa
7600a51039
small. Patch by Josef Drexler. * Added 'view.persistance' command that makes downloads inserted into that view persist across sessions. Only call on user-created views. * Added 'ratio.*' commands that call the 'group.seeding.ratio.*' equivalents. git-svn-id: svn://rakshasa.no/libtorrent/trunk/rtorrent@1080 e378c898-3ddf-0310-93e7-cc216c733640
406 lines
16 KiB
C++
406 lines
16 KiB
C++
// rTorrent - BitTorrent client
|
|
// Copyright (C) 2005-2007, Jari Sundell
|
|
//
|
|
// This program is free software; you can redistribute it and/or modify
|
|
// it under the terms of the GNU General Public License as published by
|
|
// the Free Software Foundation; either version 2 of the License, or
|
|
// (at your option) any later version.
|
|
//
|
|
// This program is distributed in the hope that it will be useful,
|
|
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
// GNU General Public License for more details.
|
|
//
|
|
// You should have received a copy of the GNU General Public License
|
|
// along with this program; if not, write to the Free Software
|
|
// Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
|
|
//
|
|
// In addition, as a special exception, the copyright holders give
|
|
// permission to link the code of portions of this program with the
|
|
// OpenSSL library under certain conditions as described in each
|
|
// individual source file, and distribute linked combinations
|
|
// including the two.
|
|
//
|
|
// You must obey the GNU General Public License in all respects for
|
|
// all of the code used other than OpenSSL. If you modify file(s)
|
|
// with this exception, you may extend this exception to your version
|
|
// of the file(s), but you are not obligated to do so. If you do not
|
|
// wish to do so, delete this exception statement from your version.
|
|
// If you delete this exception statement from all source files in the
|
|
// program, then also delete it here.
|
|
//
|
|
// Contact: Jari Sundell <jaris@ifi.uio.no>
|
|
//
|
|
// Skomakerveien 33
|
|
// 3185 Skoppum, NORWAY
|
|
|
|
#include "config.h"
|
|
|
|
#include <cstdlib>
|
|
#include <iostream>
|
|
#include <string>
|
|
#include <sigc++/adaptors/bind.h>
|
|
#include <torrent/http.h>
|
|
#include <torrent/torrent.h>
|
|
#include <torrent/exceptions.h>
|
|
#include <rak/functional.h>
|
|
|
|
#ifdef USE_EXECINFO
|
|
#include <execinfo.h>
|
|
#endif
|
|
|
|
#include "core/dht_manager.h"
|
|
#include "core/download.h"
|
|
#include "core/download_factory.h"
|
|
#include "core/download_store.h"
|
|
#include "core/manager.h"
|
|
#include "display/canvas.h"
|
|
#include "display/window.h"
|
|
#include "display/manager.h"
|
|
#include "input/bindings.h"
|
|
|
|
#include "rpc/command_scheduler.h"
|
|
#include "rpc/command_scheduler_item.h"
|
|
#include "rpc/parse_commands.h"
|
|
#include "utils/directory.h"
|
|
|
|
#include "control.h"
|
|
#include "globals.h"
|
|
#include "signal_handler.h"
|
|
#include "option_parser.h"
|
|
|
|
void do_panic(int signum);
|
|
void print_help();
|
|
void initialize_commands();
|
|
|
|
int
|
|
parse_options(Control* c, int argc, char** argv) {
|
|
try {
|
|
OptionParser optionParser;
|
|
|
|
// Converted.
|
|
optionParser.insert_flag('h', sigc::ptr_fun(&print_help));
|
|
optionParser.insert_flag('n', OptionParser::Slot());
|
|
|
|
optionParser.insert_option('b', sigc::bind<0>(sigc::ptr_fun(&rpc::call_command_set_string), "bind"));
|
|
optionParser.insert_option('d', sigc::bind<0>(sigc::ptr_fun(&rpc::call_command_set_string), "directory"));
|
|
optionParser.insert_option('i', sigc::bind<0>(sigc::ptr_fun(&rpc::call_command_set_string), "ip"));
|
|
optionParser.insert_option('p', sigc::bind<0>(sigc::ptr_fun(&rpc::call_command_set_string), "port_range"));
|
|
optionParser.insert_option('s', sigc::bind<0>(sigc::ptr_fun(&rpc::call_command_set_string), "session"));
|
|
|
|
optionParser.insert_option('O', sigc::ptr_fun(&rpc::parse_command_single_std));
|
|
optionParser.insert_option_list('o', sigc::ptr_fun(&rpc::call_command_set_std_string));
|
|
|
|
return optionParser.process(argc, argv);
|
|
|
|
} catch (torrent::input_error& e) {
|
|
throw torrent::input_error("Failed to parse command line option: " + std::string(e.what()));
|
|
}
|
|
}
|
|
|
|
void
|
|
load_session_torrents(Control* c) {
|
|
utils::Directory entries = c->core()->download_store()->get_formated_entries();
|
|
|
|
for (utils::Directory::const_iterator first = entries.begin(), last = entries.end(); first != last; ++first) {
|
|
// We don't really support session torrents that are links. These
|
|
// would be overwritten anyway on exit, and thus not really be
|
|
// useful.
|
|
if (!first->is_file())
|
|
continue;
|
|
|
|
core::DownloadFactory* f = new core::DownloadFactory(c->core());
|
|
|
|
// Replace with session torrent flag.
|
|
f->set_session(true);
|
|
f->slot_finished(sigc::bind(sigc::ptr_fun(&rak::call_delete_func<core::DownloadFactory>), f));
|
|
f->load(entries.path() + first->d_name);
|
|
f->commit();
|
|
}
|
|
}
|
|
|
|
void
|
|
load_arg_torrents(Control* c, char** first, char** last) {
|
|
//std::for_each(begin, end, std::bind1st(std::mem_fun(&core::Manager::insert), &c->get_core()));
|
|
for (; first != last; ++first) {
|
|
core::DownloadFactory* f = new core::DownloadFactory(c->core());
|
|
|
|
// Replace with session torrent flag.
|
|
f->set_start(true);
|
|
f->slot_finished(sigc::bind(sigc::ptr_fun(&rak::call_delete_func<core::DownloadFactory>), f));
|
|
f->load(*first);
|
|
f->commit();
|
|
}
|
|
}
|
|
|
|
static inline rak::timer
|
|
client_next_timeout(Control* c) {
|
|
if (taskScheduler.empty())
|
|
return c->is_shutdown_started() ? rak::timer::from_milliseconds(100) : rak::timer::from_seconds(60);
|
|
else if (taskScheduler.top()->time() <= cachedTime)
|
|
return 0;
|
|
else
|
|
return taskScheduler.top()->time() - cachedTime;
|
|
}
|
|
|
|
int
|
|
main(int argc, char** argv) {
|
|
try {
|
|
|
|
// Temporary.
|
|
setlocale(LC_ALL, "");
|
|
|
|
cachedTime = rak::timer::current();
|
|
|
|
control = new Control;
|
|
|
|
srandom(cachedTime.usec());
|
|
srand48(cachedTime.usec());
|
|
|
|
SignalHandler::set_ignore(SIGPIPE);
|
|
SignalHandler::set_handler(SIGINT, sigc::mem_fun(control, &Control::receive_normal_shutdown));
|
|
SignalHandler::set_handler(SIGTERM, sigc::mem_fun(control, &Control::receive_quick_shutdown));
|
|
SignalHandler::set_handler(SIGWINCH, sigc::mem_fun(control->display(), &display::Manager::force_redraw));
|
|
SignalHandler::set_handler(SIGSEGV, sigc::bind(sigc::ptr_fun(&do_panic), SIGSEGV));
|
|
SignalHandler::set_handler(SIGBUS, sigc::bind(sigc::ptr_fun(&do_panic), SIGBUS));
|
|
SignalHandler::set_handler(SIGFPE, sigc::bind(sigc::ptr_fun(&do_panic), SIGFPE));
|
|
|
|
control->core()->initialize_first();
|
|
|
|
// Initialize option handlers after libtorrent to ensure
|
|
// torrent::ConnectionManager* are valid etc.
|
|
initialize_commands();
|
|
|
|
rpc::parse_command_multiple
|
|
(rpc::make_target(),
|
|
// "system.method.insert = test.value,value\n"
|
|
// "system.method.insert = test.value2,value,6\n"
|
|
|
|
// "system.method.insert = test.string,string,6\n"
|
|
// "system.method.insert = test.bool,bool,true\n"
|
|
|
|
"system.method.insert = test.method.simple,simple,\"print=simple_test_,$argument.0=\"\n"
|
|
|
|
"system.method.insert = event.download.inserted,multi\n"
|
|
"system.method.insert = event.download.inserted_new,multi\n"
|
|
"system.method.insert = event.download.inserted_session,multi\n"
|
|
"system.method.insert = event.download.erased,multi\n"
|
|
"system.method.insert = event.download.opened,multi\n"
|
|
"system.method.insert = event.download.closed,multi\n"
|
|
"system.method.insert = event.download.resumed,multi\n"
|
|
"system.method.insert = event.download.paused,multi\n"
|
|
|
|
"system.method.insert = event.download.finished,multi\n"
|
|
"system.method.insert = event.download.hash_done,multi\n"
|
|
"system.method.insert = event.download.hash_removed,multi\n"
|
|
"system.method.insert = event.download.hash_queued,multi\n"
|
|
|
|
"system.method.set_key = event.download.inserted, 1_connect_logs, d.initialize_logs=\n"
|
|
"system.method.set_key = event.download.inserted_new, 1_prepare, \"branch=d.get_state=,view.set_visible=started,view.set_visible=stopped ;d.save_session=\"\n"
|
|
"system.method.set_key = event.download.inserted_session, 1_prepare, \"branch=d.get_state=,view.set_visible=started,view.set_visible=stopped\"\n"
|
|
|
|
"system.method.set_key = event.download.erased, 0_download_list, ui.unfocus_download=\n"
|
|
"system.method.set_key = event.download.erased, 9_delete_tied, d.delete_tied=\n"
|
|
|
|
// Allow setting 'group.view' as constant, so that we can't
|
|
// modify the value. And look into the possibility of making
|
|
// 'const' use non-heap memory, as we know they can't be
|
|
// erased.
|
|
|
|
// TODO: Remember to ensure it doesn't get restarted by watch
|
|
// dir, etc. Set ignore commands, or something.
|
|
|
|
"group.insert = seeding,seeding\n"
|
|
|
|
"system.method.insert = ratio.enable, simple|static|const,group.seeding.ratio.enable=\n"
|
|
"system.method.insert = ratio.disable,simple|static|const,group.seeding.ratio.disable=\n"
|
|
"system.method.insert = ratio.command,simple|static|const,group.seeding.ratio.command=\n"
|
|
"system.method.insert = ratio.min, simple|static|const,group.seeding.ratio.min=\n"
|
|
"system.method.insert = ratio.max, simple|static|const,group.seeding.ratio.max=\n"
|
|
"system.method.insert = ratio.upload, simple|static|const,group.seeding.ratio.upload=\n"
|
|
"system.method.insert = ratio.min.set, simple|static|const,group.seeding.ratio.min.set=$argument.0=\n"
|
|
"system.method.insert = ratio.max.set, simple|static|const,group.seeding.ratio.max.set=$argument.0=\n"
|
|
"system.method.insert = ratio.upload.set,simple|static|const,group.seeding.ratio.upload.set=$argument.0=\n"
|
|
|
|
"set_name = \"$cat=$system.hostname=,:,$system.pid=\"\n"
|
|
|
|
// Currently not doing any sorting on main.
|
|
"view_add = main\n"
|
|
"view_add = default\n"
|
|
|
|
"view_add = name\n"
|
|
"view_sort_new = name,less=d.get_name=\n"
|
|
"view_sort_current = name,less=d.get_name=\n"
|
|
|
|
"view_add = active\n"
|
|
"view_filter = active,false=\n"
|
|
|
|
"view_add = started\n"
|
|
"view_filter = started,false=\n"
|
|
"view.event_added = started,\"view.set_not_visible=stopped ;d.set_state=1 ;scheduler.simple.added=\"\n"
|
|
"view.event_removed = started,\"view.set_visible=stopped ;scheduler.simple.removed=\"\n"
|
|
|
|
"view_add = stopped\n"
|
|
"view_filter = stopped,false=\n"
|
|
"view.event_added = stopped,\"view.set_not_visible=started ;d.set_state=0\"\n"
|
|
"view.event_removed = stopped,view.set_visible=started\n"
|
|
|
|
"view_add = complete\n"
|
|
"view_filter = complete,d.get_complete=\n"
|
|
"view_filter_on = complete,event.download.hash_done,event.download.finished\n"
|
|
"view_sort_new = complete,less=d.get_state_changed=\n"
|
|
"view_sort_current = complete,less=d.get_state_changed=\n"
|
|
|
|
"view_add = incomplete\n"
|
|
"view_filter = incomplete,not=$d.get_complete=\n"
|
|
"view_filter_on = incomplete,event.download.hash_done,event.download.finished\n"
|
|
"view_sort_new = incomplete,less=d.get_state_changed=\n"
|
|
"view_sort_current = incomplete,less=d.get_state_changed=\n"
|
|
|
|
// The hashing view does not include stopped torrents.
|
|
"view_add = hashing\n"
|
|
"view_filter = hashing,d.get_hashing=\n"
|
|
"view_filter_on = hashing,event.download.hash_queued,event.download.hash_removed,event.download.hash_done\n"
|
|
// "view_sort_new = hashing,less=d.get_state_changed=\n"
|
|
// "view_sort_current = hashing,less=d.get_state_changed=\n"
|
|
|
|
"view_add = seeding\n"
|
|
"view_filter = seeding,\"and=d.get_state=,d.get_complete=\"\n"
|
|
"view_filter_on = seeding,event.download.resumed,event.download.paused\n"
|
|
"view_sort_new = seeding,less=d.get_state_changed=\n"
|
|
"view_sort_current = seeding,less=d.get_state_changed=\n"
|
|
|
|
"schedule = view_main,10,10,\"view_sort=main,20\"\n"
|
|
"schedule = view_name,10,10,\"view_sort=name,20\"\n"
|
|
|
|
"schedule = session_save,1800,1800,session_save=\n"
|
|
"schedule = low_diskspace,5,60,close_low_diskspace=500M\n"
|
|
"schedule = prune_file_status,3600,86400,system.file_status_cache.prune=\n"
|
|
|
|
"encryption=allow_incoming,prefer_plaintext,enable_retry\n"
|
|
);
|
|
|
|
if (OptionParser::has_flag('n', argc, argv))
|
|
control->core()->push_log("Ignoring ~/.rtorrent.rc.");
|
|
else
|
|
rpc::parse_command_single(rpc::make_target(), "try_import = ~/.rtorrent.rc");
|
|
|
|
int firstArg = parse_options(control, argc, argv);
|
|
|
|
control->initialize();
|
|
|
|
// Load session torrents and perform scheduled tasks to ensure
|
|
// session torrents are loaded before arg torrents.
|
|
control->dht_manager()->load_dht_cache();
|
|
load_session_torrents(control);
|
|
rak::priority_queue_perform(&taskScheduler, cachedTime);
|
|
|
|
load_arg_torrents(control, argv + firstArg, argv + argc);
|
|
|
|
// Make sure we update the display before any scheduled tasks can
|
|
// run, so that loading of torrents doesn't look like it hangs on
|
|
// startup.
|
|
control->display()->adjust_layout();
|
|
control->display()->receive_update();
|
|
|
|
while (!control->is_shutdown_completed()) {
|
|
if (control->is_shutdown_received())
|
|
control->handle_shutdown();
|
|
|
|
control->inc_tick();
|
|
|
|
cachedTime = rak::timer::current();
|
|
rak::priority_queue_perform(&taskScheduler, cachedTime);
|
|
|
|
// Do shutdown check before poll, not after.
|
|
control->core()->get_poll_manager()->poll(client_next_timeout(control));
|
|
}
|
|
|
|
control->core()->download_list()->session_save();
|
|
control->cleanup();
|
|
|
|
} catch (std::exception& e) {
|
|
control->cleanup_exception();
|
|
|
|
std::cout << "rtorrent: " << e.what() << std::endl;
|
|
return -1;
|
|
}
|
|
|
|
delete control;
|
|
|
|
return 0;
|
|
}
|
|
|
|
void
|
|
do_panic(int signum) {
|
|
// Use the default signal handler in the future to avoid infinit
|
|
// loops.
|
|
SignalHandler::set_default(signum);
|
|
display::Canvas::cleanup();
|
|
|
|
std::cout << "Caught " << SignalHandler::as_string(signum) << ", dumping stack:" << std::endl;
|
|
|
|
#ifdef USE_EXECINFO
|
|
void* stackPtrs[20];
|
|
|
|
// Print the stack and exit.
|
|
int stackSize = backtrace(stackPtrs, 20);
|
|
char** stackStrings = backtrace_symbols(stackPtrs, stackSize);
|
|
|
|
for (int i = 0; i < stackSize; ++i)
|
|
std::cout << i << ' ' << stackStrings[i] << std::endl;
|
|
|
|
#else
|
|
std::cout << "Stack dump not enabled." << std::endl;
|
|
#endif
|
|
|
|
if (signum == SIGBUS)
|
|
std::cout << "A bus error probably means you ran out of diskspace." << std::endl;
|
|
|
|
std::abort();
|
|
}
|
|
|
|
void
|
|
print_help() {
|
|
std::cout << "Rakshasa's BitTorrent client version " VERSION "." << std::endl;
|
|
std::cout << std::endl;
|
|
std::cout << "All value pairs (f.ex rate and queue size) will be in the UP/DOWN" << std::endl;
|
|
std::cout << "order. Use the up/down/left/right arrow keys to move between screens." << std::endl;
|
|
std::cout << std::endl;
|
|
std::cout << "Usage: rtorrent [OPTIONS]... [FILE]... [URL]..." << std::endl;
|
|
std::cout << " -h Display this very helpful text" << std::endl;
|
|
std::cout << " -n Don't try to load ~/.rtorrent.rc on startup" << std::endl;
|
|
std::cout << " -b <a.b.c.d> Bind the listening socket to this IP" << std::endl;
|
|
std::cout << " -i <a.b.c.d> Change the IP that is sent to the tracker" << std::endl;
|
|
std::cout << " -p <int>-<int> Set port range for incoming connections" << std::endl;
|
|
std::cout << " -d <directory> Save torrents to this directory by default" << std::endl;
|
|
std::cout << " -s <directory> Set the session directory" << std::endl;
|
|
std::cout << " -o key=opt,... Set options, see 'rtorrent.rc' file" << std::endl;
|
|
std::cout << std::endl;
|
|
std::cout << "Main view keys:" << std::endl;
|
|
std::cout << " backspace Add a torrent url or path" << std::endl;
|
|
std::cout << " ^s Start torrent" << std::endl;
|
|
std::cout << " ^d Stop torrent or delete a stopped torrent" << std::endl;
|
|
std::cout << " ^r Manually initiate hash checking" << std::endl;
|
|
std::cout << " ^q Initiate shutdown or skip shutdown process" << std::endl;
|
|
std::cout << " a,s,d,z,x,c Adjust upload throttle" << std::endl;
|
|
std::cout << " A,S,D,Z,X,C Adjust download throttle" << std::endl;
|
|
std::cout << " I Toggle whether torrent ignores ratio settings" << std::endl;
|
|
std::cout << " right View torrent" << std::endl;
|
|
std::cout << std::endl;
|
|
std::cout << "Download view keys:" << std::endl;
|
|
std::cout << " spacebar Depends on the current view" << std::endl;
|
|
std::cout << " 1,2 Adjust max uploads" << std::endl;
|
|
std::cout << " 3,4,5,6 Adjust min/max connected peers" << std::endl;
|
|
std::cout << " t/T Query tracker for more peers / Force query" << std::endl;
|
|
std::cout << " * Snub peer" << std::endl;
|
|
std::cout << " right View files" << std::endl;
|
|
std::cout << " p View peer information" << std::endl;
|
|
std::cout << " o View trackers" << std::endl;
|
|
std::cout << std::endl;
|
|
|
|
std::cout << "Report bugs to <jaris@ifi.uio.no>." << std::endl;
|
|
|
|
exit(0);
|
|
}
|