ctorrent/main.cpp

/*
 * Copyright (c) 2014, 2015 Ahmed Samy  <f.fallen45@gmail.com>
 *
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this software and associated documentation files (the "Software"), to deal
 * in the Software without restriction, including without limitation the rights
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 * copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in
 * all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
 * THE SOFTWARE.
 */
#include <ctorrent/torrent.h>
#include <net/connection.h>
#include <util/auxiliar.h>

#include <thread>
#include <functional>
#include <chrono>
#include <iostream>

#include <boost/program_options.hpp>

#ifndef _WIN32
#include <ncurses.h>
#endif

static void print_help(const char *p)
{
	std::clog << "Usage: " << p << " <options...> <torrent...>" << std::endl;
	std::clog << "Mandatory arguments to long options are mandatory for short options too." << std::endl;
	std::clog << "\t\t--version (-v), --help (-h) 	print version and help respectively then exit" << std::endl;
	std::clog << "\t\t--nodownload (-n)		Just check pieces completed, torrent size, etc." << std::endl;
	std::clog << "\t\t--piecesize (-s)		Specify piece size in KB, this will be rounded to the nearest power of two.  Default is 16 KB" << std::endl;
	std::clog << "\t\t--port (-p)			Not yet fully implemented." << std::endl;
	std::clog << "\t\t--dldir (-d)			Specify downloads directory." << std::endl;
	std::clog << "\t\t--noseed (-e)			Do not seed after download has finished." << std::endl;
	std::clog << "\t\t--torrents (-t) 		Specify torrent file(s)." << std::endl;
	std::clog << "Example: " << p << " --nodownload --torrents a.torrent b.torrent c.torrent" << std::endl;
}

#ifdef _WIN32
enum {
	COL_BLACK = 0,
	COL_GREEN = 10,
	COL_YELLOW = 14,
};

static void set_color(int col)
{
	CONSOLE_SCREEN_BUFFER_INFO i;
	GetConsoleScreenBufferInfo(GetStdHandle(STD_OUTPUT_HANDLE), &i);

	int back = (i.wAttributes / 16) % 16;
	if (col % 16 == back % 16)
		++col;
	col %= 16;
	back %= 16;

	uint16_t attrib = ((unsigned)back << 4) | (unsigned)col;
	SetConsoleTextAttribute(GetStdHandle(STD_OUTPUT_HANDLE), attrib);
}

#define printc(c, fmt, args...) do {	\
	set_color((c));	\
	printf(fmt, ##args);	\
	set_color(COL_BLACK);	\
} while (0)
#else
enum {
	COL_GREEN = 1,
	COL_YELLOW = 2,
};

#define printc(c, fmt, args...)	do {	\
	attron(COLOR_PAIR(c));	\
	printw(fmt, ##args);	\
} while (0)
#endif

static void print_stats(Torrent *t)
{
	const TorrentMeta *meta = t->meta();
	const TorrentFileManager *fm = t->fileManager();

	printc(COL_GREEN, "%s: ", meta->name().c_str());
	printc(COL_YELLOW, "%.2f Mbps (%zd / %zd downloaded %zd hash miss, %zd wasted - %.2f seconds left) ",
				t->downloadSpeed(), t->downloadedBytes(), meta->totalSize(),
				t->hashMisses(), t->wastedBytes(), t->eta());
	printc(COL_YELLOW, "[ %d/%d pieces %d peers active]\n",
				fm->completedPieces(), fm->totalPieces(), t->activePeers());
}

static void print_all_stats(Torrent *torrents, size_t total)
{
#ifdef _WIN32
	COORD coord;
	CONSOLE_SCREEN_BUFFER_INFO info;
	GetConsoleScreenBufferInfo(GetStdHandle(STD_OUTPUT_HANDLE), &info);
	coord.X = info.dwCursorPosition.X;
	coord.Y = info.dwCursorPosition.Y;
#else
	move(0, 0);
#endif
	for (size_t i = 0; i < total; ++i)
		print_stats(&torrents[i]);
#ifdef _WIN32
	SetConsoleCursorPosition(GetStdHandle(STD_OUTPUT_HANDLE), coord);
#else
	refresh();
#endif
}

int main(int argc, char *argv[])
{
	bool noseed = false;
	int startport = 6881;
	std::string dldir = "Torrents";
	std::vector<std::string> files;

	namespace po = boost::program_options;
	po::options_description opts;
	opts.add_options()
		("version,v", "print version string")
		("help,h", "print this help message")
		("port,p", po::value(&startport), "specify start port; not currently used")
		("nodownload,n", "do not download anything, just print info about torrents")
		("piecesize,s", po::value(&maxRequestSize), "specify piece block size")
		("dldir,d", po::value(&dldir), "specify downloads directory")
		("noseed,e", po::value(&noseed), "do not seed after download has finished.")
		("torrents,t", po::value<std::vector<std::string>>(&files)->required()->multitoken(), "specify torrent file(s)");

	if (argc == 1) {
		print_help(argv[0]);
		return 1;
	}

	po::variables_map vm;
	try {
		po::store(po::parse_command_line(argc, argv, opts), vm);
		po::notify(vm);
	} catch (const std::exception &e) {
		std::cerr << argv[0] << ": error parsing command line arguments: " << e.what() << std::endl;
		print_help(argv[0]);
		return 1;
	}

	if (vm.count("help")) {
		print_help(argv[0]);
		return 0;
	}

	if (vm.count("version")) {
		std::clog << "CTorrent version 1.0" << std::endl;
		return 0;
	}

	bool nodownload = false;
	if (vm.count("nodownload"))
		nodownload = true;
	else if (vm.count("piecesize"))
		maxRequestSize = 1 << (32 - __builtin_clz(maxRequestSize - 1));

#ifndef _WIN32
	initscr();
	assume_default_colors(COLOR_WHITE, COLOR_BLACK);
	init_pair(1, COLOR_GREEN, COLOR_BLACK);
	init_pair(2, COLOR_YELLOW, COLOR_BLACK);
	attrset(A_BOLD);	// boldy
	curs_set(0);		// don't show cursor
#endif

	size_t total = files.size();
	size_t completed = 0;
	size_t errors = 0;
	size_t eseed = 0;
	size_t started = 0;

	Torrent torrents[total];
	std::thread threads[total];
	for (size_t i = 0; i < files.size(); ++i) {
		std::string file = files[i];
		Torrent *t = &torrents[i];

		if (!t->open(file, dldir)) {
			std::cerr << file << ": corrupted torrent file" << std::endl;
			++errors;
			continue;
		}

		const TorrentMeta *meta = t->meta();
		if (nodownload) {
			std::clog << meta->name() << ": Total size: " << bytesToHumanReadable(meta->totalSize(), true) << std::endl;
			continue;
		}

		++started;
		threads[i] = std::thread([t, meta, &startport, &completed, &errors, &eseed, noseed]() {
			uint16_t tport = startport++;
			Torrent::DownloadState error = t->download(tport);
			switch (error) {
			case Torrent::DownloadState::Completed:
				++completed;
				break;
			case Torrent::DownloadState::AlreadyDownloaded:
				++completed;
				break;
			case Torrent::DownloadState::NetworkError:
				++errors;
				break;
			case Torrent::DownloadState::TrackerQueryFailure:
				++errors;
				break;
			}

			if (!noseed
				&& (error == Torrent::DownloadState::Completed || error == Torrent::DownloadState::AlreadyDownloaded)
				&& !t->seed(tport))
				++eseed;
		});
		threads[i].detach();
	}

	if (!nodownload && started > 0) {
		while (completed != total - errors) {
			if (errors == total)
				break;

			Connection::poll();
			print_all_stats(torrents, total);
			std::this_thread::sleep_for(std::chrono::milliseconds(5));
		}
	}

	if (!noseed) {
		while (eseed != total) {
			Connection::poll();
			print_all_stats(torrents, total);
			std::this_thread::sleep_for(std::chrono::milliseconds(10));
		}
	}

#ifndef _WIN32
	endwin();
#endif
	return 0;
}