test/java/nio/channels/Selector/Wakeup.java

Print this page




  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 /* @test
  25  * @bug 6405995
  26  * @summary Unit test for selector wakeup and interruption
  27  * @library ..
  28  */
  29 
  30 import java.io.*;
  31 import java.net.*;
  32 import java.nio.*;
  33 import java.nio.channels.*;
  34 import java.util.Random;
  35 
  36 public class Wakeup {
  37 
  38     static void sleep(int millis) {
  39         try {
  40             Thread.sleep(millis);
  41         } catch (InterruptedException x) {
  42             x.printStackTrace();
  43         }
  44     }
  45 
  46     static class Sleeper extends TestThread {
  47         volatile boolean started = false;
  48         volatile int entries = 0;
  49         volatile int wakeups = 0;
  50         volatile boolean wantInterrupt = false;
  51         volatile boolean gotInterrupt = false;
  52         volatile Exception exception = null;
  53         volatile boolean closed = false;
  54         Object gate = new Object();
  55 
  56         Selector sel;
  57 
  58         Sleeper(Selector sel) {
  59             super("Sleeper", System.err);
  60             this.sel = sel;
  61         }
  62 













  63         public void go() throws Exception {
  64             started = true;
  65             for (;;) {
  66                 synchronized (gate) { }
  67                 entries++;
  68                 try {
  69                     sel.select();
  70                 } catch (ClosedSelectorException x) {
  71                     closed = true;
  72                 }
  73                 boolean intr = Thread.currentThread().isInterrupted();
  74                 wakeups++;
  75                 System.err.println("Wakeup " + wakeups
  76                                    + (closed ? " (closed)" : "")
  77                                    + (intr ? " (intr)" : ""));
  78                 if (wakeups > 1000)
  79                     throw new Exception("Too many wakeups");
  80                 if (closed)
  81                     return;
  82                 if (wantInterrupt) {


  86                     wantInterrupt = false;
  87                 }
  88             }
  89         }
  90 
  91     }
  92 
  93     private static int checkedWakeups = 0;
  94 
  95     private static void check(Sleeper sleeper, boolean intr)
  96         throws Exception
  97     {
  98         checkedWakeups++;
  99         if (sleeper.wakeups > checkedWakeups) {
 100             sleeper.finish(100);
 101             throw new Exception("Sleeper has run ahead");
 102         }
 103         int n = 0;
 104         while (sleeper.wakeups < checkedWakeups) {
 105             sleep(50);
 106             if ((n += 50) > 1000) {
 107                 sleeper.finish(100);
 108                 throw new Exception("Sleeper appears to be dead ("
 109                                     + checkedWakeups + ")");
 110             }
 111         }
 112         if (sleeper.wakeups > checkedWakeups) {
 113             sleeper.finish(100);
 114             throw new Exception("Too many wakeups: Expected "
 115                                 + checkedWakeups
 116                                 + ", got " + sleeper.wakeups);
 117         }
 118         if (intr) {
 119             n = 0;
 120             // Interrupts can sometimes be delayed, so wait
 121             while (!sleeper.gotInterrupt) {
 122                 sleep(50);
 123                 if ((n += 50) > 1000) {
 124                     sleeper.finish(100);
 125                     throw new Exception("Interrupt never delivered");
 126                 }
 127             }
 128             sleeper.gotInterrupt = false;
 129         }
 130         System.err.println("Check " + checkedWakeups
 131                            + (intr ? " (intr " + n + ")" : ""));
 132     }
 133 
 134     public static void main(String[] args) throws Exception {
 135 
 136         Selector sel = Selector.open();
 137 
 138         // Wakeup before select
 139         sel.wakeup();
 140 
 141         Sleeper sleeper = new Sleeper(sel);
 142 
 143         sleeper.start();
 144         while (!sleeper.started)
 145             sleep(50);
 146 
 147         check(sleeper, false);          // 1
 148 
 149         for (int i = 2; i < 5; i++) {
 150             // Wakeup during select
 151             sel.wakeup();
 152             check(sleeper, false);      // 2 .. 4
 153         }
 154 
 155         // Double wakeup

 156         synchronized (sleeper.gate) {
 157             sel.wakeup();
 158             check(sleeper, false);      // 5
 159             sel.wakeup();
 160             sel.wakeup();
 161         }
 162         check(sleeper, false);          // 6
 163 
 164         // Interrupt

 165         synchronized (sleeper.gate) {
 166             sleeper.wantInterrupt = true;
 167             sleeper.interrupt();
 168             check(sleeper, true);       // 7
 169         }
 170 
 171         // Interrupt before select
 172         while (sleeper.entries < 8)
 173             Thread.yield();
 174         synchronized (sleeper.gate) {
 175             sel.wakeup();
 176             check(sleeper, false);      // 8
 177             sleeper.wantInterrupt = true;
 178             sleeper.interrupt();
 179             sleep(50);
 180         }
 181         check(sleeper, true);           // 9
 182 
 183         // Close during select
 184         while (sleeper.entries < 10)
 185             Thread.yield();
 186         synchronized (sleeper.gate) {
 187             sel.close();
 188             check(sleeper, false);      // 10
 189         }
 190 
 191         if (sleeper.finish(200) == 0)
 192             throw new Exception("Test failed");
 193         if (!sleeper.closed)
 194             throw new Exception("Selector not closed");
 195     }
 196 
 197 }


  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 /* @test
  25  * @bug 6405995
  26  * @summary Unit test for selector wakeup and interruption
  27  * @library ..
  28  */
  29 
  30 import java.nio.channels.ClosedSelectorException;
  31 import java.nio.channels.Selector;



  32 
  33 public class Wakeup {
  34 
  35     static void sleep(int millis) {
  36         try {
  37             Thread.sleep(millis);
  38         } catch (InterruptedException x) {
  39             x.printStackTrace();
  40         }
  41     }
  42 
  43     static class Sleeper extends TestThread {
  44         volatile boolean started = false;
  45         volatile int entries = 0;
  46         volatile int wakeups = 0;
  47         volatile boolean wantInterrupt = false;
  48         volatile boolean gotInterrupt = false;
  49         volatile Exception exception = null;
  50         volatile boolean closed = false;
  51         final Object gate = new Object();
  52         final Selector sel;

  53 
  54         Sleeper(Selector sel) {
  55             super("Sleeper", System.err);
  56             this.sel = sel;
  57         }
  58 
  59         void awaitStarted() throws InterruptedException {
  60             while (!started) {
  61                 Thread.sleep(10);
  62             }
  63         }
  64 
  65 
  66         void awaitAdvance(int value) throws InterruptedException {
  67             while (entries < value) {
  68                 Thread.sleep(10);
  69             }
  70         }
  71 
  72         public void go() throws Exception {
  73             started = true;
  74             for (;;) {
  75                 synchronized (gate) { }
  76                 entries++;
  77                 try {
  78                     sel.select();
  79                 } catch (ClosedSelectorException x) {
  80                     closed = true;
  81                 }
  82                 boolean intr = Thread.currentThread().isInterrupted();
  83                 wakeups++;
  84                 System.err.println("Wakeup " + wakeups
  85                                    + (closed ? " (closed)" : "")
  86                                    + (intr ? " (intr)" : ""));
  87                 if (wakeups > 1000)
  88                     throw new Exception("Too many wakeups");
  89                 if (closed)
  90                     return;
  91                 if (wantInterrupt) {


  95                     wantInterrupt = false;
  96                 }
  97             }
  98         }
  99 
 100     }
 101 
 102     private static int checkedWakeups = 0;
 103 
 104     private static void check(Sleeper sleeper, boolean intr)
 105         throws Exception
 106     {
 107         checkedWakeups++;
 108         if (sleeper.wakeups > checkedWakeups) {
 109             sleeper.finish(100);
 110             throw new Exception("Sleeper has run ahead");
 111         }
 112         int n = 0;
 113         while (sleeper.wakeups < checkedWakeups) {
 114             sleep(50);
 115             if ((n += 50) > 10000) {
 116                 sleeper.finish(100);
 117                 throw new Exception("Sleeper appears to be dead ("
 118                                     + checkedWakeups + ")");
 119             }
 120         }
 121         if (sleeper.wakeups > checkedWakeups) {
 122             sleeper.finish(100);
 123             throw new Exception("Too many wakeups: Expected "
 124                                 + checkedWakeups
 125                                 + ", got " + sleeper.wakeups);
 126         }
 127         if (intr) {
 128             n = 0;
 129             // Interrupts can sometimes be delayed, so wait
 130             while (!sleeper.gotInterrupt) {
 131                 sleep(50);
 132                 if ((n += 50) > 10000) {
 133                     sleeper.finish(100);
 134                     throw new Exception("Interrupt never delivered");
 135                 }
 136             }
 137             sleeper.gotInterrupt = false;
 138         }
 139         System.err.println("Check " + checkedWakeups
 140                            + (intr ? " (intr " + n + ")" : ""));
 141     }
 142 
 143     public static void main(String[] args) throws Exception {
 144 
 145         Selector sel = Selector.open();
 146 
 147         // Wakeup before select
 148         sel.wakeup();
 149 
 150         Sleeper sleeper = new Sleeper(sel);
 151 
 152         sleeper.start();
 153         sleeper.awaitStarted();

 154 
 155         check(sleeper, false);          // 1
 156 
 157         for (int i = 2; i < 5; i++) {
 158             // Wakeup during select
 159             sel.wakeup();
 160             check(sleeper, false);      // 2 .. 4
 161         }
 162 
 163         // Double wakeup
 164         sleeper.awaitAdvance(5);
 165         synchronized (sleeper.gate) {
 166             sel.wakeup();
 167             check(sleeper, false);      // 5
 168             sel.wakeup();
 169             sel.wakeup();
 170         }
 171         check(sleeper, false);          // 6
 172 
 173         // Interrupt
 174         sleeper.awaitAdvance(7);
 175         synchronized (sleeper.gate) {
 176             sleeper.wantInterrupt = true;
 177             sleeper.interrupt();
 178             check(sleeper, true);       // 7
 179         }
 180 
 181         // Interrupt before select
 182         sleeper.awaitAdvance(8);

 183         synchronized (sleeper.gate) {
 184             sel.wakeup();
 185             check(sleeper, false);      // 8
 186             sleeper.wantInterrupt = true;
 187             sleeper.interrupt();
 188             sleep(50);
 189         }
 190         check(sleeper, true);           // 9
 191 
 192         // Close during select
 193         sleeper.awaitAdvance(10);

 194         synchronized (sleeper.gate) {
 195             sel.close();
 196             check(sleeper, false);      // 10
 197         }
 198 
 199         if (sleeper.finish(2000) == 0)
 200             throw new Exception("Sleeper did not terminate");
 201         if (!sleeper.closed)
 202             throw new Exception("Selector not closed");
 203     }
 204 
 205 }