blob: e1fe10bd341f1eafc675659c566612c3d4db8793 (
plain)
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
68
69
70
71
72
73
74
75
|
require 'net/http'
require 'uri'
Net::HTTP.version_1_2
class WserverPlugin < Plugin
def help(plugin, topic="")
"wserver <uri> => try and determine what webserver <uri> is using"
end
def privmsg(m)
unless(m.params && m.params =~ /^\S+$/)
m.reply "incorrect usage: " + help(m.plugins)
return
end
redirect_count = 0
hostname = m.params.dup
hostname = "http://#{hostname}" unless hostname =~ /:\/\//
begin
if(redirect_count > 3)
m.reply "cowardly refusing to follow more than 3 redirects"
return
end
begin
uri = URI.parse(hostname)
rescue URI::InvalidURIError => err
m.reply "#{m.params} is not a valid URI"
return
end
unless(uri)
m.reply "incorrect usage: " + help(m.plugin)
return
end
http = @bot.httputil.get_proxy(uri)
http.open_timeout = 5
http.start {|http|
resp = http.head('/')
server = resp['Server']
if(server && server.length > 0)
m.reply "#{uri.host} is running #{server}"
else
m.reply "couldn't tell what #{uri.host} is running"
end
if(resp.code == "302" || resp.code == "301")
newloc = resp['location']
newuri = URI.parse(newloc)
# detect and ignore incorrect redirects (to relative paths etc)
if (newuri.host != nil)
if(uri.host != newuri.host)
m.reply "#{uri.host} redirects to #{newuri.scheme}://#{newuri.host}"
raise resp['location']
end
end
end
}
rescue TimeoutError => err
m.reply "timed out connecting to #{uri.host}:#{uri.port} :("
return
rescue RuntimeError => err
redirect_count += 1
hostname = err.message
retry
rescue StandardError => err
puts err
m.reply "couldn't connect to #{uri.host}:#{uri.port} :("
return
end
end
end
plugin = WserverPlugin.new
plugin.register("wserver")
|
