From c29479a57db73c4dd379788d1b819e716ff7e1c1 Mon Sep 17 00:00:00 2001
From: futian.liu <liufutianyoo@163.com>
Date: 星期二, 28 十一月 2023 10:39:39 +0800
Subject: [PATCH] 参数赋值,漏洞修复

---
 consum-base/src/main/java/com/consum/base/controller/LWhFormInventoryController.java |   74 +++++++++++++++++++++++++++++--------
 1 files changed, 58 insertions(+), 16 deletions(-)

diff --git a/consum-base/src/main/java/com/consum/base/controller/LWhFormInventoryController.java b/consum-base/src/main/java/com/consum/base/controller/LWhFormInventoryController.java
index 6f4a5e7..aa8da03 100644
--- a/consum-base/src/main/java/com/consum/base/controller/LWhFormInventoryController.java
+++ b/consum-base/src/main/java/com/consum/base/controller/LWhFormInventoryController.java
@@ -1,6 +1,7 @@
 package com.consum.base.controller;
 
 import com.consum.base.BaseController;
+import com.consum.base.core.utils.CommonUtil;
 import com.consum.base.core.utils.PageUtil;
 import com.consum.base.pojo.query.LWhFormInventoryQry;
 import com.consum.base.pojo.request.FormInventoryParam;
@@ -9,10 +10,10 @@
 import com.consum.base.pojo.response.FormInventoryGoodsVO;
 import com.consum.base.pojo.response.FormInventoryVO;
 import com.consum.base.service.FinSysTenantUserServiceImpl;
-import com.consum.base.service.LWhFormInventoryGoodsServiceImpl;
 import com.consum.base.service.LWhFormInventoryServiceImpl;
 import com.consum.model.po.FinSysTenantUser;
 import com.consum.model.po.LWhFormInventory;
+import com.iplatform.model.po.S_user_core;
 import com.walker.db.page.GenericPager;
 import com.walker.web.ResponseValue;
 import io.swagger.annotations.Api;
@@ -27,7 +28,6 @@
 import org.springframework.web.bind.annotation.DeleteMapping;
 import org.springframework.web.bind.annotation.GetMapping;
 import org.springframework.web.bind.annotation.PostMapping;
-import org.springframework.web.bind.annotation.RequestBody;
 import org.springframework.web.bind.annotation.RequestMapping;
 import org.springframework.web.bind.annotation.RestController;
 
@@ -45,8 +45,7 @@
     private LWhFormInventoryServiceImpl lWhFormInventoryService;
     @Autowired
     private FinSysTenantUserServiceImpl finSysTenantUserService;
-    @Autowired
-    private LWhFormInventoryGoodsServiceImpl inventoryGoodsService;
+
 
     /**
      * @Description 鏂板
@@ -58,7 +57,13 @@
         @ApiImplicitParam(name = "param", value = "鐩樼偣鍗曚俊鎭�", dataType = "FormInventoryParam")
     })
     @PostMapping("/add")
-    public ResponseValue add(@RequestBody FormInventoryParam param) {
+    public ResponseValue add() {
+
+        FormInventoryParam param = CommonUtil.getObjFromReqBody(FormInventoryParam.class);
+        FormInventoryParam formInventoryParam = new FormInventoryParam();
+        CommonUtil.copyProperties(param, formInventoryParam);
+        param = formInventoryParam;
+
         FinSysTenantUser sysInfo = this.getSysInfo();
         if (sysInfo == null) {
             return ResponseValue.error("鐧诲綍鐢ㄦ埛淇℃伅涓嶅瓨鍦�");
@@ -88,7 +93,12 @@
         @ApiImplicitParam(name = "param", value = "鐩樼偣鏉′欢", dataType = "LWhFormInventoryParam", required = true, paramType = "query")
     })
     @GetMapping("/list")
-    public ResponseValue queryList(LWhFormInventoryQry param) {
+    public ResponseValue queryList() {
+        LWhFormInventoryQry param = CommonUtil.getObjFromReq(LWhFormInventoryQry.class);
+        LWhFormInventoryQry param2 = new LWhFormInventoryQry();
+        CommonUtil.copyProperties(param, param2);
+        param = param2;
+
         FinSysTenantUser sysInfo = this.getSysInfo();
         if (sysInfo == null) {
             return ResponseValue.error("鐧诲綍鐢ㄦ埛淇℃伅涓嶅瓨鍦�");
@@ -127,7 +137,12 @@
         @ApiImplicitParam(name = "param", value = "鐩樼偣鍗曚俊鎭�", dataType = "FormInventoryParam")
     })
     @PostMapping("/edit")
-    public ResponseValue edit(@RequestBody FormInventoryParam param) {
+    public ResponseValue edit() {
+        FormInventoryParam param = CommonUtil.getObjFromReqBody(FormInventoryParam.class);
+        FormInventoryParam formInventoryParam = new FormInventoryParam();
+        CommonUtil.copyProperties(param, formInventoryParam);
+        param = formInventoryParam;
+
         LWhFormInventory lWhFormInventory = lWhFormInventoryService.get(new LWhFormInventory(param.getId()));
         if (lWhFormInventory == null) {
             return ResponseValue.error("缂栬緫澶辫触锛�");
@@ -139,8 +154,18 @@
         if (num == 0) {
             return ResponseValue.error("缂栬緫澶辫触锛�");
         }
-        ResponseValue add = this.add(param);
-        if (add.getCode() == ResponseValue.CODE_SUCCESS) {
+
+        //鏍规嵁鐩樼偣浜篿d鏌ヨ鐩樼偣浜�
+        FinSysTenantUser finSysTenantUser = new FinSysTenantUser();
+        finSysTenantUser.setId(param.getOperatorUserId());
+        FinSysTenantUser operatorUser = finSysTenantUserService.get(finSysTenantUser);
+        //鏍规嵁鐩戠洏浜篿d鏌ヨ鐩戠洏浜�
+        FinSysTenantUser monitorUserInfo = new FinSysTenantUser();
+        monitorUserInfo.setId(param.getMonitorUserId());
+        FinSysTenantUser monitorUser = finSysTenantUserService.get(monitorUserInfo);
+        int result = this.lWhFormInventoryService.add(param, this.getSysInfo(), operatorUser, monitorUser);
+
+        if (result > 0) {
             return ResponseValue.success();
         }
 
@@ -186,12 +211,17 @@
         @ApiImplicitParam(name = "param", value = "鐩樼偣鍗曚俊鎭�", dataType = "LWhFormInventoryDto")
     })
     @PostMapping("/temporary/storage")
-    public ResponseValue temporaryStorage(@RequestBody LWhFormInventoryParam dto) {
-        if (dto == null) {
+    public ResponseValue updateTemporaryStorage() {
+        LWhFormInventoryParam param = CommonUtil.getObjFromReqBody(LWhFormInventoryParam.class);
+        LWhFormInventoryParam formInventoryParam = new LWhFormInventoryParam();
+        CommonUtil.copyProperties(param, formInventoryParam);
+        param = formInventoryParam;
+
+        if (param == null) {
             return ResponseValue.error("鍙傛暟閿欒");
         }
 
-        int num = this.lWhFormInventoryService.updateInventoryInfo(dto, 1);
+        int num = this.lWhFormInventoryService.updateInventoryInfo(param, 1, null, null);
         return num > 0 ? ResponseValue.success(1) : ResponseValue.error("鏆傚瓨澶辫触锛�");
     }
 
@@ -206,12 +236,19 @@
         @ApiImplicitParam(name = "param", value = "鐩樼偣鍗曚俊鎭�", dataType = "LWhFormInventoryDto")
     })
     @PostMapping("/finish")
-    public ResponseValue finishPd(@RequestBody LWhFormInventoryParam dto) {
-        if (dto == null) {
+    public ResponseValue updateFinishPd() {
+        LWhFormInventoryParam param = CommonUtil.getObjFromReqBody(LWhFormInventoryParam.class);
+        LWhFormInventoryParam formInventoryParam = new LWhFormInventoryParam();
+        CommonUtil.copyProperties(param, formInventoryParam);
+        param = formInventoryParam;
+
+        if (param == null) {
             return ResponseValue.error("鍙傛暟閿欒");
         }
+        S_user_core currentUser = this.getCurrentUser();
+        FinSysTenantUser sysInfo = this.getSysInfo();
+        int num = this.lWhFormInventoryService.updateInventoryInfo(param, 2, currentUser, sysInfo);
 
-        int num = this.lWhFormInventoryService.updateInventoryInfo(dto, 2);
         return num > 0 ? ResponseValue.success(1) : ResponseValue.error("鐩樼偣澶辫触锛�");
     }
 
@@ -225,7 +262,12 @@
         @ApiImplicitParam(name = "param", value = "鐩樼偣鏉′欢", dataType = "LWhFormInventoryParam", paramType = "query")
     })
     @GetMapping("/list/PdDetail")
-    public ResponseValue queryPdDetailList(LWhFormInventoryQry param) {
+    public ResponseValue queryPdDetailList() {
+        LWhFormInventoryQry param = CommonUtil.getObjFromReq(LWhFormInventoryQry.class);
+        LWhFormInventoryQry param2 = new LWhFormInventoryQry();
+        CommonUtil.copyProperties(param, param2);
+        param = param2;
+
         FinSysTenantUser sysInfo = this.getSysInfo();
         if (sysInfo == null) {
             return ResponseValue.error("鐧诲綍鐢ㄦ埛淇℃伅涓嶅瓨鍦�");

--
Gitblit v1.9.1