/*

 * Copyright (c) 2007, 2019, Oracle and/or its affiliates. All rights reserved.

 */

/*

 * 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 com.sun.org.apache.xalan.internal.xsltc.trax;

import com.sun.org.apache.xalan.internal.XalanConstants;

import com.sun.org.apache.xalan.internal.utils.FeaturePropertyBase;

import com.sun.org.apache.xalan.internal.utils.ObjectFactory;

import com.sun.org.apache.xalan.internal.utils.XMLSecurityManager;

import com.sun.org.apache.xalan.internal.utils.XMLSecurityPropertyManager.Property;

import com.sun.org.apache.xalan.internal.utils.XMLSecurityPropertyManager;

import com.sun.org.apache.xalan.internal.xsltc.compiler.Constants;

import com.sun.org.apache.xalan.internal.xsltc.compiler.SourceLoader;

import com.sun.org.apache.xalan.internal.xsltc.compiler.XSLTC;

import com.sun.org.apache.xalan.internal.xsltc.compiler.util.ErrorMsg;

import com.sun.org.apache.xalan.internal.xsltc.dom.XSLTCDTMManager;

import com.sun.org.apache.xml.internal.utils.StopParseException;

import com.sun.org.apache.xml.internal.utils.StylesheetPIHandler;

import java.io.File;

import java.io.FileInputStream;

import java.io.FileNotFoundException;

import java.io.FilenameFilter;

import java.io.IOException;

import java.io.InputStream;

import java.net.MalformedURLException;

import java.net.URL;

import java.util.ArrayList;

import java.util.Enumeration;

import java.util.List;

import java.util.Map;

import java.util.Properties;

import java.util.zip.ZipEntry;

import java.util.zip.ZipFile;

import javax.xml.XMLConstants;

import javax.xml.catalog.CatalogException;

import javax.xml.catalog.CatalogFeatures.Feature;

import javax.xml.catalog.CatalogFeatures;

import javax.xml.catalog.CatalogManager;

import javax.xml.catalog.CatalogResolver;

import javax.xml.transform.ErrorListener;

import javax.xml.transform.Source;

import javax.xml.transform.Templates;

import javax.xml.transform.Transformer;

import javax.xml.transform.TransformerConfigurationException;

import javax.xml.transform.TransformerException;

import javax.xml.transform.URIResolver;

import javax.xml.transform.dom.DOMResult;

import javax.xml.transform.dom.DOMSource;

import javax.xml.transform.sax.SAXResult;

import javax.xml.transform.sax.SAXSource;

import javax.xml.transform.sax.SAXTransformerFactory;

import javax.xml.transform.sax.TemplatesHandler;

import javax.xml.transform.sax.TransformerHandler;

import javax.xml.transform.stax.*;

import javax.xml.transform.stream.StreamResult;

import javax.xml.transform.stream.StreamSource;

import jdk.xml.internal.JdkXmlFeatures;

import jdk.xml.internal.JdkXmlUtils;

import jdk.xml.internal.SecuritySupport;

import jdk.xml.internal.TransformErrorListener;

import org.xml.sax.InputSource;

import org.xml.sax.SAXException;

import org.xml.sax.XMLFilter;

import org.xml.sax.XMLReader;

/**

 * Implementation of a JAXP TransformerFactory for Translets.

 * @author G. Todd Miller

 * @author Morten Jorgensen

 * @author Santiago Pericas-Geertsen

 * @LastModified: Aug 2019

 */

public class TransformerFactoryImpl

    extends SAXTransformerFactory implements SourceLoader

{

    // Public constants for attributes supported by the XSLTC TransformerFactory.

    public final static String TRANSLET_NAME = "translet-name";

    public final static String DESTINATION_DIRECTORY = "destination-directory";

    public final static String PACKAGE_NAME = "package-name";

    public final static String JAR_NAME = "jar-name";

    public final static String GENERATE_TRANSLET = "generate-translet";

    public final static String AUTO_TRANSLET = "auto-translet";

    public final static String USE_CLASSPATH = "use-classpath";

    public final static String DEBUG = "debug";

    public final static String ENABLE_INLINING = "enable-inlining";

    public final static String INDENT_NUMBER = "indent-number";

    /**

     * Default error listener

     */

    private final ErrorListener _defaultListener = new TransformErrorListener();

    /**

     * This error listener is used only for this factory and is not passed to

     * the Templates or Transformer objects that we create.

     */

    private ErrorListener _errorListener = _defaultListener;

    // flag indicating whether there's an user's ErrorListener

    private boolean _hasUserErrListener;

    /**

     * This URIResolver is passed to all created Templates and Transformers

     */

    private URIResolver _uriResolver = null;

    /**

     * As Gregor Samsa awoke one morning from uneasy dreams he found himself

     * transformed in his bed into a gigantic insect. He was lying on his hard,

     * as it were armour plated, back, and if he lifted his head a little he

     * could see his big, brown belly divided into stiff, arched segments, on

     * top of which the bed quilt could hardly keep in position and was about

     * to slide off completely. His numerous legs, which were pitifully thin

     * compared to the rest of his bulk, waved helplessly before his eyes.

     * "What has happened to me?", he thought. It was no dream....

     */

    protected final static String DEFAULT_TRANSLET_NAME = "GregorSamsa";

    /**

     * The class name of the translet

     */

    private String _transletName = DEFAULT_TRANSLET_NAME;

    /**

     * The destination directory for the translet

     */

    private String _destinationDirectory = null;

    /**

     * The package name prefix for all generated translet classes

     */

    private static final String DEFAULT_TRANSLATE_PACKAGE = "die.verwandlung";

    private String _packageName = DEFAULT_TRANSLATE_PACKAGE;

    /**

     * The jar file name which the translet classes are packaged into

     */

    private String _jarFileName = null;

    /**

     * This Map is used to store parameters for locating

     * <?xml-stylesheet ...?> processing instructions in XML docs.

     */

    private Map<Source, PIParamWrapper> _piParams = null;

    /**

     * The above Map stores objects of this class.

     */

    private static class PIParamWrapper {

        public String _media = null;

        public String _title = null;

        public String _charset = null;

        public PIParamWrapper(String media, String title, String charset) {

            _media = media;

            _title = title;

            _charset = charset;

        }

    }

    /**

     * Set to <code>true</code> when debugging is enabled.

     */

    private boolean _debug = false;

    /**

     * Set to <code>true</code> when templates are inlined.

     */

    private boolean _enableInlining = false;

    /**

     * Set to <code>true</code> when we want to generate

     * translet classes from the stylesheet.

     */

    private boolean _generateTranslet = false;

    /**

     * If this is set to <code>true</code>, we attempt to use translet classes

     * for transformation if possible without compiling the stylesheet. The

     * translet class is only used if its timestamp is newer than the timestamp

     * of the stylesheet.

     */

    private boolean _autoTranslet = false;

    /**

     * If this is set to <code>true</code>, we attempt to load the translet

     * from the CLASSPATH.

     */

    private boolean _useClasspath = false;

    /**

     * Number of indent spaces when indentation is turned on.

     */

    private int _indentNumber = -1;

    /**

     * <p>State of secure processing feature.</p>

     */

    private boolean _isNotSecureProcessing = true;

    /**

     * <p>State of secure mode.</p>

     */

    private boolean _isSecureMode = false;

    /**

     * Indicates whether 3rd party parser may be used to override the system-default

     * Note the default value (false) is the safe option.

     * Note same as the old property useServicesMechanism

     */

    private boolean _overrideDefaultParser;

    /**

     * protocols allowed for external references set by the stylesheet

     * processing instruction, Import and Include element.

     */

    private String _accessExternalStylesheet = XalanConstants.EXTERNAL_ACCESS_DEFAULT;

     /**

     * protocols allowed for external DTD references in source file and/or stylesheet.

     */

    private String _accessExternalDTD = XalanConstants.EXTERNAL_ACCESS_DEFAULT;

    private XMLSecurityPropertyManager _xmlSecurityPropertyMgr;

    private XMLSecurityManager _xmlSecurityManager;

    private final JdkXmlFeatures _xmlFeatures;

    private ClassLoader _extensionClassLoader = null;

    // Unmodifiable view of external extension function from xslt compiler

    // It will be populated by user-specified extension functions during the

    // type checking

    private Map<String, Class<?>> _xsltcExtensionFunctions;

    CatalogResolver _catalogUriResolver;

    CatalogFeatures _catalogFeatures;

    CatalogFeatures.Builder cfBuilder = CatalogFeatures.builder();

    // Catalog features

    String _catalogFiles = null;

    String _catalogDefer = null;

    String _catalogPrefer = null;

    String _catalogResolve = null;

    int _cdataChunkSize = JdkXmlUtils.CDATA_CHUNK_SIZE_DEFAULT;

    /**

     * javax.xml.transform.sax.TransformerFactory implementation.

     */

    public TransformerFactoryImpl() {

        if (System.getSecurityManager() != null) {

            _isSecureMode = true;

            _isNotSecureProcessing = false;

        }

        _xmlFeatures = new JdkXmlFeatures(!_isNotSecureProcessing);

        _overrideDefaultParser = _xmlFeatures.getFeature(

                JdkXmlFeatures.XmlFeature.JDK_OVERRIDE_PARSER);

        _xmlSecurityPropertyMgr = new XMLSecurityPropertyManager();

        _accessExternalDTD = _xmlSecurityPropertyMgr.getValue(

                Property.ACCESS_EXTERNAL_DTD);

        _accessExternalStylesheet = _xmlSecurityPropertyMgr.getValue(

                Property.ACCESS_EXTERNAL_STYLESHEET);

        //Parser's security manager

        _xmlSecurityManager = new XMLSecurityManager(true);

        //Unmodifiable hash map with loaded external extension functions

        _xsltcExtensionFunctions = null;

    }

    public Map<String, Class<?>> getExternalExtensionsMap() {

        return _xsltcExtensionFunctions;

    }

    /**

     * javax.xml.transform.sax.TransformerFactory implementation.

     * Set the error event listener for the TransformerFactory, which is used

     * for the processing of transformation instructions, and not for the

     * transformation itself.

     *

     * @param listener The error listener to use with the TransformerFactory

     * @throws IllegalArgumentException

     */

    @Override

    public void setErrorListener(ErrorListener listener)

        throws IllegalArgumentException

    {

        if (listener == null) {

            ErrorMsg err = new ErrorMsg(ErrorMsg.ERROR_LISTENER_NULL_ERR,

                                        "TransformerFactory");

            throw new IllegalArgumentException(err.toString());

        }

        _hasUserErrListener = true;

        _errorListener = listener;

    }

    /**

     * javax.xml.transform.sax.TransformerFactory implementation.

     * Get the error event handler for the TransformerFactory.

     *

     * @return The error listener used with the TransformerFactory

     */

    @Override

    public ErrorListener getErrorListener() {

        return _errorListener;

    }

    /**

     * Returns the package name.

     */

    String getPackageName() {

        return _packageName;

    }

    /**

     * javax.xml.transform.sax.TransformerFactory implementation.

     * Returns the value set for a TransformerFactory attribute

     *

     * @param name The attribute name

     * @return An object representing the attribute value

     * @throws IllegalArgumentException

     */

    @Override

    public Object getAttribute(String name)

        throws IllegalArgumentException

    {

        // Return value for attribute 'translet-name'

        if (name.equals(TRANSLET_NAME)) {

            return _transletName;

        }

        else if (name.equals(GENERATE_TRANSLET)) {

            return _generateTranslet;

        }

        else if (name.equals(AUTO_TRANSLET)) {

            return _autoTranslet;

        }

        else if (name.equals(ENABLE_INLINING)) {

            if (_enableInlining)

              return Boolean.TRUE;

            else

              return Boolean.FALSE;

        } else if (name.equals(XalanConstants.SECURITY_MANAGER)) {

            return _xmlSecurityManager;

        } else if (name.equals(XalanConstants.JDK_EXTENSION_CLASSLOADER)) {

           return _extensionClassLoader;

        } else if (JdkXmlUtils.CATALOG_FILES.equals(name)) {

            return _catalogFiles;

        } else if (JdkXmlUtils.CATALOG_DEFER.equals(name)) {

            return _catalogDefer;

        } else if (JdkXmlUtils.CATALOG_PREFER.equals(name)) {

            return _catalogPrefer;

        } else if (JdkXmlUtils.CATALOG_RESOLVE.equals(name)) {

            return _catalogResolve;

        } else if (JdkXmlFeatures.CATALOG_FEATURES.equals(name)) {

            return buildCatalogFeatures();

        } else if (JdkXmlUtils.CDATA_CHUNK_SIZE.equals(name)) {

            return _cdataChunkSize;

        }

        /** Check to see if the property is managed by the security manager **/

        String propertyValue = (_xmlSecurityManager != null) ?

                _xmlSecurityManager.getLimitAsString(name) : null;

        if (propertyValue != null) {

            return propertyValue;

        } else {

            propertyValue = (_xmlSecurityPropertyMgr != null) ?

                _xmlSecurityPropertyMgr.getValue(name) : null;

            if (propertyValue != null) {

                return propertyValue;

            }

        }

        // Throw an exception for all other attributes

        ErrorMsg err = new ErrorMsg(ErrorMsg.JAXP_INVALID_ATTR_ERR, name);

        throw new IllegalArgumentException(err.toString());

    }

    /**

     * javax.xml.transform.sax.TransformerFactory implementation.

     * Sets the value for a TransformerFactory attribute.

     *

     * @param name The attribute name

     * @param value An object representing the attribute value

     * @throws IllegalArgumentException

     */

    @Override

    public void setAttribute(String name, Object value)

        throws IllegalArgumentException

    {

        // Set the default translet name (ie. class name), which will be used

        // for translets that cannot be given a name from their system-id.

        if (name.equals(TRANSLET_NAME) && value instanceof String) {

            _transletName = (String) value;

            return;

        }

        else if (name.equals(DESTINATION_DIRECTORY) && value instanceof String) {

            _destinationDirectory = (String) value;

            return;

        }

        else if (name.equals(PACKAGE_NAME) && value instanceof String) {

            _packageName = (String) value;

            return;

        }

        else if (name.equals(JAR_NAME) && value instanceof String) {

            _jarFileName = (String) value;

            return;

        }

        else if (name.equals(GENERATE_TRANSLET)) {

            if (value instanceof Boolean) {

                _generateTranslet = ((Boolean) value);

                return;

            }

            else if (value instanceof String) {

                _generateTranslet = ((String) value).equalsIgnoreCase("true");

                return;

            }

        }

        else if (name.equals(AUTO_TRANSLET)) {

            if (value instanceof Boolean) {

                _autoTranslet = ((Boolean) value);

                return;

            }

            else if (value instanceof String) {

                _autoTranslet = ((String) value).equalsIgnoreCase("true");

                return;

            }

        }

        else if (name.equals(USE_CLASSPATH)) {

            if (value instanceof Boolean) {

                _useClasspath = ((Boolean) value);

                return;

            }

            else if (value instanceof String) {

                _useClasspath = ((String) value).equalsIgnoreCase("true");

                return;

            }

        }

        else if (name.equals(DEBUG)) {

            if (value instanceof Boolean) {

                _debug = ((Boolean) value);

                return;

            }

            else if (value instanceof String) {

                _debug = ((String) value).equalsIgnoreCase("true");

                return;

            }

        }

        else if (name.equals(ENABLE_INLINING)) {

            if (value instanceof Boolean) {

                _enableInlining = ((Boolean) value);

                return;

            }

            else if (value instanceof String) {

                _enableInlining = ((String) value).equalsIgnoreCase("true");

                return;

            }

        }

        else if (name.equals(INDENT_NUMBER)) {

            if (value instanceof String) {

                try {

                    _indentNumber = Integer.parseInt((String) value);

                    return;

                }

                catch (NumberFormatException e) {

                    // Falls through

                }

            }

            else if (value instanceof Integer) {

                _indentNumber = ((Integer) value);

                return;

            }

        }

        else if ( name.equals(XalanConstants.JDK_EXTENSION_CLASSLOADER)) {

            if (value instanceof ClassLoader) {

                _extensionClassLoader = (ClassLoader) value;

                return;

            } else {

                final ErrorMsg err

                    = new ErrorMsg(ErrorMsg.JAXP_INVALID_ATTR_VALUE_ERR, "Extension Functions ClassLoader");

                throw new IllegalArgumentException(err.toString());

            }

        } else if (JdkXmlUtils.CATALOG_FILES.equals(name)) {

            _catalogFiles = (String) value;

            cfBuilder = CatalogFeatures.builder().with(Feature.FILES, _catalogFiles);

            return;

        } else if (JdkXmlUtils.CATALOG_DEFER.equals(name)) {

            _catalogDefer = (String) value;

            cfBuilder = CatalogFeatures.builder().with(Feature.DEFER, _catalogDefer);

            return;

        } else if (JdkXmlUtils.CATALOG_PREFER.equals(name)) {

            _catalogPrefer = (String) value;

            cfBuilder = CatalogFeatures.builder().with(Feature.PREFER, _catalogPrefer);