com.predic8.membrane.core.interceptor.authentication.session.StaticUserDataProvider Maven / Gradle / Ivy
/* Copyright 2012 predic8 GmbH, www.predic8.com
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 com.predic8.membrane.core.interceptor.authentication.session;
import com.google.api.client.repackaged.org.apache.commons.codec.binary.Base64;
import com.predic8.membrane.annot.MCAttribute;
import com.predic8.membrane.annot.MCChildElement;
import com.predic8.membrane.annot.MCElement;
import com.predic8.membrane.annot.MCOtherAttributes;
import com.predic8.membrane.core.Router;
import org.apache.commons.codec.digest.Crypt;
import org.bouncycastle.jce.provider.BouncyCastleProvider;
import java.io.UnsupportedEncodingException;
import java.security.MessageDigest;
import java.security.NoSuchAlgorithmException;
import java.security.SecureRandom;
import java.util.*;
import java.util.regex.Pattern;
/**
* @description A user data provider listing all user data in-place in the config file.
* @explanation
* the staticuserdataprovider can be used to statically list user data within the config file.
*
*
* each user must have a unique username attribute as well as a password attribute.
*
*
* arbitrary attributes can be set on a user element. other sub-components of the login interceptor
* might use those: for example, the telekomsmstokenprovider uses the sms property as the
* user's cell phone number. for example, the totptokenprovider uses the secret property to
* initialize the token sequence.
*
*/
@MCElement(name="staticUserDataProvider")
public class StaticUserDataProvider implements UserDataProvider {
private List users = new ArrayList();
private Map usersByName = new HashMap();
private BouncyCastleProvider provider = new BouncyCastleProvider();
private SecureRandom random = new SecureRandom();
private int saltByteSize = 128;
@Override
public Map verify(Map postData) {
String username = postData.get("username");
if (username == null)
throw new NoSuchElementException();
if (username.equals("error"))
throw new RuntimeException();
User userAttributes;
userAttributes = getUsersByName().get(username);
if (userAttributes == null)
throw new NoSuchElementException();
String pw = null;
String postDataPassword = postData.get("password");
if(userAttributes.getPassword() != null && isHashedPassword(userAttributes.getPassword())) {
String userHash = userAttributes.getPassword();
String[] userHashSplit = userHash.split(Pattern.quote("$"));
String algo = userHashSplit[1];
String salt = userHashSplit[2];
try {
pw = createPasswdCompatibleHash(algo,postDataPassword,salt);
} catch (UnsupportedEncodingException e) {
throw new RuntimeException(e.getMessage());
} catch (NoSuchAlgorithmException e) {
throw new RuntimeException(e.getMessage());
}
}else
pw = postDataPassword;
String pw2;
pw2 = userAttributes.getPassword();
if (pw2 == null || !pw2.equals(pw))
throw new NoSuchElementException();
return userAttributes.getAttributes();
}
private boolean isHashedPassword(String postDataPassword) {
// TODO do a better check here
String[] split = postDataPassword.split(Pattern.quote("$"));
if(split.length != 4)
return false;
if(!split[0].isEmpty())
return false;
if(split[3].length() < 20)
return false;
return true;
}
private String createPasswdCompatibleHash(String algo, String password, String salt) throws UnsupportedEncodingException, NoSuchAlgorithmException {
return Crypt.crypt(password, "$" + algo + "$" + salt);
}
private String createPasswdCompatibleHash(String algo, String password) throws UnsupportedEncodingException, NoSuchAlgorithmException {
byte[] salt = new byte[saltByteSize];
random.nextBytes(salt);
String saltString = Base64.encodeBase64String(salt);
if(saltString.length() > 8)
saltString = saltString.substring(0, 8);
saltString.replaceAll(Pattern.quote("+"),Pattern.quote("."));
return createPasswdCompatibleHash(algo,password,saltString);
}
@MCElement(name="user", topLevel=false, id="staticUserDataProvider-user")
public static class User {
Map attributes = new HashMap();
public User() {}
public User(String username, String password){
setUsername(username);
setPassword(password);
}
public String getUsername() {
return attributes.get("username");
}
/**
* @description The user's login.
*/
@MCAttribute
public void setUsername(String value) {
attributes.put("username", value);
}
public String getPassword() {
return attributes.get("password");
}
/**
* @description The user's password.
*/
@MCAttribute
public void setPassword(String value) {
attributes.put("password", value);
}
public String getSms() {
return attributes.get("sms");
}
/**
* @description The user's phone number (if used in combination with the {@link TelekomSMSTokenProvider}).
*/
@MCAttribute
public void setSms(String value) {
attributes.put("sms", value);
}
public String getSecret() {
return attributes.get("secret");
}
/**
* @description The user's shared TOTP secret (if used in combination with the {@link TOTPTokenProvider}).
*/
@MCAttribute
public void setSecret(String value) {
attributes.put("secret", value);
}
public Map getAttributes() {
return attributes;
}
/**
* @description Other user attributes. For example any attribute starting with "header" will be added
* when HTTP requests are forwarded when authorized by this user.
*/
@MCOtherAttributes
public void setAttributes(Map attributes) {
for (Map.Entry e : attributes.entrySet())
this.attributes.put(e.getKey(), e.getValue());
}
}
public List getUsers() {
return users;
}
@MCChildElement
public void setUsers(List users) {
getUsersByName().clear();
for(User user : users){
getUsersByName().put(user.getUsername(), user);
}
this.users = users;
}
public Map getUsersByName() {
return usersByName;
}
public void setUsersByName(Map usersByName) {
this.usersByName = usersByName;
}
@Override
public void init(Router router) {
for (User user : users)
getUsersByName().put(user.getUsername(), user);
}
}