Hash工具类（MD5,SHA-256,SHA-1）

最新推荐文章于 2024-07-17 18:45:00 发布

顽皮的小木藕

最新推荐文章于 2024-07-17 18:45:00 发布

阅读量294

点赞数 1

分类专栏： Java 文章标签： java

本文链接：https://blog.csdn.net/weixin_44240277/article/details/115730063

版权

Java 专栏收录该内容

4 篇文章 0 订阅

订阅专栏

这段代码提供了计算数据的MD5、SHA-256和SHA-1哈希值的方法。它接受RandomAccessFile、BufferedInputStream或字节数组作为输入，并确保数据转换为合适的格式。如果输入不是预定义类型，会抛出异常。更新消息摘要的过程使用了16KB的缓冲区进行优化。

摘要由CSDN通过智能技术生成

/**
     * Returns MD5 hash of given data and it's length.
     *
     * @param data must be {@link RandomAccessFile}, {@link BufferedInputStream} or byte array.
     * @param len length of data to be read for hash calculation.
     */
    public static String md5Hash(Object data, int len)
            throws NoSuchAlgorithmException, IOException, InsufficientDataException, XmlParserException {

        if (data != null
                && !(data instanceof InputStream
                || data instanceof RandomAccessFile
                || data instanceof byte[])) {
            byte[] bytes;
            if (data instanceof CharSequence) {
                bytes = data.toString().getBytes(StandardCharsets.UTF_8);
            } else {
                bytes = Xml.marshal(data).getBytes(StandardCharsets.UTF_8);
            }

            data = bytes;
            len = bytes.length;
        }

        MessageDigest md5Digest = MessageDigest.getInstance("MD5");

        if (data instanceof BufferedInputStream || data instanceof RandomAccessFile) {
            updateDigests(data, len, null, md5Digest);
        } else if (data instanceof byte[]) {
            md5Digest.update((byte[]) data, 0, len);
        } else {
            throw new VianException(
                    "Unknown data source to calculate sha256 hash. This should not happen.");
        }

//        return BaseEncoding.base64().encode(md5Digest.digest());
        return byteArrToHex(md5Digest.digest());
    }

    /**
     * Returns MD5 hash of given data and it's length.
     *
     * @param data must be {@link RandomAccessFile}, {@link BufferedInputStream} or byte array.
     * @param len length of data to be read for hash calculation.
     */
    public static String sha256Hash(Object data, int len)
            throws NoSuchAlgorithmException, IOException, InsufficientDataException, XmlParserException {

        if (data != null
                && !(data instanceof InputStream
                || data instanceof RandomAccessFile
                || data instanceof byte[])) {
            byte[] bytes;
            if (data instanceof CharSequence) {
                bytes = data.toString().getBytes(StandardCharsets.UTF_8);
            } else {
                bytes = Xml.marshal(data).getBytes(StandardCharsets.UTF_8);
            }

            data = bytes;
            len = bytes.length;
        }

        MessageDigest sha256Digest = MessageDigest.getInstance("SHA-256");

        if (data instanceof BufferedInputStream || data instanceof RandomAccessFile) {
            updateDigests(data, len, sha256Digest, null);
        } else if (data instanceof byte[]) {
            sha256Digest.update((byte[]) data, 0, len);
        } else {
            throw new VianException(
                    "Unknown data source to calculate sha256 hash. This should not happen.");
        }

//        return BaseEncoding.base64().encode(md5Digest.digest());
        return byteArrToHex(sha256Digest.digest());
    }

    /**
     * Returns MD5 hash of given data and it's length.
     *
     * @param data must be {@link RandomAccessFile}, {@link BufferedInputStream} or byte array.
     * @param len length of data to be read for hash calculation.
     */
    public static String sha1Hash(Object data, int len)
            throws NoSuchAlgorithmException, IOException, InsufficientDataException, XmlParserException {

        if (data != null
                && !(data instanceof InputStream
                || data instanceof RandomAccessFile
                || data instanceof byte[])) {
            byte[] bytes;
            if (data instanceof CharSequence) {
                bytes = data.toString().getBytes(StandardCharsets.UTF_8);
            } else {
                bytes = Xml.marshal(data).getBytes(StandardCharsets.UTF_8);
            }

            data = bytes;
            len = bytes.length;
        }

        MessageDigest md5Digest = MessageDigest.getInstance("SHA-1");

        if (data instanceof BufferedInputStream || data instanceof RandomAccessFile) {
            updateDigests(data, len, null, md5Digest);
        } else if (data instanceof byte[]) {
            md5Digest.update((byte[]) data, 0, len);
        } else {
            throw new VianException(
                    "Unknown data source to calculate sha256 hash. This should not happen.");
        }

//        return BaseEncoding.base64().encode(md5Digest.digest());
        return byteArrToHex(md5Digest.digest());
    }

    /** Updated MessageDigest with bytes read from file and stream. */
    private static int updateDigests(
            Object inputStream, int len, MessageDigest sha256Digest, MessageDigest md5Digest)
            throws IOException, InsufficientDataException {
        RandomAccessFile file = null;
        BufferedInputStream stream = null;
        if (inputStream instanceof RandomAccessFile) {
            file = (RandomAccessFile) inputStream;
        } else if (inputStream instanceof BufferedInputStream) {
            stream = (BufferedInputStream) inputStream;
        }

        // hold current position of file/stream to reset back to this position.
        long pos = 0;
        if (file != null) {
            pos = file.getFilePointer();
        } else {
            stream.mark(len);
        }

        // 16KiB buffer for optimization
        byte[] buf = new byte[16384];
        int bytesToRead = buf.length;
        int bytesRead = 0;
        int totalBytesRead = 0;
        while (totalBytesRead < len) {
            if ((len - totalBytesRead) < bytesToRead) {
                bytesToRead = len - totalBytesRead;
            }

            if (file != null) {
                bytesRead = file.read(buf, 0, bytesToRead);
            } else {
                bytesRead = stream.read(buf, 0, bytesToRead);
            }

            if (bytesRead < 0) {
                // reached EOF
                throw new InsufficientDataException(
                        "Insufficient data.  bytes read " + totalBytesRead + " expected " + len);
            }

            if (bytesRead > 0) {
                if (sha256Digest != null) {
                    sha256Digest.update(buf, 0, bytesRead);
                }

                if (md5Digest != null) {
                    md5Digest.update(buf, 0, bytesRead);
                }

                totalBytesRead += bytesRead;
            }
        }

        // reset back to saved position.
        if (file != null) {
            file.seek(pos);
        } else {
            stream.reset();
        }

        return totalBytesRead;
    }

    /**
     * 数组转16进制字符串
     * @param btArr
     * @return
     */
    private static String byteArrToHex(byte[] btArr) {
        char HexCharArr[] = {'0','1','2','3','4','5','6','7','8','9','a','b','c','d','e','f'};
        char strArr[] = new char[btArr.length * 2];
        int i = 0;
        for (byte bt : btArr) {
            strArr[i++] = HexCharArr[bt>>>4 & 0xf];
            strArr[i++] = HexCharArr[bt & 0xf];
        }
        return new String(strArr);
    }