Detection of local requests

[friendica.git] / src / Worker / UpdateGServer.php

diff --git a/src/Worker/UpdateGServer.php b/src/Worker/UpdateGServer.php
index b94c5343a0f58815786f5b94410b7982e92c7d42..bc836253a5c110ab90fe74d351331718cc35d358 100644 (file)
--- a/src/Worker/UpdateGServer.php
+++ b/src/Worker/UpdateGServer.php
@@ -1,10 +1,28 @@
 <?php
 /**
- * @file src/Worker/UpdateGServer.php
+ * @copyright Copyright (C) 2010-2021, the Friendica project
+ *
+ * @license GNU AGPL version 3 or any later version
+ *
+ * This program is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU Affero General Public License as
+ * published by the Free Software Foundation, either version 3 of the
+ * License, or (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU Affero General Public License for more details.
+ *
+ * You should have received a copy of the GNU Affero General Public License
+ * along with this program.  If not, see <https://www.gnu.org/licenses/>.
+ *
  */
+
 namespace Friendica\Worker;
 
 use Friendica\Core\Logger;
+use Friendica\Database\DBA;
 use Friendica\Model\GServer;
 use Friendica\Util\Strings;
 
@@ -12,20 +30,33 @@ class UpdateGServer
 {
        /**
         * Update the given server
-        * @param string $server_url Server URL
+        * @param string  $server_url    Server URL
+        * @param boolean $only_nodeinfo Only use nodeinfo for server detection
         */
-       public static function execute($server_url)
+       public static function execute(string $server_url, bool $only_nodeinfo = false)
        {
                if (empty($server_url)) {
                        return;
                }
 
-               $server_url = filter_var($server_url, FILTER_SANITIZE_URL);
-               if (substr(Strings::normaliseLink($server_url), 0, 7) != 'http://') {
+               $filtered = filter_var($server_url, FILTER_SANITIZE_URL);
+               if (substr(Strings::normaliseLink($filtered), 0, 7) != 'http://') {
+                       GServer::setFailure($server_url);
+                       return;
+               }
+
+               if (($filtered != $server_url) && DBA::exists('gserver', ['nurl' => Strings::normaliseLink($server_url)])) {
+                       GServer::setFailure($server_url);
+                       return;
+               }
+
+               $cleaned = GServer::cleanURL($server_url);
+               if (($cleaned != $server_url) && DBA::exists('gserver', ['nurl' => Strings::normaliseLink($server_url)])) {
+                       GServer::setFailure($server_url);
                        return;
                }
 
-               $ret = GServer::check($server_url);
-               Logger::info('Updated gserver', ['url' => $server_url, 'result' => $ret]);
+               $ret = GServer::check($filtered, '', true, $only_nodeinfo);
+               Logger::info('Updated gserver', ['url' => $filtered, 'result' => $ret]);
        }
 }