/* * Copyright (c) Microsoft Corporation. All rights reserved. * Licensed under the MIT License. */ import { AccountFilter, CredentialFilter, ValidCredentialType, AppMetadataFilter, AppMetadataCache, TokenKeys, TenantProfileFilter, } from "./utils/CacheTypes.js"; import { CacheRecord } from "./entities/CacheRecord.js"; import { CredentialType, APP_METADATA, THE_FAMILY_ID, AUTHORITY_METADATA_CONSTANTS, AuthenticationScheme, } from "../utils/Constants.js"; import { CredentialEntity } from "./entities/CredentialEntity.js"; import { ScopeSet } from "../request/ScopeSet.js"; import { AccountEntity } from "./entities/AccountEntity.js"; import { AccessTokenEntity } from "./entities/AccessTokenEntity.js"; import { IdTokenEntity } from "./entities/IdTokenEntity.js"; import { RefreshTokenEntity } from "./entities/RefreshTokenEntity.js"; import { ICacheManager } from "./interface/ICacheManager.js"; import { createClientAuthError, ClientAuthErrorCodes, } from "../error/ClientAuthError.js"; import { AccountInfo, TenantProfile, updateAccountTenantProfileData, } from "../account/AccountInfo.js"; import { AppMetadataEntity } from "./entities/AppMetadataEntity.js"; import { ServerTelemetryEntity } from "./entities/ServerTelemetryEntity.js"; import { ThrottlingEntity } from "./entities/ThrottlingEntity.js"; import { extractTokenClaims } from "../account/AuthToken.js"; import { ICrypto } from "../crypto/ICrypto.js"; import { AuthorityMetadataEntity } from "./entities/AuthorityMetadataEntity.js"; import { BaseAuthRequest } from "../request/BaseAuthRequest.js"; import { Logger } from "../logger/Logger.js"; import { name, version } from "../packageMetadata.js"; import { StoreInCache } from "../request/StoreInCache.js"; import { getAliasesFromStaticSources } from "../authority/AuthorityMetadata.js"; import { StaticAuthorityOptions } from "../authority/AuthorityOptions.js"; import { TokenClaims } from "../account/TokenClaims.js"; import { IPerformanceClient } from "../telemetry/performance/IPerformanceClient.js"; import { createCacheError } from "../error/CacheError.js"; import { AuthError } from "../error/AuthError.js"; /** * Interface class which implement cache storage functions used by MSAL to perform validity checks, and store tokens. * @internal */ export abstract class CacheManager implements ICacheManager { protected clientId: string; protected cryptoImpl: ICrypto; // Instance of logger for functions defined in the msal-common layer private commonLogger: Logger; private staticAuthorityOptions?: StaticAuthorityOptions; protected performanceClient: IPerformanceClient; constructor( clientId: string, cryptoImpl: ICrypto, logger: Logger, performanceClient: IPerformanceClient, staticAuthorityOptions?: StaticAuthorityOptions ) { this.clientId = clientId; this.cryptoImpl = cryptoImpl; this.commonLogger = logger.clone(name, version); this.staticAuthorityOptions = staticAuthorityOptions; this.performanceClient = performanceClient; } /** * fetch the account entity from the platform cache * @param accountKey */ abstract getAccount( accountKey: string, correlationId: string ): AccountEntity | null; /** * set account entity in the platform cache * @param account * @param correlationId */ abstract setAccount( account: AccountEntity, correlationId: string ): Promise; /** * fetch the idToken entity from the platform cache * @param idTokenKey */ abstract getIdTokenCredential( idTokenKey: string, correlationId: string ): IdTokenEntity | null; /** * set idToken entity to the platform cache * @param idToken * @param correlationId */ abstract setIdTokenCredential( idToken: IdTokenEntity, correlationId: string ): Promise; /** * fetch the idToken entity from the platform cache * @param accessTokenKey */ abstract getAccessTokenCredential( accessTokenKey: string, correlationId: string ): AccessTokenEntity | null; /** * set accessToken entity to the platform cache * @param accessToken * @param correlationId */ abstract setAccessTokenCredential( accessToken: AccessTokenEntity, correlationId: string ): Promise; /** * fetch the idToken entity from the platform cache * @param refreshTokenKey */ abstract getRefreshTokenCredential( refreshTokenKey: string, correlationId: string ): RefreshTokenEntity | null; /** * set refreshToken entity to the platform cache * @param refreshToken * @param correlationId */ abstract setRefreshTokenCredential( refreshToken: RefreshTokenEntity, correlationId: string ): Promise; /** * fetch appMetadata entity from the platform cache * @param appMetadataKey */ abstract getAppMetadata(appMetadataKey: string): AppMetadataEntity | null; /** * set appMetadata entity to the platform cache * @param appMetadata */ abstract setAppMetadata( appMetadata: AppMetadataEntity, correlationId: string ): void; /** * fetch server telemetry entity from the platform cache * @param serverTelemetryKey */ abstract getServerTelemetry( serverTelemetryKey: string ): ServerTelemetryEntity | null; /** * set server telemetry entity to the platform cache * @param serverTelemetryKey * @param serverTelemetry */ abstract setServerTelemetry( serverTelemetryKey: string, serverTelemetry: ServerTelemetryEntity, correlationId: string ): void; /** * fetch cloud discovery metadata entity from the platform cache * @param key */ abstract getAuthorityMetadata(key: string): AuthorityMetadataEntity | null; /** * */ abstract getAuthorityMetadataKeys(): Array; /** * set cloud discovery metadata entity to the platform cache * @param key * @param value */ abstract setAuthorityMetadata( key: string, value: AuthorityMetadataEntity ): void; /** * fetch throttling entity from the platform cache * @param throttlingCacheKey */ abstract getThrottlingCache( throttlingCacheKey: string ): ThrottlingEntity | null; /** * set throttling entity to the platform cache * @param throttlingCacheKey * @param throttlingCache */ abstract setThrottlingCache( throttlingCacheKey: string, throttlingCache: ThrottlingEntity, correlationId: string ): void; /** * Function to remove an item from cache given its key. * @param key */ abstract removeItem(key: string, correlationId: string): void; /** * Function which retrieves all current keys from the cache. */ abstract getKeys(): string[]; /** * Function which retrieves all account keys from the cache */ abstract getAccountKeys(): string[]; /** * Function which retrieves all token keys from the cache */ abstract getTokenKeys(): TokenKeys; /** * Returns credential cache key from the entity * @param credential */ abstract generateCredentialKey(credential: CredentialEntity): string; /** * Returns the account cache key from the account info * @param account */ abstract generateAccountKey(account: AccountInfo): string; /** * Returns all the accounts in the cache that match the optional filter. If no filter is provided, all accounts are returned. * @param accountFilter - (Optional) filter to narrow down the accounts returned * @returns Array of AccountInfo objects in cache */ getAllAccounts( accountFilter: AccountFilter, correlationId: string ): AccountInfo[] { return this.buildTenantProfiles( this.getAccountsFilteredBy(accountFilter, correlationId), correlationId, accountFilter ); } /** * Gets first tenanted AccountInfo object found based on provided filters */ getAccountInfoFilteredBy( accountFilter: AccountFilter, correlationId: string ): AccountInfo | null { const allAccounts = this.getAllAccounts(accountFilter, correlationId); if (allAccounts.length > 1) { // If one or more accounts are found, prioritize accounts that have an ID token const sortedAccounts = allAccounts.sort((account) => { return account.idTokenClaims ? -1 : 1; }); return sortedAccounts[0]; } else if (allAccounts.length === 1) { // If only one account is found, return it regardless of whether a matching ID token was found return allAccounts[0]; } else { return null; } } /** * Returns a single matching * @param accountFilter * @returns */ getBaseAccountInfo( accountFilter: AccountFilter, correlationId: string ): AccountInfo | null { const accountEntities = this.getAccountsFilteredBy( accountFilter, correlationId ); if (accountEntities.length > 0) { return accountEntities[0].getAccountInfo(); } else { return null; } } /** * Matches filtered account entities with cached ID tokens that match the tenant profile-specific account filters * and builds the account info objects from the matching ID token's claims * @param cachedAccounts * @param accountFilter * @returns Array of AccountInfo objects that match account and tenant profile filters */ private buildTenantProfiles( cachedAccounts: AccountEntity[], correlationId: string, accountFilter?: AccountFilter ): AccountInfo[] { return cachedAccounts.flatMap((accountEntity) => { return this.getTenantProfilesFromAccountEntity( accountEntity, correlationId, accountFilter?.tenantId, accountFilter ); }); } private getTenantedAccountInfoByFilter( accountInfo: AccountInfo, tokenKeys: TokenKeys, tenantProfile: TenantProfile, correlationId: string, tenantProfileFilter?: TenantProfileFilter ): AccountInfo | null { let tenantedAccountInfo: AccountInfo | null = null; let idTokenClaims: TokenClaims | undefined; if (tenantProfileFilter) { if ( !this.tenantProfileMatchesFilter( tenantProfile, tenantProfileFilter ) ) { return null; } } const idToken = this.getIdToken( accountInfo, correlationId, tokenKeys, tenantProfile.tenantId ); if (idToken) { idTokenClaims = extractTokenClaims( idToken.secret, this.cryptoImpl.base64Decode ); if ( !this.idTokenClaimsMatchTenantProfileFilter( idTokenClaims, tenantProfileFilter ) ) { // ID token sourced claims don't match so this tenant profile is not a match return null; } } // Expand tenant profile into account info based on matching tenant profile and if available matching ID token claims tenantedAccountInfo = updateAccountTenantProfileData( accountInfo, tenantProfile, idTokenClaims, idToken?.secret ); return tenantedAccountInfo; } private getTenantProfilesFromAccountEntity( accountEntity: AccountEntity, correlationId: string, targetTenantId?: string, tenantProfileFilter?: TenantProfileFilter ): AccountInfo[] { const accountInfo = accountEntity.getAccountInfo(); let searchTenantProfiles: Map = accountInfo.tenantProfiles || new Map(); const tokenKeys = this.getTokenKeys(); // If a tenant ID was provided, only return the tenant profile for that tenant ID if it exists if (targetTenantId) { const tenantProfile = searchTenantProfiles.get(targetTenantId); if (tenantProfile) { // Reduce search field to just this tenant profile searchTenantProfiles = new Map([ [targetTenantId, tenantProfile], ]); } else { // No tenant profile for search tenant ID, return empty array return []; } } const matchingTenantProfiles: AccountInfo[] = []; searchTenantProfiles.forEach((tenantProfile: TenantProfile) => { const tenantedAccountInfo = this.getTenantedAccountInfoByFilter( accountInfo, tokenKeys, tenantProfile, correlationId, tenantProfileFilter ); if (tenantedAccountInfo) { matchingTenantProfiles.push(tenantedAccountInfo); } }); return matchingTenantProfiles; } private tenantProfileMatchesFilter( tenantProfile: TenantProfile, tenantProfileFilter: TenantProfileFilter ): boolean { if ( !!tenantProfileFilter.localAccountId && !this.matchLocalAccountIdFromTenantProfile( tenantProfile, tenantProfileFilter.localAccountId ) ) { return false; } if ( !!tenantProfileFilter.name && !(tenantProfile.name === tenantProfileFilter.name) ) { return false; } if ( tenantProfileFilter.isHomeTenant !== undefined && !(tenantProfile.isHomeTenant === tenantProfileFilter.isHomeTenant) ) { return false; } return true; } private idTokenClaimsMatchTenantProfileFilter( idTokenClaims: TokenClaims, tenantProfileFilter?: TenantProfileFilter ): boolean { // Tenant Profile filtering if (tenantProfileFilter) { if ( !!tenantProfileFilter.localAccountId && !this.matchLocalAccountIdFromTokenClaims( idTokenClaims, tenantProfileFilter.localAccountId ) ) { return false; } if ( !!tenantProfileFilter.loginHint && !this.matchLoginHintFromTokenClaims( idTokenClaims, tenantProfileFilter.loginHint ) ) { return false; } if ( !!tenantProfileFilter.username && !this.matchUsername( idTokenClaims.preferred_username, tenantProfileFilter.username ) ) { return false; } if ( !!tenantProfileFilter.name && !this.matchName(idTokenClaims, tenantProfileFilter.name) ) { return false; } if ( !!tenantProfileFilter.sid && !this.matchSid(idTokenClaims, tenantProfileFilter.sid) ) { return false; } } return true; } /** * saves a cache record * @param cacheRecord {CacheRecord} * @param storeInCache {?StoreInCache} * @param correlationId {?string} correlation id */ async saveCacheRecord( cacheRecord: CacheRecord, correlationId: string, storeInCache?: StoreInCache ): Promise { if (!cacheRecord) { throw createClientAuthError( ClientAuthErrorCodes.invalidCacheRecord ); } try { if (!!cacheRecord.account) { await this.setAccount(cacheRecord.account, correlationId); } if (!!cacheRecord.idToken && storeInCache?.idToken !== false) { await this.setIdTokenCredential( cacheRecord.idToken, correlationId ); } if ( !!cacheRecord.accessToken && storeInCache?.accessToken !== false ) { await this.saveAccessToken( cacheRecord.accessToken, correlationId ); } if ( !!cacheRecord.refreshToken && storeInCache?.refreshToken !== false ) { await this.setRefreshTokenCredential( cacheRecord.refreshToken, correlationId ); } if (!!cacheRecord.appMetadata) { this.setAppMetadata(cacheRecord.appMetadata, correlationId); } } catch (e: unknown) { this.commonLogger?.error(`CacheManager.saveCacheRecord: failed`); if (e instanceof AuthError) { throw e; } else { throw createCacheError(e); } } } /** * saves access token credential * @param credential */ private async saveAccessToken( credential: AccessTokenEntity, correlationId: string ): Promise { const accessTokenFilter: CredentialFilter = { clientId: credential.clientId, credentialType: credential.credentialType, environment: credential.environment, homeAccountId: credential.homeAccountId, realm: credential.realm, tokenType: credential.tokenType, requestedClaimsHash: credential.requestedClaimsHash, }; const tokenKeys = this.getTokenKeys(); const currentScopes = ScopeSet.fromString(credential.target); tokenKeys.accessToken.forEach((key) => { if ( !this.accessTokenKeyMatchesFilter(key, accessTokenFilter, false) ) { return; } const tokenEntity = this.getAccessTokenCredential( key, correlationId ); if ( tokenEntity && this.credentialMatchesFilter(tokenEntity, accessTokenFilter) ) { const tokenScopeSet = ScopeSet.fromString(tokenEntity.target); if (tokenScopeSet.intersectingScopeSets(currentScopes)) { this.removeAccessToken(key, correlationId); } } }); await this.setAccessTokenCredential(credential, correlationId); } /** * Retrieve account entities matching all provided tenant-agnostic filters; if no filter is set, get all account entities in the cache * Not checking for casing as keys are all generated in lower case, remember to convert to lower case if object properties are compared * @param accountFilter - An object containing Account properties to filter by */ getAccountsFilteredBy( accountFilter: AccountFilter, correlationId: string ): AccountEntity[] { const allAccountKeys = this.getAccountKeys(); const matchingAccounts: AccountEntity[] = []; allAccountKeys.forEach((cacheKey) => { const entity: AccountEntity | null = this.getAccount( cacheKey, correlationId ); // Match base account fields if (!entity) { return; } if ( !!accountFilter.homeAccountId && !this.matchHomeAccountId(entity, accountFilter.homeAccountId) ) { return; } if ( !!accountFilter.username && !this.matchUsername(entity.username, accountFilter.username) ) { return; } if ( !!accountFilter.environment && !this.matchEnvironment(entity, accountFilter.environment) ) { return; } if ( !!accountFilter.realm && !this.matchRealm(entity, accountFilter.realm) ) { return; } if ( !!accountFilter.nativeAccountId && !this.matchNativeAccountId( entity, accountFilter.nativeAccountId ) ) { return; } if ( !!accountFilter.authorityType && !this.matchAuthorityType(entity, accountFilter.authorityType) ) { return; } // If at least one tenant profile matches the tenant profile filter, add the account to the list of matching accounts const tenantProfileFilter: TenantProfileFilter = { localAccountId: accountFilter?.localAccountId, name: accountFilter?.name, }; const matchingTenantProfiles = entity.tenantProfiles?.filter( (tenantProfile: TenantProfile) => { return this.tenantProfileMatchesFilter( tenantProfile, tenantProfileFilter ); } ); if (matchingTenantProfiles && matchingTenantProfiles.length === 0) { // No tenant profile for this account matches filter, don't add to list of matching accounts return; } matchingAccounts.push(entity); }); return matchingAccounts; } /** * Returns whether or not the given credential entity matches the filter * @param entity * @param filter * @returns */ credentialMatchesFilter( entity: ValidCredentialType, filter: CredentialFilter ): boolean { if (!!filter.clientId && !this.matchClientId(entity, filter.clientId)) { return false; } if ( !!filter.userAssertionHash && !this.matchUserAssertionHash(entity, filter.userAssertionHash) ) { return false; } /* * homeAccountId can be undefined, and we want to filter out cached items that have a homeAccountId of "" * because we don't want a client_credential request to return a cached token that has a homeAccountId */ if ( typeof filter.homeAccountId === "string" && !this.matchHomeAccountId(entity, filter.homeAccountId) ) { return false; } if ( !!filter.environment && !this.matchEnvironment(entity, filter.environment) ) { return false; } if (!!filter.realm && !this.matchRealm(entity, filter.realm)) { return false; } if ( !!filter.credentialType && !this.matchCredentialType(entity, filter.credentialType) ) { return false; } if (!!filter.familyId && !this.matchFamilyId(entity, filter.familyId)) { return false; } /* * idTokens do not have "target", target specific refreshTokens do exist for some types of authentication * Resource specific refresh tokens case will be added when the support is deemed necessary */ if (!!filter.target && !this.matchTarget(entity, filter.target)) { return false; } // If request OR cached entity has requested Claims Hash, check if they match if (filter.requestedClaimsHash || entity.requestedClaimsHash) { // Don't match if either is undefined or they are different if (entity.requestedClaimsHash !== filter.requestedClaimsHash) { return false; } } // Access Token with Auth Scheme specific matching if ( entity.credentialType === CredentialType.ACCESS_TOKEN_WITH_AUTH_SCHEME ) { if ( !!filter.tokenType && !this.matchTokenType(entity, filter.tokenType) ) { return false; } // KeyId (sshKid) in request must match cached SSH certificate keyId because SSH cert is bound to a specific key if (filter.tokenType === AuthenticationScheme.SSH) { if (filter.keyId && !this.matchKeyId(entity, filter.keyId)) { return false; } } } return true; } /** * retrieve appMetadata matching all provided filters; if no filter is set, get all appMetadata * @param filter */ getAppMetadataFilteredBy(filter: AppMetadataFilter): AppMetadataCache { const allCacheKeys = this.getKeys(); const matchingAppMetadata: AppMetadataCache = {}; allCacheKeys.forEach((cacheKey) => { // don't parse any non-appMetadata type cache entities if (!this.isAppMetadata(cacheKey)) { return; } // Attempt retrieval const entity = this.getAppMetadata(cacheKey); if (!entity) { return; } if ( !!filter.environment && !this.matchEnvironment(entity, filter.environment) ) { return; } if ( !!filter.clientId && !this.matchClientId(entity, filter.clientId) ) { return; } matchingAppMetadata[cacheKey] = entity; }); return matchingAppMetadata; } /** * retrieve authorityMetadata that contains a matching alias * @param filter */ getAuthorityMetadataByAlias(host: string): AuthorityMetadataEntity | null { const allCacheKeys = this.getAuthorityMetadataKeys(); let matchedEntity = null; allCacheKeys.forEach((cacheKey) => { // don't parse any non-authorityMetadata type cache entities if ( !this.isAuthorityMetadata(cacheKey) || cacheKey.indexOf(this.clientId) === -1 ) { return; } // Attempt retrieval const entity = this.getAuthorityMetadata(cacheKey); if (!entity) { return; } if (entity.aliases.indexOf(host) === -1) { return; } matchedEntity = entity; }); return matchedEntity; } /** * Removes all accounts and related tokens from cache. */ removeAllAccounts(correlationId: string): void { const accounts = this.getAllAccounts({}, correlationId); accounts.forEach((account) => { this.removeAccount(account, correlationId); }); } /** * Removes the account and related tokens for a given account key * @param account */ removeAccount(account: AccountInfo, correlationId: string): void { this.removeAccountContext(account, correlationId); const accountKeys = this.getAccountKeys(); const keyFilter = (key: string): boolean => { return ( key.includes(account.homeAccountId) && key.includes(account.environment) ); }; accountKeys.filter(keyFilter).forEach((key) => { this.removeItem(key, correlationId); this.performanceClient.incrementFields( { accountsRemoved: 1 }, correlationId ); }); } /** * Removes credentials associated with the provided account * @param account */ removeAccountContext(account: AccountInfo, correlationId: string): void { const allTokenKeys = this.getTokenKeys(); const keyFilter = (key: string): boolean => { return ( key.includes(account.homeAccountId) && key.includes(account.environment) ); }; allTokenKeys.idToken.filter(keyFilter).forEach((key) => { this.removeIdToken(key, correlationId); }); allTokenKeys.accessToken.filter(keyFilter).forEach((key) => { this.removeAccessToken(key, correlationId); }); allTokenKeys.refreshToken.filter(keyFilter).forEach((key) => { this.removeRefreshToken(key, correlationId); }); } /** * Removes accessToken from the cache * @param key * @param correlationId */ removeAccessToken(key: string, correlationId: string): void { const credential = this.getAccessTokenCredential(key, correlationId); this.removeItem(key, correlationId); this.performanceClient.incrementFields( { accessTokensRemoved: 1 }, correlationId ); if ( !credential || credential.credentialType.toLowerCase() !== CredentialType.ACCESS_TOKEN_WITH_AUTH_SCHEME.toLowerCase() || credential.tokenType !== AuthenticationScheme.POP ) { // If the credential is not a PoP token, we can return return; } // Remove Token Binding Key from key store for PoP Tokens Credentials const kid = credential.keyId; if (kid) { void this.cryptoImpl.removeTokenBindingKey(kid).catch(() => { this.commonLogger.error( `Failed to remove token binding key ${kid}`, correlationId ); this.performanceClient?.incrementFields( { removeTokenBindingKeyFailure: 1 }, correlationId ); }); } } /** * Removes all app metadata objects from cache. */ removeAppMetadata(correlationId: string): boolean { const allCacheKeys = this.getKeys(); allCacheKeys.forEach((cacheKey) => { if (this.isAppMetadata(cacheKey)) { this.removeItem(cacheKey, correlationId); } }); return true; } /** * Retrieve IdTokenEntity from cache * @param account {AccountInfo} * @param tokenKeys {?TokenKeys} * @param targetRealm {?string} * @param performanceClient {?IPerformanceClient} * @param correlationId {?string} */ getIdToken( account: AccountInfo, correlationId: string, tokenKeys?: TokenKeys, targetRealm?: string, performanceClient?: IPerformanceClient ): IdTokenEntity | null { this.commonLogger.trace("CacheManager - getIdToken called"); const idTokenFilter: CredentialFilter = { homeAccountId: account.homeAccountId, environment: account.environment, credentialType: CredentialType.ID_TOKEN, clientId: this.clientId, realm: targetRealm, }; const idTokenMap: Map = this.getIdTokensByFilter( idTokenFilter, correlationId, tokenKeys ); const numIdTokens = idTokenMap.size; if (numIdTokens < 1) { this.commonLogger.info("CacheManager:getIdToken - No token found"); return null; } else if (numIdTokens > 1) { let tokensToBeRemoved: Map = idTokenMap; // Multiple tenant profiles and no tenant specified, pick home account if (!targetRealm) { const homeIdTokenMap: Map = new Map< string, IdTokenEntity >(); idTokenMap.forEach((idToken, key) => { if (idToken.realm === account.tenantId) { homeIdTokenMap.set(key, idToken); } }); const numHomeIdTokens = homeIdTokenMap.size; if (numHomeIdTokens < 1) { this.commonLogger.info( "CacheManager:getIdToken - Multiple ID tokens found for account but none match account entity tenant id, returning first result" ); return idTokenMap.values().next().value; } else if (numHomeIdTokens === 1) { this.commonLogger.info( "CacheManager:getIdToken - Multiple ID tokens found for account, defaulting to home tenant profile" ); return homeIdTokenMap.values().next().value; } else { // Multiple ID tokens for home tenant profile, remove all and return null tokensToBeRemoved = homeIdTokenMap; } } // Multiple tokens for a single tenant profile, remove all and return null this.commonLogger.info( "CacheManager:getIdToken - Multiple matching ID tokens found, clearing them" ); tokensToBeRemoved.forEach((idToken, key) => { this.removeIdToken(key, correlationId); }); if (performanceClient && correlationId) { performanceClient.addFields( { multiMatchedID: idTokenMap.size }, correlationId ); } return null; } this.commonLogger.info("CacheManager:getIdToken - Returning ID token"); return idTokenMap.values().next().value; } /** * Gets all idTokens matching the given filter * @param filter * @returns */ getIdTokensByFilter( filter: CredentialFilter, correlationId: string, tokenKeys?: TokenKeys ): Map { const idTokenKeys = (tokenKeys && tokenKeys.idToken) || this.getTokenKeys().idToken; const idTokens: Map = new Map< string, IdTokenEntity >(); idTokenKeys.forEach((key) => { if ( !this.idTokenKeyMatchesFilter(key, { clientId: this.clientId, ...filter, }) ) { return; } const idToken = this.getIdTokenCredential(key, correlationId); if (idToken && this.credentialMatchesFilter(idToken, filter)) { idTokens.set(key, idToken); } }); return idTokens; } /** * Validate the cache key against filter before retrieving and parsing cache value * @param key * @param filter * @returns */ idTokenKeyMatchesFilter( inputKey: string, filter: CredentialFilter ): boolean { const key = inputKey.toLowerCase(); if ( filter.clientId && key.indexOf(filter.clientId.toLowerCase()) === -1 ) { return false; } if ( filter.homeAccountId && key.indexOf(filter.homeAccountId.toLowerCase()) === -1 ) { return false; } return true; } /** * Removes idToken from the cache * @param key */ removeIdToken(key: string, correlationId: string): void { this.removeItem(key, correlationId); } /** * Removes refresh token from the cache * @param key */ removeRefreshToken(key: string, correlationId: string): void { this.removeItem(key, correlationId); } /** * Retrieve AccessTokenEntity from cache * @param account {AccountInfo} * @param request {BaseAuthRequest} * @param correlationId {?string} * @param tokenKeys {?TokenKeys} * @param performanceClient {?IPerformanceClient} */ getAccessToken( account: AccountInfo, request: BaseAuthRequest, tokenKeys?: TokenKeys, targetRealm?: string ): AccessTokenEntity | null { const correlationId = request.correlationId; this.commonLogger.trace( "CacheManager - getAccessToken called", correlationId ); const scopes = ScopeSet.createSearchScopes(request.scopes); const authScheme = request.authenticationScheme || AuthenticationScheme.BEARER; /* * Distinguish between Bearer and PoP/SSH token cache types * Cast to lowercase to handle "bearer" from ADFS */ const credentialType = authScheme && authScheme.toLowerCase() !== AuthenticationScheme.BEARER.toLowerCase() ? CredentialType.ACCESS_TOKEN_WITH_AUTH_SCHEME : CredentialType.ACCESS_TOKEN; const accessTokenFilter: CredentialFilter = { homeAccountId: account.homeAccountId, environment: account.environment, credentialType: credentialType, clientId: this.clientId, realm: targetRealm || account.tenantId, target: scopes, tokenType: authScheme, keyId: request.sshKid, requestedClaimsHash: request.requestedClaimsHash, }; const accessTokenKeys = (tokenKeys && tokenKeys.accessToken) || this.getTokenKeys().accessToken; const accessTokens: AccessTokenEntity[] = []; accessTokenKeys.forEach((key) => { // Validate key if ( this.accessTokenKeyMatchesFilter(key, accessTokenFilter, true) ) { const accessToken = this.getAccessTokenCredential( key, correlationId ); // Validate value if ( accessToken && this.credentialMatchesFilter(accessToken, accessTokenFilter) ) { accessTokens.push(accessToken); } } }); const numAccessTokens = accessTokens.length; if (numAccessTokens < 1) { this.commonLogger.info( "CacheManager:getAccessToken - No token found", correlationId ); return null; } else if (numAccessTokens > 1) { this.commonLogger.info( "CacheManager:getAccessToken - Multiple access tokens found, clearing them", correlationId ); accessTokens.forEach((accessToken) => { this.removeAccessToken( this.generateCredentialKey(accessToken), correlationId ); }); this.performanceClient.addFields( { multiMatchedAT: accessTokens.length }, correlationId ); return null; } this.commonLogger.info( "CacheManager:getAccessToken - Returning access token", correlationId ); return accessTokens[0]; } /** * Validate the cache key against filter before retrieving and parsing cache value * @param key * @param filter * @param keyMustContainAllScopes * @returns */ accessTokenKeyMatchesFilter( inputKey: string, filter: CredentialFilter, keyMustContainAllScopes: boolean ): boolean { const key = inputKey.toLowerCase(); if ( filter.clientId && key.indexOf(filter.clientId.toLowerCase()) === -1 ) { return false; } if ( filter.homeAccountId && key.indexOf(filter.homeAccountId.toLowerCase()) === -1 ) { return false; } if (filter.realm && key.indexOf(filter.realm.toLowerCase()) === -1) { return false; } if ( filter.requestedClaimsHash && key.indexOf(filter.requestedClaimsHash.toLowerCase()) === -1 ) { return false; } if (filter.target) { const scopes = filter.target.asArray(); for (let i = 0; i < scopes.length; i++) { if ( keyMustContainAllScopes && !key.includes(scopes[i].toLowerCase()) ) { // When performing a cache lookup a missing scope would be a cache miss return false; } else if ( !keyMustContainAllScopes && key.includes(scopes[i].toLowerCase()) ) { // When performing a cache write, any token with a subset of requested scopes should be replaced return true; } } } return true; } /** * Gets all access tokens matching the filter * @param filter * @returns */ getAccessTokensByFilter( filter: CredentialFilter, correlationId: string ): AccessTokenEntity[] { const tokenKeys = this.getTokenKeys(); const accessTokens: AccessTokenEntity[] = []; tokenKeys.accessToken.forEach((key) => { if (!this.accessTokenKeyMatchesFilter(key, filter, true)) { return; } const accessToken = this.getAccessTokenCredential( key, correlationId ); if ( accessToken && this.credentialMatchesFilter(accessToken, filter) ) { accessTokens.push(accessToken); } }); return accessTokens; } /** * Helper to retrieve the appropriate refresh token from cache * @param account {AccountInfo} * @param familyRT {boolean} * @param correlationId {?string} * @param tokenKeys {?TokenKeys} * @param performanceClient {?IPerformanceClient} */ getRefreshToken( account: AccountInfo, familyRT: boolean, correlationId: string, tokenKeys?: TokenKeys, performanceClient?: IPerformanceClient ): RefreshTokenEntity | null { this.commonLogger.trace("CacheManager - getRefreshToken called"); const id = familyRT ? THE_FAMILY_ID : undefined; const refreshTokenFilter: CredentialFilter = { homeAccountId: account.homeAccountId, environment: account.environment, credentialType: CredentialType.REFRESH_TOKEN, clientId: this.clientId, familyId: id, }; const refreshTokenKeys = (tokenKeys && tokenKeys.refreshToken) || this.getTokenKeys().refreshToken; const refreshTokens: RefreshTokenEntity[] = []; refreshTokenKeys.forEach((key) => { // Validate key if (this.refreshTokenKeyMatchesFilter(key, refreshTokenFilter)) { const refreshToken = this.getRefreshTokenCredential( key, correlationId ); // Validate value if ( refreshToken && this.credentialMatchesFilter( refreshToken, refreshTokenFilter ) ) { refreshTokens.push(refreshToken); } } }); const numRefreshTokens = refreshTokens.length; if (numRefreshTokens < 1) { this.commonLogger.info( "CacheManager:getRefreshToken - No refresh token found." ); return null; } // address the else case after remove functions address environment aliases if (numRefreshTokens > 1 && performanceClient && correlationId) { performanceClient.addFields( { multiMatchedRT: numRefreshTokens }, correlationId ); } this.commonLogger.info( "CacheManager:getRefreshToken - returning refresh token" ); return refreshTokens[0] as RefreshTokenEntity; } /** * Validate the cache key against filter before retrieving and parsing cache value * @param key * @param filter */ refreshTokenKeyMatchesFilter( inputKey: string, filter: CredentialFilter ): boolean { const key = inputKey.toLowerCase(); if ( filter.familyId && key.indexOf(filter.familyId.toLowerCase()) === -1 ) { return false; } // If familyId is used, clientId is not in the key if ( !filter.familyId && filter.clientId && key.indexOf(filter.clientId.toLowerCase()) === -1 ) { return false; } if ( filter.homeAccountId && key.indexOf(filter.homeAccountId.toLowerCase()) === -1 ) { return false; } return true; } /** * Retrieve AppMetadataEntity from cache */ readAppMetadataFromCache(environment: string): AppMetadataEntity | null { const appMetadataFilter: AppMetadataFilter = { environment, clientId: this.clientId, }; const appMetadata: AppMetadataCache = this.getAppMetadataFilteredBy(appMetadataFilter); const appMetadataEntries: AppMetadataEntity[] = Object.keys( appMetadata ).map((key) => appMetadata[key]); const numAppMetadata = appMetadataEntries.length; if (numAppMetadata < 1) { return null; } else if (numAppMetadata > 1) { throw createClientAuthError( ClientAuthErrorCodes.multipleMatchingAppMetadata ); } return appMetadataEntries[0] as AppMetadataEntity; } /** * Return the family_id value associated with FOCI * @param environment * @param clientId */ isAppMetadataFOCI(environment: string): boolean { const appMetadata = this.readAppMetadataFromCache(environment); return !!(appMetadata && appMetadata.familyId === THE_FAMILY_ID); } /** * helper to match account ids * @param value * @param homeAccountId */ private matchHomeAccountId( entity: AccountEntity | CredentialEntity, homeAccountId: string ): boolean { return !!( typeof entity.homeAccountId === "string" && homeAccountId === entity.homeAccountId ); } /** * helper to match account ids * @param entity * @param localAccountId * @returns */ private matchLocalAccountIdFromTokenClaims( tokenClaims: TokenClaims, localAccountId: string ): boolean { const idTokenLocalAccountId = tokenClaims.oid || tokenClaims.sub; return localAccountId === idTokenLocalAccountId; } private matchLocalAccountIdFromTenantProfile( tenantProfile: TenantProfile, localAccountId: string ): boolean { return tenantProfile.localAccountId === localAccountId; } /** * helper to match names * @param entity * @param name * @returns true if the downcased name properties are present and match in the filter and the entity */ private matchName(claims: TokenClaims, name: string): boolean { return !!(name.toLowerCase() === claims.name?.toLowerCase()); } /** * helper to match usernames * @param entity * @param username * @returns */ private matchUsername( cachedUsername?: string, filterUsername?: string ): boolean { return !!( cachedUsername && typeof cachedUsername === "string" && filterUsername?.toLowerCase() === cachedUsername.toLowerCase() ); } /** * helper to match assertion * @param value * @param oboAssertion */ private matchUserAssertionHash( entity: CredentialEntity, userAssertionHash: string ): boolean { return !!( entity.userAssertionHash && userAssertionHash === entity.userAssertionHash ); } /** * helper to match environment * @param value * @param environment */ private matchEnvironment( entity: AccountEntity | CredentialEntity | AppMetadataEntity, environment: string ): boolean { // Check static authority options first for cases where authority metadata has not been resolved and cached yet if (this.staticAuthorityOptions) { const staticAliases = getAliasesFromStaticSources( this.staticAuthorityOptions, this.commonLogger ); if ( staticAliases.includes(environment) && staticAliases.includes(entity.environment) ) { return true; } } // Query metadata cache if no static authority configuration has aliases that match enviroment const cloudMetadata = this.getAuthorityMetadataByAlias(environment); if ( cloudMetadata && cloudMetadata.aliases.indexOf(entity.environment) > -1 ) { return true; } return false; } /** * helper to match credential type * @param entity * @param credentialType */ private matchCredentialType( entity: CredentialEntity, credentialType: string ): boolean { return ( entity.credentialType && credentialType.toLowerCase() === entity.credentialType.toLowerCase() ); } /** * helper to match client ids * @param entity * @param clientId */ private matchClientId( entity: CredentialEntity | AppMetadataEntity, clientId: string ): boolean { return !!(entity.clientId && clientId === entity.clientId); } /** * helper to match family ids * @param entity * @param familyId */ private matchFamilyId( entity: CredentialEntity | AppMetadataEntity, familyId: string ): boolean { return !!(entity.familyId && familyId === entity.familyId); } /** * helper to match realm * @param entity * @param realm */ private matchRealm( entity: AccountEntity | CredentialEntity, realm: string ): boolean { return !!(entity.realm?.toLowerCase() === realm.toLowerCase()); } /** * helper to match nativeAccountId * @param entity * @param nativeAccountId * @returns boolean indicating the match result */ private matchNativeAccountId( entity: AccountEntity, nativeAccountId: string ): boolean { return !!( entity.nativeAccountId && nativeAccountId === entity.nativeAccountId ); } /** * helper to match loginHint which can be either: * 1. login_hint ID token claim * 2. username in cached account object * 3. upn in ID token claims * @param entity * @param loginHint * @returns */ private matchLoginHintFromTokenClaims( tokenClaims: TokenClaims, loginHint: string ): boolean { if (tokenClaims.login_hint === loginHint) { return true; } if (tokenClaims.preferred_username === loginHint) { return true; } if (tokenClaims.upn === loginHint) { return true; } return false; } /** * Helper to match sid * @param entity * @param sid * @returns true if the sid claim is present and matches the filter */ private matchSid(idTokenClaims: TokenClaims, sid: string): boolean { return idTokenClaims.sid === sid; } private matchAuthorityType( entity: AccountEntity, authorityType: string ): boolean { return !!( entity.authorityType && authorityType.toLowerCase() === entity.authorityType.toLowerCase() ); } /** * Returns true if the target scopes are a subset of the current entity's scopes, false otherwise. * @param entity * @param target */ private matchTarget(entity: CredentialEntity, target: ScopeSet): boolean { const isNotAccessTokenCredential = entity.credentialType !== CredentialType.ACCESS_TOKEN && entity.credentialType !== CredentialType.ACCESS_TOKEN_WITH_AUTH_SCHEME; if (isNotAccessTokenCredential || !entity.target) { return false; } const entityScopeSet: ScopeSet = ScopeSet.fromString(entity.target); return entityScopeSet.containsScopeSet(target); } /** * Returns true if the credential's tokenType or Authentication Scheme matches the one in the request, false otherwise * @param entity * @param tokenType */ private matchTokenType( entity: CredentialEntity, tokenType: AuthenticationScheme ): boolean { return !!(entity.tokenType && entity.tokenType === tokenType); } /** * Returns true if the credential's keyId matches the one in the request, false otherwise * @param entity * @param keyId */ private matchKeyId(entity: CredentialEntity, keyId: string): boolean { return !!(entity.keyId && entity.keyId === keyId); } /** * returns if a given cache entity is of the type appmetadata * @param key */ private isAppMetadata(key: string): boolean { return key.indexOf(APP_METADATA) !== -1; } /** * returns if a given cache entity is of the type authoritymetadata * @param key */ protected isAuthorityMetadata(key: string): boolean { return key.indexOf(AUTHORITY_METADATA_CONSTANTS.CACHE_KEY) !== -1; } /** * returns cache key used for cloud instance metadata */ generateAuthorityMetadataCacheKey(authority: string): string { return `${AUTHORITY_METADATA_CONSTANTS.CACHE_KEY}-${this.clientId}-${authority}`; } /** * Helper to convert serialized data to object * @param obj * @param json */ static toObject(obj: T, json: object): T { for (const propertyName in json) { obj[propertyName] = json[propertyName]; } return obj; } } /** @internal */ export class DefaultStorageClass extends CacheManager { async setAccount(): Promise { throw createClientAuthError(ClientAuthErrorCodes.methodNotImplemented); } getAccount(): AccountEntity { throw createClientAuthError(ClientAuthErrorCodes.methodNotImplemented); } async setIdTokenCredential(): Promise { throw createClientAuthError(ClientAuthErrorCodes.methodNotImplemented); } getIdTokenCredential(): IdTokenEntity { throw createClientAuthError(ClientAuthErrorCodes.methodNotImplemented); } async setAccessTokenCredential(): Promise { throw createClientAuthError(ClientAuthErrorCodes.methodNotImplemented); } getAccessTokenCredential(): AccessTokenEntity { throw createClientAuthError(ClientAuthErrorCodes.methodNotImplemented); } async setRefreshTokenCredential(): Promise { throw createClientAuthError(ClientAuthErrorCodes.methodNotImplemented); } getRefreshTokenCredential(): RefreshTokenEntity { throw createClientAuthError(ClientAuthErrorCodes.methodNotImplemented); } setAppMetadata(): void { throw createClientAuthError(ClientAuthErrorCodes.methodNotImplemented); } getAppMetadata(): AppMetadataEntity { throw createClientAuthError(ClientAuthErrorCodes.methodNotImplemented); } setServerTelemetry(): void { throw createClientAuthError(ClientAuthErrorCodes.methodNotImplemented); } getServerTelemetry(): ServerTelemetryEntity { throw createClientAuthError(ClientAuthErrorCodes.methodNotImplemented); } setAuthorityMetadata(): void { throw createClientAuthError(ClientAuthErrorCodes.methodNotImplemented); } getAuthorityMetadata(): AuthorityMetadataEntity | null { throw createClientAuthError(ClientAuthErrorCodes.methodNotImplemented); } getAuthorityMetadataKeys(): Array { throw createClientAuthError(ClientAuthErrorCodes.methodNotImplemented); } setThrottlingCache(): void { throw createClientAuthError(ClientAuthErrorCodes.methodNotImplemented); } getThrottlingCache(): ThrottlingEntity { throw createClientAuthError(ClientAuthErrorCodes.methodNotImplemented); } removeItem(): boolean { throw createClientAuthError(ClientAuthErrorCodes.methodNotImplemented); } getKeys(): string[] { throw createClientAuthError(ClientAuthErrorCodes.methodNotImplemented); } getAccountKeys(): string[] { throw createClientAuthError(ClientAuthErrorCodes.methodNotImplemented); } getTokenKeys(): TokenKeys { throw createClientAuthError(ClientAuthErrorCodes.methodNotImplemented); } generateCredentialKey(): string { throw createClientAuthError(ClientAuthErrorCodes.methodNotImplemented); } generateAccountKey(): string { throw createClientAuthError(ClientAuthErrorCodes.methodNotImplemented); } }