65 lines
2.0 KiB
Java
65 lines
2.0 KiB
Java
package com.novitechie.rules;
|
|
|
|
import com.janetfilter.core.commons.DebugInfo;
|
|
import com.janetfilter.core.models.FilterRule;
|
|
import com.novitechie.LogUtil;
|
|
|
|
import java.nio.file.Path;
|
|
import java.util.Collections;
|
|
import java.util.List;
|
|
|
|
/**
|
|
* @author YeloChick
|
|
*/
|
|
public class SystemRule {
|
|
|
|
private static final String JB_VM_OPTIONS_FILE = "jb.vmOptionsFile";
|
|
|
|
private static List<FilterRule> hideEnvRules = Collections.emptyList();
|
|
|
|
private static List<FilterRule> hidePropertyRules = Collections.emptyList();
|
|
|
|
public static void initRules(List<FilterRule> hideEnvRules, List<FilterRule> hidePropertyRules) {
|
|
SystemRule.hideEnvRules = hideEnvRules;
|
|
SystemRule.hidePropertyRules = hidePropertyRules;
|
|
}
|
|
|
|
public static boolean checkEnv(String name) {
|
|
if (checkHideEnv(name) && StackTraceRule.check()) {
|
|
DebugInfo.output("======================Hide Env: " + name);
|
|
LogUtil.printStackTrace();
|
|
return true;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
public static boolean checkProperty(String name) {
|
|
if ((checkVMOptionsProperty(name) || checkHideProperty(name)) && StackTraceRule.check()) {
|
|
DebugInfo.output("======================Hide Property: " + name);
|
|
LogUtil.printStackTrace();
|
|
return true;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
public static String hookProperty(String name) {
|
|
if (checkVMOptionsProperty(name)) {
|
|
Path path = VMOptionsRule.hook();
|
|
return path == null ? null : path.toString();
|
|
}
|
|
return null;
|
|
}
|
|
|
|
private static boolean checkHideEnv(String name) {
|
|
return hideEnvRules.stream().anyMatch(rule -> rule.test(name));
|
|
}
|
|
|
|
private static boolean checkHideProperty(String name) {
|
|
return hidePropertyRules.stream().anyMatch(rule -> rule.test(name));
|
|
}
|
|
|
|
private static boolean checkVMOptionsProperty(String name) {
|
|
return JB_VM_OPTIONS_FILE.equals(name);
|
|
}
|
|
}
|