org.xnio.sasl.SaslWrappingConduit Maven / Gradle / Ivy
/*
* JBoss, Home of Professional Open Source
*
* Copyright 2013 Red Hat, Inc. and/or its affiliates.
*
* Licensed 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.xnio.sasl;
import java.io.IOException;
import java.nio.ByteBuffer;
import org.xnio.Buffers;
import org.xnio.conduits.AbstractMessageSinkConduit;
import org.xnio.conduits.Conduits;
import org.xnio.conduits.MessageSinkConduit;
/**
* @author David M. Lloyd
*/
public final class SaslWrappingConduit extends AbstractMessageSinkConduit implements MessageSinkConduit {
private final SaslWrapper wrapper;
private ByteBuffer buffer;
public SaslWrappingConduit(final MessageSinkConduit next, final SaslWrapper wrapper) {
super(next);
this.wrapper = wrapper;
}
public boolean send(final ByteBuffer src) throws IOException {
if (! doSend()) {
return false;
}
ByteBuffer wrapped = ByteBuffer.wrap(wrapper.wrap(src));
if (! next.send(wrapped)) {
buffer = wrapped;
}
return true;
}
public boolean send(final ByteBuffer[] srcs, final int offs, final int len) throws IOException {
if (! doSend()) {
return false;
}
final ByteBuffer wrapped = ByteBuffer.wrap(wrapper.wrap(Buffers.take(srcs, offs, len)));
if (! next.send(wrapped)) {
this.buffer = wrapped;
}
return true;
}
@Override
public boolean sendFinal(ByteBuffer src) throws IOException {
return Conduits.sendFinalBasic(this, src);
}
@Override
public boolean sendFinal(ByteBuffer[] srcs, int offs, int len) throws IOException {
return Conduits.sendFinalBasic(this, srcs, offs, len);
}
private boolean doSend() throws IOException {
final ByteBuffer buffer = this.buffer;
if (buffer != null && next.send(buffer)) {
this.buffer = null;
return true;
}
return false;
}
public boolean flush() throws IOException {
return doSend() && next.flush();
}
}