/*
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF 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.apache.shiro.crypto.hash;
import org.apache.shiro.codec.Base64;
import org.apache.shiro.codec.CodecException;
import org.apache.shiro.codec.Hex;
import org.apache.shiro.crypto.UnknownAlgorithmException;
import org.apache.shiro.util.ByteSource;
import org.apache.shiro.util.StringUtils;
import java.security.MessageDigest;
import java.security.NoSuchAlgorithmException;
import java.util.Arrays;
/**
* A {@code Hash} implementation that allows any {@link java.security.MessageDigest MessageDigest} algorithm name to
* be used. This class is a less type-safe variant than the other {@code AbstractHash} subclasses
* (e.g. {@link Sha512Hash}, etc), but it does allow for any algorithm name to be specified in case the other subclass
* implementations do not represent an algorithm that you may want to use.
* <p/>
* As of Shiro 1.1, this class effectively replaces the (now-deprecated) {@link AbstractHash} class. It subclasses
* {@code AbstractHash} only to retain backwards-compatibility.
*
* @since 1.1
*/
public class SimpleHash extends AbstractHash {
private static final int DEFAULT_ITERATIONS = 1;
/**
* The {@link java.security.MessageDigest MessageDigest} algorithm name to use when performing the hash.
*/
private final String algorithmName;
/**
* The hashed data
*/
private byte[] bytes;
/**
* Supplied salt, if any.
*/
private ByteSource salt;
/**
* Number of hash iterations to perform. Defaults to 1 in the constructor.
*/
private int iterations;
/**
* Cached value of the {@link #toHex() toHex()} call so multiple calls won't incur repeated overhead.
*/
private transient String hexEncoded = null;
/**
* Cached value of the {@link #toBase64() toBase64()} call so multiple calls won't incur repeated overhead.
*/
private transient String base64Encoded = null;
/**
* Creates an new instance with only its {@code algorithmName} set - no hashing is performed.
* <p/>
* Because all other constructors in this class hash the {@code source} constructor argument, this
* constructor is useful in scenarios when you have a byte array that you know is already hashed and
* just want to set the bytes in their raw form directly on an instance. After using this constructor,
* you can then immediately call {@link #setBytes setBytes} to have a fully-initialized instance.
* <p/>
* <b>N.B.</b>The algorithm identified by the {@code algorithmName} parameter must be available on the JVM. If it
* is not, a {@link UnknownAlgorithmException} will be thrown when the hash is performed (not at instantiation).
*
* @param algorithmName the {@link java.security.MessageDigest MessageDigest} algorithm name to use when
* performing the hash.
* @see UnknownAlgorithmException
*/
public SimpleHash(String algorithmName) {
this.algorithmName = algorithmName;
this.iterations = DEFAULT_ITERATIONS;
}
/**
* Creates an {@code algorithmName}-specific hash of the specified {@code source} with no {@code salt} using a
* single hash iteration.
* <p/>
* This is a convenience constructor that merely executes <code>this( algorithmName, source, null, 1);</code>.
* <p/>
* Please see the
* {@link #SimpleHash(String algorithmName, Object source, Object salt, int numIterations) SimpleHashHash(algorithmName, Object,Object,int)}
* constructor for the types of Objects that may be passed into this constructor, as well as how to support further
* types.
*
* @param algorithmName the {@link java.security.MessageDigest MessageDigest} algorithm name to use when
* performing the hash.
* @param source the object to be hashed.
* @throws org.apache.shiro.codec.CodecException
* if the specified {@code source} cannot be converted into a byte array (byte[]).
* @throws UnknownAlgorithmException if the {@code algorithmName} is not available.
*/
public SimpleHash(String algorithmName, Object source) throws CodecException, UnknownAlgorithmException {
//noinspection NullableProblems
this(algorithmName, source, null, DEFAULT_ITERATIONS);
}
/**
* Creates an {@code algorithmName}-specific hash of the specified {@code source} using the given {@code salt}
* using a single hash iteration.
* <p/>
* It is a convenience constructor that merely executes <code>this( algorithmName, source, salt, 1);</code>.
* <p/>
* Please see the
* {@link #SimpleHash(String algorithmName, Object source, Object salt, int numIterations) SimpleHashHash(algorithmName, Object,Object,int)}
* constructor for the types of Objects that may be passed into this constructor, as well as how to support further
* types.
*
* @param algorithmName the {@link java.security.MessageDigest MessageDigest} algorithm name to use when
* performing the hash.
* @param source the source object to be hashed.
* @param salt the salt to use for the hash
* @throws CodecException if either constructor argument cannot be converted into a byte array.
* @throws UnknownAlgorithmException if the {@code algorithmName} is not available.
*/
public SimpleHash(String algorithmName, Object source, Object salt) throws CodecException, UnknownAlgorithmException {
this(algorithmName, source, salt, DEFAULT_ITERATIONS);
}
/**
* Creates an {@code algorithmName}-specific hash of the specified {@code source} using the given
* {@code salt} a total of {@code hashIterations} times.
* <p/>
* By default, this class only supports Object method arguments of
* type {@code byte[]}, {@code char[]}, {@link String}, {@link java.io.File File},
* {@link java.io.InputStream InputStream} or {@link org.apache.shiro.util.ByteSource ByteSource}. If either
* argument is anything other than these types a {@link org.apache.shiro.codec.CodecException CodecException}
* will be thrown.
* <p/>
* If you want to be able to hash other object types, or use other salt types, you need to override the
* {@link #toBytes(Object) toBytes(Object)} method to support those specific types. Your other option is to
* convert your arguments to one of the default supported types first before passing them in to this
* constructor}.
*
* @param algorithmName the {@link java.security.MessageDigest MessageDigest} algorithm name to use when
* performing the hash.
* @param source the source object to be hashed.
* @param salt the salt to use for the hash
* @param hashIterations the number of times the {@code source} argument hashed for attack resiliency.
* @throws CodecException if either Object constructor argument cannot be converted into a byte array.
* @throws UnknownAlgorithmException if the {@code algorithmName} is not available.
*/
public SimpleHash(String algorithmName, Object source, Object salt, int hashIterations)
throws CodecException, UnknownAlgorithmException {
if (!StringUtils.hasText(algorithmName)) {
throw new NullPointerException("algorithmName argument cannot be null or empty.");
}
this.algorithmName = algorithmName;
this.iterations = Math.max(DEFAULT_ITERATIONS, hashIterations);
ByteSource saltBytes = null;
if (salt != null) {
saltBytes = convertSaltToBytes(salt);
this.salt = saltBytes;
}
ByteSource sourceBytes = convertSourceToBytes(source);
hash(sourceBytes, saltBytes, hashIterations);
}
/**
* Acquires the specified {@code source} argument's bytes and returns them in the form of a {@code ByteSource} instance.
* <p/>
* This implementation merely delegates to the convenience {@link #toByteSource(Object)} method for generic
* conversion. Can be overridden by subclasses for source-specific conversion.
*
* @param source the source object to be hashed.
* @return the source's bytes in the form of a {@code ByteSource} instance.
* @since 1.2
*/
protected ByteSource convertSourceToBytes(Object source) {
return toByteSource(source);
}
/**
* Acquires the specified {@code salt} argument's bytes and returns them in the form of a {@code ByteSource} instance.
* <p/>
* This implementation merely delegates to the convenience {@link #toByteSource(Object)} method for generic
* conversion. Can be overridden by subclasses for salt-specific conversion.
*
* @param salt the salt to be use for the hash.
* @return the salt's bytes in the form of a {@code ByteSource} instance.
* @since 1.2
*/
protected ByteSource convertSaltToBytes(Object salt) {
return toByteSource(salt);
}
/**
* Converts a given object into a {@code ByteSource} instance. Assumes the object can be converted to bytes.
*
* @param o the Object to convert into a {@code ByteSource} instance.
* @return the {@code ByteSource} representation of the specified object's bytes.
* @since 1.2
*/
protected ByteSource toByteSource(Object o) {
if (o == null) {
return null;
}
if (o instanceof ByteSource) {
return (ByteSource) o;
}
byte[] bytes = toBytes(o);
return ByteSource.Util.bytes(bytes);
}
private void hash(ByteSource source, ByteSource salt, int hashIterations) throws CodecException, UnknownAlgorithmException {
byte[] saltBytes = salt != null ? salt.getBytes() : null;
byte[] hashedBytes = hash(source.getBytes(), saltBytes, hashIterations);
setBytes(hashedBytes);
}
/**
* Returns the {@link java.security.MessageDigest MessageDigest} algorithm name to use when performing the hash.
*
* @return the {@link java.security.MessageDigest MessageDigest} algorithm name to use when performing the hash.
*/
public String getAlgorithmName() {
return this.algorithmName;
}
public ByteSource getSalt() {
return this.salt;
}
public int getIterations() {
return this.iterations;
}
public byte[] getBytes() {
return this.bytes;
}
/**
* Sets the raw bytes stored by this hash instance.
* <p/>
* The bytes are kept in raw form - they will not be hashed/changed. This is primarily a utility method for
* constructing a Hash instance when the hashed value is already known.
*
* @param alreadyHashedBytes the raw already-hashed bytes to store in this instance.
*/
public void setBytes(byte[] alreadyHashedBytes) {
this.bytes = alreadyHashedBytes;
this.hexEncoded = null;
this.base64Encoded = null;
}
/**
* Sets the iterations used to previously compute AN ALREADY GENERATED HASH.
* <p/>
* This is provided <em>ONLY</em> to reconstitute an already-created Hash instance. It should ONLY ever be
* invoked when re-constructing a hash instance from an already-hashed value.
*
* @param iterations the number of hash iterations used to previously create the hash/digest.
* @since 1.2
*/
public void setIterations(int iterations) {
this.iterations = Math.max(DEFAULT_ITERATIONS, iterations);
}
/**
* Sets the salt used to previously compute AN ALREADY GENERATED HASH.
* <p/>
* This is provided <em>ONLY</em> to reconstitute a Hash instance that has already been computed. It should ONLY
* ever be invoked when re-constructing a hash instance from an already-hashed value.
*
* @param salt the salt used to previously create the hash/digest.
* @since 1.2
*/
public void setSalt(ByteSource salt) {
this.salt = salt;
}
/**
* Returns the JDK MessageDigest instance to use for executing the hash.
*
* @param algorithmName the algorithm to use for the hash, provided by subclasses.
* @return the MessageDigest object for the specified {@code algorithm}.
* @throws UnknownAlgorithmException if the specified algorithm name is not available.
*/
protected MessageDigest getDigest(String algorithmName) throws UnknownAlgorithmException {
try {
return MessageDigest.getInstance(algorithmName);
} catch (NoSuchAlgorithmException e) {
String msg = "No native '" + algorithmName + "' MessageDigest instance available on the current JVM.";
throw new UnknownAlgorithmException(msg, e);
}
}
/**
* Hashes the specified byte array without a salt for a single iteration.
*
* @param bytes the bytes to hash.
* @return the hashed bytes.
* @throws UnknownAlgorithmException if the configured {@link #getAlgorithmName() algorithmName} is not available.
*/
protected byte[] hash(byte[] bytes) throws UnknownAlgorithmException {
return hash(bytes, null, DEFAULT_ITERATIONS);
}
/**
* Hashes the specified byte array using the given {@code salt} for a single iteration.
*
* @param bytes the bytes to hash
* @param salt the salt to use for the initial hash
* @return the hashed bytes
* @throws UnknownAlgorithmException if the configured {@link #getAlgorithmName() algorithmName} is not available.
*/
protected byte[] hash(byte[] bytes, byte[] salt) throws UnknownAlgorithmException {
return hash(bytes, salt, DEFAULT_ITERATIONS);
}
/**
* Hashes the specified byte array using the given {@code salt} for the specified number of iterations.
*
* @param bytes the bytes to hash
* @param salt the salt to use for the initial hash
* @param hashIterations the number of times the the {@code bytes} will be hashed (for attack resiliency).
* @return the hashed bytes.
* @throws UnknownAlgorithmException if the {@link #getAlgorithmName() algorithmName} is not available.
*/
protected byte[] hash(byte[] bytes, byte[] salt, int hashIterations) throws UnknownAlgorithmException {
MessageDigest digest = getDigest(getAlgorithmName());
if (salt != null) {
digest.reset();
digest.update(salt);
}
byte[] hashed = digest.digest(bytes);
int iterations = hashIterations - 1; //already hashed once above
//iterate remaining number:
for (int i = 0; i < iterations; i++) {
digest.reset();
hashed = digest.digest(hashed);
}
return hashed;
}
public boolean isEmpty() {
return this.bytes == null || this.bytes.length == 0;
}
/**
* Returns a hex-encoded string of the underlying {@link #getBytes byte array}.
* <p/>
* This implementation caches the resulting hex string so multiple calls to this method remain efficient.
* However, calling {@link #setBytes setBytes} will null the cached value, forcing it to be recalculated the
* next time this method is called.
*
* @return a hex-encoded string of the underlying {@link #getBytes byte array}.
*/
public String toHex() {
if (this.hexEncoded == null) {
this.hexEncoded = Hex.encodeToString(getBytes());
}
return this.hexEncoded;
}
/**
* Returns a Base64-encoded string of the underlying {@link #getBytes byte array}.
* <p/>
* This implementation caches the resulting Base64 string so multiple calls to this method remain efficient.
* However, calling {@link #setBytes setBytes} will null the cached value, forcing it to be recalculated the
* next time this method is called.
*
* @return a Base64-encoded string of the underlying {@link #getBytes byte array}.
*/
public String toBase64() {
if (this.base64Encoded == null) {
//cache result in case this method is called multiple times.
this.base64Encoded = Base64.encodeToString(getBytes());
}
return this.base64Encoded;
}
/**
* Simple implementation that merely returns {@link #toHex() toHex()}.
*
* @return the {@link #toHex() toHex()} value.
*/
public String toString() {
return toHex();
}
/**
* Returns {@code true} if the specified object is a Hash and its {@link #getBytes byte array} is identical to
* this Hash's byte array, {@code false} otherwise.
*
* @param o the object (Hash) to check for equality.
* @return {@code true} if the specified object is a Hash and its {@link #getBytes byte array} is identical to
* this Hash's byte array, {@code false} otherwise.
*/
public boolean equals(Object o) {
if (o instanceof Hash) {
Hash other = (Hash) o;
return MessageDigest.isEqual(getBytes(), other.getBytes());
}
return false;
}
/**
* Simply returns toHex().hashCode();
*
* @return toHex().hashCode()
*/
public int hashCode() {
if (this.bytes == null || this.bytes.length == 0) {
return 0;
}
return Arrays.hashCode(this.bytes);
}
}