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

背景

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

代码实现

工具类代码

import java.util.HashMap;
import java.util.Map;

/**
 * @author lixk
 * @description 魔数工具类
 */
public class MagicNumberUtils {
    /**
     * 魔数到文件类型的映射集合
     */
    public static final Map<String, String> TYPES = new HashMap<>();

    static {
        // 图片，此处只提取前六位作为魔数
        TYPES.put("FFD8FF", "jpg");
        TYPES.put("89504E", "png");
        TYPES.put("474946", "gif");
        TYPES.put("524946", "webp");
    }

    /**
     * 根据文件的字节数据获取文件类型
     *
     * @param data 文件字节数组数据
     * @return
     */
    public static String getFileType(byte[] data) {
        //提取前六位作为魔数
        String magicNumberHex = getHex(data, 6);
        return TYPES.get(magicNumberHex);
    }

    /**
     * 获取16进制表示的魔数
     *
     * @param data              字节数组形式的文件数据
     * @param magicNumberLength 魔数长度
     * @return
     */
    public static String getHex(byte[] data, int magicNumberLength) {
        //提取文件的魔数
        StringBuilder magicNumber = new StringBuilder();
        //一个字节对应魔数的两位
        int magicNumberByteLength = magicNumberLength / 2;
        for (int i = 0; i < magicNumberByteLength; i++) {
            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;

/**
 * @author lixk
 * @description 文件魔数测试
 */
@SpringBootTest
public class MagicNumberUtilsTests {

    /**
     * 读取文件字节数据
     *
     * @param filePath
     * @return
     * @throws IOException
     */
    public static byte[] getFileBytesData(String filePath) throws IOException {
        InputStream fs = new FileInputStream(filePath);
        byte[] b = new byte[fs.available()];
        fs.read(b);
        return b;
    }

    /**
     * 根据文件魔数获取文件类型
     */
    @Test
    public 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")));
    }
}

运行结果

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