Merge pull request openedx#18988 from edx/robrap/ARCH-238-login-redirect

ARCH-238: Add ability to redirect to subdomain for login.

common/djangoapps/student/helpers.py

@@ -273,7 +273,7 @@ def get_next_url_for_login_page(request):
 
     If THIRD_PARTY_AUTH_HINT is set, then `tpa_hint=<hint>` is added as a query parameter.
     """
-    redirect_to = get_redirect_to(request)
+    redirect_to = _get_redirect_to(request)
     if not redirect_to:
         try:
             redirect_to = reverse('dashboard')
@@ -308,7 +308,7 @@ def get_next_url_for_login_page(request):
     return redirect_to
 
 
-def get_redirect_to(request):
+def _get_redirect_to(request):
     """
     Determine the redirect url and return if safe
     :argument
@@ -325,7 +325,7 @@ def get_redirect_to(request):
     # get information about a user on edx.org. In any such case drop the parameter.
     if redirect_to:
         mime_type, _ = mimetypes.guess_type(redirect_to, strict=False)
-        if not http.is_safe_url(redirect_to, allowed_hosts={request.get_host()}, require_https=True):
+        if not _is_safe_redirect(request, redirect_to):
             log.warning(
                 u'Unsafe redirect parameter detected after login page: %(redirect_to)r',
                 {"redirect_to": redirect_to}
@@ -368,6 +368,19 @@ def get_redirect_to(request):
     return redirect_to
 
 
+def _is_safe_redirect(request, redirect_to):
+    """
+    Determine if the given redirect URL/path is safe for redirection.
+    """
+    request_host = request.get_host()  # e.g. 'courses.edx.org'
+
+    login_redirect_whitelist = set(getattr(settings, 'LOGIN_REDIRECT_WHITELIST', []))
+    login_redirect_whitelist.add(request_host)
+
+    is_safe_url = http.is_safe_url(redirect_to, allowed_hosts=login_redirect_whitelist, require_https=True)
+    return is_safe_url
+
+
 def generate_activation_email_context(user, registration):
     """
     Constructs a dictionary for use in activation email contexts

common/djangoapps/student/tests/test_helpers.py

@@ -67,8 +67,11 @@ def test_unsafe_next(self, log_level, log_name, unsafe_url, http_accept, user_ag
     @ddt.data(
         ('/dashboard', 'testserver', '/dashboard'),
         ('https://edx.org/courses', 'edx.org', 'https://edx.org/courses'),
+        ('https://test.edx.org/courses', 'edx.org', 'https://test.edx.org/courses'),
+        ('https://test2.edx.org/courses', 'edx.org', 'https://test2.edx.org/courses'),
     )
     @ddt.unpack
+    @override_settings(LOGIN_REDIRECT_WHITELIST=['test.edx.org', 'test2.edx.org'])
     def test_safe_next(self, url, host, expected_url):
         """ Test safe next parameter """
         req = self.request.get(reverse("login") + "?next={url}".format(url=url), HTTP_HOST=host)

lms/envs/aws.py

@@ -421,6 +421,9 @@
 # by end users.
 CSRF_COOKIE_SECURE = ENV_TOKENS.get('CSRF_COOKIE_SECURE', False)
 
+# Whitelist of domains to which the login/logout pages will redirect.
+LOGIN_REDIRECT_WHITELIST = ENV_TOKENS.get('LOGIN_REDIRECT_WHITELIST', LOGIN_REDIRECT_WHITELIST)
+
 ############# CORS headers for cross-domain requests #################
 
 if FEATURES.get('ENABLE_CORS_HEADERS') or FEATURES.get('ENABLE_CROSS_DOMAIN_CSRF_COOKIE'):

lms/envs/common.py

@@ -2508,6 +2508,8 @@ def _make_locale_paths(settings):
 # to simulate cross-domain requests.
 XDOMAIN_PROXY_CACHE_TIMEOUT = 60 * 15
 
+LOGIN_REDIRECT_WHITELIST = []
+
 ###################### Registration ##################################
 
 # For each of the fields, give one of the following values:

lms/envs/devstack.py

@@ -228,6 +228,8 @@ def should_show_debug_toolbar(request):
 CORS_ORIGIN_WHITELIST = ()
 CORS_ORIGIN_ALLOW_ALL = True
 
+LOGIN_REDIRECT_WHITELIST = []
+
 ###################### JWTs ######################
 JWT_AUTH.update({
     'JWT_ISSUER': OAUTH_OIDC_ISSUER,