jdk/test/java/nio/channels/SocketChannel/AdaptSocket.java
changeset 2 90ce3da70b43
child 5506 202f599c92aa
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/jdk/test/java/nio/channels/SocketChannel/AdaptSocket.java	Sat Dec 01 00:00:00 2007 +0000
@@ -0,0 +1,162 @@
+/*
+ * Copyright 2001-2002 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.
+ */
+
+/* @test
+ * @summary Unit test for socket-channel adaptors
+ * @library ..
+ */
+
+import java.io.*;
+import java.net.*;
+import java.nio.*;
+import java.nio.channels.*;
+import java.nio.charset.*;
+
+
+public class AdaptSocket {
+
+    static java.io.PrintStream out = System.out;
+
+    static final int ECHO_PORT = 7;
+    static final int DAYTIME_PORT = 13;
+    static final String REMOTE_HOST = TestUtil.HOST;
+    static final String VERY_REMOTE_HOST = TestUtil.FAR_HOST;
+
+    static void test(String hn, int timeout, boolean shouldTimeout)
+        throws Exception
+    {
+        out.println();
+
+        InetSocketAddress isa
+            = new InetSocketAddress(InetAddress.getByName(hn),
+                                    DAYTIME_PORT);
+        SocketChannel sc = SocketChannel.open();
+        Socket so = sc.socket();
+        out.println("opened: " + so);
+        out.println("        " + sc);
+
+        //out.println("opts:   " + sc.options());
+        so.setTcpNoDelay(true);
+        //so.setTrafficClass(SocketOpts.IP.TOS_THROUGHPUT);
+        so.setKeepAlive(true);
+        so.setSoLinger(true, 42);
+        so.setOOBInline(true);
+        so.setReceiveBufferSize(512);
+        so.setSendBufferSize(512);
+        //out.println("        " + sc.options());
+
+        if (timeout == 0)
+            so.connect(isa);
+        else {
+            try {
+                so.connect(isa, timeout);
+            } catch (SocketTimeoutException x) {
+                if (shouldTimeout) {
+                    out.println("Connection timed out, as expected");
+                    return;
+                } else {
+                    throw x;
+                }
+            }
+            if (shouldTimeout)
+                throw new Exception("Connection did not time out");
+        }
+        out.println("connected: " + so);
+        out.println("           " + sc);
+        byte[] bb = new byte[100];
+        int n = so.getInputStream().read(bb);
+        String s = new String(bb, 0, n - 2, "US-ASCII");
+        out.println(isa + " says: \"" + s + "\"");
+        so.shutdownInput();
+        out.println("ishut: " + sc);
+        so.shutdownOutput();
+        out.println("oshut: " + sc);
+        so.close();
+        out.println("closed: " + so);
+        out.println("        " + sc);
+    }
+
+    static String dataString = "foo\r\n";
+
+    static void testRead(Socket so, boolean shouldTimeout)
+        throws Exception
+    {
+        String data = "foo\r\n";
+        so.getOutputStream().write(dataString.getBytes("US-ASCII"));
+        InputStream is = so.getInputStream();
+        try {
+            byte[] b = new byte[100];
+            int n = is.read(b);
+            if (n != 5)
+                throw new Exception("Incorrect number of bytes read: " + n);
+            if (!dataString.equals(new String(b, 0, n, "US-ASCII")))
+                throw new Exception("Incorrect data read: " + n);
+        } catch (SocketTimeoutException x) {
+            if (shouldTimeout) {
+                out.println("Read timed out, as expected");
+                return;
+            }
+            throw x;
+        }
+        if (shouldTimeout)
+            throw new Exception("Read did not time out");
+    }
+
+    static void testRead(String hn, int timeout, boolean shouldTimeout)
+        throws Exception
+    {
+        out.println();
+
+        InetSocketAddress isa
+            = new InetSocketAddress(InetAddress.getByName(hn), ECHO_PORT);
+        SocketChannel sc = SocketChannel.open();
+        sc.connect(isa);
+        Socket so = sc.socket();
+        out.println("connected: " + so);
+        out.println("           " + sc);
+
+        if (timeout > 0)
+            so.setSoTimeout(timeout);
+        out.println("timeout: " + so.getSoTimeout());
+
+        testRead(so, shouldTimeout);
+        if (!TestUtil.onME())
+            for (int i = 0; i < 4; i++)
+                testRead(so, shouldTimeout);
+
+        sc.close();
+    }
+
+    public static void main(String[] args) throws Exception {
+
+        test(REMOTE_HOST, 0, false);
+        test(REMOTE_HOST, 1000, false);
+        test(VERY_REMOTE_HOST, 10, true);
+
+        testRead(REMOTE_HOST, 0, false);
+        testRead(REMOTE_HOST, 8000, false);
+        testRead(VERY_REMOTE_HOST, 10, true);
+
+    }
+
+}