libjava/classpath/testsuite/java.net/DatagramSocketSendReceiveTest.java

   1 import java.net.*;
   2
   3 /*
   4  * Start one thread for receiving a packet, wait for it to set up,
   5  * send a packet to it, and wait until it completes.  Compare the
   6  * packet to make sure it came thru without errors.
   7  */
   8
   9 public class DatagramSocketSendReceiveTest
  10   implements Runnable
  11 {
  12   public static final int port = 4000 + (int)(java.lang.Math.random() * 1000);
  13   public static final String message = "hello";
  14   public static int count = 0;
  15   public static String received;
  16
  17   void send()
  18     throws Exception
  19     {
  20       DatagramSocket sender = new DatagramSocket();
  21       InetAddress local = sender.getLocalAddress();
  22       byte []message_bytes = message.getBytes();
  23
  24       DatagramPacket packet = new DatagramPacket(message_bytes,
  25                                                  message_bytes.length,
  26                                                  local, port);
  27
  28       sender.send(packet);
  29       sender.close();
  30     }
  31   void receive()
  32     throws Exception
  33     {
  34       DatagramSocket socket = new DatagramSocket(port);
  35       socket.setSoTimeout(10);
  36
  37       byte[] buffer = new byte[100];
  38       DatagramPacket packet = new DatagramPacket(buffer, buffer.length);
  39
  40       synchronized(this) {
  41         notifyAll();
  42       }
  43
  44       socket.receive(packet);
  45       socket.close();
  46
  47       received = new String(buffer, 0, packet.getLength());
  48
  49       count++;
  50       if ( message.length() != received.length() )
  51         throw new Exception("Receved "+ received.length()+
  52                             " bytes but sent "+message.length() + " bytes");
  53
  54       if ( ! message.equals(received) )
  55         throw new Exception("Receved \""+ received+
  56                             "\" but sent \""+message + "\"");
  57
  58     }
  59
  60   public void run()
  61     {
  62       String name = Thread.currentThread().getName();
  63       if (name.equals("timer")) {
  64         try {
  65           Thread.sleep(10000);
  66         } catch (InterruptedException e){}
  67         System.out.println("FAILED: timer triggered");
  68         System.exit(0);
  69       }
  70       try {
  71         receive();
  72       } catch (Exception e) {
  73         System.out.println("FAILED: receiver: " + e);
  74         System.exit(0);
  75       }
  76     }
  77   public static void main(String args[])
  78     {
  79       try {
  80         DatagramSocketSendReceiveTest sender =
  81           new DatagramSocketSendReceiveTest();
  82         DatagramSocketSendReceiveTest receiver =
  83           new DatagramSocketSendReceiveTest();
  84         Thread receiver_thread = new Thread(receiver);
  85
  86         /* Make sure the test terminates even if it hangs on network */
  87         DatagramSocketSendReceiveTest timer = new DatagramSocketSendReceiveTest();
  88         Thread timer_thread = new Thread(timer, "timer");
  89         timer_thread.start();
  90
  91         synchronized(receiver) {
  92           receiver_thread.start();
  93           receiver.wait();
  94         }
  95         try {
  96           sender.send();
  97         } catch (Exception e) {
  98           System.out.println("FAILED: sender: " + e);
  99           System.exit(0);
 100         }
 101         receiver_thread.join();
 102
 103         if (0 == count)
 104           throw new Exception("Nothing received");
 105
 106         System.out.println("PASSED: DatagramSocket send/receive count="+count+
 107                            " message="+received);
 108         System.exit(0);
 109       } catch (Exception e) {
 110         System.out.println("FAILED: " + e);
 111         System.exit(0);
 112       }
 113     }
 114 }