package org.apache.commons.openpgp;

 /*

  * Licensed to the Apache Software Foundation (ASF) under one or more

  * contributor license agreements.  See the NOTICE file distributed with

  * this work for additional information regarding copyright ownership.

  * The ASF licenses this file to You under the Apache License, Version 2.0

  * (the "License"); you may not use this file except in compliance with

  * the License.  You may obtain a copy of the License at

  *

  *      http://www.apache.org/licenses/LICENSE-2.0

  *

  * Unless required by applicable law or agreed to in writing, software

  * distributed under the License is distributed on an "AS IS" BASIS,

  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.

  * See the License for the specific language governing permissions and

  * limitations under the License.

  */

 import java.io.IOException;

 import java.io.InputStream;

 import java.util.HashMap;

 import java.util.Map;

 import org.bouncycastle.openpgp.PGPException;

 import org.bouncycastle.openpgp.PGPObjectFactory;

 import org.bouncycastle.openpgp.PGPPublicKey;

 import org.bouncycastle.openpgp.PGPPublicKeyRing;

 import org.bouncycastle.openpgp.PGPSecretKey;

 import org.bouncycastle.openpgp.PGPSecretKeyRing;

 import org.bouncycastle.openpgp.PGPUtil;

 /**

  * Bouncy Castle implementation of the OpenPGP key ring.

  * 

  * @author <a href="mailto:brett@apache.org">Brett Porter</a>

  * @todo password is not secure

  */

 public class BouncyCastleKeyRing implements KeyRing

 {

     private String firstKeyId;

     private final Map<Long, PGPSecretKey> pgpSec = new HashMap<Long, PGPSecretKey>();

     private char[] password;

     private final Map<Long, PGPPublicKey> pgpPub = new HashMap<Long, PGPPublicKey>();

     private static final long MASK = 0xFFFFFFFFL;

     public BouncyCastleKeyRing()

     {

     }

     public BouncyCastleKeyRing( InputStream secretKeyRingStream, InputStream publicKeyRingStream, char[] password )

         throws IOException, PGPException

     {

         addSecretKeyRing( secretKeyRingStream, password );

         addPublicKeyRing( publicKeyRingStream );

     }

     public void addPublicKeyRing( InputStream publicKeyRingStream )

         throws IOException, PGPException

     {

         PGPObjectFactory pgpFact = new PGPObjectFactory( PGPUtil.getDecoderStream( publicKeyRingStream ) );

         Object obj;

         while ( ( obj = pgpFact.nextObject() ) != null )

         {

             if ( !( obj instanceof PGPPublicKeyRing ) )

             {

                 throw new PGPException( obj.getClass().getName() + " found where PGPPublicKeyRing expected" );

             }

             PGPPublicKeyRing keyRing = (PGPPublicKeyRing) obj;

             long key = keyRing.getPublicKey().getKeyID() & MASK;

             pgpPub.put( key, keyRing.getPublicKey() );

         }

     }

     public void addSecretKeyRing( InputStream secretKeyRingStream, char[] password )

         throws IOException, PGPException

     {

         PGPObjectFactory pgpFact = new PGPObjectFactory( PGPUtil.getDecoderStream( secretKeyRingStream ) );

         Object obj;

         while ( ( obj = pgpFact.nextObject() ) != null )

         {

             if ( !( obj instanceof PGPSecretKeyRing ) )

             {

                 throw new PGPException( obj.getClass().getName() + " found where PGPSecretKeyRing expected" );

             }

             PGPSecretKeyRing pgpSecret = (PGPSecretKeyRing) obj;

             long key = pgpSecret.getSecretKey().getKeyID() & MASK;

             if ( pgpSec.isEmpty() )

             {

                 // Convert the keyId to a hexadecimal upper case String

                 firstKeyId = Long.toHexString( key ).toUpperCase();

             }

             pgpSec.put( key, pgpSecret.getSecretKey() );

         }

         this.password = password;

     }

     public String getFirstKeyId()

     {

         return firstKeyId.toString();

     }

     public char[] getPassword()

     {

         return password;

     }

     public PGPSecretKey getSecretKey( String keyId )

     {

         return pgpSec.get( Long.valueOf( keyId, 16 ) );

     }

     public PGPPublicKey getPublicKey( String keyId )

     {

         return pgpPub.get( Long.valueOf( keyId, 16 ) );

     }

     public PGPSecretKey getSecretKey( long keyId )

     {

         return pgpSec.get(keyId & MASK);

     }

     public PGPPublicKey getPublicKey( long keyId )

     {

         return pgpPub.get(keyId & MASK);

     }

 }