1/*
2 * Copyright (c) 2001, Oracle and/or its affiliates. All rights reserved.
3 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
4 *
5 * This code is free software; you can redistribute it and/or modify it
6 * under the terms of the GNU General Public License version 2 only, as
7 * published by the Free Software Foundation.
8 *
9 * This code is distributed in the hope that it will be useful, but WITHOUT
10 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
11 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
12 * version 2 for more details (a copy is included in the LICENSE file that
13 * accompanied this code).
14 *
15 * You should have received a copy of the GNU General Public License version
16 * 2 along with this work; if not, write to the Free Software Foundation,
17 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
18 *
19 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
20 * or visit www.oracle.com if you need additional information or have any
21 * questions.
22 */
23
24/*
25 * Tests that a thread blocked in ServerSocket.accept
26 * throws a SocketException if the socket is asynchronously closed.
27 */
28import java.io.IOException;
29import java.net.*;
30import java.util.concurrent.CountDownLatch;
31
32public class ServerSocket_accept extends AsyncCloseTest implements Runnable {
33    private final ServerSocket ss;
34    private final int timeout;
35    private final CountDownLatch latch;
36
37    public ServerSocket_accept() throws IOException {
38       this(0);
39    }
40
41    public ServerSocket_accept(int timeout) throws IOException {
42        this.timeout = timeout;
43        latch = new CountDownLatch(1);
44        ss = new ServerSocket(0);
45    }
46
47    public String description() {
48        String s = "ServerSocket.accept()";
49        if (timeout > 0) {
50            s += " (with timeout)";
51        }
52        return s;
53    }
54
55    public void run() {
56        try {
57            latch.countDown();
58            Socket s = ss.accept();
59            failed("ServerSocket.accept() returned unexpectly!!");
60        } catch (SocketException se) {
61            closed();
62        } catch (Exception e) {
63            failed(e.getMessage());
64        }
65    }
66
67    public AsyncCloseTest go(){
68        try {
69            Thread thr = new Thread(this);
70            thr.start();
71            latch.await();
72            Thread.sleep(5000); //sleep, so ServerSocket.accept() can block
73            ss.close();
74            thr.join();
75
76            if (isClosed()) {
77                return passed();
78            } else {
79                return failed("ServerSocket.accept() wasn't preempted");
80            }
81        } catch (Exception x) {
82            failed(x.getMessage());
83            throw new RuntimeException(x);
84        }
85    }
86}
87