svn commit: r1340432 - /mina/sshd/trunk/sshd-core/src/main/java/org/apache/sshd/server/shell/InvertedShellWrapper.java

[gnodet]

Author: gnodet
Date: Sat May 19 12:20:51 2012
New Revision: 1340432

URL: http://svn.apache.org/viewvc?rev=1340432&view=rev
Log:
[SSHD-163] Allow InvertedShellWrapper to implement SessionAware

Modified:
    
mina/sshd/trunk/sshd-core/src/main/java/org/apache/sshd/server/shell/InvertedShellWrapper.java

Modified: 
mina/sshd/trunk/sshd-core/src/main/java/org/apache/sshd/server/shell/InvertedShellWrapper.java
URL: 
http://svn.apache.org/viewvc/mina/sshd/trunk/sshd-core/src/main/java/org/apache/sshd/server/shell/InvertedShellWrapper.java?rev=1340432&r1=1340431&r2=1340432&view=diff
==============================================================================
--- 
mina/sshd/trunk/sshd-core/src/main/java/org/apache/sshd/server/shell/InvertedShellWrapper.java
 (original)
+++ 
mina/sshd/trunk/sshd-core/src/main/java/org/apache/sshd/server/shell/InvertedShellWrapper.java
 Sat May 19 12:20:51 2012
@@ -28,6 +28,8 @@ import org.apache.mina.util.NamePreservi
 import org.apache.sshd.server.Command;
 import org.apache.sshd.server.Environment;
 import org.apache.sshd.server.ExitCallback;
+import org.apache.sshd.server.SessionAware;
+import org.apache.sshd.server.session.ServerSession;
 
 /**
  * A shell implementation that wraps an instance of {@link InvertedShell}
@@ -38,7 +40,7 @@ import org.apache.sshd.server.ExitCallba
  *
  * @author <a href="mailto:d...@mina.apache.org";>Apache MINA SSHD Project</a>
  */
-public class InvertedShellWrapper implements Command {
+public class InvertedShellWrapper implements Command, SessionAware {
 
     /** default buffer size for the IO pumps. */
     public static final int DEFAULT_BUFFER_SIZE = 8192;
@@ -88,6 +90,12 @@ public class InvertedShellWrapper implem
         this.callback = callback;
     }
 
+    public void setSession(ServerSession session) {
+        if (shell instanceof SessionAware) {
+            ((SessionAware) shell).setSession(session);
+        }
+    }
+
     public void start(Environment env) throws IOException {
         // TODO propagate the Environment itself and support signal sending.
         shell.start(env.getEnv());