ziver/hal
1
1
Fork 0
Code Pull requests Releases Activity

Bug fixes and added a SQLQuery class that is incomplete

Browse source
This commit is contained in:
Ziver Koc 2010-12-26 18:17:35 +00:00
parent f7f02123c2
commit d2bf44365c
4 changed files with 168 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

15
src/zutil/jee/upload/AjaxFileUpload.java
View file

@ -5,10 +5,12 @@ import java.io.IOException;
import java.io.PrintWriter;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.HashSet;
import java.util.Iterator;
import java.util.LinkedList;
import java.util.List;
import java.util.Map;
import java.util.logging.Level;
import java.util.logging.Logger;
import javax.servlet.ServletConfig;
@ -76,6 +78,7 @@ public abstract class AjaxFileUpload extends HttpServlet {
public static File TEMPFILE_PATH = null;
public static String JAVASCRIPT = "";
public static HashSet<String> ALLOWED_EXTENSIONS = new HashSet<String>();
public void init(ServletConfig config) throws ServletException {
super.init(config);
@ -95,6 +98,14 @@ public abstract class AjaxFileUpload extends HttpServlet {
else
TEMPFILE_PATH = new File( config.getInitParameter("TEMP_PATH") );
}
// Read allowed file types
if(config.getInitParameter("ALLOWED_EXTENSIONS") != null){
String[] tmp = config.getInitParameter("TEMP_PATH").split(",");
for( String ext : tmp ){
ALLOWED_EXTENSIONS.add(ext.trim());
}
}
} catch (IOException e) {
e.printStackTrace();
@ -179,6 +190,8 @@ public abstract class AjaxFileUpload extends HttpServlet {
FileItemIterator it = upload.getItemIterator( request );
while( it.hasNext() ) {
FileItemStream item = it.next();
if( !ALLOWED_EXTENSIONS.contains( FileUtil.fileExtension(item.getName()) ))
throw new Exception("Filetype "+FileUtil.fileExtension(item.getName())+" not allowed!");
listener.setFileName( item.getName() );
FileItem fileItem = factory.createItem(item.getFieldName(),
item.getContentType(), item.isFormField(), item.getName());
@ -204,7 +217,7 @@ public abstract class AjaxFileUpload extends HttpServlet {
// Done
listener.setStatus( Status.Done );
} catch (Exception e) {
e.printStackTrace();
logger.log(Level.WARNING, null, e);
listener.setStatus(Status.Error);
listener.setFileName("");
listener.setMessage( e.getMessage() );