java校验图片类型

原创

mob64ca12d9081f 2024-07-09 06:39:26 ©著作权

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

实现Java校验图片类型

1. 整体流程

flowchart TD
    A(开始) --> B(读取文件头部信息);
    B --> C(根据文件头信息判断文件类型);
    C --> D{文件类型是否符合要求};
    D -- 是 --> E(输出校验成功信息);
    D -- 否 --> F(输出校验失败信息);
    F --> G(结束);
    E --> G;

2. 每一步具体操作

步骤1: 读取文件头部信息

// 读取文件头部信息
// 将文件头部信息读入byte数组
byte[] fileBytes = new byte[4];
try (InputStream is = new FileInputStream(filePath)) {
    is.read(fileBytes, 0, 4);
} catch (IOException e) {
    e.printStackTrace();
}

步骤2: 根据文件头信息判断文件类型

// 根据文件头信息判断文件类型
String fileSignature = bytesToHex(fileBytes);
String fileType = null;

// JPEG文件头信息
String jpgSignature = "FFD8FF";

if (fileSignature.startsWith(jpgSignature)) {
    fileType = "JPEG";
} else {
    fileType = "Unknown";
}

步骤3: 输出校验结果信息

// 输出校验结果信息
if ("JPEG".equals(fileType)) {
    System.out.println("文件类型为JPEG，校验通过！");
} else {
    System.out.println("文件类型不符合要求，校验失败！");
}

3. 完整代码示例

import java.io.FileInputStream;
import java.io.IOException;
import java.io.InputStream;

public class ImageValidator {

    public static void main(String[] args) {
        String filePath = "path/to/image.jpg";
        byte[] fileBytes = new byte[4];

        try (InputStream is = new FileInputStream(filePath)) {
            is.read(fileBytes, 0, 4);
        } catch (IOException e) {
            e.printStackTrace();
        }

        String fileSignature = bytesToHex(fileBytes);
        String fileType = null;

        String jpgSignature = "FFD8FF";

        if (fileSignature.startsWith(jpgSignature)) {
            fileType = "JPEG";
        } else {
            fileType = "Unknown";
        }

        if ("JPEG".equals(fileType)) {
            System.out.println("文件类型为JPEG，校验通过！");
        } else {
            System.out.println("文件类型不符合要求，校验失败！");
        }
    }

    private static String bytesToHex(byte[] bytes) {
        StringBuilder sb = new StringBuilder();
        for (byte b : bytes) {
            sb.append(String.format("%02X", b));
        }
        return sb.toString();
    }
}