/*
 * CDDL HEADER START
 *
 * The contents of this file are subject to the terms of the
 * Common Development and Distribution License (the "License").
 * You may not use this file except in compliance with the License.
 *
 * 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.
 *
 */

//  CSrvMsg.java:     Message class for SLP service reply.
//  Author:           James Kempf
//  Created On:       Thu Oct  9 15:09:32 1997
//  Last Modified By: James Kempf
//  Last Modified On: Tue Oct 27 11:01:45 1998
//  Update Count:     127
//

package com.sun.slp;

import java.util.*;
import java.io.*;

/**
 * The CSrvMsg class models the SLP client side service message.
 *
 * @author James Kempf
 */

class CSrvMsg extends SrvLocMsgImpl {

    Vector serviceURLs = new Vector();		// vector of ServiceURL objects
    Hashtable  URLSignatures = new Hashtable();	// authentication block lists.

    // Only used for testing.

    protected CSrvMsg() { }

    // Construct a CSrvMsg from the byte input stream. This is a SrvRply.
    //  error code is already parsed.

    CSrvMsg(SLPHeaderV2 hdr, DataInputStream dis) 
	throws ServiceLocationException, IOException {
	super(hdr, SrvLocHeader.SrvRply);

	// Don't parse the rest if there's an error.

	if (hdr.errCode != ServiceLocationException.OK) {
	    return;
	}

	// Note that we ignore the overflow flag here, because the spec
	//  disallows partial URL entries, and so we should be able
	//  to parse in the rest of the message even if there is overflow.
	//  This is different from other messages.

	parseServiceURLsIn(hdr, dis);
    }

    // Parse in a vector of service URLs including lifetime.

    protected void parseServiceURLsIn(SLPHeaderV2 hdr, DataInputStream dis)
	throws ServiceLocationException, IOException {

	// Get the number of service URL's.

	int i, n = hdr.getInt(dis);

	// Get the service URL's including lifetime.

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

	    ServiceURL surl = 
		hdr.parseServiceURLIn(dis, URLSignatures,
				      ServiceLocationException.PARSE_ERROR);

	    serviceURLs.addElement(surl);

	    // Verify the signature if any. Doing it here saves muss and
	    //  fuss in the upper layers.

	    Hashtable auth = (Hashtable) URLSignatures.get(surl);

	    if (auth != null) {
		AuthBlock.verifyAll(auth);
	    }
	}

	// Set the header number of replies received.

	hdr.iNumReplies = serviceURLs.size();

    }

    // Construct a CSrvMsg from the arguments.

    CSrvMsg(Locale locale,
	    ServiceType serviceType,
	    Vector scopes,
	    String query) 
	throws ServiceLocationException {

	this.initialize(locale, serviceType, scopes, query);

    }

    // Initialize as a SLPv2 SrvRqst.

    protected void 
	initialize(Locale locale,
		   ServiceType serviceType,
		   Vector scopes,
		   String query)
	throws ServiceLocationException {

	SLPHeaderV2 hdr = new SLPHeaderV2(SrvLocHeader.SrvReq, false, locale);
	this.hdr = hdr;
	hdr.scopes = (Vector)scopes.clone();

	// Set up for previous responders.

	hdr.previousResponders = new Vector();

	// Create the payload for the message.

	ByteArrayOutputStream baos = new ByteArrayOutputStream();

	// Escape scope strings.

	hdr.escapeScopeStrings(scopes);

	// Retrieve the configured SPI, if any
	String spi = "";
	if (SLPConfig.getSLPConfig().getHasSecurity()) {
	    LinkedList spiList = AuthBlock.getSPIList("sun.net.slp.SPIs");
	    if (spiList != null && !spiList.isEmpty()) {
		// There can be only one configured SPI for UAs
		spi = (String) spiList.getFirst();
	    }
	}

	// Write out the service type.

	hdr.putString(serviceType.toString(), baos);

	// Write out scopes.

	hdr.parseCommaSeparatedListOut(scopes, baos);

	// Write out query. 

	hdr.putString(query, baos);

	// Write out SPI

	hdr.putString(spi, baos);

	hdr.payload = baos.toByteArray();
    }

    //
    // Property accessors
    //

    final Hashtable getURLSignature(ServiceURL URL) {

	return (Hashtable)(URLSignatures.get(URL));
    }

    final void setURLSignature(ServiceURL URL, Hashtable sig) 
	throws IllegalArgumentException {

	if (sig == null) {
	    URLSignatures.remove(URL);
	} else {
	    URLSignatures.put(URL, sig);
	}
    }

}