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.handler.codec.http;
17
18 import io.netty.buffer.ByteBuf;
19 import io.netty.util.CharsetUtil;
20
21 import static io.netty.handler.codec.http.HttpConstants.*;
22
23 /**
24 * Encodes an {@link HttpRequest} or an {@link HttpContent} into
25 * a {@link ByteBuf}.
26 */
27 public class HttpRequestEncoder extends HttpObjectEncoder<HttpRequest> {
28 private static final char SLASH = '/';
29 private static final char QUESTION_MARK = '?';
30
31 @Override
32 public boolean acceptOutboundMessage(Object msg) throws Exception {
33 return super.acceptOutboundMessage(msg) && !(msg instanceof HttpResponse);
34 }
35
36 @Override
37 protected void encodeInitialLine(ByteBuf buf, HttpRequest request) throws Exception {
38 request.getMethod().encode(buf);
39 buf.writeByte(SP);
40
41 // Add / as absolute path if no is present.
42 // See http://tools.ietf.org/html/rfc2616#section-5.1.2
43 String uri = request.getUri();
44
45 if (uri.isEmpty()) {
46 uri += SLASH;
47 } else {
48 int start = uri.indexOf("://");
49 if (start != -1 && uri.charAt(0) != SLASH) {
50 int startIndex = start + 3;
51 // Correctly handle query params.
52 // See https://github.com/netty/netty/issues/2732
53 int index = uri.indexOf(QUESTION_MARK, startIndex);
54 if (index == -1) {
55 if (uri.lastIndexOf(SLASH) < startIndex) {
56 uri += SLASH;
57 }
58 } else {
59 if (uri.lastIndexOf(SLASH, index) < startIndex) {
60 uri = new StringBuilder(uri).insert(index, SLASH).toString();
61 }
62 }
63 }
64 }
65
66 buf.writeBytes(uri.getBytes(CharsetUtil.UTF_8));
67
68 buf.writeByte(SP);
69 request.getProtocolVersion().encode(buf);
70 buf.writeBytes(CRLF);
71 }
72 }