JspSmartUpload源码.rar

package com.jspsmart.upload; import java.io.*; import java.sql.ResultSet; import java.sql.SQLException; import java.util.Vector; import javax.servlet.*; import javax.servlet.http.*; import javax.servlet.jsp.JspWriter; import javax.servlet.jsp.PageContext; // Referenced classes of package com.jspsmart.upload: // Files, Request, SmartUploadException, File public class SmartUpload { protected byte m_binArray[]; protected HttpServletRequest m_request; protected HttpServletResponse m_response; protected ServletContext m_application; private int m_totalBytes; private int m_currentIndex; private int m_startData; private int m_endData; private String m_boundary; private long m_totalMaxFileSize; private long m_maxFileSize; private Vector m_deniedFilesList; private Vector m_allowedFilesList; private boolean m_denyPhysicalPath; private boolean m_forcePhysicalPath; private String m_contentDisposition; public static final int SAVE_AUTO = 0; public static final int SAVE_VIRTUAL = 1; public static final int SAVE_PHYSICAL = 2; private Files m_files; private Request m_formRequest; public SmartUpload() { m_totalBytes = 0; m_currentIndex = 0; m_startData = 0; m_endData = 0; m_boundary = new String(); m_totalMaxFileSize = 0L; m_maxFileSize = 0L; m_deniedFilesList = new Vector(); m_allowedFilesList = new Vector(); m_denyPhysicalPath = false; m_forcePhysicalPath = false; m_contentDisposition = new String(); m_files = new Files(); m_formRequest = new Request(); } public final void init(ServletConfig config) throws ServletException { m_application = config.getServletContext(); } public void service(HttpServletRequest request, HttpServletResponse response) throws IOException, ServletException { m_request = request; m_response = response; } public final void initialize(ServletConfig config, HttpServletRequest request, HttpServletResponse response) throws ServletException { m_application = config.getServletContext(); m_request = request; m_response = response; } public final void initialize(PageContext pageContext) throws ServletException { m_application = pageContext.getServletContext(); m_request = (HttpServletRequest) pageContext.getRequest(); m_response = (HttpServletResponse) pageContext.getResponse(); } public final void initialize(ServletContext application, HttpSession session, HttpServletRequest request, HttpServletResponse response, JspWriter out) throws ServletException { m_application = application; m_request = request; m_response = response; } public void upload() throws SmartUploadException, IOException, ServletException { int totalRead = 0; int readBytes = 0; long totalFileSize = 0L; boolean found = false; String dataHeader = new String(); String fieldName = new String(); String fileName = new String(); String fileExt = new String(); String filePathName = new String(); String contentType = new String(); String contentDisp = new String(); String typeMIME = new String(); String subTypeMIME = new String(); boolean isFile = false; m_totalBytes = m_request.getContentLength(); m_binArray = new byte[m_totalBytes]; for (; totalRead < m_totalBytes; totalRead += readBytes) try { m_request.getInputStream(); readBytes = m_request.getInputStream().read(m_binArray, totalRead, m_totalBytes - totalRead); } catch (Exception e) { throw new SmartUploadException("Unable to upload."); } for (; !found && m_currentIndex < m_totalBytes; m_currentIndex++) if (m_binArray[m_currentIndex] == 13) found = true; else m_boundary = m_boundary + (char) m_binArray[m_currentIndex]; if (m_currentIndex == 1) return; m_currentIndex++; do { if (m_currentIndex >= m_totalBytes) break; dataHeader = getDataHeader(); m_currentIndex = m_currentIndex + 2; isFile = dataHeader.indexOf("filename") > 0; fieldName = getDataFieldValue(dataHeader, "name"); if (isFile) { filePathName = getDataFieldValue(dataHeader, "filename"); fileName = getFileName(filePathName); fileExt = getFileExt(fileName); contentType = getContentType(dataHeader); contentDisp = getContentDisp(dataHeader); typeMIME = getTypeMIME(contentType); subTypeMIME = getSubTypeMIME(contentType); } getDataSection(); if (isFile && fileName.length() > 0) { if (m_deniedFilesList.contains(fileExt)) throw new SecurityException( "The extension of the file is denied to be uploaded (1015)."); if (!m_allowedFilesList.isEmpty() && !m_allowedFilesList.contains(fileExt)) throw new SecurityException( "The extension of the file is not allowed to be uploaded (1010)."); if (m_maxFileSize > (long) 0 && (long) ((m_endData - m_startData) + 1) > m_maxFileSize) throw new SecurityException(String .valueOf((new StringBuffer( "Size exceeded for this file : ")).append( fileName).append(" (1105)."))); totalFileSize += (m_endData - m_startData) + 1; if (m_totalMaxFileSize > (long) 0 && totalFileSize > m_totalMaxFileSize) throw new SecurityException( "Total File Size exceeded (1110)."); } if (isFile) { com.jspsmart.upload.File newFile = new com.jspsmart.upload.File(); newFile.setParent(this); newFile.setFieldName(fieldName); newFile.setFileName(fileName); newFile.setFileExt(fileExt); newFile.setFilePathName(filePathName); newFile.setIsMissing(filePathName.length() == 0); newFile.setContentType(contentType); newFile.setContentDisp(contentDisp); newFile.setTypeMIME(typeMIME); newFile.setSubTypeMIME(subTypeMIME); if (contentType.indexOf("application/x-macbinary") > 0) m_startData = m_startData + 128; newFile.setSize((m_endData - m_startData) + 1); newFile.setStartData(m_startData); newFile.setEndData(m_endData); m_files.addFile(newFile); } else { String value = new String(m_binArray, m_startData, (m_endData - m_startData) + 1, "UTF-8"); m_formRequest.putParameter(fieldName, value); } if ((char) m_binArray[m_currentIndex + 1] == '-') break; m_currentIndex = m_currentIndex + 2; } while (true); } public int save(String destPathName) throws SmartUploadException, IOException, ServletException { return save(destPathName, 0); } public int save(String destPathName, int option) throws SmartUploadException, IOException, ServletException { int count = 0; if (destPathName == null) destPathName = m_application.getRealPath("/"); if (destPathName.indexOf("/") != -1) { if (destPathName.charAt(destPathName.length() - 1) != '/') destPathName = String.valueOf(destPathName).concat("/"); } else if (destPathName.charAt(destPathName.length() - 1) != '\\') destPathName = String.valueOf(destPathName).concat("\\"); for (int i = 0; i < m_files.getCount(); i++) if (!m_files.getFile(i).isMissing()) { m_files .getFile(i) .saveAs( destPathName + m_files.getFile(i).getFileName(), option); count++; } return count; } public int getSize() { return m_totalBytes; } public byte getBinaryData(int index) { byte retval; try { retval = m_binArray[index]; } catch (Exception e) { throw new ArrayIndexOutOfBoundsException( "Index out of range (1005)."); } return retval; } public Files getFiles() { return m_files; } public Request getRequest() { return m_formRequest; } public void downloadFile(String sourceFilePathName) throws SmartUploadException, IOException, ServletException { downloadFile(sourceFilePathName, null, null); } public void downloadFil