Java 讀取圖片檔案的型別（MimeType）
阿新 • • 發佈：2019-06-19
一、問題描述
在專案開發的時候，我們經常會遇到一類檔案上傳的問題，就是獲取圖片是哪種格式。很多情況下，很多人都是用字尾名去判斷，如下所示。
if(filename.endsWith(".png") || filename.endsWith(".jpg"))
{
//儲存圖片
}else{
throw new IOException("Error file format ！");
}
但是這種方式相當不可靠，我們可以嘗試將zip檔案、rmvb檔案、css、js修改後綴名位jpg或者png上傳，也可以上傳到伺服器，這就造成我們伺服器上出現了髒資料。此外，對於有些圖片檔案，修改成錯誤的副檔名，有些瀏覽器可能無法顯示出此圖片。
二、解決方案
在計算機系統中，媒體型別（MimeType）的檔案都有【識別符號】，zip、圖片本身屬於媒體檔案，因此我們可以通過編解碼的方式判斷圖片是否合法。
1、判斷標示方法
private static boolean isBMP(byte[] buf){
byte[] markBuf = "BM".getBytes(); //BMP圖片檔案的前兩個位元組
return compare(buf, markBuf);
}
private static boolean isICON(byte[] buf) {
byte[] markBuf = {0, 0, 1, 0, 1, 0, 32, 32};
return compare(buf, markBuf);
}
private static boolean isWEBP(byte[] buf) {
byte[] markBuf = "RIFF".getBytes(); //WebP圖片識別符
return compare(buf, markBuf);
}
private static boolean isGIF(byte[] buf) {
byte[] markBuf = "GIF89a".getBytes(); //GIF識別符
if(compare(buf, markBuf))
{
return true;
}
markBuf = "GIF87a".getBytes(); //GIF識別符
if(compare(buf, markBuf))
{
return true;
}
return false;
}
private static boolean isPNG(byte[] buf) {
byte[] markBuf = {(byte) 0x89,0x50,0x4E,0x47,0x0D,0x0A,0x1A,0x0A}; //PNG識別符
// new String(buf).indexOf("PNG")>0 //也可以使用這種方式
return compare(buf, markBuf);
}
private static boolean isJPEGHeader(byte[] buf) {
byte[] markBuf = {(byte) 0xff, (byte) 0xd8}; //JPEG開始符
return compare(buf, markBuf);
}
private static boolean isJPEGFooter(byte[] buf)//JPEG結束符
{
byte[] markBuf = {(byte) 0xff, (byte) 0xd9};
return compare(buf, markBuf);
}2、核心方法
/**
* 獲取檔案的mimeType
* @param filename
* @return
*/
private static String getMimeType(String filename){
try {
String mimeType = readType(filename);
return String.format("image/%s", mimeType);
} catch (IOException e) {
e.printStackTrace();
}
return null;
}
/**
* 讀取檔案型別
* @param filename
* @return
* @throws IOException
*/
private static String readType(String filename) throws IOException {
FileInputStream fis = null;
try {
File f = new File(filename);
if(!f.exists() || f.isDirectory() || f.length()<8) {
throw new IOException("the file ["+f.getAbsolutePath()+"] is not image !");
}
fis= new FileInputStream(f);
byte[] bufHeaders = readInputStreamAt(fis,0,8);
if(isJPEGHeader(bufHeaders))
{
long skiplength = f.length()-2-8; //第一次讀取時已經讀了8個byte,因此需要減掉
byte[] bufFooters = readInputStreamAt(fis, skiplength, 2);
if(isJPEGFooter(bufFooters))
{
return "jpeg";
}
}
if(isPNG(bufHeaders))
{
return "png";
}
if(isGIF(bufHeaders)){
return "gif";
}
if(isWEBP(bufHeaders))
{
return "webp";
}
if(isBMP(bufHeaders))
{
return "bmp";
}
if(isICON(bufHeaders))
{
return "ico";
}
throw new IOException("the image's format is unkown!");
} catch (FileNotFoundException e) {
throw e;
}finally{
try {
if(fis!=null) fis.close();
} catch (Exception e) {
}
}
}
/**
* 標示一致性比較
* @param buf 待檢測標示
* @param markBuf 識別符號位元組陣列
* @return 返回false標示標示不匹配
*/
private static boolean compare(byte[] buf, byte[] markBuf) {
for (int i = 0; i < markBuf.length; i++) {
byte b = markBuf[i];
byte a = buf[i];
if(a!=b){
return false;
}
}
return true;
}
/**
*
* @param fis 輸入流物件
* @param skiplength 跳過位置長度
* @param length 要讀取的長度
* @return 位元組陣列
* @throws IOException
*/
private static byte[] readInputStreamAt(FileInputStream fis, long skiplength, int length) throws IOException
{
byte[] buf = new byte[length];
fis.skip(skiplength); //
int read = fis.read(buf,0,length);
return buf;
}
3、測試程式碼
正常測試
public class ImageType {
public static void main(String[] args) {
String filename = "oschina.jpg";
String type = getMimeType(filename);
System.out.println(type);
}
}
輸出
image/jpeg修改副檔名測試
①修改oschina.jpeg為oschina.png
②複製oschina.png刪除副檔名
public class ImageType {
public static void main(String[] args) {
String filename = "oschina.png";
String type = getMimeType(filename);
System.out.println(type);
filename = "oschina";
type = getMimeType(filename);
System.out.println(type);
}
}
輸出
image/jpeg image/jpeg