em-powerdns 0.0.1

data/.gitignore

@@ -0,0 +1,23 @@
+## MAC OS
+.DS_Store
+
+## TEXTMATE
+*.tmproj
+tmtags
+
+## EMACS
+*~
+\#*
+.\#*
+
+## VIM
+*.swp
+
+## PROJECT::GENERAL
+coverage
+rdoc
+pkg
+.bundle
+vendor
+
+## PROJECT::SPECIFIC

data/Gemfile

@@ -0,0 +1,20 @@
+source :gemcutter
+
+group :runtime do
+  gem 'eventmachine'
+end
+
+group :release do
+  gem 'jeweler'
+  gem 'rake'
+  gem 'bundler', '~> 0.9.11'
+end
+
+group :test do
+  gem 'rspec',    '>= 1.2.9', :require => 'spec'
+  gem 'reek'
+  gem 'roodi'
+  gem 'yard'
+  gem 'ruby-debug'
+  gem 'sys-uname'
+end

data/LICENSE

@@ -0,0 +1,20 @@
+Copyright (c) 2009 Martin Emde
+
+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.rdoc

@@ -0,0 +1,36 @@
+= em-powerdns
+
+You can use this plus the unix program PowerDNS to make a custom DNS server
+with the lookup done by a ruby script you write.
+
+Even though the actual PowerDNS pipe backend is synchronous, using this will
+allow a script to start up other side tasks, logging or stats for example,
+without interrupting the scripts ability to serve more requests.
+
+Include EM::P::PowerDNS in a module or class to use it as a PowerDNS backend.
+
+== Example
+
+  module RedisBackend
+    include EM::P::PowerDNS
+
+    def receive_query(query)
+      # make some logic here to resolve the DNS
+      ip = redis[query.qname]
+      data query.qname, query.qclass, :A, 3600, 1, ip
+      super
+    end
+
+    def redis
+      @redis ||= Redis.new
+    end
+  end
+
+  EM.run { EM.open_keyboard(RedisBackend) }
+
+PowerDNS uses stdin to send all it's queries so EM.open_keyboard does exactly
+what we need.
+
+== Copyright
+
+Copyright (c) 2010 Martin Emde. See LICENSE for details.

data/Rakefile

@@ -0,0 +1,86 @@
+require 'bundler'
+Bundler.setup
+Bundler.require
+
+load 'tasks/powerdns.rake'
+
+begin
+  require 'jeweler'
+  Jeweler::Tasks.new do |gem|
+    gem.name = "em-powerdns"
+    gem.summary = %Q{Ruby EventMachine PowerDNS Protocol for PDNS custom pipe backend}
+    gem.description = gem.summary
+    gem.email = "martin.emde@gmail.com"
+    gem.homepage = "http://github.com/martinemde/em-powerdns"
+    gem.authors = ["Martin Emde"]
+
+    bundle = Bundler::Definition.from_gemfile('Gemfile')
+    bundle.dependencies.each do |dep|
+      gem.add_dependency(dep.name, dep.requirement.to_s) if dep.groups.include?(:runtime)
+    end
+  end
+  Jeweler::GemcutterTasks.new
+rescue LoadError
+  puts "Jeweler (or a dependency) not available. Install it with: gem install jeweler"
+end
+
+require 'spec/rake/spectask'
+Spec::Rake::SpecTask.new(:spec) do |spec|
+  spec.libs << 'lib' << 'spec'
+  spec.spec_files = FileList['spec/**/*_spec.rb']
+end
+
+Spec::Rake::SpecTask.new(:rcov) do |spec|
+  spec.libs << 'lib' << 'spec'
+  spec.pattern = 'spec/**/*_spec.rb'
+  spec.rcov = true
+end
+
+begin
+  require 'cucumber/rake/task'
+  Cucumber::Rake::Task.new(:features) do |t|
+    t.cucumber_opts = %w{--guess}
+  end
+
+  task :features => :check_dependencies
+rescue LoadError
+  task :features do
+    abort "Cucumber is not available. In order to run features, you must: sudo gem install cucumber"
+  end
+end
+
+task :default => %w[spec features]
+
+begin
+  require 'reek/adapters/rake_task'
+  Reek::RakeTask.new do |t|
+    t.fail_on_error = true
+    t.verbose = false
+    t.source_files = 'lib/**/*.rb'
+  end
+rescue LoadError
+  task :reek do
+    abort "Reek is not available. In order to run reek, you must: sudo gem install reek"
+  end
+end
+
+begin
+  require 'roodi'
+  require 'roodi_task'
+  RoodiTask.new do |t|
+    t.verbose = false
+  end
+rescue LoadError
+  task :roodi do
+    abort "Roodi is not available. In order to run roodi, you must: sudo gem install roodi"
+  end
+end
+
+begin
+  require 'yard'
+  YARD::Rake::YardocTask.new
+rescue LoadError
+  task :yardoc do
+    abort "YARD is not available. In order to run yardoc, you must: sudo gem install yard"
+  end
+end

data/VERSION

@@ -0,0 +1 @@
+0.0.1

data/lib/em-powerdns.rb

@@ -0,0 +1,139 @@
+require 'eventmachine'
+
+# Instances of this class wrap the PowerDNS I/O stream with a nice API
+module EventMachine::Protocols
+  module PowerDNS
+    class Error < RuntimeError
+      attr_reader :original_exception
+
+      def initialize(data, exception)
+        @original_exception = exception
+        super "Unexpected error in the EM::P::PowerDNS:\nPowerDNS Data: #{data.inspect}\nError: #{exception}\n#{exception.backtrace}"
+      end
+    end
+
+    include EM::Protocols::LineText2
+    include Output
+
+    SEPARATOR = "\t"
+    EVENT_MAP = Hash.new(:receive_garbage)
+    EVENT_MAP.update(
+      'HELO' => :receive_raw_handshake,
+      'Q'    => :receive_raw_query,
+      'AXFR' => :receive_raw_axfr,
+      'PING' => :receive_raw_ping
+    )
+
+    def self.start
+      EM.open_keyboard(self)
+    end
+
+    def initialize
+      $stdin.sync = true
+      $stdout.sync = true
+    end
+
+    def receive_line(data)
+      parts = data.split SEPARATOR
+      type = parts.shift
+      send EVENT_MAP[type], *parts
+    rescue => e
+      receive_error Error.new(data, e)
+    end
+
+    def receive_raw_handshake(*parts)
+      @version = parts.first.to_i
+      if [1,2].include?(@version)
+        receive_handshake(@version)
+      else
+        fail "Received unexpected handshake: #{parts.inspect}"
+      end
+    end
+
+    def receive_handshake(version)
+      ok "Backend starting version #{version}."
+    end
+
+    class Query < Struct.new(:qname, :qclass, :qtype, :id, :remote_ip, :local_ip)
+    end
+
+    def receive_raw_query(*parts)
+      if parts.size != parts_size_for_version
+        fail "Received unexpected format for Q: #{parts.inspect}"
+      else
+        query = Query.new(*parts)
+        receive_query(query)
+      end
+    end
+
+    def parts_size_for_version
+      @version == 1 ? 5 : 6 # already chopped off the 'Q'
+    end
+
+    # Stub method for users implementing this protocol
+    def receive_query(query)
+      done
+    end
+
+    def receive_raw_axfr(*parts)
+      if parts.size != 1
+        fail "Received unexpected format for AXFR: #{parts.inspect}"
+      else
+        receive_axfr(parts.first)
+      end
+    end
+
+    # Stub method for users implementing this protocol
+    def receive_axfr(soa_resource)
+      done
+    end
+
+    def receive_raw_ping(*parts)
+      receive_ping
+    end
+
+    # Stub method for users implementing this protocol
+    def receive_ping
+      done
+    end
+
+    def receive_garbage(line)
+      fail "Unknown Question: #{line.inspect}"
+    end
+
+    def receive_error(error)
+      fail "An unexpected error occurred in backend: #{error.original_exception.message}"
+      raise error
+    end
+    module Output
+    def ok(line)
+      send_line "OK", line
+    end
+
+    def data(*parts)
+      send_line "DATA", *parts
+    end
+
+    def log(line)
+      send_line "LOG", line
+    end
+
+    def fail(line = nil)
+      log(line) if line
+      send_line "FAIL"
+    end
+
+    def done(line = nil)
+      log(line) if line
+      send_line "END"
+    end
+
+    def send_line(*parts)
+      send_data parts.join(PowerDNS::Connection::SEPARATOR)
+    end
+
+    def send_data(data)
+      $stdout.puts data
+    end
+  end
+end

data/tasks/powerdns.rake

@@ -0,0 +1,219 @@
+# Inspired by powerdns, which was inspired by rabbitmq.rake the Redbox project at http://github.com/rick/redbox/tree/master
+require 'rake'
+require 'fileutils'
+require 'pathname'
+require 'sys/uname'
+
+class PDNS
+  SOURCE_URI  = "http://downloads.powerdns.com/releases/pdns-2.9.22.tar.gz"
+  BOOST_URI   = "http://downloads.sourceforge.net/project/boost/boost/1.42.0/boost_1_42_0.tar.gz?use_mirror=superb-sea2"
+  DTACH_URI   = "http://downloads.sourceforge.net/project/dtach/dtach/0.8/dtach-0.8.tar.gz"
+  ROOT        = (Pathname.new(File.dirname(__FILE__)) + '../powerdns').cleanpath
+  PDNS_SERVER = 'pdns_server'
+  BACKEND     = 'backend'
+
+  def self.bin() ROOT + 'bin' end
+  def self.etc() ROOT + 'etc' end
+  def self.src() ROOT + 'src' end
+  def self.tmp() ROOT + 'tmp' end
+
+  def self.source_path()    src + 'powerdns' end
+  def self.compiled_path()  source_path + 'pdns' end
+  def self.boost_path()     src + 'boost' end
+
+  def self.server_command() bin + PDNS_SERVER end
+  def self.pipe_command()   bin + BACKEND end
+  def self.config()         etc + 'pdns.conf' end
+
+  def self.dtach_socket
+    tmp.mkpath
+    tmp + 'powerdns.dtach'
+  end
+
+  # Just check for existance of dtach socket
+  def self.running?
+    File.exists? dtach_socket
+  end
+
+  def self.start
+    puts 'Detach with Ctrl+\  Re-attach with rake powerdns:attach'
+    sleep 3
+    exec "dtach -A #{dtach_socket} #{server_command}"
+  end
+
+  def self.start_detached
+    system "dtach -n #{dtach_socket} #{server_command}"
+  end
+
+  def self.attach
+    exec "dtach -a #{dtach_socket}"
+  end
+
+  def self.stop
+    system 'echo "SHUTDOWN" | nc localhost 6379'
+  end
+
+end
+
+namespace :powerdns do
+
+  desc 'About powerdns'
+  task :about do
+    puts "\nSee http://www.powerdns.com/ for information about PowerDNS.\n\n"
+  end
+
+  desc 'Start powerdns'
+  task :start do
+    PDNS.start
+  end
+
+  desc 'Stop powerdns'
+  task :stop do
+    PDNS.stop
+  end
+
+  desc 'Restart powerdns'
+  task :restart do
+    PDNS.stop
+    PDNS.start
+  end
+
+  desc 'Attach to powerdns dtach socket'
+  task :attach do
+    PDNS.attach
+  end
+
+  desc 'Install PowerDNS server and configuration file'
+  task :install => %w[about install:server install:backend install:config]
+
+  namespace :install do
+    desc 'Install PowerDNS server'
+    task :server => [:paths] do
+      script = PDNS.server_command
+      script.delete if script.exist?
+      script.open('w') do |f|
+        f.puts <<-SCRIPT
+#!/bin/bash
+sudo #{PDNS.compiled_path.realpath + PDNS::PDNS_SERVER} --config-dir=#{PDNS.etc}
+        SCRIPT
+      end
+      script.chmod(0755)
+      puts "Installed #{script}."
+    end
+
+    desc 'Install PowerDNS config'
+    task :config => [:backend, :paths] do
+      config = PDNS.config
+      config.delete if config.exist?
+      config.open('w') do |file|
+        file.puts <<-CONFIG
+launch=pipe
+pipe-command=#{PDNS.pipe_command}
+pipebackend-abi-version=2
+        CONFIG
+      end
+      `cat #{PDNS.compiled_path.realpath + 'pdns.conf-dist'} >> #{config}`
+      puts "Installed #{config} and configured it for testing.\n You should double check this file."
+    end
+
+    desc 'Install backend wrapper'
+    task :backend => [:paths] do
+      gem_root = File.expand_path('../..', __FILE__)
+      pipe = PDNS.pipe_command
+      pipe.delete if pipe.exist?
+      pipe.open('w') do |f|
+        f.puts <<-SCRIPT
+#!/bin/bash
+cd #{gem_root}
+bundle exec bin/backend
+        SCRIPT
+      end
+      pipe.chmod(0755)
+      puts "Linked #{pipe} to #{executable}."
+    end
+  end
+
+  desc "Configure and make PowerDNS"
+  task :make => [:paths, :download, :boost] do
+    cflags = "-I#{PDNS.boost_path.realpath}"
+    cflags << " -DDARWIN" if Sys::Uname.sysname =~ /Darwin/i
+
+    system <<-COMMAND
+cd #{PDNS.source_path}
+
+echo "Configuring PowerDNS for pipe backend ..."
+CXXFLAGS="#{cflags}" ./configure --with-modules=pipe --without-pgsql --without-mysql --without-sqlite --without-sqlite3
+
+echo "make clean ..."
+make clean
+
+echo "make ..."
+make
+    COMMAND
+  end
+
+  desc "Download PowerDNS source"
+  task :download => [:paths, :boost] do
+    source_path = PDNS.source_path
+    unless source_path.exist?
+      source_path.mkpath
+      puts "Downloading PowerDNS source..."
+      system "curl -L #{PDNS::SOURCE_URI} | tar zx --strip-components 1 -C #{source_path}"
+    end
+  end
+
+  desc "PowerDNS requires boost libs at compile time"
+  task :boost => :paths do
+    boost_path = PDNS.boost_path
+    unless boost_path.exist?
+      boost_path.mkpath
+      puts "Downloading Boost source..."
+      system "curl -L #{PDNS::BOOST_URI} | tar zx --strip-components 1 -C #{boost_path}"
+    end
+  end
+
+  task :paths do
+    [PDNS.bin, PDNS.etc, PDNS.src].each { |path| path.mkpath }
+  end
+
+  namespace :purge do
+    def purge(*paths)
+      paths.each do |path|
+        path.rmtree if path.exist?
+        path.mkpath
+      end
+    end
+
+    task :installs do
+      purge(PDNS.bin, PDNS.etc)
+    end
+    
+    task :src do
+      purge(PDNS.src)
+    end
+  end
+end
+
+namespace :dtach do
+  desc 'About dtach'
+  task :about do
+    puts <<-ABOUT
+dtach allows background proccess that are not attached to a term.
+See http://dtach.sourceforge.net/ for more information about dtach.
+
+ABOUT
+  end
+
+  desc 'Install dtach 0.8 from source'
+  task :install => [:about] do
+    dtach_path = PDNS.source_path + 'dtach'
+
+    unless dtach.exist?
+      dtach.mkpath
+      system "curl -L #{PDNS::DTACH_URI} | tar zx --strip-components 1 -C #{dtach_path}"
+    end
+
+    system "cd #{dtach_path} && ./configure && make && ln -s dtach #{PDNS.bin + 'dtach'}"
+    puts "dtach installed to #{PDNS.bin}."
+  end
+end

metadata

@@ -0,0 +1,81 @@
+--- !ruby/object:Gem::Specification 
+name: em-powerdns
+version: !ruby/object:Gem::Version 
+  prerelease: false
+  segments: 
+  - 0
+  - 0
+  - 1
+  version: 0.0.1
+platform: ruby
+authors: 
+- Martin Emde
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2010-04-10 00:00:00 -07:00
+default_executable: 
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: eventmachine
+  prerelease: false
+  requirement: &id001 !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        segments: 
+        - 0
+        version: "0"
+  type: :runtime
+  version_requirements: *id001
+description: Ruby EventMachine PowerDNS Protocol for PDNS custom pipe backend
+email: martin.emde@gmail.com
+executables: []
+
+extensions: []
+
+extra_rdoc_files: 
+- LICENSE
+- README.rdoc
+files: 
+- .gitignore
+- Gemfile
+- LICENSE
+- README.rdoc
+- Rakefile
+- VERSION
+- lib/em-powerdns.rb
+- tasks/powerdns.rake
+has_rdoc: true
+homepage: http://github.com/martinemde/em-powerdns
+licenses: []
+
+post_install_message: 
+rdoc_options: 
+- --charset=UTF-8
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      segments: 
+      - 0
+      version: "0"
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      segments: 
+      - 0
+      version: "0"
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.3.6
+signing_key: 
+specification_version: 3
+summary: Ruby EventMachine PowerDNS Protocol for PDNS custom pipe backend
+test_files: []
+