1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
|
/*******************************************************************************
* Copyright (c) 2009, 2010 VMware Inc.
*
* All rights reserved. This program and the accompanying materials
* are made available under the terms of the Eclipse Public License v1.0
* and Apache License v2.0 which accompanies this distribution.
* The Eclipse Public License is available at
* http://www.eclipse.org/legal/epl-v10.html
* and the Apache License v2.0 is available at
* http://www.opensource.org/licenses/apache2.0.php.
* You may elect to redistribute this code under either of these licenses.
*
* Contributors:
* VMware Inc. - initial contribution
*******************************************************************************/
package org.eclipse.gemini.web.internal.url;
import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.io.OutputStream;
import java.net.URL;
import java.net.URLConnection;
import java.util.Dictionary;
import java.util.Enumeration;
import java.util.jar.Attributes;
import java.util.jar.JarFile;
import java.util.jar.JarOutputStream;
import java.util.jar.Manifest;
import java.util.zip.ZipEntry;
import org.osgi.service.url.AbstractURLStreamHandlerService;
import org.osgi.service.url.URLStreamHandlerService;
import org.eclipse.gemini.web.core.InstallationOptions;
import org.eclipse.gemini.web.core.WebBundleManifestTransformer;
import org.eclipse.gemini.web.internal.WebContainerUtils;
import org.eclipse.gemini.web.internal.url.DirTransformer.DirTransformerCallback;
import org.eclipse.virgo.util.io.IOUtils;
import org.eclipse.virgo.util.io.JarTransformer;
import org.eclipse.virgo.util.io.JarTransformingURLConnection;
import org.eclipse.virgo.util.io.JarTransformer.JarTransformerCallback;
import org.eclipse.virgo.util.io.PathReference;
import org.eclipse.virgo.util.osgi.manifest.BundleManifest;
import org.eclipse.virgo.util.osgi.manifest.BundleManifestFactory;
/**
* {@link URLStreamHandlerService} that transforms bundles installed with the <code>war:</code> protocol.
* <p/>
* Transformations are applied using the {@link WebBundleManifestTransformer}.
*
* @see WebBundleManifestTransformer
*/
public final class WebBundleUrlStreamHandlerService extends AbstractURLStreamHandlerService {
private static final String FILE_PROTOCOL = "file";
private final WebBundleManifestTransformer transformer;
public WebBundleUrlStreamHandlerService(WebBundleManifestTransformer transformer) {
this.transformer = transformer;
}
@Override
public URLConnection openConnection(URL u) throws IOException {
WebBundleUrl url = new WebBundleUrl(u);
URL actualUrl = new URL(url.getLocation());
if (FILE_PROTOCOL.equals(actualUrl.getProtocol()) && new File(actualUrl.getPath()).isDirectory()) {
DirTransformer dirTransformer = new DirTransformer(new Callback(actualUrl, url, this.transformer));
return new DirTransformingURLConnection(actualUrl, dirTransformer, true);
} else {
JarTransformer jarTransformer = new JarTransformer(new Callback(actualUrl, url, this.transformer));
return new JarTransformingURLConnection(actualUrl, jarTransformer, true);
}
}
private static final class Callback implements JarTransformerCallback, DirTransformerCallback {
private static final String META_INF = "META-INF";
private static final String MANIFEST_MF = "MANIFEST.MF";
private final WebBundleManifestTransformer transformer;
private final URL sourceURL;
private final WebBundleUrl webBundleUrl;
Callback(URL sourceURL, WebBundleUrl url, WebBundleManifestTransformer transformer) {
this.sourceURL = sourceURL;
this.webBundleUrl = url;
this.transformer = transformer;
}
public boolean transformEntry(String entryName, InputStream is, JarOutputStream jos) throws IOException {
if (JarFile.MANIFEST_NAME.equals(entryName)) {
jos.putNextEntry(new ZipEntry(entryName));
transformManifest(is, jos);
jos.closeEntry();
return true;
}
// Delete signature files. Should be generalised into another transformer type.
return isSignatureFile(entryName);
}
private void transformManifest(InputStream inputStream, OutputStream outputStream) throws IOException {
InputStreamReader reader = new InputStreamReader(inputStream);
BundleManifest manifest = BundleManifestFactory.createBundleManifest(reader);
InstallationOptions options = new InstallationOptions(this.webBundleUrl.getOptions());
if (manifest.getHeader(WebContainerUtils.HEADER_SPRINGSOURCE_DEFAULT_WAB_HEADERS) != null) {
options.setDefaultWABHeaders(true);
}
boolean webBundle = WebContainerUtils.isWebApplicationBundle(manifest);
this.transformer.transform(manifest, sourceURL, options, webBundle);
toManifest(manifest.toDictionary()).write(outputStream);
}
private boolean isSignatureFile(String entryName) {
String[] entryNameComponents = entryName.split("/");
if (entryNameComponents.length == 2) {
if (META_INF.equals(entryNameComponents[0])) {
String entryFileName = entryNameComponents[1];
if (entryFileName.endsWith(".SF") || entryFileName.endsWith(".DSA") || entryFileName.endsWith(".RSA")) {
return true;
}
}
}
return false;
}
private static Manifest toManifest(Dictionary<String, String> headers) {
Manifest manifest = new Manifest();
Attributes attributes = manifest.getMainAttributes();
Enumeration<String> names = headers.keys();
while (names.hasMoreElements()) {
String name = names.nextElement();
String value = headers.get(name);
attributes.putValue(name, value);
}
return manifest;
}
public boolean transformFile(InputStream inputStream, PathReference toFile) throws IOException {
if (MANIFEST_MF.equals(toFile.getName()) && META_INF.equals(toFile.getParent().getName())) {
toFile.getParent().createDirectory();
OutputStream outputStream = null;
try {
outputStream = new FileOutputStream(toFile.toFile());
transformManifest(inputStream, outputStream);
} finally {
IOUtils.closeQuietly(outputStream);
}
return true;
}
// Delete signature files. Should be generalized into another
// transformer type.
return isSignatureFile(toFile);
}
private boolean isSignatureFile(PathReference file) {
if (META_INF.equals(file.getParent().getName())) {
String fileName = file.getName();
if (fileName.endsWith(".SF") || fileName.endsWith(".DSA") || fileName.endsWith(".RSA")) {
return true;
}
}
return false;
}
}
}
|
