mirror of https://github.com/jacekkow/keycloak-protocol-cas

github-actions
2020-06-03 1fba7a0a60eac476992fd81a00a8248749835601
commit | author | age
74023a 1 package org.keycloak.protocol.cas.endpoints;
EH 2
3 import org.jboss.logging.Logger;
4 import org.jboss.resteasy.spi.HttpRequest;
5 import org.keycloak.common.ClientConnection;
6 import org.keycloak.events.Errors;
7 import org.keycloak.events.EventBuilder;
8 import org.keycloak.models.*;
9 import org.keycloak.protocol.ProtocolMapper;
10 import org.keycloak.protocol.cas.CASLoginProtocol;
11 import org.keycloak.protocol.cas.mappers.CASAttributeMapper;
12 import org.keycloak.protocol.cas.representations.CASErrorCode;
13 import org.keycloak.protocol.cas.utils.CASValidationException;
c140ce 14 import org.keycloak.protocol.oidc.utils.OAuth2CodeParser;
74023a 15 import org.keycloak.protocol.oidc.utils.RedirectUtils;
EH 16 import org.keycloak.services.managers.AuthenticationManager;
17 import org.keycloak.services.util.DefaultClientSessionContext;
18
19 import javax.ws.rs.core.Context;
20 import javax.ws.rs.core.HttpHeaders;
21 import javax.ws.rs.core.Response;
22 import java.util.HashMap;
23 import java.util.Map;
24 import java.util.Set;
25
26 public abstract class AbstractValidateEndpoint {
27     protected final Logger logger = Logger.getLogger(getClass());
28     @Context
29     protected KeycloakSession session;
30     @Context
31     protected ClientConnection clientConnection;
32     @Context
33     protected HttpRequest request;
34     @Context
35     protected HttpHeaders headers;
36     protected RealmModel realm;
37     protected EventBuilder event;
38     protected ClientModel client;
39     protected AuthenticatedClientSessionModel clientSession;
40
41     public AbstractValidateEndpoint(RealmModel realm, EventBuilder event) {
42         this.realm = realm;
43         this.event = event;
44     }
45
46     protected void checkSsl() {
47         if (!session.getContext().getUri().getBaseUri().getScheme().equals("https") && realm.getSslRequired().isRequired(clientConnection)) {
48             throw new CASValidationException(CASErrorCode.INVALID_REQUEST, "HTTPS required", Response.Status.FORBIDDEN);
49         }
50     }
51
52     protected void checkRealm() {
53         if (!realm.isEnabled()) {
54             throw new CASValidationException(CASErrorCode.INTERNAL_ERROR, "Realm not enabled", Response.Status.FORBIDDEN);
55         }
56     }
57
58     protected void checkClient(String service) {
59         if (service == null) {
60             event.error(Errors.INVALID_REQUEST);
61             throw new CASValidationException(CASErrorCode.INVALID_REQUEST, "Missing parameter: " + CASLoginProtocol.SERVICE_PARAM, Response.Status.BAD_REQUEST);
62         }
63
64         client = realm.getClients().stream()
65                 .filter(c -> CASLoginProtocol.LOGIN_PROTOCOL.equals(c.getProtocol()))
019db5 66                 .filter(c -> RedirectUtils.verifyRedirectUri(session, service, c) != null)
74023a 67                 .findFirst().orElse(null);
EH 68         if (client == null) {
69             event.error(Errors.CLIENT_NOT_FOUND);
70             throw new CASValidationException(CASErrorCode.INVALID_SERVICE, "Client not found", Response.Status.BAD_REQUEST);
71         }
72
73         if (!client.isEnabled()) {
74             event.error(Errors.CLIENT_DISABLED);
75             throw new CASValidationException(CASErrorCode.INVALID_SERVICE, "Client disabled", Response.Status.BAD_REQUEST);
76         }
77
78         event.client(client.getClientId());
79
80         session.getContext().setClient(client);
81     }
82
83     protected void checkTicket(String ticket, boolean requireReauth) {
84         if (ticket == null) {
85             event.error(Errors.INVALID_CODE);
86             throw new CASValidationException(CASErrorCode.INVALID_REQUEST, "Missing parameter: " + CASLoginProtocol.TICKET_PARAM, Response.Status.BAD_REQUEST);
87         }
88         if (!ticket.startsWith(CASLoginProtocol.SERVICE_TICKET_PREFIX)) {
89             event.error(Errors.INVALID_CODE);
90             throw new CASValidationException(CASErrorCode.INVALID_TICKET_SPEC, "Malformed service ticket", Response.Status.BAD_REQUEST);
91         }
92
93         String code = ticket.substring(CASLoginProtocol.SERVICE_TICKET_PREFIX.length());
94
c140ce 95         OAuth2CodeParser.ParseResult parseResult = OAuth2CodeParser.parseCode(session, code, realm, event);
MP 96         if (parseResult.isIllegalCode()) {
74023a 97             event.error(Errors.INVALID_CODE);
EH 98
99             // Attempt to use same code twice should invalidate existing clientSession
100             AuthenticatedClientSessionModel clientSession = parseResult.getClientSession();
101             if (clientSession != null) {
102                 clientSession.detachFromUserSession();
103             }
104
105             throw new CASValidationException(CASErrorCode.INVALID_TICKET, "Code not valid", Response.Status.BAD_REQUEST);
106         }
107
108         clientSession = parseResult.getClientSession();
109
c140ce 110         if (parseResult.isExpiredCode()) {
74023a 111             event.error(Errors.EXPIRED_CODE);
EH 112             throw new CASValidationException(CASErrorCode.INVALID_TICKET, "Code is expired", Response.Status.BAD_REQUEST);
113         }
114
115         clientSession.setNote(CASLoginProtocol.SESSION_SERVICE_TICKET, ticket);
116
117         if (requireReauth && AuthenticationManager.isSSOAuthentication(clientSession)) {
118             event.error(Errors.SESSION_EXPIRED);
119             throw new CASValidationException(CASErrorCode.INVALID_TICKET, "Interactive authentication was requested but not performed", Response.Status.BAD_REQUEST);
120         }
121
122         UserSessionModel userSession = clientSession.getUserSession();
123
124         if (userSession == null) {
125             event.error(Errors.USER_SESSION_NOT_FOUND);
126             throw new CASValidationException(CASErrorCode.INVALID_TICKET, "User session not found", Response.Status.BAD_REQUEST);
127         }
128
129         UserModel user = userSession.getUser();
130         if (user == null) {
131             event.error(Errors.USER_NOT_FOUND);
132             throw new CASValidationException(CASErrorCode.INVALID_TICKET, "User not found", Response.Status.BAD_REQUEST);
133         }
134         if (!user.isEnabled()) {
135             event.error(Errors.USER_DISABLED);
136             throw new CASValidationException(CASErrorCode.INVALID_TICKET, "User disabled", Response.Status.BAD_REQUEST);
137         }
138
139         event.user(userSession.getUser());
140         event.session(userSession.getId());
141
142         if (!client.getClientId().equals(clientSession.getClient().getClientId())) {
143             event.error(Errors.INVALID_CODE);
144             throw new CASValidationException(CASErrorCode.INVALID_SERVICE, "Auth error", Response.Status.BAD_REQUEST);
145         }
146
147         if (!AuthenticationManager.isSessionValid(realm, userSession)) {
148             event.error(Errors.USER_SESSION_NOT_FOUND);
149             throw new CASValidationException(CASErrorCode.INVALID_TICKET, "Session not active", Response.Status.BAD_REQUEST);
150         }
151     }
152
153     protected Map<String, Object> getUserAttributes() {
154         UserSessionModel userSession = clientSession.getUserSession();
155         // CAS protocol does not support scopes, so pass null scopeParam
8379a3 156         ClientSessionContext clientSessionCtx = DefaultClientSessionContext.fromClientSessionAndScopeParameter(clientSession, null, session);
74023a 157
EH 158         Set<ProtocolMapperModel> mappings = clientSessionCtx.getProtocolMappers();
159         KeycloakSessionFactory sessionFactory = session.getKeycloakSessionFactory();
160         Map<String, Object> attributes = new HashMap<>();
161         for (ProtocolMapperModel mapping : mappings) {
162             ProtocolMapper mapper = (ProtocolMapper) sessionFactory.getProviderFactory(ProtocolMapper.class, mapping.getProtocolMapper());
163             if (mapper instanceof CASAttributeMapper) {
164                 ((CASAttributeMapper) mapper).setAttribute(attributes, mapping, userSession, session, clientSessionCtx);
165             }
166         }
167         return attributes;
168     }
169 }