cloudamatic/mu

#
# Cookbook Name:: mu-tools
# Recipe:: set_local_fw
#
# Copyright:: Copyright (c) 2016 eGlobalTech, Inc., all rights reserved
#
# Licensed under the BSD-3 license (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License in the root of the project or at
#
#     http://egt-labs.com/mu/LICENSE.html
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

if !node['application_attributes']['skip_recipes'].include?('set_local_fw')
  master_ips = get_mu_master_ips
  case node['platform_family']
  when 'rhel', 'amazon'
    include_recipe 'mu-firewall'

    if elversion >= 7 and node['platform_family'] != "amazon" # Can use firewalld, but not if iptables is already rigged
      package "firewall-config" do
        not_if "/bin/systemctl list-units | grep iptables.service"
      end
      execute "restart FirewallD" do # ...but only if iptables isn't live
        command "/bin/firewall-cmd --reload"
        action :nothing
        not_if "/bin/systemctl list-units | grep iptables.service"
        only_if { ::File.exist?("/bin/firewall-cmd") }
      end
    end

    if elversion <= 6
      firewall_rule "Allow loopback in" do
        raw "-A INPUT -i lo -j ACCEPT"
      end

      firewall_rule "Allow loopback out" do
        raw "-A OUTPUT -o lo -j ACCEPT"
      end
    end

    firewall_rule "Allow eth0 out" do
      raw "-A OUTPUT -o eth0 -j ACCEPT"
    end

    firewall_rule "Allow established connections" do
      raw "-A INPUT -m state --state ESTABLISHED,RELATED -j ACCEPT"
    end

    opento = master_ips.map { |x| "#{x}/32"}

    opento.uniq.each { |src|
      [:tcp, :udp, :icmp].each { |proto|
        firewall_rule "allow all #{src} #{proto} traffic" do
          source src
          protocol proto
        end
      }
    }
  end
end