spam/spam.php

   1 <?php
   2 // $Id: spam.php,v 1.213 2008/12/30 11:41:38 henoheno Exp $
   3 // Copyright (C) 2006-2007 PukiWiki Developers Team
   4 // License: GPL v2 or (at your option) any later version
   5 //
   6 // Functions for Concept-work of spam-uri metrics
   7 //
   8 // (PHP 4 >= 4.3.0): preg_match_all(PREG_OFFSET_CAPTURE): $method['uri_XXX'] related feature
   9
  10 if (! defined('SPAM_INI_FILE'))   define('SPAM_INI_FILE',   'spam.ini.php');
  11 if (! defined('DOMAIN_INI_FILE')) define('DOMAIN_INI_FILE', 'domain.ini.php');
  12
  13 // ---------------------
  14 // Compat etc
  15
  16 // (PHP 4 >= 4.2.0): var_export(): mail-reporting and dump related
  17 if (! function_exists('var_export')) {
  18         function var_export() {
  19                 return 'var_export() is not found on this server' . "\n";
  20         }
  21 }
  22
  23 // (PHP 4 >= 4.2.0): preg_grep() enables invert option
  24 function preg_grep_invert($pattern = '//', $input = array())
  25 {
  26         static $invert;
  27         if (! isset($invert)) $invert = defined('PREG_GREP_INVERT');
  28
  29         if ($invert) {
  30                 return preg_grep($pattern, $input, PREG_GREP_INVERT);
  31         } else {
  32                 $result = preg_grep($pattern, $input);
  33                 if ($result) {
  34                         return array_diff($input, preg_grep($pattern, $input));
  35                 } else {
  36                         return $input;
  37                 }
  38         }
  39 }
  40
  41
  42 // ---------------------
  43 // Utilities
  44
  45 // Very roughly, shrink the lines of var_export()
  46 // NOTE: If the same data exists, it must be corrupted.
  47 function var_export_shrink($expression, $return = FALSE, $ignore_numeric_keys = FALSE)
  48 {
  49         $result = var_export($expression, TRUE);
  50
  51         $result = preg_replace(
  52                 // Remove a newline and spaces
  53                 '# => \n *array \(#', ' => array (',
  54                 $result
  55         );
  56
  57         if ($ignore_numeric_keys) {
  58                 $result =preg_replace(
  59                         // Remove numeric keys
  60                         '#^( *)[0-9]+ => #m', '$1',
  61                         $result
  62                 );
  63         }
  64
  65         if ($return) {
  66                 return $result;
  67         } else {
  68                 echo   $result;
  69                 return NULL;
  70         }
  71 }
  72
  73 // Data structure: Create an array they _refer_only_one_ value
  74 function one_value_array($num = 0, $value = NULL)
  75 {
  76         $num   = max(0, intval($num));
  77         $array = array();
  78
  79         for ($i = 0; $i < $num; $i++) {
  80                 $array[] = & $value;
  81         }
  82
  83         return $array;
  84 }
  85
  86 // Reverse $string with specified delimiter
  87 function delimiter_reverse($string = 'foo.bar.example.com', $from_delim = '.', $to_delim = NULL)
  88 {
  89         $to_null = ($to_delim === NULL);
  90
  91         if (! is_string($from_delim) || (! $to_null && ! is_string($to_delim))) {
  92                 return FALSE;
  93         }
  94         if (is_array($string)) {
  95                 // Map, Recurse
  96                 $count = count($string);
  97                 $from  = one_value_array($count, $from_delim);
  98                 if ($to_null) {
  99                         // Note: array_map() vanishes all keys
 100                         return array_map('delimiter_reverse', $string, $from);
 101                 } else {
 102                         $to = one_value_array($count, $to_delim);
 103                         // Note: array_map() vanishes all keys
 104                         return array_map('delimiter_reverse', $string, $from, $to);
 105                 }
 106         }
 107         if (! is_string($string)) {
 108                 return FALSE;
 109         }
 110
 111         // Returns com.example.bar.foo
 112         if ($to_null) $to_delim = & $from_delim;
 113         return implode($to_delim, array_reverse(explode($from_delim, $string)));
 114 }
 115
 116 // ksort() by domain
 117 function ksort_by_domain(& $array)
 118 {
 119         $sort = array();
 120         foreach(array_keys($array) as $key) {
 121                 $reversed = delimiter_reverse($key);
 122                 if ($reversed !== FALSE) {
 123                         $sort[$reversed] = $key;
 124                 }
 125         }
 126         ksort($sort, SORT_STRING);
 127
 128         $result = array();
 129         foreach($sort as $key) {
 130                 $result[$key] = & $array[$key];
 131         }
 132
 133         $array = $result;
 134 }
 135
 136 // Roughly strings(1) using PCRE
 137 // This function is useful to:
 138 //   * Reduce the size of data, from removing unprintable binary data
 139 //   * Detect _bare_strings_ from binary data
 140 // References:
 141 //   http://www.freebsd.org/cgi/man.cgi?query=strings (Man-page of GNU strings)
 142 //   http://www.pcre.org/pcre.txt
 143 // Note: mb_ereg_replace() is one of mbstring extension's functions
 144 //   and need to init its encoding.
 145 function strings($binary = '', $min_len = 4, $ignore_space = FALSE, $multibyte = FALSE)
 146 {
 147         // String only
 148         $binary = (is_array($binary) || $binary === TRUE) ? '' : strval($binary);
 149
 150         $regex = $ignore_space ?
 151                 '[^[:graph:] \t\n]+' :          // Remove "\0" etc, and readable spaces
 152                 '[^[:graph:][:space:]]+';       // Preserve readable spaces if possible
 153
 154         $binary = $multibyte ?
 155                 mb_ereg_replace($regex,           "\n",  $binary) :
 156                 preg_replace('/' . $regex . '/s', "\n",  $binary);
 157
 158         if ($ignore_space) {
 159                 $binary = preg_replace(
 160                         array(
 161                                 '/[ \t]{2,}/',
 162                                 '/^[ \t]/m',
 163                                 '/[ \t]$/m',
 164                         ),
 165                         array(
 166                                 ' ',
 167                                 '',
 168                                 ''
 169                         ),
 170                          $binary);
 171         }
 172
 173         if ($min_len > 1) {
 174                 // The last character seems "\n" or not
 175                 $br = (! empty($binary) && $binary[strlen($binary) - 1] == "\n") ? "\n" : '';
 176
 177                 $min_len = min(1024, intval($min_len));
 178                 $regex = '/^.{' . $min_len . ',}/S';
 179                 $binary = implode("\n", preg_grep($regex, explode("\n", $binary))) . $br;
 180         }
 181
 182         return $binary;
 183 }
 184
 185
 186 // ---------------------
 187 // Utilities: Arrays
 188
 189 // Count leaves (A leaf = value that is not an array, or an empty array)
 190 function array_count_leaves($array = array(), $count_empty = FALSE)
 191 {
 192         if (! is_array($array) || (empty($array) && $count_empty)) return 1;
 193
 194         // Recurse
 195         $count = 0;
 196         foreach ($array as $part) {
 197                 $count += array_count_leaves($part, $count_empty);
 198         }
 199         return $count;
 200 }
 201
 202 // Merge two leaves
 203 // Similar to PHP array_merge_leaves(), except strictly preserving keys as string
 204 function array_merge_leaves($array1, $array2, $sort_keys = TRUE)
 205 {
 206         // Array(s) only
 207         $is_array1 = is_array($array1);
 208         $is_array2 = is_array($array2);
 209         if ($is_array1) {
 210                 if ($is_array2) {
 211                         ;       // Pass
 212                 } else {
 213                         return $array1;
 214                 }
 215         } else if ($is_array2) {
 216                 return $array2;
 217         } else {
 218                 return $array2; // Not array ($array1 is overwritten)
 219         }
 220
 221         $keys_all = array_merge(array_keys($array1), array_keys($array2));
 222         if ($sort_keys) sort($keys_all, SORT_STRING);
 223
 224         $result = array();
 225         foreach($keys_all as $key) {
 226                 $isset1 = isset($array1[$key]);
 227                 $isset2 = isset($array2[$key]);
 228                 if ($isset1 && $isset2) {
 229                         // Recurse
 230                         $result[$key] = array_merge_leaves($array1[$key], $array2[$key], $sort_keys);
 231                 } else if ($isset1) {
 232                         $result[$key] = & $array1[$key];
 233                 } else {
 234                         $result[$key] = & $array2[$key];
 235                 }
 236         }
 237         return $result;
 238 }
 239
 240 // An array-leaves to a flat array
 241 function array_flat_leaves($array, $unique = TRUE)
 242 {
 243         if (! is_array($array)) return $array;
 244
 245         $tmp = array();
 246         foreach(array_keys($array) as $key) {
 247                 if (is_array($array[$key])) {
 248                         // Recurse
 249                         foreach(array_flat_leaves($array[$key]) as $_value) {
 250                                 $tmp[] = $_value;
 251                         }
 252                 } else {
 253                         $tmp[] = & $array[$key];
 254                 }
 255         }
 256
 257         return $unique ? array_values(array_unique($tmp)) : $tmp;
 258 }
 259
 260 // $array['something'] => $array['wanted']
 261 function array_rename_keys(& $array, $keys = array('from' => 'to'), $force = FALSE, $default = '')
 262 {
 263         if (! is_array($array) || ! is_array($keys)) return FALSE;
 264
 265         // Nondestructive test
 266         if (! $force) {
 267                 foreach(array_keys($keys) as $from) {
 268                         if (! isset($array[$from])) {
 269                                 return FALSE;
 270                         }
 271                 }
 272         }
 273
 274         foreach($keys as $from => $to) {
 275                 if ($from === $to) continue;
 276                 if (! $force || isset($array[$from])) {
 277                         $array[$to] = & $array[$from];
 278                         unset($array[$from]);
 279                 } else  {
 280                         $array[$to] = $default;
 281                 }
 282         }
 283
 284         return TRUE;
 285 }
 286
 287 // Remove redundant values from array()
 288 function array_unique_recursive($array = array())
 289 {
 290         if (! is_array($array)) return $array;
 291
 292         $tmp = array();
 293         foreach($array as $key => $value){
 294                 if (is_array($value)) {
 295                         $array[$key] = array_unique_recursive($value);
 296                 } else {
 297                         if (isset($tmp[$value])) {
 298                                 unset($array[$key]);
 299                         } else {
 300                                 $tmp[$value] = TRUE;
 301                         }
 302                 }
 303         }
 304
 305         return $array;
 306 }
 307
 308
 309 // ---------------------
 310 // Part One : Checker
 311
 312 // Rough implementation of globbing
 313 //
 314 // USAGE: $regex = '/^' . generate_glob_regex('*.txt', '/') . '$/i';
 315 //
 316 function generate_glob_regex($string = '', $divider = '/')
 317 {
 318         static $from = array(
 319                          1 => '*',
 320                         11 => '?',
 321         //              22 => '[',      // Maybe cause regex compilation error (e.g. '[]')
 322         //              23 => ']',      //
 323                 );
 324         static $mid = array(
 325                          1 => '_AST_',
 326                         11 => '_QUE_',
 327         //              22 => '_RBR_',
 328         //              23 => '_LBR_',
 329                 );
 330         static $to = array(
 331                          1 => '.*',
 332                         11 => '.',
 333         //              22 => '[',
 334         //              23 => ']',
 335                 );
 336
 337         if (! is_string($string)) return '';
 338
 339         $string = str_replace($from, $mid, $string); // Hide
 340         $string = preg_quote($string, $divider);
 341         $string = str_replace($mid, $to, $string);   // Unhide
 342
 343         return $string;
 344 }
 345
 346 // Generate host (FQDN, IPv4, ...) regex
 347 // 'localhost'     : Matches with 'localhost' only
 348 // 'example.org'   : Matches with 'example.org' only (See host_normalize() about 'www')
 349 // '.example.org'  : Matches with ALL FQDN ended with '.example.org'
 350 // '*.example.org' : Almost the same of '.example.org' except 'www.example.org'
 351 // '10.20.30.40'   : Matches with IPv4 address '10.20.30.40' only
 352 // [TODO] '192.'   : Matches with all IPv4 hosts started with '192.'
 353 // TODO: IPv4, CIDR?, IPv6
 354 function generate_host_regex($string = '', $divider = '/')
 355 {
 356         if (! is_string($string)) return '';
 357
 358         if (mb_strpos($string, '.') === FALSE) {
 359                 // localhost
 360                 return generate_glob_regex($string, $divider);
 361         }
 362
 363         if (is_ip($string)) {
 364                 // IPv4
 365                 return generate_glob_regex($string, $divider);
 366         } else {
 367                 // FQDN or something
 368                 $part = explode('.', $string, 2);
 369                 if ($part[0] == '') {
 370                         // .example.org
 371                         $part[0] = '(?:.*\.)?';
 372                 } else if ($part[0] == '*') {
 373                         // *.example.org
 374                         $part[0] = '.*\.';
 375                 } else {
 376                         // example.org, etc
 377                         return generate_glob_regex($string, $divider);
 378                 }
 379                 $part[1] = generate_glob_regex($part[1], $divider);
 380                 return implode('', $part);
 381         }
 382 }
 383
 384 // Rough hostname checker
 385 // TODO: Strict digit, 0x, CIDR, '999.999.999.999', ':', '::G'
 386 function is_ip($string = '')
 387 {
 388         if (! is_string($string)) return FALSE;
 389
 390         if (strpos($string, ':') !== FALSE) {
 391                 return 6;       // Seems IPv6
 392         }
 393
 394         if (preg_match('/^' .
 395                 '(?:[0-9]{1,3}\.){3}[0-9]{1,3}' . '|' .
 396                 '(?:[0-9]{1,3}\.){1,3}'         . '$/',
 397                 $string)) {
 398                 return 4;       // Seems IPv4(dot-decimal)
 399         }
 400
 401         return FALSE;   // Seems not IP
 402 }
 403
 404 // Load SPAM_INI_FILE and return parsed one
 405 function get_blocklist($list = '')
 406 {
 407         static $regexes;
 408
 409         if ($list === NULL) {
 410                 $regexes = NULL;        // Unset
 411                 return array();
 412         }
 413
 414         if (! isset($regexes)) {
 415                 $regexes = array();
 416                 if (file_exists(SPAM_INI_FILE)) {
 417                         $blocklist = array();
 418
 419                         include(SPAM_INI_FILE);
 420                         //      $blocklist['list'] = array(
 421                         //      //'goodhost' => FALSE;
 422                         //      'badhost' => TRUE;
 423                         // );
 424                         //      $blocklist['badhost'] = array(
 425                         //              '*.blogspot.com',       // Blog services's subdomains (only)
 426                         //              'IANA-examples' => '#^(?:.*\.)?example\.(?:com|net|org)$#',
 427                         //      );
 428
 429                         foreach(array(
 430                                         'pre',
 431                                         'list',
 432                                 ) as $special) {
 433
 434                                 if (! isset($blocklist[$special])) continue;
 435
 436                                 $regexes[$special] = $blocklist[$special];
 437
 438                                 foreach(array_keys($blocklist[$special]) as $_list) {
 439                                         if (! isset($blocklist[$_list])) continue;
 440
 441                                         foreach ($blocklist[$_list] as $key => $value) {
 442                                                 if (is_array($value)) {
 443                                                         $regexes[$_list][$key] = array();
 444                                                         foreach($value as $_key => $_value) {
 445                                                                 get_blocklist_add($regexes[$_list][$key], $_key, $_value);
 446                                                         }
 447                                                 } else {
 448                                                         get_blocklist_add($regexes[$_list], $key, $value);
 449                                                 }
 450                                         }
 451
 452                                         unset($blocklist[$_list]);
 453                                 }
 454                         }
 455                 }
 456         }
 457
 458         if ($list === '') {
 459                 return $regexes;                // ALL of
 460         } else if (isset($regexes[$list])) {
 461                 return $regexes[$list]; // A part of
 462         } else {
 463                 return array();                 // Found nothing
 464         }
 465 }
 466
 467 // Subroutine of get_blocklist(): Add new regex to the $array
 468 function get_blocklist_add(& $array, $key = 0, $value = '*.example.org/path/to/file.html')
 469 {
 470         if (is_string($key)) {
 471                 $array[$key]   = & $value; // Treat $value as a regex for FQDN(host)s
 472         } else {
 473                 $array[$value] = '#^' . generate_host_regex($value, '#') . '$#i';
 474         }
 475 }
 476
 477 // Blocklist metrics: Separate $host, to $blocked and not blocked
 478 function blocklist_distiller(& $hosts, $keys = array('goodhost', 'badhost'), $asap = FALSE)
 479 {
 480         if (! is_array($hosts)) $hosts = array($hosts);
 481         if (! is_array($keys))  $keys  = array($keys);
 482
 483         $list = get_blocklist('list');
 484         $blocked = array();
 485
 486         foreach($keys as $key){
 487                 foreach (get_blocklist($key) as $label => $regex) {
 488                         if (is_array($regex)) {
 489                                 foreach($regex as $_label => $_regex) {
 490                                         $group = preg_grep($_regex, $hosts);
 491                                         if ($group) {
 492                                                 $hosts = array_diff($hosts, $group);
 493                                                 $blocked[$key][$label][$_label] = $group;
 494                                                 if ($asap && $list[$key]) break;
 495                                         }
 496                                 }
 497                         } else {
 498                                 $group = preg_grep($regex, $hosts);
 499                                 if ($group) {
 500                                         $hosts = array_diff($hosts, $group);
 501                                         $blocked[$key][$label] = $group;
 502                                         if ($asap && $list[$key]) break;
 503                                 }
 504                         }
 505                 }
 506         }
 507
 508         return $blocked;
 509 }
 510
 511
 512 // ---------------------
 513
 514
 515 // Default (enabled) methods and thresholds (for content insertion)
 516 function check_uri_spam_method($times = 1, $t_area = 0, $rule = TRUE)
 517 {
 518         $times  = intval($times);
 519         $t_area = intval($t_area);
 520
 521         $positive = array(
 522                 // Thresholds
 523                 'quantity'     =>  8 * $times,  // Allow N URIs
 524                 'non_uniqhost' =>  3 * $times,  // Allow N duped (and normalized) Hosts
 525                 //'non_uniquri'=>  3 * $times,  // Allow N duped (and normalized) URIs
 526
 527                 // Areas
 528                 'area_anchor'  => $t_area,      // Using <a href> HTML tag
 529                 'area_bbcode'  => $t_area,      // Using [url] or [link] BBCode
 530                 //'uri_anchor' => $t_area,      // URI inside <a href> HTML tag
 531                 //'uri_bbcode' => $t_area,      // URI inside [url] or [link] BBCode
 532         );
 533         if ($rule) {
 534                 $bool = array(
 535                         // Rules
 536                         //'asap'   => TRUE,     // Quit or return As Soon As Possible
 537                         'uniqhost' => TRUE,     // Show uniq host (at block notification mail)
 538                         'badhost'  => TRUE,     // Check badhost
 539                 );
 540         } else {
 541                 $bool = array();
 542         }
 543
 544         // Remove non-$positive values
 545         foreach (array_keys($positive) as $key) {
 546                 if ($positive[$key] < 0) unset($positive[$key]);
 547         }
 548
 549         return $positive + $bool;
 550 }
 551
 552 // Simple/fast spam check
 553 function check_uri_spam($target = '', $method = array())
 554 {
 555         // Return value
 556         $progress = array(
 557                 'method'  => array(
 558                         // Theme to do  => Dummy, optional value, or optional array()
 559                         //'quantity'    => 8,
 560                         //'uniqhost'    => TRUE,
 561                         //'non_uniqhost'=> 3,
 562                         //'non_uniquri' => 3,
 563                         //'badhost'     => TRUE,
 564                         //'area_anchor' => 0,
 565                         //'area_bbcode' => 0,
 566                         //'uri_anchor'  => 0,
 567                         //'uri_bbcode'  => 0,
 568                 ),
 569                 'sum' => array(
 570                         // Theme        => Volume found (int)
 571                 ),
 572                 'is_spam' => array(
 573                         // Flag. If someting defined here,
 574                         // one or more spam will be included
 575                         // in this report
 576                 ),
 577                 'blocked' => array(
 578                         // Hosts blocked
 579                         //'category' => array(
 580                         //      'host',
 581                         //)
 582                 ),
 583                 'hosts' => array(
 584                         // Hosts not blocked
 585                 ),
 586         );
 587
 588         // ----------------------------------------
 589         // Aliases
 590
 591         $sum     = & $progress['sum'];
 592         $is_spam = & $progress['is_spam'];
 593         $progress['method'] = & $method;        // Argument
 594         $blocked = & $progress['blocked'];
 595         $hosts   = & $progress['hosts'];
 596         $asap    = isset($method['asap']);
 597
 598         // ----------------------------------------
 599         // Init
 600
 601         if (! is_array($method) || empty($method)) {
 602                 $method = check_uri_spam_method();
 603         }
 604         foreach(array_keys($method) as $key) {
 605                 if (! isset($sum[$key])) $sum[$key] = 0;
 606         }
 607         if (! isset($sum['quantity'])) $sum['quantity'] = 0;
 608
 609         // ----------------------------------------
 610         // Recurse
 611
 612         if (is_array($target)) {
 613                 foreach($target as $str) {
 614                         if (! is_string($str)) continue;
 615
 616                         $_progress = check_uri_spam($str, $method);     // Recurse
 617
 618                         // Merge $sum
 619                         $_sum = & $_progress['sum'];
 620                         foreach (array_keys($_sum) as $key) {
 621                                 if (! isset($sum[$key])) {
 622                                         $sum[$key] = & $_sum[$key];
 623                                 } else {
 624                                         $sum[$key] += $_sum[$key];
 625                                 }
 626                         }
 627
 628                         // Merge $is_spam
 629                         $_is_spam = & $_progress['is_spam'];
 630                         foreach (array_keys($_is_spam) as $key) {
 631                                 $is_spam[$key] = TRUE;
 632                                 if ($asap) break;
 633                         }
 634                         if ($asap && $is_spam) break;
 635
 636                         // Merge only
 637                         $blocked = array_merge_leaves($blocked, $_progress['blocked'], FALSE);
 638                         $hosts   = array_merge_leaves($hosts,   $_progress['hosts'],   FALSE);
 639                 }
 640
 641                 // Unique values
 642                 $blocked = array_unique_recursive($blocked);
 643                 $hosts   = array_unique_recursive($hosts);
 644
 645                 // Recount $sum['badhost']
 646                 $sum['badhost'] = array_count_leaves($blocked);
 647
 648                 return $progress;
 649         }
 650
 651         // ----------------------------------------
 652         // Area measure
 653
 654         if (! $asap || ! $is_spam) {
 655
 656                 // Method pickup
 657                 $_method = array();
 658                 foreach(array(
 659                                 'area_anchor',  // There's HTML anchor tag
 660                                 'area_bbcode',  // There's 'BBCode' linking tag
 661                         ) as $key) {
 662                         if (isset($method[$key])) $_method[$key] = TRUE;
 663                 }
 664
 665                 if ($_method) {
 666                         $_asap   = isset($method['asap']) ? array('asap' => TRUE) : array();
 667                         $_result = area_pickup($target, $_method + $_asap);
 668                         $_asap   = NULL;
 669                 } else {
 670                         $_result = FALSE;
 671                 }
 672
 673                 if ($_result) {
 674                         foreach(array_keys($_method) as $key) {
 675                                 if (isset($_result[$key])) {
 676                                         $sum[$key] = $_result[$key];
 677                                         if (isset($method[$key]) && $sum[$key] > $method[$key]) {
 678                                                 $is_spam[$key] = TRUE;
 679                                         }
 680                                 }
 681                         }
 682                 }
 683
 684                 unset($_asap, $_method, $_result);
 685         }
 686
 687         // Return if ...
 688         if ($asap && $is_spam) return $progress;
 689
 690         // ----------------------------------------
 691         // URI: Pickup
 692
 693         $pickups = spam_uri_pickup($target, $method);
 694
 695
 696         // Return if ...
 697         if (empty($pickups)) return $progress;
 698
 699         // Normalize all
 700         $pickups = uri_pickup_normalize($pickups);
 701
 702         // ----------------------------------------
 703         // Pickup some part of URI
 704
 705         $hosts = array();
 706         foreach ($pickups as $key => $pickup) {
 707                 $hosts[$key] = & $pickup['host'];
 708         }
 709
 710         // ----------------------------------------
 711         // URI: Bad host <pre-filter> (Separate good/bad hosts from $hosts)
 712
 713         if ((! $asap || ! $is_spam) && isset($method['badhost'])) {
 714                 $list    = get_blocklist('pre');
 715                 $blocked = blocklist_distiller($hosts, array_keys($list), $asap);
 716                 foreach($list as $key => $type){
 717                         if (! $type) unset($blocked[$key]); // Ignore goodhost etc
 718                 }
 719                 unset($list);
 720                 if (! empty($blocked)) $is_spam['badhost'] = TRUE;
 721         }
 722
 723         // Return if ...
 724         if ($asap && $is_spam) return $progress;
 725
 726         // Remove blocked from $pickups
 727         foreach(array_keys($pickups) as $key) {
 728                 if (! isset($hosts[$key])) {
 729                         unset($pickups[$key]);
 730                 }
 731         }
 732
 733         // ----------------------------------------
 734         // URI: Check quantity
 735
 736         $sum['quantity'] += count($pickups);
 737                 // URI quantity
 738         if ((! $asap || ! $is_spam) && isset($method['quantity']) &&
 739                 $sum['quantity'] > $method['quantity']) {
 740                 $is_spam['quantity'] = TRUE;
 741         }
 742
 743         // ----------------------------------------
 744         // URI: used inside HTML anchor tag pair
 745
 746         if ((! $asap || ! $is_spam) && isset($method['uri_anchor'])) {
 747                 $key = 'uri_anchor';
 748                 foreach($pickups as $pickup) {
 749                         if (isset($pickup['area'][$key])) {
 750                                 $sum[$key] += $pickup['area'][$key];
 751                                 if(isset($method[$key]) &&
 752                                         $sum[$key] > $method[$key]) {
 753                                         $is_spam[$key] = TRUE;
 754                                         if ($asap && $is_spam) break;
 755                                 }
 756                                 if ($asap && $is_spam) break;
 757                         }
 758                 }
 759         }
 760
 761         // ----------------------------------------
 762         // URI: used inside 'BBCode' pair
 763
 764         if ((! $asap || ! $is_spam) && isset($method['uri_bbcode'])) {
 765                 $key = 'uri_bbcode';
 766                 foreach($pickups as $pickup) {
 767                         if (isset($pickup['area'][$key])) {
 768                                 $sum[$key] += $pickup['area'][$key];
 769                                 if(isset($method[$key]) &&
 770                                         $sum[$key] > $method[$key]) {
 771                                         $is_spam[$key] = TRUE;
 772                                         if ($asap && $is_spam) break;
 773                                 }
 774                                 if ($asap && $is_spam) break;
 775                         }
 776                 }
 777         }
 778
 779         // ----------------------------------------
 780         // URI: Uniqueness (and removing non-uniques)
 781
 782         if ((! $asap || ! $is_spam) && isset($method['non_uniquri'])) {
 783
 784                 $uris = array();
 785                 foreach (array_keys($pickups) as $key) {
 786                         $uris[$key] = uri_pickup_implode($pickups[$key]);
 787                 }
 788                 $count = count($uris);
 789                 $uris  = array_unique($uris);
 790                 $sum['non_uniquri'] += $count - count($uris);
 791                 if ($sum['non_uniquri'] > $method['non_uniquri']) {
 792                         $is_spam['non_uniquri'] = TRUE;
 793                 }
 794                 if (! $asap || ! $is_spam) {
 795                         foreach (array_diff(array_keys($pickups),
 796                                 array_keys($uris)) as $remove) {
 797                                 unset($pickups[$remove]);
 798                         }
 799                 }
 800                 unset($uris);
 801         }
 802
 803         // Return if ...
 804         if ($asap && $is_spam) return $progress;
 805
 806         // ----------------------------------------
 807         // Host: Uniqueness (uniq / non-uniq)
 808
 809         $hosts = array_unique($hosts);
 810
 811         if (isset($sum['uniqhost'])) $sum['uniqhost'] += count($hosts);
 812         if ((! $asap || ! $is_spam) && isset($method['non_uniqhost'])) {
 813                 $sum['non_uniqhost'] = $sum['quantity'] - $sum['uniqhost'];
 814                 if ($sum['non_uniqhost'] > $method['non_uniqhost']) {
 815                         $is_spam['non_uniqhost'] = TRUE;
 816                 }
 817         }
 818
 819         // Return if ...
 820         if ($asap && $is_spam) return $progress;
 821
 822         // ----------------------------------------
 823         // URI: Bad host (Separate good/bad hosts from $hosts)
 824
 825         if ((! $asap || ! $is_spam) && isset($method['badhost'])) {
 826                 $list    = get_blocklist('list');
 827                 $blocked = array_merge_leaves(
 828                         $blocked,
 829                         blocklist_distiller($hosts, array_keys($list), $asap),
 830                         FALSE
 831                 );
 832                 foreach($list as $key=>$type){
 833                         if (! $type) unset($blocked[$key]); // Ignore goodhost etc
 834                 }
 835                 unset($list);
 836                 if (! empty($blocked)) $is_spam['badhost'] = TRUE;
 837         }
 838
 839         // Return if ...
 840         //if ($asap && $is_spam) return $progress;
 841
 842         // ----------------------------------------
 843         // End
 844
 845         return $progress;
 846 }
 847
 848 // ---------------------
 849 // Reporting
 850
 851 // Summarize $progress (blocked only)
 852 function summarize_spam_progress($progress = array(), $blockedonly = FALSE)
 853 {
 854         if ($blockedonly) {
 855                 $tmp = array_keys($progress['is_spam']);
 856         } else {
 857                 $tmp = array();
 858                 $method = & $progress['method'];
 859                 if (isset($progress['sum'])) {
 860                         foreach ($progress['sum'] as $key => $value) {
 861                                 if (isset($method[$key]) && $value) {
 862                                         $tmp[] = $key . '(' . $value . ')';
 863                                 }
 864                         }
 865                 }
 866         }
 867
 868         return implode(', ', $tmp);
 869 }
 870
 871 function summarize_detail_badhost($progress = array())
 872 {
 873         if (! isset($progress['blocked']) || empty($progress['blocked'])) return '';
 874
 875         // Flat per group
 876         $blocked = array();
 877         foreach($progress['blocked'] as $list => $lvalue) {
 878                 foreach($lvalue as $group => $gvalue) {
 879                         $flat = implode(', ', array_flat_leaves($gvalue));
 880                         if ($flat === $group) {
 881                                 $blocked[$list][]       = $flat;
 882                         } else {
 883                                 $blocked[$list][$group] = $flat;
 884                         }
 885                 }
 886         }
 887
 888         // Shrink per list
 889         // From: 'A-1' => array('ie.to')
 890         // To:   'A-1' => 'ie.to'
 891         foreach($blocked as $list => $lvalue) {
 892                 if (is_array($lvalue) &&
 893                    count($lvalue) == 1 &&
 894                    is_numeric(key($lvalue))) {
 895                     $blocked[$list] = current($lvalue);
 896                 }
 897         }
 898
 899         return var_export_shrink($blocked, TRUE, TRUE);
 900 }
 901
 902 function summarize_detail_newtral($progress = array())
 903 {
 904         if (! isset($progress['hosts'])    ||
 905             ! is_array($progress['hosts']) ||
 906             empty($progress['hosts'])) return '';
 907
 908         // Generate a responsible $trie
 909         $trie = array();
 910         foreach($progress['hosts'] as $value) {
 911                 // 'A.foo.bar.example.com'
 912                 $resp = whois_responsibility($value);   // 'example.com'
 913                 if (empty($resp)) {
 914                         // One or more test, or do nothing here
 915                         $resp = strval($value);
 916                         $rest = '';
 917                 } else {
 918                         $rest = rtrim(substr($value, 0, - strlen($resp)), '.'); // 'A.foo.bar'
 919                 }
 920                 $trie = array_merge_leaves($trie, array($resp => array($rest => NULL)), FALSE);
 921         }
 922
 923         // Format: var_export_shrink() -like output
 924         $result = array();
 925         ksort_by_domain($trie);
 926         foreach(array_keys($trie) as $key) {
 927                 ksort_by_domain($trie[$key]);
 928                 if (count($trie[$key]) == 1 && key($trie[$key]) == '') {
 929                         // Just one 'responsibility.example.com'
 930                         $result[] = '  \'' . $key . '\',';
 931                 } else {
 932                         // One subdomain-or-host, or several ones
 933                         $subs = array();
 934                         foreach(array_keys($trie[$key]) as $sub) {
 935                                 if ($sub == '') {
 936                                         $subs[] = $key;                 // 'example.com'
 937                                 } else {
 938                                         $subs[] = $sub . '. ';  // 'A.foo.bar. '
 939                                 }
 940                         }
 941                         $result[] = '  \'' . $key . '\' => \'' . implode(', ', $subs) . '\',';
 942                 }
 943                 unset($trie[$key]);
 944         }
 945         return
 946                 'array (' . "\n" .
 947                         implode("\n", $result) . "\n" .
 948                 ')';
 949 }
 950
 951
 952 // Check responsibility-root of the FQDN
 953 // 'foo.bar.example.com'        => 'example.com'        (.com        has the last whois for it)
 954 // 'foo.bar.example.au'         => 'example.au'         (.au         has the last whois for it)
 955 // 'foo.bar.example.edu.au'     => 'example.edu.au'     (.edu.au     has the last whois for it)
 956 // 'foo.bar.example.act.edu.au' => 'example.act.edu.au' (.act.edu.au has the last whois for it)
 957 function whois_responsibility($fqdn = 'foo.bar.example.com', $parent = FALSE, $implicit = TRUE)
 958 {
 959         static $domain;
 960
 961         if ($fqdn === NULL) {
 962                 $domain = NULL; // Unset
 963                 return '';
 964         }
 965         if (! is_string($fqdn)) return '';
 966
 967         if (is_ip($fqdn)) return $fqdn;
 968
 969         if (! isset($domain)) {
 970                 $domain = array();
 971                 if (file_exists(DOMAIN_INI_FILE)) {
 972                         include(DOMAIN_INI_FILE);       // Set
 973                 }
 974         }
 975
 976         $result  = array();
 977         $dcursor = & $domain;
 978         $array   = array_reverse(explode('.', $fqdn));
 979         $i = 0;
 980         while(TRUE) {
 981                 if (! isset($array[$i])) break;
 982                 $acursor = $array[$i];
 983                 if (is_array($dcursor) && isset($dcursor[$acursor])) {
 984                         $result[] = & $array[$i];
 985                         $dcursor  = & $dcursor[$acursor];
 986                 } else {
 987                         if (! $parent && isset($acursor)) {
 988                                 $result[] = & $array[$i];       // Whois servers must know this subdomain
 989                         }
 990                         break;
 991                 }
 992                 ++$i;
 993         }
 994
 995         // Implicit responsibility: Top-Level-Domains must not be yours
 996         // 'bar.foo.something' => 'foo.something'
 997         if ($implicit && count($result) == 1 && count($array) > 1) {
 998                 $result[] = & $array[1];
 999         }
1000
1001         return $result ? implode('.', array_reverse($result)) : '';
1002 }
1003
1004
1005 // ---------------------
1006 // Exit
1007
1008 // Freeing memories
1009 function spam_dispose()
1010 {
1011         get_blocklist(NULL);
1012         whois_responsibility(NULL);
1013 }
1014
1015 // Common bahavior for blocking
1016 // NOTE: Call this function from various blocking feature, to disgueise the reason 'why blocked'
1017 function spam_exit($mode = '', $data = array())
1018 {
1019         $exit = TRUE;
1020
1021         switch ($mode) {
1022                 case '':
1023                         echo("\n");
1024                         break;
1025                 case 'dump':
1026                         echo('<pre>' . "\n");
1027                         echo htmlspecialchars(var_export($data, TRUE));
1028                         echo('</pre>' . "\n");
1029                         break;
1030         };
1031
1032         if ($exit) exit;        // Force exit
1033 }
1034
1035
1036 // ---------------------
1037 // Simple filtering
1038
1039 // TODO: Record them
1040 // Simple/fast spam filter ($target: 'a string' or an array())
1041 function pkwk_spamfilter($action, $page, $target = array('title' => ''), $method = array(), $exitmode = '')
1042 {
1043         $progress = check_uri_spam($target, $method);
1044
1045         if (empty($progress['is_spam'])) {
1046                 spam_dispose();
1047         } else {
1048
1049 // TODO: detect encoding from $target for mbstring functions
1050 //              $tmp = array();
1051 //              foreach(array_keys($target) as $key) {
1052 //                      $tmp[strings($key, 0, FALSE, TRUE)] = strings($target[$key], 0, FALSE, TRUE);   // Removing "\0" etc
1053 //              }
1054 //              $target = & $tmp;
1055
1056                 pkwk_spamnotify($action, $page, $target, $progress, $method);
1057                 spam_exit($exitmode, $progress);
1058         }
1059 }
1060
1061 // ---------------------
1062 // PukiWiki original
1063
1064 // Mail to administrator(s)
1065 function pkwk_spamnotify($action, $page, $target = array('title' => ''), $progress = array(), $method = array())
1066 {
1067         global $notify, $notify_subject;
1068
1069         if (! $notify) return;
1070
1071         $asap = isset($method['asap']);
1072
1073         $summary['ACTION']  = 'Blocked by: ' . summarize_spam_progress($progress, TRUE);
1074         if (! $asap) {
1075                 $summary['METRICS'] = summarize_spam_progress($progress);
1076         }
1077
1078         $tmp = summarize_detail_badhost($progress);
1079         if ($tmp != '') $summary['DETAIL_BADHOST'] = $tmp;
1080
1081         $tmp = summarize_detail_newtral($progress);
1082         if (! $asap && $tmp != '') $summary['DETAIL_NEUTRAL_HOST'] = $tmp;
1083
1084         $summary['COMMENT'] = $action;
1085         $summary['PAGE']    = '[blocked] ' . (is_pagename($page) ? $page : '');
1086         $summary['URI']     = get_script_uri() . '?' . rawurlencode($page);
1087         $summary['USER_AGENT']  = TRUE;
1088         $summary['REMOTE_ADDR'] = TRUE;
1089         pkwk_mail_notify($notify_subject,  var_export($target, TRUE), $summary, TRUE);
1090 }
1091
1092 ?>