org.wso2.andes.server.security.auth.database.HashedUser.java Source code

Introduction

Here is the source code for org.wso2.andes.server.security.auth.database.HashedUser.java

Source

/*
 * Copyright (c) 2005-2014, WSO2 Inc. (http://www.wso2.org) All Rights Reserved.
 *
 * WSO2 Inc. licenses this file to you 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 org.wso2.andes.server.security.auth.database;

import org.apache.commons.codec.EncoderException;
import org.apache.commons.codec.binary.Base64;
import org.apache.log4j.Logger;

import java.io.UnsupportedEncodingException;
import java.security.MessageDigest;
import java.security.NoSuchAlgorithmException;
import java.security.Principal;

public class HashedUser implements Principal {
    private static final Logger _logger = Logger.getLogger(HashedUser.class);

    String _name;
    char[] _password;
    byte[] _encodedPassword = null;
    private boolean _modified = false;
    private boolean _deleted = false;

    HashedUser(String[] data) throws UnsupportedEncodingException {
        if (data.length != 2) {
            throw new IllegalArgumentException("User Data should be length 2, username, password");
        }

        _name = data[0];

        byte[] encoded_password = data[1].getBytes(Base64MD5PasswordFilePrincipalDatabase.DEFAULT_ENCODING);

        Base64 b64 = new Base64();
        byte[] decoded = b64.decode(encoded_password);

        _encodedPassword = encoded_password;

        _password = new char[decoded.length];

        int index = 0;
        for (byte c : decoded) {
            _password[index++] = (char) c;
        }
    }

    public HashedUser(String name, char[] password) throws UnsupportedEncodingException, NoSuchAlgorithmException {
        _name = name;
        setPassword(password, false);
    }

    public static byte[] getMD5(byte[] data) throws NoSuchAlgorithmException, UnsupportedEncodingException {
        MessageDigest md = MessageDigest.getInstance("MD5");

        for (byte b : data) {
            md.update(b);
        }

        return md.digest();
    }

    public String getName() {
        return _name;
    }

    public String toString() {
        return _name;
    }

    char[] getPassword() {
        return _password;
    }

    void setPassword(char[] password, boolean alreadyHashed)
            throws UnsupportedEncodingException, NoSuchAlgorithmException {
        if (alreadyHashed) {
            _password = password;
        } else {
            byte[] byteArray = new byte[password.length];
            int index = 0;
            for (char c : password) {
                byteArray[index++] = (byte) c;
            }

            byte[] MD5byteArray = getMD5(byteArray);

            _password = new char[MD5byteArray.length];

            index = 0;
            for (byte c : MD5byteArray) {
                _password[index++] = (char) c;
            }
        }

        _modified = true;
        _encodedPassword = null;
    }

    byte[] getEncodedPassword() throws EncoderException, UnsupportedEncodingException, NoSuchAlgorithmException {
        if (_encodedPassword == null) {
            encodePassword();
        }
        return _encodedPassword;
    }

    private void encodePassword() throws EncoderException, UnsupportedEncodingException, NoSuchAlgorithmException {
        byte[] byteArray = new byte[_password.length];
        int index = 0;
        for (char c : _password) {
            byteArray[index++] = (byte) c;
        }
        _encodedPassword = (new Base64()).encode(byteArray);
    }

    public boolean isModified() {
        return _modified;
    }

    public boolean isDeleted() {
        return _deleted;
    }

    public void delete() {
        _deleted = true;
    }

    public void saved() {
        _modified = false;
    }

}