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);
     }
 }