51 lines
		
	
	
		
			1.4 KiB
		
	
	
	
		
			Nix
		
	
	
	
	
	
			
		
		
	
	
			51 lines
		
	
	
		
			1.4 KiB
		
	
	
	
		
			Nix
		
	
	
	
	
	
# Test the firewall module.
 | 
						|
 | 
						|
import ./make-test.nix ( { pkgs, ... } : {
 | 
						|
  name = "firewall";
 | 
						|
  meta = with pkgs.stdenv.lib.maintainers; {
 | 
						|
    maintainers = [ eelco chaoflow ];
 | 
						|
  };
 | 
						|
 | 
						|
  nodes =
 | 
						|
    { walled =
 | 
						|
        { config, pkgs, nodes, ... }:
 | 
						|
        { networking.firewall.enable = true;
 | 
						|
          networking.firewall.logRefusedPackets = true;
 | 
						|
          services.httpd.enable = true;
 | 
						|
          services.httpd.adminAddr = "foo@example.org";
 | 
						|
        };
 | 
						|
 | 
						|
      attacker =
 | 
						|
        { config, pkgs, ... }:
 | 
						|
        { services.httpd.enable = true;
 | 
						|
          services.httpd.adminAddr = "foo@example.org";
 | 
						|
          networking.firewall.enable = false;
 | 
						|
        };
 | 
						|
    };
 | 
						|
 | 
						|
  testScript =
 | 
						|
    { nodes, ... }:
 | 
						|
    ''
 | 
						|
      startAll;
 | 
						|
 | 
						|
      $walled->waitForUnit("firewall");
 | 
						|
      $walled->waitForUnit("httpd");
 | 
						|
      $attacker->waitForUnit("network.target");
 | 
						|
 | 
						|
      # Local connections should still work.
 | 
						|
      $walled->succeed("curl -v http://localhost/ >&2");
 | 
						|
 | 
						|
      # Connections to the firewalled machine should fail.
 | 
						|
      $attacker->fail("curl --fail --connect-timeout 2 http://walled/ >&2");
 | 
						|
      $attacker->fail("ping -c 1 walled >&2");
 | 
						|
 | 
						|
      # Outgoing connections/pings should still work.
 | 
						|
      $walled->succeed("curl -v http://attacker/ >&2");
 | 
						|
      $walled->succeed("ping -c 1 attacker >&2");
 | 
						|
 | 
						|
      # If we stop the firewall, then connections should succeed.
 | 
						|
      $walled->stopJob("firewall");
 | 
						|
      $attacker->succeed("curl -v http://walled/ >&2");
 | 
						|
    '';
 | 
						|
})
 |