Форум программистов [Powered by Invision Power Board]

Код

package org.umek.bts;

import java.io.*;
import javax.servlet.*;
import javax.servlet.http.*;
// import java.lang.*;
import java.util.*;
import java.sql.*;
// import java.net.URLDecoder;
//==============================================================================
class FileInfo{
        //----------------------------------------------------------------------
        public String filename;
        public int start_index;
        public int last_index;
        //----------------------------------------------------------------------
        public String ToTranslit(String in){
                String alpha = "абвгдеёжзиыйклмнопрстуфхцчшщьэюя ";
                String[] _alpha = { "a","b","v","g","d","e","yo","g","z","i","y","i",
                                    "k","l","m","n","o","p","r","s","t","u",
                                    "f","h","tz","ch","sh","sh","'","e","yu","ya","_"};
                in = in.toLowerCase();
                StringBuffer nname = new StringBuffer("");
                char[] chs = in.toCharArray();
                for(int i=0; i<chs.length;i++){
                    int k = alpha.indexOf(chs[i]);
                    if(k != -1){
                        nname.append(_alpha[k]);
                    }else{
                        nname.append(chs[i]);
                    }
                }
                return nname.toString();
        }
        //----------------------------------------------------------------------
        public FileInfo(String filename, int start_index, int last_index){
           this.filename = filename;
           this.start_index = start_index;
           this.last_index = last_index;
        }
        //----------------------------------------------------------------------
        public void SaveFile (byte[] data, String directory, String bugid){
           if (!filename.equals("NO_FILE")){
              File f = new File(directory + bugid + "_" + ToTranslit(filename));
              try{
                   FileOutputStream fos = new FileOutputStream(f);
                   int length = last_index - start_index;
                   fos.write(data, start_index, length);
                   fos.close();
              }catch(Exception ex){
                   System.out.println(ex.toString());
              }
           }
        }
        //----------------------------------------------------------------------
}
//==============================================================================
public class Upload extends HttpServlet {
        Connection conn     = null;
        Statement stm       = null;
        String SQL          = "";
        HttpSession session = null;
        // String patch = "D:\\proj1\\trunk\\web\\uploads\\";
        String patch = "C:\\apache-tomcat\\webapps\\bts\\uploads\\";
        //----------------------------------------------------------------------
        public String ToTranslit(String in){
                String alpha = new String("абвгдеёжзиыйклмнопрстуфхцчшщьэюя ");
                String[] _alpha = {"a","b","v","g","d","e","yo","g","z","i","y","i","k","l","m","n","o","p","r","s","t","u", "f","h","tz","ch","sh","sh","'","e","yu","ya","_"};
                in = in.toLowerCase();
                StringBuffer nname = new StringBuffer("");
                char[] chs = in.toCharArray();
                for(int i=0; i<chs.length;i++){
                    int k = alpha.indexOf(chs[i]);
                    if(k != -1){
                        nname.append(_alpha[k]);
                    }else{
                        nname.append(chs[i]);
                    }
                }
                return nname.toString();
        }
        //----------------------------------------------------------------------
        @Override
        public void doPost (HttpServletRequest request, HttpServletResponse response)throws ServletException, IOException{
             byte[] data;
             final int size = 1024*1024;
             session = request.getSession(true);
             PrintWriter out = response.getWriter();
             ServletInputStream in = request.getInputStream();
             BufferedInputStream bf = new BufferedInputStream((InputStream)in, size);
             byte[] all_data = new byte[size];
             int b,j = 0;
             while ((b = bf.read()) != -1){
                  all_data[j] = (byte)b;
                  j++;
             }
             data = new byte[j];
             for(int i=0; i<j; i++){
                     data[i] = all_data[i];
             }
             all_data = null;
             String boundary = extractBoundary(request.getHeader("Content-Type"));
             FileInfo[] file = extractFiles(data, boundary);             
             try{
                 conn = (Connection)session.getAttribute("connect");
                 stm  = conn.createStatement();
                 for (int i=0; i < file.length; i++){                                        
                     file[i].SaveFile(data, patch, request.getParameter("bugid"));
                     if (!file[i].filename.equals("NO_FILE")){
                        SQL ="insert"; // тут запись в бд о файле
                        stm.execute(SQL);
                        stm.execute("commit");
                     }
                 }
                 stm.close();
                 out.print("<meta http-equiv=\"Refresh\" content=\"0; URL=sb?bugid="+request.getParameter("bugid")+"\">");
             }catch(SQLException ex){
                 ex.getStackTrace();
             }
        }
        //----------------------------------------------------------------------
        private String getFilename(String header){
               String filename  = "";
               header.toLowerCase();
               int index;
               if ((index = header.indexOf("filename=")) != -1){
                   int up_index = header.indexOf((int)'"',index + 1 + 9);
                   filename = header.substring(index + 9 + 1, up_index);
                   index = filename.lastIndexOf((int)'/');
                   up_index = filename.lastIndexOf((int)'\\');
                   filename = filename.substring(Math.max(index, up_index) + 1);
               }else{
                    filename = "NO_FILE";
               }
               return filename;
        }
        //----------------------------------------------------------------------
        private void extractData(FileInfo fis, String data){
           char[] ch = {'\r','\n','\r','\n'};
           String new_line = new String(ch);
           String header;
           int index = data.indexOf(new_line,2);
           if (index != -1){
                header = data.substring(0,index);
                fis.filename = getFilename(header);
                fis.start_index += index + 4;
           }
        }
        //----------------------------------------------------------------------
        private FileInfo[] extractFiles(byte[] data, String boundary){
           int i = 0, index = 0, prev_index = 0;
           String data_str = new String(data);
           Vector data_vec = new Vector();
           while((index = data_str.indexOf(boundary,index)) != -1){
              if (i != 0){
               FileInfo f_info =
                new FileInfo("NO_FILE", prev_index, index - 2);
                extractData(f_info,
                data_str.substring(prev_index,index));
                data_vec.addElement(f_info);
              }
              index += boundary.length();
              prev_index = index;
              i++;
           }
           FileInfo[] Data = new FileInfo[i-1];
           Enumeration enume = data_vec.elements();
           i=0;
           while(enume.hasMoreElements()){
                   Data[i] = (FileInfo) enume.nextElement();
                   i++;
           }
           return Data;
        }
        //----------------------------------------------------------------------
        private String extractBoundary(String str){
           int index = str.lastIndexOf("boundary=");
           String boundary = str.substring(index + 9);
           boundary = "--" + boundary;
           return boundary;
        }
        //----------------------------------------------------------------------
}