| /* |
| * Licensed to the Apache Software Foundation (ASF) under one |
| * or more contributor license agreements. See the NOTICE file |
| * distributed with this work for additional information |
| * regarding copyright ownership. The ASF licenses this file |
| * to you under the Apache License, Version 2.0 (the |
| * "License"); you may not use this file except in compliance |
| * with the License. You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, |
| * software distributed under the License is distributed on an |
| * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY |
| * KIND, either express or implied. See the License for the |
| * specific language governing permissions and limitations |
| * under the License. |
| */ |
| |
| |
| package org.apache.axis2.deployment.repository.util; |
| |
| import org.apache.axiom.om.OMElement; |
| import org.apache.axiom.om.OMNamespace; |
| import org.apache.axis2.AxisFault; |
| import org.apache.axis2.context.ConfigurationContext; |
| import org.apache.axis2.deployment.DeploymentConstants; |
| import org.apache.axis2.deployment.DeploymentErrorMsgs; |
| import org.apache.axis2.deployment.DeploymentException; |
| import org.apache.axis2.deployment.DescriptionBuilder; |
| import org.apache.axis2.deployment.ModuleBuilder; |
| import org.apache.axis2.deployment.ServiceBuilder; |
| import org.apache.axis2.deployment.ServiceGroupBuilder; |
| import org.apache.axis2.deployment.resolver.AARBasedWSDLLocator; |
| import org.apache.axis2.deployment.resolver.AARFileBasedURIResolver; |
| import org.apache.axis2.deployment.resolver.WarBasedWSDLLocator; |
| import org.apache.axis2.deployment.resolver.WarFileBasedURIResolver; |
| import org.apache.axis2.description.AxisModule; |
| import org.apache.axis2.description.AxisService; |
| import org.apache.axis2.description.AxisServiceGroup; |
| import org.apache.axis2.description.WSDL11ToAllAxisServicesBuilder; |
| import org.apache.axis2.description.WSDL11ToAxisServiceBuilder; |
| import org.apache.axis2.description.WSDL20ToAllAxisServicesBuilder; |
| import org.apache.axis2.description.WSDL2Constants; |
| import org.apache.axis2.description.WSDLToAxisServiceBuilder; |
| import org.apache.axis2.engine.AxisConfiguration; |
| import org.apache.axis2.i18n.Messages; |
| import org.apache.axis2.namespace.Constants; |
| import org.apache.axis2.util.Utils; |
| import org.apache.axis2.util.XMLUtils; |
| import org.apache.commons.logging.Log; |
| import org.apache.commons.logging.LogFactory; |
| |
| import javax.xml.stream.XMLStreamException; |
| import java.io.ByteArrayInputStream; |
| import java.io.ByteArrayOutputStream; |
| import java.io.File; |
| import java.io.FileInputStream; |
| import java.io.FileNotFoundException; |
| import java.io.IOException; |
| import java.io.InputStream; |
| import java.util.ArrayList; |
| import java.util.HashMap; |
| import java.util.List; |
| import java.util.zip.ZipEntry; |
| import java.util.zip.ZipInputStream; |
| |
| public class ArchiveReader implements DeploymentConstants { |
| private static final Log log = LogFactory.getLog(ArchiveReader.class); |
| |
| public ArrayList<AxisService> buildServiceGroup(InputStream zin, DeploymentFileData currentFile, |
| AxisServiceGroup axisServiceGroup, HashMap<String, AxisService> wsdlServices, |
| ConfigurationContext configCtx) |
| throws XMLStreamException, AxisFault { |
| |
| DescriptionBuilder builder = new DescriptionBuilder(zin, configCtx); |
| OMElement rootElement = builder.buildOM(); |
| String elementName = rootElement.getLocalName(); |
| |
| if (TAG_SERVICE.equals(elementName)) { |
| AxisService axisService = null; |
| String serviceName = DescriptionBuilder.getShortFileName(currentFile.getName()); |
| if (serviceName != null) { |
| axisService = (AxisService) wsdlServices.get(serviceName); |
| } |
| if (axisService == null) { |
| axisService = (AxisService) wsdlServices.get( |
| DescriptionBuilder.getShortFileName(currentFile.getName())); |
| } |
| if (axisService == null) { |
| axisService = new AxisService(serviceName); |
| } else { |
| axisService.setWsdlFound(true); |
| axisService.setCustomWsdl(true); |
| } |
| |
| axisService.setParent(axisServiceGroup); |
| axisService.setClassLoader(currentFile.getClassLoader()); |
| |
| ServiceBuilder serviceBuilder = new ServiceBuilder(configCtx, axisService); |
| serviceBuilder.setWsdlServiceMap(wsdlServices); |
| AxisService service = serviceBuilder.populateService(rootElement); |
| |
| ArrayList<AxisService> serviceList = new ArrayList<AxisService>(); |
| serviceList.add(service); |
| return serviceList; |
| } else if (TAG_SERVICE_GROUP.equals(elementName)) { |
| ServiceGroupBuilder groupBuilder = new ServiceGroupBuilder(rootElement, wsdlServices, |
| configCtx); |
| return groupBuilder.populateServiceGroup(axisServiceGroup); |
| } |
| throw new AxisFault("Invalid services.xml found"); |
| } |
| |
| /** |
| * Extracts Service XML files and builds the service groups. |
| * |
| * @param filename |
| * @param axisServiceGroup |
| * @param extractService |
| * @param wsdlServices |
| * @param configCtx |
| * @return Returns ArrayList. |
| * @throws DeploymentException |
| */ |
| public ArrayList<AxisService> processServiceGroup(String filename, DeploymentFileData currentFile, |
| AxisServiceGroup axisServiceGroup, |
| boolean extractService, |
| HashMap<String, AxisService> wsdlServices, |
| ConfigurationContext configCtx) |
| throws AxisFault { |
| // get attribute values |
| if (!extractService) { |
| ZipInputStream zin = null; |
| FileInputStream fin = null; |
| try { |
| fin = new FileInputStream(filename); |
| zin = new ZipInputStream(fin); |
| ZipEntry entry; |
| while ((entry = zin.getNextEntry()) != null) { |
| if (entry.getName().equalsIgnoreCase(SERVICES_XML)) { |
| axisServiceGroup.setServiceGroupName( |
| DescriptionBuilder.getShortFileName(currentFile.getName())); |
| return buildServiceGroup(zin, currentFile, axisServiceGroup, wsdlServices, |
| configCtx); |
| } |
| } |
| throw new DeploymentException( |
| Messages.getMessage(DeploymentErrorMsgs.SERVICE_XML_NOT_FOUND, filename)); |
| } catch (Exception e) { |
| throw new DeploymentException(e); |
| } finally { |
| if (zin != null) { |
| try { |
| zin.close(); |
| } catch (IOException e) { |
| log.info(Messages.getMessage("errorininputstreamclose")); |
| } |
| } |
| if (fin != null) { |
| try { |
| fin.close(); |
| } catch (IOException e) { |
| log.info(Messages.getMessage("errorininputstreamclose")); |
| } |
| } |
| } |
| } else { |
| File file = new File(filename, SERVICES_XML); |
| if (!file.exists()) { |
| // try for meta-inf |
| file = new File(filename, SERVICES_XML.toLowerCase()); |
| } |
| if (file.exists()) { |
| InputStream in = null; |
| try { |
| in = new FileInputStream(file); |
| axisServiceGroup.setServiceGroupName(currentFile.getName()); |
| return buildServiceGroup(in, currentFile, axisServiceGroup, wsdlServices, configCtx); |
| } catch (FileNotFoundException e) { |
| throw new DeploymentException( |
| Messages.getMessage(DeploymentErrorMsgs.FILE_NOT_FOUND, |
| e.getMessage())); |
| } catch (XMLStreamException e) { |
| throw new DeploymentException( |
| Messages.getMessage(DeploymentErrorMsgs.XML_STREAM_EXCEPTION, |
| e.getMessage())); |
| } finally { |
| if (in != null) { |
| try { |
| in.close(); |
| } catch (IOException e) { |
| log.info(Messages.getMessage("errorininputstreamclose")); |
| } |
| } |
| } |
| } else { |
| throw new DeploymentException( |
| Messages.getMessage(DeploymentErrorMsgs.SERVICE_XML_NOT_FOUND)); |
| } |
| } |
| } |
| |
| /** |
| * Creats AxisService. |
| * |
| * @param in |
| * @return Returns AxisService. |
| * @throws DeploymentException |
| */ |
| private List<AxisService> processWSDLFile(WSDLToAxisServiceBuilder axisServiceBuilder, |
| File serviceArchiveFile, |
| boolean isArchive, InputStream in, String baseURI) |
| throws DeploymentException { |
| try { |
| |
| if (serviceArchiveFile != null && isArchive) { |
| axisServiceBuilder.setCustomResolver( |
| new AARFileBasedURIResolver(serviceArchiveFile)); |
| if (axisServiceBuilder instanceof WSDL11ToAllAxisServicesBuilder) { |
| |
| ((WSDL11ToAllAxisServicesBuilder) axisServiceBuilder).setCustomWSDLResolver( |
| new AARBasedWSDLLocator(baseURI, serviceArchiveFile, in)); |
| |
| ((WSDL11ToAllAxisServicesBuilder) axisServiceBuilder).setDocumentBaseUri( |
| serviceArchiveFile.getCanonicalFile().toURI().toString()); |
| |
| } else if (axisServiceBuilder instanceof WSDL20ToAllAxisServicesBuilder) { |
| ((WSDL20ToAllAxisServicesBuilder) axisServiceBuilder).setCustomWSDLResolver( |
| new AARBasedWSDLLocator(baseURI, serviceArchiveFile, in)); |
| // trying to use the jar scheme as the base URI. I think this can be used to handle |
| // wsdl 1.1 as well without using a custom URI resolver. Need to look at it later. |
| axisServiceBuilder.setBaseUri( |
| "jar:file://" + serviceArchiveFile.toURI() + "!/" + baseURI); |
| } |
| } else { |
| if (serviceArchiveFile != null) { |
| axisServiceBuilder.setBaseUri( |
| serviceArchiveFile.getParentFile().toURI().toString()); |
| |
| if (axisServiceBuilder instanceof WSDL11ToAllAxisServicesBuilder) { |
| ((WSDL11ToAllAxisServicesBuilder) axisServiceBuilder).setDocumentBaseUri( |
| serviceArchiveFile.getCanonicalFile().toURI().toString()); |
| } |
| } |
| } |
| if (axisServiceBuilder instanceof WSDL11ToAllAxisServicesBuilder) { |
| return ((WSDL11ToAllAxisServicesBuilder) axisServiceBuilder).populateAllServices(); |
| } else if (axisServiceBuilder instanceof WSDL20ToAllAxisServicesBuilder) { |
| return ((WSDL20ToAllAxisServicesBuilder) axisServiceBuilder).populateAllServices(); |
| } |
| } catch (AxisFault axisFault) { |
| log.info("Trouble processing wsdl file :" + axisFault.getMessage()); |
| if (log.isDebugEnabled()) { |
| log.debug(axisFault); |
| } |
| } catch (IOException ioex) { |
| log.info("Trouble processing wsdl file :" + ioex.getMessage()); |
| if (log.isDebugEnabled()) { |
| log.debug(ioex); |
| } |
| } |
| return null; |
| } |
| |
| /** |
| * Creates service objects from wsdl file inside a service archive file. |
| * |
| * @param file <code>ArchiveFileData</code> |
| * @throws DeploymentException <code>DeploymentException</code> |
| */ |
| public HashMap<String, AxisService> processWSDLs(DeploymentFileData file) |
| throws DeploymentException { |
| File serviceFile = file.getFile(); |
| // to store service come from wsdl files |
| HashMap<String, AxisService> servicesMap = new HashMap<String, AxisService>(); |
| boolean isDirectory = serviceFile.isDirectory(); |
| if (isDirectory) { |
| try { |
| File metaInfFolder = new File(serviceFile, META_INF); |
| |
| if (!metaInfFolder.exists()) { |
| metaInfFolder = new File(serviceFile, META_INF.toLowerCase()); |
| if (!metaInfFolder.exists()) { |
| throw new DeploymentException( |
| Messages.getMessage( |
| DeploymentErrorMsgs.META_INF_MISSING, |
| serviceFile.getName())); |
| } |
| } |
| |
| processFilesInFolder(metaInfFolder, servicesMap); |
| |
| } catch (FileNotFoundException e) { |
| throw new DeploymentException(e); |
| } catch (IOException e) { |
| throw new DeploymentException(e); |
| } catch (XMLStreamException e) { |
| throw new DeploymentException(e); |
| } |
| } else { |
| ZipInputStream zin; |
| FileInputStream fin; |
| try { |
| fin = new FileInputStream(serviceFile); |
| zin = new ZipInputStream(fin); |
| |
| //TODO Check whether this WSDL is empty |
| |
| ZipEntry entry; |
| byte[] buf = new byte[1024]; |
| int read; |
| ByteArrayOutputStream out; |
| while ((entry = zin.getNextEntry()) != null) { |
| String entryName = entry.getName().toLowerCase(); |
| if (entryName.startsWith(META_INF.toLowerCase()) |
| && entryName.endsWith(SUFFIX_WSDL)) { |
| out = new ByteArrayOutputStream(); |
| |
| // we do not want to generate the services for the |
| // imported wsdl of one file. |
| if ((entryName.indexOf("/") != entryName.lastIndexOf("/")) |
| || (entryName.indexOf("wsdl_") != -1)) { |
| //only care abt the toplevel wsdl |
| continue; |
| } |
| |
| while ((read = zin.read(buf)) > 0) { |
| out.write(buf, 0, read); |
| } |
| |
| ByteArrayInputStream in = new ByteArrayInputStream(out.toByteArray()); |
| |
| // now the question is which version of WSDL file this archive contains. |
| // lets check the namespace of the root element and decide. But since we are |
| // using axiom (dude, you are becoming handy here :)), we will not build the |
| // whole thing. |
| OMNamespace documentElementNS = |
| ((OMElement) XMLUtils.toOM(in)).getNamespace(); |
| if (documentElementNS != null) { |
| WSDLToAxisServiceBuilder wsdlToAxisServiceBuilder; |
| if (WSDL2Constants.WSDL_NAMESPACE |
| .equals(documentElementNS.getNamespaceURI())) { |
| // we have a WSDL 2.0 document here. |
| wsdlToAxisServiceBuilder = new WSDL20ToAllAxisServicesBuilder( |
| new ByteArrayInputStream(out.toByteArray())); |
| wsdlToAxisServiceBuilder.setBaseUri(entryName); |
| } else if (Constants.NS_URI_WSDL11. |
| equals(documentElementNS.getNamespaceURI())) { |
| wsdlToAxisServiceBuilder = new WSDL11ToAllAxisServicesBuilder( |
| new ByteArrayInputStream(out.toByteArray())); |
| ((WSDL11ToAxisServiceBuilder) wsdlToAxisServiceBuilder).setDocumentBaseUri(entryName); |
| } else { |
| throw new DeploymentException(Messages.getMessage("invalidWSDLFound")); |
| } |
| List<AxisService> services = processWSDLFile(wsdlToAxisServiceBuilder, |
| serviceFile, true, |
| new ByteArrayInputStream( |
| out.toByteArray()), |
| entry.getName()); |
| if (services != null) { |
| for (int i = 0; i < services.size(); i++) { |
| AxisService axisService = (AxisService) services.get(i); |
| if (axisService != null) { |
| servicesMap.put(axisService.getName(), axisService); |
| } |
| } |
| } |
| } |
| } |
| } |
| try { |
| zin.close(); |
| } catch (IOException e) { |
| log.info(e); |
| } |
| try { |
| fin.close(); |
| } catch (IOException e) { |
| log.info(e); |
| } |
| } catch (FileNotFoundException e) { |
| throw new DeploymentException(e); |
| } catch (IOException e) { |
| throw new DeploymentException(e); |
| } catch (XMLStreamException e) { |
| throw new DeploymentException(e); |
| } |
| } |
| return servicesMap; |
| } |
| |
| public List<AxisService> getAxisServiceFromWsdl(InputStream in, |
| ClassLoader loader, String wsdlUrl) throws Exception { |
| // ByteArrayInputStream in = new ByteArrayInputStream(out.toByteArray()); |
| |
| // now the question is which version of WSDL file this archive contains. |
| // lets check the namespace of the root element and decide. But since we are |
| // using axiom (dude, you are becoming handy here :)), we will not build the |
| // whole thing. |
| OMElement element = (OMElement) XMLUtils.toOM(in); |
| OMNamespace documentElementNS = element.getNamespace(); |
| if (documentElementNS != null) { |
| WSDLToAxisServiceBuilder wsdlToAxisServiceBuilder; |
| ByteArrayOutputStream out = new ByteArrayOutputStream(); |
| element.serialize(out); |
| if (Constants.NS_URI_WSDL11. |
| equals(documentElementNS.getNamespaceURI())) { |
| wsdlToAxisServiceBuilder = new WSDL11ToAllAxisServicesBuilder( |
| new ByteArrayInputStream(out.toByteArray())); |
| ((WSDL11ToAllAxisServicesBuilder)wsdlToAxisServiceBuilder).setCustomWSDLResolver(new WarBasedWSDLLocator(wsdlUrl, |
| loader, |
| new ByteArrayInputStream( |
| out.toByteArray()))); |
| wsdlToAxisServiceBuilder.setCustomResolver( |
| new WarFileBasedURIResolver(loader)); |
| return ((WSDL11ToAllAxisServicesBuilder)wsdlToAxisServiceBuilder).populateAllServices(); |
| } else if (WSDL2Constants.WSDL_NAMESPACE. |
| equals(documentElementNS.getNamespaceURI())){ |
| wsdlToAxisServiceBuilder = new WSDL20ToAllAxisServicesBuilder( |
| new ByteArrayInputStream(out.toByteArray())); |
| ((WSDL20ToAllAxisServicesBuilder)wsdlToAxisServiceBuilder).setCustomWSDLResolver(new WarBasedWSDLLocator(wsdlUrl, |
| loader, |
| new ByteArrayInputStream( |
| out.toByteArray()))); |
| wsdlToAxisServiceBuilder.setCustomResolver( |
| new WarFileBasedURIResolver(loader)); |
| return ((WSDL20ToAllAxisServicesBuilder)wsdlToAxisServiceBuilder).populateAllServices(); |
| } |
| else { |
| throw new DeploymentException(Messages.getMessage("invalidWSDLFound")); |
| } |
| } |
| return null; |
| } |
| |
| public void processFilesInFolder(File folder, HashMap<String, AxisService> servicesMap) |
| throws FileNotFoundException, XMLStreamException, DeploymentException { |
| File files[] = folder.listFiles(); |
| for (int i = 0; i < files.length; i++) { |
| File file1 = files[i]; |
| if (file1.getName().toLowerCase().endsWith(SUFFIX_WSDL)) { |
| InputStream in = new FileInputStream(file1); |
| FileInputStream in2; |
| |
| // now the question is which version of WSDL file this archive contains. |
| // lets check the namespace of the root element and decide. But since we are |
| // using axiom (dude, you are becoming handy here :)), we will not build the |
| // whole thing. |
| OMNamespace documentElementNS = ((OMElement) XMLUtils.toOM(in)).getNamespace(); |
| if (documentElementNS != null) { |
| WSDLToAxisServiceBuilder wsdlToAxisServiceBuilder; |
| if (WSDL2Constants.WSDL_NAMESPACE |
| .equals(documentElementNS.getNamespaceURI())) { |
| // we have a WSDL 2.0 document here. |
| in2 = new FileInputStream(file1); |
| wsdlToAxisServiceBuilder = new WSDL20ToAllAxisServicesBuilder(in2); |
| } else if (Constants.NS_URI_WSDL11. |
| equals(documentElementNS.getNamespaceURI())) { |
| in2 = new FileInputStream(file1); |
| wsdlToAxisServiceBuilder = new WSDL11ToAllAxisServicesBuilder(in2); |
| ((WSDL11ToAxisServiceBuilder) wsdlToAxisServiceBuilder).setDocumentBaseUri(file1.toURI() |
| .toString()); |
| } else { |
| throw new DeploymentException(Messages.getMessage("invalidWSDLFound")); |
| } |
| |
| FileInputStream in3 = new FileInputStream(file1); |
| List<AxisService> services = processWSDLFile(wsdlToAxisServiceBuilder, file1, false, |
| in2, file1.toURI().toString()); |
| |
| if (services != null) { |
| for (int j = 0; j < services.size(); j++) { |
| AxisService axisService = (AxisService) services.get(j); |
| if (axisService != null) { |
| servicesMap.put(axisService.getName(), axisService); |
| } |
| } |
| } |
| try { |
| in2.close(); |
| in3.close(); |
| } catch (IOException e) { |
| log.info(e); |
| } |
| } |
| try { |
| in.close(); |
| } catch (IOException e) { |
| log.info(e); |
| } |
| } |
| } |
| } |
| |
| public void readModuleArchive(DeploymentFileData deploymentFile, |
| AxisModule module, boolean explodedDir, |
| AxisConfiguration axisConfig) |
| throws DeploymentException { |
| |
| // get attribute values |
| boolean moduleXMLFound = false; |
| String shortFileName = DescriptionBuilder.getShortFileName(deploymentFile.getName()); |
| if (!explodedDir) { |
| ZipInputStream zin; |
| FileInputStream fin; |
| try { |
| fin = new FileInputStream(deploymentFile.getAbsolutePath()); |
| zin = new ZipInputStream(fin); |
| ZipEntry entry; |
| while ((entry = zin.getNextEntry()) != null) { |
| if (entry.getName().equalsIgnoreCase(MODULE_XML)) { |
| moduleXMLFound = true; |
| ModuleBuilder builder = new ModuleBuilder(zin, module, axisConfig); |
| // setting module name |
| module.setName(Utils.getModuleName(shortFileName)); |
| module.setVersion(Utils.getModuleVersion(shortFileName)); |
| builder.populateModule(); |
| break; |
| } |
| } |
| zin.close(); |
| fin.close(); |
| if (!moduleXMLFound) { |
| throw new DeploymentException( |
| Messages.getMessage( |
| DeploymentErrorMsgs.MODULE_XML_MISSING, |
| deploymentFile.getAbsolutePath())); |
| } |
| } catch (Exception e) { |
| throw new DeploymentException(e); |
| } |
| } else { |
| File file = new File(deploymentFile.getAbsolutePath(), MODULE_XML); |
| |
| if (file.exists() || |
| (file = new File(deploymentFile.getAbsolutePath(), MODULE_XML.toLowerCase())) |
| .exists()) { |
| InputStream in = null; |
| try { |
| in = new FileInputStream(file); |
| ModuleBuilder builder = new ModuleBuilder(in, module, axisConfig); |
| // setting module name |
| module.setName(Utils.getModuleName(shortFileName)); |
| module.setVersion(Utils.getModuleVersion(shortFileName)); |
| builder.populateModule(); |
| } catch (FileNotFoundException e) { |
| throw new DeploymentException( |
| Messages.getMessage(DeploymentErrorMsgs.FILE_NOT_FOUND, |
| e.getMessage())); |
| } finally { |
| if (in != null) { |
| try { |
| in.close(); |
| } catch (IOException e) { |
| log.info(Messages.getMessage("errorininputstreamclose")); |
| } |
| } |
| } |
| } else { |
| throw new DeploymentException( |
| Messages.getMessage( |
| DeploymentErrorMsgs.MODULE_XML_MISSING, |
| deploymentFile.getAbsolutePath())); |
| } |
| } |
| } |
| } |
