forked from fanruan/demo-function-defender
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
39 lines
1.2 KiB
39 lines
1.2 KiB
6 years ago
|
package com.fr.security.function;
|
||
|
|
|
||
|
|
import com.fr.intelli.record.Focus;
|
||
|
|
import com.fr.record.analyzer.EnableMetrics;
|
||
|
|
import com.fr.security.function.conf.RemoteEvalConfig;
|
||
|
|
import com.fr.security.function.holder.HolderFunction;
|
||
|
|
import com.fr.stable.fun.impl.AbstractFunctionDefendProvider;
|
||
|
|
import com.fr.stable.script.CalculatorProvider;
|
||
|
|
import com.fr.stable.script.Function;
|
||
|
|
|
||
|
|
import java.lang.annotation.Annotation;
|
||
|
|
|
||
|
|
/**
|
||
|
|
* @author richie
|
||
|
|
* @version 10.0
|
||
|
|
* Created by richie on 2019-01-12
|
||
|
|
*/
|
||
|
|
@EnableMetrics
|
||
|
|
public class RemoteEvalFunctionDefender extends AbstractFunctionDefendProvider {
|
||
|
|
|
||
|
|
@Override
|
||
|
|
@Focus(id = "com.fr.security.function", text = "defender")
|
||
|
|
public Function replacer(CalculatorProvider calculator, Class clazz) {
|
||
|
|
return HolderFunction.ONE;
|
||
|
|
}
|
||
|
|
|
||
|
|
@Override
|
||
|
|
public boolean match(CalculatorProvider calculator, Class clazz) {
|
||
|
|
if (RemoteEvalConfig.getInstance().isEnable()) {
|
||
|
|
Boolean mark = calculator.getAttribute(RestrictScriptKey.KEY);
|
||
|
|
if (mark != null && mark) {
|
||
|
|
Annotation restrict = clazz.getAnnotation(RestrictScript.class);
|
||
|
|
return restrict != null;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
}
|