puppet nftables

# @summary Configure nftables

#

# @example

#   class{'nftables:

#     out_ntp = false,

#     out_dns = true,

#   }

#

# @param out_all

#   Allow all outbound connections. If `true` then all other

#   out parameters `out_ntp`, `out_dns`, ... will be assuemed

#   false.

#

# @param out_ntp

#   Allow outbound to ntp servers.

#

# @param out_http

#   Allow outbound to http servers.

#

# @param out_https

#   Allow outbound to https servers.

#

# @param out_https

#   Allow outbound to https servers.

#

# @param out_icmp

#   Allow outbound ICMPv4/v6 traffic.

#

# @param in_ssh

#   Allow inbound to ssh servers.

#

# @param in_icmp

#   Allow inbound ICMPv4/v6 traffic.

#

# @param nat

#   Add default tables and chains to process NAT traffic.

#

# @param log_prefix

#   String that will be used as prefix when logging packets. It can contain

#   two variables using standard sprintf() string-formatting:

#    * chain: Will be replaced by the name of the chain.

#    * comment: Allows chains to add extra comments.

#

# @param reject_with

#   How to discard packets not matching any rule. If `false`, the

#   fate of the packet will be defined by the chain policy (normally

#   drop), otherwise the packet will be rejected with the REJECT_WITH

#   policy indicated by the value of this parameter.

#

# @param in_out_conntrack

#   Adds INPUT and OUTPUT rules to allow traffic that's part of an

#   established connection and also to drop invalid packets.

#

# @param firewalld_enable

#   Configures how the firewalld systemd service unit is enabled. It might be

#   useful to set this to false if you're externaly removing firewalld from

#   the system completely.

#

class nftables (

  Boolean $in_ssh                = true,

  Boolean $in_icmp               = true,

  Boolean $out_ntp               = true,

  Boolean $out_dns               = true,

  Boolean $out_http              = true,

  Boolean $out_https             = true,

  Boolean $out_icmp              = true,

  Boolean $out_all               = false,

  Boolean $in_out_conntrack      = true,

  Boolean $nat                   = true,

  Hash $rules                    = {},

  String $log_prefix             = '[nftables] %<chain>s %<comment>s',

  Variant[Boolean[false], Pattern[

    /icmp(v6|x)? type .+|tcp reset/]]

    $reject_with                 = 'icmpx type port-unreachable',

  Variant[Boolean[false], Enum['mask']]

    $firewalld_enable            = 'mask',

) {

  package{'nftables':

    ensure => installed,

  } -> file_line{

    'enable_nftables':

      line   => 'include "/etc/nftables/puppet.nft"',

      path   => '/etc/sysconfig/nftables.conf',

      notify => Service['nftables'],

  } -> file{

    default:

      owner => 'root',

      group => 'root',

      mode  => '0640';

    '/etc/nftables/puppet-preflight':

      ensure  => directory,

      mode    => '0750',

      purge   => true,

      force   => true,

      recurse => true;

    '/etc/nftables/puppet-preflight.nft':

      ensure  => file,

      content => epp('nftables/config/puppet.nft.epp', { 'nat' => $nat });

  } ~> exec{

    'nft validate':

      refreshonly => true,

      command     => '/usr/sbin/nft -I /etc/nftables/puppet-preflight -c -f /etc/nftables/puppet-preflight.nft || ( /usr/bin/echo "#CONFIG BROKEN" >> /etc/nftables/puppet-preflight.nft && /bin/false)';

  } -> file{

    default:

      owner => 'root',

      group => 'root',

      mode  => '0640';

    '/etc/nftables/puppet.nft':

      ensure  => file,

      content => epp('nftables/config/puppet.nft.epp', { 'nat' => $nat });

    '/etc/nftables/puppet':

      ensure  => directory,

      mode    => '0750',

      purge   => true,

      force   => true,

      recurse => true;

  } ~> service{'nftables':

    ensure     => running,

    enable     => true,

    hasrestart => true,

    restart    => '/usr/bin/systemctl reload nftables',

  }

  systemd::dropin_file{'puppet_nft.conf':

    ensure => present,

    unit   => 'nftables.service',

    source => 'puppet:///modules/nftables/systemd/puppet_nft.conf',

    notify => Service['nftables'],

  }

  service{'firewalld':

    ensure => stopped,

    enable => $firewalld_enable,

  }

  include nftables::inet_filter

  if $nat {

    include nftables::ip_nat

  }

  # inject custom rules e.g. from hiera

  $rules.each |$n,$v| {

    nftables::rule{

      $n:

        * => $v

    }

  }

}