数据库文件上传：安全防范恶意代码与存储优化实践

import java.io.IOException;
import java.io.InputStream;
import java.util.Arrays;

public class FileValidator {

    // PNG文件头魔术数字: 89 50 4E 47 0D 0A 1A 0A
    private static final byte[] PNG_HEADER = new byte[] {
        (byte) 0x89, (byte) 0x50, (byte) 0x4E, (byte) 0x47, (byte) 0x0D, (byte) 0x0A, (byte) 0x1A, (byte) 0x0A
    };

    // JPEG文件头魔术数字（常见）：FF D8 FF E0 / FF D8 FF E1 / FF D8 FF E2 / FF D8 FF E3
    // 这里我们只检查起始的FF D8 FF
    private static final byte[] JPEG_HEADER_START = new byte[] {
        (byte) 0xFF, (byte) 0xD8, (byte) 0xFF
    };

    /**
     * 验证文件是否为合法的图片类型（PNG或JPEG）。
     * @param inputStream 文件的输入流
     * @return 如果是合法图片返回true，否则返回false
     * @throws IOException 读取文件流时可能发生的IO异常
     */
    public boolean isValidImage(InputStream inputStream) throws IOException {
        if (inputStream == null) {
            return false;
        }

        byte[] headerBytes = new byte[8]; // 读取足够长的字节来覆盖PNG和JPEG的常见头部
        int bytesRead = inputStream.read(headerBytes);

        if (bytesRead < 3) { // 至少需要3个字节来判断JPEG
            return false;
        }

        // 检查是否为PNG
        if (bytesRead >= PNG_HEADER.length && Arrays.equals(Arrays.copyOfRange(headerBytes, 0, PNG_HEADER.length), PNG_HEADER)) {
            return true;
        }

        // 检查是否为JPEG (只检查FF D8 FF)
        if (bytesRead >= JPEG_HEADER_START.length &&
            headerBytes[0] == JPEG_HEADER_START[0] &&
            headerBytes[1] == JPEG_HEADER_START[1] &&
            headerBytes[2] == JPEG_HEADER_START[2]) {
            return true;
        }

        // 可以添加其他图片格式的验证，例如GIF, BMP等
        return false;
    }
}

import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.util.zip.GZIPInputStream;
import java.util.zip.GZIPOutputStream;

public class CompressionUtil {

    /**
     * 使用GZIP压缩字节数组。
     * @param data 原始字节数组
     * @return 压缩后的字节数组
     * @throws IOException 压缩过程中可能发生的IO异常
     */
    public byte[] compress(byte[] data) throws IOException {
        if (data == null || data.length == 0) {
            return new byte[0];
        }
        ByteArrayOutputStream bos = new ByteArrayOutputStream(data.length);
        GZIPOutputStream gzip = new GZIPOutputStream(bos);
        try {
            gzip.write(data);
        } finally {
            gzip.close(); // 确保关闭GZIPOutputStream以刷新所有数据
            bos.close();
        }
        return bos.toByteArray();
    }

    /**
     * 使用GZIP解压缩字节数组。
     * @param compressedData 压缩后的字节数组
     * @return 解压缩后的原始字节数组
     * @throws IOException 解压缩过程中可能发生的IO异常
     */
    public byte[] decompress(byte[] compressedData) throws IOException {
        if (compressedData == null || compressedData.length == 0) {
            return new byte[0];
        }
        ByteArrayOutputStream bos = new ByteArrayOutputStream();
        GZIPInputStream gzip = new GZIPInputStream(new java.io.ByteArrayInputStream(compressedData));
        try {
            byte[] buffer = new byte[1024];
            int len;
            while ((len = gzip.read(buffer)) > 0) {
                bos.write(buffer, 0, len);
            }
        } finally {
            gzip.close();
            bos.close();
        }
        return bos.toByteArray();
    }
}