1 /*
2  * Copyright (C) 2014 The Android Open Source Project
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  *      http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16 
17 package com.android.multidex;
18 
19 import com.android.dx.cf.direct.DirectClassFile;
20 import com.android.dx.cf.direct.StdAttributeFactory;
21 
22 import java.io.ByteArrayOutputStream;
23 import java.io.File;
24 import java.io.FileNotFoundException;
25 import java.io.IOException;
26 import java.io.InputStream;
27 import java.util.ArrayList;
28 import java.util.List;
29 import java.util.regex.Pattern;
30 import java.util.zip.ZipException;
31 import java.util.zip.ZipFile;
32 
33 class Path {
34 
getClassPathElement(File file)35     static ClassPathElement getClassPathElement(File file)
36             throws ZipException, IOException {
37         if (file.isDirectory()) {
38             return new FolderPathElement(file);
39         } else if (file.isFile()) {
40             return new ArchivePathElement(new ZipFile(file));
41         } else if (file.exists()) {
42             throw new IOException("\"" + file.getPath() +
43                     "\" is not a directory neither a zip file");
44         } else {
45             throw new FileNotFoundException("File \"" + file.getPath() + "\" not found");
46         }
47     }
48 
49     List<ClassPathElement> elements = new ArrayList<ClassPathElement>();
50     private final String definition;
51     private final ByteArrayOutputStream baos = new ByteArrayOutputStream(40 * 1024);
52     private final byte[] readBuffer = new byte[20 * 1024];
53 
Path(String definition)54     Path(String definition) throws IOException {
55         this.definition = definition;
56         for (String filePath : definition.split(Pattern.quote(File.pathSeparator))) {
57             try {
58                 addElement(getClassPathElement(new File(filePath)));
59             } catch (IOException e) {
60                 throw new IOException("Wrong classpath: " + e.getMessage(), e);
61             }
62         }
63     }
64 
readStream(InputStream in, ByteArrayOutputStream baos, byte[] readBuffer)65     private static byte[] readStream(InputStream in, ByteArrayOutputStream baos, byte[] readBuffer)
66             throws IOException {
67         try {
68             for (;;) {
69                 int amt = in.read(readBuffer);
70                 if (amt < 0) {
71                     break;
72                 }
73 
74                 baos.write(readBuffer, 0, amt);
75             }
76         } finally {
77             in.close();
78         }
79         return baos.toByteArray();
80     }
81 
82     @Override
toString()83     public String toString() {
84         return definition;
85     }
86 
getElements()87     Iterable<ClassPathElement> getElements() {
88       return elements;
89     }
90 
addElement(ClassPathElement element)91     private void addElement(ClassPathElement element) {
92         assert element != null;
93         elements.add(element);
94     }
95 
getClass(String path)96     synchronized DirectClassFile getClass(String path) throws FileNotFoundException {
97         DirectClassFile classFile = null;
98         for (ClassPathElement element : elements) {
99             try {
100                 InputStream in = element.open(path);
101                 try {
102                     byte[] bytes = readStream(in, baos, readBuffer);
103                     baos.reset();
104                     classFile = new DirectClassFile(bytes, path, false);
105                     classFile.setAttributeFactory(StdAttributeFactory.THE_ONE);
106                     break;
107                 } finally {
108                     in.close();
109                 }
110             } catch (IOException e) {
111                 // search next element
112             }
113         }
114         if (classFile == null) {
115             throw new FileNotFoundException("File \"" + path + "\" not found");
116         }
117         return classFile;
118     }
119 }