Class: Wepawet::Query

Inherits:
Object
  • Object
show all
Defined in:
lib/wepawet.rb

Instance Method Summary collapse

Constructor Details

#initialize(config = { 'wepawetSubmitUrl' => 'http://wepawet.cs.ucsb.edu/services/upload.php', 'wepawetQueryUrl' => 'http://wepawet.cs.ucsb.edu/services/query.php', 'wepawetDomainUrl' => 'http://wepawet.cs.ucsb.edu/services/domain.php', 'wepawetUrlUrl' => 'http://wepawet.cs.ucsb.edu/services/url.php', }) ⇒ Query

Returns a new instance of Query.



67
68
69
70
71
72
73
74
 
# File 'lib/wepawet.rb', line 67

def initialize(config = {
	'wepawetSubmitUrl' => 'http://wepawet.cs.ucsb.edu/services/upload.php', 
	'wepawetQueryUrl' => 'http://wepawet.cs.ucsb.edu/services/query.php',
	'wepawetDomainUrl' => 'http://wepawet.cs.ucsb.edu/services/domain.php',
	'wepawetUrlUrl' => 'http://wepawet.cs.ucsb.edu/services/url.php',
})
	@config = config
end

Instance Method Details

#_parse_response(doc) ⇒ Object 



103
104
105
106
107
108
109
110
111
112
113
114
 
# File 'lib/wepawet.rb', line 103

def _parse_response(doc)
	xml = XML::Document.string(doc)
	hash = {}
	xml.child.children.each do |node|
		if node.name =~ /\w/ and node.child
			hash[node.name] = node.child.content
		end
	end
	hash
rescue Exception
	return nil
end

#by_domain(domain) ⇒ Object 



95
96
97
 
# File 'lib/wepawet.rb', line 95

def by_domain(domain)
	by_whatever('domain',domain)
end

#by_taskid(taskid) ⇒ Object Also known as: by_hash 



89
90
91
 
# File 'lib/wepawet.rb', line 89

def by_taskid(taskid)
	by_whatever('hash',taskid)
end

#by_url(url) ⇒ Object 



99
100
101
 
# File 'lib/wepawet.rb', line 99

def by_url(url)
	by_whatever('url', CGI.escape(url))
end

#by_whatever(whatever, value) ⇒ Object 



76
77
78
79
80
81
82
83
84
85
86
87
 
# File 'lib/wepawet.rb', line 76

def by_whatever(whatever, value)
	params = {'resource_type' => 'js', whatever => value}
	urlkey = (whatever == 'hash') ? 'wepawetQueryUrl' : (whatever == 'domain') ? 'wepawetDomainUrl' : (whatever == 'url') ? 'wepawetUrlUrl' : 'wepawetQueryUrl'
	uri = URI.parse(@config[urlkey]+"?"+params.map{|k,v| "#{k}=#{v}"}.join("&"))
	http = Net::HTTP.new(uri.host, uri.port)
	http.use_ssl = (uri.scheme == 'https')
	http.verify_mode = OpenSSL::SSL::VERIFY_NONE
	request = Net::HTTP::Get.new(uri.path+"?"+uri.query)
	request.add_field("User-Agent", "Ruby/#{RUBY_VERSION} wepawet gem v#{Wepawet::VERSION}")
	response = http.request(request)
	_parse_response(response.body)
end