通过后缀名和MIME-TYPE检查实现文件类型校验

最新推荐文章于 2024-08-13 14:46:30 发布

Vigoooo

最新推荐文章于 2024-08-13 14:46:30 发布

阅读量2.4k

点赞数 2

本文链接：https://blog.csdn.net/qq_36466774/article/details/88826083

版权

本文介绍了在开发中如何进行文件类型校验，特别是通过后缀名和MIME-TYPE进行安全控制。提供了一种先定义文件类型白名单，再分别进行后缀名和mime-type校验的实现方法，其中涉及到使用第三方库来获取文件的mimeType。

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

前言

文件上传是一个在开发中很常见的需求场景,通常出于安全考虑,我们会对上传的文件进行类型校验,其中常见的有后缀名校验,mime-type校验

话不多说,直接上代码

1.首先定义允许上传的文件类型白名单

private static final String[] suffixWhiteList = {"PNG","JPEG","JPG","GIF"};
private static final String[] mimeTypeWhiteList = {"image/jpeg","image/gif","image/png"};

2.后缀名校验

    /**
     * 文件后缀名校验
     *
     * @param fileName
     *            文件名称
     * @return
     */
    public static boolean suffixCheck(String fileName) {
        if(fileName == null || "".equals(fileName)){
            return false;
        }
        //从最后一个点之后截取字符串
        String suffix = fileName.substring(fileName.lastIndexOf(".") + 1);
        //白名单匹配
        boolean anyMatch = Arrays.stream(suffixWhiteList).anyMatch(x -> x.equalsIgnoreCase(suffix));
        return anyMatch;
    }