XMLConfiguration.java

/*
 * Copyright (c) 2006 jDiameter.
 * https://jdiameter.dev.java.net/
 *
 * License: GPL v3
 *
 * e-mail: erick.svenson@yahoo.com
 *
 */
package org.jdiameter.server.impl.helpers;

import org.jdiameter.api.Configuration;
import org.jdiameter.client.impl.helpers.AppConfiguration;
import static org.jdiameter.server.impl.helpers.Parameters.*;
import org.w3c.dom.Document;
import org.w3c.dom.Element;
import org.w3c.dom.Node;
import org.w3c.dom.NodeList;

import javax.xml.XMLConstants;
import javax.xml.parsers.DocumentBuilder;
import javax.xml.parsers.DocumentBuilderFactory;
import javax.xml.transform.Source;
import javax.xml.transform.dom.DOMSource;
import javax.xml.transform.stream.StreamSource;
import javax.xml.validation.Schema;
import javax.xml.validation.SchemaFactory;
import javax.xml.validation.Validator;
import java.io.File;
import java.io.InputStream;
import java.util.ArrayList;
import java.util.Hashtable;

public class XMLConfiguration extends EmptyConfiguration {

    /**
     * Create instance of class and load file from defined input stream
     * @param in input stream
     * @throws Exception
     */
    public XMLConfiguration(InputStream in) throws Exception {
        this(in, null, null, false);
    }

    /**
     * Create instance of class and load file from defined input stream
     * @param in input stream
     * @param attributes attributes for DocumentBuilderFactory
     * @param  features features for DocumentBuilderFactory
     * @throws Exception
     */
    public XMLConfiguration(InputStream in,  Hashtable<String,Object> attributes, Hashtable<String,Boolean> features) throws Exception {
        this(in, attributes, features, false);
    }

    /**
     * Create instance of class and load file from defined  file name
     * @param filename configuration file name
     * @throws Exception
     */
    public XMLConfiguration(String filename) throws Exception {
        this(filename, null, null, false);
    }

    /**
     * Create instance of class and load file from defined input stream
     * @param filename configuration file name
     * @param attributes attributes for DocumentBuilderFactory
     * @param  features features for DocumentBuilderFactory
     * @throws Exception
     */

    public XMLConfiguration(String filename, Hashtable<String,Object> attributes, Hashtable<String,Boolean> features) throws Exception {
        this(filename, attributes, features, false);
    }

    protected XMLConfiguration(Object in, Hashtable<String,Object> attributes, Hashtable<String,Boolean> features, boolean nop) throws Exception {
        DocumentBuilderFactory factory = DocumentBuilderFactory.newInstance();
        factory.setNamespaceAware(true);
        if (attributes != null)
            for (String key : attributes.keySet() )
                factory.setAttribute(key, attributes.get(key));
        if (features != null)
            for (String key : features.keySet() )
                factory.setFeature(key, features.get(key));

        DocumentBuilder builder = factory.newDocumentBuilder();
        Document document;
       
        if (in instanceof InputStream)
            document = builder.parse((InputStream)in);
        else if (in instanceof String)
            document = builder.parse(new File((String)in));
        else
            throw  new Exception("Unknown type of input data");
        validate(document);
        processing(document);
    }

    protected void validate(Document document) throws Exception {
        SchemaFactory factory = SchemaFactory.newInstance(XMLConstants.W3C_XML_SCHEMA_NS_URI);
        Source schemaFile = new StreamSource(getClass().getResourceAsStream("/META-INF/jdiameter-server.xsd"));
        Schema schema = factory.newSchema(schemaFile);
        Validator validator = schema.newValidator();
        validator.validate(new DOMSource(document));
    }

    protected void processing(Document document) {
        Element element = document.getDocumentElement();
        NodeList c = element.getChildNodes();
        for (int i = 0; i < c.getLength(); i++) {
            String nodeName = c.item(i).getNodeName();
            if (nodeName.equals("LocalPeer")) addLocalPeer(c.item(i));
            else
            if (nodeName.equals("Parameters")) addParameters(c.item(i));
            else
            if (nodeName.equals("Network")) addNetwork(c.item(i));
            else
            if (nodeName.equals("Extensions")) addExtensions(c.item(i));
        }
    }



    protected void addApplications(Node node) {
        NodeList c = node.getChildNodes();
        ArrayList<Configuration> items = new ArrayList<Configuration>();
        for (int i = 0; i < c.getLength(); i++) {
            String nodeName = c.item(i).getNodeName();
            if (nodeName.equals("ApplicationID")) {
                Configuration m = addApplicationID(c.item(i));
                if (m!=null) items.add(m);
            }
        }
        add(ApplicationId, items.toArray(EMPTY_ARRAY));
    }

     protected Configuration addApplicationID(NodeList node) {
        for (int i = 0; i < node.getLength(); i++) {
           String nodeName = node.item(i).getNodeName();
            if (nodeName.equals("ApplicationID"))
                return addApplicationID(node.item(i));
        }
        return null;
     }

    protected Configuration addApplicationID(Node node) {
        NodeList c = node.getChildNodes();
        AppConfiguration e = getInstance();
        for (int i = 0; i < c.getLength(); i++) {
            String nodeName = c.item(i).getNodeName();
            if (nodeName.equals("VendorId"))   e.add(VendorId,   getLongValue(c.item(i)));
            else
            if (nodeName.equals("AuthApplId")) e.add(AuthApplId, getLongValue(c.item(i)));
            else
            if (nodeName.equals("AcctApplId")) e.add(AcctApplId, getLongValue(c.item(i)));
        }
        return e;
    }

    protected void addParameters(Node node) {
        NodeList c = node.getChildNodes();
        for (int i = 0; i < c.getLength(); i++) {
            String nodeName = c.item(i).getNodeName();
            if (nodeName.equals("UseUriAsFqdn")) add(UseUriAsFqdn, Boolean.valueOf(getValue(c.item(i))));
            if (nodeName.equals("QueueSize")) add(QueueSize, getIntValue(c.item(i)));
            if (nodeName.equals("MessageTimeOut")) add(MessageTimeOut, getLongValue(c.item(i)));
            if (nodeName.equals("StopTimeOut")) add(StopTimeOut, getLongValue(c.item(i)));
            if (nodeName.equals("CeaTimeOut")) add(CeaTimeOut, getLongValue(c.item(i)));
            if (nodeName.equals("IacTimeOut")) add(IacTimeOut, getLongValue(c.item(i)));
            if (nodeName.equals("DwaTimeOut")) add(DwaTimeOut, getLongValue(c.item(i)));
            if (nodeName.equals("DpaTimeOut")) add(DpaTimeOut, getLongValue(c.item(i)));
            if (nodeName.equals("RecTimeOut")) add(RecTimeOut, getLongValue(c.item(i)));
            appendOtherParameter(c.item(i));
        }
    }

    protected void addNetwork(Node node) {
        NodeList c = node.getChildNodes();
        for (int i = 0; i < c.getLength(); i++) {
            String nodeName = c.item(i).getNodeName();
            if (nodeName.equals("Peers")) addPeers(c.item(i));
            else
            if (nodeName.equals("Realms")) addRealms(c.item(i));
        }
    }

    protected void addPeers(Node node) {
        NodeList c = node.getChildNodes();
        ArrayList<Configuration> items = new ArrayList<Configuration>();
        for (int i = 0; i < c.getLength(); i++) {
            String nodeName = c.item(i).getNodeName();
            if (nodeName.equals("Peer"))
                items.add( addPeer(c.item(i)) );
        }
        add(PeerTable, items.toArray(EMPTY_ARRAY));
    }

    protected void addRealms(Node node) {
        NodeList c = node.getChildNodes();
        ArrayList<Configuration> items = new ArrayList<Configuration>();
        for (int i = 0; i < c.getLength(); i++) {
            String nodeName = c.item(i).getNodeName();
            if (nodeName.equals("Realm"))
                items.add( addRealm(c.item(i)) );
        }
        add(RealmTable, items.toArray(EMPTY_ARRAY));
    }

    protected Configuration addPeer(Node node) {
        String rating = node.getAttributes().getNamedItem("rating").getNodeValue();
        String connecting = node.getAttributes().getNamedItem("attempt_connect").getNodeValue();
        String name = node.getAttributes().getNamedItem("name").getNodeValue();
        AppConfiguration c = getInstance();
        c.add( PeerRating, Integer.parseInt(rating));
        c.add( PeerAttemptConnection, Boolean.valueOf(connecting) );
        c.add( PeerName, name );
        if ( node.getAttributes().getNamedItem("ip") != null) {
            c.add( PeerIp, node.getAttributes().getNamedItem("ip").getNodeValue() );
        }
        if ( node.getAttributes().getNamedItem("portRange") != null) {
            c.add( PeerLocalPortRange, node.getAttributes().getNamedItem("portRange").getNodeValue() );
        }
        return c;
    }

    protected void addLocalPeer(Node node) {
        NodeList c = node.getChildNodes();
        for (int i = 0; i < c.getLength(); i++) {
            String nodeName = c.item(i).getNodeName();
            if (nodeName.equals("URI")) add(OwnDiameterURI, getValue(c.item(i)));
            addIPAddress(c.item(i));
            if (nodeName.equals("Realm")) add(OwnRealm, getValue(c.item(i)));
            if (nodeName.equals("VendorID")) add(OwnVendorID, getLongValue(c.item(i)));
            if (nodeName.equals("ProductName")) add(OwnProductName, getValue(c.item(i)));
            if (nodeName.equals("FirmwareRevision")) add(OwnFirmwareRevision, getLongValue(c.item(i)));
            if (nodeName.equals("Applications")) addApplications(c.item(i));
            if (nodeName.equals("OverloadMonitor")) addOverloadMonitor(c.item(i));
        }
    }

    private void addOverloadMonitor(Node node) {
        NodeList c = node.getChildNodes();
        ArrayList<Configuration> items = new ArrayList<Configuration>();
        for (int i = 0; i < c.getLength(); i++) {
            String nodeName = c.item(i).getNodeName();
            if (nodeName.equals("Entry"))
                items.add(addOverloadMonitorItem(c.item(i)));
        }
        add(OverloadMonitor, items.toArray(EMPTY_ARRAY));
    }

    private Configuration addOverloadMonitorItem(Node node) {
        return getInstance().
            add(OverloadEntryIndex, Integer.valueOf(getAttrValue(node, "index"))).
            add(OverloadEntrylowThreshold, Double.valueOf(getAttrValue(node, "lowThreshold"))).
            add(OverloadEntryhighThreshold, Double.valueOf(getAttrValue(node, "highThreshold"))).
            add(ApplicationId, addApplicationID(node.getChildNodes()));
    }

    protected void addIPAddress(Node node) {
        String nodeName = node.getNodeName();
        if (nodeName.equals("IPAddresses")) addIPAddresses(node);
    }

    private void addIPAddresses(Node node) {
        NodeList c = node.getChildNodes();
        ArrayList<Configuration> items = new ArrayList<Configuration>();
        for (int i = 0; i < c.getLength(); i++) {
            String nodeName = c.item(i).getNodeName();
            if (nodeName.equals("IPAddress")) items.add(addIPAddressItem(c.item(i)));
        }
        add(
           OwnIPAddresses, items.toArray(EMPTY_ARRAY)
        );
    }

    protected Configuration addIPAddressItem(Node node) {
        return getInstance().
                add(OwnIPAddress, getValue(node));
    }

    protected Configuration addRealm(Node node) {
        return getInstance().
            add(RealmEntry, getInstance().
                                add(ApplicationId, new Configuration[] {addApplicationID(node.getChildNodes())}).
                                add(RealmName,  getAttrValue(node, "name")).
                                add(RealmHosts, getAttrValue(node, "peers")).
                                add(RealmLocalAction,    getAttrValue(node, "local_action")).
                                add(RealmEntryIsDynamic, Boolean.valueOf(getAttrValue(node, "dynamic"))).
                                add(RealmEntryExpTime,   Long.valueOf(getAttrValue(node, "exp_time")))
            );
    }

    protected void appendOtherParameter(Node node) {
        String nodeName = node.getNodeName();
        if (nodeName.equals("DuplicateProtection")) add(DuplicateProtection, Boolean.valueOf(getValue(node)));
        if (nodeName.equals("DuplicateTimer")) add(DuplicateTimer, getLongValue(node));
        if (nodeName.equals("AcceptUndefinedPeer")) add(AcceptUndefinedPeer, Boolean.valueOf(getValue(node)));
    }

   protected void addExtensions(Node node) {
    }

    protected Long getLongValue(Node node) {
        return new Long(getValue(node));
    }

    protected Integer getIntValue(Node node) {
        return new Integer(getValue(node));
    }

    protected String getValue(Node node) {
        return node.getAttributes().getNamedItem("value").getNodeValue();
    }

    protected String getAttrValue(Node node, String name) {
        return node.getAttributes().getNamedItem(name).getNodeValue();
    }
}