Add support for nftables

Author: smortex

lib/puffy.rb

@@ -7,6 +7,7 @@
 require 'puffy/formatters/iptables'
 require 'puffy/formatters/iptables4'
 require 'puffy/formatters/iptables6'
+require 'puffy/formatters/nftables'
 require 'puffy/formatters/pf'
 require 'puffy/puppet'
 require 'puffy/resolver'

lib/puffy/formatters/nftables.rb

@@ -0,0 +1,123 @@
+# frozen_string_literal: true
+
+module Puffy
+  module Formatters
+    module Nftables # :nodoc:
+      # Returns the rule action
+      #
+      # @return [String]
+      def self.nftables_action(rule_or_action, ret: false)
+        case rule_or_action
+        when :pass      then 'accept'
+        when :block     then ret ? 'return' : 'drop'
+        when Puffy::Rule then nftables_action(rule_or_action.action, ret: rule_or_action.return)
+        end
+      end
+
+      # Nftables implementation of a Puffy Ruleset formatter.
+      class Ruleset < Puffy::Formatters::Base::Ruleset
+        alias parent_emit_ruleset emit_ruleset
+
+        # Returns a Nftables String representation of the provided +rules+ Array of Puffy::Rule.
+        def emit_ruleset(rules, policy = :block)
+          <<~FRAGMENT
+            #!/usr/sbin/nft -f
+
+            flush ruleset
+
+            #{inet_filter(rules, policy)}
+          FRAGMENT
+        end
+
+        def inet_filter(rules, policy)
+          <<~FRAGMENT.chomp
+            table inet filter {
+            #{chain_input(rules.select(&:in?), policy)}
+            #{chain_forward(rules.select(&:fwd?))}
+            #{chain_output(rules.select(&:out?), policy)}
+            }
+          FRAGMENT
+        end
+
+        def chain_input(rules, policy)
+          <<~FRAGMENT.chomp
+            \tchain input {
+            \t\ttype filter hook input priority 0;
+            \t\tpolicy #{Nftables.nftables_action(policy)};
+            \t\tct state {established, related} accept;
+            #{parent_emit_ruleset(rules)}
+            \t}
+          FRAGMENT
+        end
+
+        def chain_forward(rules)
+          <<~FRAGMENT.chomp
+            \tchain forward {
+            \t\ttype filter hook forward priority 0;
+            #{parent_emit_ruleset(rules)}
+            \t}
+          FRAGMENT
+        end
+
+        def chain_output(rules, policy)
+          <<~FRAGMENT.chomp
+            \tchain output {
+            \t\ttype filter hook output priority 0;
+            \t\tpolicy #{Nftables.nftables_action(policy)};
+            \t\tct state {established, related} accept;
+            #{parent_emit_ruleset(rules)}
+            \t}
+          FRAGMENT
+        end
+
+        def filename_fragment
+          ['nftables', 'nftables.conf']
+        end
+      end
+
+      # Nftables implementation of a Puffy Rule formatter.
+      class Rule < Puffy::Formatters::Base::Rule
+        # Returns a Nftables String representation of the provided +rule+ Puffy::Rule.
+        def emit_rule(rule)
+          parts = []
+          parts << emit_what(rule)
+          "\t\t#{parts.flatten.compact.join(' ')};"
+        end
+
+        private
+
+        def emit_what(rule)
+          parts = []
+          parts << emit_proto(rule)
+          parts += emit_from(rule)
+          parts += emit_to(rule)
+          parts << Nftables.nftables_action(rule)
+          parts
+        end
+
+        def emit_proto(rule)
+          case rule.proto
+          when :icmp
+            %w[ip protocol icmp]
+          else
+            rule.proto
+          end
+        end
+
+        def emit_from(rule)
+          parts = []
+          parts += ['saddr', rule.from_host] if rule.from_host
+          parts += ['sport', rule.from_port] if rule.from_port
+          parts
+        end
+
+        def emit_to(rule)
+          parts = []
+          parts += ['daddr', rule.to_host] if rule.to_host
+          parts += ['dport', rule.to_port] if rule.to_port
+          parts
+        end
+      end
+    end
+  end
+end