From a7aac4b6f2e7b87c3730aab3c15aeaf260e3612d Mon Sep 17 00:00:00 2001
From: Denis <denis@vaadin.com>
Date: Tue, 5 Oct 2021 11:08:49 +0300
Subject: [PATCH] fix: return immidiately if VaadinSession is not initialized
 (#11974)

fixes #11961
---
 .../com/vaadin/flow/server/VaadinSession.java | 44 +++++++++++++++----
 .../vaadin/flow/server/VaadinSessionTest.java | 20 +++++++++
 2 files changed, 56 insertions(+), 8 deletions(-)

diff --git a/flow-server/src/main/java/com/vaadin/flow/server/VaadinSession.java b/flow-server/src/main/java/com/vaadin/flow/server/VaadinSession.java
index 9b07d9830eb..68705d81ff8 100644
--- a/flow-server/src/main/java/com/vaadin/flow/server/VaadinSession.java
+++ b/flow-server/src/main/java/com/vaadin/flow/server/VaadinSession.java
@@ -167,16 +167,25 @@ public void valueBound(HttpSessionBindingEvent arg0) {
      */
     @Override
     public void valueUnbound(HttpSessionBindingEvent event) {
-        // If we are going to be unbound from the session, the session must be
-        // closing
-        // Notify the service
-        if (service == null) {
+        // The Vaadin session instance may be not yet initialized properly via
+        // {@link #refreshTransients(WrappedSession, VaadinService)}. It happens
+        // when the session is deserialized and container decides that it's
+        // expired. Such session is not known to anyone (except deserializer)
+        // and nothins should be done with it: just return immediately.
+        //
+        // Be aware that not initialized session doesn't have the
+        // correct/expected state: it has no lock, service, session, etc.
+        if (!isInitialized()) {
             getLogger().warn(
                     "A VaadinSession instance not associated to any service is getting unbound. "
                             + "Session destroy events will not be fired and UIs in the session will not get detached. "
                             + "This might happen if a session is deserialized but never used before it expires.");
-        } else if (VaadinService.getCurrentRequest() != null
-                && getCurrent() == this) {
+            return;
+        }
+        // If we are going to be unbound from the session, the session must be
+        // closing
+        // Notify the service
+        if (VaadinService.getCurrentRequest() != null && getCurrent() == this) {
             checkHasLock();
             // Ignore if the session is being moved to a different backing
             // session or if GAEVaadinServlet is doing its normal cleanup.
@@ -624,8 +633,10 @@ public void removeUI(UI ui) {
      * <code>Lock</code> interface than {@link Lock#lock()} and
      * {@link Lock#unlock()}.
      *
-     * @return the <code>Lock</code> that is used for synchronization, never
-     *         <code>null</code>
+     * @return the <code>Lock</code> that is used for synchronization, it's
+     *         never <code>null</code> for the session which is in use, i.e. if
+     *         {@link #refreshTransients(WrappedSession, VaadinService)} has
+     *         been called for it
      * @see #lock()
      * @see Lock
      */
@@ -1098,4 +1109,21 @@ public StreamResourceRegistry getResourceRegistry() {
         return resourceRegistry;
     }
 
+    /**
+     * Checks whether the session is properly initialized/in use.
+     * <p>
+     * The session is initialized if
+     * {@link #refreshTransients(WrappedSession, VaadinService)} has been called
+     * for it. If the session is just created or deserialized but not yet in
+     * real use then it's not initialized and it's state is incomplete.
+     * 
+     * @return whether the session is initialized
+     */
+    private boolean isInitialized() {
+        boolean isInitialized = service != null;
+        assert isInitialized
+                || session == null : "The wrapped session must be null if the service is null (which happens after deserialization)";
+        return isInitialized;
+    }
+
 }
diff --git a/flow-server/src/test/java/com/vaadin/flow/server/VaadinSessionTest.java b/flow-server/src/test/java/com/vaadin/flow/server/VaadinSessionTest.java
index 8e3511696c2..ee5c67e9e65 100644
--- a/flow-server/src/test/java/com/vaadin/flow/server/VaadinSessionTest.java
+++ b/flow-server/src/test/java/com/vaadin/flow/server/VaadinSessionTest.java
@@ -463,4 +463,24 @@ protected Lock getSessionLock(WrappedSession wrappedSession) {
         Assert.assertNull(vaadinSession.getSession());
     }
 
+    @Test
+    public void valueUnbound_sessionIsNotInitialized_noAnyInteractions() {
+        VaadinSession session = Mockito.spy(TestVaadinSession.class);
+
+        HttpSessionBindingEvent event = Mockito
+                .mock(HttpSessionBindingEvent.class);
+        session.valueUnbound(null);
+
+        Mockito.verify(session).valueUnbound(null);
+        Mockito.verifyNoInteractions(event);
+        Mockito.verifyNoMoreInteractions(session);
+    }
+
+    public static class TestVaadinSession extends VaadinSession {
+
+        public TestVaadinSession() {
+            super(null);
+        }
+    }
+
 }