ServiceKeystorePasswordCallback.java

package gov.samhsa.acs.pep.saml;

import java.io.IOException;
import java.util.HashMap;
import java.util.Map;

import javax.security.auth.callback.Callback;
import javax.security.auth.callback.CallbackHandler;
import javax.security.auth.callback.UnsupportedCallbackException;

import org.apache.wss4j.common.ext.WSPasswordCallback;

public class ServiceKeystorePasswordCallback implements CallbackHandler {

	private Map passwords = new HashMap();

	public ServiceKeystorePasswordCallback() {
		passwords.put("myservicekey", "skpass");
	}

	@Override
	public void handle(Callback[] callbacks) throws IOException,
			UnsupportedCallbackException {
		for (int i = 0; i < callbacks.length; i++) {
			WSPasswordCallback pc = (WSPasswordCallback) callbacks[i];
			int usage = pc.getUsage();
			if (usage == WSPasswordCallback.DECRYPT
					|| usage == WSPasswordCallback.SIGNATURE) {
				String pass = (String) passwords.get(pc.getIdentifier());
				if (pass != null) {
					pc.setPassword(pass);
					return;
				}
			}
		}
	}
}