View Javadoc
1   /*
2    * Copyright 2012 The Netty Project
3    *
4    * The Netty Project licenses this file to you under the Apache License,
5    * version 2.0 (the "License"); you may not use this file except in compliance
6    * with the License. You may obtain a copy of the License at:
7    *
8    *   http://www.apache.org/licenses/LICENSE-2.0
9    *
10   * Unless required by applicable law or agreed to in writing, software
11   * distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
12   * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
13   * License for the specific language governing permissions and limitations
14   * under the License.
15   */
16  package org.jboss.netty.example.securechat;
17  
18  import org.jboss.netty.bootstrap.ClientBootstrap;
19  import org.jboss.netty.channel.Channel;
20  import org.jboss.netty.channel.ChannelFuture;
21  import org.jboss.netty.channel.socket.nio.NioClientSocketChannelFactory;
22  import org.jboss.netty.example.telnet.TelnetClient;
23  import org.jboss.netty.handler.ssl.SslContext;
24  import org.jboss.netty.handler.ssl.util.InsecureTrustManagerFactory;
25  
26  import java.io.BufferedReader;
27  import java.io.InputStreamReader;
28  import java.net.InetSocketAddress;
29  import java.util.concurrent.Executors;
30  
31  /**
32   * Simple SSL chat client modified from {@link TelnetClient}.
33   */
34  public final class SecureChatClient {
35  
36      static final String HOST = System.getProperty("host", "127.0.0.1");
37      static final int PORT = Integer.parseInt(System.getProperty("port", "8992"));
38  
39      public static void main(String[] args) throws Exception {
40  
41          // Configure SSL.
42          final SslContext sslCtx = SslContext.newClientContext(InsecureTrustManagerFactory.INSTANCE);
43  
44          // Configure the client.
45          ClientBootstrap bootstrap = new ClientBootstrap(
46                  new NioClientSocketChannelFactory(
47                          Executors.newCachedThreadPool(),
48                          Executors.newCachedThreadPool()));
49  
50          try {
51              // Configure the pipeline factory.
52              bootstrap.setPipelineFactory(new SecureChatClientPipelineFactory(sslCtx));
53  
54              // Start the connection attempt.
55              ChannelFuture future = bootstrap.connect(new InetSocketAddress(HOST, PORT));
56  
57              // Wait until the connection attempt succeeds or fails.
58              Channel channel = future.sync().getChannel();
59  
60              // Read commands from the stdin.
61              ChannelFuture lastWriteFuture = null;
62              BufferedReader in = new BufferedReader(new InputStreamReader(System.in));
63              for (;;) {
64                  String line = in.readLine();
65                  if (line == null) {
66                      break;
67                  }
68  
69                  // Sends the received line to the server.
70                  lastWriteFuture = channel.write(line + "\r\n");
71  
72                  // If user typed the 'bye' command, wait until the server closes
73                  // the connection.
74                  if ("bye".equals(line.toLowerCase())) {
75                      channel.getCloseFuture().sync();
76                      break;
77                  }
78              }
79  
80              // Wait until all messages are flushed before closing the channel.
81              if (lastWriteFuture != null) {
82                  lastWriteFuture.sync();
83              }
84  
85              // Close the connection.  Make sure the close operation ends because
86              // all I/O operations are asynchronous in Netty.
87              channel.close().sync();
88          } finally {
89              // Shut down all thread pools to exit.
90              bootstrap.releaseExternalResources();
91          }
92      }
93  }