external link icons removed, personal toolbar cleanup, relative paths
[lhc/web/wiklou.git] / includes / Block.php
index 2aeff38..d98b07b 100644 (file)
@@ -1,4 +1,4 @@
-<?
+<?php
 # Blocks and bans object
 #
 #TODO: This could be used everywhere, but it isn't.
@@ -8,12 +8,15 @@
 #
 # To use delete(), you only need to fill $mAddress
 
+# Globals used: $wgBlockCache, $wgAutoblockExpiry
+
 class Block
 {
-       var $mAddress, $mUser, $mBy, $mReason, $mTimestamp, $mAuto, $mId;
+       /* public*/ var $mAddress, $mUser, $mBy, $mReason, $mTimestamp, $mAuto, $mId, $mExpiry;
+       /* private */ var $mNetworkBits, $mIntegerAddr;
        
        function Block( $address = "", $user = "", $by = 0, $reason = "", 
-               $timestamp = "" , $auto = 0) 
+               $timestamp = "" , $auto = 0, $expiry = "" 
        {
                $this->mAddress = $address;
                $this->mUser = $user;
@@ -21,21 +24,27 @@ class Block
                $this->mReason = $reason;
                $this->mTimestamp = $timestamp;
                $this->mAuto = $auto;
-       }
+               $this->mExpiry = $expiry;
 
-       /*static*/ function newFromDB( $address, $user = 0, $killExpired = true ) {
+               $this->initialiseRange();
+       }
+       
+       /*static*/ function newFromDB( $address, $user = 0, $killExpired = true ) 
+       {
                $ban = new Block();
                $ban->load( $address, $user, $killExpired );
                return $ban;
        }
-               
-       function clear() {
+       
+       function clear() 
+       {
                $mAddress = $mReason = $mTimestamp = "";
                $mUser = $mBy = 0;
        }
 
        # Get a ban from the DB, with either the given address or the given username
-       function load( $address, $user = 0, $killExpired = true ) {
+       function load( $address, $user = 0, $killExpired = true ) 
+       {
                $fname = "Block::load";
                $ret = false;
                $killed = false;
@@ -47,7 +56,7 @@ class Block
                                "' OR ipb_user={$user})";
                }
 
-               $res = wfQuery( $sql, $fname );
+               $res = wfQuery( $sql, DB_READ, $fname );
                if ( 0 == wfNumRows( $res ) ) {
                        # User is not blocked
                        $this->clear();
@@ -83,7 +92,8 @@ class Block
                return $ret;
        }
        
-       function initFromRow( $row ) {
+       function initFromRow( $row ) 
+       {
                $this->mAddress = $row->ipb_address;
                $this->mReason = $row->ipb_reason;
                $this->mTimestamp = $row->ipb_timestamp;
@@ -91,12 +101,32 @@ class Block
                $this->mBy = $row->ipb_by;
                $this->mAuto = $row->ipb_auto;
                $this->mId = $row->ipb_id;
+               $this->mExpiry = $row->ipb_expiry;
+
+               $this->initialiseRange();
        }       
 
+       function initialiseRange()
+       {
+               if ( $this->mUser == 0 ) {
+                       $rangeParts = explode( "/", $this->mAddress );
+                       if ( count( $rangeParts ) == 2 ) {
+                               $this->mNetworkBits = $rangeParts[1];
+                       } else {
+                               $this->mNetworkBits = 32;
+                       }
+                       $this->mIntegerAddr = ip2long( $rangeParts[0] );
+               } else {
+                       $this->mNetworkBits = false;
+                       $this->mIntegerAddr = false;
+               }
+       }
+       
        # Callback with a Block object for every block
-       /*static*/ function enumBlocks( $callback, $tag, $killExpired = true ) {
-               $sql = "SELECT * FROM ipblocks ORDER BY ipb_timestamp";
-               $res = wfQuery( $sql, "Block::enumBans" );
+       /*static*/ function enumBlocks( $callback, $tag, $killExpired = true ) 
+       {
+               $sql = "SELECT * FROM ipblocks ORDER BY ipb_timestamp DESC";
+               $res = wfQuery( $sql, DB_READ, "Block::enumBans" );
                $block = new Block();
 
                while ( $row = wfFetchObject( $res ) ) {
@@ -112,7 +142,8 @@ class Block
                wfFreeResult( $res );
        }
 
-       function delete() {
+       function delete() 
+       {
                $fname = "Block::delete";
                if ( $this->mAddress == "" ) {
                        $sql = "DELETE FROM ipblocks WHERE ipb_id={$this->mId}";
@@ -120,18 +151,24 @@ class Block
                        $sql = "DELETE FROM ipblocks WHERE ipb_address='" .
                                wfStrencode( $this->mAddress ) . "'";
                }
-               wfQuery( $sql, "Block::delete" );
+               wfQuery( $sql, DB_WRITE, "Block::delete" );
+
+               $this->clearCache();
        }
 
-       function insert() {
+       function insert() 
+       {
                $sql = "INSERT INTO ipblocks 
-                 (ipb_address, ipb_user, ipb_by, ipb_reason, ipb_timestamp, ipb_auto ) 
+                 (ipb_address, ipb_user, ipb_by, ipb_reason, ipb_timestamp, ipb_auto, ipb_expiry ) 
                  VALUES ('" . wfStrencode( $this->mAddress ) . "', {$this->mUser}, {$this->mBy}, '" . 
-                 wfStrencode( $this->mReason ) . "','{$this->mTimestamp}', {$this->mAuto})";
-               wfQuery( $sql, "Block::insert" );
+                 wfStrencode( $this->mReason ) . "','{$this->mTimestamp}', {$this->mAuto}, '{$this->mExpiry}')";
+               wfQuery( $sql, DB_WRITE, "Block::insert" );
+
+               $this->clearCache();
        }
 
-       function deleteIfExpired() {
+       function deleteIfExpired() 
+       {
                if ( $this->isExpired() ) {
                        $this->delete();
                        return true;
@@ -140,31 +177,70 @@ class Block
                }
        }
 
-       function isExpired() {
-               global $wgIPBlockExpiration, $wgUserBlockExpiration;
-               
-               $period = $this->mUser ? $wgUserBlockExpiration : $wgIPBlockExpiration;
-               
-               # Period==0 means no expiry
-               if ( !$period ) { 
+       function isExpired() 
+       {       
+               if ( !$this->mExpiry ) {
                        return false;
-               }
-               $expiry = wfTimestamp2Unix( $this->mTimestamp ) + $period;
-               $now = wfTimestamp2Unix( wfTimestampNow() );
-               if ( $now > $expiry ) {
-                       return true;
                } else {
-                       return false;
+                       return wfTimestampNow() > $this->mExpiry;
                }
        }
 
-       function isValid() {
+       function isValid() 
+       {
                return $this->mAddress != "";
        }
        
-       function updateTimestamp() {
-               wfQuery( "UPDATE ipblocks SET ipb_timestamp='" . wfTimestampNow() . 
-                       "' WHERE ipb_address='" . wfStrencode( $this->mAddress ) . "'", "Block::updateTimestamp" );
+       function updateTimestamp() 
+       {
+               if ( $this->mAuto ) {
+                       $this->mTimestamp = wfTimestampNow();
+                       $this->mExpiry = Block::getAutoblockExpiry( $this->mTimestamp );
+
+                       wfQuery( "UPDATE ipblocks SET " .
+                               "ipb_timestamp='" . $this->mTimestamp . "', " .
+                               "ipb_expiry='" . $this->mExpiry . "' " .
+                               "WHERE ipb_address='" . wfStrencode( $this->mAddress ) . "'", DB_WRITE, "Block::updateTimestamp" );
+                       
+                       $this->clearCache();
+               }
+       }
+
+       /* private */ function clearCache()
+       {
+               global $wgBlockCache;
+               if ( is_object( $wgBlockCache ) ) {
+                       $wgBlockCache->clear();
+               }
+       }
+       
+       function getIntegerAddr()
+       {
+               return $this->mIntegerAddr;
+       }
+       
+       function getNetworkBits()
+       {
+               return $this->mNetworkBits;
+       }
+
+       /* static */ function getAutoblockExpiry( $timestamp )
+       {
+               global $wgAutoblockExpiry;
+               return wfUnix2Timestamp( wfTimestamp2Unix( $timestamp ) + $wgAutoblockExpiry );
+       }
+
+       /* static */ function normaliseRange( $range )
+       {
+               $parts = explode( "/", $range );
+               if ( count( $parts ) == 2 ) {
+                       $shift = 32 - $parts[1];
+                       $ipint = ip2long( $parts[0] );
+                       $ipint = $ipint >> $shift << $shift;
+                       $newip = long2ip( $ipint );
+                       $range = "$newip/{$parts[1]}";
+               }
+               return $range;
        }
 
 }