1 /*
   2  * Copyright (c) 2010, 2015, 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 socket adapter sendUrgentData method
  26  * @bug 6963907

  27  */
  28 
  29 import java.net.*;
  30 import java.nio.ByteBuffer;
  31 import java.nio.channels.*;
  32 import java.io.IOException;
  33 import java.util.Random;
  34 
  35 public class OutOfBand {
  36 
  37     private static final Random rand = new Random();
  38 
  39     public static void main(String[] args) throws Exception {
  40         ServerSocketChannel ssc = null;
  41         SocketChannel sc1 = null;
  42         SocketChannel sc2 = null;
  43 
  44         try {
  45 
  46             // establish loopback connection
  47             ssc = ServerSocketChannel.open().bind(new InetSocketAddress(0));
  48             InetAddress lh = InetAddress.getLocalHost();
  49             SocketAddress remote =
  50                 new InetSocketAddress(lh, ssc.socket().getLocalPort());
  51             sc1 = SocketChannel.open(remote);
  52             sc2 = ssc.accept();
  53 
  54             // enable SO_OOBLINE on server side
  55             sc2.socket().setOOBInline(true);
  56 
  57             // run tests
  58             test1(sc1, sc2);
  59             test2(sc1, sc2);
  60             test3(sc1, sc2);
  61         } finally {
  62             if (sc1 != null) sc1.close();
  63             if (sc2 != null) sc2.close();
  64             if (ssc != null) ssc.close();
  65         }
  66     }
  67 
  68     /**
  69      * Basic test to check that OOB/TCP urgent byte is received.
  70      */
  71     static void test1(SocketChannel client, SocketChannel server)
  72         throws Exception
  73     {
  74         assert server.socket().getOOBInline();
  75         ByteBuffer bb = ByteBuffer.allocate(100);
  76         for (int i=0; i<1000; i++) {
  77             int b1 = -127 + rand.nextInt(384);
  78             client.socket().sendUrgentData(b1);
  79 
  80             bb.clear();
  81             if (server.read(bb) != 1)
  82                 throw new RuntimeException("One byte expected");
  83             bb.flip();
  84             byte b2 = bb.get();
  85             if ((byte)b1 != b2)
  86                 throw new RuntimeException("Unexpected byte");
  87         }
  88     }
  89 
  90     /**
  91      * Basic test to check that OOB/TCP urgent byte is received, maybe with
  92      * OOB mark changing.
  93      */
  94     static void test2(final SocketChannel client, SocketChannel server)
  95         throws Exception
  96     {
  97         assert server.socket().getOOBInline();
  98         Runnable sender = new Runnable() {
  99             public void run() {
 100                 try {
 101                     for (int i=0; i<256; i++)
 102                         client.socket().sendUrgentData(i);
 103                 } catch (IOException ioe) {
 104                     ioe.printStackTrace();
 105                 }
 106             }
 107         };
 108         Thread thr = new Thread(sender);
 109         thr.start();
 110 
 111         ByteBuffer bb = ByteBuffer.allocate(256);
 112         while (bb.hasRemaining()) {
 113             if (server.read(bb) < 0)
 114                 throw new RuntimeException("Unexpected EOF");
 115         }
 116         bb.flip();
 117         byte expect = 0;
 118         while (bb.hasRemaining()) {
 119             if (bb.get() != expect)
 120                 throw new RuntimeException("Unexpected byte");
 121             expect++;
 122         }
 123 
 124         thr.join();
 125     }
 126 
 127     /**
 128      * Test that is close to some real world examples where an urgent byte is
 129      * used to "cancel" a long running query or transaction on the server.
 130      */
 131     static void test3(SocketChannel client, final SocketChannel server)
 132         throws Exception
 133     {
 134         final int STOP = rand.nextInt(256);
 135 
 136         assert server.socket().getOOBInline();
 137         Runnable reader = new Runnable() {
 138             public void run() {
 139                 ByteBuffer bb = ByteBuffer.allocate(100);
 140                 try {
 141                     int n = server.read(bb);
 142                     if (n != 1) {
 143                         String msg = (n < 0) ? "Unexpected EOF" :
 144                                                "One byte expected";
 145                         throw new RuntimeException(msg);
 146                     }
 147                     bb.flip();
 148                     if (bb.get() != (byte)STOP)
 149                         throw new RuntimeException("Unexpected byte");
 150                     bb.flip();
 151                     server.write(bb);
 152                 } catch (IOException ioe) {
 153                     ioe.printStackTrace();
 154                 }
 155 
 156             }
 157         };
 158 
 159         Thread thr = new Thread(reader);
 160         thr.start();
 161 
 162         // "stop" server
 163         client.socket().sendUrgentData(STOP);
 164 
 165         // wait for server reply
 166         ByteBuffer bb = ByteBuffer.allocate(100);
 167         int n = client.read(bb);
 168         if (n != 1)
 169             throw new RuntimeException("Unexpected number of bytes");
 170         bb.flip();
 171         if (bb.get() != (byte)STOP)
 172             throw new RuntimeException("Unexpected reply");
 173 
 174         thr.join();
 175     }
 176 }
--- EOF ---