diff options
author | Ahmad Sherif <me@ahmadsherif.com> | 2017-06-22 01:51:46 +0200 |
---|---|---|
committer | Ahmad Sherif <me@ahmadsherif.com> | 2017-06-22 03:12:23 +0200 |
commit | 5b092d21cca71dde8f032dfcb9b7b41612a8727f (patch) | |
tree | 5a7873b2d04643a161835f824232c38ba18ee7cf /lib | |
parent | db9a8f4e3257104937d4558cd25ccbd345914a73 (diff) | |
download | gitlab-ce-5b092d21cca71dde8f032dfcb9b7b41612a8727f.tar.gz |
Encode Gitaly diff patches properlyfix/properly-encode-gitaly-diffs
Diffstat (limited to 'lib')
-rw-r--r-- | lib/gitlab/git/diff.rb | 2 | ||||
-rw-r--r-- | lib/gitlab/gitaly_client/diff_stitcher.rb | 5 |
2 files changed, 5 insertions, 2 deletions
diff --git a/lib/gitlab/git/diff.rb b/lib/gitlab/git/diff.rb index f825568f194..cf7829a583b 100644 --- a/lib/gitlab/git/diff.rb +++ b/lib/gitlab/git/diff.rb @@ -318,7 +318,7 @@ module Gitlab end def init_from_gitaly(diff) - @diff = diff.patch if diff.respond_to?(:patch) + @diff = encode!(diff.patch) if diff.respond_to?(:patch) @new_path = encode!(diff.to_path.dup) @old_path = encode!(diff.from_path.dup) @a_mode = diff.old_mode.to_s(8) diff --git a/lib/gitlab/gitaly_client/diff_stitcher.rb b/lib/gitlab/gitaly_client/diff_stitcher.rb index d84e8d752dc..65d81dc5d46 100644 --- a/lib/gitlab/gitaly_client/diff_stitcher.rb +++ b/lib/gitlab/gitaly_client/diff_stitcher.rb @@ -13,7 +13,10 @@ module Gitlab @rpc_response.each do |diff_msg| if current_diff.nil? diff_params = diff_msg.to_h.slice(*GitalyClient::Diff::FIELDS) - diff_params[:patch] = diff_msg.raw_patch_data + # gRPC uses frozen strings by default, and we need to have an unfrozen string as it + # gets processed further down the line. So we unfreeze the first chunk of the patch + # in case it's the only chunk we receive for this diff. + diff_params[:patch] = diff_msg.raw_patch_data.dup current_diff = GitalyClient::Diff.new(diff_params) else |