netfilter-ruby 4.2

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: c06c6b3915a80142bc0d8a782a2d06a290040759
+  data.tar.gz: bdf9373f496a4a8e7f567e2ae82e831427e7ab7a
+SHA512:
+  metadata.gz: 3539f2ef54357aca38eda3a68af8d8eb2f80bdddc3eb339704cb5f18792532e489cec60b69dd5e5bb8b55e9ee85bfbed05d5945788aa43b60bde7c216c07860e
+  data.tar.gz: f3fafe63c39a387d81379c78d2c010c02dd990c21fbd609075838d2c5fb12d3ce1b1b5d6d7feadc8ec885ff958546daecb463d142408a57215fb6dd97d4d4fde

data/.gitignore

@@ -0,0 +1,17 @@
+*.gem
+*.rbc
+.bundle
+.config
+.yardoc
+Gemfile.lock
+InstalledFiles
+_yardoc
+coverage
+doc/
+lib/bundler/man
+pkg
+rdoc
+spec/reports
+test/tmp
+test/version_tmp
+tmp

data/.rspec

@@ -0,0 +1,2 @@
+--color
+--format progress

data/Gemfile

@@ -0,0 +1,4 @@
+source 'https://rubygems.org'
+
+# Specify your gem's dependencies in netfilter.gemspec
+gemspec

data/LICENSE.txt

@@ -0,0 +1,22 @@
+Copyright (c) 2013 Netskin GmbH
+
+MIT License
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+"Software"), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/README.md

@@ -0,0 +1,63 @@
+# Netfilter::Ruby
+
+Awesome Netfilter (iptables & ebtables) management using ruby.
+
+
+## Installation
+
+Add this line to your application's Gemfile:
+
+    gem 'netfilter-ruby'
+
+And then execute:
+
+    $ bundle
+
+Or install it yourself as:
+
+    $ gem install netfilter-ruby
+
+
+## Usage
+
+    require "netfilter"
+
+    Netfilter.new("example") do |eb, ip4, ip6|
+      eb.table :filter do |t|
+        t.chain "eth1-o" do |c|
+          # allow dhcp requests
+          c.filter :protocol => :ipv4, :ip_proto => :udp, :ip_src => "0.0.0.0", :ip_sport => 68, :ip_dst => "255.255.255.255", :ip_dport => 67, :jump => :return
+
+          # drop everything else
+          c.filter :jump => :drop
+        end
+
+        # capture all outgoing traffic (all traffic to the interface)
+        t.chain :input do |c|
+          c.filter :in_interface => "eth1", :jump => "eth1-o"
+        end
+
+        t.chain :forward do |c|
+          c.filter :in_interface => "eth1", :jump => "eth1-o"
+        end
+      end
+    end
+
+
+## Known bugs
+
+* None
+
+
+## Contributing
+
+1. Fork it
+2. Create your feature branch (`git checkout -b my-new-feature`)
+3. Commit your changes (`git commit -am 'Add some feature'`)
+4. Push to the branch (`git push origin my-new-feature`)
+5. Create new Pull Request
+
+
+## Copyright
+
+Copyright (c) 2012 - 2013 [Netskin GmbH](http://www.netskin.com). Released unter the MIT license.

data/Rakefile

@@ -0,0 +1,5 @@
+require "bundler/gem_tasks"
+require "rspec/core/rake_task"
+
+RSpec::Core::RakeTask.new("spec")
+task :default => :spec

data/lib/netfilter.rb

@@ -0,0 +1,107 @@
+# encoding: utf-8
+
+# see http://livesin.digitalmalaya.net/wp-content/uploads/2011/10/PacketFlow.png?9d7bd4
+
+require "active_support/core_ext/module/delegation"
+require "active_support/hash_with_indifferent_access"
+require "active_support/inflector"
+
+require "netfilter/filter"
+require "netfilter/chain"
+require "netfilter/table"
+require "netfilter/tool"
+require "netfilter/eb_tables"
+require "netfilter/ip_tables"
+require "netfilter/ip6_tables"
+require "netfilter/version"
+
+class Netfilter
+  NATIVE_TABLES = %w(filter nat mangle raw)
+  NATIVE_CHAINS = %w(input output forward prerouting postrouting)
+  NATIVE_TARGETS = %w(accept drop continue return reject dnat snat arpreply ct mark)
+
+  SystemError = Class.new(StandardError)
+
+  attr_accessor :eb_tables, :ip_tables, :ip6_tables
+
+  def self.import(data)
+    data = data.symbolize_keys
+    new.tap do |netfilter|
+      netfilter.eb_tables = data[:eb_tables] ? EbTables.import(data[:eb_tables]) : EbTables.new
+      netfilter.ip_tables = data[:ip_tables] ? IpTables.import(data[:ip_tables]) : IpTables.new
+      netfilter.ip6_tables = data[:ip6_tables] ? Ip6Tables.import(data[:ip6_tables]) : Ip6Tables.new
+    end
+  end
+
+  def initialize(namespace = nil)
+    self.eb_tables = EbTables.new(namespace)
+    self.ip_tables = IpTables.new(namespace)
+    self.ip6_tables = Ip6Tables.new(namespace)
+    yield(eb_tables, ip_tables, ip6_tables) if block_given?
+  end
+
+  def ip_tables
+    return yield(@ip_tables) if block_given?
+    @ip_tables
+  end
+
+  def ip6_tables
+    return yield(@ip6_tables) if block_given?
+    @ip6_tables
+  end
+
+  def eb_tables
+    return yield(@eb_tables) if block_given?
+    @eb_tables
+  end
+
+  def up
+    done = []
+    [:eb_tables, :ip_tables, :ip6_tables].each do |tool|
+      send(tool).up
+      done << tool
+    end
+  rescue => e
+    done.reverse.each{ |tool| send(tool).down }
+    raise e
+  end
+
+  def down
+    done = []
+    [:eb_tables, :ip_tables, :ip6_tables].each do |tool|
+      send(tool).down
+      done << tool
+    end
+  rescue => e
+    done.reverse.each{ |tool| send(tool).up }
+    raise e
+  end
+
+  def pp
+    puts "Eb-Tables"
+    puts "-" * 80
+    eb_tables.pp
+    puts
+    puts "Ip-Tables"
+    puts "-" * 80
+    ip_tables.pp
+    puts
+    puts "Ip6-Tables"
+    puts "-" * 80
+    ip6_tables.pp
+  end
+
+  def namespace=(name)
+    eb_tables.namespace = name
+    ip_tables.namespace = name
+    ip6_tables.namespace = name
+  end
+
+  def export
+    {
+      :eb_tables => eb_tables.export,
+      :ip_tables => ip_tables.export,
+      :ip6_tables => ip6_tables.export,
+    }
+  end
+end

data/lib/netfilter/chain.rb

@@ -0,0 +1,61 @@
+class Netfilter
+  class Chain
+    attr_accessor :table
+    attr_accessor :name, :filters
+
+    delegate :namespace, :to => :table
+
+    def self.import(table, data)
+      data = data.symbolize_keys
+      new(table, data[:name]).tap do |chain|
+        data[:filters].each do |data|
+          chain.filters << Filter.import(chain, data)
+        end
+      end
+    end
+
+    def initialize(table, name)
+      self.table = table
+      self.name = name.to_s
+      self.filters = []
+      self.name.upcase! if native?
+      yield(self) if block_given?
+    end
+
+    def append(definition)
+      filters << Filter.new(self, "append", definition)
+    end
+
+    def insert(definition)
+      filters.unshift Filter.new(self, "append", definition)
+    end
+
+    def filter(definition)
+      append(definition)
+    end
+
+    def native?
+      NATIVE_CHAINS.include?(name.downcase)
+    end
+
+    def name_as_argument
+      (namespace && !native?) ? "#{namespace}-#{name}" : name
+    end
+
+    def commands
+      [].tap do |commands|
+        commands << ["--new-chain #{name_as_argument}"] unless native?
+        filters.each do |filter|
+          commands << ["--#{filter.type} #{name_as_argument}", *filter.args]
+        end
+      end
+    end
+
+    def export
+      {
+        :name => name,
+        :filters => filters.map{ |filter| filter.export },
+      }
+    end
+  end
+end

data/lib/netfilter/eb_tables.rb

@@ -0,0 +1,18 @@
+# encoding: utf-8
+class Netfilter
+  class EbTables < Tool
+    def self.parse
+      {}.tap do |data|
+        string = execute("#{executable} --list").strip << "\n"
+        string.split(/^Bridge table:\s+/).select{ |s| s != "" }.each do |string|
+          table, string = string.match(/(.+?)\n\n(.*)/m).to_a[1..-1]
+          data[table] = {}
+          string.split(/^Bridge chain:\s+/).select{ |s| s != "" }.each do |string|
+            chain, string = string.match(/(.+?),.+?\n(.*)/m).to_a[1..-1]
+            data[table][chain] = string.split("\n").map(&:strip)
+          end
+        end
+      end
+    end
+  end
+end

data/lib/netfilter/filter.rb

@@ -0,0 +1,52 @@
+class Netfilter
+  class Filter
+    attr_accessor :chain
+    attr_accessor :type
+    attr_accessor :definition
+
+    delegate :namespace, :to => :chain
+
+    def self.import(chain, data)
+      data = data.symbolize_keys
+      new(chain, data[:type], data[:definition])
+    end
+
+    def initialize(chain, type, definition)
+      self.chain = chain
+      self.type = type
+      self.definition = definition
+    end
+
+    def to_s
+      args*" "
+    end
+
+    def args
+      [].tap do |args|
+        definition.each_pair do |key, value|
+          key = object_to_argument(key)
+          value = object_to_argument(value)
+          value = "#{namespace}-#{value}" if key == "jump" && namespace && !NATIVE_TARGETS.include?(value.downcase)
+          args << "--#{key} #{value}"
+        end
+      end
+    end
+
+    def export
+      {
+        :type => type,
+        :definition => definition,
+      }
+    end
+
+    private
+
+    def object_to_argument(data)
+      data = data.to_s
+      data = data.gsub("_", "-")
+      data = data.upcase if NATIVE_TARGETS.include?(data.downcase)
+      data = data.downcase if %w(arpreply mark).include?(data.downcase)
+      data
+    end
+  end
+end

data/lib/netfilter/ip6_tables.rb

@@ -0,0 +1,5 @@
+# encoding: utf-8
+class Netfilter
+  class Ip6Tables < IpTables
+  end
+end

data/lib/netfilter/ip_tables.rb

@@ -0,0 +1,23 @@
+# encoding: utf-8
+class Netfilter
+  class IpTables < Tool
+    def self.parse
+      {}.tap do |data|
+        string = execute("#{executable}-save").strip << "\n"
+        string = string.split("\n").reject{ |s| s[0] == "#" }.join("\n")
+        string.split(/^\*/).select{ |s| s != "" }.each do |string|
+          table, string = string.match(/(.+?)\n(.*)/m).to_a[1..-1]
+          data[table] = {}
+
+          string.scan(/^:(.+?)\s+/).each do |match|
+            data[table][match[0]] = []
+          end
+
+          string.scan(/^-A (.+?) (.+?)\n/).each do |match|
+            data[table][match[0]] << match[1]
+          end
+        end
+      end
+    end
+  end
+end

data/lib/netfilter/table.rb

@@ -0,0 +1,62 @@
+class Netfilter
+  class Table
+    attr_accessor :tool
+    attr_accessor :name, :chains
+
+    delegate :namespace, :to => :tool
+
+    def self.import(tool, data)
+      data = data.symbolize_keys
+      new(tool, data[:name]).tap do |table|
+        data[:chains].each do |data|
+          chain = Chain.import(table, data)
+          table.chains[chain.name.to_s.downcase] = chain
+        end
+      end
+    end
+
+    def initialize(tool, name)
+      self.tool = tool
+      self.name = name.to_s
+      self.chains = {}
+      raise ArgumentError, "unsupported table '#{name}'" unless native?
+      yield(self) if block_given?
+    end
+
+    def chain(name, &block)
+      key = name.to_s.downcase
+      (chains[key] || Chain.new(self, name)).tap do |chain|
+        chains[key] = chain
+        block.call(chain) if block
+      end
+    end
+
+    def native?
+      NATIVE_TABLES.include?(name)
+    end
+
+    def commands
+      [].tap do |commands|
+        chains.values.each do |chain|
+          chain.commands.each do |command|
+            commands << command.unshift("--table #{name}")
+          end
+        end
+
+        cmds = [[], []]
+        commands.each do |cmd|
+          index = cmd[1].include?("--new-chain") ? 0 : 1
+          cmds[index] << cmd
+        end
+        commands.replace(cmds[0] + cmds[1])
+      end
+    end
+
+    def export
+      {
+        :name => name,
+        :chains => chains.values.map{ |chain| chain.export },
+      }
+    end
+  end
+end