1 /*
2  * Copyright 2014 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
17 package io.netty.handler.codec.http;
18
19 import io.netty.buffer.ByteBuf;
20 import io.netty.buffer.ByteBufUtil;
21 import io.netty.util.AsciiString;
22 import io.netty.util.CharsetUtil;
23
24 import static io.netty.handler.codec.http.HttpConstants.*;
25 import static io.netty.handler.codec.http.HttpObjectEncoder.CRLF_SHORT;
26
27 final class HttpHeadersEncoder {
28     private static final int COLON_AND_SPACE_SHORT = (COLON << 8) | SP;
29
30     private HttpHeadersEncoder() {
31     }
32
33     static void encoderHeader(CharSequence name, CharSequence value, ByteBuf buf) {
34         final int nameLen = name.length();
35         final int valueLen = value.length();
36         final int entryLen = nameLen + valueLen + 4;
37         buf.ensureWritable(entryLen);
38         int offset = buf.writerIndex();
39         writeAscii(buf, offset, name);
40         offset += nameLen;
41         ByteBufUtil.setShortBE(buf, offset, COLON_AND_SPACE_SHORT);
42         offset += 2;
43         writeAscii(buf, offset, value);
44         offset += valueLen;
45         ByteBufUtil.setShortBE(buf, offset, CRLF_SHORT);
46         offset += 2;
47         buf.writerIndex(offset);
48     }
49
50     private static void writeAscii(ByteBuf buf, int offset, CharSequence value) {
51         if (value instanceof AsciiString) {
52             ByteBufUtil.copy((AsciiString) value, 0, buf, offset, value.length());
53         } else {
54             buf.setCharSequence(offset, value, CharsetUtil.US_ASCII);
55         }
56     }
57 }
58