'commit'

QingLong/src/main/java/com/dsideal/QingLong/HuiYa/Controller/HuiYaController.java

@@ -294,6 +294,10 @@ public class HuiYaController extends Controller {
     @EmptyInterface({"token", "source_img_url"})
     @IsNumericInterface({"model_id", "count"})
     public void wxAddTask(String token, int model_id, String source_img_url, int prompt_id) {
+        if (!source_img_url.endsWith(".jpg") && !source_img_url.endsWith(".jpeg")) {
+            renderJson(Kv.by("success", false).set("message", "图片格式不正确"));
+            return;
+        }
         if (prompt_id == 0) prompt_id = 1;
         //验证token
         Kv res = ym.checkToken(token);