| /* |
| * Copyright (c) 1997, 2016, Oracle and/or its affiliates. All rights reserved. |
| * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER. |
| * |
| * This code is free software; you can redistribute it and/or modify it |
| * under the terms of the GNU General Public License version 2 only, as |
| * published by the Free Software Foundation. Oracle designates this |
| * particular file as subject to the "Classpath" exception as provided |
| * by Oracle in the LICENSE file that accompanied this code. |
| * |
| * This code is distributed in the hope that it will be useful, but WITHOUT |
| * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or |
| * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License |
| * version 2 for more details (a copy is included in the LICENSE file that |
| * accompanied this code). |
| * |
| * You should have received a copy of the GNU General Public License version |
| * 2 along with this work; if not, write to the Free Software Foundation, |
| * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA. |
| * |
| * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA |
| * or visit www.oracle.com if you need additional information or have any |
| * questions. |
| */ |
| |
| package com.sun.tools.internal.ws.wscompile; |
| |
| import com.sun.codemodel.internal.JCodeModel; |
| import com.sun.tools.internal.ws.processor.generator.GeneratorExtension; |
| import com.sun.tools.internal.ws.resources.ConfigurationMessages; |
| import com.sun.tools.internal.ws.resources.WscompileMessages; |
| import com.sun.tools.internal.ws.util.ForkEntityResolver; |
| import com.sun.tools.internal.ws.wsdl.document.jaxws.JAXWSBindingsConstants; |
| import com.sun.tools.internal.ws.wsdl.document.schema.SchemaConstants; |
| import com.sun.tools.internal.xjc.api.SchemaCompiler; |
| import com.sun.tools.internal.xjc.api.SpecVersion; |
| import com.sun.tools.internal.xjc.api.XJC; |
| import com.sun.tools.internal.xjc.reader.Util; |
| import com.sun.xml.internal.ws.api.streaming.XMLStreamReaderFactory; |
| import com.sun.xml.internal.ws.streaming.XMLStreamReaderUtil; |
| import com.sun.xml.internal.ws.util.ServiceFinder; |
| import com.sun.xml.internal.ws.util.JAXWSUtils; |
| import com.sun.xml.internal.ws.util.xml.XmlUtil; |
| import org.w3c.dom.Element; |
| import org.xml.sax.EntityResolver; |
| import org.xml.sax.InputSource; |
| import org.xml.sax.helpers.LocatorImpl; |
| |
| import javax.xml.namespace.QName; |
| import javax.xml.stream.XMLStreamReader; |
| |
| import java.io.ByteArrayInputStream; |
| import java.io.ByteArrayOutputStream; |
| import java.io.File; |
| import java.io.IOException; |
| import java.io.InputStream; |
| import java.io.Reader; |
| import java.lang.reflect.Array; |
| import java.net.MalformedURLException; |
| import java.net.URL; |
| import java.util.ArrayList; |
| import java.util.Arrays; |
| import java.util.List; |
| import java.util.HashMap; |
| import java.util.logging.Level; |
| import java.util.logging.Logger; |
| |
| /** |
| * @author Vivek Pandey |
| */ |
| public class WsimportOptions extends Options { |
| /** |
| * -wsdlLocation |
| */ |
| public String wsdlLocation; |
| |
| /** |
| * Actually stores {@link com.sun.org.apache.xml.internal.resolver.tools.CatalogResolver}, but the field |
| * type is made to {@link org.xml.sax.EntityResolver} so that XJC can be |
| * used even if resolver.jar is not available in the classpath. |
| */ |
| public EntityResolver entityResolver = null; |
| |
| /** |
| * The -p option that should control the default Java package that |
| * will contain the generated code. Null if unspecified. |
| */ |
| public String defaultPackage = null; |
| |
| /** |
| * The -clientjar option to package client artifacts as jar |
| */ |
| public String clientjar = null; |
| |
| /** |
| * -XadditionalHeaders |
| */ |
| public boolean additionalHeaders; |
| |
| /** |
| * The option indicates the dir where the jwsImpl will be generated. |
| */ |
| public File implDestDir = null; |
| |
| /** |
| * optional, generated impl file only for the ordered serviceName |
| * Note: It is a QName string, formatted as: "{" + Namespace URI + "}" + local part |
| */ |
| public String implServiceName = null; |
| |
| /** |
| * optional, generated impl file only for the ordered portName |
| * Note: It is a QName string, formatted as: "{" + Namespace URI + "}" + local part |
| */ |
| public String implPortName = null; |
| |
| /** |
| * optional, if true JWS file is generated |
| */ |
| public boolean isGenerateJWS = false; |
| |
| /** |
| * Setting disableSSLHostVerification to true disables the SSL Hostname verification while fetching the wsdls. |
| * -XdisableSSLHostVerification |
| */ |
| public boolean disableSSLHostnameVerification; |
| |
| /** |
| * Setting useBaseResourceAndURLToLoadWSDL to true causes generated Service classes to load the WSDL file from |
| * a URL generated from the base resource. |
| * -XuseBaseResourceAndURLToLoadWSDL |
| */ |
| public boolean useBaseResourceAndURLToLoadWSDL = false; |
| |
| /** |
| * Java module name in {@code module-info.java}. |
| */ |
| private String javaModule = null; |
| |
| /** |
| * JAXB's {@link SchemaCompiler} to be used for handling the schema portion. |
| * This object is also configured through options. |
| */ |
| private SchemaCompiler schemaCompiler = XJC.createSchemaCompiler(); |
| |
| /** |
| * Authentication file |
| */ |
| public File authFile = null; |
| |
| //can user.home value be null? |
| public static final String defaultAuthfile |
| = System.getProperty("user.home") + System.getProperty("file.separator") |
| + ".metro" + System.getProperty("file.separator") + "auth"; |
| |
| /** |
| * Setting disableAuthenticator to true disables the DefaultAuthenticator. |
| * -XdisableAuthenticator |
| */ |
| public boolean disableAuthenticator; |
| |
| public String proxyAuth = null; |
| private String proxyHost = null; |
| private String proxyPort = null; |
| |
| /** |
| * Additional arguments |
| */ |
| public HashMap<String, String> extensionOptions = new HashMap<String, String>(); |
| |
| /** |
| * All discovered {@link Plugin}s. |
| * This is lazily parsed, so that we can take '-cp' option into account. |
| * |
| * @see #getAllPlugins() |
| */ |
| private List<Plugin> allPlugins; |
| |
| /** |
| * {@link Plugin}s that are enabled in this compilation. |
| */ |
| public final List<Plugin> activePlugins = new ArrayList<Plugin>(); |
| |
| public JCodeModel getCodeModel() { |
| if(codeModel == null) |
| codeModel = new JCodeModel(); |
| return codeModel; |
| } |
| |
| public SchemaCompiler getSchemaCompiler() { |
| schemaCompiler.setTargetVersion(SpecVersion.parse(target.getVersion())); |
| if(entityResolver != null) { |
| //set if its not null so as not to override catalog option specified via xjc args |
| schemaCompiler.setEntityResolver(entityResolver); |
| } |
| return schemaCompiler; |
| } |
| |
| public void setCodeModel(JCodeModel codeModel) { |
| this.codeModel = codeModel; |
| } |
| |
| private JCodeModel codeModel; |
| |
| /** |
| * This captures jars passed on the commandline and passes them to XJC and puts them in the classpath for compilation |
| */ |
| public List<String> cmdlineJars = new ArrayList<String>(); |
| |
| /** |
| * Gets all the {@link Plugin}s discovered so far. |
| * |
| * <p> |
| * A plugins are enumerated when this method is called for the first time, |
| * by taking {@link #classpath} into account. That means |
| * "-cp plugin.jar" has to come before you specify options to enable it. |
| */ |
| public List<Plugin> getAllPlugins() { |
| if(allPlugins==null) { |
| allPlugins = new ArrayList<Plugin>(); |
| allPlugins.addAll(Arrays.asList(findServices(Plugin.class, getClassLoader()))); |
| } |
| return allPlugins; |
| } |
| |
| /** |
| * Gets Java module name option. |
| * @return Java module name option or {@code null} if this option was not set. |
| */ |
| public String getModuleName() { |
| return javaModule; |
| } |
| |
| /** |
| * Parses arguments and fill fields of this object. |
| * |
| * @exception BadCommandLineException |
| * thrown when there's a problem in the command-line arguments |
| */ |
| @Override |
| public final void parseArguments( String[] args ) throws BadCommandLineException { |
| |
| for (int i = 0; i < args.length; i++) { |
| if(args[i].length()==0) |
| throw new BadCommandLineException(); |
| if (args[i].charAt(0) == '-') { |
| int j = parseArguments(args,i); |
| if(j==0) |
| throw new BadCommandLineException(WscompileMessages.WSCOMPILE_INVALID_OPTION(args[i])); |
| i += (j-1); |
| } else { |
| if(args[i].endsWith(".jar")) { |
| |
| try { |
| cmdlineJars.add(args[i]); |
| schemaCompiler.getOptions().scanEpisodeFile(new File(args[i])); |
| |
| } catch (com.sun.tools.internal.xjc.BadCommandLineException e) { |
| //Driver.usage(jaxbOptions,false); |
| throw new BadCommandLineException(e.getMessage(), e); |
| } |
| } else{ |
| addFile(args[i]); |
| } |
| } |
| } |
| |
| if (encoding != null && schemaCompiler.getOptions().encoding == null) { |
| try { |
| schemaCompiler.getOptions().parseArgument( |
| new String[] {"-encoding", encoding}, 0); |
| } catch (com.sun.tools.internal.xjc.BadCommandLineException ex) { |
| Logger.getLogger(WsimportOptions.class.getName()).log(Level.SEVERE, null, ex); |
| } |
| } |
| |
| if(destDir == null) |
| destDir = new File("."); |
| if(sourceDir == null) |
| sourceDir = destDir; |
| } |
| |
| /** -Xno-addressing-databinding option to disable addressing namespace data binding. This is |
| * experimental switch and will be working as a temporary workaround till |
| * jaxb can provide a better way to selelctively disable compiling of an |
| * schema component. |
| * **/ |
| public boolean noAddressingBbinding; |
| |
| @Override |
| public int parseArguments(String[] args, int i) throws BadCommandLineException { |
| int j = super.parseArguments(args ,i); |
| if(j>0) return j; // understood by the super class |
| |
| if (args[i].equals("-b")) { |
| addBindings(requireArgument("-b", args, ++i)); |
| return 2; |
| } else if (args[i].equals("-wsdllocation")) { |
| wsdlLocation = requireArgument("-wsdllocation", args, ++i); |
| return 2; |
| } else if (args[i].equals("-XadditionalHeaders")) { |
| additionalHeaders = true; |
| return 1; |
| } else if (args[i].equals("-XdisableSSLHostnameVerification")) { |
| disableSSLHostnameVerification = true; |
| return 1; |
| } else if (args[i].equals("-p")) { |
| defaultPackage = requireArgument("-p", args, ++i); |
| return 2; |
| } else if (args[i].equals("-m")) { |
| javaModule = requireArgument("-m", args, ++i); |
| return 2; |
| } else if (args[i].equals("-catalog")) { |
| String catalog = requireArgument("-catalog", args, ++i); |
| try { |
| if (entityResolver == null) { |
| if (catalog != null && catalog.length() > 0) |
| entityResolver = XmlUtil.createEntityResolver(JAXWSUtils.getFileOrURL(JAXWSUtils.absolutize(Util.escapeSpace(catalog)))); |
| } else if (catalog != null && catalog.length() > 0) { |
| EntityResolver er = XmlUtil.createEntityResolver(JAXWSUtils.getFileOrURL(JAXWSUtils.absolutize(Util.escapeSpace(catalog)))); |
| entityResolver = new ForkEntityResolver(er, entityResolver); |
| } |
| } catch (IOException e) { |
| throw new BadCommandLineException(WscompileMessages.WSIMPORT_FAILED_TO_PARSE(catalog, e.getMessage())); |
| } |
| return 2; |
| } else if (args[i].startsWith("-httpproxy:")) { |
| String value = args[i].substring(11); |
| if (value.length() == 0) { |
| throw new BadCommandLineException(WscompileMessages.WSCOMPILE_INVALID_OPTION(args[i])); |
| } |
| parseProxy(value); |
| if (proxyHost != null || proxyPort != null) { |
| System.setProperty("proxySet", "true"); |
| } |
| if (proxyHost != null) { |
| System.setProperty("proxyHost", proxyHost); |
| } |
| if (proxyPort != null) { |
| System.setProperty("proxyPort", proxyPort); |
| } |
| return 1; |
| } else if (args[i].equals("-Xno-addressing-databinding")) { |
| noAddressingBbinding = true; |
| return 1; |
| } else if (args[i].startsWith("-B")) { |
| // JAXB option pass through. |
| String[] subCmd = new String[args.length-i]; |
| System.arraycopy(args,i,subCmd,0,subCmd.length); |
| subCmd[0] = subCmd[0].substring(2); // trim off the first "-B" |
| |
| com.sun.tools.internal.xjc.Options jaxbOptions = schemaCompiler.getOptions(); |
| try { |
| int r = jaxbOptions.parseArgument(subCmd, 0); |
| if(r==0) { |
| //Driver.usage(jaxbOptions,false); |
| throw new BadCommandLineException(WscompileMessages.WSIMPORT_NO_SUCH_JAXB_OPTION(subCmd[0])); |
| } |
| return r; |
| } catch (com.sun.tools.internal.xjc.BadCommandLineException e) { |
| //Driver.usage(jaxbOptions,false); |
| throw new BadCommandLineException(e.getMessage(),e); |
| } |
| } else if (args[i].equals("-Xauthfile")) { |
| String authfile = requireArgument("-Xauthfile", args, ++i); |
| authFile = new File(authfile); |
| return 2; |
| } else if (args[i].equals("-clientjar")) { |
| clientjar = requireArgument("-clientjar", args, ++i); |
| return 2; |
| } else if (args[i].equals("-implDestDir")) { |
| implDestDir = new File(requireArgument("-implDestDir", args, ++i)); |
| if (!implDestDir.exists()) |
| throw new BadCommandLineException(WscompileMessages.WSCOMPILE_NO_SUCH_DIRECTORY(implDestDir.getPath())); |
| return 2; |
| } else if (args[i].equals("-implServiceName")) { |
| implServiceName = requireArgument("-implServiceName", args, ++i); |
| return 2; |
| } else if (args[i].equals("-implPortName")) { |
| implPortName = requireArgument("-implPortName", args, ++i); |
| return 2; |
| } else if (args[i].equals("-generateJWS")) { |
| isGenerateJWS = true; |
| return 1; |
| } else if (args[i].equals("-XuseBaseResourceAndURLToLoadWSDL")) { |
| useBaseResourceAndURLToLoadWSDL = true; |
| return 1; |
| } else if (args[i].equals("-XdisableAuthenticator")) { |
| disableAuthenticator = true; |
| return 1; |
| } |
| |
| // handle additional options |
| for (GeneratorExtension f:ServiceFinder.find(GeneratorExtension.class)) { |
| if (f.validateOption(args[i])) { |
| extensionOptions.put(args[i], requireArgument(args[i], args, ++i)); |
| return 2; |
| } |
| } |
| |
| // see if this is one of the extensions |
| for( Plugin plugin : getAllPlugins() ) { |
| try { |
| if(('-' + plugin.getOptionName()).equals(args[i])) { |
| activePlugins.add(plugin); |
| plugin.onActivated(this); |
| return 1; |
| } |
| int r = plugin.parseArgument(this, args, i); |
| if (r != 0) { |
| return r; |
| } |
| } catch (IOException e) { |
| throw new BadCommandLineException(e.getMessage(),e); |
| } |
| } |
| |
| return 0; // what's this option? |
| } |
| |
| public void validate() throws BadCommandLineException { |
| if (wsdls.isEmpty()) { |
| throw new BadCommandLineException(WscompileMessages.WSIMPORT_MISSING_FILE()); |
| } |
| |
| if(wsdlLocation !=null && clientjar != null) { |
| throw new BadCommandLineException(WscompileMessages.WSIMPORT_WSDLLOCATION_CLIENTJAR()); |
| } |
| if(wsdlLocation == null){ |
| wsdlLocation = wsdls.get(0).getSystemId(); |
| } |
| |
| |
| } |
| |
| @Override |
| protected void addFile(String arg) throws BadCommandLineException { |
| addFile(arg, wsdls, ".wsdl"); |
| } |
| |
| private final List<InputSource> wsdls = new ArrayList<InputSource>(); |
| private final List<InputSource> schemas = new ArrayList<InputSource>(); |
| private final List<InputSource> bindingFiles = new ArrayList<InputSource>(); |
| private final List<InputSource> jaxwsCustomBindings = new ArrayList<InputSource>(); |
| private final List<InputSource> jaxbCustomBindings = new ArrayList<InputSource>(); |
| private final List<Element> handlerConfigs = new ArrayList<Element>(); |
| |
| /** |
| * There is supposed to be one handler chain per generated SEI. |
| * TODO: There is possible bug, how to associate a @HandlerChain |
| * with each port on the generated SEI. For now lets preserve the JAXWS 2.0 FCS |
| * behaviour and generate only one @HandlerChain on the SEI |
| */ |
| public Element getHandlerChainConfiguration(){ |
| if(handlerConfigs.size() > 0) |
| return handlerConfigs.get(0); |
| return null; |
| } |
| |
| public void addHandlerChainConfiguration(Element config){ |
| handlerConfigs.add(config); |
| } |
| |
| public InputSource[] getWSDLs() { |
| return wsdls.toArray(new InputSource[wsdls.size()]); |
| } |
| |
| public InputSource[] getSchemas() { |
| return schemas.toArray(new InputSource[schemas.size()]); |
| } |
| |
| public InputSource[] getWSDLBindings() { |
| return jaxwsCustomBindings.toArray(new InputSource[jaxwsCustomBindings.size()]); |
| } |
| |
| public InputSource[] getSchemaBindings() { |
| return jaxbCustomBindings.toArray(new InputSource[jaxbCustomBindings.size()]); |
| } |
| |
| public void addWSDL(File source) { |
| addWSDL(fileToInputSource(source)); |
| } |
| |
| public void addWSDL(InputSource is) { |
| wsdls.add(absolutize(is)); |
| } |
| |
| public void addSchema(File source) { |
| addSchema(fileToInputSource(source)); |
| } |
| |
| public void addSchema(InputSource is) { |
| schemas.add(is); |
| } |
| |
| private InputSource fileToInputSource(File source) { |
| try { |
| String url = source.toURL().toExternalForm(); |
| return new InputSource(Util.escapeSpace(url)); |
| } catch (MalformedURLException e) { |
| return new InputSource(source.getPath()); |
| } |
| } |
| |
| /** |
| * Recursively scan directories and add all XSD files in it. |
| */ |
| public void addGrammarRecursive(File dir) { |
| addRecursive(dir, ".wsdl", wsdls); |
| addRecursive(dir, ".xsd", schemas); |
| } |
| |
| /** |
| * Adds a new input schema. |
| */ |
| public void addWSDLBindFile(InputSource is) { |
| jaxwsCustomBindings.add(new RereadInputSource(absolutize(is))); |
| } |
| |
| public void addSchemmaBindFile(InputSource is) { |
| jaxbCustomBindings.add(new RereadInputSource(absolutize(is))); |
| } |
| |
| private void addRecursive(File dir, String suffix, List<InputSource> result) { |
| File[] files = dir.listFiles(); |
| if (files == null) return; // work defensively |
| |
| for (File f : files) { |
| if (f.isDirectory()) |
| addRecursive(f, suffix, result); |
| else if (f.getPath().endsWith(suffix)) |
| result.add(absolutize(fileToInputSource(f))); |
| } |
| } |
| |
| private InputSource absolutize(InputSource is) { |
| // absolutize all the system IDs in the input, |
| // so that we can map system IDs to DOM trees. |
| try { |
| URL baseURL = new File(".").getCanonicalFile().toURL(); |
| is.setSystemId(new URL(baseURL, is.getSystemId()).toExternalForm()); |
| } catch (IOException e) { |
| // ignore |
| } |
| return is; |
| } |
| |
| public void addBindings(String name) throws BadCommandLineException { |
| addFile(name, bindingFiles, null); |
| } |
| |
| /** |
| * Parses a token to a file (or a set of files) |
| * and add them as {@link InputSource} to the specified list. |
| * |
| * @param suffix If the given token is a directory name, we do a recusive search |
| * and find all files that have the given suffix. |
| */ |
| private void addFile(String name, List<InputSource> target, String suffix) throws BadCommandLineException { |
| Object src; |
| try { |
| src = Util.getFileOrURL(name); |
| } catch (IOException e) { |
| throw new BadCommandLineException(WscompileMessages.WSIMPORT_NOT_A_FILE_NOR_URL(name)); |
| } |
| if (src instanceof URL) { |
| target.add(absolutize(new InputSource(Util.escapeSpace(((URL) src).toExternalForm())))); |
| } else { |
| File fsrc = (File) src; |
| if (fsrc.isDirectory()) { |
| addRecursive(fsrc, suffix, target); |
| } else { |
| target.add(absolutize(fileToInputSource(fsrc))); |
| } |
| } |
| } |
| |
| |
| /** |
| * Exposing it as a public method to allow external tools such as NB to read from wsdl model and work on it. |
| * TODO: WSDL model needs to be exposed - basically at tool time we need to use the runtimw wsdl model |
| * |
| * Binding files could be jaxws or jaxb. This method identifies jaxws and jaxb binding files and keeps them separately. jaxb binding files are given separately |
| * to JAXB in {@link com.sun.tools.internal.ws.processor.modeler.wsdl.JAXBModelBuilder} |
| * |
| * @param receiver {@link ErrorReceiver} |
| */ |
| public final void parseBindings(ErrorReceiver receiver){ |
| for (InputSource is : bindingFiles) { |
| XMLStreamReader reader = |
| XMLStreamReaderFactory.create(is,true); |
| XMLStreamReaderUtil.nextElementContent(reader); |
| if (reader.getName().equals(JAXWSBindingsConstants.JAXWS_BINDINGS)) { |
| jaxwsCustomBindings.add(new RereadInputSource(is)); |
| } else if (reader.getName().equals(JAXWSBindingsConstants.JAXB_BINDINGS) || |
| reader.getName().equals(new QName(SchemaConstants.NS_XSD, "schema"))) { |
| jaxbCustomBindings.add(new RereadInputSource(is)); |
| } else { |
| LocatorImpl locator = new LocatorImpl(); |
| locator.setSystemId(reader.getLocation().getSystemId()); |
| locator.setPublicId(reader.getLocation().getPublicId()); |
| locator.setLineNumber(reader.getLocation().getLineNumber()); |
| locator.setColumnNumber(reader.getLocation().getColumnNumber()); |
| receiver.warning(locator, ConfigurationMessages.CONFIGURATION_NOT_BINDING_FILE(is.getSystemId())); |
| } |
| } |
| } |
| |
| /** |
| * Get extension argument |
| */ |
| public String getExtensionOption(String argument) { |
| return extensionOptions.get(argument); |
| } |
| |
| private void parseProxy(String text) throws BadCommandLineException { |
| int i = text.lastIndexOf('@'); |
| int j = text.lastIndexOf(':'); |
| |
| if (i > 0) { |
| proxyAuth = text.substring(0, i); |
| if (j > i) { |
| proxyHost = text.substring(i + 1, j); |
| proxyPort = text.substring(j + 1); |
| } else { |
| proxyHost = text.substring(i + 1); |
| proxyPort = "8080"; |
| } |
| } else { |
| //no auth info |
| if (j < 0) { |
| //no port |
| proxyHost = text; |
| proxyPort = "8080"; |
| } else { |
| proxyHost = text.substring(0, j); |
| proxyPort = text.substring(j + 1); |
| } |
| } |
| try { |
| Integer.valueOf(proxyPort); |
| } catch (NumberFormatException e) { |
| throw new BadCommandLineException(WscompileMessages.WSIMPORT_ILLEGAL_PROXY(text)); |
| } |
| } |
| |
| /** |
| * Looks for all "META-INF/services/[className]" files and |
| * create one instance for each class name found inside this file. |
| */ |
| private static <T> T[] findServices(Class<T> clazz, ClassLoader classLoader) { |
| ServiceFinder<T> serviceFinder = ServiceFinder.find(clazz, classLoader); |
| List<T> r = new ArrayList<T>(); |
| for (T t : serviceFinder) { |
| r.add(t); |
| } |
| return r.toArray((T[]) Array.newInstance(clazz, r.size())); |
| } |
| |
| private static final class ByteStream extends ByteArrayOutputStream { |
| byte[] getBuffer() { |
| return buf; |
| } |
| } |
| |
| private static final class RereadInputStream extends InputStream { |
| private InputStream is; |
| private ByteStream bs; |
| |
| RereadInputStream(InputStream is) { |
| this.is = is; |
| this.bs = new ByteStream(); |
| } |
| |
| @Override |
| public int available() throws IOException { |
| return is.available(); |
| } |
| |
| @Override |
| public void close() throws IOException { |
| if (bs != null) { |
| InputStream i = new ByteArrayInputStream(bs.getBuffer()); |
| bs = null; |
| is.close(); |
| is = i; |
| } |
| } |
| |
| @Override |
| public synchronized void mark(int readlimit) { |
| is.mark(readlimit); |
| } |
| |
| @Override |
| public boolean markSupported() { |
| return is.markSupported(); |
| } |
| |
| @Override |
| public int read() throws IOException { |
| int r = is.read(); |
| if (bs != null) |
| bs.write(r); |
| return r; |
| } |
| |
| @Override |
| public int read(byte[] b, int off, int len) throws IOException { |
| int r = is.read(b, off, len); |
| if (r > 0 && bs != null) |
| bs.write(b, off, r); |
| return r; |
| } |
| |
| @Override |
| public int read(byte[] b) throws IOException { |
| int r = is.read(b); |
| if (r > 0 && bs != null) |
| bs.write(b, 0, r); |
| return r; |
| } |
| |
| @Override |
| public synchronized void reset() throws IOException { |
| is.reset(); |
| } |
| } |
| |
| private static final class RereadInputSource extends InputSource { |
| private InputSource is; |
| |
| RereadInputSource(InputSource is) { |
| this.is = is; |
| } |
| |
| @Override |
| public InputStream getByteStream() { |
| InputStream i = is.getByteStream(); |
| if (i != null && !(i instanceof RereadInputStream)) { |
| i = new RereadInputStream(i); |
| is.setByteStream(i); |
| } |
| return i; |
| } |
| |
| @Override |
| public Reader getCharacterStream() { |
| // TODO Auto-generated method stub |
| return is.getCharacterStream(); |
| } |
| |
| @Override |
| public String getEncoding() { |
| return is.getEncoding(); |
| } |
| |
| @Override |
| public String getPublicId() { |
| return is.getPublicId(); |
| } |
| |
| @Override |
| public String getSystemId() { |
| return is.getSystemId(); |
| } |
| |
| @Override |
| public void setByteStream(InputStream byteStream) { |
| is.setByteStream(byteStream); |
| } |
| |
| @Override |
| public void setCharacterStream(Reader characterStream) { |
| is.setCharacterStream(characterStream); |
| } |
| |
| @Override |
| public void setEncoding(String encoding) { |
| is.setEncoding(encoding); |
| } |
| |
| @Override |
| public void setPublicId(String publicId) { |
| is.setPublicId(publicId); |
| } |
| |
| @Override |
| public void setSystemId(String systemId) { |
| is.setSystemId(systemId); |
| } |
| } |
| |
| @Override |
| protected void disableXmlSecurity() { |
| super.disableXmlSecurity(); |
| schemaCompiler.getOptions().disableXmlSecurity = true; |
| } |
| } |