View Javadoc

1   /*
2    *  Licensed to the Apache Software Foundation (ASF) under one
3    *  or more contributor license agreements.  See the NOTICE file
4    *  distributed with this work for additional information
5    *  regarding copyright ownership.  The ASF licenses this file
6    *  to you under the Apache License, Version 2.0 (the
7    *  "License"); you may not use this file except in compliance
8    *  with the License.  You may obtain a copy of the License at
9    *
10   *    http://www.apache.org/licenses/LICENSE-2.0
11   *
12   *  Unless required by applicable law or agreed to in writing,
13   *  software distributed under the License is distributed on an
14   *  "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
15   *  KIND, either express or implied.  See the License for the
16   *  specific language governing permissions and limitations
17   *  under the License.
18   *
19   */
20  package org.apache.mina.example.proxy;
21  
22  import java.net.SocketAddress;
23  
24  import org.apache.mina.core.RuntimeIoException;
25  import org.apache.mina.core.future.ConnectFuture;
26  import org.apache.mina.core.future.IoFutureListener;
27  import org.apache.mina.core.service.IoConnector;
28  import org.apache.mina.core.session.IoSession;
29  
30  /**
31   * Handles the client to proxy part of the proxied connection.
32   *
33   * @author The Apache MINA Project (dev@mina.apache.org)
34   * @version $Rev$, $Date$
35   *
36   */
37  public class ClientToProxyIoHandler extends AbstractProxyIoHandler {
38      private final ServerToProxyIoHandler connectorHandler = new ServerToProxyIoHandler();
39  
40      private final IoConnector connector;
41  
42      private final SocketAddress remoteAddress;
43  
44      public ClientToProxyIoHandler(IoConnector connector,
45              SocketAddress remoteAddress) {
46          this.connector = connector;
47          this.remoteAddress = remoteAddress;
48          connector.setHandler(connectorHandler);
49      }
50  
51      @Override
52      public void sessionOpened(final IoSession session) throws Exception {
53  
54          connector.connect(remoteAddress).addListener(new IoFutureListener<ConnectFuture>() {
55              public void operationComplete(ConnectFuture future) {
56                  try {
57                      future.getSession().setAttribute(OTHER_IO_SESSION, session);
58                      session.setAttribute(OTHER_IO_SESSION, future.getSession());
59                      IoSession session2 = future.getSession();
60                      session2.resumeRead();
61                      session2.resumeWrite();
62                  } catch (RuntimeIoException e) {
63                      // Connect failed
64                      session.close(true);
65                  } finally {
66                  	session.resumeRead();
67                      session.resumeWrite();
68                  }
69              }
70          });
71      }
72  }