一、问题描述
在项目开发的时候，我们经常会遇到一类文件上传的问题，就是获取图片是哪种格式。很多情况下，很多人都是用后缀名去判断，如下所示。
|
1
2
3
4
5
6
|
if(filename.endsWith(".png") || filename.endsWith(".jpg"))
{
//保存图片
}else{
throw new IOException("Error file format ！");
}
|
但是这种方式相当不可靠，我们可以尝试将zip文件、rmvb文件、css、js修改后缀名位jpg或者png上传，也可以上传到服务器，这就造成我们服务器上出现了脏数据。此外，对于有些图片文件，修改成错误的扩展名，有些浏览器可能无法显示出此图片。
二、解决方案
在计算机系统中，媒体类型的文件都有【标识符】，zip、图片本身属于媒体文件，因此我们可以通过编解码的方式判断图片是否合法。
1、判断标示方法
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
|
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、核心方法
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
|
/**
* 获取文件的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、测试代码
正常测试
|
1
2
3
4
5
6
7
|
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删除扩展名
|
1
2
3
4
5
6
7
8
9
10
11
12
13
|
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
以上就是本文的全部内容，希望对大家的学习有所帮助，也希望大家多多支持服务器之家。
原文链接：https://my.oschina.net/ososchina/blog/1610685