#!/usr/sbin/nft -f

<%- unless @interfaces.empty? %>
define external-interfaces = { <%= @interfaces.sort.uniq.join(", ") %> }
<%- end %>

define ip-private-addresses = { 0.0.0.0, 10.0.0.0/8, 127.0.0.0/8, 169.254.0.0/16, 172.16.0.0/12, 192.0.2.0/24, 192.168.0.0/16 }
define ip-multicast-addresses = { 224.0.0.0/4 }
define ip6-private-addresses = { 2001:db8::/32, fc00::/7 }
define ip6-multicast-addresses = { ff00::/8 }

table inet chef-filter {
  set ip-osm-addresses {
    type ipv4_addr
<%- unless Array(@hosts["inet"]).empty? %>
    elements = { <%= Array(@hosts["inet"]).sort.join(", ") %> }
<%- end %>
  }

  set ip6-osm-addresses {
    type ipv6_addr
<%- unless Array(@hosts["inet"]).empty? %>
    elements = { <%= Array(@hosts["inet6"]).sort.join(", ") %> }
<%- end %>
  }

  set ip-blocklist {
    type ipv4_addr
    flags interval
  }

  set ip6-blocklist {
    type ipv6_addr
    flags interval
  }

  set ratelimit-icmp-echo-ip {
    type ipv4_addr
    flags dynamic
    timeout 120s
  }

  set ratelimit-icmp-echo-ip6 {
    type ipv6_addr
    flags dynamic
    timeout 120s
  }

<%- node[:networking][:firewall][:sets].each do |set| %>
  set <%= set %> {
<%- if set.end_with?("-ip") %>
    type ipv4_addr
<%- elsif set.end_with?("-ip6") %>
    type ipv6_addr
<%- end %>
    flags dynamic
<%- unless set.start_with?("connlimit-") %>
    timeout 120s
<%- end %>
  }

<%- end %>

<%- node[:networking][:firewall][:helpers].each do |helper| %>
  ct helper <%= helper[:name] %> {
    type "<%= helper[:helper] %>" protocol <%= helper[:protocol] %>
  }

<%- end %>
  chain log-and-drop {
    limit rate 1/second log
    drop
  }

  chain log-and-reject {
    limit rate 1/second log
    reject
  }

  chain incoming {
<%- if node[:networking][:firewall][:allowlist].empty? %>
    ip saddr { $ip-private-addresses, $ip-multicast-addresses } jump log-and-drop
<%- else %>
    ip saddr { $ip-private-addresses, $ip-multicast-addresses } ip saddr != { <%= node[:networking][:firewall][:allowlist].sort.join(", ") %> } jump log-and-drop
<%- end %>
    ip6 saddr { $ip6-private-addresses, $ip6-multicast-addresses } jump log-and-drop

    ip saddr @ip-blocklist jump log-and-drop
    ip6 saddr @ip6-blocklist jump log-and-drop

    icmp type { echo-request } update @ratelimit-icmp-echo-ip { ip saddr limit rate 1/second } accept
    icmp type { echo-request } drop

    icmpv6 type { nd-neighbor-solicit, nd-neighbor-advert, nd-router-solicit, nd-router-advert } accept
    icmpv6 type { echo-request } update @ratelimit-icmp-echo-ip6 { ip6 saddr limit rate 1/second } accept
    icmpv6 type { echo-request } drop

    ct state { established, related } accept

    meta l4proto { icmp, icmpv6 } jump log-and-drop

    tcp flags & (fin|syn|rst|psh|ack|urg) == fin|psh|urg jump log-and-drop
    tcp flags & (fin|syn|rst|psh|ack|urg) == 0x0 jump log-and-drop
    tcp flags & (syn|rst) == syn|rst jump log-and-drop
    tcp flags & (fin|rst) == fin|rst jump log-and-drop
    tcp flags & (fin|syn) == fin|syn jump log-and-drop
    tcp flags & (fin|psh|ack) == fin|psh jump log-and-drop
    tcp sport 0 tcp flags & (fin|syn|rst|ack) == syn jump log-and-drop

<%- node[:networking][:firewall][:incoming].uniq.each do |rule| %>
    <%= rule %>
<%- end %>

    jump log-and-drop
  }

  chain outgoing {
<%- if node[:networking][:firewall][:allowlist].empty? %>
    ip daddr { $ip-private-addresses } jump log-and-drop
<%- else %>
    ip daddr { $ip-private-addresses } ip daddr != { <%= node[:networking][:firewall][:allowlist].sort.join(", ") %> } jump log-and-drop
<%- end %>
    ip6 daddr { $ip6-private-addresses } jump log-and-drop

<%- node[:networking][:firewall][:outgoing].each do |rule| %>
    <%= rule %>
<%- end %>

    accept
  }

  chain input {
    type filter hook input priority filter;

<%- unless @interfaces.empty? %>
    iifname { $external-interfaces } jump incoming
<%- end %>

    accept
  }

  chain forward {
    type filter hook forward priority filter;

<%- unless @interfaces.empty? %>
    iifname { $external-interfaces } jump incoming
    oifname { $external-interfaces } jump outgoing
<%- end %>

    accept
  }

  chain output {
    type filter hook output priority filter;

<%- unless @interfaces.empty? %>
    oifname { $external-interfaces } jump outgoing
<%- end %>

    accept
  }
}
<%- if node[:roles].include?("gateway") %>

table ip chef-nat {
  chain postrouting {
    type nat hook postrouting priority srcnat;

<%- node.interfaces(:role => :external, :family => :inet).each do |external| %>
<%- node.interfaces(:role => :internal, :family => :inet).each do |internal| %>
    oifname { <%= external[:interface] %> } ip saddr { <%= internal[:network] %>/<%= internal[:prefix] %> } snat <%= external[:address] %>
<%- end %>
<%- end %>
  }
}
<%- end %>