Write a Custom User Store Manager¶
The following sections provide information that you need to be aware of when writing a custom user store manager.
Available methods and implementations¶
There are a set of methods available in the AbstractUserStoreManager
class. These methods are used when interacting with user stores. When we implement a custom user store manager, it is important to identify the methods that must be implemented or overridden.
Overriding methods
You must select the methods to override based on your requirement. For example, if you want to change the way you encrypt the password, you only need to implement the preparePassword
method. If you want to implement a completely new read/write user store manager, you need to implement all the methods listed in the tables given below.
If the user store is read-only, you can implement only the important methods and read methods (if you extend from AbstractUserStoreManager
, you have to keep the unrelated methods empty).
There are a few other methods used for internal purposes. You do not need to override those methods.
The following list briefly explains the use of the available methods in the AbstractUserStoreManager
class. Most of the methods provide a configuration option through properties. It is recommended to use these methods with the provided customization options.
Important methods¶
Available methods | Default behavior | Reason for overriding |
---|---|---|
boolean doAuthenticate(String userName, Object credential) |
This method returns details on whether the given username and password are matched or not. The credential is usually a string literal. | If you want to change the authentication logic, you can override this method and write your own implementation. The default task of this method is to compare the given password with the stored password. The given credentials are passed to the preparePassword method to do the salting or encryption before the comparison takes place. |
String preparePassword(Object password, String saltValue) |
This returns the encrypted or plain-text password based on the configurations. | You can override this method if you need to change the way you encrypt the password. If you want to change the algorithm that is used for encryption, you can configure it. |
Properties getDefaultUserStoreProperties() |
The default properties of the user store are returned using this method. These properties are used in user store related operations. Be sure to manually add the following property when you implement the class to control whether the user store is enabled or disabled.: setOptionalProperty("Disabled", "false", "Whether user store is disabled"); |
By overriding this method, you can programmatically change the configuration of the user store manager implementation. |
boolean checkUserNameValid(String userName) |
Returns whether the given username is compatible with the defined criteria. | This is the criteria used for defining a valid username that can be configured as a regex in user store configurations. If you want to change the way user name validation is done, override this method. |
boolean checkUserPasswordValid(Object credential) |
This returns whether the given password is compatible with the defined criteria. This is invoked when creating a user, updating a password, and authorization. | Similar to the username, you can configure the format of a valid password in configuration. If you want to change that behavior, you can override this method. |
Write methods¶
Available methods | Default behavior |
---|---|
|
This method is responsible to create a new user based on the given values. We can change the JDBC query or LDAP attribute name with the user store configuration. |
|
This removes the user store record related to the given username. |
|
This is responsible to update the credential of the given username after authenticating with the existing password. |
|
Admin users can use this method to update the credentials of a given user. This can be done without validating the existing password. |
|
This creates a new user role with the given roleName and maps the given users to the newly created role. The shared parameter indicates whether this role is shared among tenants or not. |
|
This method removes the given role and related mappings from the user store. |
|
This method is used to update the name of the existing roles. |
|
This is used to delete the existing mappings between the given user and |
|
This is used to delete the existing mappings between the given role and |
|
This is responsible for creating a new claim for a given user and profile with the given claim URI and value. |
|
This is responsible for creating a new claim for a given user and profile with the given list of claim URIs and values. |
|
Removes the existing claim details mapped with the given user and profile |
|
Removes the given list of claims from a given user |
|
This method is used to persist tokens in the user store. |
|
This method is used to create a group with given parameters. |
|
This method is used to create a group with given user uuid list. |
|
This method is used to update the name of the group with the given group uuid.. |
|
This method is used to update the user list of a group. |
|
This method is used to delete the group with the given id. |
|
This method is used to create a group with the given name and given set of user names. |
|
This method is used to update name of the group with the given current name. |
|
This method is used to delete the group with the name |
Read methods¶
Available methods | Default behavior |
---|---|
boolean doCheckExistingUser(String userName) |
Returns whether the provided userName already exists in the user store |
boolean doCheckExistingRole(String roleName) |
Returns whether the provided roleName already exists in the user store |
String[] doListUsers(String filter, int maxItemLimit) |
This method returns a list of usernames that match the given filter string. |
String[] doGetRoleNames(String filter, int maxItemLimit) |
Returns a list of role names that match with the given filter string. |
String[] doGetExternalRoleListOfUser(String userName, String filter) |
Returns a list of external role names of a given user that match with the given filter string |
String[] doGetSharedRoleListOfUser(String userName, String tenantDomain, String filter) |
This method returns a list of shared role names of a given user that match the given filter string. |
Map<String, String> getUserPropertyValues(String userName, String[] propertyNames, String profileName) |
This method returns values for the given propertyNames for a given userName and profileName . |
String[] getUserListFromProperties(String property, String value, String profileName) |
This returns a list of usernames that match with the given value of the given property and profileName . |
String[] doGetDisplayNamesForInternalRole(String[] userNames) |
Returns names to display in the UI for given usernames |
Date getPasswordExpirationTime(String userName) |
This returns the password expiry date of a given user. The default value is null. |
int getUserId(String username) |
This method returns the identifier of a given username. The default value is 0. |
boolean doCheckIsUserInRole(String userName, String roleName) |
True is returned if the given user is already mapped to the given role name. |
String[] getProfileNames(String userName) |
Returns a list of profile names mapped with a given username. |
String[] doGetSharedRoleNames(String tenantDomain, String filter, int maxItemLimit) |
This returns a list of role names that are associated with the given tenant domain and match with the filter. |
String[] doGetUserListOfRole(String roleName, String filter) |
This method returns a list of usernames that are mapped with the given rolename. |
String[] getAllProfileNames() |
All the profile names are returned including the default profile. |
boolean isValidRememberMeToken(String userName, String token) |
This method is used to check if the given token exists for the given user. |
boolean isMultipleProfilesAllowed() |
This returns whether this user store is allowed to have multiple profiles per user. The default value is false . |
boolean isBulkImportSupported() |
This method returns whether this user store allows bulk transactions or not. |
String doGetGroupIdFromGroupName(String groupName) |
This method returns group uuid which has the given group name. |
String doGetGroupNameFromGroupId(String groupId) |
This method returns group name which has the given group uuid. |
Group doGetGroupFromGroupName(String groupName, List<String> requiredAttributes) |
This method returns the group with the given group name. |
Group doGetGroupFromGroupId(String groupId, List<String> requiredAttributes) |
This method returns the group with the given group uuid. |
List<Group> doGetGroupListOfUser(String userId, int limit, int offset, String sortBy, String sortOrder) |
This method returns the group list of a given user. |
List<Group> doListGroups(Condition condition, int limit, int offset, String sortBy, String sortOrder) |
This method return the group list which matches with the given condition. |
Implementations¶
In WSO2 Identity Server, there are four user store manager classes that implement the AbstractUserStoreManager
class. You can select one of those classes according to the user store that you have in your environment.
userstore manager class | When you would use it |
---|---|
org.wso2.carbon.user.core.jdbc.UniqueIDJDBCUserStoreManager |
If your user details are stored in a database, you must use this user store manager implementation. With the abstraction provided in this implementation, most of the JDBC-based scenarios can be handled without writing a custom user store manager. |
org.wso2.carbon.user.core.ldap.UniqueIDReadOnlyLDAPUserStoreManager |
You can use this class if you have an LDAP user store. This implementation does not allow you to insert or update users from the WSO2 Identity Server side. Instead, you can only read and use them in the product. |
org.wso2.carbon.user.core.ldap.UniqueIDReadWriteLDAPUserStoreManager |
If you want to allow WSO2 Identity Server to manipulate user store data, you need to use this implementation. |
org.wso2.carbon.user.core.ldap.UniqueIDActiveDirectoryLDAPUserStoreManager |
Active Directory also can be used as the user store of WSO2 Identity Server and you can configure it using this user store manager implementation. |
Implement a custom JDBC user store manager¶
The instructions in this section are focused on implementing a sample JDBC user store manager. For this sample, the following tools are used to implement the custom user store manager.
- Java 1.6.0
- IDE (Eclipse, InteliJ IDEA, etc.)
- Apache Maven
Set up the implementation¶
To set up this implementation, follow the instructions given below.
- Create a new Apache Maven project with the help of the IDE that you are using. The project should be a simple Apache Maven project and you can use any desired artifact and group ID.
-
Add the WSO2 user store management .jar file as a dependency on our project. Since this .jar file is stored in WSO2's Maven repository, you need to add the WSO2 repository to your POM file. Given below is a sample POM file.
Note
The version number of the carbon dependencies given below has to be updated according to the carbon kernel version that the particular product version is compatible with.
Sample
pom.xml
file<?xml version="1.0" encoding="UTF-8"?> <project xmlns="http://maven.apache.org/POM/4.0.0" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd"> <modelVersion>4.0.0</modelVersion> <groupId>org.wso2.sample</groupId> <artifactId>CustomReadOnlyJDBCUserStoreManager</artifactId> <version>1.0.0</version> <packaging>bundle</packaging> <repositories> <repository> <id>wso2-nexus</id> <name>WSO2 internal Repository</name> <url>http://maven.wso2.org/nexus/content/groups/wso2-public/</url> <releases> <enabled>true</enabled> <updatePolicy>daily</updatePolicy> <checksumPolicy>ignore</checksumPolicy> </releases> </repository> </repositories> <dependencies> <dependency> <groupId>org.wso2.carbon</groupId> <artifactId>org.wso2.carbon.user.core</artifactId> <version>4.4.11</version> </dependency> <dependency> <groupId>org.wso2.carbon</groupId> <artifactId>org.wso2.carbon.utils</artifactId> <version>4.4.11</version> </dependency> <dependency> <groupId>org.wso2.carbon</groupId> <artifactId>org.wso2.carbon.user.api</artifactId> <version>4.4.11</version> </dependency> </dependencies> <build> <plugins> <plugin> <artifactId>maven-compiler-plugin</artifactId> <version>2.3.1</version> <inherited>true</inherited> <configuration> <encoding>UTF-8</encoding> <source>1.7</source> <target>1.7</target> </configuration> </plugin> <plugin> <groupId>org.apache.felix</groupId> <artifactId>maven-scr-plugin</artifactId> <version>1.7.2</version> <executions> <execution> <id>generate-scr-scrdescriptor</id> <goals> <goal>scr</goal> </goals> </execution> </executions> </plugin> <plugin> <groupId>org.apache.felix</groupId> <artifactId>maven-bundle-plugin</artifactId> <version>2.3.5</version> <extensions>true</extensions> <configuration> <instructions> <Bundle-SymbolicName>${project.artifactId}</Bundle-SymbolicName> <Bundle-Name>${project.artifactId}</Bundle-Name> <Private-Package> org.wso2.sample.user.store.manager.internal </Private-Package> <Export-Package> !org.wso2.sample.user.store.manager.internal, org.wso2.sample.user.store.manager.*, </Export-Package> <Import-Package> org.wso2.carbon.*, org.apache.commons.logging.*, org.osgi.framework.*, org.osgi.service.component.* </Import-Package> </instructions> </configuration> </plugin> </plugins> </build> </project>
Now your basic implementation is ready.
Write a custom user store manager for a sample scenario¶
Consider a sample scenario where you want to use a custom hashing method using a 3rd party library such as Jasypt and to do this, you need to override the doAuthentication
and preparePassword
methods.
Follow the steps given below to write a custom user store manager.
-
Include the required dependencies in your development environment. To do that, include the relevant Apache Maven dependency details or manually add the
.jar
files to your classpath. For example, add the following XML snippet under the dependencies tag in yourpom.xml
file to include the Jasypt dependency.<dependency> <groupId>org.jasypt</groupId> <artifactId>jasypt</artifactId> <version>1.9.2</version> </dependency>
-
Create a new class by extending the existing
JDBCUserStoreManager
implementation. The following code is an example of how this would look.Code sample
package com.wso2.custom.usermgt; package org.wso2.custom.user.store; import org.apache.commons.logging.Log; import org.apache.commons.logging.LogFactory; import org.jasypt.util.password.StrongPasswordEncryptor; import org.wso2.carbon.user.api.RealmConfiguration; import org.wso2.carbon.user.core.UserCoreConstants; import org.wso2.carbon.user.core.UserRealm; import org.wso2.carbon.user.core.UserStoreException; import org.wso2.carbon.user.core.claim.ClaimManager; import org.wso2.carbon.user.core.common.AuthenticationResult; import org.wso2.carbon.user.core.common.FailureReason; import org.wso2.carbon.user.core.common.User; import org.wso2.carbon.user.core.jdbc.JDBCRealmConstants; import org.wso2.carbon.user.core.jdbc.UniqueIDJDBCUserStoreManager; import org.wso2.carbon.user.core.profile.ProfileConfigurationManager; import org.wso2.carbon.utils.Secret; import java.sql.Connection; import java.sql.PreparedStatement; import java.sql.ResultSet; import java.sql.SQLException; import java.sql.Timestamp; import java.util.Date; import java.util.GregorianCalendar; import java.util.Map; /** * This class implements the Custom User Store Manager. */ public class CustomUserStoreManager extends UniqueIDJDBCUserStoreManager { private static final Log log = LogFactory.getLog(CustomUserStoreManager.class); private static final StrongPasswordEncryptor passwordEncryptor = new StrongPasswordEncryptor(); public CustomUserStoreManager() { } public CustomUserStoreManager(RealmConfiguration realmConfig, Map<String, Object> properties, ClaimManager claimManager, ProfileConfigurationManager profileManager, UserRealm realm, Integer tenantId) throws UserStoreException { super(realmConfig, properties, claimManager, profileManager, realm, tenantId); log.info("CustomUserStoreManager initialized..."); } @Override public AuthenticationResult doAuthenticateWithUserName(String userName, Object credential) throws UserStoreException { boolean isAuthenticated = false; String userID = null; User user; // In order to avoid unnecessary db queries. if (!isValidUserName(userName)) { String reason = "Username validation failed."; if (log.isDebugEnabled()) { log.debug(reason); } return getAuthenticationResult(reason); } if (!isValidCredentials(credential)) { String reason = "Password validation failed."; if (log.isDebugEnabled()) { log.debug(reason); } return getAuthenticationResult(reason); } try { String candidatePassword = String.copyValueOf(((Secret) credential).getChars()); Connection dbConnection = null; ResultSet rs = null; PreparedStatement prepStmt = null; String sql = null; dbConnection = this.getDBConnection(); dbConnection.setAutoCommit(false); // get the SQL statement used to select user details sql = this.realmConfig.getUserStoreProperty(JDBCRealmConstants.SELECT_USER_NAME); if (log.isDebugEnabled()) { log.debug(sql); } prepStmt = dbConnection.prepareStatement(sql); prepStmt.setString(1, userName); // check whether tenant id is used if (sql.contains(UserCoreConstants.UM_TENANT_COLUMN)) { prepStmt.setInt(2, this.tenantId); } rs = prepStmt.executeQuery(); if (rs.next()) { userID = rs.getString(1); String storedPassword = rs.getString(3); // check whether password is expired or not boolean requireChange = rs.getBoolean(5); Timestamp changedTime = rs.getTimestamp(6); GregorianCalendar gc = new GregorianCalendar(); gc.add(GregorianCalendar.HOUR, -24); Date date = gc.getTime(); if (!(requireChange && changedTime.before(date))) { // compare the given password with the stored password using jasypt isAuthenticated = passwordEncryptor.checkPassword(candidatePassword, storedPassword); } } dbConnection.commit(); log.info(userName + " is authenticated? " + isAuthenticated); } catch (SQLException exp) { try { this.getDBConnection().rollback(); } catch (SQLException e1) { throw new UserStoreException("Transaction rollback connection error occurred while" + " retrieving user authentication info. Authentication Failure.", e1); } log.error("Error occurred while retrieving user authentication info.", exp); throw new UserStoreException("Authentication Failure"); } if (isAuthenticated) { user = getUser(userID, userName); AuthenticationResult authenticationResult = new AuthenticationResult( AuthenticationResult.AuthenticationStatus.SUCCESS); authenticationResult.setAuthenticatedUser(user); return authenticationResult; } else { AuthenticationResult authenticationResult = new AuthenticationResult( AuthenticationResult.AuthenticationStatus.FAIL); authenticationResult.setFailureReason(new FailureReason("Invalid credentials.")); return authenticationResult; } } @Override protected String preparePassword(Object password, String saltValue) throws UserStoreException { if (password != null) { String candidatePassword = String.copyValueOf(((Secret) password).getChars()); // ignore saltValue for the time being log.info("Generating hash value using jasypt..."); return passwordEncryptor.encryptPassword(candidatePassword); } else { log.error("Password cannot be null"); throw new UserStoreException("Authentication Failure"); } } private AuthenticationResult getAuthenticationResult(String reason) { AuthenticationResult authenticationResult = new AuthenticationResult( AuthenticationResult.AuthenticationStatus.FAIL); authenticationResult.setFailureReason(new FailureReason(reason)); return authenticationResult; } }
Note
The default constructor is not enough when you implement a custom user store manager and you must implement a constructor with relevant arguments.
Deploy and configure the custom user store manager¶
Follow the instructions given below to deploy and configure the custom user store manager.
-
Copy the artifact of your project (custom-userstore.jar, in this case) to the
<IS_HOME>/repository/components/dropins
directory. Also copy all the OSGI bundles to this location. If you have any dependency .jar files, copy them to the<IS_HOME>/repository/components/lib
directory. -
Add the following configuration to the
<IS_HOME>/repository/conf/deployment.toml
file to use our custom implementation for user store management.Although the existing
UniqueID
user stores are allowed by default, when adding a new user store, note that both existing user stores, as well as new user stores, must be configured as shown below.Format
[user_store_mgt] custom_user_stores=["<new userstore>"]
Sample
[user_store_mgt] custom_user_stores=["com.wso2.custom.usermgt.CustomUserStoreManager"]
Tip
This step provides instructions on configuring your custom user store manager as a primary user store. Alternatively, you can configure this as a secondary user store if you already have a different primary user store configured. For more information configuring user stores in WSO2 Identity Server, see Manage user stores.
You do not need to change anything else since you extend the
JDBCUserStoreManager
class, so the configurations will remain the same.
You have now implemented a custom user store manager for WSO2 Identity Server. Once you have done this, start the product and see the log messages that you have placed inside overridden methods when you create a new user or log in. This ensures that all your configurations work as intended.
Do you want to create a custom user store that only has a few enabled methods?
- Sign in to the WSO2 Identity Server management console (
https://<IS_HOST>:<PORT>/carbon
). - Click User Attributes & Stores > User stores > New User Store.
- Select the custom user store you just created.
- Deselect the Claim Operations Supported property.
- Click Finish.
Try out the sample user store manager¶
- Build the sample user store manager here using
mvn clean install
command. - Copy the generated
org.wso2.custom.user.store-1.0.0.jar
to the<IS_HOME>/repository/components/dropins
directory. -
Add the following configuration to the
<IS_HOME>/repository/conf/deployment.toml
file to use our custom implementation for user store management.[user_store_mgt] custom_user_stores=["org.wso2.custom.user.store.CustomUserStoreManager"]
-
Restart the Identity Server.
Note
Once you add the custom user store manager to the Identity server, you have to configure it as a primary or a secondary user store.