Merge topic/xxe into develop

[mikutoga/Pmd2XML.git] / src / main / java / jp / sfjp / mikutoga / pmd2xml / XmlInputUtil.java

diff --git a/src/main/java/jp/sfjp/mikutoga/pmd2xml/XmlInputUtil.java b/src/main/java/jp/sfjp/mikutoga/pmd2xml/XmlInputUtil.java
index 0772100..e1a87b8 100644 (file)
--- a/src/main/java/jp/sfjp/mikutoga/pmd2xml/XmlInputUtil.java
+++ b/src/main/java/jp/sfjp/mikutoga/pmd2xml/XmlInputUtil.java
@@ -7,6 +7,13 @@
 
 package jp.sfjp.mikutoga.pmd2xml;
 
+import java.io.BufferedInputStream;
+import java.io.File;
+import java.io.IOException;
+import java.io.InputStream;
+import java.net.MalformedURLException;
+import java.net.URI;
+import java.net.URL;
 import javax.xml.parsers.ParserConfigurationException;
 import javax.xml.parsers.SAXParser;
 import javax.xml.parsers.SAXParserFactory;
@@ -14,16 +21,16 @@ import javax.xml.validation.Schema;
 import jp.sfjp.mikutoga.pmd.model.xml.Schema101009;
 import jp.sfjp.mikutoga.pmd.model.xml.Schema130128;
 import jp.sfjp.mikutoga.xml.BotherHandler;
-import jp.sfjp.mikutoga.xml.LocalXmlResource;
+import jp.sfjp.mikutoga.xml.NoopEntityResolver;
 import jp.sfjp.mikutoga.xml.SchemaUtil;
-import jp.sfjp.mikutoga.xml.XmlResourceResolver;
+import org.xml.sax.InputSource;
 import org.xml.sax.SAXException;
 import org.xml.sax.XMLReader;
 
 /**
  * XML入力に関する各種ユーティリティ。
  */
-class XmlInputUtil {
+final class XmlInputUtil {
 
     /**
      * 隠しコンストラクタ。
@@ -35,6 +42,60 @@ class XmlInputUtil {
 
 
     /**
+     * 実在ファイルからXML入力ソースを得る。
+     * @param file 実在ファイル
+     * @return XML入力ソース
+     */
+    static InputSource fileToSource(File file){
+        assert file.exists();
+
+        URI uri = file.toURI();
+
+        URL url;
+        try{
+            url = uri.toURL();
+        }catch(MalformedURLException e){
+            // 実在File由来のURLでは起こりえない
+            assert false;
+            throw new AssertionError(e);
+        }
+
+        String systemId = url.toString();
+
+        InputSource source = new InputSource(systemId);
+
+        return source;
+    }
+
+    /**
+     * InputSourceからInputStreamを得る。
+     * <p>入力ソースには、少なくともバイトストリームか
+     * URL文字列(SystemId)のいずれかが設定されていなければならない。
+     * @param source 入力ソース
+     * @return 入力バイトストリーム
+     * @throws IllegalArgumentException 入力ソースの設定が足りない。
+     * @throws IOException 入力ソースにアクセス不能。
+     */
+    static InputStream openInputSource(InputSource source)
+            throws IllegalArgumentException, IOException{
+        InputStream is;
+
+        is = source.getByteStream();
+
+        if(is == null){
+            String systemId = source.getSystemId();
+            if(systemId == null) throw new IllegalArgumentException();
+
+            URL url = new URL(systemId);
+            is = url.openStream();
+        }
+
+        is = new BufferedInputStream(is);
+
+        return is;
+    }
+
+    /**
      * SAXパーサファクトリを生成する。
      * <ul>
      * <li>XML名前空間機能は有効になる。
@@ -68,10 +129,7 @@ class XmlInputUtil {
         SAXParser parser;
         try{
             parser = factory.newSAXParser();
-        }catch(ParserConfigurationException e){
-            assert false;
-            throw new AssertionError(e);
-        }catch(SAXException e){
+        }catch(ParserConfigurationException | SAXException e){
             assert false;
             throw new AssertionError(e);
         }
@@ -87,31 +145,36 @@ class XmlInputUtil {
      * @param xmlInType 入力XML種別
      * @return スキーマ
      */
-    private static Schema builsSchema(XmlResourceResolver resolver,
-                                        ModelFileType xmlInType ){
-        LocalXmlResource[] schemaArray;
+    private static Schema buildSchema(ModelFileType xmlInType ){
+        URI[] schemaUris;
         switch(xmlInType){
         case XML_101009:
-            schemaArray = new LocalXmlResource[]{
-                Schema101009.SINGLETON,
+            schemaUris = new URI[]{
+                Schema101009.RES_SCHEMA_PMDXML,
             };
             break;
         case XML_130128:
-            schemaArray = new LocalXmlResource[]{
-                Schema130128.SINGLETON,
+            schemaUris = new URI[]{
+                Schema130128.RES_SCHEMA_PMDXML,
             };
             break;
         case XML_AUTO:
-            schemaArray = new LocalXmlResource[]{
-                Schema101009.SINGLETON,
-                Schema130128.SINGLETON,
+            schemaUris = new URI[]{
+                Schema101009.RES_SCHEMA_PMDXML,
+                Schema130128.RES_SCHEMA_PMDXML,
             };
             break;
         default:
             throw new IllegalStateException();
         }
 
-        Schema schema = SchemaUtil.newSchema(resolver, schemaArray);
+        Schema schema;
+        try{
+            schema = SchemaUtil.newSchema(schemaUris);
+        }catch(IOException | SAXException e){
+            assert false;
+            throw new AssertionError(e);
+        }
 
         return schema;
     }
@@ -123,9 +186,7 @@ class XmlInputUtil {
      * @return XMLリーダ
      */
     static XMLReader buildReader(ModelFileType xmlInType){
-        XmlResourceResolver resolver = new XmlResourceResolver();
-
-        Schema schema = builsSchema(resolver, xmlInType);
+        Schema schema = buildSchema(xmlInType);
 
         SAXParser parser = buildParser(schema);
 
@@ -137,7 +198,7 @@ class XmlInputUtil {
             throw new AssertionError(e);
         }
 
-        reader.setEntityResolver(resolver);
+        reader.setEntityResolver(NoopEntityResolver.NOOP_RESOLVER);
         reader.setErrorHandler(BotherHandler.HANDLER);
 
         return reader;