-<?
+<?php
# Blocks and bans object
-
+#
#TODO: This could be used everywhere, but it isn't.
-
+#
# All the functions in this class assume the object is either explicitly
-# loaded or filled. It is not load-on-demand.
-
+# loaded or filled. It is not load-on-demand. There are no accessors.
+#
# To use delete(), you only need to fill $mAddress
+# Globals used: $wgBlockCache, $wgAutoblockExpiry
+
class Block
{
- var $mAddress, $mUser, $mBy, $mReason, $mTimestamp;
+ /* public*/ var $mAddress, $mUser, $mBy, $mReason, $mTimestamp, $mAuto, $mId, $mExpiry;
+ /* private */ var $mNetworkBits, $mIntegerAddr;
- function Block( $address = "", $user = "", $by = 0, $reason = "", $timestamp = "" )
+ function Block( $address = "", $user = "", $by = 0, $reason = "",
+ $timestamp = "" , $auto = 0, $expiry = "" )
{
$this->mAddress = $address;
$this->mUser = $user;
$this->mBy = $by;
$this->mReason = $reason;
$this->mTimestamp = $timestamp;
- }
+ $this->mAuto = $auto;
+ $this->mExpiry = $expiry;
+ $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;
$killed = false;
if ( 0 == $user ) {
- $sql = "SELECT * FROM ipblocks WHERE ipb_address='$address'";
+ $sql = "SELECT * FROM ipblocks WHERE ipb_address='" . wfStrencode( $address ) . "'";
} else {
- $sql = "SELECT * FROM ipblocks WHERE (ipb_address='$address' OR ipb_user={$user})";
+ $sql = "SELECT * FROM ipblocks WHERE (ipb_address='" . wfStrencode( $address ) .
+ "' OR ipb_user={$user})";
}
-
- $res = wfQuery( $sql, $fname );
+ $res = wfQuery( $sql, DB_READ, $fname );
if ( 0 == wfNumRows( $res ) ) {
# User is not blocked
$this->clear();
$this->initFromRow( $row );
if ( $killExpired ) {
-
# If requested, delete expired rows
do {
$killed = $this->deleteIfExpired();
- $row = wfFetchObject( $res );
+ if ( $killed ) {
+ $row = wfFetchObject( $res );
+ if ( $row ) {
+ $this->initFromRow( $row );
+ }
+ }
} while ( $killed && $row );
# If there were any left after the killing finished, return true
- if ( $row == false ) {
+ if ( !$row ) {
$ret = false;
$this->clear();
} else {
$this->mTimestamp = $row->ipb_timestamp;
$this->mUser = $row->ipb_user;
$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 )
+ /*static*/ function enumBlocks( $callback, $tag, $killExpired = true )
{
- $sql = "SELECT * FROM ipblocks ORDER BY ipb_timestamp";
- $res = wfQuery( $sql, "Block::enumBans" );
+ $sql = "SELECT * FROM ipblocks ORDER BY ipb_timestamp DESC";
+ $res = wfQuery( $sql, DB_READ, "Block::enumBans" );
$block = new Block();
while ( $row = wfFetchObject( $res ) ) {
wfFreeResult( $res );
}
- function delete()
+ function delete()
{
- wfQuery( "DELETE FROM ipblocks WHERE ipb_address='{$this->mAddress}'",
- "Block::delete" );
+ $fname = "Block::delete";
+ if ( $this->mAddress == "" ) {
+ $sql = "DELETE FROM ipblocks WHERE ipb_id={$this->mId}";
+ } else {
+ $sql = "DELETE FROM ipblocks WHERE ipb_address='" .
+ wfStrencode( $this->mAddress ) . "'";
+ }
+ 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 ) VALUES ('{$this->mAddress}', {$this->mUser}, " .
- "{$this->mBy}, '" . wfStrencode( $this->mReason ) . "','{$this->mTimestamp}')";
- wfQuery( $sql, "Block::insert" );
+ $sql = "INSERT INTO ipblocks
+ (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}, '{$this->mExpiry}')";
+ wfQuery( $sql, DB_WRITE, "Block::insert" );
+
+ $this->clearCache();
}
- function deleteIfExpired()
+ function deleteIfExpired()
{
if ( $this->isExpired() ) {
$this->delete();
}
}
- 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;
}
}
{
return $this->mAddress != "";
}
+
+ 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 updateTimestamp()
+ function getIntegerAddr()
{
- $sql = "UPDATE ipblocks SET ipb_timestamp='" . wfTimestampNow() . "' WHERE ipb_address='{$this->mAddress}'";
- wfQuery( "UPDATE ipblocks SET ipb_timestamp='" . wfTimestampNow() .
- "' WHERE ipb_address='{$this->mAddress}'", "Block::updateTimestamp" );
+ 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;
+ }
+
}
?>