-
Notifications
You must be signed in to change notification settings - Fork 9
/
botmatch.rb
67 lines (55 loc) · 1.36 KB
/
botmatch.rb
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
require 'resolv'
require 'ipaddr'
module Legitbot
##
# Represents a bot instance match. Typical methods are
# +valid?+, +fake?+ and +detected_as+
#
class BotMatch
def initialize(ip, resolver_config = nil)
@dns = Resolv::DNS.new(resolver_config)
@ip = ip
end
##
# Returns a Resolv::DNS::Name instance with
# the reverse name
def reverse_domain
@reverse_domain ||= @dns.getname(@ip)
rescue Resolv::ResolvError
@reverse_domain ||= nil
end
##
# Returns a String with the reverse name
def reverse_name
reverse_domain&.to_s
end
##
# Returns a String with IP created from the reverse name
def reversed_ip
return nil if reverse_name.nil?
@reverse_ip ||= @dns.getaddress(reverse_name)
@reverse_ip.to_s
end
def reverse_resolves?
@ip == reversed_ip
end
def subdomain_of?(*domains)
return false if reverse_name.nil?
domains.any? { |d|
reverse_domain.subdomain_of? Resolv::DNS::Name.create(d)
}
end
def detected_as
self.class.name.split('::').last.downcase.to_sym
end
def fake?
!valid?
end
def self.valid?(ip, resolver_config = nil)
self.new(ip, resolver_config).valid?
end
def self.fake?(ip, resolver_config = nil)
self.new(ip, resolver_config).fake?
end
end
end