定时任务目标字符串验证包名白名单

RuoYi

2022-01-06 0dc26533584ea2269917d00aebc5b12a1017d121

 src/main/java/com/ruoyi/common/utils/job/ScheduleUtils.java

@@ -10,9 +10,11 @@
import org.quartz.SchedulerException;

import org.quartz.TriggerBuilder;

import org.quartz.TriggerKey;

import com.ruoyi.common.constant.Constants;

import com.ruoyi.common.constant.ScheduleConstants;

import com.ruoyi.common.exception.job.TaskException;

import com.ruoyi.common.exception.job.TaskException.Code;

import com.ruoyi.common.utils.StringUtils;

import com.ruoyi.project.monitor.domain.SysJob;



/**

@@ -110,4 +112,24 @@
                        + "' cannot be used in cron schedule tasks", Code.CONFIG_ERROR);

        }

    }

}


    /**

     * 检查包名是否为白名单配置

     * 
     * @param invokeTarget 目标字符串

     * @return 结果

     */

    public static boolean whiteList(String invokeTarget)

    {

        String packageName = StringUtils.substringBefore(invokeTarget, ")");

        int count = StringUtils.countMatches(packageName, ".");

        if (count > 1)

        {

            if (!StringUtils.containsAnyIgnoreCase(invokeTarget, Constants.JOB_WHITELIST_STR))

            {

                return false;

            }

        }

        return true;

    }

}