1/*
2 * Copyright (c) 2001, 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/*
25 * @test
26 * @bug 4092038
27 * @summary TCP Urgent data support
28 *
29 */
30
31import java.net.*;
32import java.io.*;
33
34public class UrgentDataTest {
35
36    Object opref;
37    boolean isClient, isServer;
38    String  clHost;
39    ServerSocket listener;
40    Socket client, server;
41    int  clPort;
42    InputStream clis, sis;
43    OutputStream clos, sos;
44
45    static void usage () {
46        System.out.println ("   usage: java UrgentDataTest <runs client and server together>");
47        System.out.println ("   usage: java UrgentDataTest -server <runs server alone>");
48        System.out.println ("   usage: java UrgentDataTest -client host port <runs client and connects to"+
49            " specified server>");
50    }
51
52    public static void main (String args[]) {
53        try {
54            UrgentDataTest test = new UrgentDataTest ();
55            if (args.length == 0) {
56                test.listener = new ServerSocket (0);
57                test.isClient = true;
58                test.isServer = true;
59                test.clHost = "127.0.0.1";
60                test.clPort = test.listener.getLocalPort();
61                test.run();
62            } else if (args[0].equals ("-server")) {
63                test.listener = new ServerSocket (0);
64                System.out.println ("Server listening on port " + test.listener.getLocalPort());
65                test.isClient = false;
66                test.isServer = true;
67                test.run();
68                System.out.println ("Server: Completed OK");
69            } else if (args[0].equals ("-client")) {
70                test.isClient = true;
71                test.isServer = false;
72                test.clHost = args [1];
73                test.clPort = Integer.parseInt (args[2]);
74                test.run();
75                System.out.println ("Client: Completed OK");
76            } else {
77                usage ();
78            }
79        }
80        catch (ArrayIndexOutOfBoundsException e) {
81            usage();
82        }
83        catch (NumberFormatException e) {
84            usage();
85        }
86        catch (Exception e) {
87            e.printStackTrace ();
88            throw new RuntimeException ("Exception caught");
89        }
90    }
91
92    public void run () throws Exception {
93        try {
94            if (isClient) {
95                client = new Socket (clHost, clPort);
96                clis = client.getInputStream();
97                clos = client.getOutputStream();
98                client.setOOBInline (true);
99                if (client.getOOBInline() != true) {
100                    throw new RuntimeException ("Setting OOBINLINE failed");
101                }
102            }
103            if (isServer) {
104                server = listener.accept ();
105                sis = server.getInputStream();
106                sos = server.getOutputStream();
107            }
108            if (isClient) {
109                clos.write ("Hello".getBytes ());
110                client.sendUrgentData (100);
111                clos.write ("world".getBytes ());
112            }
113            // read Hello world from server (during which oob byte must have been dropped)
114            String s = "Helloworld";
115            if (isServer) {
116                for (int y=0; y<s.length(); y++) {
117                    int c = sis.read ();
118                    if (c != (int)s.charAt (y)) {
119                        throw new RuntimeException ("Unexpected character read");
120                    }
121                }
122                // Do the same from server to client
123                sos.write ("Hello".getBytes ());
124                server.sendUrgentData (101);
125                sos.write ("World".getBytes ());
126            }
127            if (isClient) {
128                // read Hello world from client (during which oob byte must have been read)
129                s="Hello";
130                for (int y=0; y<s.length(); y++) {
131                    int c = clis.read ();
132                    if (c != (int)s.charAt (y)) {
133                        throw new RuntimeException ("Unexpected character read");
134                    }
135                }
136                if (clis.read() != 101) {
137                    throw new RuntimeException ("OOB byte not received");
138                }
139                s="World";
140                for (int y=0; y<s.length(); y++) {
141                    int c = clis.read ();
142                    if (c != (int)s.charAt (y)) {
143                        throw new RuntimeException ("Unexpected character read");
144                    }
145                }
146            }
147        } finally {
148            if (listener != null) listener.close();
149            if (client != null) client.close ();
150            if (server != null) server.close ();
151        }
152    }
153}
154