SpringBoot中如何驗證用戶上傳的圖片資源

小編給大家分享一下SpringBoot中如何驗證用戶上傳的圖片資源，相信大部分人都還不怎么了解，因此分享這篇文章給大家參考一下，希望大家閱讀完這篇文章后大有收獲，下面讓我們一起去了解一下吧！

允許用戶上傳圖片資源（頭像，發帖）是APP常見的需求，特別需要把用戶的資源IO到磁盤情況下，需要防止壞人提交一些非法的文件，例如木馬，webshell，可執行程序等等。這類非法文件不僅會導致客戶端圖片資源顯示失敗，而且還會給服務器帶來安全問題。

通過文件后綴判斷文件的合法性

這種方式比較常見，也很簡單，是目前大多數APP選擇的做法。

public Object upload (@RequestParam("file") MultipartFile multipartFile) throws IllegalStateException, IOException {
	
	// 原始文件名稱
	String fileName = multipartFile.getOriginalFilename();
	
	// 解析到文件后綴，判斷是否合法
	int index = fileName.lastIndexOf(".");
	String suffix = null;
	if (index == -1 || (suffix = fileName.substring(index + 1)).isEmpty()) {
		return "文件后綴不能為空";
	}
	
	// 允許上傳的文件后綴列表
	Set<String> allowSuffix = new HashSet<>(Arrays.asList("jpg", "jpeg", "png", "gif"));
	if (!allowSuffix.contains(suffix.toLowerCase())) {
		return "非法的文件，不允許的文件類型：" + suffix;
	}
	
	// 序列化到磁盤中的文件上傳目錄， /upload
	// FileCopyUtils.copy 方法會自動關閉流資源
	FileCopyUtils.copy(multipartFile.getInputStream(), Files.newOutputStream(Paths.get("D://upload", fileName), StandardOpenOption.CREATE_NEW));
	
	// 返回相對訪問路徑，文件名極有可能帶中文或者空格等字符，進行uri編碼
	return  "/" + UriUtils.encode(fileName, StandardCharsets.UTF_8);
}

使用 ImageIO 判斷是否是圖片

這個方法就比較嚴格了，在判斷后綴的基礎上，使用Java的ImageIO類去加載圖片，嘗試讀取其寬高信息，如果不是合法的圖片資源。則無法讀取到這兩個數據。就算是把非法文件修改了后綴，也可以檢測出來。

public Object upload (@RequestParam("file") MultipartFile multipartFile) throws IllegalStateException, IOException {
	
	// 原始文件名稱
	String fileName = multipartFile.getOriginalFilename();
	
	// 解析到文件后綴
	int index = fileName.lastIndexOf(".");
	String suffix = null;
	if (index == -1 || (suffix = fileName.substring(index + 1)).isEmpty()) {
		return "文件后綴不能為空";
	}
	
	// 允許上傳的文件后綴列表
	Set<String> allowSuffix = new HashSet<>(Arrays.asList("jpg", "jpeg", "png", "gif"));
	if (!allowSuffix.contains(suffix.toLowerCase())) {
		return "非法的文件，不允許的文件類型：" + suffix;
	}
	
	// 臨時文件
	File tempFile = new File(System.getProperty("java.io.tmpdir"), fileName);
	
	try {
		// 先把文件序列化到臨時目錄
		multipartFile.transferTo(tempFile);
		try {
			// 嘗試IO文件，判斷文件的合法性
			BufferedImage  bufferedImage = ImageIO.read(tempFile);
			bufferedImage.getWidth();
			bufferedImage.getHeight();
		} catch (Exception e) {
			// IO異常，不是合法的圖片文件，返回異常信息
			return "文件不是圖片文件";
		}
		// 復制到到上傳目錄
		FileCopyUtils.copy(new FileInputStream(tempFile), Files.newOutputStream(Paths.get("D://upload", fileName), StandardOpenOption.CREATE_NEW));
		// 返回相對訪問路徑
		return  "/" + UriUtils.encode(fileName, StandardCharsets.UTF_8);
	} finally {
		// 響應客戶端后，始終刪除臨時文件
		tempFile.delete();
	}
}

以上是“SpringBoot中如何驗證用戶上傳的圖片資源”這篇文章的所有內容，感謝各位的閱讀！相信大家都有了一定的了解，希望分享的內容對大家有所幫助，如果還想學習更多知識，歡迎關注億速云行業資訊頻道！