Fixes for phar

danog · Mar 21, 2018 · a18918c · a18918c
1 parent 5db10dd
commit a18918c
Showing 1 changed file with 12 additions and 2 deletions.
diff --git a/lib/danog/PrimeModule.php b/lib/danog/PrimeModule.php
@@ -82,7 +82,12 @@ public static function python_single($what)
         if (function_exists('shell_exec')) {
             $res = trim(shell_exec('timeout 10 python '.__DIR__.'/prime.py '.$what.' 2>&1'));
             if ($res == '' || is_null($res) || !is_numeric($res)) {
-                return false;
+                copy(__DIR__.'/prime.py', getcwd().'/.prime.py');
+                $res = trim(shell_exec('timeout 10 python '.getcwd().'/.prime.py '.$what.' 2>&1'));
+                unlink(getcwd().'/.prime.py');
+                if ($res == '' || is_null($res) || !is_numeric($res)) {
+                    return false;
+                }
             }
             $newval = intval($res);
             if (is_int($newval)) {
@@ -116,7 +121,12 @@ public static function python_single_alt($what)
         if (function_exists('shell_exec')) {
             $res = trim(shell_exec('python '.__DIR__.'/alt_prime.py '.$what.' 2>&1'));
             if ($res == '' || is_null($res) || !is_numeric($res)) {
-                return false;
+                copy(__DIR__.'/alt_prime.py', getcwd().'/.alt_prime.py');
+                $res = trim(shell_exec('python '.getcwd().'/.alt_prime.py '.$what.' 2>&1'));
+                unlink(getcwd().'/.alt_prime.py');
+                if ($res == '' || is_null($res) || !is_numeric($res)) {
+                    return false;
+                }
             }
             $newval = intval($res);
             if (is_int($newval)) {