4 # :title: rbot HTTP provider
6 # Author:: Tom Gilbert <tom@linuxbrit.co.uk>
7 # Author:: Giuseppe "Oblomov" Bilotta <giuseppe.bilotta@gmail.com>
8 # Author:: Dmitry "jsn" Kim <dmitry point kim at gmail point com>
10 # Copyright:: (C) 2002-2005 Tom Gilbert
11 # Copyright:: (C) 2006 Tom Gilbert, Giuseppe Bilotta
12 # Copyright:: (C) 2007 Giuseppe Bilotta, Dmitry Kim
20 error "Couldn't load 'net/https': #{e.inspect}"
21 error "Secured HTTP connections will fail"
24 # To handle Gzipped pages
30 attr_accessor :no_cache
31 if !instance_methods.include?('raw_body')
35 def body_charset(str=self.raw_body)
36 ctype = self['content-type'] || 'text/html'
37 return nil unless ctype =~ /^text/i || ctype =~ /x(ht)?ml/i
39 charsets = ['latin1'] # should be in config
41 if self['content-type'].match(/charset=["']?([^\s"']+)["']?/i)
43 debug "charset #{charsets.last} added from header"
47 when /<\?xml\s[^>]*encoding=['"]([^\s"'>]+)["'][^>]*\?>/i
49 debug "xml charset #{charsets.last} added from xml pi"
50 when /<(meta\s[^>]*http-equiv=["']?Content-Type["']?[^>]*)>/i
52 if meta =~ /charset=['"]?([^\s'";]+)['"]?/
54 debug "html charset #{charsets.last} added from meta"
61 charsets = self.body_charset(str) or return str
63 charsets.reverse_each { |charset|
65 return Iconv.iconv('utf-8//ignore', charset, str).first
67 debug "conversion failed for #{charset}"
73 def decompress_body(str)
74 method = self['content-encoding']
79 debug "gunzipping body"
80 return Zlib::GzipReader.new(StringIO.new(str)).read
82 raise "Unhandled content encoding #{method}"
87 return self.body_to_utf(self.decompress_body(self.raw_body))
90 # Read chunks from the body until we have at least _size_ bytes, yielding
91 # the partial text at each chunk. Return the partial body.
92 def partial_body(size=0, &block)
97 self.read_body { |chunk|
99 yield self.body_to_utf(partial) if block_given?
100 break if size and size > 0 and partial.length >= size
103 return self.body_to_utf(partial)
108 Net::HTTP.version_1_2
113 # class for making http requests easier (mainly for plugins to use)
114 # this class can check the bot proxy configuration to determine if a proxy
115 # needs to be used, which includes support for per-url proxy configuration.
117 BotConfig.register BotConfigBooleanValue.new('http.use_proxy',
118 :default => false, :desc => "should a proxy be used for HTTP requests?")
119 BotConfig.register BotConfigStringValue.new('http.proxy_uri', :default => false,
120 :desc => "Proxy server to use for HTTP requests (URI, e.g http://proxy.host:port)")
121 BotConfig.register BotConfigStringValue.new('http.proxy_user',
123 :desc => "User for authenticating with the http proxy (if required)")
124 BotConfig.register BotConfigStringValue.new('http.proxy_pass',
126 :desc => "Password for authenticating with the http proxy (if required)")
127 BotConfig.register BotConfigArrayValue.new('http.proxy_include',
129 :desc => "List of regexps to check against a URI's hostname/ip to see if we should use the proxy to access this URI. All URIs are proxied by default if the proxy is set, so this is only required to re-include URIs that might have been excluded by the exclude list. e.g. exclude /.*\.foo\.com/, include bar\.foo\.com")
130 BotConfig.register BotConfigArrayValue.new('http.proxy_exclude',
132 :desc => "List of regexps to check against a URI's hostname/ip to see if we should use avoid the proxy to access this URI and access it directly")
133 BotConfig.register BotConfigIntegerValue.new('http.max_redir',
135 :desc => "Maximum number of redirections to be used when getting a document")
136 BotConfig.register BotConfigIntegerValue.new('http.expire_time',
138 :desc => "After how many minutes since last use a cached document is considered to be expired")
139 BotConfig.register BotConfigIntegerValue.new('http.max_cache_time',
141 :desc => "After how many minutes since first use a cached document is considered to be expired")
142 BotConfig.register BotConfigIntegerValue.new('http.no_expire_cache',
144 :desc => "Set this to true if you want the bot to never expire the cached pages")
145 BotConfig.register BotConfigIntegerValue.new('http.info_bytes',
147 :desc => "How many bytes to download from a web page to find some information. Set to 0 to let the bot download the whole page.")
150 attr_accessor :response, :last_used, :first_used, :count, :expires, :date
152 def self.maybe_new(resp)
153 debug "maybe new #{resp}"
154 return nil if resp.no_cache
155 return nil unless Net::HTTPOK === resp ||
156 Net::HTTPMovedPermanently === resp ||
157 Net::HTTPFound === resp ||
158 Net::HTTPPartialContent === resp
160 cc = resp['cache-control']
161 return nil if cc && (cc =~ /no-cache/i)
165 date = Time.httpdate(d)
168 return nil if resp['expires'] && (Time.httpdate(resp['expires']) < date)
170 debug "creating cache obj"
177 @first_used = now if @count == 0
183 debug "checking expired?"
184 if cc = self.response['cache-control'] && cc =~ /must-revalidate/
187 return self.expires < Time.now
190 def setup_headers(hdr)
191 hdr['if-modified-since'] = self.date.rfc2822
193 debug "ims == #{hdr['if-modified-since']}"
195 if etag = self.response['etag']
196 hdr['if-none-match'] = etag
197 debug "etag: #{etag}"
201 def revalidate(resp = self.response)
204 self.date = resp.key?('date') ? Time.httpdate(resp['date']) : Time.now
206 cc = resp['cache-control']
207 if cc && (cc =~ /max-age=(\d+)/)
208 self.expires = self.date + $1.to_i
209 elsif resp.key?('expires')
210 self.expires = Time.httpdate(resp['expires'])
211 elsif lm = resp['last-modified']
212 delta = self.date - Time.httpdate(lm)
213 delta = 10 if delta <= 0
215 self.expires = self.date + delta
217 self.expires = self.date + 300
219 # self.expires = Time.now + 10 # DEBUG
220 debug "expires on #{self.expires}"
230 self.response.raw_body
231 rescue Exception => e
233 error e.backtrace.join("\n")
243 'Accept-Charset' => 'utf-8;q=1.0, *;q=0.8',
245 "rbot http util #{$version} (http://linuxbrit.co.uk/rbot/)"
247 debug "starting http cache cleanup timer"
248 @timer = @bot.timer.add(300) {
249 self.remove_stale_cache unless @bot.config['http.no_expire_cache']
254 debug 'stopping http cache cleanup timer'
255 @bot.timer.remove(@timer)
258 # if http_proxy_include or http_proxy_exclude are set, then examine the
259 # uri to see if this is a proxied uri
260 # the in/excludes are a list of regexps, and each regexp is checked against
261 # the server name, and its IP addresses
262 def proxy_required(uri)
264 if @bot.config["http.proxy_exclude"].empty? && @bot.config["http.proxy_include"].empty?
270 list.concat Resolv.getaddresses(uri.host)
271 rescue StandardError => err
272 warning "couldn't resolve host uri.host"
275 unless @bot.config["http.proxy_exclude"].empty?
276 re = @bot.config["http.proxy_exclude"].collect{|r| Regexp.new(r)}
286 unless @bot.config["http.proxy_include"].empty?
287 re = @bot.config["http.proxy_include"].collect{|r| Regexp.new(r)}
297 debug "using proxy for uri #{uri}?: #{use_proxy}"
301 # uri:: Uri to create a proxy for
303 # return a net/http Proxy object, which is configured correctly for
304 # proxying based on the bot's proxy configuration.
305 # This will include per-url proxy configuration based on the bot config
306 # +http_proxy_include/exclude+ options.
308 def get_proxy(uri, options = {})
320 if @bot.config["http.use_proxy"]
321 if (ENV['http_proxy'])
322 proxy = URI.parse ENV['http_proxy'] rescue nil
324 if (@bot.config["http.proxy_uri"])
325 proxy = URI.parse @bot.config["http.proxy_uri"] rescue nil
328 debug "proxy is set to #{proxy.host} port #{proxy.port}"
329 if proxy_required(uri)
330 proxy_host = proxy.host
331 proxy_port = proxy.port
332 proxy_user = @bot.config["http.proxy_user"]
333 proxy_pass = @bot.config["http.proxy_pass"]
338 h = Net::HTTP.new(uri.host, uri.port, proxy_host, proxy_port, proxy_user, proxy_port)
339 h.use_ssl = true if uri.scheme == "https"
341 h.read_timeout = opts[:read_timeout]
342 h.open_timeout = opts[:open_timeout]
346 def handle_response(uri, resp, opts, &block)
347 if Net::HTTPRedirection === resp && opts[:max_redir] >= 0
348 if resp.key?('location')
349 raise 'Too many redirections' if opts[:max_redir] <= 0
350 yield resp if opts[:yield] == :all && block_given?
351 loc = resp['location']
352 new_loc = URI.join(uri.to_s, loc) rescue URI.parse(loc)
354 new_opts[:max_redir] -= 1
355 case opts[:method].to_s.downcase.intern
356 when :post, :"net::http::post"
357 new_opts[:method] = :get
359 debug "following the redirect to #{new_loc}"
360 return get_response(new_loc, new_opts, &block)
362 warning ":| redirect w/o location?"
368 # Net::HTTP wants us to read the whole body here
374 # uri:: uri to query (Uri object or String)
375 # opts:: options. Currently used:
376 # :method:: request method [:get (default), :post or :head]
377 # :open_timeout:: open timeout for the proxy
378 # :read_timeout:: read timeout for the proxy
379 # :cache:: should we cache results?
380 # :yield:: if :final [default], call &block for the response object
381 # if :all, call &block for all intermediate redirects, too
382 # :max_redir:: how many redirects to follow before raising the exception
383 # if -1, don't follow redirects, just return them
384 # :range:: make a ranged request (usually GET). accepts a string
385 # for HTTP/1.1 "Range:" header (i.e. "bytes=0-1000")
386 # :body:: request body (usually for POST requests)
388 # Generic http transaction method
390 # It will return a HTTP::Response object or raise an exception
392 # If a block is given, it will yield the response (see :yield option)
394 def get_response(uri_or_s, options = {}, &block)
395 uri = uri_or_s.kind_of?(URI) ? uri_or_s : URI.parse(uri_or_s.to_s)
397 :max_redir => @bot.config['http.max_redir'],
406 req_class = case opts[:method].to_s.downcase.intern
407 when :head, :"net::http::head"
408 opts[:max_redir] = -1
410 when :get, :"net::http::get"
412 when :post, :"net::http::post"
414 opts[:body] or raise 'post request w/o a body?'
415 warning "refusing to cache POST request" if options[:cache]
418 warning "unsupported method #{opts[:method]}, doing GET"
422 if req_class != Net::HTTP::Get && opts[:range]
423 warning "can't request ranges for #{req_class}"
427 cache_key = "#{opts[:range]}|#{req_class}|#{uri.to_s}"
429 if req_class != Net::HTTP::Get && req_class != Net::HTTP::Head
431 warning "can't cache #{req_class.inspect} requests, working w/o cache"
436 debug "get_response(#{uri}, #{opts.inspect})"
438 if opts[:cache] && cached = @cache[cache_key]
443 return handle_response(uri, cached.response, opts, &block)
447 headers = @headers.dup.merge(opts[:headers] || {})
448 headers['Range'] = opts[:range] if opts[:range]
450 cached.setup_headers(headers) if cached && (req_class == Net::HTTP::Get)
451 req = req_class.new(uri.request_uri, headers)
452 req.basic_auth(uri.user, uri.password) if uri.user && uri.password
453 req.body = opts[:body] if req_class == Net::HTTP::Post
454 debug "prepared request: #{req.to_hash.inspect}"
456 get_proxy(uri, opts).start do |http|
457 http.request(req) do |resp|
458 resp['x-rbot-location'] = uri.to_s
459 if Net::HTTPNotModified === resp
462 cached.revalidate(resp)
463 rescue Exception => e
465 error e.backtrace.join("\n")
467 debug "reusing cached"
468 resp = cached.response
469 elsif Net::HTTPServerError === resp || Net::HTTPClientError === resp
470 debug "http error, deleting cached obj" if cached
471 @cache.delete(cache_key)
474 return handle_response(uri, resp, opts, &block)
476 if cached = CachedObject.maybe_new(resp) rescue nil
477 debug "storing to cache"
478 @cache[cache_key] = cached
483 return handle_response(uri, resp, opts, &block)
488 # uri:: uri to query (Uri object)
490 # simple get request, returns (if possible) response body following redirs
491 # and caching if requested
492 def get(uri, opts = {}, &block)
494 resp = get_response(uri, opts, &block)
495 raise "http error: #{resp}" unless Net::HTTPOK === resp ||
496 Net::HTTPPartialContent === resp
498 rescue Exception => e
500 error e.backtrace.join("\n")
505 def head(uri, options = {}, &block)
506 opts = {:method => :head}.merge(options)
508 resp = get_response(uri, opts, &block)
509 raise "http error #{resp}" if Net::HTTPClientError === resp ||
510 Net::HTTPServerError == resp
512 rescue Exception => e
514 error e.backtrace.join("\n")
519 def post(uri, data, options = {}, &block)
520 opts = {:method => :post, :body => data, :cache => false}.merge(options)
522 resp = get_response(uri, opts, &block)
523 raise 'http error' unless Net::HTTPOK === resp
525 rescue Exception => e
527 error e.backtrace.join("\n")
532 def get_partial(uri, nbytes = @bot.config['http.info_bytes'], options = {}, &block)
533 opts = {:range => "bytes=0-#{nbytes}"}.merge(options)
534 return get(uri, opts, &block)
537 def remove_stale_cache
538 debug "Removing stale cache"
540 max_last = @bot.config['http.expire_time'] * 60
541 max_first = @bot.config['http.max_cache_time'] * 60
542 debug "#{@cache.size} pages before"
544 @cache.reject! { |k, val|
545 (now - val.last_used > max_last) || (now - val.first_used > max_first)
548 error "Failed to remove stale cache: #{e.inspect}"
550 debug "#{@cache.size} pages after"
557 class HttpUtilPlugin < CoreBotModule
560 debug 'initializing httputil'
561 @bot.httputil = Irc::Utils::HttpUtil.new(@bot)
565 debug 'shutting down httputil'
566 @bot.httputil.cleanup