1 //
2 // ========================================================================
3 // Copyright (c) 1995-2013 Mort Bay Consulting Pty. Ltd.
4 // ------------------------------------------------------------------------
5 // All rights reserved. This program and the accompanying materials
6 // are made available under the terms of the Eclipse Public License v1.0
7 // and Apache License v2.0 which accompanies this distribution.
8 //
9 // The Eclipse Public License is available at
10 // http://www.eclipse.org/legal/epl-v10.html
11 //
12 // The Apache License v2.0 is available at
13 // http://www.opensource.org/licenses/apache2.0.php
14 //
15 // You may elect to redistribute this code under either of these licenses.
16 // ========================================================================
17 //
18
19 package org.eclipse.jetty.spdy.generator;
20
21 import java.nio.ByteBuffer;
22
23 import org.eclipse.jetty.spdy.ByteBufferPool;
24 import org.eclipse.jetty.spdy.api.DataInfo;
25 import org.eclipse.jetty.spdy.frames.DataFrame;
26
27 public class DataFrameGenerator
28 {
29 private final ByteBufferPool bufferPool;
30
31 public DataFrameGenerator(ByteBufferPool bufferPool)
32 {
33 this.bufferPool = bufferPool;
34 }
35
36 public ByteBuffer generate(int streamId, int length, DataInfo dataInfo)
37 {
38 ByteBuffer buffer = bufferPool.acquire(DataFrame.HEADER_LENGTH + length, true);
39 buffer.position(DataFrame.HEADER_LENGTH);
40 // Guaranteed to always be >= 0
41 int read = dataInfo.readInto(buffer);
42
43 buffer.putInt(0, streamId & 0x7F_FF_FF_FF);
44 buffer.putInt(4, read & 0x00_FF_FF_FF);
45
46 byte flags = dataInfo.getFlags();
47 if (dataInfo.available() > 0)
48 flags &= ~DataInfo.FLAG_CLOSE;
49 buffer.put(4, flags);
50
51 buffer.flip();
52 return buffer;
53 }
54 }