2010-05-21 01:07:21 +02:00
|
|
|
# This test runs a Bittorrent tracker on one machine, and verifies
|
|
|
|
# that two client machines can download the torrent using
|
|
|
|
# `transmission'. The first client (behind a NAT router) downloads
|
|
|
|
# from the initial seeder running on the tracker. Then we kill the
|
|
|
|
# initial seeder. The second client downloads from the first client,
|
|
|
|
# which only works if the first client successfully uses the UPnP-IGD
|
|
|
|
# protocol to poke a hole in the NAT.
|
|
|
|
|
2019-09-10 15:59:50 +02:00
|
|
|
import ./make-test-python.nix ({ pkgs, ... }:
|
2010-05-20 16:56:04 +02:00
|
|
|
|
|
|
|
let
|
|
|
|
|
|
|
|
# Some random file to serve.
|
2017-01-26 02:44:05 +01:00
|
|
|
file = pkgs.hello.src;
|
2010-05-21 01:07:21 +02:00
|
|
|
|
2018-09-15 15:15:32 +02:00
|
|
|
internalRouterAddress = "192.168.3.1";
|
|
|
|
internalClient1Address = "192.168.3.2";
|
|
|
|
externalRouterAddress = "80.100.100.1";
|
|
|
|
externalClient2Address = "80.100.100.2";
|
|
|
|
externalTrackerAddress = "80.100.100.3";
|
2010-05-20 16:56:04 +02:00
|
|
|
in
|
|
|
|
|
|
|
|
{
|
2014-06-28 16:04:49 +02:00
|
|
|
name = "bittorrent";
|
2015-07-12 12:09:40 +02:00
|
|
|
meta = with pkgs.stdenv.lib.maintainers; {
|
2019-02-22 16:14:13 +01:00
|
|
|
maintainers = [ domenkozar eelco rob bobvanderlinden ];
|
2015-07-12 12:09:40 +02:00
|
|
|
};
|
2010-05-20 16:56:04 +02:00
|
|
|
|
|
|
|
nodes =
|
2011-09-14 20:20:50 +02:00
|
|
|
{ tracker =
|
2018-07-20 22:56:59 +02:00
|
|
|
{ pkgs, ... }:
|
2018-09-15 15:15:32 +02:00
|
|
|
{ environment.systemPackages = [ pkgs.transmission ];
|
|
|
|
|
|
|
|
virtualisation.vlans = [ 1 ];
|
|
|
|
networking.interfaces.eth1.ipv4.addresses = [
|
|
|
|
{ address = externalTrackerAddress; prefixLength = 24; }
|
|
|
|
];
|
2010-05-20 16:56:04 +02:00
|
|
|
|
|
|
|
# We need Apache on the tracker to serve the torrents.
|
|
|
|
services.httpd.enable = true;
|
|
|
|
services.httpd.adminAddr = "foo@example.org";
|
|
|
|
services.httpd.documentRoot = "/tmp";
|
2014-04-11 17:15:56 +02:00
|
|
|
|
2018-09-15 15:15:32 +02:00
|
|
|
networking.firewall.enable = false;
|
|
|
|
|
|
|
|
services.opentracker.enable = true;
|
|
|
|
|
|
|
|
services.transmission.enable = true;
|
|
|
|
services.transmission.settings.dht-enabled = false;
|
|
|
|
services.transmission.settings.port-forwaring-enabled = false;
|
2010-05-20 16:56:04 +02:00
|
|
|
};
|
|
|
|
|
2011-09-14 20:20:50 +02:00
|
|
|
router =
|
2018-09-15 15:15:32 +02:00
|
|
|
{ pkgs, nodes, ... }:
|
|
|
|
{ virtualisation.vlans = [ 1 2 ];
|
2011-03-21 14:14:50 +01:00
|
|
|
networking.nat.enable = true;
|
2014-04-15 14:35:34 +02:00
|
|
|
networking.nat.internalInterfaces = [ "eth2" ];
|
2011-03-21 14:14:50 +01:00
|
|
|
networking.nat.externalInterface = "eth1";
|
2018-09-15 15:15:32 +02:00
|
|
|
networking.firewall.enable = true;
|
|
|
|
networking.firewall.trustedInterfaces = [ "eth2" ];
|
|
|
|
networking.interfaces.eth0.ipv4.addresses = [];
|
|
|
|
networking.interfaces.eth1.ipv4.addresses = [
|
|
|
|
{ address = externalRouterAddress; prefixLength = 24; }
|
|
|
|
];
|
|
|
|
networking.interfaces.eth2.ipv4.addresses = [
|
|
|
|
{ address = internalRouterAddress; prefixLength = 24; }
|
|
|
|
];
|
|
|
|
services.miniupnpd = {
|
|
|
|
enable = true;
|
|
|
|
externalInterface = "eth1";
|
|
|
|
internalIPs = [ "eth2" ];
|
|
|
|
appendConfig = ''
|
|
|
|
ext_ip=${externalRouterAddress}
|
|
|
|
'';
|
|
|
|
};
|
2010-05-21 01:07:21 +02:00
|
|
|
};
|
|
|
|
|
2011-09-14 20:20:50 +02:00
|
|
|
client1 =
|
2018-07-20 22:56:59 +02:00
|
|
|
{ pkgs, nodes, ... }:
|
2018-09-15 15:15:32 +02:00
|
|
|
{ environment.systemPackages = [ pkgs.transmission pkgs.miniupnpc ];
|
2010-05-21 01:07:21 +02:00
|
|
|
virtualisation.vlans = [ 2 ];
|
2018-09-15 15:15:32 +02:00
|
|
|
networking.interfaces.eth0.ipv4.addresses = [];
|
|
|
|
networking.interfaces.eth1.ipv4.addresses = [
|
|
|
|
{ address = internalClient1Address; prefixLength = 24; }
|
|
|
|
];
|
|
|
|
networking.defaultGateway = internalRouterAddress;
|
2014-04-11 17:15:56 +02:00
|
|
|
networking.firewall.enable = false;
|
2018-09-15 15:15:32 +02:00
|
|
|
services.transmission.enable = true;
|
|
|
|
services.transmission.settings.dht-enabled = false;
|
|
|
|
services.transmission.settings.message-level = 3;
|
2010-05-21 01:07:21 +02:00
|
|
|
};
|
|
|
|
|
2011-09-14 20:20:50 +02:00
|
|
|
client2 =
|
2018-07-20 22:56:59 +02:00
|
|
|
{ pkgs, ... }:
|
2010-05-20 16:56:04 +02:00
|
|
|
{ environment.systemPackages = [ pkgs.transmission ];
|
2018-09-15 15:15:32 +02:00
|
|
|
virtualisation.vlans = [ 1 ];
|
|
|
|
networking.interfaces.eth0.ipv4.addresses = [];
|
|
|
|
networking.interfaces.eth1.ipv4.addresses = [
|
|
|
|
{ address = externalClient2Address; prefixLength = 24; }
|
|
|
|
];
|
2014-04-11 17:15:56 +02:00
|
|
|
networking.firewall.enable = false;
|
2018-09-15 15:15:32 +02:00
|
|
|
services.transmission.enable = true;
|
|
|
|
services.transmission.settings.dht-enabled = false;
|
|
|
|
services.transmission.settings.port-forwaring-enabled = false;
|
2010-05-20 16:56:04 +02:00
|
|
|
};
|
|
|
|
};
|
|
|
|
|
|
|
|
testScript =
|
2010-05-21 16:31:05 +02:00
|
|
|
{ nodes, ... }:
|
2010-05-20 16:56:04 +02:00
|
|
|
''
|
2019-09-10 15:59:50 +02:00
|
|
|
start_all()
|
2010-05-20 16:56:04 +02:00
|
|
|
|
2018-09-15 15:15:32 +02:00
|
|
|
# Wait for network and miniupnpd.
|
2019-09-10 15:59:50 +02:00
|
|
|
router.wait_for_unit("network-online.target")
|
|
|
|
router.wait_for_unit("miniupnpd")
|
2010-05-21 01:07:21 +02:00
|
|
|
|
2010-05-20 16:56:04 +02:00
|
|
|
# Create the torrent.
|
2019-09-10 15:59:50 +02:00
|
|
|
tracker.succeed("mkdir /tmp/data")
|
|
|
|
tracker.succeed(
|
|
|
|
"cp ${file} /tmp/data/test.tar.bz2"
|
|
|
|
)
|
|
|
|
tracker.succeed(
|
|
|
|
"transmission-create /tmp/data/test.tar.bz2 --private --tracker http://${externalTrackerAddress}:6969/announce --outfile /tmp/test.torrent"
|
|
|
|
)
|
|
|
|
tracker.succeed("chmod 644 /tmp/test.torrent")
|
2010-05-20 16:56:04 +02:00
|
|
|
|
|
|
|
# Start the tracker. !!! use a less crappy tracker
|
2019-09-10 15:59:50 +02:00
|
|
|
tracker.wait_for_unit("network-online.target")
|
|
|
|
tracker.wait_for_unit("opentracker.service")
|
|
|
|
tracker.wait_for_open_port(6969)
|
2010-05-20 16:56:04 +02:00
|
|
|
|
|
|
|
# Start the initial seeder.
|
2019-09-10 15:59:50 +02:00
|
|
|
tracker.succeed(
|
|
|
|
"transmission-remote --add /tmp/test.torrent --no-portmap --no-dht --download-dir /tmp/data"
|
|
|
|
)
|
2010-05-20 16:56:04 +02:00
|
|
|
|
2010-05-21 01:07:21 +02:00
|
|
|
# Now we should be able to download from the client behind the NAT.
|
2019-09-10 15:59:50 +02:00
|
|
|
tracker.wait_for_unit("httpd")
|
|
|
|
client1.wait_for_unit("network-online.target")
|
|
|
|
client1.succeed(
|
|
|
|
"transmission-remote --add http://${externalTrackerAddress}/test.torrent --download-dir /tmp >&2 &"
|
|
|
|
)
|
|
|
|
client1.wait_for_file("/tmp/test.tar.bz2")
|
|
|
|
client1.succeed(
|
|
|
|
"cmp /tmp/test.tar.bz2 ${file}"
|
|
|
|
)
|
2010-05-21 01:07:21 +02:00
|
|
|
|
|
|
|
# Bring down the initial seeder.
|
2019-09-10 15:59:50 +02:00
|
|
|
# tracker.stop_job("transmission")
|
2010-05-21 01:07:21 +02:00
|
|
|
|
|
|
|
# Now download from the second client. This can only succeed if
|
|
|
|
# the first client created a NAT hole in the router.
|
2019-09-10 15:59:50 +02:00
|
|
|
client2.wait_for_unit("network-online.target")
|
|
|
|
client2.succeed(
|
|
|
|
"transmission-remote --add http://${externalTrackerAddress}/test.torrent --no-portmap --no-dht --download-dir /tmp >&2 &"
|
|
|
|
)
|
|
|
|
client2.wait_for_file("/tmp/test.tar.bz2")
|
|
|
|
client2.succeed(
|
|
|
|
"cmp /tmp/test.tar.bz2 ${file}"
|
|
|
|
)
|
2010-05-20 16:56:04 +02:00
|
|
|
'';
|
2011-09-14 20:20:50 +02:00
|
|
|
|
2014-04-14 14:02:44 +02:00
|
|
|
})
|