/*
* 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.qpid.server.security.auth.manager;
import java.io.IOException;
import java.security.Principal;
import java.util.Collections;
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.NameCallback;
import javax.security.auth.callback.PasswordCallback;
import javax.security.auth.callback.UnsupportedCallbackException;
import javax.security.sasl.AuthorizeCallback;
import javax.security.sasl.Sasl;
import javax.security.sasl.SaslException;
import javax.security.sasl.SaslServer;
import org.apache.log4j.Logger;
import org.apache.qpid.server.security.auth.AuthenticationResult;
import org.apache.qpid.server.security.auth.UsernamePrincipal;
import org.apache.qpid.server.security.auth.sasl.plain.PlainPasswordCallback;
import org.apache.qpid.server.security.auth.sasl.plain.PlainSaslServer;
public class SimpleAuthenticationManager implements AuthenticationManager
{
private static final Logger _logger = Logger.getLogger(SimpleAuthenticationManager.class);
private static final String PLAIN_MECHANISM = "PLAIN";
private static final String CRAM_MD5_MECHANISM = "CRAM-MD5";
private Map<String, String> _users;
public SimpleAuthenticationManager(String userName, String userPassword)
{
this(Collections.singletonMap(userName, userPassword));
}
public SimpleAuthenticationManager(Map<String, String> users)
{
_users = new HashMap<String, String>(users);
}
@Override
public void initialise()
{
}
@Override
public String getMechanisms()
{
return PLAIN_MECHANISM + " " + CRAM_MD5_MECHANISM;
}
@Override
public SaslServer createSaslServer(String mechanism, String localFQDN, Principal externalPrincipal) throws SaslException
{
if (PLAIN_MECHANISM.equals(mechanism))
{
return new PlainSaslServer(new SimplePlainCallbackHandler());
}
else if (CRAM_MD5_MECHANISM.equals(mechanism))
{
return Sasl.createSaslServer(mechanism, "AMQP", localFQDN, null, new SimpleCramMd5CallbackHandler());
}
else
{
throw new SaslException("Unknown mechanism: " + mechanism);
}
}
@Override
public AuthenticationResult authenticate(SaslServer server, byte[] response)
{
try
{
// Process response from the client
byte[] challenge = server.evaluateResponse(response != null ? response : new byte[0]);
if (server.isComplete())
{
String authorizationID = server.getAuthorizationID();
_logger.debug("Authenticated as " + authorizationID);
return new AuthenticationResult(new UsernamePrincipal(authorizationID));
}
else
{
return new AuthenticationResult(challenge, AuthenticationResult.AuthenticationStatus.CONTINUE);
}
}
catch (SaslException e)
{
return new AuthenticationResult(AuthenticationResult.AuthenticationStatus.ERROR, e);
}
}
@Override
public AuthenticationResult authenticate(String username, String password)
{
if (_users.containsKey(username))
{
String userPassword = _users.get(username);
if (userPassword.equals(password))
{
return new AuthenticationResult(new UsernamePrincipal(username));
}
}
return new AuthenticationResult(AuthenticationResult.AuthenticationStatus.ERROR);
}
@Override
public void close()
{
}
@Override
public void onCreate()
{
// nothing to do, no external resource is required
}
@Override
public void onDelete()
{
// nothing to do, no external resource is used
}
private class SimpleCramMd5CallbackHandler implements CallbackHandler
{
public void handle(Callback[] callbacks) throws IOException, UnsupportedCallbackException
{
String username = null;
for (Callback callback : callbacks)
{
if (callback instanceof NameCallback)
{
username = ((NameCallback) callback).getDefaultName();
}
else if (callback instanceof PasswordCallback)
{
if (_users.containsKey(username))
{
String password = _users.get(username);
((PasswordCallback) callback).setPassword(password.toCharArray());
}
else
{
throw new SaslException("Authentication failed");
}
}
else if (callback instanceof AuthorizeCallback)
{
((AuthorizeCallback) callback).setAuthorized(true);
}
else
{
throw new UnsupportedCallbackException(callback);
}
}
}
}
private class SimplePlainCallbackHandler implements CallbackHandler
{
public void handle(Callback[] callbacks) throws IOException, UnsupportedCallbackException
{
String username = null;
for (Callback callback : callbacks)
{
if (callback instanceof NameCallback)
{
username = ((NameCallback) callback).getDefaultName();
}
else if (callback instanceof PlainPasswordCallback)
{
if (_users.containsKey(username))
{
PlainPasswordCallback plainPasswordCallback = (PlainPasswordCallback) callback;
String password = plainPasswordCallback.getPlainPassword();
plainPasswordCallback.setAuthenticated(password.equals(_users.get(username)));
}
}
else if (callback instanceof AuthorizeCallback)
{
((AuthorizeCallback) callback).setAuthorized(true);
}
else
{
throw new UnsupportedCallbackException(callback);
}
}
}
}
}