1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19 package org.eclipse.jetty.fcgi.generator;
20
21 import java.nio.ByteBuffer;
22 import java.nio.charset.Charset;
23 import java.util.ArrayList;
24 import java.util.List;
25
26 import org.eclipse.jetty.fcgi.FCGI;
27 import org.eclipse.jetty.http.HttpField;
28 import org.eclipse.jetty.http.HttpFields;
29 import org.eclipse.jetty.http.HttpStatus;
30 import org.eclipse.jetty.io.ByteBufferPool;
31 import org.eclipse.jetty.util.BufferUtil;
32 import org.eclipse.jetty.util.Callback;
33
34 public class ServerGenerator extends Generator
35 {
36 private static final byte[] STATUS = new byte[]{'S', 't', 'a', 't', 'u', 's'};
37 private static final byte[] COLON = new byte[]{':', ' '};
38 private static final byte[] EOL = new byte[]{'\r', '\n'};
39
40 private final boolean sendStatus200;
41
42 public ServerGenerator(ByteBufferPool byteBufferPool)
43 {
44 this(byteBufferPool, true);
45 }
46
47 public ServerGenerator(ByteBufferPool byteBufferPool, boolean sendStatus200)
48 {
49 super(byteBufferPool);
50 this.sendStatus200 = sendStatus200;
51 }
52
53 public Result generateResponseHeaders(int request, int code, String reason, HttpFields fields, Callback callback)
54 {
55 request &= 0xFF_FF;
56
57 Charset utf8 = Charset.forName("UTF-8");
58 List<byte[]> bytes = new ArrayList<>(fields.size() * 2);
59 int length = 0;
60
61 if (code != 200 || sendStatus200)
62 {
63
64 bytes.add(STATUS);
65 length += STATUS.length + COLON.length;
66 if (reason == null)
67 reason = HttpStatus.getMessage(code);
68 byte[] responseBytes = (code + " " + reason).getBytes(utf8);
69 bytes.add(responseBytes);
70 length += responseBytes.length + EOL.length;
71 }
72
73
74 for (HttpField field : fields)
75 {
76 String name = field.getName();
77 byte[] nameBytes = name.getBytes(utf8);
78 bytes.add(nameBytes);
79
80 String value = field.getValue();
81 byte[] valueBytes = value.getBytes(utf8);
82 bytes.add(valueBytes);
83
84 length += nameBytes.length + COLON.length;
85 length += valueBytes.length + EOL.length;
86 }
87
88 length += EOL.length;
89
90 final ByteBuffer buffer = byteBufferPool.acquire(length, true);
91 BufferUtil.clearToFill(buffer);
92
93 for (int i = 0; i < bytes.size(); i += 2)
94 buffer.put(bytes.get(i)).put(COLON).put(bytes.get(i + 1)).put(EOL);
95 buffer.put(EOL);
96
97 buffer.flip();
98
99 return generateContent(request, buffer, true, false, callback, FCGI.FrameType.STDOUT);
100 }
101
102 public Result generateResponseContent(int request, ByteBuffer content, boolean lastContent, boolean aborted, Callback callback)
103 {
104 if (aborted)
105 {
106 Result result = new Result(byteBufferPool, callback);
107 if (lastContent)
108 result.append(generateEndRequest(request, true), true);
109 else
110 result.append(BufferUtil.EMPTY_BUFFER, false);
111 return result;
112 }
113 else
114 {
115 Result result = generateContent(request, content, false, lastContent, callback, FCGI.FrameType.STDOUT);
116 if (lastContent)
117 result.append(generateEndRequest(request, false), true);
118 return result;
119 }
120 }
121
122 private ByteBuffer generateEndRequest(int request, boolean aborted)
123 {
124 request &= 0xFF_FF;
125 ByteBuffer endRequestBuffer = byteBufferPool.acquire(8, false);
126 BufferUtil.clearToFill(endRequestBuffer);
127 endRequestBuffer.putInt(0x01_03_00_00 + request);
128 endRequestBuffer.putInt(0x00_08_00_00);
129 endRequestBuffer.putInt(aborted ? 1 : 0);
130 endRequestBuffer.putInt(0);
131 endRequestBuffer.flip();
132 return endRequestBuffer;
133 }
134 }