Use streams instead of loading raw message generator into memory

bytestream · bytestream · commit f1e77e458773 · 2024-05-29T11:26:09.000+01: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();
