diff options
author | Michael Kozono <mkozono@gmail.com> | 2018-08-14 21:20:21 +0300 |
---|---|---|
committer | Jose Vargas <jvargas@gitlab.com> | 2018-08-29 00:36:25 +0300 |
commit | b44ccb283f68c10e33ac6c51b695fd0afd70642a (patch) | |
tree | 0dc809228a2800d6d7e0051c56d49811e3e14149 /lib | |
parent | 4ef1447c4f54af7a609c2a76445704995a29612e (diff) |
Resolve "Orphaned upload files are accessible via project exports"
Diffstat (limited to 'lib')
-rw-r--r-- | lib/gitlab/import_export/uploads_manager.rb | 19 |
1 files changed, 11 insertions, 8 deletions
diff --git a/lib/gitlab/import_export/uploads_manager.rb b/lib/gitlab/import_export/uploads_manager.rb index 07875ebb56a..e0d4235e65b 100644 --- a/lib/gitlab/import_export/uploads_manager.rb +++ b/lib/gitlab/import_export/uploads_manager.rb @@ -13,13 +13,11 @@ module Gitlab end def save - copy_files(@from, uploads_export_path) if File.directory?(@from) - if File.file?(@from) && @relative_export_path == 'avatar' copy_files(@from, File.join(uploads_export_path, @project.avatar.filename)) end - copy_from_object_storage + copy_project_uploads true rescue => e @@ -48,14 +46,19 @@ module Gitlab UploadService.new(@project, File.open(upload, 'r'), FileUploader, uploader_context).execute end - def copy_from_object_storage - return unless Gitlab::ImportExport.object_storage? - + def copy_project_uploads each_uploader do |uploader| next unless uploader.file - next if uploader.upload.local? # Already copied, using the old method - download_and_copy(uploader) + if uploader.upload.local? + next unless uploader.upload.exist? + + copy_files(uploader.absolute_path, File.join(uploads_export_path, uploader.upload.path)) + else + next unless Gitlab::ImportExport.object_storage? + + download_and_copy(uploader) + end end end |