org.jmrtd.lds.DG15File Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of jmrtd Show documentation
Java Machine Readable Travel Documents API.
There is a newer version: 0.7.42
/*
 * JMRTD - A Java API for accessing machine readable travel documents.
 *
 * Copyright (C) 2006 - 2014  The JMRTD team
 *
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2.1 of the License, or (at your option) any later version.
 *
 * This library is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with this library; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
 *
 * $Id: DG15File.java 1559 2014-11-14 12:46:26Z martijno $
 */

package org.jmrtd.lds;

import java.io.DataInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.security.GeneralSecurityException;
import java.security.InvalidAlgorithmParameterException;
import java.security.KeyFactory;
import java.security.PublicKey;
import java.security.spec.InvalidKeySpecException;
import java.security.spec.X509EncodedKeySpec;
import java.util.logging.Logger;

/**
 * File structure for the EF_DG15 file.
 * Datagroup 15 contains the public key used in AA.
 * 
 * @author Cees-Bart Breunesse ([email protected])
 * @author Martijn Oostdijk ([email protected])
 * 
 * @version $Revision: 1559 $
 */
public class DG15File extends DataGroup {

	private static final long serialVersionUID = 3834304239673755744L;

	private static final Logger LOGGER = Logger.getLogger("org.jmrtd");

	private PublicKey publicKey;

	/**
	 * Constructs a new file.
	 * 
	 * @param publicKey the key to store in this file
	 */
	public DG15File(PublicKey publicKey) {
		super(EF_DG15_TAG);
		this.publicKey = publicKey;
	}

	/**
	 * Constructs a new file from binary representation.
	 * 
	 * @param inputStream an input stream
	 * 
	 * @throws IOException on error reading from input stream
	 */
	public DG15File(InputStream inputStream) throws IOException {
		super(EF_DG15_TAG, inputStream);
	}

	protected void readContent(InputStream inputStream) throws IOException {
		DataInputStream dataInputStream = inputStream instanceof DataInputStream ? (DataInputStream)inputStream : new DataInputStream(inputStream);
		try {
			byte[] value = new byte[getLength()];
			dataInputStream.readFully(value);

			publicKey = getPublicKey(value);
		} catch (GeneralSecurityException e) {
			throw new IllegalArgumentException(e.toString());
		}
	}

	private static PublicKey getPublicKey(byte[] keyBytes) throws GeneralSecurityException {
		X509EncodedKeySpec pubKeySpec = new X509EncodedKeySpec(keyBytes);

		String[] algorithms = { "RSA", "EC" };

		for (String algorithm: algorithms) {
			try {
				KeyFactory keyFactory = KeyFactory.getInstance(algorithm);
				PublicKey publicKey = keyFactory.generatePublic(pubKeySpec);
				return publicKey;
			} catch (InvalidKeySpecException ikse) {
				/* NOTE: Ignore, try next algorithm. */
			}
		}
		throw new InvalidAlgorithmParameterException();
	}

	protected void writeContent(OutputStream out) throws IOException {
		out.write(publicKey.getEncoded());
	}

	/**
	 * Gets the public key stored in this file.
	 * 
	 * @return the public key
	 */
	public PublicKey getPublicKey() {
		return publicKey;
	}

	public boolean equals(Object obj) {
		if (obj == null) { return false; }
		if (obj.getClass() != this.getClass()) { return false; }
		DG15File other = (DG15File)obj;
		return publicKey.equals(other.publicKey);
	}

	public int hashCode() {
		return 5 * publicKey.hashCode() + 61;
	}

	public String toString() {
		return "DG15File [" + publicKey.toString() + "]";
	}
}