jdk-24/test/jdk/sun/net/InetAddress/nameservice/simple/DefaultCaching.java
Conor Cleary 8bc8542e3f 8262195: Harden tests that use the HostsFileNameService (jdk.net.hosts.file property)
Reviewed-by: michaelm, aefimov, dfuchs, chegar
2021-03-01 10:18:12 +00:00

148 lines
5.4 KiB
Java

/*
* Copyright (c) 2006, 2021, 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 6442088
* @summary Change default DNS caching behavior for code not running under
* security manager.
* @run main/othervm/timeout=200 -Djdk.net.hosts.file=DefaultCachingHosts
* -Dsun.net.inetaddr.ttl=20 DefaultCaching
*/
import java.net.InetAddress;
import java.net.UnknownHostException;
import java.io.FileWriter;
import java.io.PrintWriter;
import java.io.BufferedWriter;
public class DefaultCaching {
public static void main(String args[]) throws Exception {
String hostsFileName = System.getProperty("jdk.net.hosts.file");
addMappingToHostsFile("theclub", "129.156.220.219", hostsFileName, false);
test("theclub", "129.156.220.219", true); // lk: 1
test("luster", "1.16.20.2", false); // lk: 2
// name service now needs to know about luster
addMappingToHostsFile("luster", "10.5.18.21", hostsFileName, true);
test("luster", "1.16.20.2", false); // lk: 2
sleep(10+1);
test("luster", "10.5.18.21", true, 3); // lk: 3
sleep(5);
// new mapping for theclub and rewrite existing foo and luster mappings
addMappingToHostsFile("theclub", "129.156.220.1", hostsFileName, false);
addMappingToHostsFile("foo", "10.5.18.22", hostsFileName, true);
addMappingToHostsFile("luster", "10.5.18.21", hostsFileName, true);
test("theclub", "129.156.220.219", true, 3);
test("luster", "10.5.18.21", true, 3);
test("bar", "10.5.18.22", false, 4);
test("foo", "10.5.18.22", true, 5);
// now delay to see if theclub has expired
sleep(5);
test("foo", "10.5.18.22", true, 5);
test("theclub", "129.156.220.1", true, 6);
sleep(11);
// now see if luster has expired
test("luster", "10.5.18.21", true, 7);
test("theclub", "129.156.220.1", true, 7);
// now delay to see if 3rd has expired
sleep(10+6);
test("theclub", "129.156.220.1", true, 8);
test("luster", "10.5.18.21", true, 8);
test("foo", "10.5.18.22", true, 9);
}
/* throws RuntimeException if it fails */
static void test(String host, String address,
boolean shouldSucceed, int count) {
test(host, address, shouldSucceed);
}
static void sleep(int seconds) {
try {
sleepms(seconds * 1000);
} catch (InterruptedException e) {
e.printStackTrace();
}
}
static long sleepms(long millis) throws InterruptedException {
long start = System.nanoTime();
long ms = millis;
while (ms > 0) {
assert ms < Long.MAX_VALUE/1000_000L;
Thread.sleep(ms);
long elapsedms = (System.nanoTime() - start)/1000_000L;
ms = millis - elapsedms;
}
return millis - ms;
}
static void test(String host, String address, boolean shouldSucceed) {
InetAddress addr = null;
try {
addr = InetAddress.getByName(host);
if (!shouldSucceed) {
throw new RuntimeException(host+":"+address+": should fail (got "
+ addr + ")");
}
if (!address.equals(addr.getHostAddress())) {
throw new RuntimeException(host+"/"+address+": compare failed (found "
+ addr + ")");
}
System.out.println("test: " + host + "/" + address
+ " succeeded - got " + addr);
} catch (UnknownHostException e) {
if (shouldSucceed) {
throw new RuntimeException(host+":"+address+": should succeed");
} else {
System.out.println("test: " + host + "/" + address
+ " succeeded - got expected " + e);
}
}
}
private static void addMappingToHostsFile(String host,
String addr,
String hostsFileName,
boolean append)
throws Exception {
String mapping = addr + " " + host;
try (PrintWriter hfPWriter = new PrintWriter(new BufferedWriter(
new FileWriter(hostsFileName, append)))) {
hfPWriter.println(mapping);
}
}
}