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 }