View Javadoc
1   /*
2    * Licensed to the Apache Software Foundation (ASF) under one or more
3    * contributor license agreements. See the NOTICE file distributed with
4    * this work for additional information regarding copyright ownership.
5    * The ASF licenses this file to You under the Apache license, Version 2.0
6    * (the "License"); you may not use this file except in compliance with
7    * the License. You may obtain a copy of the License at
8    *
9    *      http://www.apache.org/licenses/LICENSE-2.0
10   *
11   * Unless required by applicable law or agreed to in writing, software
12   * distributed under the License is distributed on an "AS IS" BASIS,
13   * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14   * See the license for the specific language governing permissions and
15   * limitations under the license.
16   */
17  package org.apache.logging.log4j.core.util;
18  
19  import java.io.File;
20  import java.io.IOException;
21  import java.io.UnsupportedEncodingException;
22  import java.net.MalformedURLException;
23  import java.net.URI;
24  import java.net.URL;
25  import java.net.URLDecoder;
26  import java.nio.charset.StandardCharsets;
27  import java.util.Objects;
28  
29  import org.apache.logging.log4j.Logger;
30  import org.apache.logging.log4j.status.StatusLogger;
31  
32  /**
33   * File utilities.
34   */
35  public final class FileUtils {
36  
37      /** Constant for the file URL protocol. */
38      private static final String PROTOCOL_FILE = "file";
39  
40      private static final String JBOSS_FILE = "vfsfile";
41  
42      private static final Logger LOGGER = StatusLogger.getLogger();
43  
44      private FileUtils() {
45      }
46  
47      /**
48       * Tries to convert the specified URI to a file object. If this fails, <b>null</b> is returned.
49       *
50       * @param uri the URI
51       * @return the resulting file object
52       */
53      public static File fileFromUri(URI uri) {
54          // There MUST be a better way to do this. TODO Search other ASL projects...
55          if (uri == null
56                  || (uri.getScheme() != null && (!PROTOCOL_FILE.equals(uri.getScheme()) && !JBOSS_FILE.equals(uri
57                          .getScheme())))) {
58              return null;
59          }
60          if (uri.getScheme() == null) {
61              File file = new File(uri.toString());
62              if (file.exists()) {
63                  return file;
64              }
65              try {
66                  final String path = uri.getPath();
67                  file = new File(path);
68                  if (file.exists()) {
69                      return file;
70                  }
71                  uri = new File(path).toURI();
72              } catch (final Exception ex) {
73                  LOGGER.warn("Invalid URI {}", uri);
74                  return null;
75              }
76          }
77          final String charsetName = StandardCharsets.UTF_8.name();
78          try {
79              String fileName = uri.toURL().getFile();
80              if (new File(fileName).exists()) { // LOG4J2-466
81                  return new File(fileName); // allow files with '+' char in name
82              }
83              fileName = URLDecoder.decode(fileName, charsetName);
84              return new File(fileName);
85          } catch (final MalformedURLException ex) {
86              LOGGER.warn("Invalid URL {}", uri, ex);
87          } catch (final UnsupportedEncodingException uee) {
88              LOGGER.warn("Invalid encoding: {}", charsetName, uee);
89          }
90          return null;
91      }
92  
93      public static boolean isFile(final URL url) {
94          return url != null && (url.getProtocol().equals(PROTOCOL_FILE) || url.getProtocol().equals(JBOSS_FILE));
95      }
96  
97      public static String getFileExtension(final File file) {
98          final String fileName = file.getName();
99          if (fileName.lastIndexOf(".") != -1 && fileName.lastIndexOf(".") != 0) {
100             return fileName.substring(fileName.lastIndexOf(".") + 1);
101         }
102         return null;
103     }
104 
105     /**
106      * Asserts that the given directory exists and creates it if necessary.
107      * 
108      * @param dir the directory that shall exist
109      * @param createDirectoryIfNotExisting specifies if the directory shall be created if it does not exist.
110      * @throws java.io.IOException thrown if the directory could not be created.
111      */
112     public static void mkdir(final File dir, final boolean createDirectoryIfNotExisting) throws IOException {
113         // commons io FileUtils.forceMkdir would be useful here, we just want to omit this dependency
114         if (!dir.exists()) {
115             if (!createDirectoryIfNotExisting) {
116                 throw new IOException("The directory " + dir.getAbsolutePath() + " does not exist.");
117             }
118             if (!dir.mkdirs()) {
119                 throw new IOException("Could not create directory " + dir.getAbsolutePath());
120             }
121         }
122         if (!dir.isDirectory()) {
123             throw new IOException("File " + dir + " exists and is not a directory. Unable to create directory.");
124         }
125     }
126     
127     /**
128      * Creates the parent directories for the given File.
129      * 
130      * @param file
131      * @throws IOException
132      */
133     public static void makeParentDirs(final File file) throws IOException {
134         File parent = Objects.requireNonNull(file, "file").getCanonicalFile().getParentFile();
135         if (parent != null) {
136             mkdir(parent, true);
137         }
138     }
139 
140 }