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   */
35  public class ClientToProxyIoHandler extends AbstractProxyIoHandler {
36      private final ServerToProxyIoHandler connectorHandler = new ServerToProxyIoHandler();
37  
38      private final IoConnector connector;
39  
40      private final SocketAddress remoteAddress;
41  
42      public ClientToProxyIoHandler(IoConnector connector,
43              SocketAddress remoteAddress) {
44          this.connector = connector;
45          this.remoteAddress = remoteAddress;
46          connector.setHandler(connectorHandler);
47      }
48  
49      @Override
50      public void sessionOpened(final IoSession session) throws Exception {
51  
52          connector.connect(remoteAddress).addListener(new IoFutureListener<ConnectFuture>() {
53              public void operationComplete(ConnectFuture future) {
54                  try {
55                      future.getSession().setAttribute(OTHER_IO_SESSION, session);
56                      session.setAttribute(OTHER_IO_SESSION, future.getSession());
57                      IoSession session2 = future.getSession();
58                      session2.resumeRead();
59                      session2.resumeWrite();
60                  } catch (RuntimeIoException e) {
61                      // Connect failed
62                      session.close(true);
63                  } finally {
64                      session.resumeRead();
65                      session.resumeWrite();
66                  }
67              }
68          });
69      }
70  }