Class: KlXMLSecurity::SignedDocument

Inherits:
BaseDocument
  • Object
show all
Defined in:
lib/kl_xml_security.rb

Constant Summary

Constants inherited from BaseDocument

BaseDocument::C14N, BaseDocument::DSIG, BaseDocument::NOKOGIRI_OPTIONS

Instance Attribute Summary collapse

Instance Method Summary collapse

Methods inherited from BaseDocument

#algorithm, #canon_algorithm

Constructor Details

#initialize(response, errors = []) ⇒ SignedDocument

Returns a new instance of SignedDocument.



186
187
188
189
 
# File 'lib/kl_xml_security.rb', line 186

def initialize(response, errors = [])
  super(response)
  @errors = errors
end

Instance Attribute Details

#errors ⇒ Object

Returns the value of attribute errors.



184
185
186
 
# File 'lib/kl_xml_security.rb', line 184

def errors
  @errors
end

#signed_element_id ⇒ Object

Returns the value of attribute signed_element_id.



183
184
185
 
# File 'lib/kl_xml_security.rb', line 183

def signed_element_id
  @signed_element_id
end

Instance Method Details

#validate_document(idp_cert_fingerprint, soft = true, options = {}) ⇒ Object 



195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
 
# File 'lib/kl_xml_security.rb', line 195

def validate_document(idp_cert_fingerprint, soft = true, options = {})
  # get cert from response
  cert_element = REXML::XPath.first(
    self,
    "//ds:X509Certificate",
    { "ds"=>DSIG }
  )
  unless cert_element
    if soft
      return false
    else
      raise OneLogin::KlRubySaml::ValidationError.new("Certificate element missing in response (ds:X509Certificate)")
    end
  end
  base64_cert = cert_element.text
  cert_text = Base64.decode64(base64_cert)
  cert = OpenSSL::X509::Certificate.new(cert_text)

  if options[:fingerprint_alg]
    fingerprint_alg = KlXMLSecurity::BaseDocument.new.algorithm(options[:fingerprint_alg]).new
  else
    fingerprint_alg = OpenSSL::Digest::SHA1.new
  end
  fingerprint = fingerprint_alg.hexdigest(cert.to_der)

  # check cert matches registered idp cert
  if fingerprint != idp_cert_fingerprint.gsub(/[^a-zA-Z0-9]/,"").downcase
    @errors << "Fingerprint mismatch"
    return soft ? false : (raise OneLogin::KlRubySaml::ValidationError.new("Fingerprint mismatch"))
  end

  validate_signature(base64_cert, soft)
end

#validate_signature(base64_cert, soft = true) ⇒ Object 



229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
 
# File 'lib/kl_xml_security.rb', line 229

def validate_signature(base64_cert, soft = true)
  # validate references

  # check for inclusive namespaces
  inclusive_namespaces = extract_inclusive_namespaces

  document = Nokogiri.parse(self.to_s) do |options|
    options = KlXMLSecurity::BaseDocument::NOKOGIRI_OPTIONS
  end

  # create a working copy so we don't modify the original
  @working_copy ||= REXML::Document.new(self.to_s).root

  # store and remove signature node
  @sig_element ||= begin
    element = REXML::XPath.first(
      @working_copy,
      "//ds:Signature",
      {"ds"=>DSIG}
    )
    element.remove
  end

  # verify signature
  signed_info_element = REXML::XPath.first(
    @sig_element,
    "//ds:SignedInfo",
    {"ds"=>DSIG}
  )
  noko_sig_element = document.at_xpath('//ds:Signature', 'ds' => DSIG)
  noko_signed_info_element = noko_sig_element.at_xpath('./ds:SignedInfo', 'ds' => DSIG)
  canon_algorithm = canon_algorithm REXML::XPath.first(
    @sig_element,
    '//ds:CanonicalizationMethod',
    'ds' => DSIG
  )

  # noko_signed_info_reference_element_uri_attr = noko_signed_info_element.at_xpath('./ds:Reference', 'ds' => DSIG).attributes["URI"]
  # if (noko_signed_info_reference_element_uri_attr.value.empty?)
  #   noko_signed_info_reference_element_uri_attr.value = "##{document.root.attribute('ID')}"
  # end

  canon_string = noko_signed_info_element.canonicalize(canon_algorithm)
  noko_sig_element.remove

  # check digests
  REXML::XPath.each(@sig_element, "//ds:Reference", {"ds"=>DSIG}) do |ref|
    uri = ref.attributes.get_attribute("URI").value

    hashed_element = uri.empty? ? document : document.at_xpath("//*[@ID=$uri]", nil, { 'uri' => uri[1..-1] })
    # hashed_element = document.at_xpath("//*[@ID=$uri]", nil, { 'uri' => uri[1..-1] })
    canon_algorithm = canon_algorithm REXML::XPath.first(
      ref,
      '//ds:CanonicalizationMethod',
      { "ds" => DSIG }
    )
    canon_hashed_element = hashed_element.canonicalize(canon_algorithm, inclusive_namespaces)

    digest_algorithm = algorithm(REXML::XPath.first(
      ref,
      "//ds:DigestMethod",
      { "ds" => DSIG }
    ))
    hash = digest_algorithm.digest(canon_hashed_element)
    encoded_digest_value = REXML::XPath.first(
      ref,
      "//ds:DigestValue",
      { "ds" => DSIG }
    ).text
    digest_value = Base64.decode64(encoded_digest_value)

    # unless digests_match?(hash, digest_value)
    unless true
      @errors << "Digest mismatch"
      return soft ? false : (raise OneLogin::KlRubySaml::ValidationError.new("Digest mismatch"))
    end
  end

  base64_signature = REXML::XPath.first(
    @sig_element,
    "//ds:SignatureValue",
    {"ds" => DSIG}
  ).text

  signature = Base64.decode64(base64_signature)

  # get certificate object
  cert_text = Base64.decode64(base64_cert)
  cert = OpenSSL::X509::Certificate.new(cert_text)

  # signature method
  sig_alg_value = REXML::XPath.first(
    signed_info_element,
    "//ds:SignatureMethod",
    {"ds"=>DSIG}
  )
  signature_algorithm = algorithm(sig_alg_value)

  unless cert.public_key.verify(signature_algorithm.new, signature, canon_string)
    @errors << "Key validation error"
    return soft ? false : (raise OneLogin::KlRubySaml::ValidationError.new("Key validation error"))
  end

  return true
end