Java获取文件的真实类型（通过文件魔数判断文件类型）

背景

通常情况下&＃xff0c;我们取文件的后缀名作为文件的类型即可&＃xff0c;但有时为了防止用户随便修改后缀名&＃xff0c;就需要获取文件的真实类型。这时候可以通过文件的魔数来判别。魔数就是文件开头的那几个字节的数据。

代码实现

工具类代码

import java.util.HashMap;
import java.util.Map;/*** &＃64;author lixk* &＃64;description 魔数工具类*/
public class MagicNumberUtils {/*** 魔数到文件类型的映射集合*/public static final Map<String, String> TYPES &＃61; new HashMap<>();static {// 图片&＃xff0c;此处只提取前六位作为魔数TYPES.put("FFD8FF", "jpg");TYPES.put("89504E", "png");TYPES.put("474946", "gif");TYPES.put("524946", "webp");}/*** 根据文件的字节数据获取文件类型** &＃64;param data 文件字节数组数据* &＃64;return*/public static String getFileType(byte[] data) {//提取前六位作为魔数String magicNumberHex &＃61; getHex(data, 6);return TYPES.get(magicNumberHex);}/*** 获取16进制表示的魔数** &＃64;param data 字节数组形式的文件数据* &＃64;param magicNumberLength 魔数长度* &＃64;return*/public static String getHex(byte[] data, int magicNumberLength) {//提取文件的魔数StringBuilder magicNumber &＃61; new StringBuilder();//一个字节对应魔数的两位int magicNumberByteLength &＃61; magicNumberLength / 2;for (int i &＃61; 0; i < magicNumberByteLength; i&＃43;&＃43;) {magicNumber.append(Integer.toHexString(data[i] >> 4 & 0xF));magicNumber.append(Integer.toHexString(data[i] & 0xF));}return magicNumber.toString().toUpperCase();}}


单元测试类

import com.kalikali.userbiz.util.MagicNumberUtils;
import org.junit.Test;
import org.springframework.boot.test.context.SpringBootTest;import java.io.FileInputStream;
import java.io.IOException;
import java.io.InputStream;/*** &＃64;author lixk* &＃64;description 文件魔数测试*/
&＃64;SpringBootTest
public class MagicNumberUtilsTests {/*** 读取文件字节数据** &＃64;param filePath* &＃64;return* &＃64;throws IOException*/public static byte[] getFileBytesData(String filePath) throws IOException {InputStream fs &＃61; new FileInputStream(filePath);byte[] b &＃61; new byte[fs.available()];fs.read(b);return b;}/*** 根据文件魔数获取文件类型*/&＃64;Testpublic void getFileType() throws Exception {System.out.println(MagicNumberUtils.getFileType(getFileBytesData("src/test/resources/images/1.jpg")));System.out.println(MagicNumberUtils.getFileType(getFileBytesData("src/test/resources/images/2.jpg")));System.out.println(MagicNumberUtils.getFileType(getFileBytesData("src/test/resources/images/1.png")));System.out.println(MagicNumberUtils.getFileType(getFileBytesData("src/test/resources/images/1.gif")));System.out.println(MagicNumberUtils.getFileType(getFileBytesData("src/test/resources/images/3.gif")));System.out.println(MagicNumberUtils.getFileType(getFileBytesData("src/test/resources/images/1.webp")));}
}

运行结果

如果需要判断其他类型文件的真实类型&＃xff0c;可以自行添加完善 魔数-类型 map.