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.http.snoop;
17  
18  import io.netty.bootstrap.Bootstrap;
19  import io.netty.channel.Channel;
20  import io.netty.channel.EventLoopGroup;
21  import io.netty.channel.nio.NioEventLoopGroup;
22  import io.netty.channel.socket.nio.NioSocketChannel;
23  import io.netty.handler.codec.http.ClientCookieEncoder;
24  import io.netty.handler.codec.http.DefaultCookie;
25  import io.netty.handler.codec.http.DefaultFullHttpRequest;
26  import io.netty.handler.codec.http.HttpHeaderNames;
27  import io.netty.handler.codec.http.HttpHeaderValues;
28  import io.netty.handler.codec.http.HttpMethod;
29  import io.netty.handler.codec.http.HttpRequest;
30  import io.netty.handler.codec.http.HttpVersion;
31  import io.netty.handler.ssl.SslContext;
32  import io.netty.handler.ssl.util.InsecureTrustManagerFactory;
33  
34  import java.net.URI;
35  
36  /**
37   * A simple HTTP client that prints out the content of the HTTP response to
38   * {@link System#out} to test {@link HttpSnoopServer}.
39   */
40  public final class HttpSnoopClient {
41  
42      static final String URL = System.getProperty("url", "http://127.0.0.1:8080/");
43  
44      public static void main(String[] args) throws Exception {
45          URI uri = new URI(URL);
46          String scheme = uri.getScheme() == null? "http" : uri.getScheme();
47          String host = uri.getHost() == null? "127.0.0.1" : uri.getHost();
48          int port = uri.getPort();
49          if (port == -1) {
50              if ("http".equalsIgnoreCase(scheme)) {
51                  port = 80;
52              } else if ("https".equalsIgnoreCase(scheme)) {
53                  port = 443;
54              }
55          }
56  
57          if (!"http".equalsIgnoreCase(scheme) && !"https".equalsIgnoreCase(scheme)) {
58              System.err.println("Only HTTP(S) is supported.");
59              return;
60          }
61  
62          // Configure SSL context if necessary.
63          final boolean ssl = "https".equalsIgnoreCase(scheme);
64          final SslContext sslCtx;
65          if (ssl) {
66              sslCtx = SslContext.newClientContext(InsecureTrustManagerFactory.INSTANCE);
67          } else {
68              sslCtx = null;
69          }
70  
71          // Configure the client.
72          EventLoopGroup group = new NioEventLoopGroup();
73          try {
74              Bootstrap b = new Bootstrap();
75              b.group(group)
76               .channel(NioSocketChannel.class)
77               .handler(new HttpSnoopClientInitializer(sslCtx));
78  
79              // Make the connection attempt.
80              Channel ch = b.connect(host, port).sync().channel();
81  
82              // Prepare the HTTP request.
83              HttpRequest request = new DefaultFullHttpRequest(
84                      HttpVersion.HTTP_1_1, HttpMethod.GET, uri.getRawPath());
85              request.headers().set(HttpHeaderNames.HOST, host);
86              request.headers().set(HttpHeaderNames.CONNECTION, HttpHeaderValues.CLOSE);
87              request.headers().set(HttpHeaderNames.ACCEPT_ENCODING, HttpHeaderValues.GZIP);
88  
89              // Set some example cookies.
90              request.headers().set(
91                      HttpHeaderNames.COOKIE,
92                      ClientCookieEncoder.encode(
93                              new DefaultCookie("my-cookie", "foo"),
94                              new DefaultCookie("another-cookie", "bar")));
95  
96              // Send the HTTP request.
97              ch.writeAndFlush(request);
98  
99              // Wait for the server to close the connection.
100             ch.closeFuture().sync();
101         } finally {
102             // Shut down executor threads to exit.
103             group.shutdownGracefully();
104         }
105     }
106 }