import java.io.*;
import java.util.*;
import javax.servlet.*;
import javax.servlet.http.*;
public class Upload {
private String saveDir = "."; // 要保存文件的路径
private String contentType = ""; // 文档类型
private String charset = ""; // 字符集
private ArrayList tmpFileName = new
ArrayList(); // 临时存放文件名的数据结构
private Hashtable parameter = new Hashtable();
// 存放参数名和值的数据结构
private ServletContext context; //
程序上下文,用于初始化
private HttpServletRequest request; //
用于传入请求对象的实例
private String boundary = ""; // 内存数据的分隔符
private int len = 0; // 每次从内在中实际读到的字节长度
private String queryString;
private int count; // 上载的文件总数
private String[] fileName; // 上载的文件名数组
private long maxFileSize = 1024 * 1024 * 10;
// 最大文件上载字节;
private String tagFileName = "";
public final void init(HttpServletRequest
request) throws ServletException {
this.request = request;
boundary =
request.getContentType().substring(30); // 得到内存中数据分界符
queryString =
request.getQueryString();
}
public String getParameter(String s) { //
用于得到指定字段的参数值,重写request.getParameter(String
// s)
if (parameter.isEmpty())
{
return
null;
}
return (String)
parameter.get(s);
}
public String[] getParameterValues(String s) {
// 用于得到指定同名字段的参数数组,重写request.getParameterValues(String
// s)
ArrayList al = new
ArrayList();
if (parameter.isEmpty())
{
return
null;
}
Enumeration e =
parameter.keys();
while (e.hasMoreElements())
{
String key =
(String) e.nextElement();
if (-1 !=
key.indexOf(s + "||||||||||") || key.equals(s)) {
al.add(parameter.get(key));
}
}
if (al.size() == 0) {
return
null;
}
String[] value = new
String[al.size()];
for (int i = 0; i
< value.length; i++) {
value[i] =
(String) al.get(i);
}
return value;
}
public String getQueryString() {
return queryString;
}
public int getCount() {
return count;
}
public String[] getFileName() {
return fileName;
}
public void setMaxFileSize(long size) {
maxFileSize = size;
}
public void setTagFileName(String filename)
{
tagFileName = filename;
}
public void setSaveDir(String saveDir) { //
设置上载文件要保存的路径
this.saveDir = saveDir;
File testdir = new
File(saveDir); // 为了保证目录存在,如果没有则新建该目录
if (!testdir.exists()) {
testdir.mkdirs();
}
}
public void setCharset(String charset) { //
设置字符集
this.charset = charset;
}
public boolean uploadFile() throws
ServletException, IOException { // 用户调用的上载方法
setCharset(request.getCharacterEncoding());
return
uploadFile(request.getInputStream());
}
private boolean uploadFile(ServletInputStream
servletinputstream) throws // 取得央存数据的主方法
ServletException,
IOException {
String line = null;
byte[] buffer = new
byte[256];
while ((line = readLine(buffer,
servletinputstream, charset)) != null) {
if
(line.startsWith("Content-Disposition: form-data;")) {
int
i = line.indexOf("filename=");
if
(i >= 0) { //
如果一段分界符内的描述中有filename=,说明是文件的编码内容
String
fName = getFileName(line);
if
(fName.equals("")) {
continue;
}
if
(count == 0 && tagFileName.length()
!= 0) {
String
ext = fName
.substring((fName.lastIndexOf(".")
+ 1));
fName
= tagFileName + "." + ext;
}
tmpFileName.add(fName);
count++;
while
((line = readLine(buffer, servletinputstream, charset)) != null)
{
if
(line.length() <= 2) {
break;
}
}
File
f = new File(saveDir, fName);
FileOutputStream
dos = new FileOutputStream(f);
long
size = 0l;
while
((line = readLine(buffer, servletinputstream, null)) != null)
{
if
(line.indexOf(boundary) != -1) {
break;
}
size
+= len;
if
(size > maxFileSize) {
throw
new IOException("文件超过" + maxFileSize + "字节!");
}
dos.write(buffer,
0, len);
}
dos.close();
}
else { // 否则是字段编码的内容
String
key = getKey(line);
String
value = "";
while
((line = readLine(buffer, servletinputstream, charset)) != null)
{
if
(line.length() <= 2) {
break;
}
}
while
((line = readLine(buffer, servletinputstream, charset)) != null)
{
if
(line.indexOf(boundary) != -1) {
break;
}
value
+= line;
}
put(key,
value.trim(), parameter);
}
}
}
if (queryString != null)
{
String[] each
= split(queryString, "&");
for (int k =
0; k < each.length; k++) {
String[]
nv = split(each[k], "=");
if
(nv.length == 2) {
put(nv[0],
nv[1], parameter);
}
}
}
fileName = new
String[tmpFileName.size()];
for (int k = 0; k
< fileName.length; k++) {
fileName[k] =
(String) tmpFileName.get(k); // 把ArrayList中临时文件名倒入数据中供用户调用
}
if (fileName.length == 0)
{
return false;
// 如果fileName数据为空说明没有上载任何文件
}
return true;
}
private void put(String key, String value,
Hashtable ht) {
if (!ht.containsKey(key))
{
ht.put(key,
value);
} else { //
如果已经有了同名的KEY,就要把当前的key更名,同时要注意不能构成和KEY同名
try {
Thread.currentThread().sleep(1);
// 为了不在同一ms中产生两个相同的key
} catch
(Exception e) {
}
key +=
"||||||||||" + System.currentTimeMillis();
ht.put(key,
value);
}
}
private String readLine(byte[] Linebyte,
ServletInputStream
servletinputstream, String charset) {
try {
len =
servletinputstream.readLine(Linebyte, 0, Linebyte.length);
if (len ==
-1) {
return
null;
}
if (charset
== null) {
return
new String(Linebyte, 0, len);
} else
{
return
new String(Linebyte, 0, len, charset);
}
} catch (Exception _ex)
{
return
null;
}
}
private String getFileName(String line) { //
从描述字符串中分离出文件名
if (line == null) {
return
"";
}
int i =
line.indexOf("filename=");
line = line.substring(i +
9).trim();
i = line.lastIndexOf("");
if (i < 0 || i
>= line.length() - 1) {
i =
line.lastIndexOf("\\");
if
(line.equals("\"\"")) {
return
"";
}
if (i
< 0 || i >= line.length() - 1)
{
return
line;
}
}
return line.substring(i + 1,
line.length() - 1);
}
private String getKey(String line) {
//从描述字符串中分离出字段名
if (line == null) {
return
"";
}
int i =
line.indexOf("name=");
line = line.substring(i +
5).trim();
return line.substring(1,
line.length() - 1);
}
public static String[] split(String strOb,
String mark) {
if (strOb == null) {
return
null;
}
StringTokenizer st = new
StringTokenizer(strOb, mark);
ArrayList tmp = new
ArrayList();
while (st.hasMoreTokens())
{
tmp.add(st.nextToken());
}
String[] strArr = new
String[tmp.size()];
for (int i = 0; i
< tmp.size(); i++) {
strArr[i] =
(String) tmp.get(i);
}
return strArr;
}
}