快捷導(dǎo)航

Java實(shí)現(xiàn)限制文件上傳類型功能

更新時間：2025年07月03日 10:33:54 作者：思靜魚

這篇文章主要為大家詳細(xì)介紹了在Java中如何實(shí)現(xiàn)限制文件上傳類型功能,文中的示例代碼講解詳細(xì),感興趣的小伙伴可以跟隨小編一起學(xué)習(xí)一下

一、限制文件類型的常見做法

方法 1：根據(jù)MultipartFile.getContentType()判斷 MIME 類型

public void validateFileType(MultipartFile file) {
    List<String> allowedTypes = List.of("image/png", "image/jpeg", "application/pdf");
    String contentType = file.getContentType();

    if (!allowedTypes.contains(contentType)) {
        throw new IllegalArgumentException("不支持的文件類型: " + contentType);
    }
}

注意：瀏覽器偽造 Content-Type 是可能的，因此推薦再結(jié)合后綴判斷或做文件頭檢查（Magic Number）

方法 2：根據(jù)文件名后綴判斷類型（不安全但能擋住大部分無意誤傳）

public void validateExtension(MultipartFile file) {
    String filename = file.getOriginalFilename().toLowerCase();
    List<String> allowedExtensions = List.of(".jpg", ".jpeg", ".png", ".pdf");

    if (allowedExtensions.stream().noneMatch(filename::endsWith)) {
        throw new IllegalArgumentException("不支持的文件后綴");
    }
}

方法 3：安全性更高 —— 檢查文件頭（Magic Number）

每種文件格式都有“魔數(shù)”（Magic Number），可使用第三方庫如 Apache Tika、JMimeMagic、FileTypeDetector 來檢測。

使用 Apache Tika 示例：

import org.apache.tika.Tika;

public void validateByTika(MultipartFile file) throws IOException {
    Tika tika = new Tika();
    String detectedType = tika.detect(file.getInputStream());

    List<String> allowedTypes = List.of("image/jpeg", "image/png", "application/pdf");
    if (!allowedTypes.contains(detectedType)) {
        throw new IllegalArgumentException("文件類型不合法: " + detectedType);
    }
}

二、實(shí)戰(zhàn)應(yīng)用整合到上傳接口中（Spring Boot）

@PostMapping("/upload")
public ResponseEntity<?> upload(@RequestParam MultipartFile file) throws IOException {
    validateFileType(file);      // 判斷 MIME 類型
    validateExtension(file);     // 判斷文件名后綴
    validateByTika(file);        // 推薦：更準(zhǔn)確安全

    // 保存邏輯略...
    return ResponseEntity.ok("上傳成功");
}

三、可配置化（支持配置文件控制允許類型）

application.yml

file:
  allowed-types:
    - image/png
    - image/jpeg
    - application/pdf

配置類注入

@ConfigurationProperties(prefix = "file")
@Data
public class FileConfig {
    private List<String> allowedTypes;
}

再注入 FileConfig 來校驗(yàn)：

@Autowired
private FileConfig fileConfig;

if (!fileConfig.getAllowedTypes().contains(contentType)) {
    throw new IllegalArgumentException("不支持的文件類型");
}

四、其他建議

場景	建議
前端限制文件選擇	<input accept="image/*">
后端強(qiáng)校驗(yàn)必不可少	防止繞過前端
存儲前統(tǒng)一重命名處理	防止惡意腳本如 xx.jpg.exe
加白名單優(yōu)于黑名單	拒絕所有不認(rèn)識的類型

到此這篇關(guān)于Java實(shí)現(xiàn)限制文件上傳類型功能的文章就介紹到這了,更多相關(guān)Java限制文件上傳類型內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: