[Mime] Use streams instead of loading raw message generator into memory

bytestream · fabpot · commit 5471940a0c4b · 2024-06-03T07:58:22.000+02:00
diff --git a/src/Symfony/Component/Mime/RawMessage.php b/src/Symfony/Component/Mime/RawMessage.php
@@ -18,20 +18,32 @@
  */
 class RawMessage
 {
-    private iterable|string $message;
+    /** @var iterable|string|resource */
+    private $message;
     private bool $isGeneratorClosed;
 
     public function __construct(iterable|string $message)
     {
         $this->message = $message;
     }
 
+    public function __destruct()
+    {
+        if (\is_resource($this->message)) {
+            fclose($this->message);
+        }
+    }
+
     public function toString(): string
     {
         if (\is_string($this->message)) {
             return $this->message;
         }
 
+        if (\is_resource($this->message)) {
+            return stream_get_contents($this->message, -1, 0);
+        }
+
         $message = '';
         foreach ($this->message as $chunk) {
             $message .= $chunk;
@@ -53,10 +65,19 @@ public function toIterable(): iterable
             return;
         }
 
+        if (\is_resource($this->message)) {
+            rewind($this->message);
+            while ($line = fgets($this->message)) {
+                yield $line;
+            }
+
+            return;
+        }
+
         if ($this->message instanceof \Generator) {
-            $message = '';
+            $message = fopen('php://temp', 'w+');
             foreach ($this->message as $chunk) {
-                $message .= $chunk;
+                fwrite($message, $chunk);
                 yield $chunk;
             }
             $this->isGeneratorClosed = !$this->message->valid();
