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.http.snoop;
17
18 import io.netty.buffer.ByteBuf;
19 import io.netty.buffer.Unpooled;
20 import io.netty.channel.ChannelFutureListener;
21 import io.netty.channel.ChannelHandlerContext;
22 import io.netty.channel.SimpleChannelInboundHandler;
23 import io.netty.handler.codec.DecoderResult;
24 import io.netty.handler.codec.http.DefaultFullHttpResponse;
25 import io.netty.handler.codec.http.FullHttpResponse;
26 import io.netty.handler.codec.http.HttpContent;
27 import io.netty.handler.codec.http.HttpHeaderNames;
28 import io.netty.handler.codec.http.HttpUtil;
29 import io.netty.handler.codec.http.HttpHeaderValues;
30 import io.netty.handler.codec.http.HttpHeaders;
31 import io.netty.handler.codec.http.HttpObject;
32 import io.netty.handler.codec.http.HttpRequest;
33 import io.netty.handler.codec.http.LastHttpContent;
34 import io.netty.handler.codec.http.QueryStringDecoder;
35 import io.netty.handler.codec.http.cookie.Cookie;
36 import io.netty.handler.codec.http.cookie.ServerCookieDecoder;
37 import io.netty.handler.codec.http.cookie.ServerCookieEncoder;
38 import io.netty.util.CharsetUtil;
39
40 import java.util.List;
41 import java.util.Map;
42 import java.util.Map.Entry;
43 import java.util.Set;
44
45 import static io.netty.handler.codec.http.HttpResponseStatus.*;
46 import static io.netty.handler.codec.http.HttpVersion.*;
47
48 public class HttpSnoopServerHandler extends SimpleChannelInboundHandler<Object> {
49
50 private HttpRequest request;
51 /** Buffer that stores the response content */
52 private final StringBuilder buf = new StringBuilder();
53
54 @Override
55 public void channelReadComplete(ChannelHandlerContext ctx) {
56 ctx.flush();
57 }
58
59 @Override
60 protected void channelRead0(ChannelHandlerContext ctx, Object msg) {
61 if (msg instanceof HttpRequest) {
62 HttpRequest request = this.request = (HttpRequest) msg;
63
64 if (HttpUtil.is100ContinueExpected(request)) {
65 send100Continue(ctx);
66 }
67
68 buf.setLength(0);
69 buf.append("WELCOME TO THE WILD WILD WEB SERVER\r\n");
70 buf.append("===================================\r\n");
71
72 buf.append("VERSION: ").append(request.protocolVersion()).append("\r\n");
73 buf.append("HOSTNAME: ").append(request.headers().get(HttpHeaderNames.HOST, "unknown")).append("\r\n");
74 buf.append("REQUEST_URI: ").append(request.uri()).append("\r\n\r\n");
75
76 HttpHeaders headers = request.headers();
77 if (!headers.isEmpty()) {
78 for (Map.Entry<String, String> h: headers) {
79 CharSequence key = h.getKey();
80 CharSequence value = h.getValue();
81 buf.append("HEADER: ").append(key).append(" = ").append(value).append("\r\n");
82 }
83 buf.append("\r\n");
84 }
85
86 QueryStringDecoder queryStringDecoder = new QueryStringDecoder(request.uri());
87 Map<String, List<String>> params = queryStringDecoder.parameters();
88 if (!params.isEmpty()) {
89 for (Entry<String, List<String>> p: params.entrySet()) {
90 String key = p.getKey();
91 List<String> vals = p.getValue();
92 for (String val : vals) {
93 buf.append("PARAM: ").append(key).append(" = ").append(val).append("\r\n");
94 }
95 }
96 buf.append("\r\n");
97 }
98
99 appendDecoderResult(buf, request);
100 }
101
102 if (msg instanceof HttpContent) {
103 HttpContent httpContent = (HttpContent) msg;
104
105 ByteBuf content = httpContent.content();
106 if (content.isReadable()) {
107 buf.append("CONTENT: ");
108 buf.append(content.toString(CharsetUtil.UTF_8));
109 buf.append("\r\n");
110 appendDecoderResult(buf, request);
111 }
112
113 if (msg instanceof LastHttpContent) {
114 buf.append("END OF CONTENT\r\n");
115
116 LastHttpContent trailer = (LastHttpContent) msg;
117 if (!trailer.trailingHeaders().isEmpty()) {
118 buf.append("\r\n");
119 for (CharSequence name: trailer.trailingHeaders().names()) {
120 for (CharSequence value: trailer.trailingHeaders().getAll(name)) {
121 buf.append("TRAILING HEADER: ");
122 buf.append(name).append(" = ").append(value).append("\r\n");
123 }
124 }
125 buf.append("\r\n");
126 }
127
128 if (!writeResponse(trailer, ctx)) {
129 // If keep-alive is off, close the connection once the content is fully written.
130 ctx.writeAndFlush(Unpooled.EMPTY_BUFFER).addListener(ChannelFutureListener.CLOSE);
131 }
132 }
133 }
134 }
135
136 private static void appendDecoderResult(StringBuilder buf, HttpObject o) {
137 DecoderResult result = o.decoderResult();
138 if (result.isSuccess()) {
139 return;
140 }
141
142 buf.append(".. WITH DECODER FAILURE: ");
143 buf.append(result.cause());
144 buf.append("\r\n");
145 }
146
147 private boolean writeResponse(HttpObject currentObj, ChannelHandlerContext ctx) {
148 // Decide whether to close the connection or not.
149 boolean keepAlive = HttpUtil.isKeepAlive(request);
150 // Build the response object.
151 FullHttpResponse response = new DefaultFullHttpResponse(
152 HTTP_1_1, currentObj.decoderResult().isSuccess()? OK : BAD_REQUEST,
153 Unpooled.copiedBuffer(buf.toString(), CharsetUtil.UTF_8));
154
155 response.headers().set(HttpHeaderNames.CONTENT_TYPE, "text/plain; charset=UTF-8");
156
157 if (keepAlive) {
158 // Add 'Content-Length' header only for a keep-alive connection.
159 response.headers().setInt(HttpHeaderNames.CONTENT_LENGTH, response.content().readableBytes());
160 // Add keep alive header as per:
161 // - https://www.w3.org/Protocols/HTTP/1.1/draft-ietf-http-v11-spec-01.html#Connection
162 response.headers().set(HttpHeaderNames.CONNECTION, HttpHeaderValues.KEEP_ALIVE);
163 }
164
165 // Encode the cookie.
166 String cookieString = request.headers().get(HttpHeaderNames.COOKIE);
167 if (cookieString != null) {
168 Set<Cookie> cookies = ServerCookieDecoder.STRICT.decode(cookieString);
169 if (!cookies.isEmpty()) {
170 // Reset the cookies if necessary.
171 for (Cookie cookie: cookies) {
172 response.headers().add(HttpHeaderNames.SET_COOKIE, ServerCookieEncoder.STRICT.encode(cookie));
173 }
174 }
175 } else {
176 // Browser sent no cookie. Add some.
177 response.headers().add(HttpHeaderNames.SET_COOKIE, ServerCookieEncoder.STRICT.encode("key1", "value1"));
178 response.headers().add(HttpHeaderNames.SET_COOKIE, ServerCookieEncoder.STRICT.encode("key2", "value2"));
179 }
180
181 // Write the response.
182 ctx.write(response);
183
184 return keepAlive;
185 }
186
187 private static void send100Continue(ChannelHandlerContext ctx) {
188 FullHttpResponse response = new DefaultFullHttpResponse(HTTP_1_1, CONTINUE, Unpooled.EMPTY_BUFFER);
189 ctx.write(response);
190 }
191
192 @Override
193 public void exceptionCaught(ChannelHandlerContext ctx, Throwable cause) {
194 cause.printStackTrace();
195 ctx.close();
196 }
197 }
以上源码来自： 即时通讯网 - 即时通讯开发者社区！