From 89148473d76316a111e59740cbbd791be7d12017 Mon Sep 17 00:00:00 2001
From: Daniel Ramos <dan@daramos.com>
Date: Thu, 31 Mar 2022 21:15:42 +0000
Subject: [PATCH] SAML 1.1 - return "SAMLart" instead of "ticket" if "TARGET" is specified

---
 src/main/java/org/keycloak/protocol/cas/CASLoginProtocol.java                |   11 ++++++++++-
 src/main/java/org/keycloak/protocol/cas/endpoints/AuthorizationEndpoint.java |   11 +++++++++++
 2 files changed, 21 insertions(+), 1 deletions(-)

diff --git a/src/main/java/org/keycloak/protocol/cas/CASLoginProtocol.java b/src/main/java/org/keycloak/protocol/cas/CASLoginProtocol.java
index 14f276a..718ae6b 100644
--- a/src/main/java/org/keycloak/protocol/cas/CASLoginProtocol.java
+++ b/src/main/java/org/keycloak/protocol/cas/CASLoginProtocol.java
@@ -36,6 +36,7 @@
     public static final String FORMAT_PARAM = "format";
 
     public static final String TICKET_RESPONSE_PARAM = "ticket";
+    public static final String SAMLART_RESPONSE_PARAM = "SAMLart";
 
     public static final String SERVICE_TICKET_PREFIX = "ST-";
     public static final String SESSION_SERVICE_TICKET = "service_ticket";
@@ -102,7 +103,15 @@
         String code = OAuth2CodeParser.persistCode(session, clientSession, codeData);
 
         KeycloakUriBuilder uriBuilder = KeycloakUriBuilder.fromUri(service);
-        uriBuilder.queryParam(TICKET_RESPONSE_PARAM, SERVICE_TICKET_PREFIX + code);
+
+        String loginTicket = SERVICE_TICKET_PREFIX + code;
+
+        if (authSession.getClientNotes().containsKey(CASLoginProtocol.TARGET_PARAM)) {
+            // This was a SAML 1.1 auth request so return the ticket ID as "SAMLart" instead of "ticket"
+            uriBuilder.queryParam(SAMLART_RESPONSE_PARAM, loginTicket);
+        } else {
+            uriBuilder.queryParam(TICKET_RESPONSE_PARAM, loginTicket);
+        }
 
         URI redirectUri = uriBuilder.build();
 
diff --git a/src/main/java/org/keycloak/protocol/cas/endpoints/AuthorizationEndpoint.java b/src/main/java/org/keycloak/protocol/cas/endpoints/AuthorizationEndpoint.java
index 1526d21..d046247 100644
--- a/src/main/java/org/keycloak/protocol/cas/endpoints/AuthorizationEndpoint.java
+++ b/src/main/java/org/keycloak/protocol/cas/endpoints/AuthorizationEndpoint.java
@@ -35,6 +35,13 @@
     public Response build() {
         MultivaluedMap<String, String> params = session.getContext().getUri().getQueryParameters();
         String service = params.getFirst(CASLoginProtocol.SERVICE_PARAM);
+
+        boolean isSaml11Request = false;
+        if (service == null && params.containsKey(CASLoginProtocol.TARGET_PARAM)) {
+            // SAML 1.1 authorization uses the TARGET parameter instead of service
+            service = params.getFirst(CASLoginProtocol.TARGET_PARAM);
+            isSaml11Request = true;
+        }
         boolean renew = params.containsKey(CASLoginProtocol.RENEW_PARAM);
         boolean gateway = params.containsKey(CASLoginProtocol.GATEWAY_PARAM);
 
@@ -54,6 +61,10 @@
         if (gateway) {
             authenticationSession.setClientNote(CASLoginProtocol.GATEWAY_PARAM, "true");
         }
+        if (isSaml11Request) {
+            // Flag the session so we can return the ticket as "SAMLart" in the response
+            authenticationSession.setClientNote(CASLoginProtocol.TARGET_PARAM, "true");
+        }
 
         this.event.event(EventType.LOGIN);
         return handleBrowserAuthenticationRequest(authenticationSession, new CASLoginProtocol(session, realm, session.getContext().getUri(), headers, event), gateway, false);

--
Gitblit v1.9.1