SpringBoot 文件上传基于MD5 文件内容校验工具类

我就是我 2023-10-10 10:29 20阅读 0赞

1、业务场景：实现文件上传功能时。我们需要校验上传文件在传输过程中是否被注入脚本或者是被修改，所有采用md5 算法+ 文件内容生成一个识别码，传递给后端，让后端判断文件是否发生修改或变更。

import java.io.File;
    import java.io.IOException;
    import java.io.InputStream;
    import java.nio.file.Files;
    import java.nio.file.StandardOpenOption;
    import java.security.MessageDigest;
    import java.security.NoSuchAlgorithmException;
    
    /**
     * 上传文件校验工具类
     * @author zzg
     *
     */
    public class UploadFileUtil {
    	// 密码提花
    	private static final char[] hexCode = "0123456789ABCDEF".toCharArray();
    	
    	// 文件类取MD5
        public static String calcMD5(File file){
            try (InputStream stream = Files.newInputStream(file.toPath(), StandardOpenOption.READ)) {
                return calcMD5(stream);
            }catch (IOException e) {
                e.printStackTrace();
                return "";
            }
        }
        // 输入流取MD5
        public static String calcMD5(InputStream stream) {
            try {
                MessageDigest digest = MessageDigest.getInstance("MD5");
                byte[] buf = new byte[8192];
                int len;
                while ((len = stream.read(buf)) > 0) {
                    digest.update(buf, 0, len);
                }
                return toHexString(digest.digest());
            } catch (IOException e) {
                e.printStackTrace();
                return "";
            } catch (NoSuchAlgorithmException e) {
                e.printStackTrace();
                return "";
            }
        }
    
        public static String toHexString(byte[] data) {
            StringBuilder r = new StringBuilder(data.length * 2);
            for (byte b : data) {
                r.append(hexCode[(b >> 4) & 0xF]);
                r.append(hexCode[(b & 0xF)]);
            }
            return r.toString();
        }
    	
    }