001    /**
002     *  Licensed to the Apache Software Foundation (ASF) under one or more
003     *  contributor license agreements.  See the NOTICE file distributed with
004     *  this work for additional information regarding copyright ownership.
005     *  The ASF licenses this file to You under the Apache License, Version 2.0
006     *  (the "License"); you may not use this file except in compliance with
007     *  the License.  You may obtain a copy of the License at
008     *
009     *     http://www.apache.org/licenses/LICENSE-2.0
010     *
011     *  Unless required by applicable law or agreed to in writing, software
012     *  distributed under the License is distributed on an "AS IS" BASIS,
013     *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
014     *  See the License for the specific language governing permissions and
015     *  limitations under the License.
016     */
017    package org.apache.geronimo.kernel.classloader;
018    
019    import java.io.File;
020    import java.io.FileInputStream;
021    import java.io.IOException;
022    import java.net.MalformedURLException;
023    import java.util.jar.Manifest;
024    
025    /**
026     * @version $Rev: 757301 $ $Date: 2009-03-23 10:30:59 +0800 (Mon, 23 Mar 2009) $
027     */
028    public class DirectoryResourceLocation extends AbstractUrlResourceLocation {
029        private final File baseDir;
030        private boolean manifestLoaded = false;
031        private Manifest manifest;
032    
033        public DirectoryResourceLocation(File baseDir) throws MalformedURLException {
034            super(baseDir.toURL());
035            this.baseDir = baseDir;
036        }
037    
038        public ResourceHandle getResourceHandle(String resourceName) {
039            File file = new File(baseDir, resourceName);
040            if (!file.exists() || !isLocal(file)) {
041                return null;
042            }
043    
044            try {
045                ResourceHandle resourceHandle = new DirectoryResourceHandle(resourceName, file, baseDir, getManifestSafe());
046                return resourceHandle;
047            } catch (MalformedURLException e) {
048                return null;
049            }
050        }
051    
052        private boolean isLocal(File file) {
053            try {
054                String base = baseDir.getCanonicalPath();
055                String relative = file.getCanonicalPath();
056                return (relative.startsWith(base));
057            } catch (IOException e) {
058                return false;
059            }
060        }
061    
062        public Manifest getManifest() throws IOException {
063            if (!manifestLoaded) {
064                File manifestFile = new File(baseDir, "META-INF/MANIFEST.MF");
065    
066                if (manifestFile.isFile() && manifestFile.canRead()) {
067                    FileInputStream in = null;
068                    try {
069                        in = new FileInputStream(manifestFile);
070                        manifest = new Manifest(in);
071                    } finally {
072                        IoUtil.close(in);
073                    }
074                }
075                manifestLoaded = true;
076            }
077            return manifest;
078        }
079    
080        private Manifest getManifestSafe() {
081            Manifest manifest = null;
082            try {
083                manifest = getManifest();
084            } catch (IOException e) {
085                // ignore
086            }
087            return manifest;
088        }
089    }