Version 1

[yaffs-website] / web / core / lib / Drupal / Core / Routing / LocalAwareRedirectResponseTrait.php

diff --git a/web/core/lib/Drupal/Core/Routing/LocalAwareRedirectResponseTrait.php b/web/core/lib/Drupal/Core/Routing/LocalAwareRedirectResponseTrait.php
new file mode 100644 (file)
index 0000000..d30ebef
--- /dev/null
+++ b/web/core/lib/Drupal/Core/Routing/LocalAwareRedirectResponseTrait.php
@@ -0,0 +1,52 @@
+<?php
+
+namespace Drupal\Core\Routing;
+
+use Drupal\Component\Utility\UrlHelper;
+
+/**
+ * Provides a trait which ensures that a URL is safe to redirect to.
+ */
+trait LocalAwareRedirectResponseTrait {
+
+  /**
+   * The request context.
+   *
+   * @var \Drupal\Core\Routing\RequestContext
+   */
+  protected $requestContext;
+
+  /**
+   * {@inheritdoc}
+   */
+  protected function isLocal($url) {
+    return !UrlHelper::isExternal($url) || UrlHelper::externalIsLocal($url, $this->getRequestContext()->getCompleteBaseUrl());
+  }
+
+  /**
+   * Returns the request context.
+   *
+   * @return \Drupal\Core\Routing\RequestContext
+   */
+  protected function getRequestContext() {
+    if (!isset($this->requestContext)) {
+      $this->requestContext = \Drupal::service('router.request_context');
+    }
+    return $this->requestContext;
+  }
+
+  /**
+   * Sets the request context.
+   *
+   * @param \Drupal\Core\Routing\RequestContext $request_context
+   *   The request context.
+   *
+   * @return $this
+   */
+  public function setRequestContext(RequestContext $request_context) {
+    $this->requestContext = $request_context;
+
+    return $this;
+  }
+
+}