1 /*

     2  * Copyright 2002 Sun Microsystems, Inc.  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 Sun Microsystems, Inc., 4150 Network Circle, Santa Clara,

    20  * CA 95054 USA or visit www.sun.com if you need additional information or

    21  * have any questions.

    22  */

    23 

    24 /**

    25  *

    26  * Utility class for tests. A simple server, which waits for a connection,

    27  * writes out n bytes and waits.

    28  * @author kladko

    29  */

    30 

    31 import java.net.Socket;

    32 import java.net.ServerSocket;

    33 

    34 public class ByteServer {

    35 

    36     public static final int PORT = 31415;

    37     public static final String LOCALHOST = "localhost";

    38     private int bytecount;

    39     private Socket  socket;

    40     private ServerSocket  serversocket;

    41     private Thread serverthread;

    42     volatile Exception savedException;

    43 

    44     public ByteServer(int bytecount) throws Exception{

    45         this.bytecount = bytecount;

    46         serversocket = new ServerSocket(PORT);

    47     }

    48 

    49     public void start() {

    50         serverthread = new Thread() {

    51             public void run() {

    52                 try {

    53                     socket = serversocket.accept();

    54                     socket.getOutputStream().write(new byte[bytecount]);

    55                     socket.getOutputStream().flush();

    56                 } catch (Exception e) {

    57                     System.err.println("Exception in ByteServer: " + e);

    58                     System.exit(1);

    59                 }

    60             }

    61         };

    62         serverthread.start();

    63     }

    64 

    65     public void exit() throws Exception {

    66         serverthread.join();

    67         socket.close();

    68         serversocket.close();

    69     }

    70 }