Add function is_local_url() to check if url is local.

2 files changed, 41 insertions, 0 deletions

diff --git a/include/network.php b/include/network.php
index fa408e602..a236a6f8e 100644
--- a/include/network.php
+++ b/include/network.php
@@ -559,6 +559,14 @@ function z_dns_check($h,$check_mx = 0) {
 	return((@dns_get_record($h,$opts) || filter_var($h, FILTER_VALIDATE_IP)) ? true : false);
 }
 
+function is_local_url($url) {
+  if (str_starts_with($url, z_root()) || str_starts_with($url, '/')) {
+    return true;
+  }
+
+  return false;
+}
+
 /**
  * @brief Validates a given URL.
  *
diff --git a/tests/unit/includes/NetworkTest.php b/tests/unit/includes/NetworkTest.php
new file mode 100644
index 000000000..0b9b42e00
--- /dev/null
+++ b/tests/unit/includes/NetworkTest.php
@@ -0,0 +1,33 @@
+<?php
+/**
+ * tests function from include/network.php
+ *
+ * @package test.util
+ */
+
+use PHPUnit\Framework\TestCase;
+
+require_once('include/network.php');
+
+class NetworkTest extends TestCase {
+
+  public function setup() : void {
+    \App::set_baseurl("https://mytest.org");
+  }
+
+  /**
+   * @dataProvider localUrlTestProvider
+   */
+  public function testIsLocalURL($url, $expected) {
+    $this->assertEquals($expected, is_local_url($url));
+  }
+
+  public function localUrlTestProvider() : array {
+    return [
+      [ '/some/path', true ],
+      [ 'https://mytest.org/some/path', true ],
+      [ 'https://other.site/some/path', false ],
+    ];
+  }
+}
+