lib/rbot/core/utils/httputil.rb

   1 #-- vim:sw=2:et
   2 #++
   3 #
   4 # :title: rbot HTTP provider
   5 #
   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>
   9 #
  10 # Copyright:: (C) 2002-2005 Tom Gilbert
  11 # Copyright:: (C) 2006 Tom Gilbert, Giuseppe Bilotta
  12 # Copyright:: (C) 2007 Giuseppe Bilotta, Dmitry Kim
  13
  14 require 'resolv'
  15 require 'net/http'
  16 require 'cgi'
  17 require 'iconv'
  18 begin
  19   require 'net/https'
  20 rescue LoadError => e
  21   error "Couldn't load 'net/https':  #{e.pretty_inspect}"
  22   error "Secured HTTP connections will fail"
  23 end
  24
  25 # To handle Gzipped pages
  26 require 'stringio'
  27 require 'zlib'
  28
  29 module ::Net
  30   class HTTPResponse
  31     attr_accessor :no_cache
  32     if !instance_methods.include?('raw_body')
  33       alias :raw_body :body
  34     end
  35
  36     def body_charset(str=self.raw_body)
  37       ctype = self['content-type'] || 'text/html'
  38       return nil unless ctype =~ /^text/i || ctype =~ /x(ht)?ml/i
  39
  40       charsets = ['latin1'] # should be in config
  41
  42       if self['content-type'].match(/charset=["']?([^\s"']+)["']?/i)
  43         charsets << $1
  44         debug "charset #{charsets.last} added from header"
  45       end
  46
  47       case str
  48       when /<\?xml\s[^>]*encoding=['"]([^\s"'>]+)["'][^>]*\?>/i
  49         charsets << $1
  50         debug "xml charset #{charsets.last} added from xml pi"
  51       when /<(meta\s[^>]*http-equiv=["']?Content-Type["']?[^>]*)>/i
  52         meta = $1
  53         if meta =~ /charset=['"]?([^\s'";]+)['"]?/
  54           charsets << $1
  55           debug "html charset #{charsets.last} added from meta"
  56         end
  57       end
  58       return charsets.uniq
  59     end
  60
  61     def body_to_utf(str)
  62       charsets = self.body_charset(str) or return str
  63
  64       charsets.reverse_each do |charset|
  65         # XXX: this one is really ugly, but i don't know how to make it better
  66         #  -jsn
  67
  68         0.upto(5) do |off|
  69           begin
  70             debug "trying #{charset} / offset #{off}"
  71             return Iconv.iconv('utf-8//ignore',
  72                                charset,
  73                                str.slice(0 .. (-1 - off))).first
  74           rescue
  75             debug "conversion failed for #{charset} / offset #{off}"
  76           end
  77         end
  78       end
  79       return str
  80     end
  81
  82     def decompress_body(str)
  83       method = self['content-encoding']
  84       case method
  85       when nil
  86         return str
  87       when /gzip/ # Matches gzip, x-gzip, and the non-rfc-compliant gzip;q=\d sent by some servers
  88         debug "gunzipping body"
  89         begin
  90           return Zlib::GzipReader.new(StringIO.new(str)).read
  91         rescue Zlib::Error => e
  92           # If we can't unpack the whole stream (e.g. because we're doing a
  93           # partial read
  94           debug "full gunzipping failed (#{e}), trying to recover as much as possible"
  95           ret = ""
  96           begin
  97             Zlib::GzipReader.new(StringIO.new(str)).each_byte { |byte|
  98               ret << byte
  99             }
 100           rescue
 101           end
 102           return ret
 103         end
 104       else
 105         raise "Unhandled content encoding #{method}"
 106       end
 107     end
 108
 109     def cooked_body
 110       return self.body_to_utf(self.decompress_body(self.raw_body))
 111     end
 112
 113     # Read chunks from the body until we have at least _size_ bytes, yielding
 114     # the partial text at each chunk. Return the partial body.
 115     def partial_body(size=0, &block)
 116
 117       self.no_cache = true
 118       partial = String.new
 119
 120       self.read_body { |chunk|
 121         partial << chunk
 122         yield self.body_to_utf(self.decompress_body(partial)) if block_given?
 123         break if size and size > 0 and partial.length >= size
 124       }
 125
 126       return self.body_to_utf(self.decompress_body(partial))
 127     end
 128   end
 129 end
 130
 131 Net::HTTP.version_1_2
 132
 133 module ::Irc
 134 module Utils
 135
 136 # class for making http requests easier (mainly for plugins to use)
 137 # this class can check the bot proxy configuration to determine if a proxy
 138 # needs to be used, which includes support for per-url proxy configuration.
 139 class HttpUtil
 140     Bot::Config.register Bot::Config::BooleanValue.new('http.use_proxy',
 141       :default => false, :desc => "should a proxy be used for HTTP requests?")
 142     Bot::Config.register Bot::Config::StringValue.new('http.proxy_uri', :default => false,
 143       :desc => "Proxy server to use for HTTP requests (URI, e.g http://proxy.host:port)")
 144     Bot::Config.register Bot::Config::StringValue.new('http.proxy_user',
 145       :default => nil,
 146       :desc => "User for authenticating with the http proxy (if required)")
 147     Bot::Config.register Bot::Config::StringValue.new('http.proxy_pass',
 148       :default => nil,
 149       :desc => "Password for authenticating with the http proxy (if required)")
 150     Bot::Config.register Bot::Config::ArrayValue.new('http.proxy_include',
 151       :default => [],
 152       :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")
 153     Bot::Config.register Bot::Config::ArrayValue.new('http.proxy_exclude',
 154       :default => [],
 155       :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")
 156     Bot::Config.register Bot::Config::IntegerValue.new('http.max_redir',
 157       :default => 5,
 158       :desc => "Maximum number of redirections to be used when getting a document")
 159     Bot::Config.register Bot::Config::IntegerValue.new('http.expire_time',
 160       :default => 60,
 161       :desc => "After how many minutes since last use a cached document is considered to be expired")
 162     Bot::Config.register Bot::Config::IntegerValue.new('http.max_cache_time',
 163       :default => 60*24,
 164       :desc => "After how many minutes since first use a cached document is considered to be expired")
 165     Bot::Config.register Bot::Config::IntegerValue.new('http.no_expire_cache',
 166       :default => false,
 167       :desc => "Set this to true if you want the bot to never expire the cached pages")
 168     Bot::Config.register Bot::Config::IntegerValue.new('http.info_bytes',
 169       :default => 8192,
 170       :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.")
 171
 172   class CachedObject
 173     attr_accessor :response, :last_used, :first_used, :count, :expires, :date
 174
 175     def self.maybe_new(resp)
 176       debug "maybe new #{resp}"
 177       return nil if resp.no_cache
 178       return nil unless Net::HTTPOK === resp ||
 179       Net::HTTPMovedPermanently === resp ||
 180       Net::HTTPFound === resp ||
 181       Net::HTTPPartialContent === resp
 182
 183       cc = resp['cache-control']
 184       return nil if cc && (cc =~ /no-cache/i)
 185
 186       date = Time.now
 187       if d = resp['date']
 188         date = Time.httpdate(d)
 189       end
 190
 191       return nil if resp['expires'] && (Time.httpdate(resp['expires']) < date)
 192
 193       debug "creating cache obj"
 194
 195       self.new(resp)
 196     end
 197
 198     def use
 199       now = Time.now
 200       @first_used = now if @count == 0
 201       @last_used = now
 202       @count += 1
 203     end
 204
 205     def expired?
 206       debug "checking expired?"
 207       if cc = self.response['cache-control'] && cc =~ /must-revalidate/
 208         return true
 209       end
 210       return self.expires < Time.now
 211     end
 212
 213     def setup_headers(hdr)
 214       hdr['if-modified-since'] = self.date.rfc2822
 215
 216       debug "ims == #{hdr['if-modified-since']}"
 217
 218       if etag = self.response['etag']
 219         hdr['if-none-match'] = etag
 220         debug "etag: #{etag}"
 221       end
 222     end
 223
 224     def revalidate(resp = self.response)
 225       @count = 0
 226       self.use
 227       self.date = resp.key?('date') ? Time.httpdate(resp['date']) : Time.now
 228
 229       cc = resp['cache-control']
 230       if cc && (cc =~ /max-age=(\d+)/)
 231         self.expires = self.date + $1.to_i
 232       elsif resp.key?('expires')
 233         self.expires = Time.httpdate(resp['expires'])
 234       elsif lm = resp['last-modified']
 235         delta = self.date - Time.httpdate(lm)
 236         delta = 10 if delta <= 0
 237         delta /= 5
 238         self.expires = self.date + delta
 239       else
 240         self.expires = self.date + 300
 241       end
 242       # self.expires = Time.now + 10 # DEBUG
 243       debug "expires on #{self.expires}"
 244
 245       return true
 246     end
 247
 248     private
 249     def initialize(resp)
 250       @response = resp
 251       begin
 252         self.revalidate
 253         self.response.raw_body
 254       rescue Exception => e
 255         error e
 256         raise e
 257       end
 258     end
 259   end
 260
 261   def initialize(bot)
 262     @bot = bot
 263     @cache = Hash.new
 264     @headers = {
 265       'Accept-Charset' => 'utf-8;q=1.0, *;q=0.8',
 266       'Accept-Encoding' => 'gzip;q=1, identity;q=0.8, *;q=0.2',
 267       'User-Agent' =>
 268         "rbot http util #{$version} (http://linuxbrit.co.uk/rbot/)"
 269     }
 270     debug "starting http cache cleanup timer"
 271     @timer = @bot.timer.add(300) {
 272       self.remove_stale_cache unless @bot.config['http.no_expire_cache']
 273     }
 274   end
 275
 276   def cleanup
 277     debug 'stopping http cache cleanup timer'
 278     @bot.timer.remove(@timer)
 279   end
 280
 281   # if http_proxy_include or http_proxy_exclude are set, then examine the
 282   # uri to see if this is a proxied uri
 283   # the in/excludes are a list of regexps, and each regexp is checked against
 284   # the server name, and its IP addresses
 285   def proxy_required(uri)
 286     use_proxy = true
 287     if @bot.config["http.proxy_exclude"].empty? && @bot.config["http.proxy_include"].empty?
 288       return use_proxy
 289     end
 290
 291     list = [uri.host]
 292     begin
 293       list.concat Resolv.getaddresses(uri.host)
 294     rescue StandardError => err
 295       warning "couldn't resolve host uri.host"
 296     end
 297
 298     unless @bot.config["http.proxy_exclude"].empty?
 299       re = @bot.config["http.proxy_exclude"].collect{|r| Regexp.new(r)}
 300       re.each do |r|
 301         list.each do |item|
 302           if r.match(item)
 303             use_proxy = false
 304             break
 305           end
 306         end
 307       end
 308     end
 309     unless @bot.config["http.proxy_include"].empty?
 310       re = @bot.config["http.proxy_include"].collect{|r| Regexp.new(r)}
 311       re.each do |r|
 312         list.each do |item|
 313           if r.match(item)
 314             use_proxy = true
 315             break
 316           end
 317         end
 318       end
 319     end
 320     debug "using proxy for uri #{uri}?: #{use_proxy}"
 321     return use_proxy
 322   end
 323
 324   # uri:: Uri to create a proxy for
 325   #
 326   # return a net/http Proxy object, which is configured correctly for
 327   # proxying based on the bot's proxy configuration.
 328   # This will include per-url proxy configuration based on the bot config
 329   # +http_proxy_include/exclude+ options.
 330
 331   def get_proxy(uri, options = {})
 332     opts = {
 333       :read_timeout => 10,
 334       :open_timeout => 20
 335     }.merge(options)
 336
 337     proxy = nil
 338     proxy_host = nil
 339     proxy_port = nil
 340     proxy_user = nil
 341     proxy_pass = nil
 342
 343     if @bot.config["http.use_proxy"]
 344       if (ENV['http_proxy'])
 345         proxy = URI.parse ENV['http_proxy'] rescue nil
 346       end
 347       if (@bot.config["http.proxy_uri"])
 348         proxy = URI.parse @bot.config["http.proxy_uri"] rescue nil
 349       end
 350       if proxy
 351         debug "proxy is set to #{proxy.host} port #{proxy.port}"
 352         if proxy_required(uri)
 353           proxy_host = proxy.host
 354           proxy_port = proxy.port
 355           proxy_user = @bot.config["http.proxy_user"]
 356           proxy_pass = @bot.config["http.proxy_pass"]
 357         end
 358       end
 359     end
 360
 361     h = Net::HTTP.new(uri.host, uri.port, proxy_host, proxy_port, proxy_user, proxy_port)
 362     h.use_ssl = true if uri.scheme == "https"
 363
 364     h.read_timeout = opts[:read_timeout]
 365     h.open_timeout = opts[:open_timeout]
 366     return h
 367   end
 368
 369   def handle_response(uri, resp, opts, &block)
 370     if Net::HTTPRedirection === resp && opts[:max_redir] >= 0
 371       if resp.key?('location')
 372         raise 'Too many redirections' if opts[:max_redir] <= 0
 373         yield resp if opts[:yield] == :all && block_given?
 374         loc = resp['location']
 375         new_loc = URI.join(uri.to_s, loc) rescue URI.parse(loc)
 376         new_opts = opts.dup
 377         new_opts[:max_redir] -= 1
 378         case opts[:method].to_s.downcase.intern
 379         when :post, :"net::http::post"
 380           new_opts[:method] = :get
 381         end
 382         debug "following the redirect to #{new_loc}"
 383         return get_response(new_loc, new_opts, &block)
 384       else
 385         warning ":| redirect w/o location?"
 386       end
 387     end
 388     class << resp
 389       undef_method :body
 390       alias :body :cooked_body
 391     end
 392     if block_given?
 393       yield(resp)
 394     else
 395       # Net::HTTP wants us to read the whole body here
 396       resp.raw_body
 397     end
 398     return resp
 399   end
 400
 401   # uri::         uri to query (Uri object or String)
 402   # opts::        options. Currently used:
 403   # :method::     request method [:get (default), :post or :head]
 404   # :open_timeout::     open timeout for the proxy
 405   # :read_timeout::     read timeout for the proxy
 406   # :cache::            should we cache results?
 407   # :yield::      if :final [default], call &block for the response object
 408   #               if :all, call &block for all intermediate redirects, too
 409   # :max_redir::  how many redirects to follow before raising the exception
 410   #               if -1, don't follow redirects, just return them
 411   # :range::      make a ranged request (usually GET). accepts a string
 412   #               for HTTP/1.1 "Range:" header (i.e. "bytes=0-1000")
 413   # :body::       request body (usually for POST requests)
 414   #
 415   # Generic http transaction method
 416   #
 417   # It will return a Net::HTTPResponse object or raise an exception
 418   #
 419   # If a block is given, it will yield the response (see :yield option)
 420
 421   def get_response(uri_or_s, options = {}, &block)
 422     uri = uri_or_s.kind_of?(URI) ? uri_or_s : URI.parse(uri_or_s.to_s)
 423     opts = {
 424       :max_redir => @bot.config['http.max_redir'],
 425       :yield => :final,
 426       :cache => true,
 427       :method => :GET
 428     }.merge(options)
 429
 430     resp = nil
 431     cached = nil
 432
 433     req_class = case opts[:method].to_s.downcase.intern
 434                 when :head, :"net::http::head"
 435                   opts[:max_redir] = -1
 436                   Net::HTTP::Head
 437                 when :get, :"net::http::get"
 438                   Net::HTTP::Get
 439                 when :post, :"net::http::post"
 440                   opts[:cache] = false
 441                   opts[:body] or raise 'post request w/o a body?'
 442                   warning "refusing to cache POST request" if options[:cache]
 443                   Net::HTTP::Post
 444                 else
 445                   warning "unsupported method #{opts[:method]}, doing GET"
 446                   Net::HTTP::Get
 447                 end
 448
 449     if req_class != Net::HTTP::Get && opts[:range]
 450       warning "can't request ranges for #{req_class}"
 451       opts.delete(:range)
 452     end
 453
 454     cache_key = "#{opts[:range]}|#{req_class}|#{uri.to_s}"
 455
 456     if req_class != Net::HTTP::Get && req_class != Net::HTTP::Head
 457       if opts[:cache]
 458         warning "can't cache #{req_class.inspect} requests, working w/o cache"
 459         opts[:cache] = false
 460       end
 461     end
 462
 463     debug "get_response(#{uri}, #{opts.inspect})"
 464
 465     if opts[:cache] && cached = @cache[cache_key]
 466       debug "got cached"
 467       if !cached.expired?
 468         debug "using cached"
 469         cached.use
 470         return handle_response(uri, cached.response, opts, &block)
 471       end
 472     end
 473
 474     headers = @headers.dup.merge(opts[:headers] || {})
 475     headers['Range'] = opts[:range] if opts[:range]
 476     headers['Authorization'] = opts[:auth_head] if opts[:auth_head]
 477
 478     cached.setup_headers(headers) if cached && (req_class == Net::HTTP::Get)
 479     req = req_class.new(uri.request_uri, headers)
 480     if uri.user && uri.password
 481       req.basic_auth(uri.user, uri.password)
 482       opts[:auth_head] = req['Authorization']
 483     end
 484     req.body = opts[:body] if req_class == Net::HTTP::Post
 485     debug "prepared request: #{req.to_hash.inspect}"
 486
 487     get_proxy(uri, opts).start do |http|
 488       http.request(req) do |resp|
 489         resp['x-rbot-location'] = uri.to_s
 490         if Net::HTTPNotModified === resp
 491           debug "not modified"
 492           begin
 493             cached.revalidate(resp)
 494           rescue Exception => e
 495             error e
 496           end
 497           debug "reusing cached"
 498           resp = cached.response
 499         elsif Net::HTTPServerError === resp || Net::HTTPClientError === resp
 500           debug "http error, deleting cached obj" if cached
 501           @cache.delete(cache_key)
 502         elsif opts[:cache]
 503           begin
 504             return handle_response(uri, resp, opts, &block)
 505           ensure
 506             if cached = CachedObject.maybe_new(resp) rescue nil
 507               debug "storing to cache"
 508               @cache[cache_key] = cached
 509             end
 510           end
 511           return ret
 512         end
 513         return handle_response(uri, resp, opts, &block)
 514       end
 515     end
 516   end
 517
 518   # uri::         uri to query (Uri object)
 519   #
 520   # simple get request, returns (if possible) response body following redirs
 521   # and caching if requested
 522   def get(uri, opts = {}, &block)
 523     begin
 524       resp = get_response(uri, opts, &block)
 525       raise "http error: #{resp}" unless Net::HTTPOK === resp ||
 526         Net::HTTPPartialContent === resp
 527       return resp.body
 528     rescue Exception => e
 529       error e
 530     end
 531     return nil
 532   end
 533
 534   def head(uri, options = {}, &block)
 535     opts = {:method => :head}.merge(options)
 536     begin
 537       resp = get_response(uri, opts, &block)
 538       raise "http error #{resp}" if Net::HTTPClientError === resp ||
 539         Net::HTTPServerError == resp
 540       return resp
 541     rescue Exception => e
 542       error e
 543     end
 544     return nil
 545   end
 546
 547   def post(uri, data, options = {}, &block)
 548     opts = {:method => :post, :body => data, :cache => false}.merge(options)
 549     begin
 550       resp = get_response(uri, opts, &block)
 551       raise 'http error' unless Net::HTTPOK === resp
 552       return resp
 553     rescue Exception => e
 554       error e
 555     end
 556     return nil
 557   end
 558
 559   def get_partial(uri, nbytes = @bot.config['http.info_bytes'], options = {}, &block)
 560     opts = {:range => "bytes=0-#{nbytes}"}.merge(options)
 561     return get(uri, opts, &block)
 562   end
 563
 564   def remove_stale_cache
 565     debug "Removing stale cache"
 566     now = Time.new
 567     max_last = @bot.config['http.expire_time'] * 60
 568     max_first = @bot.config['http.max_cache_time'] * 60
 569     debug "#{@cache.size} pages before"
 570     begin
 571       @cache.reject! { |k, val|
 572         (now - val.last_used > max_last) || (now - val.first_used > max_first)
 573       }
 574     rescue => e
 575       error "Failed to remove stale cache: #{e.pretty_inspect}"
 576     end
 577     debug "#{@cache.size} pages after"
 578   end
 579
 580 end
 581 end
 582 end
 583
 584 class HttpUtilPlugin < CoreBotModule
 585   def initialize(*a)
 586     super(*a)
 587     debug 'initializing httputil'
 588     @bot.httputil = Irc::Utils::HttpUtil.new(@bot)
 589   end
 590
 591   def cleanup
 592     debug 'shutting down httputil'
 593     @bot.httputil.cleanup
 594     @bot.httputil = nil
 595     super
 596   end
 597 end
 598
 599 HttpUtilPlugin.new