doh_client 0.1.1 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 0b7bac2931e731855c7f89ec0b6957031ba9d65ebea0e030df7224986ace9a25
-  data.tar.gz: 437fdbeece0d4517d39c0017c7207b395c804ed224fa0dd6820407db45235fb0
+  metadata.gz: 39fc3cbc3ac6a57ae2bdef218ed12d1078903713622b1f333e60fe0c3ea15f0b
+  data.tar.gz: 73b7f165fc5bdf37ec4ebc1fb55bda75948b03596176f6e383c4f4dbad0b25b0
 SHA512:
-  metadata.gz: cc5732efa9139bcddba08bfc5f78ec5bad3880b253abe972bba869d3d75e8de4b70cf9eecf25dcb57c56e460a3d70893aed2b3eb4696a6e277aaefec579bd8af
-  data.tar.gz: e6cd0c9ca267d08d02c10865b27cf13fcacde426b63876582a41d488f995be9fa31944df86524eb2148a66beabd6f0d7e02397e4c33b6cda9bd2064048c91283
+  metadata.gz: f2720b92c65401f621aa41a30694e876c19e4b2427c5744534581418551cadaaa93c7a52a02e4a6111e915fb60cfb37c86fb3175f99c6be385b44b62d5c58c35
+  data.tar.gz: 9a84d7da0f96a247dff2d1d9d4cf7be2e060807e43cadfe9840656d2c6f4d62dd5f0cc02b860e0a5bd919d93b76ce83e14af3fac6eea4124d5dcd6841151fa4b

data/README.md

@@ -4,7 +4,7 @@
 [![Maintainability](https://api.codeclimate.com/v1/badges/e9119ee2021f1cfb895d/maintainability)](https://codeclimate.com/github/ninoseki/doh_client/maintainability)
 [![Coverage Status](https://coveralls.io/repos/github/ninoseki/doh_client/badge.svg?branch=master)](https://coveralls.io/github/ninoseki/doh_client?branch=master)
 
-DNS over HTTPS(DoH) client for Ruby
+DNS over HTTPS (DoH) client for Ruby
 
 ## Installation
 
@@ -46,14 +46,18 @@ DoHClient::Cloudflare.resolve("example.com", { type: "A", do: true, cd: false })
 ```bash
 $ doh_client
 Commands:
-  console help [COMMAND]  # Describe available commands or one specific command
-  console resolve [NAME]  # resolve a given name
+  doh_client act_as_server   # act as a local DNS server on a given port (default: 5300)
+  doh_client help [COMMAND]  # Describe available commands or one specific command
+  doh_client resolve [NAME]  # resolve a given name
 
 Options:
   [--resolver=RESOLVER]  # a resolver to use ('google' or 'cloudflare', default: google)
 
 $ doh_client resolve example.com --type A
-# => {"Status":0,"TC":false,"RD":true,"RA":true,"AD":true,"CD":false,"Question":[{"name":"example.com.","type":1}],"Answer":[{"name":"example.com.","type":1,"TTL":5169,"data":"93.184.216.34"}]}
+# => {"Status":0,"TC":false,"RD":true,"RA":true,"AD":true,"CD":false,"Question":[{"name":"example.com.","type":1}],"Answer":[{"name":"example.com.","type":1,"TTL":5169,"data":"93.184.216.34"}]
+
+$ doh_client act_as_server
+# => Starting DNS server 0.0.0.0:5300 (tcp/udp)
 ```
 
 ## License

data/doh_client.gemspec

@@ -24,6 +24,7 @@ Gem::Specification.new do |spec|
   spec.executables   = spec.files.grep(%r{^exe/}) { |f| File.basename(f) }
   spec.require_paths = ["lib"]
 
+  spec.add_development_dependency "async-rspec", "~> 1.11"
   spec.add_development_dependency "bundler", "~> 1.16"
   spec.add_development_dependency "coveralls", "~> 0.8"
   spec.add_development_dependency "rake", "~> 12.3"
@@ -31,6 +32,7 @@ Gem::Specification.new do |spec|
   spec.add_development_dependency "vcr", "~> 4.0"
   spec.add_development_dependency "webmock", "~> 3.4"
 
+  spec.add_dependency "async-dns", "~> 1.2"
   spec.add_dependency "http", "~> 3.3"
   spec.add_dependency "thor", "~> 0.19"
 end

data/lib/doh_client.rb

@@ -8,6 +8,8 @@ require "doh_client/client/base"
 require "doh_client/client/cloudflare"
 require "doh_client/client/google"
 
+require "doh_client/server"
+
 require "doh_client/cli"
 
 module DoHClient

data/lib/doh_client/cli.rb

@@ -18,6 +18,22 @@ module DoHClient
       puts hash.to_json
     end
 
+    desc "act_as_server", "act as a local DNS server on a given port (default: 5300)"
+    method_option :port, type: :numeric, default: 5300
+    def act_as_server
+      port = options[:port]
+      interfaces = [[:udp, "0.0.0.0", port], [:tcp, "0.0.0.0", port]]
+      server = DoHClient::Server.new(interfaces)
+      puts "Starting DNS server 0.0.0.0:#{port} (tcp/udp)"
+      begin
+        server.run
+      rescue Interrupt
+        puts "\nStopping DNS server..."
+      ensure
+        puts "Stopped"
+      end
+    end
+
     no_commands do
       def resolver
         case options[:resolver]

data/lib/doh_client/server.rb

@@ -0,0 +1,33 @@
+# frozen_string_literal: true
+
+require 'async/dns'
+
+module DoHClient
+  class Server < Async::DNS::Server
+    attr_reader :client
+    def initialize(resolver)
+      super
+      @client = resolver == "cloudflare" ? DoHClient::Client::Cloudflare : DoHClient::Client::Google
+    end
+
+    def process(name, resource_class, transaction)
+      type = resource_class.to_s.split('::').last
+
+      begin
+        response = client.resolve(name, type: type)
+      rescue ResponseError => _
+        return transaction.fail!(:NXDomain) unless answers
+      end
+
+      answers = response["Answer"]
+      return transaction.fail!(:NXDomain) unless answers
+
+      transaction.append_question!
+      answers.each do |answer|
+        next unless klass = Resolv::DNS::Resource.get_class(answer["type"], resource_class::ClassValue)
+        resource = klass < Resolv::DNS::Resource::DomainName ? klass.new(Resolv::DNS::Name.create(answer["data"])) : klass.new(answer["data"])
+        transaction.response.add_answer(answer["name"], answer["TTL"], resource)
+      end
+    end
+  end
+end

data/lib/doh_client/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module DoHClient
-  VERSION = "0.1.1"
+  VERSION = "0.2.0"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: doh_client
 version: !ruby/object:Gem::Version
-  version: 0.1.1
+  version: 0.2.0
 platform: ruby
 authors:
 - Manabu Niseki
@@ -10,6 +10,20 @@ bindir: exe
 cert_chain: []
 date: 2018-08-26 00:00:00.000000000 Z
 dependencies:
+- !ruby/object:Gem::Dependency
+  name: async-rspec
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.11'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.11'
 - !ruby/object:Gem::Dependency
   name: bundler
   requirement: !ruby/object:Gem::Requirement
@@ -94,6 +108,20 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: '3.4'
+- !ruby/object:Gem::Dependency
+  name: async-dns
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.2'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.2'
 - !ruby/object:Gem::Dependency
   name: http
   requirement: !ruby/object:Gem::Requirement
@@ -148,6 +176,7 @@ files:
 - lib/doh_client/client/google.rb
 - lib/doh_client/client/request.rb
 - lib/doh_client/error.rb
+- lib/doh_client/server.rb
 - lib/doh_client/version.rb
 homepage: https://github.com/ninoseki/doh_client
 licenses: