/*

 * CDDL HEADER START

 *

 * The contents of this file are subject to the terms of the

 *

 * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE

 * or http://www.opensolaris.org/os/licensing.

 * See the License for the specific language governing permissions

 * and limitations under the License.

 *

 * When distributing Covered Code, include this CDDL HEADER in each

 * file and include the License file at usr/src/OPENSOLARIS.LICENSE.

 * If applicable, add the following below this CDDL HEADER, with the

 * fields enclosed by brackets "[]" replaced with your own identifying

 * information: Portions Copyright [yyyy] [name of copyright owner]

 *

 * CDDL HEADER END

 */

/*

 * Copyright (c) 1999 by Sun Microsystems, Inc.

 * All rights reserved.

 *

 */

//  ServiceStoreFactory.java: Factory for creating ServiceStore objects.

//  Author:           James Kempf

//  Created On:       Fri Apr 17 12:14:12 1998

//  Last Modified By: James Kempf

//  Last Modified On: Mon Jan  4 15:26:34 1999

//  Update Count:     34

//

package com.sun.slp;

import java.util.*;

import java.io.*;

/**

 * The ServiceStoreFactory provides a way to obtain a ServiceStore

 * object. The exact implementation will depend on how the

 * DA/slpd is configured. It could be an in-memory database,

 * a connection to an LDAP server, or a persistent object

 * database.

 *

 * @author James Kempf

 */

class ServiceStoreFactory extends Object {

    private static final String DEFAULT_SERVICE_STORE =

	"com.sun.slp.ServiceStoreInMemory";

    private static final String SERVICE_STORE_PROPERTY =

	"sun.net.slp.serviceStoreClass";

    // Comment characters for deserialization.

    final private static char COMMENT_CHAR1 = '#';

    final private static char COMMENT_CHAR2 = ';';

    // Character for URL list separator.

    final private static String URL_LIST_SEP = ", ";

    // Identifies scopes pseudo-attribute.

    final private static String SCOPES_ATTR_ID = "scopes";

    /**

     * Return the ServiceStore for the SLP agent.

     *

     * @return An object supporting the ServiceStore interface.

     * @exception ServiceLocationException Thrown

     *			if the ServiceStore object can't be

     *			created or if the

     *			class implementing the ServiceStore required

     *			a network connnection (for example, an LDAP

     *			server) and the connection couldn't be made.

     */

    static ServiceStore createServiceStore()

	throws ServiceLocationException {

	return createServiceStoreFromProperty(SERVICE_STORE_PROPERTY);

    }

    // Create the appropriate ServiceStore object from the property.

    private static ServiceStore

	createServiceStoreFromProperty(String property)

	throws ServiceLocationException {

	Properties props = System.getProperties();

	String storeClassName =

	    props.getProperty(property,

			      DEFAULT_SERVICE_STORE);

	Class storeClass = null;

	try {

	    storeClass = Class.forName(storeClassName);

	} catch (ClassNotFoundException ex) {

	    throw

		new ServiceLocationException(

				ServiceLocationException.INTERNAL_SYSTEM_ERROR,

				"ssf_no_class",

				new Object[] {storeClassName});

	}

	ServiceStore store = null;

	try {

	    store = (ServiceStore)storeClass.newInstance();

	} catch (InstantiationException ex) {

	    throw

		new ServiceLocationException(

				ServiceLocationException.INTERNAL_SYSTEM_ERROR,

				"ssf_inst_ex",

				new Object[] {

		    storeClassName,

			ex.getMessage()});

	} catch (IllegalAccessException ex) {

	    throw

		new ServiceLocationException(

				ServiceLocationException.INTERNAL_SYSTEM_ERROR,

				"ssf_ill_ex",

				new Object[] {

		    storeClassName,

			ex.getMessage()});

	} catch (ClassCastException ex) {

	    throw

		new ServiceLocationException(

				ServiceLocationException.INTERNAL_SYSTEM_ERROR,

				"ssf_class_cast",

				new Object[] {storeClassName});

	}

	return store;

    }

    /**

     * Deserialize a service store from the open stream.

     *

     * @param is The object input stream for the service store.

     * @return ServiceStore deserialized from the stream.

     * @exception ServiceLocationException If anything goes

     *				wrong with the deserialization.

     */

    static ServiceStore deserializeServiceStore(BufferedReader is)

	throws ServiceLocationException {

	ServiceStore ss = new ServiceStoreInMemory();

	try {

	    deserialize(is, ss);

	} catch (IOException ex) {

	    throw

		new ServiceLocationException(

				ServiceLocationException.PARSE_ERROR,

				"ssf_io_deser",

				new Object[] {ex.getMessage()});

	}

	return ss;

    }

    // Read the service store in the standard format from the input

    private static void deserialize(BufferedReader in, ServiceStore store)

	throws IOException, ServiceLocationException {

	SLPConfig conf = SLPConfig.getSLPConfig();

	int linecount = 0;

	int scopeLinenum = 0;

	// Parse input file until no bytes left.

	while (in.ready()) {

	    linecount++;

	    String line = in.readLine().trim();

	    // Skip any empty lines at this level.

	    if (line.length() <= 0) {

		continue;

	    }

	    char cc = line.charAt(0);

	    // If initial character is "#" or ";", ignore the line.

	    //  It's a comment. Also if the line is empty.

	    if (cc == COMMENT_CHAR1 ||

		cc == COMMENT_CHAR2) {

		continue;

	    }

	    // At this level, the line must be a URL registration,

	    //  with format:

	    //

	    // service-url ", " language ", " lifetime [ ", " type ]

	    //

	    //

	    //  We allow arbitrary whitespace around commas.

	    StringTokenizer tk = new StringTokenizer(line, URL_LIST_SEP);

	    String surl = null;

	    String slang = null;

	    String slifetime = null;

	    String sType = null;

	    if (tk.hasMoreTokens()) {

		surl = tk.nextToken().trim();

		if (tk.hasMoreTokens()) {

		    slang = tk.nextToken().trim();

		    if (tk.hasMoreTokens()) {

			slifetime = tk.nextToken().trim();

			if (tk.hasMoreTokens()) {

			    sType = tk.nextToken().trim();

			    if (tk.hasMoreTokens()) {

				slang = null;

					// should be nothing more on the line.

			    }

			}

		    }

		}

	    }

	    // Check for errors.

	    if (surl == null || slifetime == null || slang == null) {

		throw

		    new ServiceLocationException(

				ServiceLocationException.PARSE_ERROR,

				"ssf_not_valid_url",

				new Object[] {line});

	    }

	    // Create the service: URL.

	    Locale locale = SLPConfig.langTagToLocale(slang);

	    ServiceURL url = null;

	    try {

		int lifetime = Integer.parseInt(slifetime);

		// If lifetime is maximum, then set to LIFETIME_PERMANENT.

		if (lifetime == ServiceURL.LIFETIME_MAXIMUM) {

		    lifetime = ServiceURL.LIFETIME_PERMANENT;

		}

		url = new ServiceURL(surl, lifetime);

		if (sType != null) {

		    // Check if it's OK for this service URL.

		    ServiceType utype = url.getServiceType();

		    if (utype.isServiceURL()) {

			conf.writeLog("ssf_set_servc_err",

				      new Object[] {

			    surl,

				utype});

		    } else {

			ServiceType t = new ServiceType(sType);

			if (!t.isServiceURL() &&

			    !t.equals(url.getServiceType())) {

			    url.setServiceType(t);

			}

		    }

		}

	    } catch (NumberFormatException ex) {

		throw

		    new ServiceLocationException(

				ServiceLocationException.PARSE_ERROR,

				"ssf_not_valid_lifetime",

				new Object[] {

			slifetime, new Integer(linecount)});

	    } catch (IllegalArgumentException ex) {

		throw

		    new ServiceLocationException(

				ServiceLocationException.PARSE_ERROR,

				"ssf_syntax_err",

				new Object[] {

			ex.getMessage(), new Integer(linecount)});

	    }

	    // Get attributes. Format should be:

	    //

	    //      attr-line    = attr-assign | keyword

	    //	attr-assign  = attr-id "=" attrval-list

	    //	keyword      = attr-id

	    //	attrval-list = attrval | attrval ", " attrval-list

	    Vector attrs = new Vector();

	    Hashtable ht = new Hashtable();

	    ServiceLocationAttribute scopeAttr = null;

	    boolean firstLine = true;

	    try {

		while (in.ready()) {

		    linecount++;

		    line = in.readLine();

		    // Empty line indicates we're done with attributes.

		    if (line.length() <= 0) {

			break;

		    }

		    // Format the line for creating. Check whether it's a

		    // keyword or not.

		    if (line.indexOf("=") != -1) {

			line = "(" + line + ")";

		    }

		    // Create the attribute from the string.

		    ServiceLocationAttribute attr =

			new ServiceLocationAttribute(line, false);

		    // If this is the scope attribute, save until later.

		    if (firstLine) {

			firstLine = false;

			if (attr.getId().equalsIgnoreCase(SCOPES_ATTR_ID)) {

			    scopeAttr = attr;

			    continue; // do NOT save as a regular attribute.

			}

		    }

		    ServiceLocationAttribute.mergeDuplicateAttributes(attr,

								      ht,

								      attrs,

								      false);

		}

	    } catch (ServiceLocationException e) {

		// tack on the line count

		e.makeAddendum(" (line " + linecount + ")");

		throw e;

	    }

	    Vector scopes = null;

	    // Use scopes we've been configured with if none.

	    if (scopeAttr == null) {

		scopes = conf.getSAConfiguredScopes();

	    } else {

		scopes = (Vector)scopeAttr.getValues();

		try {

		    // Unescape scope strings.

		    SLPHeaderV2.unescapeScopeStrings(scopes);

		    // Validate, lower case scope names.

		    DATable.validateScopes(scopes, locale);

		} catch (ServiceLocationException e) {

		    e.makeAddendum(" (line " + scopeLinenum + ")");

		    throw e;

		}

	    }

	    // We've got the attributes, the service URL, scope, and

	    //  locale, so add a record. Note that any crypto is

	    //  added when the registration is actually done.

	    store.register(url, attrs, scopes, locale, null, null);

	    // Create a CSrvReg for forwarding

	    CSrvReg creg = new CSrvReg(true, locale, url, scopes,

				       attrs, null, null);

	    ServerDATable daTable = ServerDATable.getServerDATable();

	    daTable.forwardSAMessage(creg, conf.getLoopback());

	}

    }

    // Write the service store in the standard format to the output

    // stream.

    static void serialize(BufferedWriter out, ServiceStore store)

	throws IOException, ServiceLocationException {

	Enumeration recs = store.getServiceRecordsByScope(null);

	while (recs.hasMoreElements()) {

	    ServiceStore.ServiceRecord rec =

		(ServiceStore.ServiceRecord)recs.nextElement();

	    ServiceURL url = rec.getServiceURL();

	    String surl = url.toString();

	    Vector attrs = (Vector)rec.getAttrList().clone();

	    Locale locale = rec.getLocale();

	    Vector scopes = rec.getScopes();

	    StringBuffer line = new StringBuffer();

	    // Compose the registration line.

	    line.append(surl);

	    line.append(", ");

	    line.append(SLPConfig.localeToLangTag(locale));

	    line.append(", ");

	    line.append(Integer.toString(url.getLifetime()));

	    // Put out the service type and naming authority if the

	    //  URL is not a service URL.

	    if (!surl.startsWith(Defaults.SERVICE_PREFIX)) {

		ServiceType type = url.getServiceType();

		line.append(", ");

		line.append(type.toString());

	    }

	    // Write out line.

	    out.write(line.toString(), 0, line.length());

	    out.newLine();

	    // Zero line buffer.

	    line.setLength(0);

	    // Insert a scope attribute, if the scope isn't simply "DEFAULT".

	    if (scopes.size() > 1 &&

		!Defaults.DEFAULT_SCOPE.equals((String)scopes.elementAt(0))) {

		attrs.insertElementAt(

				new ServiceLocationAttribute(SCOPES_ATTR_ID,

							     scopes),

				0);

	    }

	    // Write out the attributes.

	    int i, n = attrs.size();

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

		ServiceLocationAttribute attr =

		    (ServiceLocationAttribute)attrs.elementAt(i);

		Vector vals = attr.getValues();

		line.append(

		ServiceLocationAttribute.escapeAttributeString(attr.getId(),

							       false));

		// Add the escaped values.

		if (vals != null) {

		    line.append("=");

		    int j, m = vals.size();

		    for (j = 0; j < m; j++) {

			Object v = vals.elementAt(j);

			if (j > 0) {

			    line.append(", ");

			}

			line.append(ServiceLocationAttribute.escapeValue(v));

		    }

		}

		out.write(line.toString(), 0, line.length());

		out.newLine();

		// Zero out string buffer.

		line.setLength(0);

	    }

	    // End of registration.

	    out.newLine();

	}

	out.flush();