dvrtools 0.0.1

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 3ddb800c33e118f64a660d98b3eee2eb80da814c
+  data.tar.gz: aec02509baece2b80544dbef260250675dd091b3
+SHA512:
+  metadata.gz: 1c87f1fd6bfdec071b10a88f7b6464a3a773b06495e047f7eedf5b134e520149c691f6631b309298fb95706ba324e11657c8a33413aaa890075eb339d2979b5a
+  data.tar.gz: ce2e281d1400adf211bba648947c0a28e5532e1c9ed5be50e33760f86c61c39e024945cb9a21809891a55ad5c36cef1a2a76f70ed5abb3ffc291ba6583714dc9

data/.gitignore

@@ -0,0 +1,14 @@
+/.bundle/
+/.yardoc
+/Gemfile.lock
+/_yardoc/
+/coverage/
+/doc/
+/pkg/
+/spec/reports/
+/tmp/
+*.bundle
+*.so
+*.o
+*.a
+mkmf.log

data/Gemfile

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

data/LICENSE.txt

@@ -0,0 +1,22 @@
+Copyright (c) 2015 Masayoshi Mizutani
+
+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,31 @@
+# Dvrtools
+
+TODO: Write a gem description
+
+## Installation
+
+Add this line to your application's Gemfile:
+
+```ruby
+gem 'dvrtools'
+```
+
+And then execute:
+
+    $ bundle
+
+Or install it yourself as:
+
+    $ gem install dvrtools
+
+## Usage
+
+TODO: Write usage instructions here
+
+## Contributing
+
+1. Fork it ( https://github.com/[my-github-username]/dvrtools/fork )
+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 a new Pull Request

data/Rakefile

@@ -0,0 +1,2 @@
+require "bundler/gem_tasks"
+

data/bin/dnsmap

@@ -0,0 +1,155 @@
+#!/usr/bin/env ruby
+
+require 'optparse'
+require 'digest/md5'
+require 'msgpack'
+require 'eventmachine'
+require 'gviz'
+require 'pp'
+
+opt = {
+  :output => 'dnsmap',
+  :interval => 0,
+  :aggregate => false,
+  :layout => 'neato',
+  :verbose => false,
+}
+
+optpsr = OptionParser.new
+optpsr.on('-o VAL', 'output file name') {|v| opt[:output] = v }
+optpsr.on('-i VAL', 'interval (sec)') {|v| opt[:interval] = v.to_i } 
+optpsr.on('-a', 'aggregate IP address by /24') {|v| opt[:aggregate] = v }
+optpsr.on('-l VAL', 'graphviz layout, default: neato') {|v| opt[:layout] = v}
+optpsr.on('-q', 'quiet mode') {|v| $VERBOSE = nil}
+optpsr.on('-v', 'verbose mode') {|v| opt[:verbose] = v }
+
+args = optpsr.parse(ARGV)
+
+class DnsMap < EventMachine::Connection
+  def initialize(opt)
+    super()
+    @unpkr = MessagePack::Unpacker.new
+    @aggregate = false
+    @aggregate = opt[:aggregate]
+    @layout = opt[:layout]
+    @output_file = opt[:output]
+    @verbose = opt[:verbose]
+    renew_data
+  end
+  
+  def renew_data
+    @node_map = Hash.new{|h,k|
+      h[k] = {:id => ('n' + Digest::MD5.new.update(k).to_s).to_sym,
+              :type => ''}
+    }
+    @addrs = Hash.new{|h,k|
+      h[k] = Set.new
+    }
+    @dnsmap = Hash.new{|h,k| h[k] = Set.new}
+  end
+
+  def receive_data(data)
+    @unpkr.feed(data)
+    @unpkr.each {|msg|
+      recieve_object(msg)
+    }
+  end
+
+  def recieve_object(msg)
+    if msg[0] == 'dns.log'
+      obj = msg[2]
+      name = obj['name']
+      data = obj['data']
+
+      if @aggregate and /^\d{1,3}\.\d{1,3}\.\d{1,3}\.\d{1,3}$/.match(data)
+        data = data.split('.')[0,3].join('.') + '.*'
+        @addrs[data].add(obj['data'])
+      end
+      
+      @node_map[name][:type] = obj['type']
+      @node_map[data][:type] = ''
+      @dnsmap[name].add(data)
+    end
+  end
+
+  
+  def flush_data
+    dnsmap = @dnsmap
+    node_map  = @node_map
+    addrs  = @addrs
+    outfile = @output_file.to_sym
+    layout = @layout
+
+    warn "node count: #{@node_map.size}"
+    warn "flushing data..."
+
+    gv = Gviz.new
+    gv.graph do
+      global layout:"#{layout}", overlap:true, bgcolor:'white'
+      nodes fontname:'Verdana', fontsize:10, color:'black', fontcolor:'black'
+      edges color:'#222222'
+      node_map.each {|name, obj|
+        label = name
+        if addrs.has_key? name
+          names = addrs[name].to_a
+          case names.size
+          when 1;    label = names[0]
+          when 2..4; label = names.join('\n')
+          else;      label = name + " [#{names.size}]"
+          end
+        end
+        
+        if /^\d{1,3}\.\d{1,3}\.\d{1,3}\.\S+$/.match(name)
+          color = '#F9E3AA'
+          shape = 'box'
+          style = ''
+        else
+          case obj[:type]
+          when 'A'; color = '#6A8CC7'
+          when 'AAAA'; color = '#6A8CC7'
+          when 'CNAME'; color = '#AEC1E3'
+          else; color = '#EAA8BF'
+          end
+          shape = 'box'
+          style = ',rounded'
+        end
+        node obj[:id], label:"#{label}", style:'filled' + style, shape:"#{shape}", fillcolor:color
+      }
+        
+      dnsmap.each {|name, dset|
+        src = node_map[name][:id]
+        route src => dset.map{|d| node_map[d][:id]}
+      }
+      edges len: '2'
+    end
+
+    gv.save(outfile, :png)
+
+    renew_data
+  end
+  
+  def unbind
+    # closed STDIN
+    flush_data
+    EM.stop
+  end
+
+  def set_timer(pt); @pt = pt end    
+end
+  
+
+
+if args.size == 0
+  EM.run do
+    dm = EM.attach(STDIN, DnsMap, opt)
+    
+    if opt[:interval] > 0
+      pt = EventMachine::PeriodicTimer.new(interval) do
+        dm.flush_data
+      end
+      dm.set_timer(pt)
+    end
+
+  end
+end
+    

data/dvrtools.gemspec

@@ -0,0 +1,26 @@
+# coding: utf-8
+lib = File.expand_path('../lib', __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require 'dvrtools/version'
+
+Gem::Specification.new do |spec|
+  spec.name          = "dvrtools"
+  spec.version       = Dvrtools::VERSION
+  spec.authors       = ["Masayoshi Mizutani"]
+  spec.email         = ["mizutani@sfc.wide.ad.jp"]
+  spec.summary       = %q{Tools for Devourer.}
+  spec.description   = %q{Tool set of packet management.}
+  spec.homepage      = ""
+  spec.license       = "MIT"
+
+  spec.files         = `git ls-files -z`.split("\x0")
+  spec.executables   = spec.files.grep(%r{^bin/}) { |f| File.basename(f) }
+  spec.test_files    = spec.files.grep(%r{^(test|spec|features)/})
+  spec.require_paths = ["lib"]
+
+  spec.add_development_dependency "bundler", "~> 1.7"
+  spec.add_development_dependency "rake", "~> 10.0"
+  spec.add_dependency "msgpack", "~> 0.5.9"
+  spec.add_dependency "eventmachine", "~> 1.0.4"
+  spec.add_dependency "gviz", "~> 0.3.4"
+end

data/lib/dvrtools.rb

@@ -0,0 +1,5 @@
+require "dvrtools/version"
+
+module Dvrtools
+  # Your code goes here...
+end

data/lib/dvrtools/version.rb

@@ -0,0 +1,3 @@
+module Dvrtools
+  VERSION = "0.0.1"
+end

metadata

@@ -0,0 +1,124 @@
+--- !ruby/object:Gem::Specification
+name: dvrtools
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+platform: ruby
+authors:
+- Masayoshi Mizutani
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2015-01-05 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.7'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.7'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '10.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '10.0'
+- !ruby/object:Gem::Dependency
+  name: msgpack
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.5.9
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.5.9
+- !ruby/object:Gem::Dependency
+  name: eventmachine
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 1.0.4
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 1.0.4
+- !ruby/object:Gem::Dependency
+  name: gviz
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.3.4
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.3.4
+description: Tool set of packet management.
+email:
+- mizutani@sfc.wide.ad.jp
+executables:
+- dnsmap
+extensions: []
+extra_rdoc_files: []
+files:
+- ".gitignore"
+- Gemfile
+- LICENSE.txt
+- README.md
+- Rakefile
+- bin/dnsmap
+- dvrtools.gemspec
+- lib/dvrtools.rb
+- lib/dvrtools/version.rb
+homepage: ''
+licenses:
+- MIT
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.4.5
+signing_key: 
+specification_version: 4
+summary: Tools for Devourer.
+test_files: []