1 files changed, 6 insertions, 14 deletions
diff --git a/manifests/redirect.pp b/manifests/redirect.pp
index dee9a98..de50f86 100644
--- a/manifests/redirect.pp
+++ b/manifests/redirect.pp
@@ -1,16 +1,8 @@
-class firewall::redirect::ssh($destinationport) {
-  # When the box is in an internal network and we want to provide
-  # and external access through a shared real IP, we have to
-  # redirect requests coming from another port to port 22.
-  $ip = lookup('firewall::external_ip', undef, undef, $::ipaddress)
-
-  shorewall::rule { "ssh-redirect-1":
-    action          => 'DNAT',
-    source          => 'net',
-    destination     => "fw:$ip:22",
-    proto           => 'tcp',
-    destinationport => $destinationport,
-    ratelimit       => '-',
-    order           => $destinationport,
+class firewall::redirect::ssh(
+  $implementation = lookup('firewall::implementation', undef, undef, 'shorewall'),
+  $destinationport,
+) {
+  class { "firewall::implementations::${implementation}::redirect::ssh":
+    destinationport = $destinationport,
   }
 }