/* * Copyright (c) 1997, 2013, 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; /** * 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 extensionOptions = new HashMap(); /** * All discovered {@link Plugin}s. * This is lazily parsed, so that we can take '-cp' option into account. * * @see #getAllPlugins() */ private List allPlugins; /** * {@link Plugin}s that are enabled in this compilation. */ public final List activePlugins = new ArrayList(); 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 cmdlineJars = new ArrayList(); /** * Gets all the {@link Plugin}s discovered so far. * *

* 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 getAllPlugins() { if(allPlugins==null) { allPlugins = new ArrayList(); allPlugins.addAll(Arrays.asList(findServices(Plugin.class, getClassLoader()))); } return allPlugins; } /** * 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("-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 wsdls = new ArrayList(); private final List schemas = new ArrayList(); private final List bindingFiles = new ArrayList(); private final List jaxwsCustomBindings = new ArrayList(); private final List jaxbCustomBindings = new ArrayList(); private final List handlerConfigs = new ArrayList(); /** * 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 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 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[] findServices(Class clazz, ClassLoader classLoader) { ServiceFinder serviceFinder = ServiceFinder.find(clazz, classLoader); List r = new ArrayList(); 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); } } }