psemembershiptest.java

jxta_src_2.41b jxta 2.41b 最新版源码 from www.jxta.org

/*
 *
 * $Id: pseMembershipTest.java,v 1.4 2005/11/09 20:29:49 bondolo Exp $
 *
 * Copyright (c) 2002 Sun Microsystems, Inc.  All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 *
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 *
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in
 *    the documentation and/or other materials provided with the
 *    distribution.
 *
 * 3. The end-user documentation included with the redistribution,
 *    if any, must include the following acknowledgment:
 *       "This product includes software developed by the
 *       Sun Microsystems, Inc. for Project JXTA."
 *    Alternately, this acknowledgment may appear in the software itself,
 *    if and wherever such third-party acknowledgments normally appear.
 *
 * 4. The names "Sun", "Sun Microsystems, Inc.", "JXTA" and "Project JXTA"
 *    must not be used to endorse or promote products derived from this
 *    software without prior written permission. For written
 *    permission, please contact Project JXTA at http://www.jxta.org.
 *
 * 5. Products derived from this software may not be called "JXTA",
 *    nor may "JXTA" appear in their name, without prior written
 *    permission of Sun.
 *
 * THIS SOFTWARE IS PROVIDED ``AS IS'' AND ANY EXPRESSED OR IMPLIED
 * WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
 * DISCLAIMED.  IN NO EVENT SHALL SUN MICROSYSTEMS OR
 * ITS CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF
 * USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
 * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
 * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT
 * OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
 * SUCH DAMAGE.
 *
 * ====================================================================
 *
 * This software consists of voluntary contributions made by many
 * individuals on behalf of Project JXTA.  For more information on
 * Project JXTA, please see <http://www.jxta.org/>.
 *
 * This license is based on the BSD license adopted by the Apache Foundation.
 *
 */
package net.jxta.impl.membership.pse;

import java.io.Reader;
import java.io.StringReader;
import java.io.StringWriter;
import java.net.URI;
import javax.crypto.EncryptedPrivateKeyInfo;
import java.security.PrivateKey;
import java.util.Arrays;
import java.util.Map;

import java.net.URISyntaxException;

import junit.framework.*;

import net.jxta.credential.Credential;
import net.jxta.credential.PrivilegedOperation;
import net.jxta.document.AdvertisementFactory;
import net.jxta.document.Element;
import net.jxta.document.MimeMediaType;
import net.jxta.document.StructuredDocument;
import net.jxta.document.StructuredDocumentFactory;
import net.jxta.document.XMLDocument;
import net.jxta.document.XMLElement;
import net.jxta.discovery.DiscoveryService;
import net.jxta.id.ID;
import net.jxta.id.IDFactory;
import net.jxta.membership.MembershipService;
import net.jxta.peergroup.PeerGroup;
import net.jxta.peergroup.PeerGroupFactory;
import net.jxta.platform.ModuleSpecID;
import net.jxta.protocol.ModuleImplAdvertisement;
import net.jxta.protocol.PeerGroupAdvertisement;
import net.jxta.credential.AuthenticationCredential;
import net.jxta.credential.Credential;
import net.jxta.membership.InteractiveAuthenticator;
import net.jxta.membership.MembershipService;

import net.jxta.impl.peergroup.StdPeerGroupParamAdv;
import net.jxta.impl.membership.pse.PSEMembershipService;
import net.jxta.impl.membership.pse.PSEUtils;
import net.jxta.impl.membership.pse.PSEUtils.IssuerInfo;
import net.jxta.impl.protocol.PSEConfigAdv;

public class pseMembershipTest extends TestCase {
    
    static PeerGroup npg = null;
    static PeerGroup pg = null;
    
    public pseMembershipTest(java.lang.String testName) {
        super(testName);
        synchronized( pseMembershipTest.class ) {
            try {
                if( null == npg ) {
                    npg = PeerGroupFactory.newNetPeerGroup();
                    //                   npg.startApp( new String[0]);
                    
                    ModuleImplAdvertisement newGroupImpl = npg.getAllPurposePeerGroupImplAdvertisement();
                    
                    StdPeerGroupParamAdv params = new StdPeerGroupParamAdv( newGroupImpl.getParam() );
                    
                    Map services = params.getServices();
                    
                    ModuleImplAdvertisement aModuleAdv = (ModuleImplAdvertisement) services.get( PeerGroup.membershipClassID );
                    services.remove( PeerGroup.membershipClassID );
                    
                    ModuleImplAdvertisement implAdv = (ModuleImplAdvertisement) AdvertisementFactory.newAdvertisement( ModuleImplAdvertisement.getAdvertisementType());
                    
                    implAdv.setModuleSpecID( PSEMembershipService.pseMembershipSpecID);
                    implAdv.setCompat(aModuleAdv.getCompat() );
                    implAdv.setCode( PSEMembershipService.class.getName() );
                    implAdv.setUri(aModuleAdv.getUri());
                    implAdv.setProvider(aModuleAdv.getProvider());
                    implAdv.setDescription("PSE Membership Service");
                    
                    // replace it
                    services.put( PeerGroup.membershipClassID, implAdv );
                    
                    newGroupImpl.setParam( (Element) params.getDocument( MimeMediaType.XMLUTF8 ) );
                    
                    // XXX bondolo 20041014 if we knew we were going to create many of this type of group we would use a well known id.
                    newGroupImpl.setModuleSpecID( IDFactory.newModuleSpecID( newGroupImpl.getModuleSpecID().getBaseClass() ) );
                    
                    npg.getDiscoveryService().publish(newGroupImpl, PeerGroup.DEFAULT_LIFETIME, PeerGroup.DEFAULT_EXPIRATION);
                    
                    npg.getDiscoveryService().remotePublish(newGroupImpl, PeerGroup.DEFAULT_LIFETIME );
                    
                    PeerGroupAdvertisement newPGAdv = (PeerGroupAdvertisement) AdvertisementFactory.newAdvertisement( PeerGroupAdvertisement.getAdvertisementType() );
                    
                    newPGAdv.setPeerGroupID(IDFactory.newPeerGroupID());
                    newPGAdv.setModuleSpecID(newGroupImpl.getModuleSpecID());
                    newPGAdv.setName( "Test Group" );
                    newPGAdv.setDescription("Created by Unit Test");
                    
                    PSEConfigAdv pseConf = (PSEConfigAdv) AdvertisementFactory.newAdvertisement(PSEConfigAdv.getAdvertisementType());
                    
                    PSEUtils.IssuerInfo info = PSEUtils.genCert( "bob", null);
                    
                    pseConf.setCertificate(info.cert);
                    pseConf.setPrivateKey(info.subjectPkey, "password".toCharArray());
                    
                    XMLDocument pseDoc = (XMLDocument) pseConf.getDocument(MimeMediaType.XMLUTF8);
                    
                    newPGAdv.putServiceParam( PeerGroup.membershipClassID, pseDoc );
                    
                    npg.getDiscoveryService().publish(newPGAdv, PeerGroup.DEFAULT_LIFETIME,PeerGroup.DEFAULT_EXPIRATION );
                    
                    npg.getDiscoveryService().remotePublish(newPGAdv, PeerGroup.DEFAULT_LIFETIME);
                    
                    pg = npg.newGroup( newPGAdv );
                }
            }   catch( Throwable all ) {
                all.printStackTrace();
                fail("exception thrown : " + all.getMessage());
            }
        }
    }
    
    public static void main(java.lang.String[] args) {
        junit.textui.TestRunner.run(suite());

        synchronized( pseMembershipTest.class ) {
            if( null != pg ) {
                pg.stopApp();
                pg.unref();
                pg = null;
            }
            
            if( null != npg ) {
                npg.stopApp();
                npg.unref();
                npg = null;
            }
        }

        System.err.flush();
        System.out.flush();
     }
    
    public static Test suite() {
        TestSuite suite = new TestSuite(pseMembershipTest.class);
        return suite;
    }
    
    public void testLogin() {
        try {
            MembershipService membership = pg.getMembershipService();
            
            membership.resign();
            
            assertTrue( "Should be null default", (null == membership.getDefaultCredential()) );
            
            AuthenticationCredential authCred = new AuthenticationCredential( pg, "StringAuthentication", null );
            
            StringAuthenticator auth = null;
            try {
                auth = (StringAuthenticator) membership.apply( authCred );
            } catch( Exception failed ) {
                ;
            }
            
            if( null != auth ) {
                auth.setAuth1_KeyStorePassword( "password".toCharArray() );
                auth.setAuth2Identity( pg.getPeerID() );
                auth.setAuth3_IdentityPassword( "password".toCharArray() );
                
                assertTrue( "should have been ready",  auth.isReadyForJoin() );
                
                Credential newCred = membership.join( auth );
                
                assertTrue( "Should have returned a credential", (null != newCred) );
                
                assertTrue( "Should be default credential", (newCred == membership.getDefaultCredential()) );
            }
        } catch( Throwable all ) {
            all.printStackTrace();
            fail("exception thrown : " + all.getMessage());
        }
    }
    
//    public void testInteractiveLogin() {
//        try {
//            MembershipService membership = pg.getMembershipService();
//            
//            membership.resign();
//            
//            AuthenticationCredential authCred = new AuthenticationCredential( pg, "InteractiveAuthentication", null );
//            
//            InteractiveAuthenticator auth = (InteractiveAuthenticator) membership.apply( authCred );
//            
//            if( auth.interact() ) {
//                assertTrue( "should have been ready",  auth.isReadyForJoin() );
//                membership.join( auth );
//            }
//        }   catch( Throwable all ) {
//            all.printStackTrace();
//            fail("exception thrown : " + all.getMessage());
//        }
//    }
    
    public void testPKCS5() {
        try {
            IssuerInfo test = PSEUtils.genCert( "test", null );
            
            EncryptedPrivateKeyInfo encPrivKey = PSEUtils.pkcs5_Encrypt_pbePrivateKey( "password".toCharArray(), test.subjectPkey, 500 );
            
            assertNotNull( "Could not encrypt Private Key", encPrivKey );
            
            PrivateKey decPrivKey = PSEUtils.pkcs5_Decrypt_pbePrivateKey( "password".toCharArray(), test.subjectPkey.getAlgorithm(), encPrivKey );
            
            assertNotNull( "Could not decrypt Private Key", decPrivKey );
            
            Arrays.equals( test.subjectPkey.getEncoded(), decPrivKey.getEncoded() );
            
            byte [] encPrivKeyDer = encPrivKey.getEncoded();
            
            EncryptedPrivateKeyInfo deserialedencPrivKey = new EncryptedPrivateKeyInfo( encPrivKeyDer) ;
            
            decPrivKey = PSEUtils.pkcs5_Decrypt_pbePrivateKey( "password".toCharArray(), test.subjectPkey.getAlgorithm(), deserialedencPrivKey );
            
            assertNotNull( "Could not decrypt Private Key", decPrivKey );
            
            Arrays.equals( test.subjectPkey.getEncoded(), decPrivKey.getEncoded() );
        } catch (Exception caught) {
            caught.printStackTrace();
            fail("exception thrown : " + caught.getMessage());
        }
    }
}