resolver_replace 0.0.1

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 13a898dc3155fb550affbfe858cb070d033e2d74
+  data.tar.gz: dcfc046f9e40bade813df3f46715667965e2bf46
+SHA512:
+  metadata.gz: e498affc9788ec2e3bac1dd092d9bac53eceba8a83e925b567537b8fbc7025366777bf55324e186685824f55ae22675328824f9d6530255f1ea5caa0d8be57ab
+  data.tar.gz: f5c747339d275a779a66defce9c1575eef9845afd71353ac2fa64372296237c2bd67d074fab1879ee42fb5c0fc764d2d4704600c21fb9fd3c14a6a43da14df85

data/CHANGELOG.md

@@ -0,0 +1,3 @@
+## 0.0.1
+
+First version

data/Gemfile

@@ -0,0 +1,4 @@
+source 'https://rubygems.org'
+
+gemspec
+gem 'mysql2' # for sample plugin test

data/Gemfile.lock

@@ -0,0 +1,44 @@
+PATH
+  remote: .
+  specs:
+    resolver_replace (0.0.1)
+
+GEM
+  remote: https://rubygems.org/
+  specs:
+    coderay (1.1.0)
+    diff-lcs (1.2.5)
+    method_source (0.8.2)
+    mysql2 (0.3.16)
+    pry (0.9.12.6)
+      coderay (~> 1.0)
+      method_source (~> 0.8)
+      slop (~> 3.4)
+    pry-nav (0.2.3)
+      pry (~> 0.9.10)
+    rake (10.3.2)
+    rspec (3.0.0)
+      rspec-core (~> 3.0.0)
+      rspec-expectations (~> 3.0.0)
+      rspec-mocks (~> 3.0.0)
+    rspec-core (3.0.2)
+      rspec-support (~> 3.0.0)
+    rspec-expectations (3.0.2)
+      diff-lcs (>= 1.2.0, < 2.0)
+      rspec-support (~> 3.0.0)
+    rspec-mocks (3.0.2)
+      rspec-support (~> 3.0.0)
+    rspec-support (3.0.2)
+    slop (3.5.0)
+
+PLATFORMS
+  ruby
+
+DEPENDENCIES
+  bundler (~> 1.6)
+  mysql2
+  pry
+  pry-nav
+  rake
+  resolver_replace!
+  rspec

data/LICENSE.txt

@@ -0,0 +1,22 @@
+Copyright (c) 2014 Naotoshi Seo
+
+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,72 @@
+# resolver-replace
+
+Replace the DNS resolver.
+
+## Installation
+
+```bash
+gem install resolver_replace
+```
+
+## How it works
+
+Ruby comes with a pure Ruby replacement, [resolv-replace](https://github.com/ruby/ruby/blob/b1f2effda85efd03bd4ad5c06e0aae5e14f3f864/lib/resolv-replace.rb). 
+It replaces the libc resolver with the built-in [Resolv](http://apidock.com/ruby/Resolv) class which is a thread-aware DNS resolver library.
+
+This gem works similarly, but allows to replace the resolver with your favorite resolver. 
+
+## How to use
+
+For example, if you want to replace the resolver with `Resolv::DNS`, write as followings:
+
+```ruby
+require 'resolve' # Resolve::DNS
+require 'resolver-replace' # ResolverReplace
+
+resolver = Resolv::DNS.new(:nameserver => ['210.251.121.21'],
+                           :search => ['ruby-lang.org'],
+                           :ndots => 1)
+
+ResolverReplace.configure(
+  getaddress: Proc.new {|host| resolver.getaddress(host) },
+  getaddresses: Proc.new {|host| resolver.getaddresses(host) },
+  error_class: Resolv::ResolvError,
+)
+```
+
+## Plugin
+
+Some gems like `mysql2` implement its connection in its C extension without using ruby library. 
+In such case, another monkeypatch must be applied. A plugin scheme of ResolveReplace is ready for such case.
+
+The `mysql2` plugin is available for example. Use as followings:
+
+```ruby
+ResolverReplace.load_plugin('mysql2')
+ResolverReplace.configure(
+  getaddress: Proc.new {|host| resolver.getaddress(host) },
+  getaddresses: Proc.new {|host| resolver.getaddresses(host) },
+  error_class: Resolv::ResolvError,
+)
+```
+
+## ChangeLog
+
+See [CHANGELOG.md](CHANGELOG.md) for details.
+
+## See Also
+
+* [例えば Resolver をすげかえる - sonots:blog](http://blog.livedoor.jp/sonots/archives/38822072.html)
+
+## 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](../../pull/new/master)
+
+## Copyright
+
+Copyright (c) 2014 Naotoshi Seo. See [LICENSE.txt](LICENSE.txt) for details.
+

data/Rakefile

@@ -0,0 +1,14 @@
+require "bundler/gem_tasks"
+
+require 'rspec/core'
+require 'rspec/core/rake_task'
+RSpec::Core::RakeTask.new(:spec) do |spec|
+  spec.pattern = FileList['spec/**/*_spec.rb']
+end
+task :default => :spec
+
+desc 'Open an irb session preloaded with the gem library'
+task :console do
+    sh 'irb -rubygems -I lib -r resolver_replace'
+end
+task :c => :console

data/example/example.rb

@@ -0,0 +1,11 @@
+require 'resolve'
+
+resolver = Resolv::DNS.new(:nameserver => ['210.251.121.21'],
+                           :search => ['ruby-lang.org'],
+                           :ndots => 1)
+
+ResolverReplace.configure(
+  getaddress_proc: Proc.new {|host| resolver.getaddress(host) },
+  getaddresses_proc: Proc.new {|host| resolver.getaddresses(host) },
+  error_class: Resolv::ResolvError,
+)

data/lib/resolver_replace/plugins/mysql2.rb

@@ -0,0 +1,15 @@
+if defined?(Mysql2)
+  module Mysql2
+    class Client
+      alias original_resolver_connect connect
+      def connect user, pass, host, port, database, socket, flags
+        begin
+          ip = ResolverReplace.getaddress(host).to_s
+          original_resolver_connect user, pass, ip, port, database, socket, flags
+        rescue ResolverReplace.error_class => e
+          raise Mysql2::Error, "#{e.class}: #{e.message}"
+        end
+      end
+    end
+  end
+end

data/lib/resolver_replace.rb

@@ -0,0 +1,102 @@
+class ResolverReplace
+  def self.register!(params = {})
+    unless @getaddress = params[:getaddress]
+      raise ArgumentError, 'getaddress is required'
+    end
+    unless @getaddresses = params[:getaddresses]
+      raise ArgumentError, 'getaddresses is required'
+    end
+    @error_class = params[:error_class] || StandardError
+  end
+
+  def self.getaddress(host)
+    @getaddress.call(host)
+  end
+
+  def self.getaddresses(host)
+    @getaddresses.call(host)
+  end
+
+  def self.error_class
+    @error_class
+  end
+
+  def self.load_plugin(name)
+    load "resolver_replace/plugins/#{name}.rb"
+  end
+end
+
+# reference: ruby/lib/resolv-replace.rb
+require 'socket'
+
+class << IPSocket
+  # :stopdoc:
+  alias original_resolver_getaddress getaddress
+  # :startdoc:
+  def getaddress(host)
+    begin
+      return ResolverReplace.getaddress(host).to_s
+    rescue ResolverReplace.error_class => e
+      raise SocketError, "#{e.class} #{e.message}"
+    end
+  end
+end
+ 
+class TCPSocket < IPSocket
+  # :stopdoc:
+  alias original_resolver_initialize initialize
+  # :startdoc:
+  def initialize(host, serv, *rest)
+    rest[0] = IPSocket.getaddress(rest[0]) unless rest.empty?
+    original_resolver_initialize(IPSocket.getaddress(host), serv, *rest)
+  end
+end
+ 
+class UDPSocket < IPSocket
+  # :stopdoc:
+  alias original_resolver_bind bind
+  # :startdoc:
+  def bind(host, port)
+    host = IPSocket.getaddress(host) if host != ""
+    original_resolver_bind(host, port)
+  end
+ 
+  # :stopdoc:
+  alias original_resolver_connect connect
+  # :startdoc:
+  def connect(host, port)
+    original_resolver_connect(IPSocket.getaddress(host), port)
+  end
+ 
+  # :stopdoc:
+  alias original_resolver_send send
+  # :startdoc:
+  def send(mesg, flags, *rest)
+    if rest.length == 2
+      host, port = rest
+      begin
+        addrs = ResolverReplace.getaddresses(host)
+      rescue ResolverReplace.error_class => e
+        raise SocketError, "#{e.class} #{e.message}"
+      end
+      addrs[0...-1].each {|addr|
+        begin
+          return original_resolver_send(mesg, flags, addr, port)
+        rescue SystemCallError
+        end
+      }
+      original_resolver_send(mesg, flags, addrs[-1], port)
+    else
+      original_resolver_send(mesg, flags, *rest)
+    end
+  end
+end
+ 
+class SOCKSSocket < TCPSocket
+  # :stopdoc:
+  alias original_resolver_initialize initialize
+  # :startdoc:
+  def initialize(host, serv)
+    original_resolver_initialize(IPSocket.getaddress(host), port)
+  end
+end if defined? SOCKSSocket

data/resolver_replace.gemspec

@@ -0,0 +1,25 @@
+# coding: utf-8
+lib = File.expand_path('../lib', __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+
+Gem::Specification.new do |spec|
+  spec.name          = "resolver_replace"
+  spec.version       = "0.0.1"
+  spec.authors       = ["Naotoshi Seo"]
+  spec.email         = ["sonots@gmail.com"]
+  spec.summary       = %q{Replace the DNS resolver}
+  spec.description   = %q{Replace the DNS resolver.}
+  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.6"
+  spec.add_development_dependency "rake"
+  spec.add_development_dependency "rspec"
+  spec.add_development_dependency "pry"
+  spec.add_development_dependency "pry-nav"
+end

data/spec/resolver_replace_spec.rb

@@ -0,0 +1,42 @@
+require_relative 'spec_helper'
+require 'resolver_replace'
+
+describe ResolverReplace do
+  let(:host) { 'ruby-lang.org' }
+  let(:ip)   { '127.0.0.1' }
+
+  before do
+    ResolverReplace.register!(
+      getaddress: Proc.new {|host| ip },
+      getaddresses: Proc.new {|host| ip },
+      error_class: StandardError,
+    )
+  end
+
+  it 'IPSocket' do
+    expect(IPSocket.getaddress(host)).to be == ip
+  end
+
+  it 'net/http' do
+    require 'net/http'
+    begin
+      Net::HTTP.get_print(host, '/index.html')
+    rescue => e
+      expect(e.message).to match(/Connection refused.*127.0.0.1/)
+    end
+  end
+
+  it 'mysql2' do
+    require 'mysql2'
+    ResolverReplace.load_plugin("mysql2")
+    begin
+      Mysql2::Client.new(
+        host: 'foo',
+        username: 'root',
+        database: 'foo',
+      )
+    rescue => e
+      expect(e.message).not_to match(/Unknown MySQL server/)
+    end
+  end
+end

data/spec/spec_helper.rb

@@ -0,0 +1,15 @@
+# encoding: UTF-8
+require 'rubygems'
+require 'bundler'
+Bundler.setup(:default, :test)
+Bundler.require(:default, :test)
+require 'pry'
+ROOT = File.dirname(File.dirname(__FILE__))
+
+if ENV['TRAVIS']
+  require 'coveralls'
+  Coveralls.wear!
+end
+
+$TESTING=true
+$:.unshift File.join(File.dirname(__FILE__), '..', 'lib')

metadata

@@ -0,0 +1,128 @@
+--- !ruby/object:Gem::Specification
+name: resolver_replace
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+platform: ruby
+authors:
+- Naotoshi Seo
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2014-06-25 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.6'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.6'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: pry
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: pry-nav
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+description: Replace the DNS resolver.
+email:
+- sonots@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- CHANGELOG.md
+- Gemfile
+- Gemfile.lock
+- LICENSE.txt
+- README.md
+- Rakefile
+- example/example.rb
+- lib/resolver_replace.rb
+- lib/resolver_replace/plugins/mysql2.rb
+- resolver_replace.gemspec
+- spec/resolver_replace_spec.rb
+- spec/spec_helper.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.2.2
+signing_key: 
+specification_version: 4
+summary: Replace the DNS resolver
+test_files:
+- spec/resolver_replace_spec.rb
+- spec/spec_helper.rb