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 io.netty.example.securechat;
17  
18  import io.netty.bootstrap.Bootstrap;
19  import io.netty.channel.Channel;
20  import io.netty.channel.ChannelFuture;
21  import io.netty.channel.EventLoopGroup;
22  import io.netty.channel.nio.NioEventLoopGroup;
23  import io.netty.channel.socket.nio.NioSocketChannel;
24  import io.netty.example.telnet.TelnetClient;
25  import io.netty.handler.ssl.SslContext;
26  import io.netty.handler.ssl.SslContextBuilder;
27  import io.netty.handler.ssl.util.InsecureTrustManagerFactory;
28  
29  import java.io.BufferedReader;
30  import java.io.InputStreamReader;
31  
32  /**
33   * Simple SSL chat client modified from {@link TelnetClient}.
34   */
35  public final class SecureChatClient {
36  
37      static final String HOST = System.getProperty("host", "127.0.0.1");
38      static final int PORT = Integer.parseInt(System.getProperty("port", "8992"));
39  
40      public static void main(String[] args) throws Exception {
41          // Configure SSL.
42          final SslContext sslCtx = SslContextBuilder.forClient()
43              .trustManager(InsecureTrustManagerFactory.INSTANCE).build();
44  
45          EventLoopGroup group = new NioEventLoopGroup();
46          try {
47              Bootstrap b = new Bootstrap();
48              b.group(group)
49               .channel(NioSocketChannel.class)
50               .handler(new SecureChatClientInitializer(sslCtx));
51  
52              // Start the connection attempt.
53              Channel ch = b.connect(HOST, PORT).sync().channel();
54  
55              // Read commands from the stdin.
56              ChannelFuture lastWriteFuture = null;
57              BufferedReader in = new BufferedReader(new InputStreamReader(System.in));
58              for (;;) {
59                  String line = in.readLine();
60                  if (line == null) {
61                      break;
62                  }
63  
64                  // Sends the received line to the server.
65                  lastWriteFuture = ch.writeAndFlush(line + "\r\n");
66  
67                  // If user typed the 'bye' command, wait until the server closes
68                  // the connection.
69                  if ("bye".equals(line.toLowerCase())) {
70                      ch.closeFuture().sync();
71                      break;
72                  }
73              }
74  
75              // Wait until all messages are flushed before closing the channel.
76              if (lastWriteFuture != null) {
77                  lastWriteFuture.sync();
78              }
79          } finally {
80              // The connection is closed automatically on shutdown.
81              group.shutdownGracefully();
82          }
83      }
84  }