修复升级漏洞

2023-06-09 22:14:29 +08:00 · 2023-06-09 22:14:29 +08:00 · ee0bc82fbd
commit ee0bc82fbd
parent 4bb2cd7219
1 changed files with 7 additions and 0 deletions
--- a/modules/equipment/api_upgrade.go
+++ b/modules/equipment/api_upgrade.go
@ -218,6 +218,13 @@ func (this *apiComp) Upgrade(session comm.IUserSession, req *pb.EquipmentUpgrade
 					} else {
 						equipments[i] = equipment
 						confs[i] = conf
+						if suitconfs[i], err = this.module.configure.getEquipSuit(confs[i].Suittype); err != nil {
+							errdata = &pb.ErrorData{
+								Code:    pb.ErrorCode_ConfigNoFound,
+								Title:   pb.ErrorCode_ConfigNoFound.ToString(),
+								Message: err.Error(),
+							}
+						}
 					}
 					suit = nil
 					for _, suit = range suits {