Fetching new queue tasks in a bulk to increase speed

[friendica.git] / include / gprobe.php

diff --git a/include/gprobe.php b/include/gprobe.php
index 5ca42729a701c60159bfd2b0213831ee86bef229..788a9eb0447fef3ca489dfb0b26cbbbd90380092 100644 (file)
--- a/include/gprobe.php
+++ b/include/gprobe.php
@@ -1,65 +1,56 @@
 <?php
 
-require_once("boot.php");
-require_once('include/Scrape.php');
-require_once('include/socgraph.php');
+use Friendica\Core\Config;
 
-function gprobe_run($argv, $argc){
-       global $a, $db;
+require_once 'include/probe.php';
+require_once 'include/socgraph.php';
+require_once 'include/datetime.php';
 
-       if(is_null($a)) {
-               $a = new App;
+function gprobe_run(&$argv, &$argc){
+       if ($argc != 2) {
+               return;
        }
-  
-       if(is_null($db)) {
-           @include(".htconfig.php");
-       require_once("dba.php");
-           $db = new dba($db_host, $db_user, $db_pass, $db_data);
-       unset($db_host, $db_user, $db_pass, $db_data);
-       };
+       $url = hex2bin($argv[1]);
 
-       require_once('include/session.php');
-       require_once('include/datetime.php');
+       $r = q("SELECT `id`, `url`, `network` FROM `gcontact` WHERE `nurl` = '%s' ORDER BY `id` LIMIT 1",
+               dbesc(normalise_link($url))
+       );
 
-       load_config('config');
-       load_config('system');
+       logger("gprobe start for ".normalise_link($url), LOGGER_DEBUG);
 
-       $a->set_baseurl(get_config('system','url'));
+       if (!dbm::is_result($r)) {
 
-       load_hooks();
+               // Is it a DDoS attempt?
+               $urlparts = parse_url($url);
 
-       if($argc != 2)
-               return;
-
-       $url = hex2bin($argv[1]);
+               $result = Cache::get("gprobe:".$urlparts["host"]);
+               if (!is_null($result)) {
+                       if (in_array($result["network"], array(NETWORK_FEED, NETWORK_PHANTOM))) {
+                               logger("DDoS attempt detected for ".$urlparts["host"]." by ".$_SERVER["REMOTE_ADDR"].". server data: ".print_r($_SERVER, true), LOGGER_DEBUG);
+                               return;
+                       }
+               }
 
-       $r = q("select * from gcontact where nurl = '%s' limit 1",
-               dbesc(normalise_link($url))
-       );
+               $arr = probe_url($url);
 
-       if(! count($r)) {
+               if (is_null($result)) {
+                       Cache::set("gprobe:".$urlparts["host"], $arr);
+               }
 
-               $arr = probe_url($url);
-               if(count($arr) && x($arr,'network') && $arr['network'] === NETWORK_DFRN) {
-                       q("insert into `gcontact` (`name`,`url`,`nurl`,`photo`)
-                               values ( '%s', '%s', '%s', '%s') ",
-                               dbesc($arr['name']),
-                               dbesc($arr['url']),
-                               dbesc(normalise_link($arr['url'])),
-                               dbesc($arr['photo'])
-                       );
+               if (!in_array($arr["network"], array(NETWORK_FEED, NETWORK_PHANTOM))) {
+                       update_gcontact($arr);
                }
-               $r = q("select * from gcontact where nurl = '%s' limit 1",
+
+               $r = q("SELECT `id`, `url`, `network` FROM `gcontact` WHERE `nurl` = '%s' ORDER BY `id` LIMIT 1",
                        dbesc(normalise_link($url))
                );
        }
-       if(count($r))
-               poco_load(0,0,$r[0]['id'], str_replace('/profile/','/poco/',$r[0]['url']));
-               
-       return;
-}
+       if (dbm::is_result($r)) {
+               // Check for accessibility and do a poco discovery
+               if (poco_last_updated($r[0]['url'], true) && ($r[0]["network"] == NETWORK_DFRN))
+                       poco_load(0,0,$r[0]['id'], str_replace('/profile/','/poco/',$r[0]['url']));
+       }
 
-if (array_search(__file__,get_included_files())===0){
-  gprobe_run($argv,$argc);
-  killme();
+       logger("gprobe end for ".normalise_link($url), LOGGER_DEBUG);
+       return;
 }