Deal with selecting relying party endpoint sooner and populating subject confirmation...
[java-idp.git] / src / edu / internet2 / middleware / shibboleth / idp / profile / AbstractSAMLProfileHandler.java
index f6a712d..643cbf7 100644 (file)
@@ -25,6 +25,7 @@ import org.opensaml.common.IdentifierGenerator;
 import org.opensaml.common.SAMLObject;
 import org.opensaml.common.binding.decoding.MessageDecoderFactory;
 import org.opensaml.common.binding.encoding.MessageEncoderFactory;
 import org.opensaml.common.SAMLObject;
 import org.opensaml.common.binding.decoding.MessageDecoderFactory;
 import org.opensaml.common.binding.encoding.MessageEncoderFactory;
+import org.opensaml.saml2.metadata.Endpoint;
 import org.opensaml.saml2.metadata.EntityDescriptor;
 import org.opensaml.saml2.metadata.RoleDescriptor;
 import org.opensaml.saml2.metadata.provider.MetadataProvider;
 import org.opensaml.saml2.metadata.EntityDescriptor;
 import org.opensaml.saml2.metadata.RoleDescriptor;
 import org.opensaml.saml2.metadata.provider.MetadataProvider;
@@ -158,13 +159,16 @@ public abstract class AbstractSAMLProfileHandler extends
      * @param <StatusType> type of Status object
      */
     protected class SAMLProfileRequestContext<StatusType extends SAMLObject> extends ShibbolethProfileRequestContext {
      * @param <StatusType> type of Status object
      */
     protected class SAMLProfileRequestContext<StatusType extends SAMLObject> extends ShibbolethProfileRequestContext {
-
+        
         /** Entity descriptor for the asserting party. */
         private EntityDescriptor assertingPartyMetadata;
 
         /** Role descriptor meatadata for the asserting party. */
         private RoleDescriptor assertingPartyRoleMetadata;
 
         /** Entity descriptor for the asserting party. */
         private EntityDescriptor assertingPartyMetadata;
 
         /** Role descriptor meatadata for the asserting party. */
         private RoleDescriptor assertingPartyRoleMetadata;
 
+        /** Endpoint of relying party. */
+        private Endpoint relyingPartyEndpoint;
+        
         /** Entity descriptor for the relying party. */
         private EntityDescriptor relyingPartyMetadata;
 
         /** Entity descriptor for the relying party. */
         private EntityDescriptor relyingPartyMetadata;
 
@@ -217,6 +221,24 @@ public abstract class AbstractSAMLProfileHandler extends
         public void setAssertingPartyRoleMetadata(RoleDescriptor descriptor) {
             assertingPartyRoleMetadata = descriptor;
         }
         public void setAssertingPartyRoleMetadata(RoleDescriptor descriptor) {
             assertingPartyRoleMetadata = descriptor;
         }
+        
+        /**
+         * Gets the endpoint for the relying party.
+         * 
+         * @return endpoint for the relying party
+         */
+        public Endpoint getRelyingPartyEndpoint(){
+            return relyingPartyEndpoint;
+        }
+        
+        /**
+         * Sets the endpoint for the relying party.
+         * 
+         * @param endpoint endpoint for the relying party
+         */
+        public void setRelyingPartyEndpoint(Endpoint endpoint){
+            relyingPartyEndpoint = endpoint;
+        }
 
         /**
          * Gets the metadata for the relying party.
 
         /**
          * Gets the metadata for the relying party.