您的位置:首页 > 编程语言 > Java开发

Java获取文件的类型

2014-01-24 08:53 295 查看
在web应用中经常需要进行上传的操作,从安全的角度考虑,这个功能一般是需要限制文件的类型的!

通过扩展名获取文件类型是一种很不靠谱的方式。一般使用lastIndexOf和substring来获取。

下面介绍一种较文件扩展名来说可信度更大的方式——文件头信息内容判断!

package com.wzwahl36.up.file;

import java.io.File;
import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.IOException;
import java.io.InputStream;
import java.util.HashMap;
import java.util.Iterator;
import java.util.Map;
import java.util.Map.Entry;

/**
* 文件操作的一个类,包括文件的类型获取,文件、文件夹操作等
*
* @author Xewee.Zhiwei.Wang
* @version 2011-9-18 下午12:34:26
*/
public class FileUtils {

public final static Map<String, String> FILE_TYPE_MAP = new HashMap<String, String>();
static {
getAllFileType(); // 初始化文件类型信息
}

/**
* 将常见文件类型放入到map中
*
* @author Xewee.Zhiwei.Wang
* @version 2011-9-18 下午12:35:22
*/
private static void getAllFileType() {
FILE_TYPE_MAP.put("jpg", "FFD8FF"); // JPEG (jpg)
FILE_TYPE_MAP.put("png", "89504E47"); // PNG (png)
FILE_TYPE_MAP.put("gif", "47494638"); // GIF (gif)
FILE_TYPE_MAP.put("tif", "49492A00"); // TIFF (tif)
FILE_TYPE_MAP.put("bmp", "424D"); // Windows Bitmap (bmp)
FILE_TYPE_MAP.put("dwg", "41433130"); // CAD (dwg)
FILE_TYPE_MAP.put("html", "68746D6C3E"); // HTML (html)
FILE_TYPE_MAP.put("rtf", "7B5C727466"); // Rich Text Format (rtf)
FILE_TYPE_MAP.put("xml", "3C3F786D6C");
FILE_TYPE_MAP.put("zip", "504B0304");
FILE_TYPE_MAP.put("rar", "52617221");
FILE_TYPE_MAP.put("psd", "38425053"); // Photoshop (psd)
FILE_TYPE_MAP.put("eml", "44656C69766572792D646174653A"); // Email
FILE_TYPE_MAP.put("dbx", "CFAD12FEC5FD746F"); // Outlook Express (dbx)
FILE_TYPE_MAP.put("pst", "2142444E"); // Outlook (pst)
FILE_TYPE_MAP.put("xls", "D0CF11E0"); // MS Word
FILE_TYPE_MAP.put("doc", "D0CF11E0"); // MS Excel 注意:word 和 excel的文件头一样
FILE_TYPE_MAP.put("mdb", "5374616E64617264204A"); // MS Access (mdb)
FILE_TYPE_MAP.put("wpd", "FF575043"); // WordPerfect (wpd)
FILE_TYPE_MAP.put("eps", "252150532D41646F6265");
FILE_TYPE_MAP.put("ps", "252150532D41646F6265");
FILE_TYPE_MAP.put("pdf", "255044462D312E"); // Adobe Acrobat (pdf)
FILE_TYPE_MAP.put("qdf", "AC9EBD8F"); // Quicken (qdf)
FILE_TYPE_MAP.put("pwl", "E3828596"); // Windows Password (pwl)
FILE_TYPE_MAP.put("wav", "57415645"); // Wave (wav)
FILE_TYPE_MAP.put("avi", "41564920");
FILE_TYPE_MAP.put("ram", "2E7261FD"); // Real Audio (ram)
FILE_TYPE_MAP.put("rm", "2E524D46"); // Real Media (rm)
FILE_TYPE_MAP.put("mpg", "000001BA"); //
FILE_TYPE_MAP.put("mov", "6D6F6F76"); // Quicktime (mov)
FILE_TYPE_MAP.put("asf", "3026B2758E66CF11"); // Windows Media (asf)
FILE_TYPE_MAP.put("mid", "4D546864"); // MIDI (mid)
}

//  /**
//   * 获取图片文件实际类型,若不是图片则返回null]
//   *
//   * @author Xewee.Zhiwei.Wang
//   * @version 2011-9-18 下午12:35:59
//   * @param f
//   * @return
//   */
//  public final static String getImageFileType(File f) {
//      if (isImage(f)) {
//          try {
//              ImageInputStream iis = ImageIO.createImageInputStream(f);
//              Iterator<ImageReader> iter = ImageIO.getImageReaders(iis);
//              if (!iter.hasNext()) {
//                  return null;
//              }
//              ImageReader reader = iter.next();
//              iis.close();
//              return reader.getFormatName();
//          } catch (IOException e) {
//              return null;
//          } catch (Exception e) {
//              return null;
//          }
//      }
//      return null;
//  }

/**
* 获取文件类型,包括图片,若格式不是已配置的,则返回null
*
* @author Xewee.Zhiwei.Wang
* @version 2011-9-18 下午12:36:32
* @param file
* @return
*/
public static String getFileTypeByFile(File file) {
String filetype = null;
byte[] b = new byte[50];
try {
InputStream is = new FileInputStream(file);
is.read(b);
filetype = getFileTypeByStream(b);
is.close();
} catch (FileNotFoundException e) {
e.printStackTrace();
} catch (IOException e) {
e.printStackTrace();
}
return filetype;
}

/**
* 通过字节流获得文件类型
*
* @author Xewee.Zhiwei.Wang
* @version 2011-9-18 下午12:37:03
* @param b
* @return
*/
public static String getFileTypeByStream(byte[] b) {
String filetypeHex = String.valueOf(getFileHexString(b));
Iterator<Entry<String, String>> entryiterator = FILE_TYPE_MAP
.entrySet().iterator();
while (entryiterator.hasNext()) {
Entry<String, String> entry = entryiterator.next();
String fileTypeHexValue = entry.getValue();
if (filetypeHex.toUpperCase().startsWith(fileTypeHexValue)) {
return entry.getKey();
}
}
return null;
}

//  /**
//   * 判断文件是不是图片
//   *
//   * @author Xewee.Zhiwei.Wang
//   * @version 2011-9-18 下午12:37:54
//   * @param file
//   * @return
//   */
//  public static final boolean isImage(File file) {
//      boolean flag = false;
//      try {
//          BufferedImage bufreader = ImageIO.read(file);
//          int width = bufreader.getWidth();
//          int height = bufreader.getHeight();
//          if (width == 0 || height == 0) {
//              flag = false;
//          }
//          else {
//              flag = true;
//          }
//      } catch (IOException e) {
//          flag = false;
//      } catch (Exception e) {
//          flag = false;
//      }
//      return flag;
//  }

/**
* 获得文件的16进制数据
*
* @author Xewee.Zhiwei.Wang
* @version 2011-9-18 下午12:38:26
* @param b
* @return
*/
public static String getFileHexString(byte[] b) {
StringBuilder stringBuilder = new StringBuilder();
if (b == null || b.length <= 0) {
return null;
}
for (int i = 0; i < b.length; i++) {
int v = b[i] & 0xFF;
String hv = Integer.toHexString(v);
if (hv.length() < 2) {
stringBuilder.append(0);
}
stringBuilder.append(hv);
}
return stringBuilder.toString();
}

/**
* 删除path(该path可能是文件也可能是文件夹)
*
* @author Xewee.Zhiwei.Wang
* @version 2011-9-18 下午12:49:15
* @param file 要删除的文件或者目录
* @param includeSubFile 如果被删出的是目录,是否循环删除该目录中的子目录
* @return 操作是否成功
*/
public final boolean deleteFile(File file, boolean includeSubFile) {
if (! file.exists()) {
return false;
}
if (file.isFile()) {
System.out.println("del file:" + file.toString());
return file.delete();
}
else if (file.isDirectory()) {

File[] fileList = file.listFiles();
if (includeSubFile) {
for (int i = 0; i < fileList.length; i++) {
deleteFile(fileList[i], includeSubFile);
}
}
else {
for (int i = 0; i < fileList.length; i++) {
if (fileList[i].isFile()) {
deleteFile(fileList[i], includeSubFile);
}
}
}
}
return true;
}

/**
* 删除path(该path可能是文件也可能是文件夹)
*
* @author Xewee.Zhiwei.Wang
* @version 2011-9-18 下午12:49:15
* @param path 要删除的文件或者目录
* @param includeSubFile 如果被删出的是目录,是否循环删除该目录中的子目录
* @return 操作是否成功
*/
public final boolean deleteFile(String path, boolean includeSubFile) {
return deleteFile(new File(path), includeSubFile);
}

/**
* 赋值文件scrFile到目地目录destFolder
* @author Xewee.Zhiwei.Wang
* @version 2011-9-18 下午12:48:34
*
* @param scrFile 源文件,可能是文件,可能是目录
* @param destFolder 目地文件,只能是目录
* @param includeSubFile 如果源文件是目录,是否循环复制源文件的子目录
* @return 操作是否成功
* @return
*/
public final boolean copyFile(File scrFile, File destFolder, boolean includeSubFile) {
//未实现
return true;
}

/**
* 赋值文件scrFile到目地目录destFolder
* @author Xewee.Zhiwei.Wang
* @version 2011-9-18 下午12:48:34
*
* @param scrFile 源文件,可能是文件,可能是目录
* @param destFolder 目地文件,只能是目录
* @param includeSubFile 如果源文件是目录,是否循环复制源文件的子目录
* @return 操作是否成功
* @return
*/
public final boolean copyFile(String scrFile, String destFolder, boolean includeSubFile) {
//未实现
return copyFile(new File(scrFile), new File(destFolder), includeSubFile);
}

/**
* 创建文件或者目录
* @author Xewee.Zhiwei.Wang
* @version 2011-9-18 下午12:48:15
* @param file
* @return
*/
public final boolean createFile(File file) {
try {
if (file.isDirectory()) {
return file.mkdirs();
}
else {
return file.createNewFile();
}
} catch (IOException e) {
return false;
}
}
/**
* 创建文件或者目录
*
* @author Xewee.Zhiwei.Wang
* @version 2011-9-18 下午12:48:05
* @param file
* @return
*/
public final boolean createFile(String file) {
return createFile(new File(file));
}
}


转自:http://50vip.com/blog.php?i=32
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: