mirror of
https://github.com/discourse/discourse.git
synced 2024-12-18 12:44:00 +08:00
4078b22887
This commit prevents a 500 error from occurring if someone is trying to setup their discourse instance as a sso provider and they don't pass in a `return_sso_url` in their payload.
41 lines
1.3 KiB
Ruby
41 lines
1.3 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
class SingleSignOnProvider < SingleSignOn
|
|
class BlankSecret < RuntimeError; end
|
|
|
|
def self.parse(payload, sso_secret = nil)
|
|
set_return_sso_url(payload)
|
|
if sso_secret.blank? && self.sso_secret.blank?
|
|
host = URI.parse(@return_sso_url).host
|
|
Rails.logger.warn("SSO failed; website #{host} is not in the `sso_provider_secrets` site settings")
|
|
raise BlankSecret
|
|
end
|
|
|
|
super
|
|
end
|
|
|
|
def self.set_return_sso_url(payload)
|
|
parsed = Rack::Utils.parse_query(payload)
|
|
decoded = Base64.decode64(parsed["sso"])
|
|
decoded_hash = Rack::Utils.parse_query(decoded)
|
|
|
|
raise ParseError unless decoded_hash.key? 'return_sso_url'
|
|
@return_sso_url = decoded_hash['return_sso_url']
|
|
end
|
|
|
|
def self.sso_secret
|
|
return nil unless @return_sso_url && SiteSetting.enable_sso_provider
|
|
|
|
provider_secrets = SiteSetting.sso_provider_secrets.split(/[|\n]/)
|
|
provider_secrets_hash = Hash[*provider_secrets]
|
|
return_url_host = URI.parse(@return_sso_url).host
|
|
# moves wildcard domains to the end of hash
|
|
sorted_secrets = provider_secrets_hash.sort_by { |k, _| k }.reverse.to_h
|
|
|
|
secret = sorted_secrets.select do |domain, _|
|
|
WildcardDomainChecker.check_domain(domain, return_url_host)
|
|
end
|
|
secret.present? ? secret.values.first : nil
|
|
end
|
|
end
|