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 public ServerGenerator(ByteBufferPool byteBufferPool)
41 {
42 super(byteBufferPool);
43 }
44
45 public Result generateResponseHeaders(int request, int code, String reason, HttpFields fields, Callback callback)
46 {
47 request &= 0xFF_FF;
48
49 Charset utf8 = Charset.forName("UTF-8");
50 List<byte[]> bytes = new ArrayList<>(fields.size() * 2);
51 int length = 0;
52
53
54 bytes.add(STATUS);
55 length += STATUS.length + COLON.length;
56 if (reason == null)
57 reason = HttpStatus.getMessage(code);
58 byte[] responseBytes = (code + " " + reason).getBytes(utf8);
59 bytes.add(responseBytes);
60 length += responseBytes.length + EOL.length;
61
62
63 for (HttpField field : fields)
64 {
65 String name = field.getName();
66 byte[] nameBytes = name.getBytes(utf8);
67 bytes.add(nameBytes);
68
69 String value = field.getValue();
70 byte[] valueBytes = value.getBytes(utf8);
71 bytes.add(valueBytes);
72
73 length += nameBytes.length + COLON.length;
74 length += valueBytes.length + EOL.length;
75 }
76
77 length += EOL.length;
78
79 final ByteBuffer buffer = byteBufferPool.acquire(length, true);
80 BufferUtil.clearToFill(buffer);
81
82 for (int i = 0; i < bytes.size(); i += 2)
83 buffer.put(bytes.get(i)).put(COLON).put(bytes.get(i + 1)).put(EOL);
84 buffer.put(EOL);
85
86 buffer.flip();
87
88 return generateContent(request, buffer, true, false, callback, FCGI.FrameType.STDOUT);
89 }
90
91 public Result generateResponseContent(int request, ByteBuffer content, boolean lastContent, boolean aborted, Callback callback)
92 {
93 if (aborted)
94 {
95 Result result = new Result(byteBufferPool, callback);
96 if (lastContent)
97 result.append(generateEndRequest(request, true), true);
98 else
99 result.append(BufferUtil.EMPTY_BUFFER, false);
100 return result;
101 }
102 else
103 {
104 Result result = generateContent(request, content, false, lastContent, callback, FCGI.FrameType.STDOUT);
105 if (lastContent)
106 result.append(generateEndRequest(request, false), true);
107 return result;
108 }
109 }
110
111 private ByteBuffer generateEndRequest(int request, boolean aborted)
112 {
113 request &= 0xFF_FF;
114 ByteBuffer endRequestBuffer = byteBufferPool.acquire(8, false);
115 BufferUtil.clearToFill(endRequestBuffer);
116 endRequestBuffer.putInt(0x01_03_00_00 + request);
117 endRequestBuffer.putInt(0x00_08_00_00);
118 endRequestBuffer.putInt(aborted ? 1 : 0);
119 endRequestBuffer.putInt(0);
120 endRequestBuffer.flip();
121 return endRequestBuffer;
122 }
123 }