RuoYi пре 4 година
родитељ
комит
68db4092ed

+ 6 - 0
ruoyi-common/ruoyi-common-core/src/main/java/com/ruoyi/common/core/constant/Constants.java

@@ -112,4 +112,10 @@ public class Constants
112 112
      * 资源映射路径 前缀
113 113
      */
114 114
     public static final String RESOURCE_PREFIX = "/profile";
115
+
116
+    /**
117
+     * 定时任务违规的字符
118
+     */
119
+    public static final String[] JOB_ERROR_STR = { "java.net.URL", "javax.naming.InitialContext", "org.yaml.snakeyaml",
120
+            "org.springframework.jndi" };
115 121
 }

+ 8 - 0
ruoyi-modules/ruoyi-job/src/main/java/com/ruoyi/job/controller/SysJobController.java

@@ -99,6 +99,10 @@ public class SysJobController extends BaseController
99 99
         {
100 100
             return error("新增任务'" + job.getJobName() + "'失败,目标字符串不允许'http(s)//'调用");
101 101
         }
102
+        else if (StringUtils.containsAnyIgnoreCase(job.getInvokeTarget(), Constants.JOB_ERROR_STR))
103
+        {
104
+            return error("新增任务'" + job.getJobName() + "'失败,目标字符串存在违规");
105
+        }
102 106
         job.setCreateBy(SecurityUtils.getUsername());
103 107
         return toAjax(jobService.insertJob(job));
104 108
     }
@@ -127,6 +131,10 @@ public class SysJobController extends BaseController
127 131
         {
128 132
             return error("修改任务'" + job.getJobName() + "'失败,目标字符串不允许'http(s)//'调用");
129 133
         }
134
+        else if (StringUtils.containsAnyIgnoreCase(job.getInvokeTarget(), Constants.JOB_ERROR_STR))
135
+        {
136
+            return error("修改任务'" + job.getJobName() + "'失败,目标字符串存在违规");
137
+        }
130 138
         job.setUpdateBy(SecurityUtils.getUsername());
131 139
         return toAjax(jobService.updateJob(job));
132 140
     }