pithyless / char_converter.rb

Created September 05, 2012


    
    View char_converter.rb 
  
  
      char_converter.rb
    
    
        Raw
      
    
    File suppressed. Click to show.
  
  
            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
          
          
            # config/initializers/char_converter.rb
require 'uri'
 
module Support
  class CharConverter
 
    SANITIZE_ENV_KEYS = [
      "HTTP_COOKIE",     # bad cookie encodings kill rack: https://github.com/rack/rack/issues/225
      "HTTP_REFERER",
      "PATH_INFO",
      "QUERY_STRING",
      "REQUEST_PATH",
      "REQUEST_URI",
    ]
 
    def initialize(app)
      @app = app
    end
 
    def call(env)
      @app.call(sanitize_env(env))
    end
 
    def sanitize_env(env)
      SANITIZE_ENV_KEYS.each do |key|
        next unless value = env[key]
        value = sanitize_string(URI.decode(value))
        env[key] = URI.encode(value)
      end
      env
    end
 
    def sanitize_string(string)
      return string unless string.is_a? String
 
      # Try it as UTF-8 directly
      cleaned = string.dup.force_encoding('UTF-8')
      if cleaned.valid_encoding?
        cleaned
      else
        # Some of it might be old Windows code page
        string.encode(Encoding::UTF_8, Encoding::Windows_1250)
      end
    rescue EncodingError
      # Force it to UTF-8, throwing out invalid bits
      string.encode!('UTF-8', invalid: :replace, undef: :replace)
    end
 
  end
end
 
Rails.application.config.middleware.insert_before(0, Support::CharConverter)