--- /dev/null Thu Jan 01 00:00:00 1970 +0000
+++ b/jdk/src/sample/share/nio/server/RequestHandler.java Sun Aug 17 15:54:13 2014 +0100
@@ -0,0 +1,198 @@
+/*
+ * Copyright (c) 2004, 2011, Oracle and/or its affiliates. All rights reserved.
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions
+ * are met:
+ *
+ * - Redistributions of source code must retain the above copyright
+ * notice, this list of conditions and the following disclaimer.
+ *
+ * - Redistributions in binary form must reproduce the above copyright
+ * notice, this list of conditions and the following disclaimer in the
+ * documentation and/or other materials provided with the distribution.
+ *
+ * - Neither the name of Oracle nor the names of its
+ * contributors may be used to endorse or promote products derived
+ * from this software without specific prior written permission.
+ *
+ * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS
+ * IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO,
+ * THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
+ * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR
+ * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
+ * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
+ * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
+ * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
+ * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
+ * NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
+ * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+ */
+
+/*
+ * This source code is provided to illustrate the usage of a given feature
+ * or technique and has been deliberately simplified. Additional steps
+ * required for a production-quality application, such as security checks,
+ * input validation and proper error handling, might not be present in
+ * this sample code.
+ */
+
+
+import java.io.*;
+import java.nio.*;
+import java.nio.channels.*;
+
+/**
+ * Primary driver class used by non-blocking Servers to receive,
+ * prepare, send, and shutdown requests.
+ *
+ * @author Mark Reinhold
+ * @author Brad R. Wetmore
+ */
+class RequestHandler implements Handler {
+
+ private ChannelIO cio;
+ private ByteBuffer rbb = null;
+
+ private boolean requestReceived = false;
+ private Request request = null;
+ private Reply reply = null;
+
+ private static int created = 0;
+
+ RequestHandler(ChannelIO cio) {
+ this.cio = cio;
+
+ // Simple heartbeat to let user know we're alive.
+ synchronized (RequestHandler.class) {
+ created++;
+ if ((created % 50) == 0) {
+ System.out.println(".");
+ created = 0;
+ } else {
+ System.out.print(".");
+ }
+ }
+ }
+
+ // Returns true when request is complete
+ // May expand rbb if more room required
+ //
+ private boolean receive(SelectionKey sk) throws IOException {
+ ByteBuffer tmp = null;
+
+ if (requestReceived) {
+ return true;
+ }
+
+ if (!cio.doHandshake(sk)) {
+ return false;
+ }
+
+ if ((cio.read() < 0) || Request.isComplete(cio.getReadBuf())) {
+ rbb = cio.getReadBuf();
+ return (requestReceived = true);
+ }
+ return false;
+ }
+
+ // When parse is successfull, saves request and returns true
+ //
+ private boolean parse() throws IOException {
+ try {
+ request = Request.parse(rbb);
+ return true;
+ } catch (MalformedRequestException x) {
+ reply = new Reply(Reply.Code.BAD_REQUEST,
+ new StringContent(x));
+ }
+ return false;
+ }
+
+ // Ensures that reply field is non-null
+ //
+ private void build() throws IOException {
+ Request.Action action = request.action();
+ if ((action != Request.Action.GET) &&
+ (action != Request.Action.HEAD)) {
+ reply = new Reply(Reply.Code.METHOD_NOT_ALLOWED,
+ new StringContent(request.toString()));
+ }
+ reply = new Reply(Reply.Code.OK,
+ new FileContent(request.uri()), action);
+ }
+
+ public void handle(SelectionKey sk) throws IOException {
+ try {
+
+ if (request == null) {
+ if (!receive(sk))
+ return;
+ rbb.flip();
+ if (parse())
+ build();
+ try {
+ reply.prepare();
+ } catch (IOException x) {
+ reply.release();
+ reply = new Reply(Reply.Code.NOT_FOUND,
+ new StringContent(x));
+ reply.prepare();
+ }
+ if (send()) {
+ // More bytes remain to be written
+ sk.interestOps(SelectionKey.OP_WRITE);
+ } else {
+ // Reply completely written; we're done
+ if (cio.shutdown()) {
+ cio.close();
+ reply.release();
+ }
+ }
+ } else {
+ if (!send()) { // Should be rp.send()
+ if (cio.shutdown()) {
+ cio.close();
+ reply.release();
+ }
+ }
+ }
+ } catch (IOException x) {
+ String m = x.getMessage();
+ if (!m.equals("Broken pipe") &&
+ !m.equals("Connection reset by peer")) {
+ System.err.println("RequestHandler: " + x.toString());
+ }
+
+ try {
+ /*
+ * We had a failure here, so we'll try to be nice
+ * before closing down and send off a close_notify,
+ * but if we can't get the message off with one try,
+ * we'll just shutdown.
+ */
+ cio.shutdown();
+ } catch (IOException e) {
+ // ignore
+ }
+
+ cio.close();
+ if (reply != null) {
+ reply.release();
+ }
+ }
+
+ }
+
+ private boolean send() throws IOException {
+ try {
+ return reply.send(cio);
+ } catch (IOException x) {
+ if (x.getMessage().startsWith("Resource temporarily")) {
+ System.err.println("## RTA");
+ return true;
+ }
+ throw x;
+ }
+ }
+}