From bcd90775e7c9e80cd3fd7b528cf0d37b16ee7132 Mon Sep 17 00:00:00 2001
From: Rostislav Wolny <rosta@mailpoet.com>
Date: Tue, 12 Jan 2021 13:20:40 +0100
Subject: [PATCH] Fix full path disclosure script for files with return on firs
 line

[MAILPOET-3364]
---
 tasks/fix-full-path-disclosure.php | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/tasks/fix-full-path-disclosure.php b/tasks/fix-full-path-disclosure.php
index c48bd05eb3..a323d2bc6f 100644
--- a/tasks/fix-full-path-disclosure.php
+++ b/tasks/fix-full-path-disclosure.php
@@ -89,6 +89,11 @@ foreach ($files as $file) {
     }
   }
 
+  // Add new line after <?php in case <?php return... is detected on the first line and we want to append after the first line
+  if ($line === 1 && is_array($tokens[1]) && $tokens[1][0] === T_RETURN && $tokens[1][2] === 1) {
+    $data = preg_replace('/<\?php/', "<?php\n", $data, 1);
+  }
+
   // inject $code after line give by detected $line
   // NOTE: UTF-8 'u' modifier is not added on purpose since we only need to count '\n' occurrences
   //       and 'u' modifier breaks on some symbols (i.e. those used in Symfony's mb_string polyfill)