rdbms: use Database::getDomainId for transaction logging calls
[lhc/web/wiklou.git] / includes / MediaWiki.php
1 <?php
2 /**
3 * Helper class for the index.php entry point.
4 *
5 * This program is free software; you can redistribute it and/or modify
6 * it under the terms of the GNU General Public License as published by
7 * the Free Software Foundation; either version 2 of the License, or
8 * (at your option) any later version.
9 *
10 * This program is distributed in the hope that it will be useful,
11 * but WITHOUT ANY WARRANTY; without even the implied warranty of
12 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
13 * GNU General Public License for more details.
14 *
15 * You should have received a copy of the GNU General Public License along
16 * with this program; if not, write to the Free Software Foundation, Inc.,
17 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
18 * http://www.gnu.org/copyleft/gpl.html
19 *
20 * @file
21 */
22
23 use MediaWiki\Logger\LoggerFactory;
24 use Psr\Log\LoggerInterface;
25 use MediaWiki\MediaWikiServices;
26 use Wikimedia\Rdbms\ChronologyProtector;
27 use Wikimedia\Rdbms\LBFactory;
28 use Wikimedia\Rdbms\DBConnectionError;
29 use Liuggio\StatsdClient\Sender\SocketSender;
30
31 /**
32 * The MediaWiki class is the helper class for the index.php entry point.
33 */
34 class MediaWiki {
35 /**
36 * @var IContextSource
37 */
38 private $context;
39
40 /**
41 * @var Config
42 */
43 private $config;
44
45 /**
46 * @var String Cache what action this request is
47 */
48 private $action;
49
50 /**
51 * @param IContextSource|null $context
52 */
53 public function __construct( IContextSource $context = null ) {
54 if ( !$context ) {
55 $context = RequestContext::getMain();
56 }
57
58 $this->context = $context;
59 $this->config = $context->getConfig();
60 }
61
62 /**
63 * Parse the request to get the Title object
64 *
65 * @throws MalformedTitleException If a title has been provided by the user, but is invalid.
66 * @return Title Title object to be $wgTitle
67 */
68 private function parseTitle() {
69 $request = $this->context->getRequest();
70 $curid = $request->getInt( 'curid' );
71 $title = $request->getVal( 'title' );
72 $action = $request->getVal( 'action' );
73
74 if ( $request->getCheck( 'search' ) ) {
75 // Compatibility with old search URLs which didn't use Special:Search
76 // Just check for presence here, so blank requests still
77 // show the search page when using ugly URLs (T10054).
78 $ret = SpecialPage::getTitleFor( 'Search' );
79 } elseif ( $curid ) {
80 // URLs like this are generated by RC, because rc_title isn't always accurate
81 $ret = Title::newFromID( $curid );
82 } else {
83 $ret = Title::newFromURL( $title );
84 // Alias NS_MEDIA page URLs to NS_FILE...we only use NS_MEDIA
85 // in wikitext links to tell Parser to make a direct file link
86 if ( !is_null( $ret ) && $ret->getNamespace() == NS_MEDIA ) {
87 $ret = Title::makeTitle( NS_FILE, $ret->getDBkey() );
88 }
89 $contLang = MediaWikiServices::getInstance()->getContentLanguage();
90 // Check variant links so that interwiki links don't have to worry
91 // about the possible different language variants
92 if (
93 $contLang->hasVariants() && !is_null( $ret ) && $ret->getArticleID() == 0
94 ) {
95 $contLang->findVariantLink( $title, $ret );
96 }
97 }
98
99 // If title is not provided, always allow oldid and diff to set the title.
100 // If title is provided, allow oldid and diff to override the title, unless
101 // we are talking about a special page which might use these parameters for
102 // other purposes.
103 if ( $ret === null || !$ret->isSpecialPage() ) {
104 // We can have urls with just ?diff=,?oldid= or even just ?diff=
105 $oldid = $request->getInt( 'oldid' );
106 $oldid = $oldid ?: $request->getInt( 'diff' );
107 // Allow oldid to override a changed or missing title
108 if ( $oldid ) {
109 $rev = Revision::newFromId( $oldid );
110 $ret = $rev ? $rev->getTitle() : $ret;
111 }
112 }
113
114 // Use the main page as default title if nothing else has been provided
115 if ( $ret === null
116 && strval( $title ) === ''
117 && !$request->getCheck( 'curid' )
118 && $action !== 'delete'
119 ) {
120 $ret = Title::newMainPage();
121 }
122
123 if ( $ret === null || ( $ret->getDBkey() == '' && !$ret->isExternal() ) ) {
124 // If we get here, we definitely don't have a valid title; throw an exception.
125 // Try to get detailed invalid title exception first, fall back to MalformedTitleException.
126 Title::newFromTextThrow( $title );
127 throw new MalformedTitleException( 'badtitletext', $title );
128 }
129
130 return $ret;
131 }
132
133 /**
134 * Get the Title object that we'll be acting on, as specified in the WebRequest
135 * @return Title
136 */
137 public function getTitle() {
138 if ( !$this->context->hasTitle() ) {
139 try {
140 $this->context->setTitle( $this->parseTitle() );
141 } catch ( MalformedTitleException $ex ) {
142 $this->context->setTitle( SpecialPage::getTitleFor( 'Badtitle' ) );
143 }
144 }
145 return $this->context->getTitle();
146 }
147
148 /**
149 * Returns the name of the action that will be executed.
150 *
151 * @return string Action
152 */
153 public function getAction() {
154 if ( $this->action === null ) {
155 $this->action = Action::getActionName( $this->context );
156 }
157
158 return $this->action;
159 }
160
161 /**
162 * Performs the request.
163 * - bad titles
164 * - read restriction
165 * - local interwiki redirects
166 * - redirect loop
167 * - special pages
168 * - normal pages
169 *
170 * @throws MWException|PermissionsError|BadTitleError|HttpError
171 * @return void
172 */
173 private function performRequest() {
174 global $wgTitle;
175
176 $request = $this->context->getRequest();
177 $requestTitle = $title = $this->context->getTitle();
178 $output = $this->context->getOutput();
179 $user = $this->context->getUser();
180
181 if ( $request->getVal( 'printable' ) === 'yes' ) {
182 $output->setPrintable();
183 }
184
185 $unused = null; // To pass it by reference
186 Hooks::run( 'BeforeInitialize', [ &$title, &$unused, &$output, &$user, $request, $this ] );
187
188 // Invalid titles. T23776: The interwikis must redirect even if the page name is empty.
189 if ( is_null( $title ) || ( $title->getDBkey() == '' && !$title->isExternal() )
190 || $title->isSpecial( 'Badtitle' )
191 ) {
192 $this->context->setTitle( SpecialPage::getTitleFor( 'Badtitle' ) );
193 try {
194 $this->parseTitle();
195 } catch ( MalformedTitleException $ex ) {
196 throw new BadTitleError( $ex );
197 }
198 throw new BadTitleError();
199 }
200
201 // Check user's permissions to read this page.
202 // We have to check here to catch special pages etc.
203 // We will check again in Article::view().
204 $permErrors = $title->isSpecial( 'RunJobs' )
205 ? [] // relies on HMAC key signature alone
206 : $title->getUserPermissionsErrors( 'read', $user );
207 if ( count( $permErrors ) ) {
208 // T34276: allowing the skin to generate output with $wgTitle or
209 // $this->context->title set to the input title would allow anonymous users to
210 // determine whether a page exists, potentially leaking private data. In fact, the
211 // curid and oldid request parameters would allow page titles to be enumerated even
212 // when they are not guessable. So we reset the title to Special:Badtitle before the
213 // permissions error is displayed.
214
215 // The skin mostly uses $this->context->getTitle() these days, but some extensions
216 // still use $wgTitle.
217 $badTitle = SpecialPage::getTitleFor( 'Badtitle' );
218 $this->context->setTitle( $badTitle );
219 $wgTitle = $badTitle;
220
221 throw new PermissionsError( 'read', $permErrors );
222 }
223
224 // Interwiki redirects
225 if ( $title->isExternal() ) {
226 $rdfrom = $request->getVal( 'rdfrom' );
227 if ( $rdfrom ) {
228 $url = $title->getFullURL( [ 'rdfrom' => $rdfrom ] );
229 } else {
230 $query = $request->getValues();
231 unset( $query['title'] );
232 $url = $title->getFullURL( $query );
233 }
234 // Check for a redirect loop
235 if ( !preg_match( '/^' . preg_quote( $this->config->get( 'Server' ), '/' ) . '/', $url )
236 && $title->isLocal()
237 ) {
238 // 301 so google et al report the target as the actual url.
239 $output->redirect( $url, 301 );
240 } else {
241 $this->context->setTitle( SpecialPage::getTitleFor( 'Badtitle' ) );
242 try {
243 $this->parseTitle();
244 } catch ( MalformedTitleException $ex ) {
245 throw new BadTitleError( $ex );
246 }
247 throw new BadTitleError();
248 }
249 // Handle any other redirects.
250 // Redirect loops, titleless URL, $wgUsePathInfo URLs, and URLs with a variant
251 } elseif ( !$this->tryNormaliseRedirect( $title ) ) {
252 // Prevent information leak via Special:MyPage et al (T109724)
253 if ( $title->isSpecialPage() ) {
254 $specialPage = SpecialPageFactory::getPage( $title->getDBkey() );
255 if ( $specialPage instanceof RedirectSpecialPage ) {
256 $specialPage->setContext( $this->context );
257 if ( $this->config->get( 'HideIdentifiableRedirects' )
258 && $specialPage->personallyIdentifiableTarget()
259 ) {
260 list( , $subpage ) = SpecialPageFactory::resolveAlias( $title->getDBkey() );
261 $target = $specialPage->getRedirect( $subpage );
262 // target can also be true. We let that case fall through to normal processing.
263 if ( $target instanceof Title ) {
264 $query = $specialPage->getRedirectQuery() ?: [];
265 $request = new DerivativeRequest( $this->context->getRequest(), $query );
266 $request->setRequestURL( $this->context->getRequest()->getRequestURL() );
267 $this->context->setRequest( $request );
268 // Do not varnish cache these. May vary even for anons
269 $this->context->getOutput()->lowerCdnMaxage( 0 );
270 $this->context->setTitle( $target );
271 $wgTitle = $target;
272 // Reset action type cache. (Special pages have only view)
273 $this->action = null;
274 $title = $target;
275 $output->addJsConfigVars( [
276 'wgInternalRedirectTargetUrl' => $target->getFullURL( $query ),
277 ] );
278 $output->addModules( 'mediawiki.action.view.redirect' );
279 }
280 }
281 }
282 }
283
284 // Special pages ($title may have changed since if statement above)
285 if ( $title->isSpecialPage() ) {
286 // Actions that need to be made when we have a special pages
287 SpecialPageFactory::executePath( $title, $this->context );
288 } else {
289 // ...otherwise treat it as an article view. The article
290 // may still be a wikipage redirect to another article or URL.
291 $article = $this->initializeArticle();
292 if ( is_object( $article ) ) {
293 $this->performAction( $article, $requestTitle );
294 } elseif ( is_string( $article ) ) {
295 $output->redirect( $article );
296 } else {
297 throw new MWException( "Shouldn't happen: MediaWiki::initializeArticle()"
298 . " returned neither an object nor a URL" );
299 }
300 }
301 }
302 }
303
304 /**
305 * Handle redirects for uncanonical title requests.
306 *
307 * Handles:
308 * - Redirect loops.
309 * - No title in URL.
310 * - $wgUsePathInfo URLs.
311 * - URLs with a variant.
312 * - Other non-standard URLs (as long as they have no extra query parameters).
313 *
314 * Behaviour:
315 * - Normalise title values:
316 * /wiki/Foo%20Bar -> /wiki/Foo_Bar
317 * - Normalise empty title:
318 * /wiki/ -> /wiki/Main
319 * /w/index.php?title= -> /wiki/Main
320 * - Don't redirect anything with query parameters other than 'title' or 'action=view'.
321 *
322 * @param Title $title
323 * @return bool True if a redirect was set.
324 * @throws HttpError
325 */
326 private function tryNormaliseRedirect( Title $title ) {
327 $request = $this->context->getRequest();
328 $output = $this->context->getOutput();
329
330 if ( $request->getVal( 'action', 'view' ) != 'view'
331 || $request->wasPosted()
332 || ( $request->getVal( 'title' ) !== null
333 && $title->getPrefixedDBkey() == $request->getVal( 'title' ) )
334 || count( $request->getValueNames( [ 'action', 'title' ] ) )
335 || !Hooks::run( 'TestCanonicalRedirect', [ $request, $title, $output ] )
336 ) {
337 return false;
338 }
339
340 if ( $title->isSpecialPage() ) {
341 list( $name, $subpage ) = SpecialPageFactory::resolveAlias( $title->getDBkey() );
342 if ( $name ) {
343 $title = SpecialPage::getTitleFor( $name, $subpage );
344 }
345 }
346 // Redirect to canonical url, make it a 301 to allow caching
347 $targetUrl = wfExpandUrl( $title->getFullURL(), PROTO_CURRENT );
348 if ( $targetUrl == $request->getFullRequestURL() ) {
349 $message = "Redirect loop detected!\n\n" .
350 "This means the wiki got confused about what page was " .
351 "requested; this sometimes happens when moving a wiki " .
352 "to a new server or changing the server configuration.\n\n";
353
354 if ( $this->config->get( 'UsePathInfo' ) ) {
355 $message .= "The wiki is trying to interpret the page " .
356 "title from the URL path portion (PATH_INFO), which " .
357 "sometimes fails depending on the web server. Try " .
358 "setting \"\$wgUsePathInfo = false;\" in your " .
359 "LocalSettings.php, or check that \$wgArticlePath " .
360 "is correct.";
361 } else {
362 $message .= "Your web server was detected as possibly not " .
363 "supporting URL path components (PATH_INFO) correctly; " .
364 "check your LocalSettings.php for a customized " .
365 "\$wgArticlePath setting and/or toggle \$wgUsePathInfo " .
366 "to true.";
367 }
368 throw new HttpError( 500, $message );
369 }
370 $output->setCdnMaxage( 1200 );
371 $output->redirect( $targetUrl, '301' );
372 return true;
373 }
374
375 /**
376 * Initialize the main Article object for "standard" actions (view, etc)
377 * Create an Article object for the page, following redirects if needed.
378 *
379 * @return Article|string An Article, or a string to redirect to another URL
380 */
381 private function initializeArticle() {
382 $title = $this->context->getTitle();
383 if ( $this->context->canUseWikiPage() ) {
384 // Try to use request context wiki page, as there
385 // is already data from db saved in per process
386 // cache there from this->getAction() call.
387 $page = $this->context->getWikiPage();
388 } else {
389 // This case should not happen, but just in case.
390 // @TODO: remove this or use an exception
391 $page = WikiPage::factory( $title );
392 $this->context->setWikiPage( $page );
393 wfWarn( "RequestContext::canUseWikiPage() returned false" );
394 }
395
396 // Make GUI wrapper for the WikiPage
397 $article = Article::newFromWikiPage( $page, $this->context );
398
399 // Skip some unnecessary code if the content model doesn't support redirects
400 if ( !ContentHandler::getForTitle( $title )->supportsRedirects() ) {
401 return $article;
402 }
403
404 $request = $this->context->getRequest();
405
406 // Namespace might change when using redirects
407 // Check for redirects ...
408 $action = $request->getVal( 'action', 'view' );
409 $file = ( $page instanceof WikiFilePage ) ? $page->getFile() : null;
410 if ( ( $action == 'view' || $action == 'render' ) // ... for actions that show content
411 && !$request->getVal( 'oldid' ) // ... and are not old revisions
412 && !$request->getVal( 'diff' ) // ... and not when showing diff
413 && $request->getVal( 'redirect' ) != 'no' // ... unless explicitly told not to
414 // ... and the article is not a non-redirect image page with associated file
415 && !( is_object( $file ) && $file->exists() && !$file->getRedirected() )
416 ) {
417 // Give extensions a change to ignore/handle redirects as needed
418 $ignoreRedirect = $target = false;
419
420 Hooks::run( 'InitializeArticleMaybeRedirect',
421 [ &$title, &$request, &$ignoreRedirect, &$target, &$article ] );
422 $page = $article->getPage(); // reflect any hook changes
423
424 // Follow redirects only for... redirects.
425 // If $target is set, then a hook wanted to redirect.
426 if ( !$ignoreRedirect && ( $target || $page->isRedirect() ) ) {
427 // Is the target already set by an extension?
428 $target = $target ?: $page->followRedirect();
429 if ( is_string( $target ) ) {
430 if ( !$this->config->get( 'DisableHardRedirects' ) ) {
431 // we'll need to redirect
432 return $target;
433 }
434 }
435 if ( is_object( $target ) ) {
436 // Rewrite environment to redirected article
437 $rpage = WikiPage::factory( $target );
438 $rpage->loadPageData();
439 if ( $rpage->exists() || ( is_object( $file ) && !$file->isLocal() ) ) {
440 $rarticle = Article::newFromWikiPage( $rpage, $this->context );
441 $rarticle->setRedirectedFrom( $title );
442
443 $article = $rarticle;
444 $this->context->setTitle( $target );
445 $this->context->setWikiPage( $article->getPage() );
446 }
447 }
448 } else {
449 // Article may have been changed by hook
450 $this->context->setTitle( $article->getTitle() );
451 $this->context->setWikiPage( $article->getPage() );
452 }
453 }
454
455 return $article;
456 }
457
458 /**
459 * Perform one of the "standard" actions
460 *
461 * @param Page $page
462 * @param Title $requestTitle The original title, before any redirects were applied
463 */
464 private function performAction( Page $page, Title $requestTitle ) {
465 $request = $this->context->getRequest();
466 $output = $this->context->getOutput();
467 $title = $this->context->getTitle();
468 $user = $this->context->getUser();
469
470 if ( !Hooks::run( 'MediaWikiPerformAction',
471 [ $output, $page, $title, $user, $request, $this ] )
472 ) {
473 return;
474 }
475
476 $act = $this->getAction();
477 $action = Action::factory( $act, $page, $this->context );
478
479 if ( $action instanceof Action ) {
480 // Narrow DB query expectations for this HTTP request
481 $trxLimits = $this->config->get( 'TrxProfilerLimits' );
482 $trxProfiler = Profiler::instance()->getTransactionProfiler();
483 if ( $request->wasPosted() && !$action->doesWrites() ) {
484 $trxProfiler->setExpectations( $trxLimits['POST-nonwrite'], __METHOD__ );
485 $request->markAsSafeRequest();
486 }
487
488 # Let CDN cache things if we can purge them.
489 if ( $this->config->get( 'UseSquid' ) &&
490 in_array(
491 // Use PROTO_INTERNAL because that's what getCdnUrls() uses
492 wfExpandUrl( $request->getRequestURL(), PROTO_INTERNAL ),
493 $requestTitle->getCdnUrls()
494 )
495 ) {
496 $output->setCdnMaxage( $this->config->get( 'SquidMaxage' ) );
497 }
498
499 $action->show();
500 return;
501 }
502 // NOTE: deprecated hook. Add to $wgActions instead
503 if ( Hooks::run(
504 'UnknownAction',
505 [
506 $request->getVal( 'action', 'view' ),
507 $page
508 ],
509 '1.19'
510 ) ) {
511 $output->setStatusCode( 404 );
512 $output->showErrorPage( 'nosuchaction', 'nosuchactiontext' );
513 }
514 }
515
516 /**
517 * Run the current MediaWiki instance; index.php just calls this
518 */
519 public function run() {
520 try {
521 $this->setDBProfilingAgent();
522 try {
523 $this->main();
524 } catch ( ErrorPageError $e ) {
525 // T64091: while exceptions are convenient to bubble up GUI errors,
526 // they are not internal application faults. As with normal requests, this
527 // should commit, print the output, do deferred updates, jobs, and profiling.
528 $this->doPreOutputCommit();
529 $e->report(); // display the GUI error
530 }
531 } catch ( Exception $e ) {
532 $context = $this->context;
533 $action = $context->getRequest()->getVal( 'action', 'view' );
534 if (
535 $e instanceof DBConnectionError &&
536 $context->hasTitle() &&
537 $context->getTitle()->canExist() &&
538 in_array( $action, [ 'view', 'history' ], true ) &&
539 HTMLFileCache::useFileCache( $this->context, HTMLFileCache::MODE_OUTAGE )
540 ) {
541 // Try to use any (even stale) file during outages...
542 $cache = new HTMLFileCache( $context->getTitle(), $action );
543 if ( $cache->isCached() ) {
544 $cache->loadFromFileCache( $context, HTMLFileCache::MODE_OUTAGE );
545 print MWExceptionRenderer::getHTML( $e );
546 exit;
547 }
548 }
549
550 MWExceptionHandler::handleException( $e );
551 } catch ( Error $e ) {
552 // Type errors and such: at least handle it now and clean up the LBFactory state
553 MWExceptionHandler::handleException( $e );
554 }
555
556 $this->doPostOutputShutdown( 'normal' );
557 }
558
559 private function setDBProfilingAgent() {
560 $services = MediaWikiServices::getInstance();
561 // Add a comment for easy SHOW PROCESSLIST interpretation
562 $name = $this->context->getUser()->getName();
563 $services->getDBLoadBalancerFactory()->setAgentName(
564 mb_strlen( $name ) > 15 ? mb_substr( $name, 0, 15 ) . '...' : $name
565 );
566 }
567
568 /**
569 * @see MediaWiki::preOutputCommit()
570 * @param callable|null $postCommitWork [default: null]
571 * @since 1.26
572 */
573 public function doPreOutputCommit( callable $postCommitWork = null ) {
574 self::preOutputCommit( $this->context, $postCommitWork );
575 }
576
577 /**
578 * This function commits all DB changes as needed before
579 * the user can receive a response (in case commit fails)
580 *
581 * @param IContextSource $context
582 * @param callable|null $postCommitWork [default: null]
583 * @since 1.27
584 */
585 public static function preOutputCommit(
586 IContextSource $context, callable $postCommitWork = null
587 ) {
588 // Either all DBs should commit or none
589 ignore_user_abort( true );
590
591 $config = $context->getConfig();
592 $request = $context->getRequest();
593 $output = $context->getOutput();
594 $lbFactory = MediaWikiServices::getInstance()->getDBLoadBalancerFactory();
595
596 // Commit all changes
597 $lbFactory->commitMasterChanges(
598 __METHOD__,
599 // Abort if any transaction was too big
600 [ 'maxWriteDuration' => $config->get( 'MaxUserDBWriteDuration' ) ]
601 );
602 wfDebug( __METHOD__ . ': primary transaction round committed' );
603
604 // Run updates that need to block the user or affect output (this is the last chance)
605 DeferredUpdates::doUpdates( 'enqueue', DeferredUpdates::PRESEND );
606 wfDebug( __METHOD__ . ': pre-send deferred updates completed' );
607
608 // Should the client return, their request should observe the new ChronologyProtector
609 // DB positions. This request might be on a foreign wiki domain, so synchronously update
610 // the DB positions in all datacenters to be safe. If this output is not a redirect,
611 // then OutputPage::output() will be relatively slow, meaning that running it in
612 // $postCommitWork should help mask the latency of those updates.
613 $flags = $lbFactory::SHUTDOWN_CHRONPROT_SYNC;
614 $strategy = 'cookie+sync';
615
616 $allowHeaders = !( $output->isDisabled() || headers_sent() );
617 if ( $output->getRedirect() && $lbFactory->hasOrMadeRecentMasterChanges( INF ) ) {
618 // OutputPage::output() will be fast, so $postCommitWork is useless for masking
619 // the latency of synchronously updating the DB positions in all datacenters.
620 // Try to make use of the time the client spends following redirects instead.
621 $domainDistance = self::getUrlDomainDistance( $output->getRedirect() );
622 if ( $domainDistance === 'local' && $allowHeaders ) {
623 $flags = $lbFactory::SHUTDOWN_CHRONPROT_ASYNC;
624 $strategy = 'cookie'; // use same-domain cookie and keep the URL uncluttered
625 } elseif ( $domainDistance === 'remote' ) {
626 $flags = $lbFactory::SHUTDOWN_CHRONPROT_ASYNC;
627 $strategy = 'cookie+url'; // cross-domain cookie might not work
628 }
629 }
630
631 // Record ChronologyProtector positions for DBs affected in this request at this point
632 $cpIndex = null;
633 $cpClientId = null;
634 $lbFactory->shutdown( $flags, $postCommitWork, $cpIndex, $cpClientId );
635 wfDebug( __METHOD__ . ': LBFactory shutdown completed' );
636
637 if ( $cpIndex > 0 ) {
638 if ( $allowHeaders ) {
639 $now = time();
640 $expires = $now + ChronologyProtector::POSITION_COOKIE_TTL;
641 $options = [ 'prefix' => '' ];
642 $value = LBFactory::makeCookieValueFromCPIndex( $cpIndex, $now, $cpClientId );
643 $request->response()->setCookie( 'cpPosIndex', $value, $expires, $options );
644 }
645
646 if ( $strategy === 'cookie+url' ) {
647 if ( $output->getRedirect() ) { // sanity
648 $safeUrl = $lbFactory->appendShutdownCPIndexAsQuery(
649 $output->getRedirect(),
650 $cpIndex
651 );
652 $output->redirect( $safeUrl );
653 } else {
654 $e = new LogicException( "No redirect; cannot append cpPosIndex parameter." );
655 MWExceptionHandler::logException( $e );
656 }
657 }
658 }
659
660 // Set a cookie to tell all CDN edge nodes to "stick" the user to the DC that handles this
661 // POST request (e.g. the "master" data center). Also have the user briefly bypass CDN so
662 // ChronologyProtector works for cacheable URLs.
663 if ( $request->wasPosted() && $lbFactory->hasOrMadeRecentMasterChanges() ) {
664 $expires = time() + $config->get( 'DataCenterUpdateStickTTL' );
665 $options = [ 'prefix' => '' ];
666 $request->response()->setCookie( 'UseDC', 'master', $expires, $options );
667 $request->response()->setCookie( 'UseCDNCache', 'false', $expires, $options );
668 }
669
670 // Avoid letting a few seconds of replica DB lag cause a month of stale data. This logic is
671 // also intimately related to the value of $wgCdnReboundPurgeDelay.
672 if ( $lbFactory->laggedReplicaUsed() ) {
673 $maxAge = $config->get( 'CdnMaxageLagged' );
674 $output->lowerCdnMaxage( $maxAge );
675 $request->response()->header( "X-Database-Lagged: true" );
676 wfDebugLog( 'replication', "Lagged DB used; CDN cache TTL limited to $maxAge seconds" );
677 }
678
679 // Avoid long-term cache pollution due to message cache rebuild timeouts (T133069)
680 if ( MessageCache::singleton()->isDisabled() ) {
681 $maxAge = $config->get( 'CdnMaxageSubstitute' );
682 $output->lowerCdnMaxage( $maxAge );
683 $request->response()->header( "X-Response-Substitute: true" );
684 }
685 }
686
687 /**
688 * @param string $url
689 * @return string Either "local", "remote" if in the farm, "external" otherwise
690 */
691 private static function getUrlDomainDistance( $url ) {
692 $clusterWiki = WikiMap::getWikiFromUrl( $url );
693 if ( $clusterWiki === wfWikiID() ) {
694 return 'local'; // the current wiki
695 } elseif ( $clusterWiki !== false ) {
696 return 'remote'; // another wiki in this cluster/farm
697 }
698
699 return 'external';
700 }
701
702 /**
703 * This function does work that can be done *after* the
704 * user gets the HTTP response so they don't block on it
705 *
706 * This manages deferred updates, job insertion,
707 * final commit, and the logging of profiling data
708 *
709 * @param string $mode Use 'fast' to always skip job running
710 * @since 1.26
711 */
712 public function doPostOutputShutdown( $mode = 'normal' ) {
713 // Perform the last synchronous operations...
714 try {
715 // Record backend request timing
716 $timing = $this->context->getTiming();
717 $timing->mark( 'requestShutdown' );
718 // Show visible profiling data if enabled (which cannot be post-send)
719 Profiler::instance()->logDataPageOutputOnly();
720 } catch ( Exception $e ) {
721 // An error may already have been shown in run(), so just log it to be safe
722 MWExceptionHandler::rollbackMasterChangesAndLog( $e );
723 }
724
725 // Disable WebResponse setters for post-send processing (T191537).
726 WebResponse::disableForPostSend();
727
728 $blocksHttpClient = true;
729 // Defer everything else if possible...
730 $callback = function () use ( $mode, &$blocksHttpClient ) {
731 try {
732 $this->restInPeace( $mode, $blocksHttpClient );
733 } catch ( Exception $e ) {
734 // If this is post-send, then displaying errors can cause broken HTML
735 MWExceptionHandler::rollbackMasterChangesAndLog( $e );
736 }
737 };
738
739 if ( function_exists( 'register_postsend_function' ) ) {
740 // https://github.com/facebook/hhvm/issues/1230
741 register_postsend_function( $callback );
742 /** @noinspection PhpUnusedLocalVariableInspection */
743 $blocksHttpClient = false;
744 } else {
745 if ( function_exists( 'fastcgi_finish_request' ) ) {
746 fastcgi_finish_request();
747 /** @noinspection PhpUnusedLocalVariableInspection */
748 $blocksHttpClient = false;
749 } else {
750 // Either all DB and deferred updates should happen or none.
751 // The latter should not be cancelled due to client disconnect.
752 ignore_user_abort( true );
753 }
754
755 $callback();
756 }
757 }
758
759 private function main() {
760 global $wgTitle;
761
762 $output = $this->context->getOutput();
763 $request = $this->context->getRequest();
764
765 // Send Ajax requests to the Ajax dispatcher.
766 if ( $request->getVal( 'action' ) === 'ajax' ) {
767 // Set a dummy title, because $wgTitle == null might break things
768 $title = Title::makeTitle( NS_SPECIAL, 'Badtitle/performing an AJAX call in '
769 . __METHOD__
770 );
771 $this->context->setTitle( $title );
772 $wgTitle = $title;
773
774 $dispatcher = new AjaxDispatcher( $this->config );
775 $dispatcher->performAction( $this->context->getUser() );
776
777 return;
778 }
779
780 // Get title from request parameters,
781 // is set on the fly by parseTitle the first time.
782 $title = $this->getTitle();
783 $action = $this->getAction();
784 $wgTitle = $title;
785
786 // Set DB query expectations for this HTTP request
787 $trxLimits = $this->config->get( 'TrxProfilerLimits' );
788 $trxProfiler = Profiler::instance()->getTransactionProfiler();
789 $trxProfiler->setLogger( LoggerFactory::getInstance( 'DBPerformance' ) );
790 if ( $request->hasSafeMethod() ) {
791 $trxProfiler->setExpectations( $trxLimits['GET'], __METHOD__ );
792 } else {
793 $trxProfiler->setExpectations( $trxLimits['POST'], __METHOD__ );
794 }
795
796 // If the user has forceHTTPS set to true, or if the user
797 // is in a group requiring HTTPS, or if they have the HTTPS
798 // preference set, redirect them to HTTPS.
799 // Note: Do this after $wgTitle is setup, otherwise the hooks run from
800 // isLoggedIn() will do all sorts of weird stuff.
801 if (
802 $request->getProtocol() == 'http' &&
803 // switch to HTTPS only when supported by the server
804 preg_match( '#^https://#', wfExpandUrl( $request->getRequestURL(), PROTO_HTTPS ) ) &&
805 (
806 $request->getSession()->shouldForceHTTPS() ||
807 // Check the cookie manually, for paranoia
808 $request->getCookie( 'forceHTTPS', '' ) ||
809 // check for prefixed version that was used for a time in older MW versions
810 $request->getCookie( 'forceHTTPS' ) ||
811 // Avoid checking the user and groups unless it's enabled.
812 (
813 $this->context->getUser()->isLoggedIn()
814 && $this->context->getUser()->requiresHTTPS()
815 )
816 )
817 ) {
818 $oldUrl = $request->getFullRequestURL();
819 $redirUrl = preg_replace( '#^http://#', 'https://', $oldUrl );
820
821 // ATTENTION: This hook is likely to be removed soon due to overall design of the system.
822 if ( Hooks::run( 'BeforeHttpsRedirect', [ $this->context, &$redirUrl ] ) ) {
823 if ( $request->wasPosted() ) {
824 // This is weird and we'd hope it almost never happens. This
825 // means that a POST came in via HTTP and policy requires us
826 // redirecting to HTTPS. It's likely such a request is going
827 // to fail due to post data being lost, but let's try anyway
828 // and just log the instance.
829
830 // @todo FIXME: See if we could issue a 307 or 308 here, need
831 // to see how clients (automated & browser) behave when we do
832 wfDebugLog( 'RedirectedPosts', "Redirected from HTTP to HTTPS: $oldUrl" );
833 }
834 // Setup dummy Title, otherwise OutputPage::redirect will fail
835 $title = Title::newFromText( 'REDIR', NS_MAIN );
836 $this->context->setTitle( $title );
837 // Since we only do this redir to change proto, always send a vary header
838 $output->addVaryHeader( 'X-Forwarded-Proto' );
839 $output->redirect( $redirUrl );
840 $output->output();
841
842 return;
843 }
844 }
845
846 if ( $title->canExist() && HTMLFileCache::useFileCache( $this->context ) ) {
847 // Try low-level file cache hit
848 $cache = new HTMLFileCache( $title, $action );
849 if ( $cache->isCacheGood( /* Assume up to date */ ) ) {
850 // Check incoming headers to see if client has this cached
851 $timestamp = $cache->cacheTimestamp();
852 if ( !$output->checkLastModified( $timestamp ) ) {
853 $cache->loadFromFileCache( $this->context );
854 }
855 // Do any stats increment/watchlist stuff, assuming user is viewing the
856 // latest revision (which should always be the case for file cache)
857 $this->context->getWikiPage()->doViewUpdates( $this->context->getUser() );
858 // Tell OutputPage that output is taken care of
859 $output->disable();
860
861 return;
862 }
863 }
864
865 // Actually do the work of the request and build up any output
866 $this->performRequest();
867
868 // GUI-ify and stash the page output in MediaWiki::doPreOutputCommit() while
869 // ChronologyProtector synchronizes DB positions or replicas accross all datacenters.
870 $buffer = null;
871 $outputWork = function () use ( $output, &$buffer ) {
872 if ( $buffer === null ) {
873 $buffer = $output->output( true );
874 }
875
876 return $buffer;
877 };
878
879 // Now commit any transactions, so that unreported errors after
880 // output() don't roll back the whole DB transaction and so that
881 // we avoid having both success and error text in the response
882 $this->doPreOutputCommit( $outputWork );
883
884 // Now send the actual output
885 print $outputWork();
886 }
887
888 /**
889 * Ends this task peacefully
890 * @param string $mode Use 'fast' to always skip job running
891 * @param bool $blocksHttpClient Whether this blocks an HTTP response to a client
892 */
893 public function restInPeace( $mode = 'fast', $blocksHttpClient = true ) {
894 $lbFactory = MediaWikiServices::getInstance()->getDBLoadBalancerFactory();
895 // Assure deferred updates are not in the main transaction
896 $lbFactory->commitMasterChanges( __METHOD__ );
897
898 // Loosen DB query expectations since the HTTP client is unblocked
899 $trxProfiler = Profiler::instance()->getTransactionProfiler();
900 $trxProfiler->resetExpectations();
901 $trxProfiler->setExpectations(
902 $this->context->getRequest()->hasSafeMethod()
903 ? $this->config->get( 'TrxProfilerLimits' )['PostSend-GET']
904 : $this->config->get( 'TrxProfilerLimits' )['PostSend-POST'],
905 __METHOD__
906 );
907
908 // Important: this must be the last deferred update added (T100085, T154425)
909 DeferredUpdates::addCallableUpdate( [ JobQueueGroup::class, 'pushLazyJobs' ] );
910
911 // Do any deferred jobs; preferring to run them now if a client will not wait on them
912 DeferredUpdates::doUpdates( $blocksHttpClient ? 'enqueue' : 'run' );
913
914 // Now that everything specific to this request is done,
915 // try to occasionally run jobs (if enabled) from the queues
916 if ( $mode === 'normal' ) {
917 $this->triggerJobs();
918 }
919
920 // Log profiling data, e.g. in the database or UDP
921 wfLogProfilingData();
922
923 // Commit and close up!
924 $lbFactory->commitMasterChanges( __METHOD__ );
925 $lbFactory->shutdown( LBFactory::SHUTDOWN_NO_CHRONPROT );
926
927 wfDebug( "Request ended normally\n" );
928 }
929
930 /**
931 * Send out any buffered statsd data according to sampling rules
932 *
933 * @param IBufferingStatsdDataFactory $stats
934 * @param Config $config
935 * @throws ConfigException
936 * @since 1.31
937 */
938 public static function emitBufferedStatsdData(
939 IBufferingStatsdDataFactory $stats, Config $config
940 ) {
941 if ( $config->get( 'StatsdServer' ) && $stats->hasData() ) {
942 try {
943 $statsdServer = explode( ':', $config->get( 'StatsdServer' ) );
944 $statsdHost = $statsdServer[0];
945 $statsdPort = $statsdServer[1] ?? 8125;
946 $statsdSender = new SocketSender( $statsdHost, $statsdPort );
947 $statsdClient = new SamplingStatsdClient( $statsdSender, true, false );
948 $statsdClient->setSamplingRates( $config->get( 'StatsdSamplingRates' ) );
949 $statsdClient->send( $stats->getData() );
950
951 $stats->clearData(); // empty buffer for the next round
952 } catch ( Exception $ex ) {
953 MWExceptionHandler::logException( $ex );
954 }
955 }
956 }
957
958 /**
959 * Potentially open a socket and sent an HTTP request back to the server
960 * to run a specified number of jobs. This registers a callback to cleanup
961 * the socket once it's done.
962 */
963 public function triggerJobs() {
964 $jobRunRate = $this->config->get( 'JobRunRate' );
965 if ( $this->getTitle()->isSpecial( 'RunJobs' ) ) {
966 return; // recursion guard
967 } elseif ( $jobRunRate <= 0 || wfReadOnly() ) {
968 return;
969 }
970
971 if ( $jobRunRate < 1 ) {
972 $max = mt_getrandmax();
973 if ( mt_rand( 0, $max ) > $max * $jobRunRate ) {
974 return; // the higher the job run rate, the less likely we return here
975 }
976 $n = 1;
977 } else {
978 $n = intval( $jobRunRate );
979 }
980
981 $logger = LoggerFactory::getInstance( 'runJobs' );
982
983 try {
984 if ( $this->config->get( 'RunJobsAsync' ) ) {
985 // Send an HTTP request to the job RPC entry point if possible
986 $invokedWithSuccess = $this->triggerAsyncJobs( $n, $logger );
987 if ( !$invokedWithSuccess ) {
988 // Fall back to blocking on running the job(s)
989 $logger->warning( "Jobs switched to blocking; Special:RunJobs disabled" );
990 $this->triggerSyncJobs( $n, $logger );
991 }
992 } else {
993 $this->triggerSyncJobs( $n, $logger );
994 }
995 } catch ( JobQueueError $e ) {
996 // Do not make the site unavailable (T88312)
997 MWExceptionHandler::logException( $e );
998 }
999 }
1000
1001 /**
1002 * @param int $n Number of jobs to try to run
1003 * @param LoggerInterface $runJobsLogger
1004 */
1005 private function triggerSyncJobs( $n, LoggerInterface $runJobsLogger ) {
1006 $trxProfiler = Profiler::instance()->getTransactionProfiler();
1007 $old = $trxProfiler->setSilenced( true );
1008 try {
1009 $runner = new JobRunner( $runJobsLogger );
1010 $runner->run( [ 'maxJobs' => $n ] );
1011 } finally {
1012 $trxProfiler->setSilenced( $old );
1013 }
1014 }
1015
1016 /**
1017 * @param int $n Number of jobs to try to run
1018 * @param LoggerInterface $runJobsLogger
1019 * @return bool Success
1020 */
1021 private function triggerAsyncJobs( $n, LoggerInterface $runJobsLogger ) {
1022 // Do not send request if there are probably no jobs
1023 $group = JobQueueGroup::singleton();
1024 if ( !$group->queuesHaveJobs( JobQueueGroup::TYPE_DEFAULT ) ) {
1025 return true;
1026 }
1027
1028 $query = [ 'title' => 'Special:RunJobs',
1029 'tasks' => 'jobs', 'maxjobs' => $n, 'sigexpiry' => time() + 5 ];
1030 $query['signature'] = SpecialRunJobs::getQuerySignature(
1031 $query, $this->config->get( 'SecretKey' ) );
1032
1033 $errno = $errstr = null;
1034 $info = wfParseUrl( $this->config->get( 'CanonicalServer' ) );
1035 $host = $info ? $info['host'] : null;
1036 $port = 80;
1037 if ( isset( $info['scheme'] ) && $info['scheme'] == 'https' ) {
1038 $host = "tls://" . $host;
1039 $port = 443;
1040 }
1041 if ( isset( $info['port'] ) ) {
1042 $port = $info['port'];
1043 }
1044
1045 Wikimedia\suppressWarnings();
1046 $sock = $host ? fsockopen(
1047 $host,
1048 $port,
1049 $errno,
1050 $errstr,
1051 // If it takes more than 100ms to connect to ourselves there is a problem...
1052 0.100
1053 ) : false;
1054 Wikimedia\restoreWarnings();
1055
1056 $invokedWithSuccess = true;
1057 if ( $sock ) {
1058 $special = SpecialPageFactory::getPage( 'RunJobs' );
1059 $url = $special->getPageTitle()->getCanonicalURL( $query );
1060 $req = (
1061 "POST $url HTTP/1.1\r\n" .
1062 "Host: {$info['host']}\r\n" .
1063 "Connection: Close\r\n" .
1064 "Content-Length: 0\r\n\r\n"
1065 );
1066
1067 $runJobsLogger->info( "Running $n job(s) via '$url'" );
1068 // Send a cron API request to be performed in the background.
1069 // Give up if this takes too long to send (which should be rare).
1070 stream_set_timeout( $sock, 2 );
1071 $bytes = fwrite( $sock, $req );
1072 if ( $bytes !== strlen( $req ) ) {
1073 $invokedWithSuccess = false;
1074 $runJobsLogger->error( "Failed to start cron API (socket write error)" );
1075 } else {
1076 // Do not wait for the response (the script should handle client aborts).
1077 // Make sure that we don't close before that script reaches ignore_user_abort().
1078 $start = microtime( true );
1079 $status = fgets( $sock );
1080 $sec = microtime( true ) - $start;
1081 if ( !preg_match( '#^HTTP/\d\.\d 202 #', $status ) ) {
1082 $invokedWithSuccess = false;
1083 $runJobsLogger->error( "Failed to start cron API: received '$status' ($sec)" );
1084 }
1085 }
1086 fclose( $sock );
1087 } else {
1088 $invokedWithSuccess = false;
1089 $runJobsLogger->error( "Failed to start cron API (socket error $errno): $errstr" );
1090 }
1091
1092 return $invokedWithSuccess;
1093 }
1094 }