Introducing wp_safe_redirect(). fixes #4606 for trunk

git-svn-id: http://svn.automattic.com/wordpress/trunk@6131 1a063a9b-81f0-0310-95a4-ce76da25c4cd
aredridel · Sep 18, 2007 · dbd3d07 · dbd3d07
1 parent bbe284b
commit dbd3d07
Show file tree

Hide file tree

Showing 3 changed files with 21 additions and 3 deletions.
diff --git a/wp-includes/pluggable.php b/wp-includes/pluggable.php
@@ -399,6 +399,24 @@ function wp_redirect($location, $status = 302) {
 }
 endif;
 
+if ( !function_exists('wp_safe_redirect') ) :
+/**
+ * performs a safe (local) redirect, using wp_redirect()
+ * @return void
+ **/
+function wp_safe_redirect($location, $status = 302) {
+	if ( $location{0} == '/' ) {
+		if ( $location{1} == '/' )
+			$location = get_option('home') . '/';
+	} else {
+		if ( substr($location, 0, strlen(get_option('home'))) != get_option('home') )
+			$location = get_option('home') . '/';
+	}
+
+	wp_redirect($location, $status);
+}
+endif;
+
 if ( !function_exists('wp_get_cookie_login') ):
 function wp_get_cookie_login() {
 	if ( empty($_COOKIE[USER_COOKIE]) || empty($_COOKIE[PASS_COOKIE]) )

diff --git a/wp-login.php b/wp-login.php
@@ -78,7 +78,7 @@ function focusit() {
 	if ( isset( $_REQUEST['redirect_to'] ) )
 		$redirect_to = $_REQUEST['redirect_to'];
 
-	wp_redirect($redirect_to);
+	wp_safe_redirect($redirect_to);
 	exit();
 
 break;
@@ -324,7 +324,7 @@ function focusit() {
 			if ( !$using_cookie )
 				wp_setcookie($user_login, $user_pass, false, '', '', $rememberme);
 			do_action('wp_login', $user_login);
-			wp_redirect($redirect_to);
+			wp_safe_redirect($redirect_to);
 			exit();
 		} else {
 			if ( $using_cookie )

diff --git a/wp-pass.php b/wp-pass.php
@@ -7,5 +7,5 @@
 // 10 days
 setcookie('wp-postpass_' . COOKIEHASH, $_POST['post_password'], time() + 864000, COOKIEPATH);
 
-wp_redirect(wp_get_referer());
+wp_safe_redirect(wp_get_referer());
 ?>