|
1
36
37 import java.io.*;
38 import java.nio.*;
39 import java.nio.charset.*;
40
41
48 class Reply implements Sendable {
49
50
53 static class Code {
54
55 private int number;
56 private String  reason;
57 private Code(int i, String r) { number = i; reason = r; }
58 public String toString() { return number + " " + reason; }
59
60 static Code OK = new Code(200, "OK");
61 static Code BAD_REQUEST = new Code(400, "Bad Request");
62 static Code NOT_FOUND = new Code(404, "Not Found");
63 static Code METHOD_NOT_ALLOWED = new Code(405, "Method Not Allowed");
64
65 }
66
67 private Code code;
68 private Content content;
69 private boolean headersOnly;
70
71 Reply(Code rc, Content c) {
72 this(rc, c, null);
73 }
74
75 Reply(Code rc, Content c, Request.Action head) {
76 code = rc;
77 content = c;
78 headersOnly = (head == Request.Action.HEAD);
79 }
80
81 private static String CRLF = "\r\n";
82 private static Charset ascii = Charset.forName("US-ASCII");
83
84 private ByteBuffer hbb = null;
85
86 private ByteBuffer headers() {
87 CharBuffer cb = CharBuffer.allocate(1024);
88 for (;;) {
89 try {
90 cb.put("HTTP/1.0 ").put(code.toString()).put(CRLF);
91 cb.put("Server: niossl/0.1").put(CRLF);
92 cb.put("Content-type: ").put(content.type()).put(CRLF);
93 cb.put("Content-length: ")
94 .put(Long.toString(content.length())).put(CRLF);
95 cb.put(CRLF);
96 break;
97 } catch (BufferOverflowException x) {
98 assert(cb.capacity() < (1 << 16));
99 cb = CharBuffer.allocate(cb.capacity() * 2);
100 continue;
101 }
102 }
103 cb.flip();
104 return ascii.encode(cb);
105 }
106
107 public void prepare() throws IOException {
108 content.prepare();
109 hbb = headers();
110 }
111
112 public boolean send(ChannelIO cio) throws IOException {
113
114 if (hbb == null)
115 throw new IllegalStateException ();
116
117 if (hbb.hasRemaining()) {
118 if (cio.write(hbb) <= 0)
119 return true;
120 }
121
122 if (!headersOnly) {
123 if (content.send(cio))
124 return true;
125 }
126
127 if (!cio.dataFlush())
128 return true;
129
130 return false;
131 }
132
133 public void release() throws IOException {
134 content.release();
135 }
136 }
137
|
Popular Tags
|
Java API By Example, From Geeks To Geeks.