/**
* @author 於豕
* Modify on 2007-11-12
*/
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(sourceFilePa