Handle initializing authentication context when graaljs proxy objects…

… are deserialized
wso2 · Feb 21, 2024 · e625ec0 · e625ec0
1 parent 5bcb3b4
commit e625ec0
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/...ation/authentication/framework/config/model/graph/graaljs/JsGraalGraphBuilderFactory.java b/...ation/authentication/framework/config/model/graph/graaljs/JsGraalGraphBuilderFactory.java
@@ -28,6 +28,7 @@
 import org.wso2.carbon.identity.application.authentication.framework.config.model.graph.JsBaseGraphBuilder;
 import org.wso2.carbon.identity.application.authentication.framework.config.model.graph.JsBaseGraphBuilderFactory;
 import org.wso2.carbon.identity.application.authentication.framework.config.model.graph.JsSerializer;
+import org.wso2.carbon.identity.application.authentication.framework.config.model.graph.js.AbstractJSObjectWrapper;
 import org.wso2.carbon.identity.application.authentication.framework.config.model.graph.js.JsLogger;
 import org.wso2.carbon.identity.application.authentication.framework.context.AuthenticationContext;
 import org.wso2.carbon.identity.application.authentication.framework.exception.FrameworkException;
@@ -64,6 +65,9 @@ public static void restoreCurrentContext(AuthenticationContext authContext, Cont
         if (map != null) {
             for (Map.Entry<String, Object> entry : map.entrySet()) {
                 Object deserializedValue = GraalSerializer.getInstance().fromJsSerializable(entry.getValue(), context);
+                if (deserializedValue instanceof AbstractJSObjectWrapper) {
+                    ((AbstractJSObjectWrapper) deserializedValue).initializeContext(authContext);
+                }
                 bindings.putMember(entry.getKey(), deserializedValue);
             }
         }