提交代码

src/main/java/cn/keking/anti_reptile/ValidateFormService.java

@@ -59,6 +59,37 @@ public class ValidateFormService {
         return "{\"result\":false}";
     }
 
+
+    public String small(HttpServletRequest request) throws UnsupportedEncodingException {
+        //DiskFileItemFactory factory = new DiskFileItemFactory();
+        //ServletFileUpload upload = new ServletFileUpload(factory);
+        //upload.setHeaderEncoding("UTF-8");
+       // List items = null;
+       /* try {
+            items = upload.parseRequest(request);
+        } catch (FileUploadException e) {
+            e.printStackTrace();
+        }*/
+        //Map<String, String> params = new HashMap<String, String>();
+        /*for(Object object : items){
+            FileItem fileItem = (FileItem) object;
+            if (fileItem.isFormField()) {
+                params.put(fileItem.getFieldName(), fileItem.getString("UTF-8"));
+            }
+        }*/
+        //request.getParameter("")
+        String verifyId =request.getParameter("verifyId");
+        String result = request.getParameter("result");
+        String realRequestUri = request.getParameter("realRequestUri");
+        String actualResult = verifyImageUtil.getVerifyCodeFromRedis(verifyId);
+        if (actualResult != null  && actualResult.equals(result.toLowerCase())) {
+            actuator.reset(request, realRequestUri);
+            return "{\"result\":true}";
+        }
+        return "{\"result\":false}";
+    }
+
+
     public String refresh(HttpServletRequest request) {
         String verifyId = request.getParameter("verifyId");
         verifyImageUtil.deleteVerifyCodeFromRedis(verifyId);

src/main/java/cn/keking/anti_reptile/config/AntiReptileAutoConfig.java

@@ -9,6 +9,7 @@ import cn.keking.anti_reptile.rule.RuleActuator;
 import cn.keking.anti_reptile.rule.UaRule;
 import cn.keking.anti_reptile.servlet.RefreshFormServlet;
 import cn.keking.anti_reptile.servlet.ValidateFormServlet;
+import cn.keking.anti_reptile.servlet.ValidateSmallServlet;
 import cn.keking.anti_reptile.util.VerifyImageUtil;
 import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty;
 import org.springframework.boot.context.properties.EnableConfigurationProperties;
@@ -36,7 +37,10 @@ public class AntiReptileAutoConfig {
     public ServletRegistrationBean validateFormServlet() {
         return new ServletRegistrationBean(new ValidateFormServlet(), AntiReptileConsts.VALIDATE_REQUEST_URI);
     }
-
+    @Bean
+    public ServletRegistrationBean validateSmallServlet() {
+        return new ServletRegistrationBean(new ValidateSmallServlet(), AntiReptileConsts.VALIDATE_SMALL_URI);
+    }
     @Bean
     public ServletRegistrationBean refreshFormServlet() {
         return new ServletRegistrationBean(new RefreshFormServlet(), AntiReptileConsts.REFRESH_REQUEST_URI);

src/main/java/cn/keking/anti_reptile/constant/AntiReptileConsts.java

@@ -6,5 +6,6 @@ package cn.keking.anti_reptile.constant;
  */
 public class AntiReptileConsts {
     public static final String VALIDATE_REQUEST_URI = "/kk-anti-reptile/validate/";
+    public static final String VALIDATE_SMALL_URI = "/kk-anti-reptile/small/";
     public static final String REFRESH_REQUEST_URI = "/kk-anti-reptile/refresh";
 }