instrumentation/hrktorrent: add support for maximum connections

diff --git a/instrumentation/hrktorrent/core.cpp b/instrumentation/hrktorrent/core.cpp
index 013dd4d..b85cad7 100644 (file)
--- a/instrumentation/hrktorrent/core.cpp
+++ b/instrumentation/hrktorrent/core.cpp
@@ -309,6 +309,9 @@ CCore::Run()
        if(Settings->GetI("maxdown") > 0) {\r
                _session->set_download_rate_limit(Settings->GetI("maxdown")*1024);\r
        }\r
+       if(Settings->GetI("maxconnections") > 0) {\r
+               _session->set_max_connections(Settings->GetI("maxconnections"));\r
+       }\r
        try {\r
                libtorrent::add_torrent_params parms;\r
 \r

diff --git a/instrumentation/hrktorrent/main.cpp b/instrumentation/hrktorrent/main.cpp
index 9546d76..dca85d0 100644 (file)
--- a/instrumentation/hrktorrent/main.cpp
+++ b/instrumentation/hrktorrent/main.cpp
@@ -20,6 +20,8 @@ printusage(const char *argv0)
        std::cout << "\t-d, --maxdown <speed>\t\tdownload speed limit (in kb/s) (default: unlimited)\n";
        std::cout << "\t-u, --maxup <speed>\t\tupload speed limit (in kb/s) (default: unlimited)\n";
 
+       std::cout << "\t-c, --maxconnections <conn>\t\tlimit numer of connections (default: unlimited)\n";
+
        std::cout << "\t-l, --limitlocal\t\tlimit local network (default: off)\n";
        std::cout << "\t-t, --nodht\t\t\tdisable dht (default: on)\n";
        std::cout << "\t-n, --noupnp\t\t\tdisable upnp (default: on)\n";
@@ -55,13 +57,14 @@ parseargs(int argc, char** argv)
                {"maxport", 1, NULL, 'P'},
                {"maxdown", 1, NULL, 'd'},
                {"maxup", 1, NULL, 'u'},
+               {"maxconnections", 1, NULL, 'c'},
                {"forcereannounce", 1, NULL, 'f'},
                {"outputdir", 1, NULL, 'o'},
                {NULL, 0, NULL, 0}
        };
 
        while (1) {
-               opt = getopt_long(argc, argv, "hiltnsvVp:P:d:u:f:o:",
+               opt = getopt_long(argc, argv, "hiltnsvVp:P:d:u:c:f:o:",
                                long_options, &option_index);
                if (opt == -1)
                        break;
@@ -105,6 +108,9 @@ parseargs(int argc, char** argv)
                case 'u':
                        Settings->Set("maxup", atoi(optarg));
                        break;
+               case 'c':
+                       Settings->Set("maxconnections", atoi(optarg));
+                       break;
                case 'f':
                        Settings->Set("forcereannounce", atoi(optarg));
                        break;

diff --git a/instrumentation/hrktorrent/settings.cpp b/instrumentation/hrktorrent/settings.cpp
index 5c3fa7e..8166139 100644 (file)
--- a/instrumentation/hrktorrent/settings.cpp
+++ b/instrumentation/hrktorrent/settings.cpp
@@ -12,6 +12,7 @@ CSettings::CSettings()
 \r
        Set("maxup", 0);\r
        Set("maxdown", 0);\r
+       Set("maxconnections", -1);\r
 \r
        Set("dht", 1);\r
        Set("upnp", 1);\r