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