/*
* Copyright (c) 2000, 2012, Oracle and/or its affiliates. All rights reserved.
* DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
*
* This code is free software; you can redistribute it and/or modify it
* under the terms of the GNU General Public License version 2 only, as
* published by the Free Software Foundation.
*
* This code is distributed in the hope that it will be useful, but WITHOUT
* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
* FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
* version 2 for more details (a copy is included in the LICENSE file that
* accompanied this code).
*
* You should have received a copy of the GNU General Public License version
* 2 along with this work; if not, write to the Free Software Foundation,
* Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
*
* Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
* or visit www.oracle.com if you need additional information or have any
* questions.
*/
/*
* @test
*
* @bug 4204320
*
* @summary DatagramSocket.send should throw exception when connected
* to an invalid destination (on platforms that support it).
*/
import java.net.*;
import java.util.*;
import java.io.InterruptedIOException;
public class SendDatagramToBadAddress {
static boolean debug = false;
public static boolean OSsupportsFeature () {
Properties p = System.getProperties ();
String v;
if (p.getProperty ("os.name").equals ("Windows 2000"))
return (true);
if (p.getProperty ("os.name").equals ("Linux"))
return (true);
if (p.getProperty ("os.name").startsWith ("Mac OS"))
return (true);
// Check for specific Solaris version from here
v = p.getProperty ("os.arch");
if (!v.equalsIgnoreCase ("sparc"))
return (false);
v = p.getProperty ("os.name");
if (!v.equalsIgnoreCase ("Solaris") && !v.equalsIgnoreCase ("SunOS"))
return (false);
v = p.getProperty ("os.version");
if (v.equals ("5.8") || v.equals ("8"))
return (false);
return (true);
}
static void print (String s) {
if (debug)
System.out.println (s);
}
class Server {
DatagramSocket server;
byte[] buf = new byte [128];
DatagramPacket pack = new DatagramPacket (buf, buf.length);
public Server (DatagramSocket s) {
server = s;
}
public void receive (int loop, boolean expectError) throws Exception {
for (int i=0; i<loop; i++) {
try {
server.receive (pack);
} catch (Exception e) {
if (expectError) {
print ("Got expected error: " + e);
continue;
} else {
print ("Got: " + new String (pack.getData()));
print ("Expected: " + new String (buf));
throw new Exception ("Error reading data: Iter " +i);
}
}
String s1 = "Hello, server"+i;
byte[] buf = s1.getBytes();
if (!s1.equals (new String (pack.getData(),
pack.getOffset(),pack.getLength()))) {
print ("Got: " + new String (pack.getData()));
print ("Expected: " + new String (buf));
throw new Exception ("Error comparing data: Iter " +i);
}
}
}
};
public static void main (String args[]) throws Exception {
if (args.length >=1 && args[0].equals ("-d")) {
debug = true;
}
SendDatagramToBadAddress ud = new SendDatagramToBadAddress ();
ud.run ();
}
public void run() throws Exception {
if (OSsupportsFeature()) {
print ("running on OS that supports ICMP port unreachable");
}
String host = "127.0.0.1";
InetAddress addr = InetAddress.getByName(host);
DatagramSocket sock = new DatagramSocket();
DatagramSocket serversock = new DatagramSocket(0);
DatagramPacket p;
byte[] buf;
int port = serversock.getLocalPort ();
final int loop = 5;
Server s = new Server (serversock);
int i;
print ("Checking send to connected address ...");
sock.connect(addr, port);
for (i = 0; i < loop; i++) {
try {
buf = ("Hello, server"+i).getBytes();
if (i % 2 == 1)
p = new DatagramPacket(buf, buf.length, addr, port);
else
p = new DatagramPacket(buf, buf.length);
sock.send(p);
} catch (Exception ex) {
print ("Got unexpected exception: " + ex);
throw new Exception ("Error sending data: ");
}
}
s.receive (loop, false);
// check disconnect() works
print ("Checking send to non-connected address ...");
sock.disconnect ();
buf = ("Hello, server"+0).getBytes();
p = new DatagramPacket(buf, buf.length, addr, port);
sock.send (p);
s.receive (1, false);
// check send() to invalid destination followed by a blocking receive
// returns an error
print ("Checking send to invalid address ...");
sock.connect(addr, port);
serversock.close ();
try {
sock.setSoTimeout (4000);
} catch (Exception e) {
print ("could not set timeout");
throw e;
}
boolean goterror = false;
for (i = 0; i < loop; i++) {
try {
buf = ("Hello, server"+i).getBytes();
p = new DatagramPacket(buf, buf.length, addr, port);
sock.send(p);
p = new DatagramPacket(buf, buf.length, addr, port);
sock.receive (p);
} catch (InterruptedIOException ex) {
print ("socket timeout");
} catch (Exception ex) {
print ("Got expected exception: " + ex);
goterror = true;
}
}
if (!goterror && OSsupportsFeature ()) {
print ("Didnt get expected exception: ");
throw new Exception ("send did not return expected error");
}
}
}