jdk/test/java/nio/channels/SocketChannel/FinishConnect.java
author dfuchs
Wed, 07 Nov 2012 13:24:39 +0100
changeset 14415 7a31b0e0cfaf
parent 5506 202f599c92aa
permissions -rw-r--r--
6720349: (ch) Channels tests depending on hosts inside Sun Summary: This changeset make the nio tests start small TCP or UDP servers from within the tests, instead of relying on external services. Reviewed-by: alanb
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
2
90ce3da70b43 Initial load
duke
parents:
diff changeset
     1
/*
14415
7a31b0e0cfaf 6720349: (ch) Channels tests depending on hosts inside Sun
dfuchs
parents: 5506
diff changeset
     2
 * Copyright (c) 2001, 2012, Oracle and/or its affiliates. All rights reserved.
2
90ce3da70b43 Initial load
duke
parents:
diff changeset
     3
 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
90ce3da70b43 Initial load
duke
parents:
diff changeset
     4
 *
90ce3da70b43 Initial load
duke
parents:
diff changeset
     5
 * This code is free software; you can redistribute it and/or modify it
90ce3da70b43 Initial load
duke
parents:
diff changeset
     6
 * under the terms of the GNU General Public License version 2 only, as
90ce3da70b43 Initial load
duke
parents:
diff changeset
     7
 * published by the Free Software Foundation.
90ce3da70b43 Initial load
duke
parents:
diff changeset
     8
 *
90ce3da70b43 Initial load
duke
parents:
diff changeset
     9
 * This code is distributed in the hope that it will be useful, but WITHOUT
90ce3da70b43 Initial load
duke
parents:
diff changeset
    10
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
90ce3da70b43 Initial load
duke
parents:
diff changeset
    11
 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
90ce3da70b43 Initial load
duke
parents:
diff changeset
    12
 * version 2 for more details (a copy is included in the LICENSE file that
90ce3da70b43 Initial load
duke
parents:
diff changeset
    13
 * accompanied this code).
90ce3da70b43 Initial load
duke
parents:
diff changeset
    14
 *
90ce3da70b43 Initial load
duke
parents:
diff changeset
    15
 * You should have received a copy of the GNU General Public License version
90ce3da70b43 Initial load
duke
parents:
diff changeset
    16
 * 2 along with this work; if not, write to the Free Software Foundation,
90ce3da70b43 Initial load
duke
parents:
diff changeset
    17
 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
90ce3da70b43 Initial load
duke
parents:
diff changeset
    18
 *
5506
202f599c92aa 6943119: Rebrand source copyright notices
ohair
parents: 2
diff changeset
    19
 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
202f599c92aa 6943119: Rebrand source copyright notices
ohair
parents: 2
diff changeset
    20
 * or visit www.oracle.com if you need additional information or have any
202f599c92aa 6943119: Rebrand source copyright notices
ohair
parents: 2
diff changeset
    21
 * questions.
2
90ce3da70b43 Initial load
duke
parents:
diff changeset
    22
 */
90ce3da70b43 Initial load
duke
parents:
diff changeset
    23
90ce3da70b43 Initial load
duke
parents:
diff changeset
    24
/* @test
90ce3da70b43 Initial load
duke
parents:
diff changeset
    25
 * @summary Test SocketChannel.finishConnect
90ce3da70b43 Initial load
duke
parents:
diff changeset
    26
 * @library ..
90ce3da70b43 Initial load
duke
parents:
diff changeset
    27
 */
90ce3da70b43 Initial load
duke
parents:
diff changeset
    28
90ce3da70b43 Initial load
duke
parents:
diff changeset
    29
import java.net.*;
90ce3da70b43 Initial load
duke
parents:
diff changeset
    30
import java.nio.*;
90ce3da70b43 Initial load
duke
parents:
diff changeset
    31
import java.nio.channels.*;
90ce3da70b43 Initial load
duke
parents:
diff changeset
    32
import java.nio.channels.spi.SelectorProvider;
90ce3da70b43 Initial load
duke
parents:
diff changeset
    33
import java.nio.charset.*;
90ce3da70b43 Initial load
duke
parents:
diff changeset
    34
import java.util.*;
90ce3da70b43 Initial load
duke
parents:
diff changeset
    35
90ce3da70b43 Initial load
duke
parents:
diff changeset
    36
90ce3da70b43 Initial load
duke
parents:
diff changeset
    37
public class FinishConnect {
90ce3da70b43 Initial load
duke
parents:
diff changeset
    38
90ce3da70b43 Initial load
duke
parents:
diff changeset
    39
    public static void main(String[] args) throws Exception {
14415
7a31b0e0cfaf 6720349: (ch) Channels tests depending on hosts inside Sun
dfuchs
parents: 5506
diff changeset
    40
        try (TestServers.DayTimeServer dayTimeServer
7a31b0e0cfaf 6720349: (ch) Channels tests depending on hosts inside Sun
dfuchs
parents: 5506
diff changeset
    41
                = TestServers.DayTimeServer.startNewServer(100)) {
7a31b0e0cfaf 6720349: (ch) Channels tests depending on hosts inside Sun
dfuchs
parents: 5506
diff changeset
    42
            test1(dayTimeServer, true, true);
7a31b0e0cfaf 6720349: (ch) Channels tests depending on hosts inside Sun
dfuchs
parents: 5506
diff changeset
    43
            test1(dayTimeServer, true, false);
7a31b0e0cfaf 6720349: (ch) Channels tests depending on hosts inside Sun
dfuchs
parents: 5506
diff changeset
    44
            test1(dayTimeServer, false, true);
7a31b0e0cfaf 6720349: (ch) Channels tests depending on hosts inside Sun
dfuchs
parents: 5506
diff changeset
    45
            test1(dayTimeServer, false, false);
7a31b0e0cfaf 6720349: (ch) Channels tests depending on hosts inside Sun
dfuchs
parents: 5506
diff changeset
    46
            test2(dayTimeServer);
7a31b0e0cfaf 6720349: (ch) Channels tests depending on hosts inside Sun
dfuchs
parents: 5506
diff changeset
    47
        }
2
90ce3da70b43 Initial load
duke
parents:
diff changeset
    48
    }
90ce3da70b43 Initial load
duke
parents:
diff changeset
    49
14415
7a31b0e0cfaf 6720349: (ch) Channels tests depending on hosts inside Sun
dfuchs
parents: 5506
diff changeset
    50
    static void test1(TestServers.DayTimeServer daytimeServer,
7a31b0e0cfaf 6720349: (ch) Channels tests depending on hosts inside Sun
dfuchs
parents: 5506
diff changeset
    51
                      boolean select,
7a31b0e0cfaf 6720349: (ch) Channels tests depending on hosts inside Sun
dfuchs
parents: 5506
diff changeset
    52
                      boolean setBlocking)
7a31b0e0cfaf 6720349: (ch) Channels tests depending on hosts inside Sun
dfuchs
parents: 5506
diff changeset
    53
        throws Exception
7a31b0e0cfaf 6720349: (ch) Channels tests depending on hosts inside Sun
dfuchs
parents: 5506
diff changeset
    54
    {
2
90ce3da70b43 Initial load
duke
parents:
diff changeset
    55
        InetSocketAddress isa
14415
7a31b0e0cfaf 6720349: (ch) Channels tests depending on hosts inside Sun
dfuchs
parents: 5506
diff changeset
    56
            = new InetSocketAddress(daytimeServer.getAddress(),
7a31b0e0cfaf 6720349: (ch) Channels tests depending on hosts inside Sun
dfuchs
parents: 5506
diff changeset
    57
                                    daytimeServer.getPort());
2
90ce3da70b43 Initial load
duke
parents:
diff changeset
    58
        SocketChannel sc = SocketChannel.open();
90ce3da70b43 Initial load
duke
parents:
diff changeset
    59
        sc.configureBlocking(false);
90ce3da70b43 Initial load
duke
parents:
diff changeset
    60
        boolean connected = sc.connect(isa);
90ce3da70b43 Initial load
duke
parents:
diff changeset
    61
        int attempts = 0;
90ce3da70b43 Initial load
duke
parents:
diff changeset
    62
90ce3da70b43 Initial load
duke
parents:
diff changeset
    63
        try {
90ce3da70b43 Initial load
duke
parents:
diff changeset
    64
            sc.connect(isa);
90ce3da70b43 Initial load
duke
parents:
diff changeset
    65
            throw new RuntimeException("Allowed another connect call");
90ce3da70b43 Initial load
duke
parents:
diff changeset
    66
        } catch (IllegalStateException ise) {
90ce3da70b43 Initial load
duke
parents:
diff changeset
    67
            // Correct behavior
90ce3da70b43 Initial load
duke
parents:
diff changeset
    68
        }
90ce3da70b43 Initial load
duke
parents:
diff changeset
    69
90ce3da70b43 Initial load
duke
parents:
diff changeset
    70
        if (setBlocking)
90ce3da70b43 Initial load
duke
parents:
diff changeset
    71
            sc.configureBlocking(true);
90ce3da70b43 Initial load
duke
parents:
diff changeset
    72
90ce3da70b43 Initial load
duke
parents:
diff changeset
    73
        if (!connected && select && !setBlocking) {
90ce3da70b43 Initial load
duke
parents:
diff changeset
    74
            Selector selector = SelectorProvider.provider().openSelector();
90ce3da70b43 Initial load
duke
parents:
diff changeset
    75
            sc.register(selector, SelectionKey.OP_CONNECT);
90ce3da70b43 Initial load
duke
parents:
diff changeset
    76
            while (!connected) {
90ce3da70b43 Initial load
duke
parents:
diff changeset
    77
                int keysAdded = selector.select(100);
90ce3da70b43 Initial load
duke
parents:
diff changeset
    78
                if (keysAdded > 0) {
90ce3da70b43 Initial load
duke
parents:
diff changeset
    79
                    Set readyKeys = selector.selectedKeys();
90ce3da70b43 Initial load
duke
parents:
diff changeset
    80
                    Iterator i = readyKeys.iterator();
90ce3da70b43 Initial load
duke
parents:
diff changeset
    81
                    while (i.hasNext()) {
90ce3da70b43 Initial load
duke
parents:
diff changeset
    82
                        SelectionKey sk = (SelectionKey)i.next();
90ce3da70b43 Initial load
duke
parents:
diff changeset
    83
                        SocketChannel nextReady =
90ce3da70b43 Initial load
duke
parents:
diff changeset
    84
                            (SocketChannel)sk.channel();
90ce3da70b43 Initial load
duke
parents:
diff changeset
    85
                        connected = sc.finishConnect();
90ce3da70b43 Initial load
duke
parents:
diff changeset
    86
                    }
90ce3da70b43 Initial load
duke
parents:
diff changeset
    87
                }
90ce3da70b43 Initial load
duke
parents:
diff changeset
    88
            }
90ce3da70b43 Initial load
duke
parents:
diff changeset
    89
            selector.close();
90ce3da70b43 Initial load
duke
parents:
diff changeset
    90
        }
90ce3da70b43 Initial load
duke
parents:
diff changeset
    91
90ce3da70b43 Initial load
duke
parents:
diff changeset
    92
        while (!connected) {
90ce3da70b43 Initial load
duke
parents:
diff changeset
    93
            if (attempts++ > 30)
90ce3da70b43 Initial load
duke
parents:
diff changeset
    94
                throw new RuntimeException("Failed to connect");
90ce3da70b43 Initial load
duke
parents:
diff changeset
    95
            Thread.sleep(100);
90ce3da70b43 Initial load
duke
parents:
diff changeset
    96
            connected = sc.finishConnect();
90ce3da70b43 Initial load
duke
parents:
diff changeset
    97
        }
90ce3da70b43 Initial load
duke
parents:
diff changeset
    98
90ce3da70b43 Initial load
duke
parents:
diff changeset
    99
        ByteBuffer bb = ByteBuffer.allocateDirect(100);
90ce3da70b43 Initial load
duke
parents:
diff changeset
   100
        int bytesRead = 0;
90ce3da70b43 Initial load
duke
parents:
diff changeset
   101
        int totalRead = 0;
90ce3da70b43 Initial load
duke
parents:
diff changeset
   102
        while (totalRead < 20) {
90ce3da70b43 Initial load
duke
parents:
diff changeset
   103
            bytesRead = sc.read(bb);
90ce3da70b43 Initial load
duke
parents:
diff changeset
   104
            if (bytesRead > 0)
90ce3da70b43 Initial load
duke
parents:
diff changeset
   105
                totalRead += bytesRead;
90ce3da70b43 Initial load
duke
parents:
diff changeset
   106
            if (bytesRead < 0)
90ce3da70b43 Initial load
duke
parents:
diff changeset
   107
                throw new RuntimeException("Message shorter than expected");
90ce3da70b43 Initial load
duke
parents:
diff changeset
   108
        }
90ce3da70b43 Initial load
duke
parents:
diff changeset
   109
        bb.position(bb.position() - 2);         // Drop CRLF
90ce3da70b43 Initial load
duke
parents:
diff changeset
   110
        bb.flip();
90ce3da70b43 Initial load
duke
parents:
diff changeset
   111
        CharBuffer cb = Charset.forName("US-ASCII").newDecoder().decode(bb);
90ce3da70b43 Initial load
duke
parents:
diff changeset
   112
        System.err.println(isa + " says: \"" + cb + "\"");
90ce3da70b43 Initial load
duke
parents:
diff changeset
   113
        sc.close();
90ce3da70b43 Initial load
duke
parents:
diff changeset
   114
    }
90ce3da70b43 Initial load
duke
parents:
diff changeset
   115
14415
7a31b0e0cfaf 6720349: (ch) Channels tests depending on hosts inside Sun
dfuchs
parents: 5506
diff changeset
   116
    static void test2(TestServers.DayTimeServer daytimeServer) throws Exception {
2
90ce3da70b43 Initial load
duke
parents:
diff changeset
   117
        InetSocketAddress isa
14415
7a31b0e0cfaf 6720349: (ch) Channels tests depending on hosts inside Sun
dfuchs
parents: 5506
diff changeset
   118
            = new InetSocketAddress(daytimeServer.getAddress(),
7a31b0e0cfaf 6720349: (ch) Channels tests depending on hosts inside Sun
dfuchs
parents: 5506
diff changeset
   119
                                    daytimeServer.getPort());
2
90ce3da70b43 Initial load
duke
parents:
diff changeset
   120
        boolean done = false;
90ce3da70b43 Initial load
duke
parents:
diff changeset
   121
        int globalAttempts = 0;
14415
7a31b0e0cfaf 6720349: (ch) Channels tests depending on hosts inside Sun
dfuchs
parents: 5506
diff changeset
   122
        int connectSuccess = 0;
2
90ce3da70b43 Initial load
duke
parents:
diff changeset
   123
        while (!done) {
14415
7a31b0e0cfaf 6720349: (ch) Channels tests depending on hosts inside Sun
dfuchs
parents: 5506
diff changeset
   124
            // When using a local daytime server it is not always possible
7a31b0e0cfaf 6720349: (ch) Channels tests depending on hosts inside Sun
dfuchs
parents: 5506
diff changeset
   125
            // to get a pending connection, as sc.connect(isa) may always
7a31b0e0cfaf 6720349: (ch) Channels tests depending on hosts inside Sun
dfuchs
parents: 5506
diff changeset
   126
            // return true.
7a31b0e0cfaf 6720349: (ch) Channels tests depending on hosts inside Sun
dfuchs
parents: 5506
diff changeset
   127
            // So we're going to throw the exception only if there was
7a31b0e0cfaf 6720349: (ch) Channels tests depending on hosts inside Sun
dfuchs
parents: 5506
diff changeset
   128
            // at least 1 case where we did not manage to connect.
7a31b0e0cfaf 6720349: (ch) Channels tests depending on hosts inside Sun
dfuchs
parents: 5506
diff changeset
   129
            if (globalAttempts++ > 50) {
7a31b0e0cfaf 6720349: (ch) Channels tests depending on hosts inside Sun
dfuchs
parents: 5506
diff changeset
   130
                if (globalAttempts == connectSuccess + 1) {
7a31b0e0cfaf 6720349: (ch) Channels tests depending on hosts inside Sun
dfuchs
parents: 5506
diff changeset
   131
                    System.out.println("Can't fully test on "
7a31b0e0cfaf 6720349: (ch) Channels tests depending on hosts inside Sun
dfuchs
parents: 5506
diff changeset
   132
                            + System.getProperty("os.name"));
7a31b0e0cfaf 6720349: (ch) Channels tests depending on hosts inside Sun
dfuchs
parents: 5506
diff changeset
   133
                    break;
7a31b0e0cfaf 6720349: (ch) Channels tests depending on hosts inside Sun
dfuchs
parents: 5506
diff changeset
   134
                }
2
90ce3da70b43 Initial load
duke
parents:
diff changeset
   135
                throw new RuntimeException("Failed to connect");
14415
7a31b0e0cfaf 6720349: (ch) Channels tests depending on hosts inside Sun
dfuchs
parents: 5506
diff changeset
   136
            }
2
90ce3da70b43 Initial load
duke
parents:
diff changeset
   137
            SocketChannel sc = SocketChannel.open();
90ce3da70b43 Initial load
duke
parents:
diff changeset
   138
            sc.configureBlocking(false);
90ce3da70b43 Initial load
duke
parents:
diff changeset
   139
            boolean connected = sc.connect(isa);
90ce3da70b43 Initial load
duke
parents:
diff changeset
   140
            int localAttempts = 0;
90ce3da70b43 Initial load
duke
parents:
diff changeset
   141
            while (!connected) {
90ce3da70b43 Initial load
duke
parents:
diff changeset
   142
                if (localAttempts++ > 500)
90ce3da70b43 Initial load
duke
parents:
diff changeset
   143
                    throw new RuntimeException("Failed to connect");
90ce3da70b43 Initial load
duke
parents:
diff changeset
   144
                connected = sc.finishConnect();
90ce3da70b43 Initial load
duke
parents:
diff changeset
   145
                if (connected) {
90ce3da70b43 Initial load
duke
parents:
diff changeset
   146
                    done = true;
90ce3da70b43 Initial load
duke
parents:
diff changeset
   147
                    break;
90ce3da70b43 Initial load
duke
parents:
diff changeset
   148
                }
90ce3da70b43 Initial load
duke
parents:
diff changeset
   149
                Thread.sleep(10);
90ce3da70b43 Initial load
duke
parents:
diff changeset
   150
            }
14415
7a31b0e0cfaf 6720349: (ch) Channels tests depending on hosts inside Sun
dfuchs
parents: 5506
diff changeset
   151
            if (connected) {
7a31b0e0cfaf 6720349: (ch) Channels tests depending on hosts inside Sun
dfuchs
parents: 5506
diff changeset
   152
                connectSuccess++;
7a31b0e0cfaf 6720349: (ch) Channels tests depending on hosts inside Sun
dfuchs
parents: 5506
diff changeset
   153
            }
2
90ce3da70b43 Initial load
duke
parents:
diff changeset
   154
            sc.close();
90ce3da70b43 Initial load
duke
parents:
diff changeset
   155
        }
90ce3da70b43 Initial load
duke
parents:
diff changeset
   156
    }
90ce3da70b43 Initial load
duke
parents:
diff changeset
   157
90ce3da70b43 Initial load
duke
parents:
diff changeset
   158
}