Merge pull request #1285 from Bidaya0/fix/narmal-vul-upload-and-heade…

…r-vul

Fix/narmal vul upload and header vul

dongtai_engine/plugins/strategy_headers.py

@@ -98,7 +98,8 @@ def check_response_header(method_pool):
                      method_pool,
                      position='HTTP Response Header')
     except Exception as e:
-        logger.error("check_response_header failed, reason: " + str(e))
+        logger.error("check_response_header failed, reason: " + str(e),
+                     exc_info=e)
 
 
 def save_vul(vul_type, method_pool, position=None, data=None):
@@ -185,8 +186,8 @@ def save_vul(vul_type, method_pool, position=None, data=None):
             client_ip=method_pool.clent_ip,
             param_name=None,
             method_pool_id=method_pool.id,
-            project_version_id=vul.agent.project_version_id,
-            project_id=vul.agent.bind_project_id,
+            project_version_id=method_pool.agent.project_version_id,
+            project_id=method_pool.agent.bind_project_id,
             language=method_pool.agent.language,
             server_id=method_pool.agent.server_id,
         )

dongtai_protocol/report/handler/narmal_vul_handler.py

@@ -279,7 +279,7 @@ def save(self):
             strategy_id=strategy_id,
             uri=self.http_uri,
             http_method=self.http_method,
-            project_version_id=iast_vul.agent.project_version_id,
+            project_version_id=self.agent.project_version_id,
             pk__lt=iast_vul.id,
         ).delete()
         header_vul = None

dongtai_web/projecttemplate/update_department_data.py

@@ -10,5 +10,5 @@ def update_department_data():
         else:
             department_dict[department.id] = f"{department_dict[department.parent_id]}-{department.id}"
         department.department_path = department_dict[department.id]
-        department.token = Token.generate_key()
+        department.token = Token().generate_key()
         department.save()