Do not traverse unaccessible subdirectories (issue 51)

lemval

2012-01-31 1c30dad2115fc513791d8a5b292ad0f7d7b85749

 src/com/gitblit/utils/FileUtils.java

@@ -15,10 +15,16 @@
 */

package com.gitblit.utils;



import java.io.BufferedInputStream;

import java.io.BufferedReader;

import java.io.BufferedWriter;

import java.io.File;

import java.io.FileInputStream;

import java.io.FileNotFoundException;

import java.io.FileOutputStream;

import java.io.IOException;

import java.io.InputStreamReader;

import java.io.OutputStreamWriter;

import java.nio.charset.Charset;



/**

@@ -34,7 +40,7 @@
    * 

    * @param file

    * @param lineEnding

    * @return

    * @return the string content of the file

    */

   public static String readContent(File file, String lineEnding) {

      StringBuilder sb = new StringBuilder();

@@ -56,4 +62,91 @@
      }

      return sb.toString();

   }



   /**

    * Writes the string content to the file.

    * 
    * @param file

    * @param content

    */

   public static void writeContent(File file, String content) {

      try {

         OutputStreamWriter os = new OutputStreamWriter(new FileOutputStream(file),

               Charset.forName("UTF-8"));

         BufferedWriter writer = new BufferedWriter(os);

         writer.append(content);

         writer.close();

      } catch (Throwable t) {

         System.err.println("Failed to write content of " + file.getAbsolutePath());

         t.printStackTrace();

      }

   }



   /**

    * Recursively traverses a folder and its subfolders to calculate the total

    * size in bytes.

    * 
    * @param directory

    * @return folder size in bytes

    */

   public static long folderSize(File directory) {

      if (directory == null || !directory.exists()) {

         return -1;

      }

      if (directory.isFile()) {

         return directory.length();

      }

      long length = 0;

      for (File file : directory.listFiles()) {

         if (file.isFile()) {

            length += file.length();

         } else {

            length += folderSize(file);

         }

      }

      return length;

   }



   /**

    * Copies a file or folder (recursively) to a destination folder.

    * 
    * @param destinationFolder

    * @param filesOrFolders

    * @return

    * @throws FileNotFoundException

    * @throws IOException

    */

   public static void copy(File destinationFolder, File... filesOrFolders)

         throws FileNotFoundException, IOException {

      destinationFolder.mkdirs();

      for (File file : filesOrFolders) {

         if (file.isDirectory()) {

            copy(new File(destinationFolder, file.getName()), file.listFiles());

         } else {

            File dFile = new File(destinationFolder, file.getName());

            BufferedInputStream bufin = null;

            FileOutputStream fos = null;

            try {

               bufin = new BufferedInputStream(new FileInputStream(file));

               fos = new FileOutputStream(dFile);

               int len = 8196;

               byte[] buff = new byte[len];

               int n = 0;

               while ((n = bufin.read(buff, 0, len)) != -1) {

                  fos.write(buff, 0, n);

               }

            } finally {

               try {

                  bufin.close();

               } catch (Throwable t) {

               }

               try {

                  fos.close();

               } catch (Throwable t) {

               }

            }

            dFile.setLastModified(file.lastModified());

         }

      }

   }

}