| /* |
| * Copyright (c) 2001, Oracle and/or its affiliates. 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 Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA |
| * or visit www.oracle.com if you need additional information or have any |
| * questions. |
| */ |
| |
| /* @test |
| * @summary Unit test for datagram-socket-channel adaptors |
| * @library .. |
| */ |
| |
| import java.io.*; |
| import java.net.*; |
| import java.nio.*; |
| import java.nio.channels.*; |
| import java.nio.charset.*; |
| import java.util.*; |
| |
| |
| public class AdaptDatagramSocket { |
| |
| static java.io.PrintStream out = System.out; |
| |
| static Random rand = new Random(); |
| |
| static final int ECHO_PORT = 7; |
| static final int DISCARD_PORT = 9; |
| static final String REMOTE_HOST = TestUtil.HOST; |
| |
| static final InetSocketAddress echoAddress |
| = new InetSocketAddress(REMOTE_HOST, ECHO_PORT); |
| static final InetSocketAddress discardAddress |
| = new InetSocketAddress(REMOTE_HOST, DISCARD_PORT); |
| |
| static String toString(DatagramPacket dp) { |
| return ("DatagramPacket[off=" + dp.getOffset() |
| + ", len=" + dp.getLength() |
| + "]"); |
| } |
| |
| static void test(DatagramSocket ds, InetSocketAddress dst, |
| boolean shouldTimeout) |
| throws Exception |
| { |
| DatagramPacket op = new DatagramPacket(new byte[100], 13, 42, dst); |
| rand.nextBytes(op.getData()); |
| DatagramPacket ip = new DatagramPacket(new byte[100], 19, 100 - 19); |
| out.println("pre op: " + toString(op) + " ip: " + toString(ip)); |
| |
| long start = System.currentTimeMillis(); |
| ds.send(op); |
| |
| for (;;) { |
| try { |
| ds.receive(ip); |
| if (ip.getLength() == 0) { // ## Not sure why this happens |
| ip.setLength(100 - 19); |
| continue; |
| } |
| } catch (SocketTimeoutException x) { |
| if (shouldTimeout) { |
| out.println("Receive timed out, as expected"); |
| return; |
| } |
| throw x; |
| } |
| if (shouldTimeout) |
| throw new Exception("Receive did not time out"); |
| break; |
| } |
| |
| out.println("rtt: " + (System.currentTimeMillis() - start)); |
| out.println("post op: " + toString(op) + " ip: " + toString(ip)); |
| |
| for (int i = 0; i < ip.getLength(); i++) |
| if (ip.getData()[ip.getOffset() + i] |
| != op.getData()[op.getOffset() + i]) |
| throw new Exception("Incorrect data received"); |
| } |
| |
| static void test(InetSocketAddress dst, |
| int timeout, boolean shouldTimeout, |
| boolean connect) |
| throws Exception |
| { |
| out.println(); |
| out.println("dst: " + dst); |
| |
| DatagramSocket ds; |
| if (false) { |
| // Original |
| ds = new DatagramSocket(); |
| } else { |
| DatagramChannel dc = DatagramChannel.open(); |
| ds = dc.socket(); |
| ds.bind(new InetSocketAddress(0)); |
| } |
| |
| out.println("socket: " + ds); |
| if (connect) { |
| ds.connect(dst); |
| out.println("connect: " + ds); |
| } |
| InetSocketAddress src = new InetSocketAddress(ds.getLocalAddress(), |
| ds.getLocalPort()); |
| out.println("src: " + src); |
| |
| if (timeout > 0) |
| ds.setSoTimeout(timeout); |
| out.println("timeout: " + ds.getSoTimeout()); |
| |
| for (int i = 0; i < 5; i++) |
| test(ds, dst, shouldTimeout); |
| |
| // Leave the socket open so that we don't reuse the old src address |
| //ds.close(); |
| |
| } |
| |
| public static void main(String[] args) throws Exception { |
| test(echoAddress, 0, false, false); |
| test(echoAddress, 0, false, true); |
| test(echoAddress, 5000, false, false); |
| test(discardAddress, 10, true, false); |
| } |
| |
| } |