Class: PassiveDNS::DNSDB

Inherits:
PassiveDB show all
Defined in:
lib/passivedns/client/providers/dnsdb.rb

Instance Attribute Summary collapse

Class Method Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(options = {}) ⇒ DNSDB

Returns a new instance of DNSDB.



23
24
25
26
27
 
# File 'lib/passivedns/client/providers/dnsdb.rb', line 23

def initialize(options={})
  @debug = options[:debug] || false
    @key = options["APIKEY"] || raise("APIKEY option required for #{self.class}")
    @base = options["URL"] || "https://api.dnsdb.info/lookup"
end

Instance Attribute Details

#debug ⇒ Object

Returns the value of attribute debug.



22
23
24
 
# File 'lib/passivedns/client/providers/dnsdb.rb', line 22

def debug
  @debug
end

Class Method Details

.config_section_name ⇒ Object

override



14
15
16
 
# File 'lib/passivedns/client/providers/dnsdb.rb', line 14

def self.config_section_name
  "dnsdb"
end

.name ⇒ Object

override



10
11
12
 
# File 'lib/passivedns/client/providers/dnsdb.rb', line 10

def self.name
  "DNSDB"
end

.option_letter ⇒ Object

override



18
19
20
 
# File 'lib/passivedns/client/providers/dnsdb.rb', line 18

def self.option_letter
  "d"
end

Instance Method Details

#lookup(label, limit = nil) ⇒ Object 



52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
 
# File 'lib/passivedns/client/providers/dnsdb.rb', line 52

def lookup(label, limit=nil)
  $stderr.puts "DEBUG: #{self.class.name}.lookup(#{label})" if @debug
  Timeout::timeout(240) {
    url = nil
    if label =~ /^\d{1,3}\.\d{1,3}\.\d{1,3}\.\d{1,3}(\/\d{1,2})?$/
      label = label.gsub(/\//,',')
      url = "#{@base}/rdata/ip/#{label}"
    else
      url = "#{@base}/rrset/name/#{label}"
    end
    url = URI.parse url
    http = Net::HTTP.new(url.host, url.port)
    http.use_ssl = (url.scheme == 'https')
    http.verify_mode = OpenSSL::SSL::VERIFY_NONE
    http.verify_depth = 5
      path = url.path
      if limit
        path << "?limit=#{limit}"
      end
    request = Net::HTTP::Get.new(path)
    request.add_field("User-Agent", "Ruby/#{RUBY_VERSION} passivedns-client rubygem v#{PassiveDNS::Client::VERSION}")
    request.add_field("X-API-Key", @key)
    request.add_field("Accept", "application/json")
    t1 = Time.now
    response = http.request(request)
    t2 = Time.now
    $stderr.puts response.body if @debug
    parse_json(response.body,t2-t1)
  }
rescue Timeout::Error => e
  $stderr.puts "#{self.class.name} lookup timed out: #{label}"
end

#parse_json(page, response_time) ⇒ Object 



29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
 
# File 'lib/passivedns/client/providers/dnsdb.rb', line 29

def parse_json(page,response_time)
  res = []
  raise "Error: unable to parse request" if page =~ /Error: unable to parse request/
  # need to remove the json_class tag or the parser will crap itself trying to find a class to align it to
  rows = page.split(/\n/)
  rows.each do |row|
    record = JSON.parse(row)
    record['rdata'] = [record['rdata']] if record['rdata'].class == String
    record['rdata'].each do |rdata|
      if record['time_first']
        res << PDNSResult.new(self.class.name,response_time,record['rrname'],rdata,record['rrtype'],0,Time.at(record['time_first'].to_i).utc.strftime("%Y-%m-%dT%H:%M:%SZ"),Time.at(record['time_last'].to_i).utc.strftime("%Y-%m-%dT%H:%M:%SZ"),record['count'])
      else
        res << PDNSResult.new(self.class.name,response_time,record['rrname'],rdata,record['rrtype'])
      end
    end
  end
  res
rescue Exception => e
  $stderr.puts "#{self.class.name} Exception: #{e}"
  $stderr.puts page
  raise e
end