Further review changes

2 files changed, 351 insertions, 356 deletions

diff --git a/src/main/java/org/traccar/api/resource/SessionResource.java b/src/main/java/org/traccar/api/resource/SessionResource.java
index a20e5f100..8240a8a6f 100644
--- a/src/main/java/org/traccar/api/resource/SessionResource.java
+++ b/src/main/java/org/traccar/api/resource/SessionResource.java
@@ -1,186 +1,188 @@
-/*
- * Copyright 2015 - 2022 Anton Tananaev (anton@traccar.org)
- *
- * Licensed under the Apache License, Version 2.0 (the "License");
- * you may not use this file except in compliance with the License.
- * You may obtain a copy of the License at
- *
- *     http://www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing, software
- * distributed under the License is distributed on an "AS IS" BASIS,
- * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- * See the License for the specific language governing permissions and
- * limitations under the License.
- */
-package org.traccar.api.resource;
-
-import org.traccar.api.BaseResource;
-import org.traccar.api.security.LoginService;
-import org.traccar.api.signature.TokenManager;
-import org.traccar.database.OpenIdProvider;
-import org.traccar.helper.DataConverter;
-import org.traccar.helper.LogAction;
-import org.traccar.helper.ServletHelper;
-import org.traccar.model.User;
-import org.traccar.storage.StorageException;
-import org.traccar.storage.query.Columns;
-import org.traccar.storage.query.Condition;
-import org.traccar.storage.query.Request;
-
-import com.nimbusds.oauth2.sdk.ParseException;
-import javax.annotation.security.PermitAll;
-import javax.inject.Inject;
-import javax.servlet.http.Cookie;
-import javax.servlet.http.HttpServletRequest;
-import javax.ws.rs.Consumes;
-import javax.ws.rs.DELETE;
-import javax.ws.rs.FormParam;
-import javax.ws.rs.GET;
-import javax.ws.rs.POST;
-import javax.ws.rs.Path;
-import javax.ws.rs.PathParam;
-import javax.ws.rs.Produces;
-import javax.ws.rs.QueryParam;
-import javax.ws.rs.WebApplicationException;
-import javax.ws.rs.core.Context;
-import javax.ws.rs.core.MediaType;
-import javax.ws.rs.core.Response;
-import java.io.IOException;
-import java.net.URLDecoder;
-import java.nio.charset.StandardCharsets;
-import java.security.GeneralSecurityException;
-import java.util.Date;
-import java.net.URI;
-
-@Path("session")
-@Produces(MediaType.APPLICATION_JSON)
-@Consumes(MediaType.APPLICATION_FORM_URLENCODED)
-public class SessionResource extends BaseResource {
-
-    public static final String USER_ID_KEY = "userId";
-    public static final String USER_COOKIE_KEY = "user";
-    public static final String PASS_COOKIE_KEY = "password";
-
-    @Inject
-    private LoginService loginService;
-
-    @Inject
-    private OpenIdProvider openIdProvider;
-
-    @Inject
-    private TokenManager tokenManager;
-
-    @Context
-    private HttpServletRequest request;
-
-    @PermitAll
-    @GET
-    public User get(@QueryParam("token") String token) throws StorageException, IOException, GeneralSecurityException {
-
-        if (token != null) {
-            User user = loginService.login(token);
-            if (user != null) {
-                request.getSession().setAttribute(USER_ID_KEY, user.getId());
-                LogAction.login(user.getId(), ServletHelper.retrieveRemoteAddress(request));
-                return user;
-            }
-        }
-
-        Long userId = (Long) request.getSession().getAttribute(USER_ID_KEY);
-        if (userId == null) {
-
-            Cookie[] cookies = request.getCookies();
-            String email = null, password = null;
-            if (cookies != null) {
-                for (Cookie cookie : cookies) {
-                    if (cookie.getName().equals(USER_COOKIE_KEY)) {
-                        byte[] emailBytes = DataConverter.parseBase64(
-                                URLDecoder.decode(cookie.getValue(), StandardCharsets.US_ASCII));
-                        email = new String(emailBytes, StandardCharsets.UTF_8);
-                    } else if (cookie.getName().equals(PASS_COOKIE_KEY)) {
-                        byte[] passwordBytes = DataConverter.parseBase64(
-                                URLDecoder.decode(cookie.getValue(), StandardCharsets.US_ASCII));
-                        password = new String(passwordBytes, StandardCharsets.UTF_8);
-                    }
-                }
-            }
-            if (email != null && password != null) {
-                User user = loginService.login(email, password);
-                if (user != null) {
-                    request.getSession().setAttribute(USER_ID_KEY, user.getId());
-                    LogAction.login(user.getId(), ServletHelper.retrieveRemoteAddress(request));
-                    return user;
-                }
-            }
-
-        } else {
-
-            User user = permissionsService.getUser(userId);
-            if (user != null) {
-                return user;
-            }
-
-        }
-
-        throw new WebApplicationException(Response.status(Response.Status.NOT_FOUND).build());
-    }
-
-    @Path("{id}")
-    @GET
-    public User get(@PathParam("id") long userId) throws StorageException {
-        permissionsService.checkUser(getUserId(), userId);
-        User user = storage.getObject(User.class, new Request(
-                new Columns.All(), new Condition.Equals("id", userId)));
-        request.getSession().setAttribute(USER_ID_KEY, user.getId());
-        LogAction.login(user.getId(), ServletHelper.retrieveRemoteAddress(request));
-        return user;
-    }
-
-    @PermitAll
-    @POST
-    public User add(
-            @FormParam("email") String email, @FormParam("password") String password) throws StorageException {
-        User user = loginService.login(email, password);
-        if (user != null) {
-            request.getSession().setAttribute(USER_ID_KEY, user.getId());
-            LogAction.login(user.getId(), ServletHelper.retrieveRemoteAddress(request));
-            return user;
-        } else {
-            LogAction.failedLogin(ServletHelper.retrieveRemoteAddress(request));
-            throw new WebApplicationException(Response.status(Response.Status.UNAUTHORIZED).build());
-        }
-    }
-
-    @DELETE
-    public Response remove() {
-        LogAction.logout(getUserId(), ServletHelper.retrieveRemoteAddress(request));
-        request.getSession().removeAttribute(USER_ID_KEY);
-        return Response.noContent().build();
-    }
-
-    @Path("token")
-    @POST
-    public String requestToken(
-            @FormParam("expiration") Date expiration) throws StorageException, GeneralSecurityException, IOException {
-        return tokenManager.generateToken(getUserId(), expiration);
-    }
-
-    @PermitAll
-    @Path("openid/auth")
-    @GET
-    public Response openIdAuth() throws IOException {
-        return Response.seeOther(openIdProvider.createAuthUri()).build();
-    }
-    
-    @PermitAll
-    @Path("openid/callback")
-    @GET
-    public Response requestToken() throws IOException, StorageException, ParseException, GeneralSecurityException {
-        StringBuilder requestUrl = new StringBuilder(request.getRequestURL().toString());
-        String queryString = request.getQueryString();
-        String requestUri = requestUrl.append('?').append(queryString).toString();
-
-        return Response.seeOther(openIdProvider.handleCallback(URI.create(requestUri), request)).build();
-    }
-}
+/*

+ * Copyright 2015 - 2022 Anton Tananaev (anton@traccar.org)

+ *

+ * Licensed under the Apache License, Version 2.0 (the "License");

+ * you may not use this file except in compliance with the License.

+ * You may obtain a copy of the License at

+ *

+ *     http://www.apache.org/licenses/LICENSE-2.0

+ *

+ * Unless required by applicable law or agreed to in writing, software

+ * distributed under the License is distributed on an "AS IS" BASIS,

+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.

+ * See the License for the specific language governing permissions and

+ * limitations under the License.

+ */

+package org.traccar.api.resource;

+

+import org.traccar.api.BaseResource;

+import org.traccar.api.security.LoginService;

+import org.traccar.api.signature.TokenManager;

+import org.traccar.database.OpenIdProvider;

+import org.traccar.helper.DataConverter;

+import org.traccar.helper.LogAction;

+import org.traccar.helper.ServletHelper;

+import org.traccar.model.User;

+import org.traccar.storage.StorageException;

+import org.traccar.storage.query.Columns;

+import org.traccar.storage.query.Condition;

+import org.traccar.storage.query.Request;

+

+import com.nimbusds.oauth2.sdk.ParseException;

+import javax.annotation.Nullable;

+import javax.annotation.security.PermitAll;

+import javax.inject.Inject;

+import javax.servlet.http.Cookie;

+import javax.servlet.http.HttpServletRequest;

+import javax.ws.rs.Consumes;

+import javax.ws.rs.DELETE;

+import javax.ws.rs.FormParam;

+import javax.ws.rs.GET;

+import javax.ws.rs.POST;

+import javax.ws.rs.Path;

+import javax.ws.rs.PathParam;

+import javax.ws.rs.Produces;

+import javax.ws.rs.QueryParam;

+import javax.ws.rs.WebApplicationException;

+import javax.ws.rs.core.Context;

+import javax.ws.rs.core.MediaType;

+import javax.ws.rs.core.Response;

+import java.io.IOException;

+import java.net.URLDecoder;

+import java.nio.charset.StandardCharsets;

+import java.security.GeneralSecurityException;

+import java.util.Date;

+import java.net.URI;

+

+@Path("session")

+@Produces(MediaType.APPLICATION_JSON)

+@Consumes(MediaType.APPLICATION_FORM_URLENCODED)

+public class SessionResource extends BaseResource {

+

+    public static final String USER_ID_KEY = "userId";

+    public static final String USER_COOKIE_KEY = "user";

+    public static final String PASS_COOKIE_KEY = "password";

+

+    @Inject

+    private LoginService loginService;

+

+    @Inject

+    @Nullable

+    private OpenIdProvider openIdProvider;

+

+    @Inject

+    private TokenManager tokenManager;

+

+    @Context

+    private HttpServletRequest request;

+

+    @PermitAll

+    @GET

+    public User get(@QueryParam("token") String token) throws StorageException, IOException, GeneralSecurityException {

+

+        if (token != null) {

+            User user = loginService.login(token);

+            if (user != null) {

+                request.getSession().setAttribute(USER_ID_KEY, user.getId());

+                LogAction.login(user.getId(), ServletHelper.retrieveRemoteAddress(request));

+                return user;

+            }

+        }

+

+        Long userId = (Long) request.getSession().getAttribute(USER_ID_KEY);

+        if (userId == null) {

+

+            Cookie[] cookies = request.getCookies();

+            String email = null, password = null;

+            if (cookies != null) {

+                for (Cookie cookie : cookies) {

+                    if (cookie.getName().equals(USER_COOKIE_KEY)) {

+                        byte[] emailBytes = DataConverter.parseBase64(

+                                URLDecoder.decode(cookie.getValue(), StandardCharsets.US_ASCII));

+                        email = new String(emailBytes, StandardCharsets.UTF_8);

+                    } else if (cookie.getName().equals(PASS_COOKIE_KEY)) {

+                        byte[] passwordBytes = DataConverter.parseBase64(

+                                URLDecoder.decode(cookie.getValue(), StandardCharsets.US_ASCII));

+                        password = new String(passwordBytes, StandardCharsets.UTF_8);

+                    }

+                }

+            }

+            if (email != null && password != null) {

+                User user = loginService.login(email, password);

+                if (user != null) {

+                    request.getSession().setAttribute(USER_ID_KEY, user.getId());

+                    LogAction.login(user.getId(), ServletHelper.retrieveRemoteAddress(request));

+                    return user;

+                }

+            }

+

+        } else {

+

+            User user = permissionsService.getUser(userId);

+            if (user != null) {

+                return user;

+            }

+

+        }

+

+        throw new WebApplicationException(Response.status(Response.Status.NOT_FOUND).build());

+    }

+

+    @Path("{id}")

+    @GET

+    public User get(@PathParam("id") long userId) throws StorageException {

+        permissionsService.checkUser(getUserId(), userId);

+        User user = storage.getObject(User.class, new Request(

+                new Columns.All(), new Condition.Equals("id", userId)));

+        request.getSession().setAttribute(USER_ID_KEY, user.getId());

+        LogAction.login(user.getId(), ServletHelper.retrieveRemoteAddress(request));

+        return user;

+    }

+

+    @PermitAll

+    @POST

+    public User add(

+            @FormParam("email") String email, @FormParam("password") String password) throws StorageException {

+        User user = loginService.login(email, password);

+        if (user != null) {

+            request.getSession().setAttribute(USER_ID_KEY, user.getId());

+            LogAction.login(user.getId(), ServletHelper.retrieveRemoteAddress(request));

+            return user;

+        } else {

+            LogAction.failedLogin(ServletHelper.retrieveRemoteAddress(request));

+            throw new WebApplicationException(Response.status(Response.Status.UNAUTHORIZED).build());

+        }

+    }

+

+    @DELETE

+    public Response remove() {

+        LogAction.logout(getUserId(), ServletHelper.retrieveRemoteAddress(request));

+        request.getSession().removeAttribute(USER_ID_KEY);

+        return Response.noContent().build();

+    }

+

+    @Path("token")

+    @POST

+    public String requestToken(

+            @FormParam("expiration") Date expiration) throws StorageException, GeneralSecurityException, IOException {

+        return tokenManager.generateToken(getUserId(), expiration);

+    }

+

+    @PermitAll

+    @Path("openid/auth")

+    @GET

+    public Response openIdAuth() throws IOException {

+        return Response.seeOther(openIdProvider.createAuthUri()).build();

+    }

+    

+    @PermitAll

+    @Path("openid/callback")

+    @GET

+    public Response requestToken() throws IOException, StorageException, ParseException, GeneralSecurityException {

+        StringBuilder requestUrl = new StringBuilder(request.getRequestURL().toString());

+        String queryString = request.getQueryString();

+        String requestUri = requestUrl.append('?').append(queryString).toString();

+

+        return Response.seeOther(openIdProvider.handleCallback(URI.create(requestUri), request)).build();

+    }

+}

diff --git a/src/main/java/org/traccar/database/OpenIdProvider.java b/src/main/java/org/traccar/database/OpenIdProvider.java
index 6f44d0e80..22e5d6b50 100644
--- a/src/main/java/org/traccar/database/OpenIdProvider.java
+++ b/src/main/java/org/traccar/database/OpenIdProvider.java
@@ -1,170 +1,163 @@
-/*
- * Copyright 2023 Daniel Raper (me@danr.uk)
- *
- * Licensed under the Apache License, Version 2.0 (the "License");
- * you may not use this file except in compliance with the License.
- * You may obtain a copy of the License at
- *
- *     http://www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing, software
- * distributed under the License is distributed on an "AS IS" BASIS,
- * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- * See the License for the specific language governing permissions and
- * limitations under the License.
- */
-package org.traccar.database;
-
-import org.traccar.config.Config;
-import org.traccar.config.Keys;
-import org.traccar.api.resource.SessionResource;
-import org.traccar.api.security.LoginService;
-import org.traccar.model.User;
-import org.traccar.storage.StorageException;
-import org.traccar.helper.LogAction;
-import org.traccar.helper.ServletHelper;
-
-import java.net.URI;
-import java.net.URISyntaxException;
-import java.security.GeneralSecurityException;
-import java.io.IOException;
-import javax.servlet.http.HttpServletRequest;
-import com.google.inject.Inject;
-import org.slf4j.Logger;
-import org.slf4j.LoggerFactory;
-
-import com.nimbusds.oauth2.sdk.http.HTTPResponse;
-import com.nimbusds.oauth2.sdk.AuthorizationCode;
-import com.nimbusds.oauth2.sdk.ResponseType;
-import com.nimbusds.oauth2.sdk.Scope;
-import com.nimbusds.oauth2.sdk.AuthorizationGrant;
-import com.nimbusds.oauth2.sdk.TokenRequest;
-import com.nimbusds.oauth2.sdk.TokenResponse;
-import com.nimbusds.oauth2.sdk.AuthorizationCodeGrant;
-import com.nimbusds.oauth2.sdk.ParseException;
-import com.nimbusds.oauth2.sdk.AuthorizationResponse;
-import com.nimbusds.oauth2.sdk.auth.Secret;
-import com.nimbusds.oauth2.sdk.auth.ClientSecretBasic;
-import com.nimbusds.oauth2.sdk.auth.ClientAuthentication;
-import com.nimbusds.oauth2.sdk.token.BearerAccessToken;
-import com.nimbusds.oauth2.sdk.id.State;
-import com.nimbusds.oauth2.sdk.id.ClientID;
-import com.nimbusds.openid.connect.sdk.OIDCTokenResponse;
-import com.nimbusds.openid.connect.sdk.Nonce;
-import com.nimbusds.openid.connect.sdk.OIDCTokenResponseParser;
-import com.nimbusds.openid.connect.sdk.UserInfoResponse;
-import com.nimbusds.openid.connect.sdk.UserInfoRequest;
-import com.nimbusds.openid.connect.sdk.AuthenticationRequest;
-
-import com.nimbusds.openid.connect.sdk.claims.UserInfo;
-
-public class OpenIdProvider {
-    private static final Logger LOGGER = LoggerFactory.getLogger(OpenIdProvider.class);
-    
-    public final Boolean force;
-    private final ClientID clientId;
-    private final ClientAuthentication clientAuth;
-    private URI callbackUrl;
-    private URI authUrl;
-    private URI tokenUrl;
-    private URI userInfoUrl;
-    private URI baseUrl;
-    private final String adminGroup;
-
-    private LoginService loginService;
-
-    @Inject
-    public OpenIdProvider(Config config, LoginService loginService) {     
-        this.loginService = loginService;
-
-        force = config.getBoolean(Keys.OPENID_FORCE);
-        clientId = new ClientID(config.getString(Keys.OPENID_CLIENTID));
-        clientAuth = new ClientSecretBasic(clientId, new Secret(config.getString(Keys.OPENID_CLIENTSECRET)));
-
-        try {
-            callbackUrl = new URI(config.getString(Keys.WEB_URL, "") + "/api/session/openid/callback");
-            authUrl = new URI(config.getString(Keys.OPENID_AUTHURL, ""));
-            tokenUrl = new URI(config.getString(Keys.OPENID_TOKENURL, ""));
-            userInfoUrl = new URI(config.getString(Keys.OPENID_USERINFOURL, ""));
-            baseUrl = new URI(config.getString(Keys.WEB_URL, ""));
-        } catch(URISyntaxException error) {
-            LOGGER.error("Invalid URIs provided in OpenID configuration");
-        }
-
-        adminGroup = config.getString(Keys.OPENID_ADMINGROUP);
-    }
-
-    public URI createAuthUri() {
-        AuthenticationRequest request = new AuthenticationRequest.Builder(
-                new ResponseType("code"),
-                new Scope("openid", "profile", "email", "groups"),
-                clientId,
-                callbackUrl)
-                .endpointURI(authUrl)
-                .state(new State())
-                .nonce(new Nonce())
-                .build();
-
-        return request.toURI();
-    }
-
-    private OIDCTokenResponse getToken(AuthorizationCode code) throws IOException, ParseException, GeneralSecurityException {
-        AuthorizationGrant codeGrant = new AuthorizationCodeGrant(code, callbackUrl);
-        TokenRequest tokenRequest = new TokenRequest(tokenUrl, clientAuth, codeGrant);
-
-        HTTPResponse tokenReq = tokenRequest.toHTTPRequest().send();
-        TokenResponse tokenResponse = OIDCTokenResponseParser.parse(tokenReq);
-        if (!tokenResponse.indicatesSuccess()) {
-            LOGGER.warn("Invalid authorization code provided to OpenID callback");
-            throw new GeneralSecurityException("Unable to authenticate with the OpenID Connect provider.");
-        }
-
-        return (OIDCTokenResponse) tokenResponse.toSuccessResponse();
-    }
-
-    private UserInfo getUserInfo(BearerAccessToken token) throws IOException, ParseException, GeneralSecurityException {
-        UserInfoResponse userInfoResponse;
-
-        HTTPResponse httpResponse = new UserInfoRequest(userInfoUrl, token)
-                .toHTTPRequest()
-                .send();
-
-        userInfoResponse = UserInfoResponse.parse(httpResponse);
-
-        if (!userInfoResponse.indicatesSuccess()) {
-            LOGGER.error("Failed to access OpenID user info endpoint");
-            throw new GeneralSecurityException("Failed to access OpenID Connect user info endpoint. Please contact your administrator.");
-        }
-
-        return userInfoResponse.toSuccessResponse().getUserInfo();
-    }
-
-    public URI handleCallback(URI requestUri, HttpServletRequest request) throws StorageException, ParseException, IOException, GeneralSecurityException {
-        AuthorizationResponse response = AuthorizationResponse.parse(requestUri);
-
-        if (!response.indicatesSuccess()) {
-            LOGGER.warn("Callback received error response from OpenID identity provider");
-            throw new GeneralSecurityException(response.toErrorResponse().getErrorObject().getDescription());
-        }
-
-        AuthorizationCode authCode = response.toSuccessResponse().getAuthorizationCode();
-
-        if (authCode == null) {
-            LOGGER.warn("Malformed OpenID callback");
-            throw new GeneralSecurityException( "Malformed OpenID callback.");
-        }
-
-        OIDCTokenResponse tokens = getToken(authCode);
-
-        BearerAccessToken bearerToken = tokens.getOIDCTokens().getBearerAccessToken();
-
-        UserInfo userInfo = getUserInfo(bearerToken);
-
-        User user = loginService.login(userInfo.getEmailAddress(), userInfo.getName(), userInfo.getStringListClaim("groups").contains(adminGroup));
-
-        request.getSession().setAttribute(SessionResource.USER_ID_KEY, user.getId());
-        LogAction.login(user.getId(), ServletHelper.retrieveRemoteAddress(request));
-
-        return baseUrl;
-    }
-}
+/*

+ * Copyright 2023 Daniel Raper (me@danr.uk)

+ *

+ * Licensed under the Apache License, Version 2.0 (the "License");

+ * you may not use this file except in compliance with the License.

+ * You may obtain a copy of the License at

+ *

+ *     http://www.apache.org/licenses/LICENSE-2.0

+ *

+ * Unless required by applicable law or agreed to in writing, software

+ * distributed under the License is distributed on an "AS IS" BASIS,

+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.

+ * See the License for the specific language governing permissions and

+ * limitations under the License.

+ */

+package org.traccar.database;

+

+import org.traccar.config.Config;

+import org.traccar.config.Keys;

+import org.traccar.api.resource.SessionResource;

+import org.traccar.api.security.LoginService;

+import org.traccar.model.User;

+import org.traccar.storage.StorageException;

+import org.traccar.helper.LogAction;

+import org.traccar.helper.ServletHelper;

+

+import java.net.URI;

+import java.net.URISyntaxException;

+import java.security.GeneralSecurityException;

+import java.io.IOException;

+import javax.servlet.http.HttpServletRequest;

+import com.google.inject.Inject;

+import org.slf4j.Logger;

+import org.slf4j.LoggerFactory;

+

+import com.nimbusds.oauth2.sdk.http.HTTPResponse;

+import com.nimbusds.oauth2.sdk.AuthorizationCode;

+import com.nimbusds.oauth2.sdk.ResponseType;

+import com.nimbusds.oauth2.sdk.Scope;

+import com.nimbusds.oauth2.sdk.AuthorizationGrant;

+import com.nimbusds.oauth2.sdk.TokenRequest;

+import com.nimbusds.oauth2.sdk.TokenResponse;

+import com.nimbusds.oauth2.sdk.AuthorizationCodeGrant;

+import com.nimbusds.oauth2.sdk.ParseException;

+import com.nimbusds.oauth2.sdk.AuthorizationResponse;

+import com.nimbusds.oauth2.sdk.auth.Secret;

+import com.nimbusds.oauth2.sdk.auth.ClientSecretBasic;

+import com.nimbusds.oauth2.sdk.auth.ClientAuthentication;

+import com.nimbusds.oauth2.sdk.token.BearerAccessToken;

+import com.nimbusds.oauth2.sdk.id.State;

+import com.nimbusds.oauth2.sdk.id.ClientID;

+import com.nimbusds.openid.connect.sdk.OIDCTokenResponse;

+import com.nimbusds.openid.connect.sdk.Nonce;

+import com.nimbusds.openid.connect.sdk.OIDCTokenResponseParser;

+import com.nimbusds.openid.connect.sdk.UserInfoResponse;

+import com.nimbusds.openid.connect.sdk.UserInfoRequest;

+import com.nimbusds.openid.connect.sdk.AuthenticationRequest;

+

+import com.nimbusds.openid.connect.sdk.claims.UserInfo;

+

+public class OpenIdProvider {

+    private static final Logger LOGGER = LoggerFactory.getLogger(OpenIdProvider.class);

+    

+    public final Boolean force;

+    private final ClientID clientId;

+    private final ClientAuthentication clientAuth;

+    private URI callbackUrl;

+    private URI authUrl;

+    private URI tokenUrl;

+    private URI userInfoUrl;

+    private URI baseUrl;

+    private final String adminGroup;

+

+    private LoginService loginService;

+

+    @Inject

+    public OpenIdProvider(Config config, LoginService loginService) {     

+        this.loginService = loginService;

+

+        force = config.getBoolean(Keys.OPENID_FORCE);

+        clientId = new ClientID(config.getString(Keys.OPENID_CLIENTID));

+        clientAuth = new ClientSecretBasic(clientId, new Secret(config.getString(Keys.OPENID_CLIENTSECRET)));

+

+        try {

+            callbackUrl = new URI(config.getString(Keys.WEB_URL, "") + "/api/session/openid/callback");

+            authUrl = new URI(config.getString(Keys.OPENID_AUTHURL, ""));

+            tokenUrl = new URI(config.getString(Keys.OPENID_TOKENURL, ""));

+            userInfoUrl = new URI(config.getString(Keys.OPENID_USERINFOURL, ""));

+            baseUrl = new URI(config.getString(Keys.WEB_URL, ""));

+        } catch(URISyntaxException error) {

+            LOGGER.error("Invalid URIs provided in OpenID configuration");

+        }

+

+        adminGroup = config.getString(Keys.OPENID_ADMINGROUP);

+    }

+

+    public URI createAuthUri() {

+        AuthenticationRequest.Builder request = new AuthenticationRequest.Builder(

+                new ResponseType("code"),

+                new Scope("openid", "profile", "email", "groups"),

+                clientId,

+                callbackUrl);

+                

+        return request.endpointURI(authUrl)

+                .state(new State())

+                .build()

+                .toURI();

+    }

+

+    private OIDCTokenResponse getToken(AuthorizationCode code) throws IOException, ParseException, GeneralSecurityException {

+        AuthorizationGrant codeGrant = new AuthorizationCodeGrant(code, callbackUrl);

+        TokenRequest tokenRequest = new TokenRequest(tokenUrl, clientAuth, codeGrant);

+

+        HTTPResponse tokenResponse = tokenRequest.toHTTPRequest().send();

+        TokenResponse token = OIDCTokenResponseParser.parse(tokenResponse);

+        if (!token.indicatesSuccess()) {

+            throw new GeneralSecurityException("Unable to authenticate with the OpenID Connect provider.");

+        }

+

+        return (OIDCTokenResponse) token.toSuccessResponse();

+    }

+

+    private UserInfo getUserInfo(BearerAccessToken token) throws IOException, ParseException, GeneralSecurityException {

+        HTTPResponse httpResponse = new UserInfoRequest(userInfoUrl, token)

+                .toHTTPRequest()

+                .send();

+

+        UserInfoResponse userInfoResponse = UserInfoResponse.parse(httpResponse);

+

+        if (!userInfoResponse.indicatesSuccess()) {

+            throw new GeneralSecurityException("Failed to access OpenID Connect user info endpoint. Please contact your administrator.");

+        }

+

+        return userInfoResponse.toSuccessResponse().getUserInfo();

+    }

+

+    public URI handleCallback(URI requestUri, HttpServletRequest request) throws StorageException, ParseException, IOException, GeneralSecurityException {

+        AuthorizationResponse response = AuthorizationResponse.parse(requestUri);

+

+        if (!response.indicatesSuccess()) {

+            throw new GeneralSecurityException(response.toErrorResponse().getErrorObject().getDescription());

+        }

+

+        AuthorizationCode authCode = response.toSuccessResponse().getAuthorizationCode();

+

+        if (authCode == null) {

+            throw new GeneralSecurityException( "Malformed OpenID callback.");

+        }

+

+        OIDCTokenResponse tokens = getToken(authCode);

+

+        BearerAccessToken bearerToken = tokens.getOIDCTokens().getBearerAccessToken();

+

+        UserInfo userInfo = getUserInfo(bearerToken);

+

+        User user = loginService.login(userInfo.getEmailAddress(), userInfo.getName(), userInfo.getStringListClaim("groups").contains(adminGroup));

+

+        request.getSession().setAttribute(SessionResource.USER_ID_KEY, user.getId());

+        LogAction.login(user.getId(), ServletHelper.retrieveRemoteAddress(request));

+

+        return baseUrl;

+    }

+}