1 files changed, 7 insertions, 9 deletions
diff --git a/manifests/virtual/mdns.pp b/manifests/virtual/mdns.pp
index db8ec25..559a00b 100644
--- a/manifests/virtual/mdns.pp
+++ b/manifests/virtual/mdns.pp
@@ -1,11 +1,9 @@
-class firewall::virtual::mdns($destination, $zone = 'fw') {
-  shorewall::rule { 'mdns-0':
-    action          => 'DNAT',
-    source          => 'net',
-    destination     => "$zone:$destination:5353",
-    proto           => 'tcp',
-    destinationport => '5353',
-    ratelimit       => '-',
-    order           => 2700,
+class firewall::virtual::mdns(
+  $implementation = lookup('firewall::implementation', undef, undef, 'shorewall'),
+  $destination, $zone = 'fw'
+) {
+  class { "firewall::implementations::${implementation}::virtual::mdns":
+    destination => $destination,
+    zone        => $zone,
   }
 }