422 lines
		
	
	
		
			12 KiB
		
	
	
	
		
			Ruby
		
	
	
	
			
		
		
	
	
			422 lines
		
	
	
		
			12 KiB
		
	
	
	
		
			Ruby
		
	
	
	
| # frozen_string_literal: true
 | |
| 
 | |
| # Check a user's access to perform a git action. All public methods in this
 | |
| # class return an instance of `GitlabAccessStatus`
 | |
| module Gitlab
 | |
|   class GitAccess
 | |
|     include Gitlab::Utils::StrongMemoize
 | |
| 
 | |
|     ForbiddenError = Class.new(StandardError)
 | |
|     NotFoundError = Class.new(StandardError)
 | |
|     ProjectCreationError = Class.new(StandardError)
 | |
|     TimeoutError = Class.new(StandardError)
 | |
|     ProjectMovedError = Class.new(NotFoundError)
 | |
| 
 | |
|     # Use the magic string '_any' to indicate we do not know what the
 | |
|     # changes are. This is also what gitlab-shell does.
 | |
|     ANY = '_any'
 | |
| 
 | |
|     ERROR_MESSAGES = {
 | |
|       upload: 'You are not allowed to upload code for this project.',
 | |
|       download: 'You are not allowed to download code from this project.',
 | |
|       auth_upload: 'You are not allowed to upload code.',
 | |
|       auth_download: 'You are not allowed to download code.',
 | |
|       deploy_key_upload: 'This deploy key does not have write access to this project.',
 | |
|       no_repo: 'A repository for this project does not exist yet.',
 | |
|       project_not_found: 'The project you were looking for could not be found.',
 | |
|       command_not_allowed: "The command you're trying to execute is not allowed.",
 | |
|       upload_pack_disabled_over_http: 'Pulling over HTTP is not allowed.',
 | |
|       receive_pack_disabled_over_http: 'Pushing over HTTP is not allowed.',
 | |
|       read_only: 'The repository is temporarily read-only. Please try again later.',
 | |
|       cannot_push_to_read_only: "You can't push code to a read-only GitLab instance.",
 | |
|       push_code: 'You are not allowed to push code to this project.'
 | |
|     }.freeze
 | |
| 
 | |
|     INTERNAL_TIMEOUT = 50.seconds.freeze
 | |
|     LOG_HEADER = <<~MESSAGE
 | |
|       Push operation timed out
 | |
| 
 | |
|       Timing information for debugging purposes:
 | |
|     MESSAGE
 | |
| 
 | |
|     DOWNLOAD_COMMANDS = %w{git-upload-pack git-upload-archive}.freeze
 | |
|     PUSH_COMMANDS = %w{git-receive-pack}.freeze
 | |
|     ALL_COMMANDS = DOWNLOAD_COMMANDS + PUSH_COMMANDS
 | |
| 
 | |
|     attr_reader :actor, :project, :protocol, :authentication_abilities, :namespace_path, :project_path, :redirected_path, :auth_result_type, :changes, :logger
 | |
| 
 | |
|     def initialize(actor, project, protocol, authentication_abilities:, namespace_path: nil, project_path: nil, redirected_path: nil, auth_result_type: nil)
 | |
|       @actor    = actor
 | |
|       @project  = project
 | |
|       @protocol = protocol
 | |
|       @authentication_abilities = authentication_abilities
 | |
|       @namespace_path = namespace_path || project&.namespace&.full_path
 | |
|       @project_path = project_path || project&.path
 | |
|       @redirected_path = redirected_path
 | |
|       @auth_result_type = auth_result_type
 | |
|     end
 | |
| 
 | |
|     def check(cmd, changes)
 | |
|       @logger = Checks::TimedLogger.new(timeout: INTERNAL_TIMEOUT, header: LOG_HEADER)
 | |
|       @changes = changes
 | |
| 
 | |
|       check_protocol!
 | |
|       check_valid_actor!
 | |
|       check_active_user!
 | |
|       check_authentication_abilities!(cmd)
 | |
|       check_command_disabled!(cmd)
 | |
|       check_command_existence!(cmd)
 | |
| 
 | |
|       custom_action = check_custom_action(cmd)
 | |
|       return custom_action if custom_action
 | |
| 
 | |
|       check_db_accessibility!(cmd)
 | |
|       check_project!(changes, cmd)
 | |
|       check_repository_existence!
 | |
| 
 | |
|       case cmd
 | |
|       when *DOWNLOAD_COMMANDS
 | |
|         check_download_access!
 | |
|       when *PUSH_COMMANDS
 | |
|         check_push_access!
 | |
|       end
 | |
| 
 | |
|       success_result(cmd)
 | |
|     end
 | |
| 
 | |
|     def guest_can_download_code?
 | |
|       Guest.can?(:download_code, project)
 | |
|     end
 | |
| 
 | |
|     def user_can_download_code?
 | |
|       authentication_abilities.include?(:download_code) && user_access.can_do_action?(:download_code)
 | |
|     end
 | |
| 
 | |
|     def build_can_download_code?
 | |
|       authentication_abilities.include?(:build_download_code) && user_access.can_do_action?(:build_download_code)
 | |
|     end
 | |
| 
 | |
|     def request_from_ci_build?
 | |
|       return false unless protocol == 'http'
 | |
| 
 | |
|       auth_result_type == :build || auth_result_type == :ci
 | |
|     end
 | |
| 
 | |
|     def protocol_allowed?
 | |
|       Gitlab::ProtocolAccess.allowed?(protocol)
 | |
|     end
 | |
| 
 | |
|     private
 | |
| 
 | |
|     def check_project!(changes, cmd)
 | |
|       check_namespace!
 | |
|       ensure_project_on_push!(cmd, changes)
 | |
|       check_project_accessibility!
 | |
|       add_project_moved_message!
 | |
|     end
 | |
| 
 | |
|     def check_custom_action(cmd)
 | |
|       nil
 | |
|     end
 | |
| 
 | |
|     def check_for_console_messages(cmd)
 | |
|       []
 | |
|     end
 | |
| 
 | |
|     def check_valid_actor!
 | |
|       return unless actor.is_a?(Key)
 | |
| 
 | |
|       unless actor.valid?
 | |
|         raise ForbiddenError, "Your SSH key #{actor.errors[:key].first}."
 | |
|       end
 | |
|     end
 | |
| 
 | |
|     def check_protocol!
 | |
|       return if request_from_ci_build?
 | |
| 
 | |
|       unless protocol_allowed?
 | |
|         raise ForbiddenError, "Git access over #{protocol.upcase} is not allowed"
 | |
|       end
 | |
|     end
 | |
| 
 | |
|     def check_namespace!
 | |
|       return if namespace_path.present?
 | |
| 
 | |
|       raise NotFoundError, ERROR_MESSAGES[:project_not_found]
 | |
|     end
 | |
| 
 | |
|     def check_active_user!
 | |
|       return unless user
 | |
| 
 | |
|       unless user_access.allowed?
 | |
|         message = Gitlab::Auth::UserAccessDeniedReason.new(user).rejection_message
 | |
|         raise ForbiddenError, message
 | |
|       end
 | |
|     end
 | |
| 
 | |
|     def check_authentication_abilities!(cmd)
 | |
|       case cmd
 | |
|       when *DOWNLOAD_COMMANDS
 | |
|         unless authentication_abilities.include?(:download_code) || authentication_abilities.include?(:build_download_code)
 | |
|           raise ForbiddenError, ERROR_MESSAGES[:auth_download]
 | |
|         end
 | |
|       when *PUSH_COMMANDS
 | |
|         unless authentication_abilities.include?(:push_code)
 | |
|           raise ForbiddenError, ERROR_MESSAGES[:auth_upload]
 | |
|         end
 | |
|       end
 | |
|     end
 | |
| 
 | |
|     def check_project_accessibility!
 | |
|       if project.blank? || !can_read_project?
 | |
|         raise NotFoundError, ERROR_MESSAGES[:project_not_found]
 | |
|       end
 | |
|     end
 | |
| 
 | |
|     def add_project_moved_message!
 | |
|       return if redirected_path.nil?
 | |
| 
 | |
|       project_moved = Checks::ProjectMoved.new(project, user, protocol, redirected_path)
 | |
| 
 | |
|       project_moved.add_message
 | |
|     end
 | |
| 
 | |
|     def check_command_disabled!(cmd)
 | |
|       if upload_pack?(cmd)
 | |
|         check_upload_pack_disabled!
 | |
|       elsif receive_pack?(cmd)
 | |
|         check_receive_pack_disabled!
 | |
|       end
 | |
|     end
 | |
| 
 | |
|     def check_upload_pack_disabled!
 | |
|       if http? && upload_pack_disabled_over_http?
 | |
|         raise ForbiddenError, ERROR_MESSAGES[:upload_pack_disabled_over_http]
 | |
|       end
 | |
|     end
 | |
| 
 | |
|     def check_receive_pack_disabled!
 | |
|       if http? && receive_pack_disabled_over_http?
 | |
|         raise ForbiddenError, ERROR_MESSAGES[:receive_pack_disabled_over_http]
 | |
|       end
 | |
|     end
 | |
| 
 | |
|     def check_command_existence!(cmd)
 | |
|       unless ALL_COMMANDS.include?(cmd)
 | |
|         raise ForbiddenError, ERROR_MESSAGES[:command_not_allowed]
 | |
|       end
 | |
|     end
 | |
| 
 | |
|     def check_db_accessibility!(cmd)
 | |
|       return unless receive_pack?(cmd)
 | |
| 
 | |
|       if Gitlab::Database.read_only?
 | |
|         raise ForbiddenError, push_to_read_only_message
 | |
|       end
 | |
|     end
 | |
| 
 | |
|     def ensure_project_on_push!(cmd, changes)
 | |
|       return if project || deploy_key?
 | |
|       return unless receive_pack?(cmd) && changes == ANY && authentication_abilities.include?(:push_code)
 | |
| 
 | |
|       namespace = Namespace.find_by_full_path(namespace_path)
 | |
| 
 | |
|       return unless user&.can?(:create_projects, namespace)
 | |
| 
 | |
|       project_params = {
 | |
|         path: project_path,
 | |
|         namespace_id: namespace.id,
 | |
|         visibility_level: Gitlab::VisibilityLevel::PRIVATE
 | |
|       }
 | |
| 
 | |
|       project = Projects::CreateService.new(user, project_params).execute
 | |
| 
 | |
|       unless project.saved?
 | |
|         raise ProjectCreationError, "Could not create project: #{project.errors.full_messages.join(', ')}"
 | |
|       end
 | |
| 
 | |
|       @project = project
 | |
|       user_access.project = @project
 | |
| 
 | |
|       Checks::ProjectCreated.new(project, user, protocol).add_message
 | |
|     end
 | |
| 
 | |
|     def check_repository_existence!
 | |
|       unless repository.exists?
 | |
|         raise NotFoundError, ERROR_MESSAGES[:no_repo]
 | |
|       end
 | |
|     end
 | |
| 
 | |
|     def check_download_access!
 | |
|       passed = deploy_key? ||
 | |
|         deploy_token? ||
 | |
|         user_can_download_code? ||
 | |
|         build_can_download_code? ||
 | |
|         guest_can_download_code?
 | |
| 
 | |
|       unless passed
 | |
|         raise ForbiddenError, ERROR_MESSAGES[:download]
 | |
|       end
 | |
|     end
 | |
| 
 | |
|     def check_push_access!
 | |
|       if project.repository_read_only?
 | |
|         raise ForbiddenError, ERROR_MESSAGES[:read_only]
 | |
|       end
 | |
| 
 | |
|       if deploy_key?
 | |
|         unless deploy_key.can_push_to?(project)
 | |
|           raise ForbiddenError, ERROR_MESSAGES[:deploy_key_upload]
 | |
|         end
 | |
|       elsif user
 | |
|         # User access is verified in check_change_access!
 | |
|       else
 | |
|         raise ForbiddenError, ERROR_MESSAGES[:upload]
 | |
|       end
 | |
| 
 | |
|       check_change_access!
 | |
|     end
 | |
| 
 | |
|     def check_change_access!
 | |
|       # Deploy keys with write access can push anything
 | |
|       return if deploy_key?
 | |
| 
 | |
|       if changes == ANY
 | |
|         can_push = user_access.can_do_action?(:push_code) ||
 | |
|           project.any_branch_allows_collaboration?(user_access.user)
 | |
| 
 | |
|         unless can_push
 | |
|           raise ForbiddenError, ERROR_MESSAGES[:push_code]
 | |
|         end
 | |
|       else
 | |
|         # If there are worktrees with a HEAD pointing to a non-existent object,
 | |
|         # calls to `git rev-list --all` will fail in git 2.15+. This should also
 | |
|         # clear stale lock files.
 | |
|         project.repository.clean_stale_repository_files
 | |
| 
 | |
|         # Iterate over all changes to find if user allowed all of them to be applied
 | |
|         changes_list.each.with_index do |change, index|
 | |
|           first_change = index == 0
 | |
| 
 | |
|           # If user does not have access to make at least one change, cancel all
 | |
|           # push by allowing the exception to bubble up
 | |
|           check_single_change_access(change, skip_lfs_integrity_check: !first_change)
 | |
|         end
 | |
|       end
 | |
|     end
 | |
| 
 | |
|     def check_single_change_access(change, skip_lfs_integrity_check: false)
 | |
|       change_access = Checks::ChangeAccess.new(
 | |
|         change,
 | |
|         user_access: user_access,
 | |
|         project: project,
 | |
|         skip_lfs_integrity_check: skip_lfs_integrity_check,
 | |
|         protocol: protocol,
 | |
|         logger: logger
 | |
|       )
 | |
| 
 | |
|       change_access.exec
 | |
|     rescue Checks::TimedLogger::TimeoutError
 | |
|       raise TimeoutError, logger.full_message
 | |
|     end
 | |
| 
 | |
|     def deploy_key
 | |
|       actor if deploy_key?
 | |
|     end
 | |
| 
 | |
|     def deploy_key?
 | |
|       actor.is_a?(DeployKey)
 | |
|     end
 | |
| 
 | |
|     def deploy_token
 | |
|       actor if deploy_token?
 | |
|     end
 | |
| 
 | |
|     def deploy_token?
 | |
|       actor.is_a?(DeployToken)
 | |
|     end
 | |
| 
 | |
|     def ci?
 | |
|       actor == :ci
 | |
|     end
 | |
| 
 | |
|     def can_read_project?
 | |
|       if deploy_key?
 | |
|         deploy_key.has_access_to?(project)
 | |
|       elsif deploy_token?
 | |
|         deploy_token.has_access_to?(project)
 | |
|       elsif user
 | |
|         user.can?(:read_project, project)
 | |
|       elsif ci?
 | |
|         true # allow CI (build without a user) for backwards compatibility
 | |
|       end || Guest.can?(:read_project, project)
 | |
|     end
 | |
| 
 | |
|     def http?
 | |
|       protocol == 'http'
 | |
|     end
 | |
| 
 | |
|     def upload_pack?(command)
 | |
|       command == 'git-upload-pack'
 | |
|     end
 | |
| 
 | |
|     def receive_pack?(command)
 | |
|       command == 'git-receive-pack'
 | |
|     end
 | |
| 
 | |
|     def upload_pack_disabled_over_http?
 | |
|       !Gitlab.config.gitlab_shell.upload_pack
 | |
|     end
 | |
| 
 | |
|     def receive_pack_disabled_over_http?
 | |
|       !Gitlab.config.gitlab_shell.receive_pack
 | |
|     end
 | |
| 
 | |
|     protected
 | |
| 
 | |
|     def success_result(cmd)
 | |
|       ::Gitlab::GitAccessResult::Success.new(console_messages: check_for_console_messages(cmd))
 | |
|     end
 | |
| 
 | |
|     def changes_list
 | |
|       @changes_list ||= Gitlab::ChangesList.new(changes == ANY ? [] : changes)
 | |
|     end
 | |
| 
 | |
|     def user
 | |
|       return @user if defined?(@user)
 | |
| 
 | |
|       @user =
 | |
|         case actor
 | |
|         when User
 | |
|           actor
 | |
|         when DeployKey
 | |
|           nil
 | |
|         when Key
 | |
|           actor.user
 | |
|         when :ci
 | |
|           nil
 | |
|         end
 | |
|     end
 | |
| 
 | |
|     def user_access
 | |
|       @user_access ||= if ci?
 | |
|                          CiAccess.new
 | |
|                        elsif user && request_from_ci_build?
 | |
|                          BuildAccess.new(user, project: project)
 | |
|                        else
 | |
|                          UserAccess.new(user, project: project)
 | |
|                        end
 | |
|     end
 | |
| 
 | |
|     def push_to_read_only_message
 | |
|       ERROR_MESSAGES[:cannot_push_to_read_only]
 | |
|     end
 | |
| 
 | |
|     def repository
 | |
|       project.repository
 | |
|     end
 | |
|   end
 | |
| end
 | |
| 
 | |
| Gitlab::GitAccess.prepend_if_ee('EE::Gitlab::GitAccess')
 |