1 /*
   2  * Copyright (c) 2012, 2016, Oracle and/or its affiliates. All rights reserved.
   3  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
   4  *
   5  * This code is free software; you can redistribute it and/or modify it
   6  * under the terms of the GNU General Public License version 2 only, as
   7  * published by the Free Software Foundation.  Oracle designates this
   8  * particular file as subject to the "Classpath" exception as provided
   9  * by Oracle in the LICENSE file that accompanied this code.
  10  *
  11  * This code is distributed in the hope that it will be useful, but WITHOUT
  12  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
  13  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
  14  * version 2 for more details (a copy is included in the LICENSE file that
  15  * accompanied this code).
  16  *
  17  * You should have received a copy of the GNU General Public License version
  18  * 2 along with this work; if not, write to the Free Software Foundation,
  19  * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
  20  *
  21  * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
  22  * or visit www.oracle.com if you need additional information or have any
  23  * questions.
  24  */
  25 
  26 package com.oracle.tools.packager;
  27 
  28 import com.oracle.tools.packager.Log;
  29 import java.io.*;
  30 import java.net.URL;
  31 import java.nio.channels.FileChannel;
  32 import java.util.Arrays;
  33 




  34 public class IOUtils {
  35 
  36     public static boolean deleteRecursive(File path) throws FileNotFoundException {
  37         boolean ret = true;
  38         if (!path.exists()) { //nothing to do
  39             return true;
  40         }
  41         if (path.isDirectory()) {
  42             File[] children = path.listFiles();
  43             if (children != null) {
  44                 for (File f : children) {
  45                     ret = ret && deleteRecursive(f);
  46                 }
  47             }
  48         }
  49         return ret && path.delete();
  50     }
  51 
  52     public static void copyFromURL(URL location, File file) throws IOException {
  53         copyFromURL(location, file, false);
  54     }
  55 
  56     public static void copyFromURL(URL location, File file, boolean append) throws IOException {
  57         if (location == null) {
  58             throw new IOException("Missing input resource!");
  59         }
  60         if (file.exists() && !append) {
  61            file.delete();
  62         }
  63         InputStream in = location.openStream();
  64         FileOutputStream out = new FileOutputStream(file, append);
  65         byte[] buffer = new byte[1024];
  66         int len;
  67         while ((len = in.read(buffer)) != -1) {
  68             out.write(buffer, 0, len);
  69         }
  70         out.close();
  71         in.close();
  72         file.setReadOnly();
  73         file.setReadable(true, false);
  74     }
  75 
  76     public static void copyFile(File sourceFile, File destFile)
  77             throws IOException {
  78         destFile.getParentFile().mkdirs();
  79 
  80         //recreate the file as existing copy may have weird permissions
  81         destFile.delete();
  82         destFile.createNewFile();
  83 
  84         FileChannel source = null;
  85         FileChannel destination = null;
  86         source = new FileInputStream(sourceFile).getChannel();
  87         destination = new FileOutputStream(destFile).getChannel();
  88         if (destination != null && source != null) {
  89             destination.transferFrom(source, 0, source.size());
  90         }
  91         if (source != null) {
  92             source.close();
  93         }
  94         if (destination != null) {
  95             destination.close();
  96         }
  97 
  98         //preserve executable bit!
  99         if (sourceFile.canExecute()) {
 100             destFile.setExecutable(true, false);
 101         }
 102         if (!sourceFile.canWrite()) {
 103             destFile.setReadOnly();
 104         }
 105         destFile.setReadable(true, false);
 106     }
 107 
 108     public static long getFolderSize(File folder) {
 109         long foldersize = 0;
 110 
 111         File[] children = folder.listFiles();
 112         if (children != null) {
 113             for (File f : children) {
 114                 if (f.isDirectory()) {
 115                     foldersize += getFolderSize(f);
 116                 } else {
 117                     foldersize += f.length();
 118                 }
 119             }
 120         }
 121 
 122         return foldersize;
 123     }
 124 
 125     //run "launcher paramfile" in the directory where paramfile is kept
 126     public static void run(String launcher, File paramFile, boolean verbose)
 127             throws IOException {
 128         if (paramFile != null && paramFile.exists()) {
 129             ProcessBuilder pb = new ProcessBuilder(launcher, paramFile.getName());
 130             pb = pb.directory(paramFile.getParentFile());
 131             exec(pb, verbose);
 132         }
 133     }
 134 
 135     public static void exec(ProcessBuilder pb, boolean verbose) throws IOException {
 136         exec(pb, verbose, false);
 137     }
 138 
 139 
 140     public static void exec(ProcessBuilder pb, boolean verbose,
 141             boolean testForPresenseOnly) throws IOException {
 142         exec(pb, verbose, testForPresenseOnly, null);
 143     }
 144 
 145     public static void exec(ProcessBuilder pb, boolean verbose,
 146             boolean testForPresenseOnly, PrintStream consumer) throws IOException {
 147         pb.redirectErrorStream(true);
 148         Log.verbose("Running " + Arrays.toString(pb.command().toArray(new String[0]))
 149                                 + (pb.directory() != null ? (" in " + pb.directory()) : ""));
 150         Process p = pb.start();
 151         InputStreamReader isr = new InputStreamReader(p.getInputStream());
 152         BufferedReader br = new BufferedReader(isr);
 153         String lineRead;
 154         while ((lineRead = br.readLine()) != null) {
 155             if (consumer != null) {
 156                 consumer.print(lineRead + '\n');
 157             } else if (verbose) {
 158                Log.info(lineRead);
 159             } else {
 160                Log.debug(lineRead);
 161             }
 162         }
 163         try {
 164             int ret = p.waitFor();
 165             if (ret != 0 && !(testForPresenseOnly && ret != 127)) {
 166                 throw new IOException("Exec failed with code " + ret +
 167                         " command [" + Arrays.toString(pb.command().toArray(new String[0])) +
 168                         " in " + (pb.directory() != null ?
 169                            pb.directory().getAbsolutePath() : "unspecified directory"));
 170             }
 171         } catch (InterruptedException ex) {
 172         }
 173     }
 174 
 175     //no good test if we are running pre-JRE7
 176     //use heuristic approach
 177     // "false positive" is better than wrong answer
 178     public static boolean isNotSymbolicLink(File file) {
 179         //no symlinks on windows
 180         if (Platform.getPlatform() == Platform.WINDOWS) {
 181             return true;
 182         }
 183         try {
 184             if (file == null || file.getParent() == null) {
 185                 return false;
 186             }
 187             File file_canonical = new File(
 188                     file.getParentFile().getCanonicalFile(), file.getName());
 189             if (file_canonical.getCanonicalFile().equals(
 190                        file_canonical.getAbsoluteFile())) {
 191                 return true;
 192             }
 193         } catch (IOException ioe) {}
 194         return false;
 195     }
 196 
 197     public static byte[] readFully(File f) throws IOException {
 198         InputStream inp = new FileInputStream(f);
 199         //read fully into memory
 200         ByteArrayOutputStream baos = new ByteArrayOutputStream();
 201         byte[] buffer = new byte[1024];
 202         int length = 0;
 203         while ((length = inp.read(buffer)) != -1) {
 204             baos.write(buffer, 0, length);
 205         }
 206         baos.close();
 207         return baos.toByteArray();
 208     }
 209 }
--- EOF ---