java-springboot文件上传校验之只允许上传excel文件,且检查不能是脚本或者有害文件或可行性文件

  1. 四重验证机制

    • 文件扩展名检查(.xlsx/.xls)
    • MIME类型检查
    • 文件魔数验证(真实文件类型)
    • 可执行文件特征检测
  2. 防御措施

    • 使用try-with-resources确保流关闭
    • 限制文件大小防止DoS攻击
    • 使用Apache POI的FileMagic进行专业验证
  3. 生产环境建议

    Yaml

    # application.yml配置 spring: servlet: multipart: max-file-size: 10MB max-request-size: 10MB

完整代码:

java 复制代码
import org.apache.poi.poifs.filesystem.FileMagic;
import org.springframework.web.multipart.MultipartFile;

import java.io.IOException;
import java.io.InputStream;
import java.util.Arrays;

// 文件上传限制-只允许上传excel文件,且检查不能是脚本或者有害文件或可行性文件
public class ExcelFileValidator {

    // 允许的Excel文件MIME类型
    private static final String[] ALLOWED_MIME_TYPES = {
            "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet", // .xlsx
            "application/vnd.ms-excel" // .xls
    };

    // 最大文件大小(10MB)
    private static final long MAX_FILE_SIZE = 10 * 1024 * 1024;

    /**
     * 验证Excel文件安全性
     *
     * @param file 上传的文件
     * @throws IOException              文件读取异常
     * @throws IllegalArgumentException 文件非法时抛出
     */
    public static void validateExcelFile(MultipartFile file) throws IOException, IllegalArgumentException {
        // 基础检查
        if (file == null || file.isEmpty()) {
            throw new IllegalArgumentException("请选择要上传的文件");
        }

        // 检查文件大小
        if (file.getSize() > MAX_FILE_SIZE) {
            throw new IllegalArgumentException("Excel文件大小不能超过10MB");
        }

        // 检查文件扩展名
        String originalFilename = file.getOriginalFilename();
        if (originalFilename == null ||
                (!originalFilename.toLowerCase().endsWith(".xlsx") &&
                        !originalFilename.toLowerCase().endsWith(".xls"))) {
            throw new IllegalArgumentException("仅支持.xlsx或.xls格式的Excel文件");
        }

        // 检查MIME类型
        String contentType = file.getContentType();
        if (contentType == null || !Arrays.asList(ALLOWED_MIME_TYPES).contains(contentType.toLowerCase())) {
            throw new IllegalArgumentException("非法的Excel文件类型");
        }

        // 使用POI检查文件魔数(真实文件类型)
        try (InputStream inputStream = file.getInputStream()) {
            FileMagic fileMagic = FileMagic.valueOf(inputStream);
            if (fileMagic != FileMagic.OLE2 && fileMagic != FileMagic.OOXML) {
                throw new IllegalArgumentException("非法的Excel文件格式");
            }

            // 基础恶意内容检查
            checkForExecutableContent(inputStream);
        }
    }

    /**
     * 检查是否包含可执行文件特征
     */
    private static void checkForExecutableContent(InputStream is) throws IOException {
        byte[] buffer = new byte[1024];
        is.read(buffer);

        // PE文件头检查(Windows可执行文件)
        if (buffer.length > 60 && buffer[0] == 0x4D && buffer[1] == 0x5A) {
            throw new IllegalArgumentException("检测到潜在有害文件内容");
        }

        // ELF文件头检查(Linux可执行文件)
        if (buffer.length > 4 && buffer[0] == 0x7F && buffer[1] == 0x45 &&
                buffer[2] == 0x4C && buffer[3] == 0x46) {
            throw new IllegalArgumentException("检测到潜在有害文件内容");
        }
    }
}
相关推荐
万能程序员-传康Kk几秒前
智能教育个性化学习平台-java
java·开发语言·学习
落笔画忧愁e10 分钟前
扣子Coze飞书多维表插件-列出全部数据表
java·服务器·飞书
鱼儿也有烦恼13 分钟前
Elasticsearch最新入门教程
java·elasticsearch·kibana
路修19 分钟前
Spring Boot + Elasticsearch + HBase 构建海量数据搜索系统
spring boot·elasticsearch·hbase
eternal__day23 分钟前
微服务架构下的服务注册与发现:Eureka 深度解析
java·spring cloud·微服务·eureka·架构·maven
一介草民丶29 分钟前
Jenkins | Linux环境部署Jenkins与部署java项目
java·linux·jenkins
武子康38 分钟前
Java-39 深入浅出 Spring - AOP切面增强 核心概念 通知类型 XML+注解方式 附代码
xml·java·大数据·开发语言·后端·spring
喜欢踢足球的老罗1 小时前
使用 Spring Boot 3.3 和 JdbcTemplate 操作 MySQL 数据库
数据库·spring boot·mysql
米粉03051 小时前
SpringBoot核心注解详解及3.0与2.0版本深度对比
java·spring boot·后端
一只帆記2 小时前
SpringBoot EhCache 缓存
spring boot·后端·缓存