From 0bb10b3c783c5f154dd19449a5c0984f0edafd20 Mon Sep 17 00:00:00 2001
From: weiweiw <14335254+weiweiw22@user.noreply.gitee.com>
Date: Wed, 22 May 2024 16:04:04 +0800
Subject: [PATCH] =?UTF-8?q?=E6=A0=B9=E6=8D=AE=E6=8A=80=E6=9C=AF=E8=AF=84?=
 =?UTF-8?q?=E5=AE=A1=E5=A7=94=E5=91=98=E4=BC=9A=E6=84=8F=E8=A7=81=E5=A2=9E?=
 =?UTF-8?q?=E5=8A=A0=E9=94=99=E8=AF=AF=E7=A0=81=E7=9A=84=E8=BF=94=E5=9B=9E?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 OCRTool/src/main/java/com/bonus/core/OCRRecognition.java | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/OCRTool/src/main/java/com/bonus/core/OCRRecognition.java b/OCRTool/src/main/java/com/bonus/core/OCRRecognition.java
index b127bc6..7ac22b7 100644
--- a/OCRTool/src/main/java/com/bonus/core/OCRRecognition.java
+++ b/OCRTool/src/main/java/com/bonus/core/OCRRecognition.java
@@ -32,8 +32,15 @@ public class OCRRecognition {
             logger.info(pythonEnv);
             logger.info(scriptEnv);
             logger.info("开始打印从请求里带过来的参数：");
+
             String typeStr = Integer.toString(recognitionParams.getType());
             logger.info(typeStr);
+
+            int type = recognitionParams.getType();
+            if (type < 0 || type >= 2) {
+                logger.info("参数检查错误");
+                return ErrorHandler.getErrorInfo(ErrorHandler.RECOGNITION_PARAM_ERROR);
+            }
             File file = new File(pythonEnv);
             if (!file.exists()) {
                 logger.info("pythonEnv不存在");