java如何根据二进制流确定图片的类型

因为仅仅通过后缀名我们并不能得知用户是否把图片的类型更改为其他类型.

public enum ImageType {
	PNG('P','N','G','-',"PNG"),
	JPG('J','F','I','F',"JPG"),
	GIF('G','I','F','-',"GIF");
	private char begin, mid, end, last;
	private String type;
	private ImageType(char begin, char mid, char end, char last, String type){
		this.begin = begin;
		this.mid = mid;
		this.end = end;
		this.last = last;
		this.type = type;
	}
	
	/**
	 * 获取图片类型
	 * @param data
	 * @return
	 */
	public static ImageType getImageType(byte[] data) {
		// Png:
		if (PNG.getBegin() == data[1] && PNG.getMid() == data[2] && PNG.getEnd() == data[3]) {
			return PNG;
		}
		// Gif:
		if (GIF.getBegin() == data[0] && GIF.getMid() == data[1] && GIF.getEnd() == data[2]) {
			return GIF;
		}
		// JPG:
		if (JPG.getBegin() == data[6] && JPG.getMid() == data[7] && JPG.getEnd() == data[8]
		        && JPG.getLast() == data[9]) {
			return JPG;
		}
		return null;
	}
}

改进:因为文件流的前10个位置是保存的文件类型的信息根据这个规律:

public enum FileType2 {
	FILE;
	private static final Map<String, String> FILE_TYPES;
	
	static {
		FILE_TYPES = new HashMap<String, String>();
		FILE_TYPES.put("PNG", "PNG");
		FILE_TYPES.put("JFIF", "JPG");
		FILE_TYPES.put("GIF", "GIF");
		FILE_TYPES.put("PDF", "PDF");
	}
 
	/**
	 * 从集合中取得相应类型
	 * 
	 * @param target
	 * @return
	 */
	public static String getType(String target) {
		if (isEmpty(target)) {
			return null;
		}
		String result = null;
		for (String key : FILE_TYPES.keySet()) {
			if (!contains(target, key)) continue;
			result = FILE_TYPES.get(key);
			break;
		}
		return result;
	}
 
	/**
	 * 是否包含
	 * 
	 * @param str
	 * @param searchStr
	 * @return
	 */
	private static boolean contains(String str, String searchStr) {
		if (str == null || searchStr == null) {
			return false;
		}
		return str.indexOf(searchStr) >= 0;
	}
 
	/**
	 * 是否为空
	 * 
	 * @param str
	 * @return
	 */
	private static boolean isEmpty(String str) {
		return str == null || str.length() == 0;
	}
}

测试:

FileInputStream fs = new FileInputStream("d:\\Koala.JPG");
		byte[] data = new byte[10];// 读10个长度就够了
		fs.read(data);
		StringBuilder str = new StringBuilder("");
		for (int i = 0; i < data.length; i++) {
			str.append((char) data[i]);
//			System.out.println(i + ":" + (char) data[i]);
		}
		System.out.println("------------------");
		System.out.println(FileType2.getType(str.toString()));
//		System.out.println("文件类型:"+ImageType.getImageType(data).getType());
		fs.close();
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值