1 /*
   2  * Copyright (c) 2000, 2017, 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 socketchannel vector IO (use -Dseed=X to set PRNG seed)
  26  * @library .. /lib/testlibrary/
  27  * @build jdk.testlibrary.RandomFactory
  28  * @run main VectorIO
  29  * @key randomness
  30  */
  31 
  32 import java.io.*;
  33 import java.net.*;
  34 import java.nio.*;
  35 import java.nio.channels.*;
  36 import java.util.*;
  37 import jdk.testlibrary.RandomFactory;
  38 
  39 public class VectorIO {
  40 
  41     private static Random generator = RandomFactory.getRandom();
  42 
  43     static int testSize;
  44 
  45     public static void main(String[] args) throws Exception {
  46         testSize = 1;
  47         runTest();
  48         for(int i=15; i<18; i++) {
  49             testSize = i;
  50             runTest();
  51         }
  52     }
  53 
  54     static void runTest() throws Exception {
  55         System.err.println("Length " + testSize);
  56         Server sv = new Server(testSize);
  57         sv.start();
  58         bufferTest(sv.port());
  59         if (sv.finish(8000) == 0)
  60             throw new Exception("Failed: Length = " + testSize);
  61     }
  62 
  63     static void bufferTest(int port) throws Exception {
  64         ByteBuffer[] bufs = new ByteBuffer[testSize];
  65         long total = 0L;
  66         for(int i=0; i<testSize; i++) {
  67             String source = "buffer" + i;
  68             if (generator.nextBoolean())
  69                 bufs[i] = ByteBuffer.allocateDirect(source.length());
  70             else
  71                 bufs[i] = ByteBuffer.allocate(source.length());
  72 
  73             bufs[i].put(source.getBytes("8859_1"));
  74             bufs[i].flip();
  75             total += bufs[i].remaining();
  76         }
  77 
  78         // Get a connection to the server
  79         InetAddress lh = InetAddress.getLocalHost();
  80         InetSocketAddress isa = new InetSocketAddress(lh, port);
  81         SocketChannel sc = SocketChannel.open();
  82         sc.connect(isa);
  83         sc.configureBlocking(generator.nextBoolean());
  84 
  85         // Write the data out
  86         long rem = total;
  87         while (rem > 0L) {
  88             long bytesWritten = sc.write(bufs);
  89             if (bytesWritten == 0) {
  90                 if (sc.isBlocking())
  91                     throw new RuntimeException("write did not block");
  92                 Thread.sleep(50);
  93             } else {
  94                 rem -= bytesWritten;
  95             }
  96         }
  97 
  98         // Clean up
  99         sc.close();
 100     }
 101 
 102     static class Server
 103         extends TestThread
 104     {
 105         final int testSize;
 106         final ServerSocketChannel ssc;
 107 
 108         Server(int testSize) throws IOException {
 109             super("Server " + testSize);
 110             this.testSize = testSize;
 111             this.ssc = ServerSocketChannel.open().bind(new InetSocketAddress(0));
 112         }
 113 
 114         int port() {
 115             return ssc.socket().getLocalPort();
 116         }
 117 
 118         void go() throws Exception {
 119             bufferTest();
 120         }
 121 
 122         void bufferTest() throws Exception {
 123             long total = 0L;
 124             ByteBuffer[] bufs = new ByteBuffer[testSize];
 125             for(int i=0; i<testSize; i++) {
 126                 String source = "buffer" + i;
 127                 if (generator.nextBoolean())
 128                     bufs[i] = ByteBuffer.allocateDirect(source.length());
 129                 else
 130                     bufs[i] = ByteBuffer.allocate(source.length());
 131                 total += bufs[i].capacity();
 132             }
 133 
 134             // Get a connection from client
 135             SocketChannel sc = null;
 136 
 137             try {
 138 
 139                 ssc.configureBlocking(false);
 140 
 141                 for (;;) {
 142                     sc = ssc.accept();
 143                     if (sc != null)
 144                         break;
 145                     Thread.sleep(50);
 146                 }
 147 
 148                 sc.configureBlocking(generator.nextBoolean());
 149 
 150                 // Read data into multiple buffers
 151                 long avail = total;
 152                 while (avail > 0) {
 153                     long bytesRead = sc.read(bufs);
 154                     if (bytesRead < 0)
 155                         break;
 156                     if (bytesRead == 0) {
 157                         if (sc.isBlocking())
 158                             throw new RuntimeException("read did not block");
 159                         Thread.sleep(50);
 160                     }
 161                     avail -= bytesRead;
 162                 }
 163 
 164                 // Check results
 165                 for(int i=0; i<testSize; i++) {
 166                     String expected = "buffer" + i;
 167                     bufs[i].flip();
 168                     int size = bufs[i].capacity();
 169                     byte[] data = new byte[size];
 170                     for(int j=0; j<size; j++)
 171                         data[j] = bufs[i].get();
 172                     String message = new String(data, "8859_1");
 173                     if (!message.equals(expected))
 174                         throw new Exception("Wrong data: Got "
 175                                             + message + ", expected "
 176                                             + expected);
 177                 }
 178 
 179             } finally {
 180                 // Clean up
 181                 ssc.close();
 182                 if (sc != null)
 183                     sc.close();
 184             }
 185 
 186         }
 187 
 188     }
 189 
 190 }