http_artifact.rb in libsaml-2.1.7

- old
+ new

@@ -1,13 +1,14 @@
 module Saml
   module Bindings
     class HTTPArtifact
+      include Saml::Notification
 
       class << self
         # @param [Saml::ArtifactResponse] artifact_response
         def create_response_xml(artifact_response)
-          Saml::Util.sign_xml(artifact_response, :soap)
+          notify('create_response', Saml::Util.sign_xml(artifact_response, :soap))
         end
 
         def create_response(artifact_response)
           {xml: create_response_xml(artifact_response), content_type: 'text/xml'}
         end
@@ -21,22 +22,23 @@
           uri.query = query.compact.join("&")
           uri.to_s
         end
 
         def receive_message(request)
-          raw_xml          = request.body.dup.read
+          raw_xml          = notify('receive_message', request.body.dup.read)
           artifact_resolve = Saml::ArtifactResolve.parse(raw_xml, single: true)
 
           Saml::Util.verify_xml(artifact_resolve, raw_xml)
         end
 
         def resolve(request, location)
           artifact         = request.params["SAMLart"]
           artifact_resolve = Saml::ArtifactResolve.new(artifact: artifact, destination: location)
 
-          response = Saml::Util.post(location, Saml::Util.sign_xml(artifact_resolve, :soap))
+          response = Saml::Util.post(location, notify('create_post', Saml::Util.sign_xml(artifact_resolve, :soap)))
 
           if response.code == "200"
+            notify('receive_response', response.body)
             artifact_response          = Saml::ArtifactResponse.parse(response.body, single: true)
             verified_artifact_response = Saml::Util.verify_xml(artifact_response, response.body)
 
             verified_artifact_response.response if artifact_response.success?
           end