353 lines
11 KiB
Java
353 lines
11 KiB
Java
|
/*
|
||
|
|
* Copyright 2003-2005 Sun Microsystems, Inc. All Rights Reserved.
|
||
|
|
* DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
|
||
|
|
*
|
||
|
|
* This code is free software; you can redistribute it and/or modify it
|
||
|
|
* under the terms of the GNU General Public License version 2 only, as
|
||
|
|
* published by the Free Software Foundation.
|
||
|
|
*
|
||
|
|
* This code is distributed in the hope that it will be useful, but WITHOUT
|
||
|
|
* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
|
||
|
|
* FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
|
||
|
|
* version 2 for more details (a copy is included in the LICENSE file that
|
||
|
|
* accompanied this code).
|
||
|
|
*
|
||
|
|
* You should have received a copy of the GNU General Public License version
|
||
|
|
* 2 along with this work; if not, write to the Free Software Foundation,
|
||
|
|
* Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
|
||
|
|
*
|
||
|
|
* Please contact Sun Microsystems, Inc., 4150 Network Circle, Santa Clara,
|
||
|
|
* CA 95054 USA or visit www.sun.com if you need additional information or
|
||
|
|
* have any questions.
|
||
|
|
*/
|
||
|
|
|
||
|
|
import java.net.*;
|
||
|
|
import java.io.*;
|
||
|
|
import java.util.*;
|
||
|
|
|
||
|
|
public class Tests {
|
||
|
|
/**
|
||
|
|
* performs a simple exchange of data between the two sockets
|
||
|
|
* and throws an exception if there is any problem.
|
||
|
|
*/
|
||
|
|
public static void simpleDataExchange (Socket s1, Socket s2)
|
||
|
|
throws Exception {
|
||
|
|
|
||
|
|
InputStream i1 = s1.getInputStream();
|
||
|
|
InputStream i2 = s2.getInputStream();
|
||
|
|
OutputStream o1 = s1.getOutputStream();
|
||
|
|
OutputStream o2 = s2.getOutputStream();
|
||
|
|
|
||
|
|
simpleWrite (o1, 100);
|
||
|
|
simpleWrite (o2, 200);
|
||
|
|
simpleRead (i2, 100);
|
||
|
|
simpleRead (i1, 200);
|
||
|
|
}
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Send a packet from s1 to s2 (ia2/s2.localPort) and check it
|
||
|
|
* Send a packet from s2 to s1 (ia1/s1.localPort) and check it
|
||
|
|
*/
|
||
|
|
public static void simpleDataExchange (DatagramSocket s1, InetAddress ia1,
|
||
|
|
DatagramSocket s2, InetAddress ia2)
|
||
|
|
throws Exception {
|
||
|
|
|
||
|
|
SocketAddress dest1 = new InetSocketAddress (ia1, s1.getLocalPort());
|
||
|
|
dprintln ("dest1 = " + dest1);
|
||
|
|
SocketAddress dest2 = new InetSocketAddress (ia2, s2.getLocalPort());
|
||
|
|
dprintln ("dest2 = " + dest2);
|
||
|
|
|
||
|
|
byte[] ba = "Hello world".getBytes();
|
||
|
|
byte[] bb = "HELLO WORLD1".getBytes();
|
||
|
|
DatagramPacket p1 = new DatagramPacket (ba, ba.length, dest1);
|
||
|
|
DatagramPacket p2 = new DatagramPacket (ba, ba.length, dest2);
|
||
|
|
|
||
|
|
DatagramPacket r1 = new DatagramPacket (new byte[256], 256);
|
||
|
|
DatagramPacket r2 = new DatagramPacket (new byte[256], 256);
|
||
|
|
|
||
|
|
s2.send (p1);
|
||
|
|
s1.send (p2);
|
||
|
|
s1.receive (r1);
|
||
|
|
s2.receive (r2);
|
||
|
|
comparePackets (p1, r1);
|
||
|
|
comparePackets (p2, r2);
|
||
|
|
}
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Send a packet from s1 to s2 (ia2/s2.localPort) and send same packet
|
||
|
|
* back from s2 to sender. Check s1 receives original packet
|
||
|
|
*/
|
||
|
|
|
||
|
|
public static void datagramEcho (DatagramSocket s1, DatagramSocket s2,
|
||
|
|
InetAddress ia2)
|
||
|
|
throws Exception {
|
||
|
|
|
||
|
|
byte[] ba = "Hello world".getBytes();
|
||
|
|
DatagramPacket p1;
|
||
|
|
|
||
|
|
SocketAddress dest2 = null;
|
||
|
|
if (ia2 != null) {
|
||
|
|
dest2 = new InetSocketAddress (ia2, s2.getLocalPort());
|
||
|
|
p1 = new DatagramPacket (ba, ba.length, dest2);
|
||
|
|
} else {
|
||
|
|
p1 = new DatagramPacket (ba, ba.length);
|
||
|
|
}
|
||
|
|
|
||
|
|
dprintln ("dest2 = " + dest2);
|
||
|
|
|
||
|
|
|
||
|
|
DatagramPacket r1 = new DatagramPacket (new byte[256], 256);
|
||
|
|
DatagramPacket r2 = new DatagramPacket (new byte[256], 256);
|
||
|
|
|
||
|
|
s1.send (p1);
|
||
|
|
s2.receive (r1);
|
||
|
|
s2.send (r1);
|
||
|
|
s1.receive (r2);
|
||
|
|
comparePackets (p1, r1);
|
||
|
|
comparePackets (p1, r2);
|
||
|
|
}
|
||
|
|
|
||
|
|
public static void comparePackets (DatagramPacket p1, DatagramPacket p2)
|
||
|
|
throws Exception {
|
||
|
|
|
||
|
|
byte[] b1 = p1.getData();
|
||
|
|
byte[] b2 = p2.getData();
|
||
|
|
int len = p1.getLength () > p2.getLength() ? p2.getLength()
|
||
|
|
: p1.getLength();
|
||
|
|
for (int i=0; i<len; i++) {
|
||
|
|
if (b1[i] != b2[i]) {
|
||
|
|
throw new Exception ("packets not the same");
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
/* check the time got is within 20% of the time expected */
|
||
|
|
public static void checkTime (long got, long expected) {
|
||
|
|
dprintln ("checkTime: got " + got + " expected " + expected);
|
||
|
|
long upper = expected + (expected / 5);
|
||
|
|
long lower = expected - (expected / 5);
|
||
|
|
if (got > upper || got < lower) {
|
||
|
|
throw new RuntimeException ("checkTime failed: got " + got + " expected " + expected);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
static boolean debug = false;
|
||
|
|
|
||
|
|
public static void checkDebug (String[] args) {
|
||
|
|
debug = args.length > 0 && args[0].equals("-d");
|
||
|
|
}
|
||
|
|
|
||
|
|
public static void dprint (String s) {
|
||
|
|
if (debug) {
|
||
|
|
System.out.print (s);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
public static void dprintln (String s) {
|
||
|
|
if (debug) {
|
||
|
|
System.out.println (s);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
static int numberInterfaces () {
|
||
|
|
try {
|
||
|
|
Enumeration ifs = NetworkInterface.getNetworkInterfaces();
|
||
|
|
int nifs=0;
|
||
|
|
while (ifs.hasMoreElements()) {
|
||
|
|
nifs++;
|
||
|
|
ifs.nextElement();
|
||
|
|
}
|
||
|
|
return nifs;
|
||
|
|
} catch (SocketException e) {
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
public static Enumeration ipv4Addresses() {
|
||
|
|
return new AddrEnum (Inet4Address.class);
|
||
|
|
}
|
||
|
|
|
||
|
|
public static Inet4Address getFirstLocalIPv4Address () {
|
||
|
|
Enumeration e = ipv4Addresses();
|
||
|
|
if (!e.hasMoreElements()) {
|
||
|
|
return null;
|
||
|
|
}
|
||
|
|
return (Inet4Address)e.nextElement();
|
||
|
|
}
|
||
|
|
|
||
|
|
public static Inet6Address getFirstLocalIPv6Address () {
|
||
|
|
Enumeration e = ipv6Addresses();
|
||
|
|
if (!e.hasMoreElements()) {
|
||
|
|
return null;
|
||
|
|
}
|
||
|
|
return (Inet6Address)e.nextElement();
|
||
|
|
}
|
||
|
|
|
||
|
|
public static Enumeration ipv6Addresses() {
|
||
|
|
return new AddrEnum (Inet6Address.class);
|
||
|
|
}
|
||
|
|
|
||
|
|
/* enumerates the Inet4Addresses or Inet6Addresses on the system */
|
||
|
|
|
||
|
|
private static class AddrEnum implements Enumeration {
|
||
|
|
|
||
|
|
Enumeration ifs;
|
||
|
|
NetworkInterface currIf = null;
|
||
|
|
InetAddress nextAddr=null;
|
||
|
|
Enumeration addrs=null;
|
||
|
|
Class filter;
|
||
|
|
|
||
|
|
static final byte[] fe80_loopback = new byte [] {
|
||
|
|
(byte)0xfe,(byte)0x80,0,0,0,0,0,0,0,0,0,0,0,0,0,1
|
||
|
|
};
|
||
|
|
|
||
|
|
AddrEnum (Class filter) {
|
||
|
|
this.filter = filter;
|
||
|
|
try {
|
||
|
|
ifs = NetworkInterface.getNetworkInterfaces();
|
||
|
|
} catch (SocketException e) {}
|
||
|
|
}
|
||
|
|
|
||
|
|
public boolean hasMoreElements () {
|
||
|
|
if (nextAddr == null) {
|
||
|
|
nextAddr = getNext();
|
||
|
|
}
|
||
|
|
return (nextAddr != null);
|
||
|
|
}
|
||
|
|
|
||
|
|
public Object nextElement () {
|
||
|
|
if (!hasMoreElements()) {
|
||
|
|
throw new NoSuchElementException ("no more addresses");
|
||
|
|
}
|
||
|
|
Object next = nextAddr;
|
||
|
|
nextAddr = null;
|
||
|
|
return next;
|
||
|
|
}
|
||
|
|
|
||
|
|
private InetAddress getNext() {
|
||
|
|
while (true) {
|
||
|
|
if (currIf == null) {
|
||
|
|
currIf = getNextIf();
|
||
|
|
if (currIf == null) {
|
||
|
|
return null;
|
||
|
|
}
|
||
|
|
addrs = currIf.getInetAddresses();
|
||
|
|
}
|
||
|
|
while (addrs.hasMoreElements()) {
|
||
|
|
InetAddress addr = (InetAddress) addrs.nextElement();
|
||
|
|
if (filter.isInstance (addr) && !addr.isLoopbackAddress()) {
|
||
|
|
if (Arrays.equals (addr.getAddress(), fe80_loopback)) {
|
||
|
|
continue;
|
||
|
|
}
|
||
|
|
return addr;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
currIf = null;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
private NetworkInterface getNextIf () {
|
||
|
|
while (ifs.hasMoreElements()) {
|
||
|
|
NetworkInterface nic = (NetworkInterface)ifs.nextElement();
|
||
|
|
try {
|
||
|
|
if (nic.isUp() && !nic.isLoopback())
|
||
|
|
return nic;
|
||
|
|
} catch (SocketException e) {
|
||
|
|
// ignore
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
return null;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Throws a RuntimeException if the boolean condition is false
|
||
|
|
*/
|
||
|
|
public static void t_assert (boolean assertion) {
|
||
|
|
if (assertion) {
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
Throwable t = new Throwable();
|
||
|
|
StackTraceElement[] strace = t.getStackTrace();
|
||
|
|
String msg = "Assertion failed at: " + strace[1].toString();
|
||
|
|
throw new RuntimeException (msg);
|
||
|
|
}
|
||
|
|
|
||
|
|
private static void simpleRead (InputStream is, int start) throws Exception {
|
||
|
|
byte b[] = new byte [2];
|
||
|
|
for (int i=start; i<start+100; i++) {
|
||
|
|
int x = is.read (b);
|
||
|
|
if (x == 1) {
|
||
|
|
x += is.read (b,1,1);
|
||
|
|
}
|
||
|
|
if (x!=2) {
|
||
|
|
throw new Exception ("read error");
|
||
|
|
}
|
||
|
|
int r = bytes (b[0], b[1]);
|
||
|
|
if (r != i) {
|
||
|
|
throw new Exception ("read " + r + " expected " +i);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
/* convert signed bytes to unisigned int */
|
||
|
|
private static int bytes (byte b1, byte b2) {
|
||
|
|
int i1 = (int)b1 & 0xFF;
|
||
|
|
int i2 = (int)b2 & 0xFF;
|
||
|
|
return i1 * 256 + i2;
|
||
|
|
}
|
||
|
|
|
||
|
|
static void simpleWrite (OutputStream os, int start) throws Exception {
|
||
|
|
byte b[] = new byte [2];
|
||
|
|
for (int i=start; i<start+100; i++) {
|
||
|
|
b[0] = (byte) (i / 256);
|
||
|
|
b[1] = (byte) (i % 256);
|
||
|
|
os.write (b);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
private static class Runner extends Thread {
|
||
|
|
Runnable runnee;
|
||
|
|
long delay;
|
||
|
|
|
||
|
|
Runner (Runnable runnee, long delay) {
|
||
|
|
super();
|
||
|
|
this.runnee = runnee;
|
||
|
|
this.delay = delay;
|
||
|
|
}
|
||
|
|
|
||
|
|
public void run () {
|
||
|
|
try {
|
||
|
|
Thread.sleep (delay);
|
||
|
|
runnee.run ();
|
||
|
|
} catch (Exception e) {
|
||
|
|
e.printStackTrace();
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
/*
|
||
|
|
* Take the given Runnable and run it in a spawned thread
|
||
|
|
* after the given time has elapsed. runAfter() returns immediately
|
||
|
|
*/
|
||
|
|
public static void runAfter (long millis, Runnable runnee) {
|
||
|
|
Runner runner = new Runner (runnee, millis);
|
||
|
|
runner.start ();
|
||
|
|
}
|
||
|
|
|
||
|
|
static String osname;
|
||
|
|
|
||
|
|
static {
|
||
|
|
osname = System.getProperty ("os.name");
|
||
|
|
}
|
||
|
|
|
||
|
|
static boolean isLinux () {
|
||
|
|
return osname.equals ("Linux");
|
||
|
|
}
|
||
|
|
|
||
|
|
static boolean isWindows () {
|
||
|
|
return osname.startsWith ("Windows");
|
||
|
|
}
|
||
|
|
}
|