[VarExporter] fix handling of __sleep()

Author: nicolas-grekas

src/Symfony/Component/VarExporter/Internal/Exporter.php

@@ -119,11 +119,10 @@ public static function prepare($values, $objectsPool, &$refsPool, &$objectsCount
             $proto = (array) $proto;
 
             foreach ($arrayValue as $name => $v) {
+                $i = 0;
                 $n = (string) $name;
                 if ('' === $n || "\0" !== $n[0]) {
                     $c = 'stdClass';
-                    $properties[$c][$n] = $v;
-                    unset($sleep[$n]);
                 } elseif ('*' === $n[1]) {
                     $n = substr($n, 3);
                     $c = $reflector->getProperty($n)->class;
@@ -132,26 +131,26 @@ public static function prepare($values, $objectsPool, &$refsPool, &$objectsCount
                     } elseif ('Exception' === $c) {
                         $c = 'ErrorException';
                     }
-                    $properties[$c][$n] = $v;
-                    unset($sleep[$n]);
                 } else {
                     $i = strpos($n, "\0", 2);
                     $c = substr($n, 1, $i - 1);
                     $n = substr($n, 1 + $i);
-                    if (null === $sleep) {
-                        $properties[$c][$n] = $v;
-                    } elseif (isset($sleep[$n]) && $c === $class) {
-                        $properties[$c][$n] = $v;
-                        unset($sleep[$n]);
+                }
+                if (null !== $sleep) {
+                    if (!isset($sleep[$n]) || ($i && $c !== $class)) {
+                        continue;
                     }
+                    $sleep[$n] = false;
                 }
-                if (\array_key_exists($name, $proto) && $proto[$name] === $v) {
-                    unset($properties[$c][$n]);
+                if (!\array_key_exists($name, $proto) || $proto[$name] !== $v) {
+                    $properties[$c][$n] = $v;
                 }
             }
             if ($sleep) {
                 foreach ($sleep as $n => $v) {
-                    trigger_error(sprintf('serialize(): "%s" returned as member variable from __sleep() but does not exist', $n), E_USER_NOTICE);
+                    if (false !== $v) {
+                        trigger_error(sprintf('serialize(): "%s" returned as member variable from __sleep() but does not exist', $n), E_USER_NOTICE);
+                    }
                 }
             }
 

src/Symfony/Component/VarExporter/Tests/Fixtures/wakeup-refl.php

@@ -0,0 +1,13 @@
+<?php
+
+return \Symfony\Component\VarExporter\Internal\Hydrator::hydrate(
+    $o = [
+        clone (\Symfony\Component\VarExporter\Internal\Registry::$prototypes['Symfony\\Component\\VarExporter\\Tests\\MyWakeup'] ?? \Symfony\Component\VarExporter\Internal\Registry::p('Symfony\\Component\\VarExporter\\Tests\\MyWakeup')),
+    ],
+    null,
+    [],
+    $o[0],
+    [
+        1 => 0,
+    ]
+);

src/Symfony/Component/VarExporter/Tests/Fixtures/wakeup.php

@@ -12,9 +12,6 @@
                 $o[1],
                 123,
             ],
-            'bis' => [
-                1 => 123,
-            ],
             'baz' => [
                 1 => 123,
             ],

src/Symfony/Component/VarExporter/Tests/VarExporterTest.php

@@ -97,6 +97,9 @@ public function testExport(string $testName, $value, bool $staticValueExpected =
         $marshalledValue = include $fixtureFile;
 
         if (!$isStaticValue) {
+            if ($value instanceof MyWakeup) {
+                $value->bis = null;
+            }
             $this->assertDumpEquals($value, $marshalledValue);
         } else {
             $this->assertSame($value, $marshalledValue);
@@ -184,6 +187,11 @@ public function provideExport()
         yield array('final-array-iterator', new FinalArrayIterator());
 
         yield array('final-stdclass', new FinalStdClass());
+
+        $value = new MyWakeup();
+        $value->bis = new \ReflectionClass($value);
+
+        yield array('wakeup-refl', $value);
     }
 }