Passing Enduser Attributes to the Backend Using JWT

JSON Web Token (JWT) is used to represent claims that are transferred between two parties such as the end user and the backend.

A claim is an attribute of the user that is mapped to the underlying user store. It is encoded as a JavaScript Object Notation (JSON) object that is used as the payload of a JSON Web Signature (JWS) structure, or as the plain text of a JSON Web Encryption (JWE) structure. This enables claims to be digitally signed.

A set of claims is called a dialect (e.g., http://wso2.org/claims ).  The general format of a JWT is {token infor}.{claims list}.{signature}. The API implementation uses information such as logging, content filtering and authentication/authorization that is stored in this token. The token is Base64-encoded and sent to the API implementation in a HTTP header variable.  The JWT is self-contained and is divided into three parts as the header, the payload and the signature. For more information on JWT, see JSON Web Token (JWT) Overview .

To authenticate end users, the API Manager passes attributes of the API invoker to the backend API implementation using JWT. In most production deployments, service calls go through the API Manager or a proxy service. If you enable JWT generation in the API Manager, each API request will carry a JWT to the back-end service. When the request goes through the API manager, the JWT is appended as a transport header to the outgoing message. The back-end service fetches the JWT and retrieves the required information about the user, application, or token.

An example of a JWT is given below:

{
    "typ":"JWT",
    "alg":"NONE"
 }{
    "iss":"wso2.org/products/am",
    "exp":1345183492181,
    "http://wso2.org/claims/subscriber":"admin",
    "http://wso2.org/claims/applicationname":"app2",
    "http://wso2.org/claims/apicontext":"/placeFinder",
    "http://wso2.org/claims/version":"1.0.0",
    "http://wso2.org/claims/tier":"Silver",
    "http://wso2.org/claims/enduser":"sumedha"
 }

The above token contains,

• Token expiration time ("exp")
• Subscriber to the API, usually the app developer ("http://wso2.org/claims/subscriber")
• Application through which API invocation is done ("http://wso2.org/claims/applicationname")
• Context of the API ("http://wso2.org/claims/apicontext")
• API version ("http://wso2.org/claims/version")
• Tier/price band for the subscription ("http://wso2.org/claims/tier")
• Enduser of the app who's action invoked the API ("http://wso2.org/claims/enduser")

Let's see how to enable and pass information in the JWT or completely alter the JWT generation logic in the API Manager: 

Configuring JWT

Before passing enduser attributes, you enable and configure the JWT implementation in the <APIM_HOME>/repository/conf/api-manager.xml file. The relevant elements are described below. If you do not configure these elements, they take their default values.

<EnableJWTGeneration>

<ClaimsRetrieverImplClass>org.wso2.carbon.apimgt.impl.token.DefaultClaimsRetriever</ClaimsRetrieverImplClass> 

Customizing the JWT generation

The JWT that is generated by default (see example above) has predefined attributes that are passed to the backend. These include basic application-specific details, subscription details, and user information that are defined in the JWT generation class that comes with the API Manager by the name org.wso2.carbon.apimgt.keymgt.token.JWTGenerator. If you want to pass additional attributes to the backend with the JWT or completely change the default JWT generation logic, do the following:

1. Write your own custom JWT implementation class by extending the default JWTGenerator class. A typical example of implementing your own claim generator is given below. It implements the populateCustomClaims()  method to generate some custom claims and adds them to the JWT. 

   import org.wso2.carbon.apimgt.keymgt.APIConstants;
   import org.wso2.carbon.apimgt.keymgt.dto.APIKeyValidationInfoDTO;
   import org.wso2.carbon.apimgt.keymgt.token.JWTGenerator;
   import org.wso2.carbon.apimgt.api.*;
   
   import java.util.Map;
   
   public class CustomTokenGenerator extends JWTGenerator {
   
       public Map populateStandardClaims(APIKeyValidationInfoDTO keyValidationInfoDTO, String apiContext, String version)
               throws APIManagementException {
           Map claims = super.populateStandardClaims(keyValidationInfoDTO, apiContext, version);
           boolean isApplicationToken =
                   keyValidationInfoDTO.getUserType().equalsIgnoreCase(APIConstants.ACCESS_TOKEN_USER_TYPE_APPLICATION) ? true : false;
           String dialect = getDialectURI();
           if (claims.get(dialect + "/enduser") != null) {
               if (isApplicationToken) {
                   claims.put(dialect + "/enduser", "null");
                   claims.put(dialect + "/enduserTenantId", "null");
               } else {
                   String enduser = claims.get(dialect + "/enduser");
                   if (enduser.endsWith("@carbon.super")) {
                       enduser = enduser.replace("@carbon.super", "");
                       claims.put(dialect + "/enduser", enduser);
                   }
               }
           }
   
           return claims;
   
       }
   
       public Map populateCustomClaims(APIKeyValidationInfoDTO keyValidationInfoDTO, String apiContext, String version, String accessToken)
               throws APIManagementException {
           Long time = System.currentTimeMillis();
           String text = "This is custom JWT";
           Map map = new HashMap();
           map.put("current_timestamp", time.toString());
           map.put("messge" , text);
           return map;
       }
   }

2. Build your class and add the JAR file to the <APIM_HOME>/repository/components/lib directory.

3. Add your class in the <JWTGeneratorImpl> element of the <APIM_HOME>/repository/conf/api-manager.xml file. 

   <JWTConfiguration>
      ....
      <JWTGeneratorImpl>org.wso2.carbon.test.CustomTokenGenerator</JWTGeneratorImpl>
      ....
   </JWTConfiguration>

4. Set the <EnableJWTGeneration> element to true in the api-manager.xml file. 
5. Restart the server.

Changing the JWT encoding to Base64URL encoding

The default JWT generator, org.wso2.carbon.apimgt.impl.token.JWTGenerator, encodes the value of the JWT using Base64 encoding. However, for certain apps you might need to have it in Base64URL encoding. To encode the JWT using Base64URL encoding, add the URLSafeJWTGenerator class in the <TokenGeneratorImpl> element in the <APIM_HOME>/repository/conf/api-manager.xml file as shown below.

<JWTConfiguration>
   ....
   <JWTGeneratorImpl>org.wso2.carbon.apimgt.keymgt.token.URLSafeJWTGenerator</JWTGeneratorImpl>
   ....
</JWTConfiguration>

Setting the expiry time of the JWT

Setting the JWT expiry time has no effect if the caching is enabled in the Gateway Manager or Key Manager. The WSO2 API-M Gateway caching is enabled by default. However, if required, you can enable or disable the caching for the Gateway Manager or the Key Manager using the <EnableGatewayTokenCache> or <EnableKeyManagerTokenCache> elements respectively in the <APIM_HOME>/repository/conf/api-manager.xml file. If caching is enabled for the Gateway Manager or the Key Manager, the JWT expiry time will be the same as that defined in <TokenCacheExpiry> property in <API-M_HOME>/repository/conf/api-manager.xml file. However, if caching is not enabled for the Gateway and Key Manager nodes, you can set the JWT expiry time by adding the expiry time in seconds in the <JWTExpiryTime> element, which is in the <APIM_HOME>/repository/conf/api-manager.xml file. 

Example:

<APIKeyValidator>
	...
	<JWTExpiryTime>60</JWTExpiryTime>
	...
</APIKeyValidator>