]> git.openstreetmap.org Git - nominatim.git/blob - lib/init-website.php
add an option for max number of sleeping threads
[nominatim.git] / lib / init-website.php
1 <?php
2         require_once('init.php');
3
4         if (CONST_ClosedForIndexing && strpos(CONST_ClosedForIndexingExceptionIPs, ','.$_SERVER["REMOTE_ADDR"].',') === false)
5         {
6                 echo "Closed for re-indexing...";
7                 exit;
8         }
9
10         $aBucketKeys = array();
11
12         if (isset($_SERVER["HTTP_REFERER"])) $aBucketKeys[] = str_replace('www.','',strtolower(parse_url($_SERVER["HTTP_REFERER"], PHP_URL_HOST)));
13         if (isset($_SERVER["REMOTE_ADDR"])) $aBucketKeys[] = $_SERVER["REMOTE_ADDR"];
14         if (isset($_GET["email"])) $aBucketKeys[] = $_GET["email"];
15
16         $fBucketVal = doBucket($aBucketKeys, 
17                         (defined('CONST_ConnectionBucket_PageType')?constant('CONST_ConnectionBucket_Cost_'.CONST_ConnectionBucket_PageType):1) + user_busy_cost(),
18                         CONST_ConnectionBucket_LeakRate, CONST_ConnectionBucket_BlockLimit);
19
20         if ($fBucketVal > CONST_ConnectionBucket_WaitLimit && $fBucketVal < CONST_ConnectionBucket_BlockLimit)
21         {
22                 $m = getBucketMemcache();
23                 $iCurrentSleeping = $m->increment('sleepCounter');
24                 if (false === $iCurrentSleeping)
25                 {
26                         $m->add('sleepCounter', 0);
27                         $iCurrentSleeping = $m->increment('sleepCounter');
28                 }
29                 if ($iCurrentSleeping >= CONST_ConnectionBucket_MaxSleeping)
30                 {
31                         // Too many threads sleeping already.  This becomes a hard block.
32                         $fBucketVal = doBucket($aBucketKeys, CONST_ConnectionBucket_BlockLimit, CONST_ConnectionBucket_LeakRate, CONST_ConnectionBucket_BlockLimit);
33                 }
34                 else
35                 {
36                         sleep(($fBucketVal - CONST_ConnectionBucket_WaitLimit)/CONST_ConnectionBucket_LeakRate);
37                         $fBucketVal = doBucket($aBucketKeys, CONST_ConnectionBucket_LeakRate, CONST_ConnectionBucket_LeakRate, CONST_ConnectionBucket_BlockLimit);
38                 }
39                 $m->decrement('sleepCounter');
40         }
41
42         if (strpos(CONST_BlockedIPs, ','.$_SERVER["REMOTE_ADDR"].',') !== false || $fBucketVal >= CONST_ConnectionBucket_BlockLimit)
43         {
44                 echo "Your IP has been blocked. \n";
45                 echo "Please create a nominatim trac ticket (http://trac.openstreetmap.org/newticket?component=nominatim) to request this to be removed. \n";
46                 echo "Information on the Nominatim usage policy can be found here: http://wiki.openstreetmap.org/wiki/Nominatim#Usage_Policy \n";
47                 exit;
48         }
49
50         header('Content-type: text/html; charset=utf-8');