org.apache.qpid.proton.engine.impl.SaslClientImpl Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of qpid-proton Show documentation
The newest version!
package org.apache.qpid.proton.engine.impl;
/*
 *
 * 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.
 *
*/


import org.apache.qpid.proton.codec.WritableBuffer;
import org.apache.qpid.proton.engine.SaslClient;
import org.apache.qpid.proton.type.Binary;
import org.apache.qpid.proton.type.Symbol;
import org.apache.qpid.proton.type.security.*;

public class SaslClientImpl extends SaslImpl implements SaslClient, SaslFrameBody.SaslFrameBodyHandler
{

    private SaslOutcome _outcome = SaslOutcome.PN_SASL_NONE;
    private SaslState _state = SaslState.PN_SASL_IDLE;
    private Symbol[] _mechanisms;
    private Symbol _chosenMechanism;
    private boolean _done;
    private String _hostname;

    public SaslClientImpl()
    {
        super();

    }


    public SaslState getState()
    {
        return _state;
    }

    public void setMechanisms(String[] mechanisms)
    {
        assert mechanisms != null;
        assert mechanisms.length == 1;

        _chosenMechanism = Symbol.valueOf(mechanisms[0]);
    }

    public String[] getRemoteMechanisms()
    {
        String[] remoteMechanisms = new String[_mechanisms.length];
        for(int i = 0; i < _mechanisms.length; i++)
        {
            remoteMechanisms[i] = _mechanisms[i].toString();
        }
        return remoteMechanisms;
    }

    public void handleMechanisms(SaslMechanisms saslMechanisms, Binary payload, Void context)
    {
        _mechanisms = saslMechanisms.getSaslServerMechanisms();

    }

    public void handleInit(SaslInit saslInit, Binary payload, Void context)
    {
        // TODO - error
    }

    public void handleChallenge(SaslChallenge saslChallenge, Binary payload, Void context)
    {
        setPending(saslChallenge.getChallenge()  == null ? null : saslChallenge.getChallenge().asByteBuffer());
    }

    public void handleResponse(SaslResponse saslResponse, Binary payload, Void context)
    {
        // TODO - error
    }

    public void handleOutcome(org.apache.qpid.proton.type.security.SaslOutcome saslOutcome,
                              Binary payload,
                              Void context)
    {
        for(SaslOutcome outcome : SaslOutcome.values())
        {
            if(outcome.getCode() == saslOutcome.getCode().byteValue())
            {
                _outcome = outcome;
                break;
            }
        }
        _done = true;
    }

    @Override
    public boolean isDone()
    {
        return _done;
    }

    @Override
    protected int process(WritableBuffer buffer)
    {
        int written = processHeader(buffer);

        if(getState() == SaslState.PN_SASL_IDLE && _chosenMechanism != null)
        {
            written += processInit(buffer);
            _state = SaslState.PN_SASL_STEP;
        }
        if(getState() == SaslState.PN_SASL_STEP && getChallengeResponse() != null)
        {
            written += processResponse(buffer);
        }
        return written;
    }

    private int processResponse(WritableBuffer buffer)
    {
        SaslResponse response = new SaslResponse();
        response.setResponse(getChallengeResponse());
        setChallengeResponse(null);
        return writeFrame(buffer, response);
    }

    private int processInit(WritableBuffer buffer)
    {
        SaslInit init = new SaslInit();
        init.setHostname(_hostname);
        init.setMechanism(_chosenMechanism);
        if(getChallengeResponse() != null)
        {
            init.setInitialResponse(getChallengeResponse());
            setChallengeResponse(null);
        }
        return writeFrame(buffer, init);
    }

    public void plain(String username, String password)
    {
        _chosenMechanism = Symbol.valueOf("PLAIN");
        byte[] usernameBytes = username.getBytes();
        byte[] passwordBytes = password.getBytes();
        byte[] data = new byte[usernameBytes.length+passwordBytes.length+2];
        System.arraycopy(usernameBytes, 0, data, 1, usernameBytes.length);
        System.arraycopy(passwordBytes, 0, data, 2+usernameBytes.length, passwordBytes.length);

        setChallengeResponse(new Binary(data));

    }

    public SaslOutcome getOutcome()
    {
        return _outcome;
    }
}