Repository: mina-sshd Updated Branches: refs/heads/master 16349da3b -> d07089cea
http://git-wip-us.apache.org/repos/asf/mina-sshd/blob/d07089ce/sshd-mina/src/main/java/org/apache/sshd/common/io/mina/MinaSession.java ---------------------------------------------------------------------- diff --git a/sshd-mina/src/main/java/org/apache/sshd/common/io/mina/MinaSession.java b/sshd-mina/src/main/java/org/apache/sshd/common/io/mina/MinaSession.java new file mode 100644 index 0000000..259f044 --- /dev/null +++ b/sshd-mina/src/main/java/org/apache/sshd/common/io/mina/MinaSession.java @@ -0,0 +1,183 @@ +/* + * 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.sshd.common.io.mina; + +import java.net.SocketAddress; +import java.util.Objects; + +import org.apache.mina.core.buffer.IoBuffer; +import org.apache.mina.core.future.IoFutureListener; +import org.apache.mina.core.future.WriteFuture; +import org.apache.sshd.common.Closeable; +import org.apache.sshd.common.future.CloseFuture; +import org.apache.sshd.common.future.DefaultCloseFuture; +import org.apache.sshd.common.future.SshFutureListener; +import org.apache.sshd.common.io.AbstractIoWriteFuture; +import org.apache.sshd.common.io.IoService; +import org.apache.sshd.common.io.IoSession; +import org.apache.sshd.common.io.IoWriteFuture; +import org.apache.sshd.common.util.NumberUtils; +import org.apache.sshd.common.util.buffer.Buffer; +import org.apache.sshd.common.util.closeable.AbstractInnerCloseable; +import org.apache.sshd.common.util.closeable.IoBaseCloseable; + +/** + * @author <a href="mailto:d...@mina.apache.org">Apache MINA SSHD Project</a> + */ +public class MinaSession extends AbstractInnerCloseable implements IoSession { + private final MinaService service; + private final org.apache.mina.core.session.IoSession session; + private final Object sessionWriteId; + + public MinaSession(MinaService service, org.apache.mina.core.session.IoSession session) { + this.service = service; + this.session = session; + this.sessionWriteId = Objects.toString(session); + } + + public org.apache.mina.core.session.IoSession getSession() { + return session; + } + + public void suspend() { + session.suspendRead(); + session.suspendWrite(); + } + + @Override + public Object getAttribute(Object key) { + return session.getAttribute(key); + } + + @Override + public Object setAttribute(Object key, Object value) { + return session.setAttribute(key, value); + } + + @Override + public Object removeAttribute(Object key) { + return session.removeAttribute(key); + } + + @Override + public SocketAddress getRemoteAddress() { + return session.getRemoteAddress(); + } + + @Override + public SocketAddress getLocalAddress() { + return session.getLocalAddress(); + } + + @Override + public long getId() { + return session.getId(); + } + + @Override + protected Closeable getInnerCloseable() { + return new IoBaseCloseable() { + @SuppressWarnings("synthetic-access") + private final DefaultCloseFuture future = new DefaultCloseFuture(MinaSession.this.toString(), lock); + + @SuppressWarnings("synthetic-access") + @Override + public boolean isClosing() { + return session.isClosing(); + } + + @SuppressWarnings("synthetic-access") + @Override + public boolean isClosed() { + return !session.isConnected(); + } + + @Override + public void addCloseFutureListener(SshFutureListener<CloseFuture> listener) { + future.addListener(listener); + } + + @Override + public void removeCloseFutureListener(SshFutureListener<CloseFuture> listener) { + future.removeListener(listener); + } + + @SuppressWarnings("synthetic-access") + @Override + public org.apache.sshd.common.future.CloseFuture close(boolean immediately) { + org.apache.mina.core.future.CloseFuture cf = immediately ? session.closeNow() : session.closeOnFlush(); + cf.addListener(f -> future.setValue(Boolean.TRUE)); + return future; + } + }; + } + + // NOTE !!! data buffer may NOT be re-used when method returns - at least until IoWriteFuture is signalled + public IoWriteFuture write(byte[] data) { + return write(data, 0, NumberUtils.length(data)); + } + + // NOTE !!! data buffer may NOT be re-used when method returns - at least until IoWriteFuture is signalled + public IoWriteFuture write(byte[] data, int offset, int len) { + return write(IoBuffer.wrap(data, offset, len)); + } + + @Override // NOTE !!! data buffer may NOT be re-used when method returns - at least until IoWriteFuture is signalled + public IoWriteFuture writePacket(Buffer buffer) { + return write(MinaSupport.asIoBuffer(buffer)); + } + + // NOTE !!! data buffer may NOT be re-used when method returns - at least until IoWriteFuture is signalled + public IoWriteFuture write(IoBuffer buffer) { + Future future = new Future(sessionWriteId, null); + session.write(buffer).addListener((IoFutureListener<WriteFuture>) cf -> { + Throwable t = cf.getException(); + if (t != null) { + future.setException(t); + } else { + future.setWritten(); + } + }); + return future; + } + + public static class Future extends AbstractIoWriteFuture { + public Future(Object id, Object lock) { + super(id, lock); + } + + public void setWritten() { + setValue(Boolean.TRUE); + } + + public void setException(Throwable exception) { + setValue(Objects.requireNonNull(exception, "No exception specified")); + } + } + + @Override + public IoService getService() { + return service; + } + + @Override + public String toString() { + return getClass().getSimpleName() + "[local=" + session.getLocalAddress() + ", remote=" + session.getRemoteAddress() + "]"; + } +} http://git-wip-us.apache.org/repos/asf/mina-sshd/blob/d07089ce/sshd-mina/src/main/java/org/apache/sshd/common/io/mina/MinaSupport.java ---------------------------------------------------------------------- diff --git a/sshd-mina/src/main/java/org/apache/sshd/common/io/mina/MinaSupport.java b/sshd-mina/src/main/java/org/apache/sshd/common/io/mina/MinaSupport.java new file mode 100644 index 0000000..cb943ba --- /dev/null +++ b/sshd-mina/src/main/java/org/apache/sshd/common/io/mina/MinaSupport.java @@ -0,0 +1,48 @@ +/* + * 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.sshd.common.io.mina; + +import org.apache.mina.core.buffer.IoBuffer; +import org.apache.sshd.common.util.Readable; +import org.apache.sshd.common.util.buffer.Buffer; + +public final class MinaSupport { + + private MinaSupport() { + throw new UnsupportedOperationException("No instance allowed"); + } + + public static Readable asReadable(final IoBuffer buffer) { + return new Readable() { + @Override + public int available() { + return buffer.remaining(); + } + + @Override + public void getRawBytes(byte[] data, int offset, int len) { + buffer.get(data, offset, len); + } + }; + } + + public static IoBuffer asIoBuffer(Buffer buffer) { + return IoBuffer.wrap(buffer.array(), buffer.rpos(), buffer.available()); + } +}