Alrighty, now we properly remove old blocks before inserting the new one. (Bug 10080...
[lhc/web/wiklou.git] / includes / Block.php
index 26fa444..2a1ad69 100644 (file)
@@ -1,7 +1,7 @@
 <?php
 /**
+ * @file
  * Blocks and bans object
- * @package MediaWiki
  */
 
 /**
@@ -9,56 +9,77 @@
  * All the functions in this class assume the object is either explicitly
  * loaded or filled. It is not load-on-demand. There are no accessors.
  *
- * To use delete(), you only need to fill $mAddress
  * Globals used: $wgAutoblockExpiry, $wgAntiLockFlags
  *
  * @todo This could be used everywhere, but it isn't.
- * @package MediaWiki
  */
 class Block
 {
        /* public*/ var $mAddress, $mUser, $mBy, $mReason, $mTimestamp, $mAuto, $mId, $mExpiry,
-                           $mRangeStart, $mRangeEnd;
-       /* private */ var $mNetworkBits, $mIntegerAddr, $mForUpdate, $mFromMaster, $mByName;
-       
+                               $mRangeStart, $mRangeEnd, $mAnonOnly, $mEnableAutoblock, $mHideName,
+                               $mBlockEmail, $mByName;
+       /* private */ var $mNetworkBits, $mIntegerAddr, $mForUpdate, $mFromMaster;
+
        const EB_KEEP_EXPIRED = 1;
        const EB_FOR_UPDATE = 2;
        const EB_RANGE_ONLY = 4;
 
-       function Block( $address = '', $user = '', $by = 0, $reason = '',
-               $timestamp = '' , $auto = 0, $expiry = '' )
+       function __construct( $address = '', $user = 0, $by = 0, $reason = '',
+               $timestamp = '' , $auto = 0, $expiry = '', $anonOnly = 0, $createAccount = 0, $enableAutoblock = 0,
+               $hideName = 0, $blockEmail = 0 )
        {
+               $this->mId = 0;
+               # Expand valid IPv6 addresses
+               $address = IP::sanitizeIP( $address );
                $this->mAddress = $address;
                $this->mUser = $user;
                $this->mBy = $by;
                $this->mReason = $reason;
                $this->mTimestamp = wfTimestamp(TS_MW,$timestamp);
                $this->mAuto = $auto;
-               if( empty( $expiry ) ) {
-                       $this->mExpiry = $expiry;
-               } else {
-                       $this->mExpiry = wfTimestamp( TS_MW, $expiry );
-               }
-
+               $this->mAnonOnly = $anonOnly;
+               $this->mCreateAccount = $createAccount;
+               $this->mExpiry = self::decodeExpiry( $expiry );
+               $this->mEnableAutoblock = $enableAutoblock;
+               $this->mHideName = $hideName;
+               $this->mBlockEmail = $blockEmail;
                $this->mForUpdate = false;
                $this->mFromMaster = false;
                $this->mByName = false;
                $this->initialiseRange();
        }
 
-       /*static*/ function newFromDB( $address, $user = 0, $killExpired = true )
+       static function newFromDB( $address, $user = 0, $killExpired = true )
        {
-               $ban = new Block();
-               $ban->load( $address, $user, $killExpired );
-               return $ban;
+               $block = new Block();
+               $block->load( $address, $user, $killExpired );
+               if ( $block->isValid() ) {
+                       return $block;
+               } else {
+                       return null;
+               }
+       }
+
+       static function newFromID( $id )
+       {
+               $dbr = wfGetDB( DB_SLAVE );
+               $res = $dbr->resultObject( $dbr->select( 'ipblocks', '*',
+                       array( 'ipb_id' => $id ), __METHOD__ ) );
+               $block = new Block;
+               if ( $block->loadFromResult( $res ) ) {
+                       return $block;
+               } else {
+                       return null;
+               }
        }
 
        function clear()
        {
                $this->mAddress = $this->mReason = $this->mTimestamp = '';
-               $this->mUser = $this->mBy = 0;
+               $this->mId = $this->mAnonOnly = $this->mCreateAccount =
+                       $this->mEnableAutoblock = $this->mAuto = $this->mUser =
+                       $this->mBy = $this->mHideName = $this->mBlockEmail = 0;
                $this->mByName = false;
-
        }
 
        /**
@@ -68,58 +89,103 @@ class Block
        {
                global $wgAntiLockFlags;
                if ( $this->mForUpdate || $this->mFromMaster ) {
-                       $db =& wfGetDB( DB_MASTER );
+                       $db = wfGetDB( DB_MASTER );
                        if ( !$this->mForUpdate || ($wgAntiLockFlags & ALF_NO_BLOCK_LOCK) ) {
-                               $options = '';
+                               $options = array();
                        } else {
-                               $options = 'FOR UPDATE';
+                               $options = array( 'FOR UPDATE' );
                        }
                } else {
-                       $db =& wfGetDB( DB_SLAVE );
-                       $options = '';
+                       $db = wfGetDB( DB_SLAVE );
+                       $options = array();
                }
                return $db;
        }
 
        /**
         * Get a ban from the DB, with either the given address or the given username
+        *
+        * @param string $address The IP address of the user, or blank to skip IP blocks
+        * @param integer $user The user ID, or zero for anonymous users
+        * @param bool $killExpired Whether to delete expired rows while loading
+        *
         */
        function load( $address = '', $user = 0, $killExpired = true )
        {
-               $fname = 'Block::load';
                wfDebug( "Block::load: '$address', '$user', $killExpired\n" );
 
-               $options = '';
+               $options = array();
                $db =& $this->getDBOptions( $options );
 
-               $ret = false;
-               $killed = false;
-               $ipblocks = $db->tableName( 'ipblocks' );
-
                if ( 0 == $user && $address == '' ) {
                        # Invalid user specification, not blocked
                        $this->clear();
                        return false;
-               } elseif ( $address == '' ) {
-                       $sql = "SELECT * FROM $ipblocks WHERE ipb_user={$user} $options";
-               } elseif ( $user == '' ) {
-                       $sql = "SELECT * FROM $ipblocks WHERE ipb_address=" . $db->addQuotes( $address ) . " $options";
-               } elseif ( $options == '' ) {
-                       # If there are no options (e.g. FOR UPDATE), use a UNION
-                       # so that the query can make efficient use of indices
-                       $sql = "SELECT * FROM $ipblocks WHERE ipb_address='" . $db->strencode( $address ) .
-                               "' UNION SELECT * FROM $ipblocks WHERE ipb_user={$user}";
-               } else {
-                       # If there are options, a UNION can not be used, use one
-                       # SELECT instead. Will do a full table scan.
-                       $sql = "SELECT * FROM $ipblocks WHERE (ipb_address='" . $db->strencode( $address ) .
-                               "' OR ipb_user={$user}) $options";
                }
 
-               $res = $db->query( $sql, $fname );
-               if ( 0 != $db->numRows( $res ) ) {
+               # Try user block
+               if ( $user ) {
+                       $res = $db->resultObject( $db->select( 'ipblocks', '*', array( 'ipb_user' => $user ),
+                               __METHOD__, $options ) );
+                       if ( $this->loadFromResult( $res, $killExpired ) ) {
+                               return true;
+                       }
+               }
+
+               # Try IP block
+               # TODO: improve performance by merging this query with the autoblock one
+               # Slightly tricky while handling killExpired as well
+               if ( $address ) {
+                       $conds = array( 'ipb_address' => $address, 'ipb_auto' => 0 );
+                       $res = $db->resultObject( $db->select( 'ipblocks', '*', $conds, __METHOD__, $options ) );
+                       if ( $this->loadFromResult( $res, $killExpired ) ) {
+                               if ( $user && $this->mAnonOnly ) {
+                                       # Block is marked anon-only
+                                       # Whitelist this IP address against autoblocks and range blocks
+                                       $this->clear();
+                                       return false;
+                               } else {
+                                       return true;
+                               }
+                       }
+               }
+
+               # Try range block
+               if ( $this->loadRange( $address, $killExpired, $user ) ) {
+                       if ( $user && $this->mAnonOnly ) {
+                               $this->clear();
+                               return false;
+                       } else {
+                               return true;
+                       }
+               }
+
+               # Try autoblock
+               if ( $address ) {
+                       $conds = array( 'ipb_address' => $address, 'ipb_auto' => 1 );
+                       if ( $user ) {
+                               $conds['ipb_anon_only'] = 0;
+                       }
+                       $res = $db->resultObject( $db->select( 'ipblocks', '*', $conds, __METHOD__, $options ) );
+                       if ( $this->loadFromResult( $res, $killExpired ) ) {
+                               return true;
+                       }
+               }
+
+               # Give up
+               $this->clear();
+               return false;
+       }
+
+       /**
+        * Fill in member variables from a result wrapper
+        */
+       function loadFromResult( ResultWrapper $res, $killExpired = true )
+       {
+               $ret = false;
+               if ( 0 != $res->numRows() ) {
                        # Get first block
-                       $row = $db->fetchObject( $res );
+                       $row = $res->fetchObject();
                        $this->initFromRow( $row );
 
                        if ( $killExpired ) {
@@ -127,7 +193,7 @@ class Block
                                do {
                                        $killed = $this->deleteIfExpired();
                                        if ( $killed ) {
-                                               $row = $db->fetchObject( $res );
+                                               $row = $res->fetchObject();
                                                if ( $row ) {
                                                        $this->initFromRow( $row );
                                                }
@@ -135,26 +201,14 @@ class Block
                                } while ( $killed && $row );
 
                                # If there were any left after the killing finished, return true
-                               if ( !$row ) {
-                                       $ret = false;
-                                       $this->clear();
-                               } else {
+                               if ( $row ) {
                                        $ret = true;
                                }
                        } else {
                                $ret = true;
                        }
                }
-               $db->freeResult( $res );
-
-               # No blocks found yet? Try looking for range blocks
-               if ( !$ret && $address != '' ) {
-                       $ret = $this->loadRange( $address, $killExpired );
-               }
-               if ( !$ret ) {
-                       $this->clear();
-               }
-
+               $res->free();
                return $ret;
        }
 
@@ -162,11 +216,9 @@ class Block
         * Search the database for any range blocks matching the given address, and
         * load the row if one is found.
         */
-       function loadRange( $address, $killExpired = true )
+       function loadRange( $address, $killExpired = true, $user = 0 )
        {
-               $fname = 'Block::loadRange';
-
-               $iaddr = wfIP2Hex( $address );
+               $iaddr = IP::toHex( $address );
                if ( $iaddr === false ) {
                        # Invalid address
                        return false;
@@ -176,40 +228,29 @@ class Block
                # Blocks should not cross a /16 boundary.
                $range = substr( $iaddr, 0, 4 );
 
-               $options = '';
+               $options = array();
                $db =& $this->getDBOptions( $options );
-               $ipblocks = $db->tableName( 'ipblocks' );
-               $sql = "SELECT * FROM $ipblocks WHERE ipb_range_start LIKE '$range%' ".
-                       "AND ipb_range_start <= '$iaddr' AND ipb_range_end >= '$iaddr' $options";
-               $res = $db->query( $sql, $fname );
-               $row = $db->fetchObject( $res );
-
-               $success = false;
-               if ( $row ) {
-                       # Found a row, initialise this object
-                       $this->initFromRow( $row );
+               $conds = array(
+                       "ipb_range_start LIKE '$range%'",
+                       "ipb_range_start <= '$iaddr'",
+                       "ipb_range_end >= '$iaddr'"
+               );
 
-                       # Is it expired?
-                       if ( !$killExpired || !$this->deleteIfExpired() ) {
-                               # No, return true
-                               $success = true;
-                       }
+               if ( $user ) {
+                       $conds['ipb_anon_only'] = 0;
                }
 
-               $db->freeResult( $res );
+               $res = $db->resultObject( $db->select( 'ipblocks', '*', $conds, __METHOD__, $options ) );
+               $success = $this->loadFromResult( $res, $killExpired );
                return $success;
        }
 
        /**
         * Determine if a given integer IPv4 address is in a given CIDR network
+        * @deprecated Use IP::isInRange
         */
        function isAddressInRange( $addr, $range ) {
-               list( $network, $bits ) = wfParseCIDR( $range );
-               if ( $network !== false && $addr >> ( 32 - $bits ) == $network >> ( 32 - $bits ) ) {
-                       return true;
-               } else {
-                       return false;
-               }
+               return IP::isInRange( $addr, $range );
        }
 
        function initFromRow( $row )
@@ -220,14 +261,17 @@ class Block
                $this->mUser = $row->ipb_user;
                $this->mBy = $row->ipb_by;
                $this->mAuto = $row->ipb_auto;
+               $this->mAnonOnly = $row->ipb_anon_only;
+               $this->mCreateAccount = $row->ipb_create_account;
+               $this->mEnableAutoblock = $row->ipb_enable_autoblock;
+               $this->mBlockEmail = $row->ipb_block_email;
+               $this->mHideName = $row->ipb_deleted;
                $this->mId = $row->ipb_id;
-               $this->mExpiry = $row->ipb_expiry ?
-                       wfTimestamp(TS_MW,$row->ipb_expiry) :
-                       $row->ipb_expiry;
+               $this->mExpiry = self::decodeExpiry( $row->ipb_expiry );
                if ( isset( $row->user_name ) ) {
                        $this->mByName = $row->user_name;
                } else {
-                       $this->mByName = false;
+                       $this->mByName = $row->ipb_by_text;
                }
                $this->mRangeStart = $row->ipb_range_start;
                $this->mRangeEnd = $row->ipb_range_end;
@@ -237,12 +281,9 @@ class Block
        {
                $this->mRangeStart = '';
                $this->mRangeEnd = '';
+
                if ( $this->mUser == 0 ) {
-                       list( $network, $bits ) = wfParseCIDR( $this->mAddress );
-                       if ( $network !== false ) {
-                               $this->mRangeStart = sprintf( '%08X', $network );
-                               $this->mRangeEnd = sprintf( '%08X', $network + (1 << (32 - $bits)) - 1 );
-                       }
+                       list( $this->mRangeStart, $this->mRangeEnd ) = IP::parseRange( $this->mAddress );
                }
        }
 
@@ -256,7 +297,7 @@ class Block
 
                $block = new Block();
                if ( $flags & Block::EB_FOR_UPDATE ) {
-                       $db =& wfGetDB( DB_MASTER );
+                       $db = wfGetDB( DB_MASTER );
                        if ( $wgAntiLockFlags & ALF_NO_BLOCK_LOCK ) {
                                $options = '';
                        } else {
@@ -264,7 +305,7 @@ class Block
                        }
                        $block->forUpdate( true );
                } else {
-                       $db =& wfGetDB( DB_SLAVE );
+                       $db = wfGetDB( DB_SLAVE );
                        $options = '';
                }
                if ( $flags & Block::EB_RANGE_ONLY ) {
@@ -275,7 +316,7 @@ class Block
 
                $now = wfTimestampNow();
 
-               extract( $db->tableNames( 'ipblocks', 'user' ) );
+               list( $ipblocks, $user ) = $db->tableNamesN( 'ipblocks', 'user' );
 
                $sql = "SELECT $ipblocks.*,user_name FROM $ipblocks,$user " .
                        "WHERE user_id=ipb_by $cond ORDER BY ipb_timestamp DESC $options";
@@ -298,30 +339,56 @@ class Block
                                call_user_func( $callback, $block, $tag );
                        }
                }
-               wfFreeResult( $res );
+               $db->freeResult( $res );
                return $num_rows;
        }
 
        function delete()
        {
-               $fname = 'Block::delete';
                if (wfReadOnly()) {
-                       return;
+                       return false;
                }
-               $dbw =& wfGetDB( DB_MASTER );
-
-               if ( $this->mAddress == '' ) {
-                       $condition = array( 'ipb_id' => $this->mId );
-               } else {
-                       $condition = array( 'ipb_address' => $this->mAddress );
+               if ( !$this->mId ) {
+                       throw new MWException( "Block::delete() now requires that the mId member be filled\n" );
                }
-               return( $dbw->delete( 'ipblocks', $condition, $fname ) > 0 ? true : false );
+
+               $dbw = wfGetDB( DB_MASTER );
+               $dbw->delete( 'ipblocks', array( 'ipb_id' => $this->mId ), __METHOD__ );
+               return $dbw->affectedRows() > 0;
        }
 
+       /**
+       * Insert a block into the block table.
+       * @return Whether or not the insertion was successful.
+       */
        function insert()
        {
                wfDebug( "Block::insert; timestamp {$this->mTimestamp}\n" );
-               $dbw =& wfGetDB( DB_MASTER );
+               $dbw = wfGetDB( DB_MASTER );
+
+               # Unset ipb_anon_only for user blocks, makes no sense
+               if ( $this->mUser ) {
+                       $this->mAnonOnly = 0;
+               }
+
+               # Unset ipb_enable_autoblock for IP blocks, makes no sense
+               if ( !$this->mUser ) {
+                       $this->mEnableAutoblock = 0;
+                       $this->mBlockEmail = 0; //Same goes for email...
+               }
+
+               if( !$this->mByName ) {
+                       if( $this->mBy ) {
+                               $this->mByName = User::whoIs( $this->mBy );
+                       } else {
+                               global $wgUser;
+                               $this->mByName = $wgUser->getName();
+                       }
+               }
+
+               # Don't collide with expired blocks
+               Block::purgeExpired();
+
                $ipb_id = $dbw->nextSequenceValue('ipblocks_ipb_id_val');
                $dbw->insert( 'ipblocks',
                        array(
@@ -329,16 +396,144 @@ class Block
                                'ipb_address' => $this->mAddress,
                                'ipb_user' => $this->mUser,
                                'ipb_by' => $this->mBy,
+                               'ipb_by_text' => $this->mByName,
                                'ipb_reason' => $this->mReason,
                                'ipb_timestamp' => $dbw->timestamp($this->mTimestamp),
                                'ipb_auto' => $this->mAuto,
-                               'ipb_expiry' => $this->mExpiry ?
-                                       $dbw->timestamp($this->mExpiry) :
-                                       $this->mExpiry,
+                               'ipb_anon_only' => $this->mAnonOnly,
+                               'ipb_create_account' => $this->mCreateAccount,
+                               'ipb_enable_autoblock' => $this->mEnableAutoblock,
+                               'ipb_expiry' => self::encodeExpiry( $this->mExpiry, $dbw ),
                                'ipb_range_start' => $this->mRangeStart,
                                'ipb_range_end' => $this->mRangeEnd,
-                       ), 'Block::insert'
+                               'ipb_deleted'   => $this->mHideName,
+                               'ipb_block_email' => $this->mBlockEmail
+                       ), 'Block::insert', array( 'IGNORE' )
                );
+               $affected = $dbw->affectedRows();
+
+               if ($affected)
+                       $this->doRetroactiveAutoblock();
+
+               return $affected;
+       }
+
+       /**
+       * Retroactively autoblocks the last IP used by the user (if it is a user)
+       * blocked by this Block.
+       *@return Whether or not a retroactive autoblock was made.
+       */
+       function doRetroactiveAutoblock() {
+               $dbr = wfGetDB( DB_SLAVE );
+               #If autoblock is enabled, autoblock the LAST IP used
+               # - stolen shamelessly from CheckUser_body.php
+
+               if ($this->mEnableAutoblock && $this->mUser) {
+                       wfDebug("Doing retroactive autoblocks for " . $this->mAddress . "\n");
+
+                       $row = $dbr->selectRow( 'recentchanges', array( 'rc_ip' ), array( 'rc_user_text' => $this->mAddress ),
+                               __METHOD__ , array( 'ORDER BY' => 'rc_timestamp DESC' ) );
+
+                       if ( !$row || !$row->rc_ip ) {
+                               #No results, don't autoblock anything
+                               wfDebug("No IP found to retroactively autoblock\n");
+                       } else {
+                               #Limit is 1, so no loop needed.
+                               $retroblockip = $row->rc_ip;
+                               return $this->doAutoblock( $retroblockip, true );
+                       }
+               }
+       }
+
+       /**
+       * Autoblocks the given IP, referring to this Block.
+       * @param string $autoblockip The IP to autoblock.
+       * @param bool $justInserted The main block was just inserted
+       * @return bool Whether or not an autoblock was inserted.
+       */
+       function doAutoblock( $autoblockip, $justInserted = false ) {
+               # If autoblocks are disabled, go away.
+               if ( !$this->mEnableAutoblock ) {
+                       return;
+               }
+
+               # Check for presence on the autoblock whitelist
+               # TODO cache this?
+               $lines = explode( "\n", wfMsgForContentNoTrans( 'autoblock_whitelist' ) );
+
+               $ip = $autoblockip;
+
+               wfDebug("Checking the autoblock whitelist..\n");
+
+               foreach( $lines as $line ) {
+                       # List items only
+                       if ( substr( $line, 0, 1 ) !== '*' ) {
+                               continue;
+                       }
+
+                       $wlEntry = substr($line, 1);
+                       $wlEntry = trim($wlEntry);
+
+                       wfDebug("Checking $ip against $wlEntry...");
+
+                       # Is the IP in this range?
+                       if (IP::isInRange( $ip, $wlEntry )) {
+                               wfDebug(" IP $ip matches $wlEntry, not autoblocking\n");
+                               #$autoblockip = null; # Don't autoblock a whitelisted IP.
+                               return; #This /SHOULD/ introduce a dummy block - but
+                                       # I don't know a safe way to do so. -werdna
+                       } else {
+                               wfDebug( " No match\n" );
+                       }
+               }
+               
+               ## Allow hooks to cancel the autoblock.
+               if (!wfRunHooks( 'AbortAutoblock', array( $autoblockip, &$this ) )) {
+                       wfDebug( "Autoblock aborted by hook." );
+                       return false;
+               }
+
+               # It's okay to autoblock. Go ahead and create/insert the block.
+
+               $ipblock = Block::newFromDB( $autoblockip );
+               if ( $ipblock ) {
+                       # If the user is already blocked. Then check if the autoblock would
+                       # exceed the user block. If it would exceed, then do nothing, else
+                       # prolong block time
+                       if ($this->mExpiry &&
+                       ($this->mExpiry < Block::getAutoblockExpiry($ipblock->mTimestamp))) {
+                               return;
+                       }
+                       # Just update the timestamp
+                       if ( !$justInserted ) {
+                               $ipblock->updateTimestamp();
+                       }
+                       return;
+               } else {
+                       $ipblock = new Block;
+               }
+
+               # Make a new block object with the desired properties
+               wfDebug( "Autoblocking {$this->mAddress}@" . $autoblockip . "\n" );
+               $ipblock->mAddress = $autoblockip;
+               $ipblock->mUser = 0;
+               $ipblock->mBy = $this->mBy;
+               $ipblock->mReason = wfMsgForContent( 'autoblocker', $this->mAddress, $this->mReason );
+               $ipblock->mTimestamp = wfTimestampNow();
+               $ipblock->mAuto = 1;
+               $ipblock->mCreateAccount = $this->mCreateAccount;
+               # Continue suppressing the name if needed
+               $ipblock->mHideName = $this->mHideName;
+
+               # If the user is already blocked with an expiry date, we don't
+               # want to pile on top of that!
+               if($this->mExpiry) {
+                       $ipblock->mExpiry = min ( $this->mExpiry, Block::getAutoblockExpiry( $this->mTimestamp ));
+               } else {
+                       $ipblock->mExpiry = Block::getAutoblockExpiry( $this->mTimestamp );
+               }
+               # Insert it
+               return $ipblock->insert();
        }
 
        function deleteIfExpired()
@@ -378,7 +573,7 @@ class Block
                        $this->mTimestamp = wfTimestamp();
                        $this->mExpiry = Block::getAutoblockExpiry( $this->mTimestamp );
 
-                       $dbw =& wfGetDB( DB_MASTER );
+                       $dbw = wfGetDB( DB_MASTER );
                        $dbw->update( 'ipblocks',
                                array( /* SET */
                                        'ipb_timestamp' => $dbw->timestamp($this->mTimestamp),
@@ -401,11 +596,18 @@ class Block
                return $this->mNetworkBits;
        }*/
 
+       /**
+        * @return The blocker user ID.
+        */
+       public function getBy() {
+               return $this->mBy;
+       }
+
+       /**
+        * @return The blocker user name.
+        */
        function getByName()
        {
-               if ( $this->mByName === false ) {
-                       $this->mByName = User::whoIs( $this->mBy );
-               }
                return $this->mByName;
        }
 
@@ -417,24 +619,139 @@ class Block
                return wfSetVar( $this->mFromMaster, $x );
        }
 
-       /* static */ function getAutoblockExpiry( $timestamp )
+       function getRedactedName() {
+               if ( $this->mAuto ) {
+                       return '#' . $this->mId;
+               } else {
+                       return $this->mAddress;
+               }
+       }
+
+       /**
+        * Encode expiry for DB
+        */
+       static function encodeExpiry( $expiry, $db ) {
+               if ( $expiry == '' || $expiry == Block::infinity() ) {
+                       return Block::infinity();
+               } else {
+                       return $db->timestamp( $expiry );
+               }
+       }
+
+       /**
+        * Decode expiry which has come from the DB
+        */
+       static function decodeExpiry( $expiry, $timestampType = TS_MW ) {
+               if ( $expiry == '' || $expiry == Block::infinity() ) {
+                       return Block::infinity();
+               } else {
+                       return wfTimestamp( $timestampType, $expiry );
+               }
+       }
+
+       static function getAutoblockExpiry( $timestamp )
        {
                global $wgAutoblockExpiry;
                return wfTimestamp( TS_MW, wfTimestamp( TS_UNIX, $timestamp ) + $wgAutoblockExpiry );
        }
 
-       /* static */ function normaliseRange( $range )
-       {
+       /**
+        * Gets rid of uneeded numbers in quad-dotted/octet IP strings
+        * For example, 127.111.113.151/24 -> 127.111.113.0/24
+        */
+       static function normaliseRange( $range ) {
                $parts = explode( '/', $range );
                if ( count( $parts ) == 2 ) {
-                       $shift = 32 - $parts[1];
-                       $ipint = wfIP2Unsigned( $parts[0] );
-                       $ipint = $ipint >> $shift << $shift;
-                       $newip = long2ip( $ipint );
-                       $range = "$newip/{$parts[1]}";
+                       // IPv6
+                       if ( IP::isIPv6($range) && $parts[1] >= 64 && $parts[1] <= 128 ) {
+                               $bits = $parts[1];
+                               $ipint = IP::toUnsigned6( $parts[0] );
+                               # Native 32 bit functions WONT work here!!!
+                               # Convert to a padded binary number
+                               $network = wfBaseConvert( $ipint, 10, 2, 128 );
+                               # Truncate the last (128-$bits) bits and replace them with zeros
+                               $network = str_pad( substr( $network, 0, $bits ), 128, 0, STR_PAD_RIGHT );
+                               # Convert back to an integer
+                               $network = wfBaseConvert( $network, 2, 10 );
+                               # Reform octet address
+                               $newip = IP::toOctet( $network );
+                               $range = "$newip/{$parts[1]}";
+                       } // IPv4
+                       else if ( IP::isIPv4($range) && $parts[1] >= 16 && $parts[1] <= 32 ) {
+                               $shift = 32 - $parts[1];
+                               $ipint = IP::toUnsigned( $parts[0] );
+                               $ipint = $ipint >> $shift << $shift;
+                               $newip = long2ip( $ipint );
+                               $range = "$newip/{$parts[1]}";
+                       }
                }
                return $range;
        }
 
+       /**
+        * Purge expired blocks from the ipblocks table
+        */
+       static function purgeExpired() {
+               $dbw = wfGetDB( DB_MASTER );
+               $dbw->delete( 'ipblocks', array( 'ipb_expiry < ' . $dbw->addQuotes( $dbw->timestamp() ) ), __METHOD__ );
+       }
+
+       static function infinity() {
+               # This is a special keyword for timestamps in PostgreSQL, and
+               # works with CHAR(14) as well because "i" sorts after all numbers.
+               return 'infinity';
+
+               /*
+               static $infinity;
+               if ( !isset( $infinity ) ) {
+                       $dbr = wfGetDB( DB_SLAVE );
+                       $infinity = $dbr->bigTimestamp();
+               }
+               return $infinity;
+                */
+       }
+       
+       /**
+        * Convert a DB-encoded expiry into a real string that humans can read.
+        */
+       static function formatExpiry( $encoded_expiry ) {
+       
+               static $msg = null;
+               
+               if( is_null( $msg ) ) {
+                       $msg = array();
+                       $keys = array( 'infiniteblock', 'expiringblock' );
+                       foreach( $keys as $key ) {
+                               $msg[$key] = wfMsgHtml( $key );
+                       }
+               }
+               
+               $expiry = Block::decodeExpiry( $encoded_expiry );
+               if ($expiry == 'infinity') {
+                       $expirystr = $msg['infiniteblock'];
+               } else {
+                       global $wgLang;
+                       $expiretimestr = $wgLang->timeanddate( $expiry, true );
+                       $expirystr = wfMsgReplaceArgs( $msg['expiringblock'], array($expiretimestr) );
+               }
+
+               return $expirystr;
+       }
+       
+       /**
+        * Convert a typed-in expiry time into something we can put into the database.
+        */
+       static function parseExpiryInput( $expiry_input ) {
+               if ( $expiry_input == 'infinite' || $expiry_input == 'indefinite' ) {
+                       $expiry = 'infinity';
+               } else {
+                       $expiry = strtotime( $expiry_input );
+                       if ($expiry < 0 || $expiry === false) {
+                               return false;
+                       }
+               }
+               
+               return $expiry;
+       }
+
 }
-?>