java的http协议文件上传 (二)

  

 MultipartRequest.java文件上半部内容如下:

 

import java.io.BufferedOutputStream;
import java.io.BufferedReader;
import java.io.ByteArrayOutputStream;
import java.io.File;
import java.io.FileOutputStream;
import java.io.FilterOutputStream;
import java.io.IOException;
import java.io.OutputStream;
import java.io.UnsupportedEncodingException;
import java.security.Principal;
import java.util.Enumeration;
import java.util.HashMap;
import java.util.Iterator;
import java.util.Locale;
import java.util.Map;
import java.util.Set;
import java.util.ArrayList;
import javax.servlet.RequestDispatcher;
import javax.servlet.ServletInputStream;
import javax.servlet.http.Cookie;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpSession;
public class MultipartRequest implements HttpServletRequest{
    private static final String TEMP_PATH = "c:/temp/";
    private long userId_;
    private HttpServletRequest req_;
    private File dir_ = new File(TEMP_PATH);
 private HashMap<String, ArrayList<String>> parameters_ = new HashMap<String, ArrayList<String>>();
    private HashMap<String, UploadedFile> files_ = new HashMap<String, UploadedFile>();
 private HashMap<File, String> exactFileNameTable_ = new HashMap<File, String>();
   
    public static boolean isMultipart(HttpServletRequest req) {
        String type = null;
        String type1 = req.getHeader("Content-Type");
        String type2 = req.getContentType();
        if (type1 == null && type2 != null) {
          type = type2;
        }
        else if (type2 == null && type1 != null) {
          type = type1;
        }
        else if (type1 != null && type2 != null) {
          type = (type1.length() > type2.length() ? type1 : type2);
        }
        if (type == null ||!type.toLowerCase().startsWith("multipart/form-data")) {
      return false ;
     }else{
      return true ;
     }
    }
   
 public MultipartRequest(HttpServletRequest req, long userId) throws IOException {
        if (req == null)
         throw new IllegalArgumentException("request cannot be null");
        if (!dir_.isDirectory())
         dir_.mkdir();
        if (!dir_.canWrite())
         throw new IllegalArgumentException("服务器上 " + TEMP_PATH + " 这个目录没有写的权限");
        req_ = req;
        userId_ = userId;
        readRequest();
    }
 
    protected void readRequest()
    throws IOException {
        int length = req_.getContentLength();
        String type = null;
        String type1 = req_.getHeader("Content-Type");
        String type2 = req_.getContentType();
        if (type1 == null && type2 != null) {
            type = type2;
        }
        else if (type2 == null && type1 != null) {
            type = type1;
        }
        else if (type1 != null && type2 != null) {
            type = (type1.length() > type2.length() ? type1 : type2);
        }
        if (type == null ||
            !type.toLowerCase().startsWith("multipart/form-data")) {
            throw new IOException("Posted content type isn't multipart/form-data");
        }
        String boundary = extractBoundary(type);
        if (boundary == null) {
            throw new IOException("Separation boundary was not specified");
        }
        MultipartInputStreamHandler in =
         new MultipartInputStreamHandler(req_.getInputStream(), length);
        String line = in.readLine();
        if (line == null) {
            throw new IOException("Corrupt form data: premature ending");
        }
        if (!line.startsWith(boundary)) {
            throw new IOException("Corrupt form data: no leading boundary");
        }
        boolean done = false;
        while (!done) {
            done = readNextPart(in, boundary);
        }
    }
  
    private String extractBoundary(String line) {
        int index = line.lastIndexOf("boundary=");
        if (index == -1) {
            return null;
        }
        String boundary = line.substring(index + 9);
        boundary = "--" + boundary;
        return boundary;
    }
  
    protected boolean readNextPart(MultipartInputStreamHandler in,
                    String boundary)
    throws IOException {
  String line = in.readLine();
  if (line == null) {
   return true;
  }
  else if (line.length() == 0) {
   return true;
  }
  String[] dispInfo = extractDispositionInfo(line);
  String strHttpParameterName = dispInfo[1];
  String strExactFileName = dispInfo[2];
  line = in.readLine();
  if (line == null) {
   return true;
  }
  String contentType = extractContentType(line);
  if (contentType != null) {
   line = in.readLine();
   if (line == null || line.length() > 0) {
    throw new IOException("Malformed line after content type: " + line);
   }
  }
  else {
   contentType = "application/octet-stream";
  }
  if (strExactFileName == null) {
   String value = readParameter(in, boundary);
   ArrayList<String> existingValues = (ArrayList<String>)parameters_.get(strHttpParameterName);
   if (existingValues == null) {
    existingValues = new ArrayList<String>();
    parameters_.put(strHttpParameterName, existingValues);
   }
   existingValues.add(value);
  }
  else {
   File temporaryFile;
   if (!strExactFileName.equals("unknown")) {
    String tempfilename = userId_ + strHttpParameterName
      + System.currentTimeMillis() + ".bin";
    temporaryFile = new File(dir_, tempfilename); 
   } else {
    temporaryFile = null;
   }
  
   readAndSaveFile(in, boundary, temporaryFile, contentType);
  
   if (!strExactFileName.equals("unknown")) {
    files_.put(strHttpParameterName,
       new UploadedFile(temporaryFile, contentType));
   }
   exactFileNameTable_.put(temporaryFile, strExactFileName);
  }
  return false;
    }
   
    protected void readAndSaveFile(MultipartInputStreamHandler in,
            String boundary,
            File temporaryFile,
            String contentType) throws IOException {
  OutputStream out = null;
 
  if (temporaryFile == null) {
  out = new ByteArrayOutputStream();
  } else {
  if (contentType.equals("application/x-macbinary")){
  out = new MacBinaryDecoderOutputStream(
  new BufferedOutputStream(
   new FileOutputStream(temporaryFile), 8 * 1024));
  }
  else {
  out = new BufferedOutputStream(
     new FileOutputStream(temporaryFile), 8 * 1024);
  }
  }
 
  byte[] bbuf = new byte[100 * 1024];
  int result;
  String line;
  boolean rnflag = false;
  while ((result = in.readLine(bbuf, 0, bbuf.length)) != -1) {
  if (result > 2 && bbuf[0] == '-' && bbuf[1] == '-') {
  line = new String(bbuf, 0, result, "GB2312");
  if (line.startsWith(boundary)) break;
  }
  if (rnflag) {
  out.write('/r'); out.write('/n');
  rnflag = false;
  }
  if (result >= 2 &&
  bbuf[result - 2] == '/r' &&
  bbuf[result - 1] == '/n') {
  out.write(bbuf, 0, result - 2);
  rnflag = true;
  }
  else {
  out.write(bbuf, 0, result);
  }
  }
  out.close();
 }
    private String[] extractDispositionInfo(String line) throws IOException {
        String[] retval = new String[3];
        String origline = line;
        line = origline.toLowerCase();
        int start = line.indexOf("content-disposition: ");
        int end = line.indexOf(";");
        if (start == -1 || end == -1) {
            throw new IOException("Content disposition corrupt: " + origline);
        }
        String disposition = line.substring(start + 21, end);
        if (!disposition.equals("form-data")) {
            throw new IOException("Invalid content disposition: " + disposition);
        }
        start = line.indexOf("name=/"", end);
        end = line.indexOf("/"", start + 7);
        if (start == -1 || end == -1) {
            throw new IOException("Content disposition corrupt: " + origline);
        }
        String name = origline.substring(start + 6, end);
        String filename = null;
        start = line.indexOf("filename=/"", end + 2);
        end = line.indexOf("/"", start + 10);     
        if (start != -1 && end != -1) {        
            filename = origline.substring(start + 10, end);
            int slash =
            Math.max(filename.lastIndexOf('/'), filename.lastIndexOf('//'));
            if (slash > -1) {
                filename = filename.substring(slash + 1);
            }
            if (filename.equals("")) filename = "unknown";
        }
        retval[0] = disposition;
        retval[1] = name;
        retval[2] = filename;
        return retval;
    }
   
    private String extractContentType(String line) throws IOException {
        String contentType = null;
        String origline = line;
        line = origline.toLowerCase();
        if (line.startsWith("content-type")) {
            int start = line.indexOf(" ");
            if (start == -1) {
                throw new IOException("Content type corrupt: " + origline);
            }
            contentType = line.substring(start + 1);
        }
        else if (line.length() != 0) {
            throw new IOException("Malformed line after disposition: " + origline);
        }
        return contentType;
    }
 
 protected String readParameter(MultipartInputStreamHandler in,
            String boundary) throws IOException {
  StringBuffer sbuf = new StringBuffer();
  String line;
  while ((line = in.readLine()) != null) {
  if (line.startsWith(boundary)) break;
  sbuf.append(line + "/r/n");
  }
  if (sbuf.length() == 0) {
  return "";
  }
  sbuf.setLength(sbuf.length() - 2);
  return sbuf.toString();
    }
   
    public File getFile(String strParameterName) {
  UploadedFile file = (UploadedFile)files_.get(strParameterName);
  if (file != null) {
   return file.getFile();
  } else {
   return null;
  }
 } 

<!--（规则）（积分）//-->