/*

 * Copyright (c) 2005, 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.xml.internal.stream;

import java.util.Hashtable;

import com.sun.org.apache.xerces.internal.impl.msg.XMLMessageFormatter;

import com.sun.org.apache.xerces.internal.util.URI;

import com.sun.org.apache.xerces.internal.util.XMLResourceIdentifierImpl;

import com.sun.org.apache.xerces.internal.xni.parser.XMLComponentManager;

import com.sun.org.apache.xerces.internal.xni.parser.XMLConfigurationException;

import com.sun.org.apache.xerces.internal.impl.XMLEntityManager;

import com.sun.org.apache.xerces.internal.impl.PropertyManager;

import com.sun.org.apache.xerces.internal.impl.XMLErrorReporter;

import com.sun.org.apache.xerces.internal.impl.Constants;

import com.sun.org.apache.xerces.internal.utils.SecuritySupport;

import java.util.Enumeration;

/**

 *

 * @author K.Venugopal SUN Microsystems

 * @author Neeraj Bajaj SUN Microsystems

 * @author Andy Clark, IBM

 *

 */

public class XMLEntityStorage {

    /** Property identifier: error reporter. */

    protected static final String ERROR_REPORTER =

    Constants.XERCES_PROPERTY_PREFIX + Constants.ERROR_REPORTER_PROPERTY;

    /** Feature identifier: warn on duplicate EntityDef */

    protected static final String WARN_ON_DUPLICATE_ENTITYDEF =

    Constants.XERCES_FEATURE_PREFIX +Constants.WARN_ON_DUPLICATE_ENTITYDEF_FEATURE;

    /** warn on duplicate Entity declaration.

     *  http://apache.org/xml/features/warn-on-duplicate-entitydef

     */

    protected boolean fWarnDuplicateEntityDef;

    /** Entities. */

    protected Hashtable fEntities = new Hashtable();

    protected Entity.ScannedEntity fCurrentEntity ;

    private XMLEntityManager fEntityManager;

    /**

     * Error reporter. This property identifier is:

     * http://apache.org/xml/properties/internal/error-reporter

     */

    protected XMLErrorReporter fErrorReporter;

    protected PropertyManager fPropertyManager ;

    /* To keep track whether an entity is declared in external or internal subset*/

    protected boolean fInExternalSubset = false;

    /** Creates a new instance of XMLEntityStorage */

    public XMLEntityStorage(PropertyManager propertyManager) {

        fPropertyManager = propertyManager ;

    }

    /** Creates a new instance of XMLEntityStorage */

    /*public XMLEntityStorage(Entity.ScannedEntity currentEntity) {

        fCurrentEntity = currentEntity ;*/

    public XMLEntityStorage(XMLEntityManager entityManager) {

        fEntityManager = entityManager;

    }

    public void reset(PropertyManager propertyManager){

        fErrorReporter = (XMLErrorReporter)propertyManager.getProperty(Constants.XERCES_PROPERTY_PREFIX + Constants.ERROR_REPORTER_PROPERTY);

        fEntities.clear();

        fCurrentEntity = null;

    }

    public void reset(){

        fEntities.clear();

        fCurrentEntity = null;

    }

    /**

     * Resets the component. The component can query the component manager

     * about any features and properties that affect the operation of the

     * component.

     *

     * @param componentManager The component manager.

     *

     * @throws SAXException Thrown by component on initialization error.

     *                      For example, if a feature or property is

     *                      required for the operation of the component, the

     *                      component manager may throw a

     *                      SAXNotRecognizedException or a

     *                      SAXNotSupportedException.

     */

    public void reset(XMLComponentManager componentManager)

    throws XMLConfigurationException {

        // xerces features

        fWarnDuplicateEntityDef = componentManager.getFeature(WARN_ON_DUPLICATE_ENTITYDEF, false);

        fErrorReporter = (XMLErrorReporter)componentManager.getProperty(ERROR_REPORTER);

        fEntities.clear();

        fCurrentEntity = null;

    } // reset(XMLComponentManager)

    /**

     * Returns entity declaration.

     *

     * @param name The name of the entity.

     *

     * @see SymbolTable

     */

    public Entity getEntity(String name) {

        return (Entity)fEntities.get(name);

    } // getEntity(String)

    public boolean hasEntities() {

            return (fEntities!=null);

    } // getEntity(String)

    public int getEntitySize() {

        return fEntities.size();

    } // getEntity(String)

    public Enumeration getEntityKeys() {

        return fEntities.keys();

    }

    /**

     * Adds an internal entity declaration.

     * <p>

     * <strong>Note:</strong> This method ignores subsequent entity

     * declarations.

     * <p>

     * <strong>Note:</strong> The name should be a unique symbol. The

     * SymbolTable can be used for this purpose.

     *

     * @param name The name of the entity.

     * @param text The text of the entity.

     *

     * @see SymbolTable

     */

    public void addInternalEntity(String name, String text) {

      if (!fEntities.containsKey(name)) {

            Entity entity = new Entity.InternalEntity(name, text, fInExternalSubset);

            fEntities.put(name, entity);

        }

        else{

            if(fWarnDuplicateEntityDef){

                fErrorReporter.reportError(XMLMessageFormatter.XML_DOMAIN,

                "MSG_DUPLICATE_ENTITY_DEFINITION",

                new Object[]{ name },

                XMLErrorReporter.SEVERITY_WARNING );

            }

        }

    } // addInternalEntity(String,String)

    /**

     * Adds an external entity declaration.

     * <p>

     * <strong>Note:</strong> This method ignores subsequent entity

     * declarations.

     * <p>

     * <strong>Note:</strong> The name should be a unique symbol. The

     * SymbolTable can be used for this purpose.

     *

     * @param name         The name of the entity.

     * @param publicId     The public identifier of the entity.

     * @param literalSystemId     The system identifier of the entity.

     * @param baseSystemId The base system identifier of the entity.

     *                     This is the system identifier of the entity

     *                     where <em>the entity being added</em> and

     *                     is used to expand the system identifier when

     *                     the system identifier is a relative URI.

     *                     When null the system identifier of the first

     *                     external entity on the stack is used instead.

     *

     * @see SymbolTable

     */

    public void addExternalEntity(String name,

    String publicId, String literalSystemId,

    String baseSystemId) {

        if (!fEntities.containsKey(name)) {

            if (baseSystemId == null) {

                // search for the first external entity on the stack

                //xxx commenting the 'size' variable..

                /**

                 * int size = fEntityStack.size();

                 * if (size == 0 && fCurrentEntity != null && fCurrentEntity.entityLocation != null) {

                 * baseSystemId = fCurrentEntity.entityLocation.getExpandedSystemId();

                 * }

                 */

                //xxx we need to have information about the current entity.

                if (fCurrentEntity != null && fCurrentEntity.entityLocation != null) {

                    baseSystemId = fCurrentEntity.entityLocation.getExpandedSystemId();

                }

                /**

                 * for (int i = size - 1; i >= 0 ; i--) {

                 * ScannedEntity externalEntity =

                 * (ScannedEntity)fEntityStack.elementAt(i);

                 * if (externalEntity.entityLocation != null && externalEntity.entityLocation.getExpandedSystemId() != null) {

                 * baseSystemId = externalEntity.entityLocation.getExpandedSystemId();

                 * break;

                 * }

                 * }

                 */

            }

            fCurrentEntity = fEntityManager.getCurrentEntity();

            Entity entity = new Entity.ExternalEntity(name,

            new XMLResourceIdentifierImpl(publicId, literalSystemId,

            baseSystemId, expandSystemId(literalSystemId, baseSystemId)),

            null, fInExternalSubset);

            //TODO :: Forced to pass true above remove it.

            //(fCurrentEntity == null) ? fasle : fCurrentEntity.isEntityDeclInExternalSubset());

            //                                  null, fCurrentEntity.isEntityDeclInExternalSubset());

            fEntities.put(name, entity);

        }

        else{

            if(fWarnDuplicateEntityDef){

                fErrorReporter.reportError(XMLMessageFormatter.XML_DOMAIN,

                "MSG_DUPLICATE_ENTITY_DEFINITION",

                new Object[]{ name },

                XMLErrorReporter.SEVERITY_WARNING );

            }

        }

    } // addExternalEntity(String,String,String,String)

    /**

     * Checks whether an entity given by name is external.

     *

     * @param entityName The name of the entity to check.

     * @returns True if the entity is external, false otherwise

     *           (including when the entity is not declared).

     */

    public boolean isExternalEntity(String entityName) {

        Entity entity = (Entity)fEntities.get(entityName);

        if (entity == null) {

            return false;

        }

        return entity.isExternal();

    }

    /**

     * Checks whether the declaration of an entity given by name is

     * // in the external subset.

     *

     * @param entityName The name of the entity to check.

     * @returns True if the entity was declared in the external subset, false otherwise

     *           (including when the entity is not declared).

     */

    public boolean isEntityDeclInExternalSubset(String entityName) {

        Entity entity = (Entity)fEntities.get(entityName);

        if (entity == null) {

            return false;

        }

        return entity.isEntityDeclInExternalSubset();

    }

    /**

     * Adds an unparsed entity declaration.

     * <p>

     * <strong>Note:</strong> This method ignores subsequent entity

     * declarations.

     * <p>

     * <strong>Note:</strong> The name should be a unique symbol. The

     * SymbolTable can be used for this purpose.

     *

     * @param name     The name of the entity.

     * @param publicId The public identifier of the entity.

     * @param systemId The system identifier of the entity.

     * @param notation The name of the notation.

     *

     * @see SymbolTable

     */

    public void addUnparsedEntity(String name,

    String publicId, String systemId,

    String baseSystemId, String notation) {

        fCurrentEntity = fEntityManager.getCurrentEntity();

        if (!fEntities.containsKey(name)) {

            Entity entity = new Entity.ExternalEntity(name, new XMLResourceIdentifierImpl(publicId, systemId, baseSystemId, null), notation, fInExternalSubset);

            //                  (fCurrentEntity == null) ? fasle : fCurrentEntity.isEntityDeclInExternalSubset());

            //                  fCurrentEntity.isEntityDeclInExternalSubset());

            fEntities.put(name, entity);

        }

        else{

            if(fWarnDuplicateEntityDef){

                fErrorReporter.reportError(XMLMessageFormatter.XML_DOMAIN,

                "MSG_DUPLICATE_ENTITY_DEFINITION",

                new Object[]{ name },

                XMLErrorReporter.SEVERITY_WARNING );

            }

        }

    } // addUnparsedEntity(String,String,String,String)

    /**

     * Checks whether an entity given by name is unparsed.

     *

     * @param entityName The name of the entity to check.

     * @returns True if the entity is unparsed, false otherwise

     *          (including when the entity is not declared).

     */

    public boolean isUnparsedEntity(String entityName) {

        Entity entity = (Entity)fEntities.get(entityName);

        if (entity == null) {

            return false;

        }

        return entity.isUnparsed();

    }

    /**

     * Checks whether an entity given by name is declared.

     *

     * @param entityName The name of the entity to check.

     * @returns True if the entity is declared, false otherwise.

     */

    public boolean isDeclaredEntity(String entityName) {

        Entity entity = (Entity)fEntities.get(entityName);

        return entity != null;

    }

    /**

     * Expands a system id and returns the system id as a URI, if

     * it can be expanded. A return value of null means that the

     * identifier is already expanded. An exception thrown

     * indicates a failure to expand the id.

     *

     * @param systemId The systemId to be expanded.

     *

     * @return Returns the URI string representing the expanded system

     *         identifier. A null value indicates that the given

     *         system identifier is already expanded.

     *

     */

    public static String expandSystemId(String systemId) {

        return expandSystemId(systemId, null);

    } // expandSystemId(String):String

    // current value of the "user.dir" property

    private static String gUserDir;

    // escaped value of the current "user.dir" property

    private static String gEscapedUserDir;

    // which ASCII characters need to be escaped

    private static boolean gNeedEscaping[] = new boolean[128];

    // the first hex character if a character needs to be escaped

    private static char gAfterEscaping1[] = new char[128];

    // the second hex character if a character needs to be escaped

    private static char gAfterEscaping2[] = new char[128];

    private static char[] gHexChs = {'0', '1', '2', '3', '4', '5', '6', '7',

    '8', '9', 'A', 'B', 'C', 'D', 'E', 'F'};

    // initialize the above 3 arrays

    static {

        for (int i = 0; i <= 0x1f; i++) {

            gNeedEscaping[i] = true;

            gAfterEscaping1[i] = gHexChs[i >> 4];

            gAfterEscaping2[i] = gHexChs[i & 0xf];

        }

        gNeedEscaping[0x7f] = true;

        gAfterEscaping1[0x7f] = '7';

        gAfterEscaping2[0x7f] = 'F';

        char[] escChs = {' ', '<', '>', '#', '%', '"', '{', '}',

        '|', '\\', '^', '~', '[', ']', '`'};

        int len = escChs.length;

        char ch;

        for (int i = 0; i < len; i++) {

            ch = escChs[i];

            gNeedEscaping[ch] = true;

            gAfterEscaping1[ch] = gHexChs[ch >> 4];

            gAfterEscaping2[ch] = gHexChs[ch & 0xf];

        }

    }

    // To escape the "user.dir" system property, by using %HH to represent

    // special ASCII characters: 0x00~0x1F, 0x7F, ' ', '<', '>', '#', '%'

    // and '"'. It's a static method, so needs to be synchronized.

    // this method looks heavy, but since the system property isn't expected

    // to change often, so in most cases, we only need to return the string

    // that was escaped before.

    // According to the URI spec, non-ASCII characters (whose value >= 128)

    // need to be escaped too.

    // REVISIT: don't know how to escape non-ASCII characters, especially

    // which encoding to use. Leave them for now.

    private static synchronized String getUserDir() {

        // get the user.dir property

        String userDir = "";

        try {

            userDir = SecuritySupport.getSystemProperty("user.dir");

        }

        catch (SecurityException se) {

        }

        // return empty string if property value is empty string.

        if (userDir.length() == 0)

            return "";

        // compute the new escaped value if the new property value doesn't

        // match the previous one

        if (userDir.equals(gUserDir)) {

            return gEscapedUserDir;

        }

        // record the new value as the global property value

        gUserDir = userDir;

        char separator = java.io.File.separatorChar;

        userDir = userDir.replace(separator, '/');

        int len = userDir.length(), ch;

        StringBuffer buffer = new StringBuffer(len*3);

        // change C:/blah to /C:/blah

        if (len >= 2 && userDir.charAt(1) == ':') {

            ch = Character.toUpperCase(userDir.charAt(0));

            if (ch >= 'A' && ch <= 'Z') {

                buffer.append('/');

            }

        }

        // for each character in the path

        int i = 0;

        for (; i < len; i++) {

            ch = userDir.charAt(i);

            // if it's not an ASCII character, break here, and use UTF-8 encoding

            if (ch >= 128)

                break;

            if (gNeedEscaping[ch]) {

                buffer.append('%');

                buffer.append(gAfterEscaping1[ch]);

                buffer.append(gAfterEscaping2[ch]);

                // record the fact that it's escaped

            }

            else {

                buffer.append((char)ch);

            }

        }

        // we saw some non-ascii character

        if (i < len) {

            // get UTF-8 bytes for the remaining sub-string

            byte[] bytes = null;

            byte b;

            try {

                bytes = userDir.substring(i).getBytes("UTF-8");

            } catch (java.io.UnsupportedEncodingException e) {

                // should never happen

                return userDir;

            }

            len = bytes.length;

            // for each byte

            for (i = 0; i < len; i++) {

                b = bytes[i];

                // for non-ascii character: make it positive, then escape

                if (b < 0) {

                    ch = b + 256;

                    buffer.append('%');

                    buffer.append(gHexChs[ch >> 4]);

                    buffer.append(gHexChs[ch & 0xf]);

                }

                else if (gNeedEscaping[b]) {

                    buffer.append('%');

                    buffer.append(gAfterEscaping1[b]);

                    buffer.append(gAfterEscaping2[b]);

                }

                else {

                    buffer.append((char)b);

                }

            }

        }

        // change blah/blah to blah/blah/

        if (!userDir.endsWith("/"))

            buffer.append('/');

        gEscapedUserDir = buffer.toString();

        return gEscapedUserDir;

    }

    /**

     * Expands a system id and returns the system id as a URI, if

     * it can be expanded. A return value of null means that the

     * identifier is already expanded. An exception thrown

     * indicates a failure to expand the id.

     *

     * @param systemId The systemId to be expanded.

     *

     * @return Returns the URI string representing the expanded system

     *         identifier. A null value indicates that the given

     *         system identifier is already expanded.

     *

     */

    public static String expandSystemId(String systemId, String baseSystemId) {

        // check for bad parameters id

        if (systemId == null || systemId.length() == 0) {