314 lines
		
	
	
		
			10 KiB
		
	
	
	
		
			PHP
		
	
	
	
			
		
		
	
	
			314 lines
		
	
	
		
			10 KiB
		
	
	
	
		
			PHP
		
	
	
	
| <?php
 | |
| 
 | |
| namespace BookStack\Access\Oidc;
 | |
| 
 | |
| use BookStack\Access\GroupSyncService;
 | |
| use BookStack\Access\LoginService;
 | |
| use BookStack\Access\RegistrationService;
 | |
| use BookStack\Exceptions\JsonDebugException;
 | |
| use BookStack\Exceptions\StoppedAuthenticationException;
 | |
| use BookStack\Exceptions\UserRegistrationException;
 | |
| use BookStack\Facades\Theme;
 | |
| use BookStack\Http\HttpRequestService;
 | |
| use BookStack\Theming\ThemeEvents;
 | |
| use BookStack\Users\Models\User;
 | |
| use Illuminate\Support\Facades\Cache;
 | |
| use League\OAuth2\Client\OptionProvider\HttpBasicAuthOptionProvider;
 | |
| use League\OAuth2\Client\Provider\Exception\IdentityProviderException;
 | |
| 
 | |
| /**
 | |
|  * Class OpenIdConnectService
 | |
|  * Handles any app-specific OIDC tasks.
 | |
|  */
 | |
| class OidcService
 | |
| {
 | |
|     public function __construct(
 | |
|         protected RegistrationService $registrationService,
 | |
|         protected LoginService $loginService,
 | |
|         protected HttpRequestService $http,
 | |
|         protected GroupSyncService $groupService
 | |
|     ) {
 | |
|     }
 | |
| 
 | |
|     /**
 | |
|      * Initiate an authorization flow.
 | |
|      * Provides back an authorize redirect URL, in addition to other
 | |
|      * details which may be required for the auth flow.
 | |
|      *
 | |
|      * @throws OidcException
 | |
|      *
 | |
|      * @return array{url: string, state: string}
 | |
|      */
 | |
|     public function login(): array
 | |
|     {
 | |
|         $settings = $this->getProviderSettings();
 | |
|         $provider = $this->getProvider($settings);
 | |
| 
 | |
|         $url = $provider->getAuthorizationUrl();
 | |
|         session()->put('oidc_pkce_code', $provider->getPkceCode() ?? '');
 | |
| 
 | |
|         return [
 | |
|             'url'   => $url,
 | |
|             'state' => $provider->getState(),
 | |
|         ];
 | |
|     }
 | |
| 
 | |
|     /**
 | |
|      * Process the Authorization response from the authorization server and
 | |
|      * return the matching, or new if registration active, user matched to the
 | |
|      * authorization server. Throws if the user cannot be auth if not authenticated.
 | |
|      *
 | |
|      * @throws JsonDebugException
 | |
|      * @throws OidcException
 | |
|      * @throws StoppedAuthenticationException
 | |
|      * @throws IdentityProviderException
 | |
|      */
 | |
|     public function processAuthorizeResponse(?string $authorizationCode): User
 | |
|     {
 | |
|         $settings = $this->getProviderSettings();
 | |
|         $provider = $this->getProvider($settings);
 | |
| 
 | |
|         // Set PKCE code flashed at login
 | |
|         $pkceCode = session()->pull('oidc_pkce_code', '');
 | |
|         $provider->setPkceCode($pkceCode);
 | |
| 
 | |
|         // Try to exchange authorization code for access token
 | |
|         $accessToken = $provider->getAccessToken('authorization_code', [
 | |
|             'code' => $authorizationCode,
 | |
|         ]);
 | |
| 
 | |
|         return $this->processAccessTokenCallback($accessToken, $settings);
 | |
|     }
 | |
| 
 | |
|     /**
 | |
|      * @throws OidcException
 | |
|      */
 | |
|     protected function getProviderSettings(): OidcProviderSettings
 | |
|     {
 | |
|         $config = $this->config();
 | |
|         $settings = new OidcProviderSettings([
 | |
|             'issuer'                => $config['issuer'],
 | |
|             'clientId'              => $config['client_id'],
 | |
|             'clientSecret'          => $config['client_secret'],
 | |
|             'authorizationEndpoint' => $config['authorization_endpoint'],
 | |
|             'tokenEndpoint'         => $config['token_endpoint'],
 | |
|             'endSessionEndpoint'    => is_string($config['end_session_endpoint']) ? $config['end_session_endpoint'] : null,
 | |
|             'userinfoEndpoint'      => $config['userinfo_endpoint'],
 | |
|         ]);
 | |
| 
 | |
|         // Use keys if configured
 | |
|         if (!empty($config['jwt_public_key'])) {
 | |
|             $settings->keys = [$config['jwt_public_key']];
 | |
|         }
 | |
| 
 | |
|         // Run discovery
 | |
|         if ($config['discover'] ?? false) {
 | |
|             try {
 | |
|                 $settings->discoverFromIssuer($this->http->buildClient(5), Cache::store(null), 15);
 | |
|             } catch (OidcIssuerDiscoveryException $exception) {
 | |
|                 throw new OidcException('OIDC Discovery Error: ' . $exception->getMessage());
 | |
|             }
 | |
|         }
 | |
| 
 | |
|         // Prevent use of RP-initiated logout if specifically disabled
 | |
|         // Or force use of a URL if specifically set.
 | |
|         if ($config['end_session_endpoint'] === false) {
 | |
|             $settings->endSessionEndpoint = null;
 | |
|         } else if (is_string($config['end_session_endpoint'])) {
 | |
|             $settings->endSessionEndpoint = $config['end_session_endpoint'];
 | |
|         }
 | |
| 
 | |
|         $settings->validate();
 | |
| 
 | |
|         return $settings;
 | |
|     }
 | |
| 
 | |
|     /**
 | |
|      * Load the underlying OpenID Connect Provider.
 | |
|      */
 | |
|     protected function getProvider(OidcProviderSettings $settings): OidcOAuthProvider
 | |
|     {
 | |
|         $provider = new OidcOAuthProvider([
 | |
|             ...$settings->arrayForOAuthProvider(),
 | |
|             'redirectUri' => url('/oidc/callback'),
 | |
|         ], [
 | |
|             'httpClient'     => $this->http->buildClient(5),
 | |
|             'optionProvider' => new HttpBasicAuthOptionProvider(),
 | |
|         ]);
 | |
| 
 | |
|         foreach ($this->getAdditionalScopes() as $scope) {
 | |
|             $provider->addScope($scope);
 | |
|         }
 | |
| 
 | |
|         return $provider;
 | |
|     }
 | |
| 
 | |
|     /**
 | |
|      * Get any user-defined addition/custom scopes to apply to the authentication request.
 | |
|      *
 | |
|      * @return string[]
 | |
|      */
 | |
|     protected function getAdditionalScopes(): array
 | |
|     {
 | |
|         $scopeConfig = $this->config()['additional_scopes'] ?: '';
 | |
| 
 | |
|         $scopeArr = explode(',', $scopeConfig);
 | |
|         $scopeArr = array_map(fn (string $scope) => trim($scope), $scopeArr);
 | |
| 
 | |
|         return array_filter($scopeArr);
 | |
|     }
 | |
| 
 | |
|     /**
 | |
|      * Processes a received access token for a user. Login the user when
 | |
|      * they exist, optionally registering them automatically.
 | |
|      *
 | |
|      * @throws OidcException
 | |
|      * @throws JsonDebugException
 | |
|      * @throws StoppedAuthenticationException
 | |
|      */
 | |
|     protected function processAccessTokenCallback(OidcAccessToken $accessToken, OidcProviderSettings $settings): User
 | |
|     {
 | |
|         $idTokenText = $accessToken->getIdToken();
 | |
|         $idToken = new OidcIdToken(
 | |
|             $idTokenText,
 | |
|             $settings->issuer,
 | |
|             $settings->keys,
 | |
|         );
 | |
| 
 | |
|         session()->put("oidc_id_token", $idTokenText);
 | |
| 
 | |
|         $returnClaims = Theme::dispatch(ThemeEvents::OIDC_ID_TOKEN_PRE_VALIDATE, $idToken->getAllClaims(), [
 | |
|             'access_token' => $accessToken->getToken(),
 | |
|             'expires_in' => $accessToken->getExpires(),
 | |
|             'refresh_token' => $accessToken->getRefreshToken(),
 | |
|         ]);
 | |
| 
 | |
|         if (!is_null($returnClaims)) {
 | |
|             $idToken->replaceClaims($returnClaims);
 | |
|         }
 | |
| 
 | |
|         if ($this->config()['dump_user_details']) {
 | |
|             throw new JsonDebugException($idToken->getAllClaims());
 | |
|         }
 | |
| 
 | |
|         try {
 | |
|             $idToken->validate($settings->clientId);
 | |
|         } catch (OidcInvalidTokenException $exception) {
 | |
|             throw new OidcException("ID token validation failed with error: {$exception->getMessage()}");
 | |
|         }
 | |
| 
 | |
|         $userDetails = $this->getUserDetailsFromToken($idToken, $accessToken, $settings);
 | |
|         if (empty($userDetails->email)) {
 | |
|             throw new OidcException(trans('errors.oidc_no_email_address'));
 | |
|         }
 | |
|         if (empty($userDetails->name)) {
 | |
|             $userDetails->name = $userDetails->externalId;
 | |
|         }
 | |
| 
 | |
|         $isLoggedIn = auth()->check();
 | |
|         if ($isLoggedIn) {
 | |
|             throw new OidcException(trans('errors.oidc_already_logged_in'));
 | |
|         }
 | |
| 
 | |
|         try {
 | |
|             $user = $this->registrationService->findOrRegister(
 | |
|                 $userDetails->name,
 | |
|                 $userDetails->email,
 | |
|                 $userDetails->externalId
 | |
|             );
 | |
|         } catch (UserRegistrationException $exception) {
 | |
|             throw new OidcException($exception->getMessage());
 | |
|         }
 | |
| 
 | |
|         if ($this->shouldSyncGroups()) {
 | |
|             $detachExisting = $this->config()['remove_from_groups'];
 | |
|             $this->groupService->syncUserWithFoundGroups($user, $userDetails->groups ?? [], $detachExisting);
 | |
|         }
 | |
| 
 | |
|         $this->loginService->login($user, 'oidc');
 | |
| 
 | |
|         return $user;
 | |
|     }
 | |
| 
 | |
|     /**
 | |
|      * @throws OidcException
 | |
|      */
 | |
|     protected function getUserDetailsFromToken(OidcIdToken $idToken, OidcAccessToken $accessToken, OidcProviderSettings $settings): OidcUserDetails
 | |
|     {
 | |
|         $userDetails = new OidcUserDetails();
 | |
|         $userDetails->populate(
 | |
|             $idToken,
 | |
|             $this->config()['external_id_claim'],
 | |
|             $this->config()['display_name_claims'] ?? '',
 | |
|             $this->config()['groups_claim'] ?? ''
 | |
|         );
 | |
| 
 | |
|         if (!$userDetails->isFullyPopulated($this->shouldSyncGroups()) && !empty($settings->userinfoEndpoint)) {
 | |
|             $provider = $this->getProvider($settings);
 | |
|             $request = $provider->getAuthenticatedRequest('GET', $settings->userinfoEndpoint, $accessToken->getToken());
 | |
|             $response = new OidcUserinfoResponse(
 | |
|                 $provider->getResponse($request),
 | |
|                 $settings->issuer,
 | |
|                 $settings->keys,
 | |
|             );
 | |
| 
 | |
|             try {
 | |
|                 $response->validate($idToken->getClaim('sub'), $settings->clientId);
 | |
|             } catch (OidcInvalidTokenException $exception) {
 | |
|                 throw new OidcException("Userinfo endpoint response validation failed with error: {$exception->getMessage()}");
 | |
|             }
 | |
| 
 | |
|             $userDetails->populate(
 | |
|                 $response,
 | |
|                 $this->config()['external_id_claim'],
 | |
|                 $this->config()['display_name_claims'] ?? '',
 | |
|                 $this->config()['groups_claim'] ?? ''
 | |
|             );
 | |
|         }
 | |
| 
 | |
|         return $userDetails;
 | |
|     }
 | |
| 
 | |
|     /**
 | |
|      * Get the OIDC config from the application.
 | |
|      */
 | |
|     protected function config(): array
 | |
|     {
 | |
|         return config('oidc');
 | |
|     }
 | |
| 
 | |
|     /**
 | |
|      * Check if groups should be synced.
 | |
|      */
 | |
|     protected function shouldSyncGroups(): bool
 | |
|     {
 | |
|         return $this->config()['user_to_groups'] !== false;
 | |
|     }
 | |
| 
 | |
|     /**
 | |
|      * Start the RP-initiated logout flow if active, otherwise start a standard logout flow.
 | |
|      * Returns a post-app-logout redirect URL.
 | |
|      * Reference: https://openid.net/specs/openid-connect-rpinitiated-1_0.html
 | |
|      * @throws OidcException
 | |
|      */
 | |
|     public function logout(): string
 | |
|     {
 | |
|         $oidcToken = session()->pull("oidc_id_token");
 | |
|         $defaultLogoutUrl = url($this->loginService->logout());
 | |
|         $oidcSettings = $this->getProviderSettings();
 | |
| 
 | |
|         if (!$oidcSettings->endSessionEndpoint) {
 | |
|             return $defaultLogoutUrl;
 | |
|         }
 | |
| 
 | |
|         $endpointParams = [
 | |
|             'id_token_hint' => $oidcToken,
 | |
|             'post_logout_redirect_uri' => $defaultLogoutUrl,
 | |
|         ];
 | |
| 
 | |
|         $joiner = str_contains($oidcSettings->endSessionEndpoint, '?') ? '&' : '?';
 | |
| 
 | |
|         return $oidcSettings->endSessionEndpoint . $joiner . http_build_query($endpointParams);
 | |
|     }
 | |
| }
 |