psemembershiptest.java

JXTA&#8482 is a set of open, generalized peer-to-peer (P2P) protocols that allow any networked devi

/*
 * Copyright (c) 2002-2007 Sun Microsystems, Inc.  All rights reserved.
 *  
 *  The Sun Project JXTA(TM) Software License
 *  
 *  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 Sun Microsystems, Inc. for JXTA(TM) technology." 
 *     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.
 *  
 *  JXTA is a registered trademark of Sun Microsystems, Inc. in the United 
 *  States and other countries.
 *  
 *  Please see the license information page at :
 *  <http://www.jxta.org/project/www/license.html> for instructions on use of 
 *  the license in source files.
 *  
 *  ====================================================================
 *  
 *  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());
        }
    }
}