X Tutup
/* * Copyright (c) 2013, 2022, 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. */ package jdk.test.lib; import java.io.BufferedReader; import java.io.File; import java.io.FileReader; import java.io.FileWriter; import java.io.FilenameFilter; import java.io.IOException; import java.net.Inet6Address; import java.net.InetAddress; import java.net.InetSocketAddress; import java.net.MalformedURLException; import java.net.ServerSocket; import java.net.URL; import java.net.URLClassLoader; import java.net.UnknownHostException; import java.nio.ByteBuffer; import java.nio.charset.StandardCharsets; import java.nio.file.CopyOption; import java.nio.file.Files; import java.nio.file.Path; import java.nio.file.Paths; import java.nio.file.attribute.AclEntry; import java.nio.file.attribute.AclEntryType; import java.nio.file.attribute.AclFileAttributeView; import java.nio.file.attribute.FileAttribute; import java.nio.channels.SocketChannel; import java.nio.file.attribute.PosixFilePermissions; import java.security.MessageDigest; import java.security.NoSuchAlgorithmException; import java.util.ArrayList; import java.util.Arrays; import java.util.Collection; import java.util.Collections; import java.util.HexFormat; import java.util.Iterator; import java.util.Map; import java.util.HashMap; import java.util.LinkedList; import java.util.List; import java.util.Random; import java.util.Set; import java.util.function.BooleanSupplier; import java.util.concurrent.TimeUnit; import java.util.function.Consumer; import java.util.function.Function; import java.util.regex.Matcher; import java.util.regex.Pattern; import java.util.stream.Collectors; import static jdk.test.lib.Asserts.assertTrue; import jdk.test.lib.process.ProcessTools; import jdk.test.lib.process.OutputAnalyzer; /** * Common library for various test helper functions. */ public final class Utils { /** * Returns the value of 'test.class.path' system property. */ public static final String TEST_CLASS_PATH = System.getProperty("test.class.path", "."); /** * Returns the sequence used by operating system to separate lines. */ public static final String NEW_LINE = System.getProperty("line.separator"); /** * Returns the value of 'test.vm.opts' system property. */ public static final String VM_OPTIONS = System.getProperty("test.vm.opts", "").trim(); /** * Returns the value of 'test.java.opts' system property. */ public static final String JAVA_OPTIONS = System.getProperty("test.java.opts", "").trim(); /** * Returns the value of 'test.src' system property. */ public static final String TEST_SRC = System.getProperty("test.src", "").trim(); /** * Returns the value of 'test.root' system property. */ public static final String TEST_ROOT = System.getProperty("test.root", "").trim(); /* * Returns the value of 'test.jdk' system property */ public static final String TEST_JDK = System.getProperty("test.jdk"); /* * Returns the value of 'compile.jdk' system property */ public static final String COMPILE_JDK = System.getProperty("compile.jdk", TEST_JDK); /** * Returns the value of 'test.classes' system property */ public static final String TEST_CLASSES = System.getProperty("test.classes", "."); /** * Returns the value of 'test.name' system property */ public static final String TEST_NAME = System.getProperty("test.name", "."); /** * Returns the value of 'test.nativepath' system property */ public static final String TEST_NATIVE_PATH = System.getProperty("test.nativepath", "."); /** * Defines property name for seed value. */ public static final String SEED_PROPERTY_NAME = "jdk.test.lib.random.seed"; /** * Returns the value of 'file.separator' system property */ public static final String FILE_SEPARATOR = System.getProperty("file.separator"); /** * Random generator with predefined seed. */ private static volatile Random RANDOM_GENERATOR; /** * Maximum number of attempts to get free socket */ private static final int MAX_SOCKET_TRIES = 10; /** * Contains the seed value used for {@link java.util.Random} creation. */ public static final long SEED; static { var seed = Long.getLong(SEED_PROPERTY_NAME); if (seed != null) { // use explicitly set seed SEED = seed; } else { var v = Runtime.version(); // promotable builds have build number, and it's greater than 0 if (v.build().orElse(0) > 0) { // promotable build -> use 1st 8 bytes of md5($version) try { var md = MessageDigest.getInstance("MD5"); var bytes = v.toString() .getBytes(StandardCharsets.UTF_8); bytes = md.digest(bytes); SEED = ByteBuffer.wrap(bytes).getLong(); } catch (NoSuchAlgorithmException e) { throw new Error(e); } } else { // "personal" build -> use random seed SEED = new Random().nextLong(); } } } /** * Returns the value of 'test.timeout.factor' system property * converted to {@code double}. */ public static final double TIMEOUT_FACTOR; static { String toFactor = System.getProperty("test.timeout.factor", "1.0"); TIMEOUT_FACTOR = Double.parseDouble(toFactor); } /** * Returns the value of JTREG default test timeout in milliseconds * converted to {@code long}. */ public static final long DEFAULT_TEST_TIMEOUT = TimeUnit.SECONDS.toMillis(120); private Utils() { // Private constructor to prevent class instantiation } /** * Returns the list of VM options with -J prefix. * * @return The list of VM options with -J prefix */ public static List getForwardVmOptions() { String[] opts = safeSplitString(VM_OPTIONS); for (int i = 0; i < opts.length; i++) { opts[i] = "-J" + opts[i]; } return Arrays.asList(opts); } /** * Returns the default JTReg arguments for a jvm running a test. * This is the combination of JTReg arguments test.vm.opts and test.java.opts. * @return An array of options, or an empty array if no options. */ public static String[] getTestJavaOpts() { List opts = new ArrayList(); Collections.addAll(opts, safeSplitString(VM_OPTIONS)); Collections.addAll(opts, safeSplitString(JAVA_OPTIONS)); return opts.toArray(new String[0]); } /** * Combines given arguments with default JTReg arguments for a jvm running a test. * This is the combination of JTReg arguments test.vm.opts and test.java.opts * @return The combination of JTReg test java options and user args. */ public static String[] prependTestJavaOpts(String... userArgs) { List opts = new ArrayList(); Collections.addAll(opts, getTestJavaOpts()); Collections.addAll(opts, userArgs); return opts.toArray(new String[0]); } /** * Combines given arguments with default JTReg arguments for a jvm running a test. * This is the combination of JTReg arguments test.vm.opts and test.java.opts * @return The combination of JTReg test java options and user args. */ public static String[] appendTestJavaOpts(String... userArgs) { List opts = new ArrayList(); Collections.addAll(opts, userArgs); Collections.addAll(opts, getTestJavaOpts()); return opts.toArray(new String[0]); } /** * Combines given arguments with default JTReg arguments for a jvm running a test. * This is the combination of JTReg arguments test.vm.opts and test.java.opts * @return The combination of JTReg test java options and user args. */ public static String[] addTestJavaOpts(String... userArgs) { return prependTestJavaOpts(userArgs); } private static final Pattern useGcPattern = Pattern.compile( "(?:\\-XX\\:[\\+\\-]Use.+GC)"); /** * Removes any options specifying which GC to use, for example "-XX:+UseG1GC". * Removes any options matching: -XX:(+/-)Use*GC * Used when a test need to set its own GC version. Then any * GC specified by the framework must first be removed. * @return A copy of given opts with all GC options removed. */ public static List removeGcOpts(List opts) { List optsWithoutGC = new ArrayList(); for (String opt : opts) { if (useGcPattern.matcher(opt).matches()) { System.out.println("removeGcOpts: removed " + opt); } else { optsWithoutGC.add(opt); } } return optsWithoutGC; } /** * Returns the default JTReg arguments for a jvm running a test without * options that matches regular expressions in {@code filters}. * This is the combination of JTReg arguments test.vm.opts and test.java.opts. * @param filters Regular expressions used to filter out options. * @return An array of options, or an empty array if no options. */ public static String[] getFilteredTestJavaOpts(String... filters) { String options[] = getTestJavaOpts(); if (filters.length == 0) { return options; } List filteredOptions = new ArrayList(options.length); Pattern patterns[] = new Pattern[filters.length]; for (int i = 0; i < filters.length; i++) { patterns[i] = Pattern.compile(filters[i]); } for (String option : options) { boolean matched = false; for (int i = 0; i < patterns.length && !matched; i++) { Matcher matcher = patterns[i].matcher(option); matched = matcher.find(); } if (!matched) { filteredOptions.add(option); } } return filteredOptions.toArray(new String[filteredOptions.size()]); } /** * Splits a string by white space. * Works like String.split(), but returns an empty array * if the string is null or empty. */ private static String[] safeSplitString(String s) { if (s == null || s.trim().isEmpty()) { return new String[] {}; } return s.trim().split("\\s+"); } /** * @return The full command line for the ProcessBuilder. */ public static String getCommandLine(ProcessBuilder pb) { StringBuilder cmd = new StringBuilder(); for (String s : pb.command()) { cmd.append(s).append(" "); } return cmd.toString(); } /** * Returns the socket address of an endpoint that refuses connections. The * endpoint is an InetSocketAddress where the address is the loopback address * and the port is a system port (1-1023 range). * This method is a better choice than getFreePort for tests that need * an endpoint that refuses connections. */ public static InetSocketAddress refusingEndpoint() { InetAddress lb = InetAddress.getLoopbackAddress(); int port = 1; while (port < 1024) { InetSocketAddress sa = new InetSocketAddress(lb, port); try { SocketChannel.open(sa).close(); } catch (IOException ioe) { return sa; } port++; } throw new RuntimeException("Unable to find system port that is refusing connections"); } /** * Returns local addresses with symbolic and numeric scopes */ public static List getAddressesWithSymbolicAndNumericScopes() { List result = new LinkedList<>(); try { NetworkConfiguration conf = NetworkConfiguration.probe(); conf.ip4Addresses().forEach(result::add); // Java reports link local addresses with symbolic scope, // but on Windows java.net.NetworkInterface generates its own scope names // which are incompatible with native Windows routines. // So on Windows test only addresses with numeric scope. // On other platforms test both symbolic and numeric scopes. conf.ip6Addresses() // test only IPv6 loopback and link-local addresses (JDK-8224775) .filter(addr -> addr.isLinkLocalAddress() || addr.isLoopbackAddress()) .forEach(addr6 -> { try { result.add(Inet6Address.getByAddress(null, addr6.getAddress(), addr6.getScopeId())); } catch (UnknownHostException e) { // cannot happen! throw new RuntimeException("Unexpected", e); } if (!Platform.isWindows()) { result.add(addr6); } }); } catch (IOException e) { // cannot happen! throw new RuntimeException("Unexpected", e); } return result; } /** * Returns the free port on the loopback address. * * @return The port number * @throws IOException if an I/O error occurs when opening the socket */ public static int getFreePort() throws IOException { try (ServerSocket serverSocket = new ServerSocket(0, 5, InetAddress.getLoopbackAddress());) { return serverSocket.getLocalPort(); } } /** * Returns the free unreserved port on the local host. * * @param reservedPorts reserved ports * @return The port number or -1 if failed to find a free port */ public static int findUnreservedFreePort(int... reservedPorts) { int numTries = 0; while (numTries++ < MAX_SOCKET_TRIES) { int port = -1; try { port = getFreePort(); } catch (IOException e) { e.printStackTrace(); } if (port > 0 && !isReserved(port, reservedPorts)) { return port; } } return -1; } private static boolean isReserved(int port, int[] reservedPorts) { for (int p : reservedPorts) { if (p == port) { return true; } } return false; } /** * Returns the name of the local host. * * @return The host name * @throws UnknownHostException if IP address of a host could not be determined */ public static String getHostname() throws UnknownHostException { InetAddress inetAddress = InetAddress.getLocalHost(); String hostName = inetAddress.getHostName(); assertTrue((hostName != null && !hostName.isEmpty()), "Cannot get hostname"); return hostName; } /** * Adjusts the provided timeout value for the TIMEOUT_FACTOR * @param tOut the timeout value to be adjusted * @return The timeout value adjusted for the value of "test.timeout.factor" * system property */ public static long adjustTimeout(long tOut) { return Math.round(tOut * Utils.TIMEOUT_FACTOR); } /** * Return the contents of the named file as a single String, * or null if not found. * @param filename name of the file to read * @return String contents of file, or null if file not found. * @throws IOException * if an I/O error occurs reading from the file or a malformed or * unmappable byte sequence is read */ public static String fileAsString(String filename) throws IOException { Path filePath = Paths.get(filename); if (!Files.exists(filePath)) return null; return new String(Files.readAllBytes(filePath)); } /** * Returns hex view of byte array * * @param bytes byte array to process * @return space separated hexadecimal string representation of bytes * @deprecated replaced by {@link java.util.HexFormat#ofDelimiter(String) * HexFormat.ofDelimiter(" ").format (byte[], char)}. */ @Deprecated public static String toHexString(byte[] bytes) { return HexFormat.ofDelimiter(" ").withUpperCase().formatHex(bytes); } /** * Returns byte array of hex view * * @param hex hexadecimal string representation * @return byte array */ public static byte[] toByteArray(String hex) { return HexFormat.of().parseHex(hex); } /** * Returns {@link java.util.Random} generator initialized with particular seed. * The seed could be provided via system property {@link Utils#SEED_PROPERTY_NAME}. * In case no seed is provided and the build under test is "promotable" * (its build number ({@code $BUILD} in {@link Runtime.Version}) is greater than 0, * the seed based on string representation of {@link Runtime#version()} is used. * Otherwise, the seed is randomly generated. * The used seed printed to stdout. * The printing is not in the synchronized block so as to prevent carrier threads starvation. * @return {@link java.util.Random} generator with particular seed. */ public static Random getRandomInstance() { if (RANDOM_GENERATOR == null) { synchronized (Utils.class) { if (RANDOM_GENERATOR == null) { RANDOM_GENERATOR = new Random(SEED); } else { return RANDOM_GENERATOR; } } System.out.printf("For random generator using seed: %d%n", SEED); System.out.printf("To re-run test with same seed value please add \"-D%s=%d\" to command line.%n", SEED_PROPERTY_NAME, SEED); } return RANDOM_GENERATOR; } /** * Returns random element of non empty collection * * @param a type of collection element * @param collection collection of elements * @return random element of collection * @throws IllegalArgumentException if collection is empty */ public static T getRandomElement(Collection collection) throws IllegalArgumentException { if (collection.isEmpty()) { throw new IllegalArgumentException("Empty collection"); } Random random = getRandomInstance(); int elementIndex = 1 + random.nextInt(collection.size() - 1); Iterator iterator = collection.iterator(); while (--elementIndex != 0) { iterator.next(); } return iterator.next(); } /** * Returns random element of non empty array * * @param a type of array element * @param array array of elements * @return random element of array * @throws IllegalArgumentException if array is empty */ public static T getRandomElement(T[] array) throws IllegalArgumentException { if (array == null || array.length == 0) { throw new IllegalArgumentException("Empty or null array"); } Random random = getRandomInstance(); return array[random.nextInt(array.length)]; } /** * Wait for condition to be true * * @param condition a condition to wait for */ public static final void waitForCondition(BooleanSupplier condition) { waitForCondition(condition, -1L, 100L); } /** * Wait until timeout for condition to be true * * @param condition a condition to wait for * @param timeout a time in milliseconds to wait for condition to be true * specifying -1 will wait forever * @return condition value, to determine if wait was successful */ public static final boolean waitForCondition(BooleanSupplier condition, long timeout) { return waitForCondition(condition, timeout, 100L); } /** * Wait until timeout for condition to be true for specified time * * @param condition a condition to wait for * @param timeout a time in milliseconds to wait for condition to be true, * specifying -1 will wait forever * @param sleepTime a time to sleep value in milliseconds * @return condition value, to determine if wait was successful */ public static final boolean waitForCondition(BooleanSupplier condition, long timeout, long sleepTime) { long startTime = System.currentTimeMillis(); while (!(condition.getAsBoolean() || (timeout != -1L && ((System.currentTimeMillis() - startTime) > timeout)))) { try { Thread.sleep(sleepTime); } catch (InterruptedException e) { Thread.currentThread().interrupt(); throw new Error(e); } } return condition.getAsBoolean(); } /** * Interface same as java.lang.Runnable but with * method {@code run()} able to throw any Throwable. */ public static interface ThrowingRunnable { void run() throws Throwable; } /** * Filters out an exception that may be thrown by the given * test according to the given filter. * * @param test method that is invoked and checked for exception. * @param filter function that checks if the thrown exception matches * criteria given in the filter's implementation. * @return exception that matches the filter if it has been thrown or * {@code null} otherwise. * @throws Throwable if test has thrown an exception that does not * match the filter. */ public static Throwable filterException(ThrowingRunnable test, Function filter) throws Throwable { try { test.run(); } catch (Throwable t) { if (filter.apply(t)) { return t; } else { throw t; } } return null; } /** * Ensures a requested class is loaded * @param aClass class to load */ public static void ensureClassIsLoaded(Class aClass) { if (aClass == null) { throw new Error("Requested null class"); } try { Class.forName(aClass.getName(), /* initialize = */ true, ClassLoader.getSystemClassLoader()); } catch (ClassNotFoundException e) { throw new Error("Class not found", e); } } /** * @param parent a class loader to be the parent for the returned one * @return an UrlClassLoader with urls made of the 'test.class.path' jtreg * property and with the given parent */ public static URLClassLoader getTestClassPathURLClassLoader(ClassLoader parent) { URL[] urls = Arrays.stream(TEST_CLASS_PATH.split(File.pathSeparator)) .map(Paths::get) .map(Path::toUri) .map(x -> { try { return x.toURL(); } catch (MalformedURLException ex) { throw new Error("Test issue. JTREG property" + " 'test.class.path'" + " is not defined correctly", ex); } }).toArray(URL[]::new); return new URLClassLoader(urls, parent); } /** * Runs runnable and checks that it throws expected exception. If exceptionException is null it means * that we expect no exception to be thrown. * @param runnable what we run * @param expectedException expected exception */ public static void runAndCheckException(ThrowingRunnable runnable, Class expectedException) { runAndCheckException(runnable, t -> { if (t == null) { if (expectedException != null) { throw new AssertionError("Didn't get expected exception " + expectedException.getSimpleName()); } } else { String message = "Got unexpected exception " + t.getClass().getSimpleName(); if (expectedException == null) { throw new AssertionError(message, t); } else if (!expectedException.isAssignableFrom(t.getClass())) { message += " instead of " + expectedException.getSimpleName(); throw new AssertionError(message, t); } } }); } /** * Runs runnable and makes some checks to ensure that it throws expected exception. * @param runnable what we run * @param checkException a consumer which checks that we got expected exception and raises a new exception otherwise */ public static void runAndCheckException(ThrowingRunnable runnable, Consumer checkException) { Throwable throwable = null; try { runnable.run(); } catch (Throwable t) { throwable = t; } checkException.accept(throwable); } /** * Converts to VM type signature * * @param type Java type to convert * @return string representation of VM type */ public static String toJVMTypeSignature(Class type) { if (type.isPrimitive()) { if (type == boolean.class) { return "Z"; } else if (type == byte.class) { return "B"; } else if (type == char.class) { return "C"; } else if (type == double.class) { return "D"; } else if (type == float.class) { return "F"; } else if (type == int.class) { return "I"; } else if (type == long.class) { return "J"; } else if (type == short.class) { return "S"; } else if (type == void.class) { return "V"; } else { throw new Error("Unsupported type: " + type); } } String result = type.getName().replaceAll("\\.", "/"); if (!type.isArray()) { return "L" + result + ";"; } return result; } public static Object[] getNullValues(Class... types) { Object[] result = new Object[types.length]; int i = 0; for (Class type : types) { result[i++] = NULL_VALUES.get(type); } return result; } private static Map, Object> NULL_VALUES = new HashMap<>(); static { NULL_VALUES.put(boolean.class, false); NULL_VALUES.put(byte.class, (byte) 0); NULL_VALUES.put(short.class, (short) 0); NULL_VALUES.put(char.class, '\0'); NULL_VALUES.put(int.class, 0); NULL_VALUES.put(long.class, 0L); NULL_VALUES.put(float.class, 0.0f); NULL_VALUES.put(double.class, 0.0d); } /* * Run uname with specified arguments. */ public static OutputAnalyzer uname(String... args) throws Throwable { String[] cmds = new String[args.length + 1]; cmds[0] = "uname"; System.arraycopy(args, 0, cmds, 1, args.length); return ProcessTools.executeCommand(cmds); } /** * Creates an empty file in "user.dir" if the property set. *

* This method is meant as a replacement for {@link Files#createTempFile(String, String, FileAttribute...)} * that doesn't leave files behind in /tmp directory of the test machine *

* If the property "user.dir" is not set, "." will be used. * * @param prefix the prefix string to be used in generating the file's name; * may be null * @param suffix the suffix string to be used in generating the file's name; * may be null, in which case ".tmp" is used * @param attrs an optional list of file attributes to set atomically when creating the file * @return the path to the newly created file that did not exist before this * method was invoked * @throws IOException if an I/O error occurs or dir does not exist * * @see Files#createTempFile(String, String, FileAttribute...) */ public static Path createTempFile(String prefix, String suffix, FileAttribute... attrs) throws IOException { Path dir = Paths.get(System.getProperty("user.dir", ".")); return Files.createTempFile(dir, prefix, suffix, attrs); } /** * Creates an empty directory in "user.dir" or "." *

* This method is meant as a replacement for {@link Files#createTempDirectory(Path, String, FileAttribute...)} * that doesn't leave files behind in /tmp directory of the test machine *

* If the property "user.dir" is not set, "." will be used. * * @param prefix the prefix string to be used in generating the directory's name; may be null * @param attrs an optional list of file attributes to set atomically when creating the directory * @return the path to the newly created directory * @throws IOException if an I/O error occurs or dir does not exist * * @see Files#createTempDirectory(Path, String, FileAttribute...) */ public static Path createTempDirectory(String prefix, FileAttribute... attrs) throws IOException { Path dir = Paths.get(System.getProperty("user.dir", ".")); return Files.createTempDirectory(dir, prefix, attrs); } /** * Converts slashes in a pathname to backslashes * if slashes is not the file separator. */ public static String convertPath(String path) { // No need to do the conversion if file separator is '/' if (FILE_SEPARATOR.length() == 1 && FILE_SEPARATOR.charAt(0) == '/') { return path; } char[] cs = path.toCharArray(); for (int i = 0; i < cs.length; i++) { if (cs[i] == '/') { cs[i] = '\\'; } } String newPath = new String(cs); return newPath; } /** * Return file directories that satisfy the specified filter. * * @param searchDirectory the base directory to search * @param filter a filename filter * @return file directories */ public static List findFiles(Path searchDirectory, FilenameFilter filter) { return Arrays.stream(searchDirectory.toFile().listFiles(filter)) .map(f -> f.toPath()) .collect(Collectors.toList()); } /** * Copy files to the target path. * * @param source the paths to the files to copy * @param target the path to the target files * @param filenameMapper mapper function applied to filenames * @param options options specifying how the copy should be done * @return the paths to the target files * @throws IOException if error occurs */ public static List copyFiles(List source, Path target, Function filenameMapper, CopyOption... options) throws IOException { List result = new ArrayList<>(); if (!target.toFile().exists()) { Files.createDirectory(target); } for (Path file : source) { if (!file.toFile().exists()) { continue; } String baseName = file.getFileName().toString(); Path targetFile = target.resolve(filenameMapper.apply(baseName)); Files.copy(file, targetFile, options); result.add(targetFile); } return result; } /** * Copy files to the target path. * * @param source the paths to the files to copy * @param target the path to the target files * @param options options specifying how the copy should be done * @return the paths to the target files * @throws IOException if error occurs */ public static List copyFiles(List source, Path target, CopyOption... options) throws IOException { return copyFiles(source, target, (s) -> s, options); } /** * Replace file string by applying the given mapper function. * * @param source the file to read * @param contentMapper the mapper function applied to file's content * @throws IOException if an I/O error occurs */ public static void replaceFileString(Path source, Function contentMapper) throws IOException { StringBuilder sb = new StringBuilder(); String lineSep = System.getProperty("line.separator"); try (BufferedReader reader = new BufferedReader(new FileReader(source.toFile()))) { String line; // read all and replace all at once?? while ((line = reader.readLine()) != null) { sb.append(contentMapper.apply(line)) .append(lineSep); } } try (FileWriter writer = new FileWriter(source.toFile())) { writer.write(sb.toString()); } } /** * Replace files' string by applying the given mapper function. * * @param source the file to read * @param contentMapper the mapper function applied to files' content * @throws IOException if an I/O error occurs */ public static void replaceFilesString(List source, Function contentMapper) throws IOException { for (Path file : source) { replaceFileString(file, contentMapper); } } /** * Grant file user access or full access to everyone. * * @param file file to grant access * @param userOnly true for user access, otherwise full access to everyone * @throws IOException if error occurs */ public static void grantFileAccess(Path file, boolean userOnly) throws IOException { Set attr = file.getFileSystem().supportedFileAttributeViews(); if (attr.contains("posix")) { String perms = userOnly ? "rwx------" : "rwxrwxrwx"; Files.setPosixFilePermissions(file, PosixFilePermissions.fromString(perms)); } else if (attr.contains("acl")) { AclFileAttributeView view = Files.getFileAttributeView(file, AclFileAttributeView.class); List acl = new ArrayList<>(); for (AclEntry thisEntry : view.getAcl()) { if (userOnly) { if (thisEntry.principal().getName() .equals(view.getOwner().getName())) { acl.add(allowAccess(thisEntry)); } else if (thisEntry.type() == AclEntryType.ALLOW) { acl.add(revokeAccess(thisEntry)); } else { acl.add(thisEntry); } } else { if (thisEntry.type() != AclEntryType.ALLOW) { acl.add(allowAccess(thisEntry)); } else { acl.add(thisEntry); } } } view.setAcl(acl); } else { throw new RuntimeException("Unsupported file attributes: " + attr); } } /** * Return an ACL entry that revokes owner access. * * @param acl original ACL entry to build from * @return an ACL entry that revokes all access */ public static AclEntry revokeAccess(AclEntry acl) { return buildAclEntry(acl, AclEntryType.DENY); } /** * Return an ACL entry that allow owner access. * @param acl original ACL entry to build from * @return an ACL entry that allows all access */ public static AclEntry allowAccess(AclEntry acl) { return buildAclEntry(acl, AclEntryType.ALLOW); } /** * Build an ACL entry with a given ACL entry type. * * @param acl original ACL entry to build from * @return an ACL entry with a given ACL entry type */ public static AclEntry buildAclEntry(AclEntry acl, AclEntryType type) { return AclEntry.newBuilder(acl) .setType(type) .build(); } /** * Grant file user access. * * @param file file to grant access * @throws IOException if error occurs */ public static void userAccess(Path file) throws IOException { grantFileAccess(file, true); } /** * Grant file full access to everyone. * * @param file file to grant access * @throws IOException if error occurs */ public static void fullAccess(Path file) throws IOException { grantFileAccess(file, false); } }

X Tutup