java File 校验文件类型

原创

mob64ca12d9b014 2023-12-25 07:06:10 ©著作权

©著作权归作者所有：来自51CTO博客作者mob64ca12d9b014的原创作品，请联系作者获取转载授权，否则将追究法律责任

Java File 校验文件类型

作为一名经验丰富的开发者，我将向你介绍如何使用 Java 来实现文件类型的校验。在本文中，我将为你展示整个流程，并提供每一步所需的代码以及注释。

流程图

graph TD
A[开始] --> B[获取文件扩展名]
B --> C[判断文件类型]
C --> D[校验文件类型]
D --> E[结束]

代码实现

1. 获取文件扩展名

首先，我们需要获取文件的扩展名。文件扩展名是文件名中最后一个点（.）之后的字符。我们可以使用 lastIndexOf() 方法来查找最后一个点的位置，并使用 substring() 方法获取扩展名。

String fileName = file.getName(); // 获取文件名
String fileExtension = fileName.substring(fileName.lastIndexOf(".") + 1); // 获取文件扩展名

2. 判断文件类型

接下来，我们需要判断文件的类型。在 Java 中，我们可以使用文件的魔数（Magic Number）来判断文件类型。每种文件类型都有独特的魔数，我们可以根据这些魔数来进行判断。

try (InputStream inputStream = new FileInputStream(file)) {
    byte[] magicNumber = new byte[4]; // 读取文件前4个字节作为魔数
    inputStream.read(magicNumber);

    // 判断文件类型
    if (Arrays.equals(magicNumber, new byte[]{0x50, 0x4B, 0x03, 0x04})) {
        // ZIP 文件
    } else if (Arrays.equals(magicNumber, new byte[]{0x25, 0x50, 0x44, 0x46})) {
        // PDF 文件
    } else if (Arrays.equals(magicNumber, new byte[]{0xFF, 0xD8, 0xFF, 0xE0})) {
        // JPEG 文件
    } else {
        // 其他文件类型
    }
} catch (IOException e) {
    e.printStackTrace();
}

上述代码中，我们通过读取文件的前4个字节，将其与不同文件类型的魔数进行比较。如果魔数匹配，则可以确定文件类型。

3. 校验文件类型

最后，我们还可以进行文件类型的校验。校验文件类型可以避免用户通过修改文件扩展名来欺骗系统。我们可以根据文件的扩展名或者魔数来进行校验。

boolean isFileTypeValid = false;

// 根据扩展名校验
String[] allowedExtensions = new String[]{"txt", "pdf", "jpg"};
for (String extension : allowedExtensions) {
    if (fileExtension.equalsIgnoreCase(extension)) {
        isFileTypeValid = true;
        break;
    }
}

// 或者根据魔数校验
try (InputStream inputStream = new FileInputStream(file)) {
    byte[] magicNumber = new byte[4];
    inputStream.read(magicNumber);

    if (Arrays.equals(magicNumber, new byte[]{0x50, 0x4B, 0x03, 0x04}) || 
        Arrays.equals(magicNumber, new byte[]{0x25, 0x50, 0x44, 0x46}) || 
        Arrays.equals(magicNumber, new byte[]{0xFF, 0xD8, 0xFF, 0xE0})) {
        isFileTypeValid = true;
    }
} catch (IOException e) {
    e.printStackTrace();
}

if (isFileTypeValid) {
    // 文件类型有效
} else {
    // 文件类型无效
}

在上述代码中，我们可以根据需求选择根据文件扩展名或者魔数进行校验。如果文件类型有效，则可以继续处理该文件；否则，需要进行相应的错误处理。