package org.bouncycastle.openpgp;
import org.bouncycastle.bcpg.BCPGOutputStream;
import org.bouncycastle.util.Strings;
import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.util.ArrayList;
import java.util.Collection;
import java.util.HashMap;
import java.util.Iterator;
import java.util.List;
import java.util.Map;
/**
* Often a PGP key ring file is made up of a succession of master/sub-key key rings.
* If you want to read an entire public key file in one hit this is the class for you.
*/
public class PGPPublicKeyRingCollection
{
private Map pubRings = new HashMap();
private List order = new ArrayList();
private PGPPublicKeyRingCollection(
Map pubRings,
List order)
{
this.pubRings = pubRings;
this.order = order;
}
public PGPPublicKeyRingCollection(
byte[] encoding)
throws IOException, PGPException
{
this(new ByteArrayInputStream(encoding));
}
/**
* Build a PGPPublicKeyRingCollection from the passed in input stream.
*
* @param in input stream containing data
* @throws IOException if a problem parsing the base stream occurs
* @throws PGPException if an object is encountered which isn't a PGPPublicKeyRing
*/
public PGPPublicKeyRingCollection(
InputStream in)
throws IOException, PGPException
{
PGPObjectFactory pgpFact = new PGPObjectFactory(in);
Object obj;
while ((obj = pgpFact.nextObject()) != null)
{
if (!(obj instanceof PGPPublicKeyRing))
{
throw new PGPException(obj.getClass().getName() + " found where PGPPublicKeyRing expected");
}
PGPPublicKeyRing pgpPub = (PGPPublicKeyRing)obj;
Long key = new Long(pgpPub.getPublicKey().getKeyID());
pubRings.put(key, pgpPub);
order.add(key);
}
}
public PGPPublicKeyRingCollection(
Collection collection)
throws IOException, PGPException
{
Iterator it = collection.iterator();
while (it.hasNext())
{
PGPPublicKeyRing pgpPub = (PGPPublicKeyRing)it.next();
Long key = new Long(pgpPub.getPublicKey().getKeyID());
pubRings.put(key, pgpPub);
order.add(key);
}
}
/**
* Return the number of rings in this collection.
*
* @return size of the collection
*/
public int size()
{
return order.size();
}
/**
* return the public key rings making up this collection.
*/
public Iterator getKeyRings()
{
return pubRings.values().iterator();
}
/**
* Return an iterator of the key rings associated with the passed in userID.
*
* @param userID the user ID to be matched.
* @return an iterator (possibly empty) of key rings which matched.
* @throws PGPException
*/
public Iterator getKeyRings(
String userID)
throws PGPException
{
return getKeyRings(userID, false, false);
}
/**
* Return an iterator of the key rings associated with the passed in userID.
* <p>
*
* @param userID the user ID to be matched.
* @param matchPartial if true userID need only be a substring of an actual ID string to match.
* @return an iterator (possibly empty) of key rings which matched.
* @throws PGPException
*/
public Iterator getKeyRings(
String userID,
boolean matchPartial)
throws PGPException
{
return getKeyRings(userID, matchPartial, false);
}
/**
* Return an iterator of the key rings associated with the passed in userID.
* <p>
*
* @param userID the user ID to be matched.
* @param matchPartial if true userID need only be a substring of an actual ID string to match.
* @param ignoreCase if true case is ignored in user ID comparisons.
* @return an iterator (possibly empty) of key rings which matched.
* @throws PGPException
*/
public Iterator getKeyRings(
String userID,
boolean matchPartial,
boolean ignoreCase)
throws PGPException
{
Iterator it = this.getKeyRings();
List rings = new ArrayList();
if (ignoreCase)
{
userID = Strings.toLowerCase(userID);
}
while (it.hasNext())
{
PGPPublicKeyRing pubRing = (PGPPublicKeyRing)it.next();
Iterator uIt = pubRing.getPublicKey().getUserIDs();
while (uIt.hasNext())
{
String next = (String)uIt.next();
if (ignoreCase)
{
next = Strings.toLowerCase(next);
}
if (matchPartial)
{
if (next.indexOf(userID) > -1)
{
rings.add(pubRing);
}
}
else
{
if (next.equals(userID))
{
rings.add(pubRing);
}
}
}
}
return rings.iterator();
}
/**
* Return the PGP public key associated with the given key id.
*
* @param keyID
* @return the PGP public key
* @throws PGPException
*/
public PGPPublicKey getPublicKey(
long keyID)
throws PGPException
{
Iterator it = this.getKeyRings();
while (it.hasNext())
{
PGPPublicKeyRing pubRing = (PGPPublicKeyRing)it.next();
PGPPublicKey pub = pubRing.getPublicKey(keyID);
if (pub != null)
{
return pub;
}
}
return null;
}
/**
* Return the public key ring which contains the key referred to by keyID.
*
* @param keyID key ID to match against
* @return the public key ring
* @throws PGPException
*/
public PGPPublicKeyRing getPublicKeyRing(
long keyID)
throws PGPException
{
Long id = new Long(keyID);
if (pubRings.containsKey(id))
{
return (PGPPublicKeyRing)pubRings.get(id);
}
Iterator it = this.getKeyRings();
while (it.hasNext())
{
PGPPublicKeyRing pubRing = (PGPPublicKeyRing)it.next();
PGPPublicKey pub = pubRing.getPublicKey(keyID);
if (pub != null)
{
return pubRing;
}
}
return null;
}
/**
* Return true if a key matching the passed in key ID is present, false otherwise.
*
* @param keyID key ID to look for.
* @return true if keyID present, false otherwise.
*/
public boolean contains(long keyID)
throws PGPException
{
return getPublicKey(keyID) != null;
}
public byte[] getEncoded()
throws IOException
{
ByteArrayOutputStream bOut = new ByteArrayOutputStream();
this.encode(bOut);
return bOut.toByteArray();
}
public void encode(
OutputStream outStream)
throws IOException
{
BCPGOutputStream out;
if (outStream instanceof BCPGOutputStream)
{
out = (BCPGOutputStream)outStream;
}
else
{
out = new BCPGOutputStream(outStream);
}
Iterator it = order.iterator();
while (it.hasNext())
{
PGPPublicKeyRing sr = (PGPPublicKeyRing)pubRings.get(it.next());
sr.encode(out);
}
}
/**
* Return a new collection object containing the contents of the passed in collection and
* the passed in public key ring.
*
* @param ringCollection the collection the ring to be added to.
* @param publicKeyRing the key ring to be added.
* @return a new collection merging the current one with the passed in ring.
* @exception IllegalArgumentException if the keyID for the passed in ring is already present.
*/
public static PGPPublicKeyRingCollection addPublicKeyRing(
PGPPublicKeyRingCollection ringCollection,
PGPPublicKeyRing publicKeyRing)
{
Long key = new Long(publicKeyRing.getPublicKey().getKeyID());
if (ringCollection.pubRings.containsKey(key))
{
throw new IllegalArgumentException("Collection already contains a key with a keyID for the passed in ring.");
}
Map newPubRings = new HashMap(ringCollection.pubRings);
List newOrder = new ArrayList(ringCollection.order);
newPubRings.put(key, publicKeyRing);
newOrder.add(key);
return new PGPPublicKeyRingCollection(newPubRings, newOrder);
}
/**
* Return a new collection object containing the contents of this collection with
* the passed in public key ring removed.
*
* @param ringCollection the collection the ring to be removed from.
* @param publicKeyRing the key ring to be removed.
* @return a new collection not containing the passed in ring.
* @exception IllegalArgumentException if the keyID for the passed in ring not present.
*/
public static PGPPublicKeyRingCollection removePublicKeyRing(
PGPPublicKeyRingCollection ringCollection,
PGPPublicKeyRing publicKeyRing)
{
Long key = new Long(publicKeyRing.getPublicKey().getKeyID());
if (!ringCollection.pubRings.containsKey(key))
{
throw new IllegalArgumentException("Collection does not contain a key with a keyID for the passed in ring.");
}
Map newPubRings = new HashMap(ringCollection.pubRings);
List newOrder = new ArrayList(ringCollection.order);
newPubRings.remove(key);
for (int i = 0; i < newOrder.size(); i++)
{
Long r = (Long)newOrder.get(i);
if (r.longValue() == key.longValue())
{
newOrder.remove(i);
break;
}
}
return new PGPPublicKeyRingCollection(newPubRings, newOrder);
}
}