/*
 * 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.
 * 
 */

package org.apache.jmeter.util.keystore;

import java.io.InputStream;
import java.security.KeyStore;
import java.security.PrivateKey;
import java.security.cert.Certificate;
import java.security.cert.X509Certificate;
import java.util.Enumeration;

/**
 * Use this Keystore to wrap the normal KeyStore implementation.
 * 
 * @author <a href="bloritsch@apache.org">Berin Loritsch</a>
 */
public class DefaultKeyStore extends JmeterKeyStore {
	private X509Certificate[] certChain;

	private PrivateKey key;

	private String alias;

	private final KeyStore store;

	public DefaultKeyStore(String type) throws Exception {
		this.store = KeyStore.getInstance(type);
	}

	public void load(InputStream is, String pword) throws Exception {
		store.load(is, pword.toCharArray());
		PrivateKey _key = null;
		X509Certificate[] _certChain = null;

		if (null != is){ // No point checking an empty keystore
			
			Enumeration aliases = store.aliases();
			while (aliases.hasMoreElements()) {
				this.alias = (String) aliases.nextElement();
				if (store.isKeyEntry(alias)) {
					_key = (PrivateKey) store.getKey(alias, pword.toCharArray());
					Certificate[] chain = store.getCertificateChain(alias);
					_certChain = new X509Certificate[chain.length];
	
					for (int i = 0; i < chain.length; i++) {
						_certChain[i] = (X509Certificate) chain[i];
					}
	
					break;
				}
			}

			if (null == _key) {
				throw new Exception("No key found");
			}
			if (null == _certChain) {
				throw new Exception("No certificate chain found");
			}
		}
		
		this.key = _key;
		this.certChain = _certChain;
	}

	public final X509Certificate[] getCertificateChain() {
		return this.certChain;
	}

	public final PrivateKey getPrivateKey() {
		return this.key;
	}

	public final String getAlias() {
		return this.alias;
	}
}
