Merge pull request #8 from newfold-labs/fix/sso-persistent-cache

Fallback to an option when transients fail

composer.json

@@ -37,5 +37,13 @@
   },
   "require": {
     "newfold-labs/wp-module-data": ">=2.1"
+  },
+  "require-dev": {
+    "newfold-labs/wp-php-standards": "^1.2"
+  },
+  "config": {
+    "allow-plugins": {
+      "dealerdirect/phpcodesniffer-composer-installer": true
+    }
   }
 }

includes/SSO_Helpers_Legacy.php

@@ -22,6 +22,9 @@ public static function handleLegacyLogin( $nonce, $salt ) {
 			exit;
 		}
 
+		$has_epoch = preg_match( '/-e(\d+)$/', $nonce, $epoch );
+		$expired   = ( $has_epoch && ( time() - $epoch[1] ) > 300 ) ? true : false;
+
 		// Too many failed attempts
 		if ( self::shouldThrottle() ) {
 			self::triggerFailure();
@@ -36,8 +39,13 @@ public static function handleLegacyLogin( $nonce, $salt ) {
 		}
 
 		// Validate token
-		$token = substr( base64_encode( hash( 'sha256', $nonce . $salt, false ) ), 0, 64 );
-		if ( get_transient( 'sso_token' ) !== $token ) {
+		$token        = substr( base64_encode( hash( 'sha256', $nonce . $salt, false ) ), 0, 64 );
+		$stored_token = get_transient( 'sso_token' );
+		if ( false === $stored_token ) {
+			$stored_token = get_option( 'sso_token' );
+			delete_option( 'sso_token' );
+		}
+		if ( $expired || $stored_token !== $token ) {
 			self::triggerFailure();
 			exit;
 		}
@@ -69,13 +77,17 @@ public static function getUser() {
 
 		// If user wasn't found, find first admin user
 		if ( ! $user ) {
-			$users = get_users( array( 'role' => 'administrator', 'number' => 1 ) );
+			$users = get_users(
+				array(
+					'role'   => 'administrator',
+					'number' => 1,
+				)
+			);
 			if ( isset( $users[0] ) && is_a( $users[0], 'WP_User' ) ) {
 				$user = $users[0];
 			}
 		}
 
 		return $user;
 	}
-
 }