00001 <?php
00012 class SquidUpdate {
00013 var $urlArr, $mMaxTitles;
00014
00015 function __construct( $urlArr = Array(), $maxTitles = false ) {
00016 global $wgMaxSquidPurgeTitles;
00017 if ( $maxTitles === false ) {
00018 $this->mMaxTitles = $wgMaxSquidPurgeTitles;
00019 } else {
00020 $this->mMaxTitles = $maxTitles;
00021 }
00022 if ( count( $urlArr ) > $this->mMaxTitles ) {
00023 $urlArr = array_slice( $urlArr, 0, $this->mMaxTitles );
00024 }
00025 $this->urlArr = $urlArr;
00026 }
00027
00028 static function newFromLinksTo( &$title ) {
00029 $fname = 'SquidUpdate::newFromLinksTo';
00030 wfProfileIn( $fname );
00031
00032 # Get a list of URLs linking to this page
00033 $dbr = wfGetDB( DB_SLAVE );
00034 $res = $dbr->select( array( 'links', 'page' ),
00035 array( 'page_namespace', 'page_title' ),
00036 array(
00037 'pl_namespace' => $title->getNamespace(),
00038 'pl_title' => $title->getDBkey(),
00039 'pl_from=page_id' ),
00040 $fname );
00041 $blurlArr = $title->getSquidURLs();
00042 if ( $dbr->numRows( $res ) <= $this->mMaxTitles ) {
00043 while ( $BL = $dbr->fetchObject ( $res ) )
00044 {
00045 $tobj = Title::makeTitle( $BL->page_namespace, $BL->page_title ) ;
00046 $blurlArr[] = $tobj->getInternalURL();
00047 }
00048 }
00049 $dbr->freeResult ( $res ) ;
00050
00051 wfProfileOut( $fname );
00052 return new SquidUpdate( $blurlArr );
00053 }
00054
00058 static function newFromTitles( $titles, $urlArr = array() ) {
00059 global $wgMaxSquidPurgeTitles;
00060 $i = 0;
00061 foreach ( $titles as $title ) {
00062 $urlArr[] = $title->getInternalURL();
00063 if ( $i++ > $wgMaxSquidPurgeTitles ) {
00064 break;
00065 }
00066 }
00067 return new SquidUpdate( $urlArr );
00068 }
00069
00070 static function newSimplePurge( &$title ) {
00071 $urlArr = $title->getSquidURLs();
00072 return new SquidUpdate( $urlArr );
00073 }
00074
00075 function doUpdate() {
00076 SquidUpdate::purge( $this->urlArr );
00077 }
00078
00079
00080
00081
00082
00083
00084 static function purge( $urlArr ) {
00085 global $wgSquidServers, $wgHTCPMulticastAddress, $wgHTCPPort;
00086
00087
00088
00089
00090
00091
00092 if( empty( $urlArr ) ) {
00093 return;
00094 }
00095
00096 if ( $wgHTCPMulticastAddress && $wgHTCPPort ) {
00097 return SquidUpdate::HTCPPurge( $urlArr );
00098 }
00099
00100 $fname = 'SquidUpdate::purge';
00101 wfProfileIn( $fname );
00102
00103 $maxsocketspersquid = 8;
00104 $urlspersocket = 400;
00105 $firsturl = SquidUpdate::expand( $urlArr[0] );
00106 unset($urlArr[0]);
00107 $urlArr = array_values($urlArr);
00108 $sockspersq = max(ceil(count($urlArr) / $urlspersocket ),1);
00109 if ($sockspersq == 1) {
00110
00111 $urlspersocket = count($urlArr);
00112 } else if ($sockspersq > $maxsocketspersquid ) {
00113 $urlspersocket = ceil(count($urlArr) / $maxsocketspersquid);
00114 $sockspersq = $maxsocketspersquid;
00115 }
00116 $totalsockets = count($wgSquidServers) * $sockspersq;
00117 $sockets = Array();
00118
00119
00120 for ($ss=0;$ss < count($wgSquidServers);$ss++) {
00121 $failed = false;
00122 $so = 0;
00123 while ($so < $sockspersq && !$failed) {
00124 if ($so == 0) {
00125
00126 @list($server, $port) = explode(':', $wgSquidServers[$ss]);
00127 if(!isset($port)) $port = 80;
00128 #$this->debug("Opening socket to $server:$port");
00129 $error = $errstr = false;
00130 $socket = @fsockopen($server, $port, $error, $errstr, 3);
00131 #$this->debug("\n");
00132 if (!$socket) {
00133 $failed = true;
00134 $totalsockets -= $sockspersq;
00135 } else {
00136 $msg = 'PURGE ' . $firsturl . " HTTP/1.0\r\n".
00137 "Connection: Keep-Alive\r\n\r\n";
00138 #$this->debug($msg);
00139 @fputs($socket,$msg);
00140 #$this->debug("...");
00141 $res = @fread($socket,512);
00142 #$this->debug("\n");
00143
00144
00145 if (strlen($res) > 250) {
00146 fclose($socket);
00147 $failed = true;
00148 $totalsockets -= $sockspersq;
00149 } else {
00150 @stream_set_blocking($socket,false);
00151 $sockets[] = $socket;
00152 }
00153 }
00154 } else {
00155
00156 list($server, $port) = explode(':', $wgSquidServers[$ss]);
00157 if(!isset($port)) $port = 80;
00158 $socket = @fsockopen($server, $port, $error, $errstr, 2);
00159 @stream_set_blocking($socket,false);
00160 $sockets[] = $socket;
00161 }
00162 $so++;
00163 }
00164 }
00165
00166 if ($urlspersocket > 0) {
00167
00168 for ($r=0;$r < $urlspersocket;$r++) {
00169 for ($s=0;$s < $totalsockets;$s++) {
00170 if($r != 0) {
00171 $res = '';
00172 $esc = 0;
00173 while (strlen($res) < 100 && $esc < 200 ) {
00174 $res .= @fread($sockets[$s],512);
00175 $esc++;
00176 usleep(20);
00177 }
00178 }
00179 $urindex = $r + $urlspersocket * ($s - $sockspersq * floor($s / $sockspersq));
00180 $url = SquidUpdate::expand( $urlArr[$urindex] );
00181 $msg = 'PURGE ' . $url . " HTTP/1.0\r\n".
00182 "Connection: Keep-Alive\r\n\r\n";
00183 #$this->debug($msg);
00184 @fputs($sockets[$s],$msg);
00185 #$this->debug("\n");
00186 }
00187 }
00188 }
00189 #$this->debug("Reading response...");
00190 foreach ($sockets as $socket) {
00191 $res = '';
00192 $esc = 0;
00193 while (strlen($res) < 100 && $esc < 200 ) {
00194 $res .= @fread($socket,1024);
00195 $esc++;
00196 usleep(20);
00197 }
00198
00199 @fclose($socket);
00200 }
00201 #$this->debug("\n");
00202 wfProfileOut( $fname );
00203 }
00204
00205 static function HTCPPurge( $urlArr ) {
00206 global $wgHTCPMulticastAddress, $wgHTCPMulticastTTL, $wgHTCPPort;
00207 $fname = 'SquidUpdate::HTCPPurge';
00208 wfProfileIn( $fname );
00209
00210 $htcpOpCLR = 4;
00211
00212
00213 if( !defined( "IPPROTO_IP" ) ) {
00214 define( "IPPROTO_IP", 0 );
00215 define( "IP_MULTICAST_LOOP", 34 );
00216 define( "IP_MULTICAST_TTL", 33 );
00217 }
00218
00219
00220 $conn = socket_create( AF_INET, SOCK_DGRAM, SOL_UDP );
00221 if ( $conn ) {
00222
00223 socket_set_option( $conn, IPPROTO_IP, IP_MULTICAST_LOOP, 0 );
00224 if ( $wgHTCPMulticastTTL != 1 )
00225 socket_set_option( $conn, IPPROTO_IP, IP_MULTICAST_TTL,
00226 $wgHTCPMulticastTTL );
00227
00228 foreach ( $urlArr as $url ) {
00229 if( !is_string( $url ) ) {
00230 throw new MWException( 'Bad purge URL' );
00231 }
00232 $url = SquidUpdate::expand( $url );
00233
00234
00235
00236
00237 $htcpTransID = rand();
00238
00239 $htcpSpecifier = pack( 'na4na*na8n',
00240 4, 'HEAD', strlen( $url ), $url,
00241 8, 'HTTP/1.0', 0 );
00242
00243 $htcpDataLen = 8 + 2 + strlen( $htcpSpecifier );
00244 $htcpLen = 4 + $htcpDataLen + 2;
00245
00246
00247
00248
00249 $htcpPacket = pack( 'nxxnCxNxxa*n',
00250 $htcpLen, $htcpDataLen, $htcpOpCLR,
00251 $htcpTransID, $htcpSpecifier, 2);
00252
00253
00254 wfDebug( "Purging URL $url via HTCP\n" );
00255 socket_sendto( $conn, $htcpPacket, $htcpLen, 0,
00256 $wgHTCPMulticastAddress, $wgHTCPPort );
00257 }
00258 } else {
00259 $errstr = socket_strerror( socket_last_error() );
00260 wfDebug( "SquidUpdate::HTCPPurge(): Error opening UDP socket: $errstr\n" );
00261 }
00262 wfProfileOut( $fname );
00263 }
00264
00265 function debug( $text ) {
00266 global $wgDebugSquid;
00267 if ( $wgDebugSquid ) {
00268 wfDebug( $text );
00269 }
00270 }
00271
00285 static function expand( $url ) {
00286 global $wgInternalServer;
00287 if( $url != '' && $url{0} == '/' ) {
00288 return $wgInternalServer . $url;
00289 }
00290 return $url;
00291 }
00292 }