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