Boa noite, senhores!
Desenvolvi um sistema com upload de arquivos que está funcionando normalmente utilizando a API do Jakarta Commons Upload File, o problema é que eu gostaria de restringir a extensão de arquivos para que o usuário suba apenas arquivos de extensão PDF.
Alguém saberia como eu poderia estar fazendo essa verificação?
Meu Servlet que faz o upload do arquivo:
/*
* To change this template, choose Tools | Templates
* and open the template in the editor.
*/
package controller;
/**
*
* @author Bruno Cesar
*/
import dao.UsuarioDao;
import java.io.File;
import java.io.IOException;
import java.io.PrintWriter;
import java.util.Date;
import java.util.List;
import javax.servlet.http.HttpSession;
import javax.servlet.ServletException;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import model.Upload;
import org.apache.tomcat.util.http.fileupload.FileItem;
import org.apache.tomcat.util.http.fileupload.FileItemFactory;
import org.apache.tomcat.util.http.fileupload.disk.DiskFileItemFactory;
import org.apache.tomcat.util.http.fileupload.servlet.ServletFileUpload;
public class Ctr2 extends HttpServlet {
@Override
protected void doPost(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
boolean isMultipart = ServletFileUpload.isMultipartContent(request);
Upload up = new Upload();
HttpSession session = request.getSession(true);
PrintWriter out = response.getWriter();
int retorno;
if (session.getAttribute("usuario_logado") == null) {
out.println("<script>");
out.println("alert('Você precisa estar logado para executar esta ação!')");
out.println("history.back()");
out.println("</script>");
} else {
if (isMultipart) {
try {
FileItemFactory factory = new DiskFileItemFactory();
ServletFileUpload upload = new ServletFileUpload(factory);
List<FileItem> items = (List<FileItem>) upload.parseRequest(request);
for (FileItem item : items) {
if (item.isFormField()) {
//se for um campo normal de form
if ("nome_arquivo".equals(item.getFieldName())) {
up.setNome_arquivo(item.getString());
}
}
if (item.isFormField()) {
if ("tipo_arquivo".equals(item.getFieldName())) {
up.setTipo_arquivo(item.getString());
}
}
if (item.isFormField()) {
if ("area_atuacao".equals(item.getFieldName())) {
up.setArea_atuacao(item.getString());
}
} else {
UsuarioDao dao = new UsuarioDao();
int conectar = dao.conectar();
if (conectar != 1) {
request.setAttribute("erro", "Não conectou ao banco...");
request.getRequestDispatcher("erronobanco.jsp").forward(request, response);
}
//FIM DA CONEXÃO COM O BANCO
File uploadedFile = new File("C:/Users/Bruno Cesar/Documents/NetBeansProjects/eDOC/web/files/" + item.getName());
item.write(uploadedFile);
up.setArquivo(item.getName());
retorno = dao.arquivar(up);
if (retorno != 1) {
request.setAttribute("erro", "Erro ao incluir arquivo...");
request.getRequestDispatcher("erro.jsp").forward(request, response);
}
out.println("<script>");
out.println("alert('Arquivo enviado com sucesso!');");
out.println("document.location=('index.jsp');");
out.println("</script>");
}
}
} catch (Exception e) {
response.getWriter().println("ocorreu um problema ao fazer o upload: " + e.getMessage());
}
}
}
}
}