1 /*
   2  * Copyright (c) 2006, 2010, Oracle and/or its affiliates. All rights reserved.
   3  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
   4  *
   5  * This code is free software; you can redistribute it and/or modify it
   6  * under the terms of the GNU General Public License version 2 only, as
   7  * published by the Free Software Foundation.
   8  *
   9  * This code is distributed in the hope that it will be useful, but WITHOUT
  10  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
  11  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
  12  * version 2 for more details (a copy is included in the LICENSE file that
  13  * accompanied this code).
  14  *
  15  * You should have received a copy of the GNU General Public License version
  16  * 2 along with this work; if not, write to the Free Software Foundation,
  17  * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
  18  *
  19  * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
  20  * or visit www.oracle.com if you need additional information or have any
  21  * questions.
  22  */
  23 
  24 /**
  25  * @test
  26  * @bug 6361557
  27  * @run main/othervm B6361557
  28  * @summary  Lightweight HTTP server quickly runs out of file descriptors on Linux
  29  */
  30 
  31 import com.sun.net.httpserver.*;
  32 
  33 import java.util.*;
  34 import java.util.concurrent.*;
  35 import java.io.*;
  36 import java.nio.*;
  37 import java.nio.channels.*;
  38 import java.net.*;
  39 
  40 /**
  41  * The test simply opens 1,000 separate connections
  42  * and invokes one http request on each. The client does
  43  * not close any sockets until after they are closed
  44  * by the server. This verifies the basic ability
  45  * of the server to manage a reasonable number of connections
  46  */
  47 public class B6361557 {
  48 
  49     public static boolean error = false;
  50     static final int NUM = 1000;
  51 
  52     static class Handler implements HttpHandler {
  53         int invocation = 1;
  54         public void handle (HttpExchange t)
  55             throws IOException
  56         {
  57             InputStream is = t.getRequestBody();
  58             Headers map = t.getRequestHeaders();
  59             Headers rmap = t.getResponseHeaders();
  60             while (is.read () != -1) ;
  61             is.close();
  62             t.sendResponseHeaders (200, -1);
  63             t.close();
  64         }
  65     }
  66 
  67     final static String request = "GET /test/foo.html HTTP/1.1\r\nContent-length: 0\r\n\r\n";
  68     final static ByteBuffer requestBuf = ByteBuffer.allocate(64).put(request.getBytes());
  69 
  70     public static void main (String[] args) throws Exception {
  71         Handler handler = new Handler();
  72         InetSocketAddress addr = new InetSocketAddress (0);
  73         HttpServer server = HttpServer.create (addr, 0);
  74         HttpContext ctx = server.createContext ("/test", handler);
  75 
  76         ExecutorService executor = Executors.newCachedThreadPool();
  77         server.setExecutor (executor);
  78         server.start ();
  79 
  80         InetSocketAddress destaddr = new InetSocketAddress (
  81                 "127.0.0.1", server.getAddress().getPort()
  82         );
  83         System.out.println ("destaddr " + destaddr);
  84 
  85         Selector selector = Selector.open ();
  86         int requests = 0;
  87         int responses = 0;
  88         while (true) {
  89             int selres = selector.select (1);
  90             Set<SelectionKey> selkeys = selector.selectedKeys();
  91             for (SelectionKey key : selkeys) {
  92                 if (key.isReadable()) {
  93                     SocketChannel chan = (SocketChannel)key.channel();
  94                     ByteBuffer buf = (ByteBuffer)key.attachment();
  95                     try {
  96                         int x = chan.read(buf);
  97                         if (x == -1 || responseComplete(buf)) {
  98                             key.attach(null);
  99                             chan.close();
 100                             responses++;
 101                         }
 102                     } catch (IOException e) {}
 103                 }
 104             }
 105             if (requests < NUM) {
 106                 SocketChannel schan = SocketChannel.open(destaddr);
 107                 requestBuf.rewind();
 108                 int c = 0;
 109                 while (requestBuf.remaining() > 0) {
 110                     c += schan.write(requestBuf);
 111                 }
 112                 schan.configureBlocking(false);
 113                 schan.register(selector, SelectionKey.OP_READ, ByteBuffer.allocate(100));
 114                 requests++;
 115             }
 116             if (responses == NUM) {
 117                 System.out.println ("Finished clients");
 118                 break;
 119             }
 120         }
 121         server.stop (1);
 122         selector.close();
 123         executor.shutdown ();
 124 
 125     }
 126 
 127     /* Look for CR LF CR LF */
 128     static boolean responseComplete(ByteBuffer buf) {
 129         int pos = buf.position();
 130         buf.flip();
 131         byte[] lookingFor = new byte[] {'\r', '\n', '\r', '\n' };
 132         int lookingForCount = 0;
 133         while (buf.hasRemaining()) {
 134             byte b = buf.get();
 135             if (b == lookingFor[lookingForCount]) {
 136                 lookingForCount++;
 137                 if (lookingForCount == 4) {
 138                     return true;
 139                 }
 140             } else {
 141                 lookingForCount = 0;
 142             }
 143         }
 144         buf.position(pos);
 145         buf.limit(buf.capacity());
 146         return false;
 147     }
 148 }