1 /*
2  * Copyright (c) 1997, 2013, Oracle and/or its affiliates. All rights reserved.
3  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
4  *
5  * This code is free software; you can redistribute it and/or modify it
6  * under the terms of the GNU General Public License version 2 only, as
7  * published by the Free Software Foundation.  Oracle designates this
8  * particular file as subject to the "Classpath" exception as provided
9  * by Oracle in the LICENSE file that accompanied this code.
10  *
11  * This code is distributed in the hope that it will be useful, but WITHOUT
12  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
13  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
14  * version 2 for more details (a copy is included in the LICENSE file that
15  * accompanied this code).
16  *
17  * You should have received a copy of the GNU General Public License version
18  * 2 along with this work; if not, write to the Free Software Foundation,
19  * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
20  *
21  * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
22  * or visit www.oracle.com if you need additional information or have any
23  * questions.
24  */
25 
26 package com.sun.tools.internal.ws.wsdl.parser;
27 
28 import com.sun.istack.internal.SAXParseException2;
29 import com.sun.tools.internal.ws.resources.WsdlMessages;
30 import org.xml.sax.Attributes;
31 import org.xml.sax.Locator;
32 import org.xml.sax.SAXException;
33 import org.xml.sax.SAXParseException;
34 import org.xml.sax.helpers.XMLFilterImpl;
35 
36 import java.io.IOException;
37 import java.net.URI;
38 import java.net.URISyntaxException;
39 import java.net.URL;
40 
41 /**
42  * XMLFilter that finds references to other schema files from
43  * SAX events.
44  *
45  * This implementation is a base implementation for typical case
46  * where we just need to look for a particular attribute which
47  * contains an URL to another schema file.
48  *
49  * @author
50  *  Kohsuke Kawaguchi (kohsuke.kawaguchi@sun.com)
51  *  Vivek Pandey
52  */
53 public abstract class AbstractReferenceFinderImpl extends XMLFilterImpl {
54     protected final DOMForest parent;
55 
AbstractReferenceFinderImpl( DOMForest _parent )56     protected AbstractReferenceFinderImpl( DOMForest _parent ) {
57         this.parent = _parent;
58     }
59 
60     /**
61      * IF the given element contains a reference to an external resource,
62      * return its URL.
63      *
64      * @param nsURI
65      *      Namespace URI of the current element
66      * @param localName
67      *      Local name of the current element
68      * @return
69      *      It's OK to return a relative URL.
70      */
findExternalResource( String nsURI, String localName, Attributes atts)71     protected abstract String findExternalResource( String nsURI, String localName, Attributes atts);
72 
73     @Override
startElement(String namespaceURI, String localName, String qName, Attributes atts)74     public void startElement(String namespaceURI, String localName, String qName, Attributes atts)
75         throws SAXException {
76         super.startElement(namespaceURI, localName, qName, atts);
77 
78         String relativeRef = findExternalResource(namespaceURI,localName,atts);
79         if(relativeRef==null)   return; // non found
80 
81         try {
82             // absolutize URL.
83             assert locator != null;
84             String lsi = locator.getSystemId();
85             String ref;
86             if (lsi.startsWith("jar:")) {
87                     int bangIdx = lsi.indexOf('!');
88                     if (bangIdx > 0) {
89                             ref = new URL(new URL(lsi), relativeRef).toString();
90                     } else
91                             ref = relativeRef;
92             } else
93                     ref = new URI(lsi).resolve(new URI(relativeRef)).toString();
94 
95             // then parse this schema as well,
96             // but don't mark this document as a root.
97             parent.parse(ref,false);
98         } catch( URISyntaxException e ) {
99             SAXParseException spe = new SAXParseException2(
100                     WsdlMessages.ABSTRACT_REFERENCE_FINDER_IMPL_UNABLE_TO_PARSE(relativeRef,e.getMessage()),
101                 locator, e );
102 
103             fatalError(spe);
104             throw spe;
105         } catch( IOException e ) {
106             SAXParseException spe = new SAXParseException2(
107                 WsdlMessages.ABSTRACT_REFERENCE_FINDER_IMPL_UNABLE_TO_PARSE(relativeRef,e.getMessage()),
108                 locator, e );
109 
110             fatalError(spe);
111             throw spe;
112         }
113     }
114 
115     private Locator locator;
116 
117     @Override
setDocumentLocator(Locator locator)118     public void setDocumentLocator(Locator locator) {
119         super.setDocumentLocator(locator);
120         this.locator = locator;
121     }
122 }
123