git_http_client_controller.rb 3.8 KB
Newer Older
J
Jacob Vosmaer 已提交
1 2 3 4 5 6
# This file should be identical in GitLab Community Edition and Enterprise Edition

class Projects::GitHttpClientController < Projects::ApplicationController
  include ActionController::HttpAuthentication::Basic
  include KerberosSpnegoHelper

P
Patricio Cano 已提交
7 8
  class MissingPersonalTokenError < StandardError; end

J
Jacob Vosmaer 已提交
9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25
  attr_reader :user

  # Git clients will not know what authenticity token to send along
  skip_before_action :verify_authenticity_token
  skip_before_action :repository
  before_action :authenticate_user
  before_action :ensure_project_found!

  private

  def authenticate_user
    if project && project.public? && download_request?
      return # Allow access
    end

    if allow_basic_auth? && basic_auth_provided?
      login, password = user_name_and_password(request)
P
Patricio Cano 已提交
26 27

      handle_authentication(login, password)
J
Jacob Vosmaer 已提交
28 29 30 31 32 33 34 35 36 37 38 39 40 41 42

      if ci? || user
        return # Allow access
      end
    elsif allow_kerberos_spnego_auth? && spnego_provided?
      @user = find_kerberos_user

      if user
        send_final_spnego_response
        return # Allow access
      end
    end

    send_challenges
    render plain: "HTTP Basic: Access denied\n", status: 401
P
Patricio Cano 已提交
43 44 45 46

  rescue MissingPersonalTokenError
    render_missing_personal_token
    return
J
Jacob Vosmaer 已提交
47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92
  end

  def basic_auth_provided?
    has_basic_credentials?(request)
  end

  def send_challenges
    challenges = []
    challenges << 'Basic realm="GitLab"' if allow_basic_auth?
    challenges << spnego_challenge if allow_kerberos_spnego_auth?
    headers['Www-Authenticate'] = challenges.join("\n") if challenges.any?
  end

  def ensure_project_found!
    render_not_found if project.blank?
  end

  def project
    return @project if defined?(@project)

    project_id, _ = project_id_with_suffix
    if project_id.blank?
      @project = nil
    else
      @project = Project.find_with_namespace("#{params[:namespace_id]}/#{project_id}")
    end
  end

  # This method returns two values so that we can parse
  # params[:project_id] (untrusted input!) in exactly one place.
  def project_id_with_suffix
    id = params[:project_id] || ''

    %w[.wiki.git .git].each do |suffix|
      if id.end_with?(suffix)
        # Be careful to only remove the suffix from the end of 'id'.
        # Accidentally removing it from the middle is how security
        # vulnerabilities happen!
        return [id.slice(0, id.length - suffix.length), suffix]
      end
    end

    # Something is wrong with params[:project_id]; do not pass it on.
    [nil, nil]
  end

93
  def render_missing_personal_token
94 95
    render plain: "HTTP Basic: Access denied\n" \
                  "You have 2FA enabled, please use a personal access token for Git over HTTP.\n" \
96 97
                  "You can generate one at #{profile_personal_access_tokens_url}",
           status: 401
98 99
  end

J
Jacob Vosmaer 已提交
100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115
  def repository
    _, suffix = project_id_with_suffix
    if suffix == '.wiki.git'
      project.wiki.repository
    else
      project.repository
    end
  end

  def render_not_found
    render plain: 'Not Found', status: :not_found
  end

  def ci?
    @ci.present?
  end
116

P
Patricio Cano 已提交
117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134
  def handle_authentication(login, password)
    auth_result = Gitlab::Auth.find_for_git_client(login, password, project: project, ip: request.ip)

    if auth_result.type == :ci && download_request?
      @ci = true
    elsif auth_result.type == :oauth && !download_request?
      # Not allowed
    elsif auth_result.type == :missing_personal_token
      raise MissingPersonalTokenError
    elsif auth_result.type == :lfs_deploy_token && download_request?
      @lfs_deploy_key = true
      @user = auth_result.user
    else
      @user = auth_result.user
    end
  end

  def lfs_deploy_key?
135
    @lfs_deploy_key.present? && (user && user.projects.include?(project))
P
Patricio Cano 已提交
136 137
  end

138 139 140
  def verify_workhorse_api!
    Gitlab::Workhorse.verify_api_request!(request.headers)
  end
J
Jacob Vosmaer 已提交
141
end