Class: XMLSecurity::Document

Inherits:
BaseDocument show all
Defined in:
lib/xml_security.rb

Constant Summary collapse

RSA_SHA1 =
"http://www.w3.org/2000/09/xmldsig#rsa-sha1"
RSA_SHA256 =
"http://www.w3.org/2001/04/xmldsig-more#rsa-sha256"
RSA_SHA384 =
"http://www.w3.org/2001/04/xmldsig-more#rsa-sha384"
RSA_SHA512 =
"http://www.w3.org/2001/04/xmldsig-more#rsa-sha512"
SHA1 =
"http://www.w3.org/2000/09/xmldsig#sha1"
SHA256 =
'http://www.w3.org/2001/04/xmlenc#sha256'
SHA384 =
"http://www.w3.org/2001/04/xmldsig-more#sha384"
SHA512 =
'http://www.w3.org/2001/04/xmlenc#sha512'
ENVELOPED_SIG =
"http://www.w3.org/2000/09/xmldsig#enveloped-signature"
INC_PREFIX_LIST =
"#default samlp saml ds xs xsi md"

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, safe_load_xml

Instance Attribute Details

#uuidObject



126
127
128
129
130
# File 'lib/xml_security.rb', line 126

def uuid
  @uuid ||= begin
    document.root.nil? ? nil : document.root.attributes['ID']
  end
end

Instance Method Details

#sign_document(private_key, certificate, signature_method = RSA_SHA1, digest_method = SHA1, check_malformed_doc = true) ⇒ Object

</Signature>



147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
# File 'lib/xml_security.rb', line 147

def sign_document(private_key, certificate, signature_method = RSA_SHA1, digest_method = SHA1, check_malformed_doc = true)
  noko = XMLSecurity::BaseDocument.safe_load_xml(self.to_s, check_malformed_doc)

  signature_element = REXML::Element.new("ds:Signature").add_namespace('ds', DSIG)
  signed_info_element = signature_element.add_element("ds:SignedInfo")
  signed_info_element.add_element("ds:CanonicalizationMethod", {"Algorithm" => C14N})
  signed_info_element.add_element("ds:SignatureMethod", {"Algorithm"=>signature_method})

  # Add Reference
  reference_element = signed_info_element.add_element("ds:Reference", {"URI" => "##{uuid}"})

  # Add Transforms
  transforms_element = reference_element.add_element("ds:Transforms")
  transforms_element.add_element("ds:Transform", {"Algorithm" => ENVELOPED_SIG})
  c14element = transforms_element.add_element("ds:Transform", {"Algorithm" => C14N})
  c14element.add_element("ec:InclusiveNamespaces", {"xmlns:ec" => C14N, "PrefixList" => INC_PREFIX_LIST})

  digest_method_element = reference_element.add_element("ds:DigestMethod", {"Algorithm" => digest_method})
  inclusive_namespaces = INC_PREFIX_LIST.split(" ")
  canon_doc = noko.canonicalize(canon_algorithm(C14N), inclusive_namespaces)
  reference_element.add_element("ds:DigestValue").text = compute_digest(canon_doc, algorithm(digest_method_element))

  # add SignatureValue
  noko_sig_element = XMLSecurity::BaseDocument.safe_load_xml(signature_element.to_s, check_malformed_doc)

  noko_signed_info_element = noko_sig_element.at_xpath('//ds:Signature/ds:SignedInfo', 'ds' => DSIG)
  canon_string = noko_signed_info_element.canonicalize(canon_algorithm(C14N))

  signature = compute_signature(private_key, algorithm(signature_method).new, canon_string)
  signature_element.add_element("ds:SignatureValue").text = signature

  # add KeyInfo
  key_info_element       = signature_element.add_element("ds:KeyInfo")
  x509_element           = key_info_element.add_element("ds:X509Data")
  x509_cert_element      = x509_element.add_element("ds:X509Certificate")
  if certificate.is_a?(String)
    certificate = OpenSSL::X509::Certificate.new(certificate)
  end
  x509_cert_element.text = Base64.encode64(certificate.to_der).gsub(/\n/, "")

  # add the signature
  issuer_element = self.elements["//saml:Issuer"]
  if issuer_element
    self.root.insert_after issuer_element, signature_element
  else
    if sp_sso_descriptor = self.elements["/md:EntityDescriptor"]
      self.root.insert_before sp_sso_descriptor, signature_element
    else
      self.root.add_element(signature_element)
    end
  end
end