diff options
author | GitLab Bot <gitlab-bot@gitlab.com> | 2021-10-27 21:13:16 +0300 |
---|---|---|
committer | GitLab Bot <gitlab-bot@gitlab.com> | 2021-10-27 21:13:16 +0300 |
commit | 4c5b7f992f2ee3134409559e35350748425fb467 (patch) | |
tree | edabd5b6b177a0f982d27ecddd45de9dccbe6186 /lib/gitlab/middleware | |
parent | f50cb4f63fae6f02a0b8d9c3a95f29b3b4f79516 (diff) |
Add latest changes from gitlab-org/gitlab@master
Diffstat (limited to 'lib/gitlab/middleware')
-rw-r--r-- | lib/gitlab/middleware/compressed_json.rb | 49 |
1 files changed, 49 insertions, 0 deletions
diff --git a/lib/gitlab/middleware/compressed_json.rb b/lib/gitlab/middleware/compressed_json.rb new file mode 100644 index 00000000000..e2dabca69a8 --- /dev/null +++ b/lib/gitlab/middleware/compressed_json.rb @@ -0,0 +1,49 @@ +# frozen_string_literal: true + +module Gitlab + module Middleware + class CompressedJson + COLLECTOR_PATH = '/api/v4/error_tracking/collector' + MAXIMUM_BODY_SIZE = 200.kilobytes.to_i + + def initialize(app) + @app = app + end + + def call(env) + if compressed_et_request?(env) + input = extract(env['rack.input']) + + if input.length > MAXIMUM_BODY_SIZE + return too_large + end + + env.delete('HTTP_CONTENT_ENCODING') + env['CONTENT_LENGTH'] = input.length + env['rack.input'] = StringIO.new(input) + end + + @app.call(env) + end + + def compressed_et_request?(env) + env['REQUEST_METHOD'] == 'POST' && + env['HTTP_CONTENT_ENCODING'] == 'gzip' && + env['CONTENT_TYPE'] == 'application/json' && + env['PATH_INFO'].start_with?((File.join(relative_url, COLLECTOR_PATH))) + end + + def too_large + [413, { 'Content-Type' => 'text/plain' }, ['Payload Too Large']] + end + + def relative_url + File.join('', Gitlab.config.gitlab.relative_url_root).chomp('/') + end + + def extract(input) + Zlib::GzipReader.new(input).read(MAXIMUM_BODY_SIZE + 1) + end + end + end +end |