1 /*
   2  * Copyright (c) 2002, 2010, 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 /* @test
  25  * @summary Test selection of ready pipe
  26  * @key randomness
  27  */
  28 
  29 import java.io.*;
  30 import java.nio.ByteBuffer;
  31 import java.nio.channels.*;
  32 import java.nio.channels.spi.SelectorProvider;
  33 import java.util.Random;
  34 
  35 /**
  36  * Testing PipeChannel
  37  */
  38 public class SelectPipe {
  39 
  40     private static Random generator = new Random();
  41 
  42     public static void main(String[] args) throws Exception {
  43 
  44         SelectorProvider sp = SelectorProvider.provider();
  45         Selector selector = Selector.open();
  46         Pipe p = sp.openPipe();
  47         Pipe.SinkChannel sink = p.sink();
  48         Pipe.SourceChannel source = p.source();
  49 
  50         source.configureBlocking(false);
  51         sink.configureBlocking(false);
  52 
  53         SelectionKey readkey = source.register(selector, SelectionKey.OP_READ);
  54         SelectionKey writekey = sink.register(selector, SelectionKey.OP_WRITE);
  55 
  56         ByteBuffer outgoingdata = ByteBuffer.allocateDirect(10);
  57         byte[] someBytes = new byte[10];
  58         generator.nextBytes(someBytes);
  59         outgoingdata.put(someBytes);
  60         outgoingdata.flip();
  61 
  62         int totalWritten = 0;
  63         while (totalWritten < 10) {
  64             int written = sink.write(outgoingdata);
  65             if (written < 0)
  66                 throw new Exception("Write failed");
  67             totalWritten += written;
  68         }
  69 
  70         if (selector.select(1000) == 0) {
  71             throw new Exception("test failed");
  72         }
  73 
  74         ByteBuffer incomingdata = ByteBuffer.allocateDirect(10);
  75         int totalRead = 0;
  76         do {
  77             int bytesRead = source.read(incomingdata);
  78             if (bytesRead > 0)
  79                 totalRead += bytesRead;
  80         } while(totalRead < 10);
  81 
  82         sink.close();
  83         source.close();
  84         selector.close();
  85 
  86         for(int i=0; i<10; i++)
  87             if (outgoingdata.get(i) != incomingdata.get(i))
  88                 throw new Exception("Pipe failed");
  89     }
  90 }