fix: verify returndatasize from setupAndEnableModule()

bcnmy · Nov 30, 2023 · a2576d6 · a2576d6
1 parent 5c5a6bf
commit a2576d6
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/contracts/smart-account/modules/SecurityPolicyManagerPlugin.sol b/contracts/smart-account/modules/SecurityPolicyManagerPlugin.sol
@@ -87,8 +87,10 @@ contract SecurityPolicyManagerPlugin is ISecurityPolicyManagerPlugin {
 
             switch success
             case 0x1 {
-                moduleInstallationSuccess := mload(ptr)
-                module := mload(add(ptr, 0x60))
+                if eq(size, 0x80) {
+                    moduleInstallationSuccess := mload(ptr)
+                    module := mload(add(ptr, 0x60))
+                }
             }
             case 0x0 {
                 revert(ptr, size)