includes/functions-auth.php

   1 <?php\r
   2 // Check for valid user. Returns true or an error message\r
   3 function yourls_is_valid_user() {\r
   4         static $valid = false;\r
   5         \r
   6         if( $valid )\r
   7                 return true;\r
   8 \r
   9         // Logout request\r
  10         if( isset( $_GET['action'] ) && $_GET['action'] == 'logout') {\r
  11                 yourls_store_cookie( null );\r
  12                 return 'Logged out successfully';\r
  13         }\r
  14         \r
  15         // Check cookies or login request. Login form has precedence.\r
  16         global $yourls_user_passwords;\r
  17         \r
  18         // Determine auth method and check credentials\r
  19         if\r
  20                 // API only: Secure (no login or pwd) and time limited token\r
  21                 // ?timestamp=12345678&signature=md5(totoblah12345678)\r
  22                 ( yourls_is_API() &&\r
  23                   isset($_REQUEST['timestamp']) && !empty($_REQUEST['timestamp']) &&\r
  24                   isset($_REQUEST['signature']) && !empty($_REQUEST['signature'])\r
  25                 )\r
  26                 {\r
  27                         $valid = yourls_check_signature_timestamp();\r
  28                 }\r
  29                 \r
  30         elseif\r
  31                 // API only: Secure (no login or pwd)\r
  32                 // ?signature=md5(totoblah)\r
  33                 ( yourls_is_API() &&\r
  34                   !isset($_REQUEST['timestamp']) &&\r
  35                   isset($_REQUEST['signature']) && !empty($_REQUEST['signature'])\r
  36                 )\r
  37                 {\r
  38                         $valid = yourls_check_signature();\r
  39                 }\r
  40         \r
  41         elseif\r
  42                 // API or normal: login with username & pwd\r
  43                 ( isset($_REQUEST['username']) && isset($_REQUEST['password'])\r
  44                   && !empty( $_REQUEST['username'] ) && !empty( $_REQUEST['password']  ) )\r
  45                 {\r
  46                         $valid = yourls_check_username_password();\r
  47                 }\r
  48         \r
  49         elseif\r
  50                 // Normal only: cookies\r
  51                 ( !yourls_is_API() && \r
  52                   isset($_COOKIE['yourls_username']) && isset($_COOKIE['yourls_password']) )\r
  53                 {\r
  54                         $valid = yourls_check_auth_cookie();\r
  55                 }\r
  56 \r
  57         // Login for the win!\r
  58         if ( $valid ) {\r
  59                 // (Re)store encrypted cookie and tell it's ok\r
  60                 if ( !yourls_is_API() ) // No need to store a cookie when used in API mode.\r
  61                         yourls_store_cookie( YOURLS_USER );\r
  62                 return true;\r
  63         }\r
  64         \r
  65         // Login failed\r
  66         if ( isset($_REQUEST['username']) || isset($_REQUEST['password']) ) {\r
  67                 return 'Invalid username or password';\r
  68         } else {\r
  69                 return 'Please log in';\r
  70         }\r
  71 }\r
  72 \r
  73 // Check auth against list of login=>pwd. Sets user if applicable, returns bool\r
  74 function yourls_check_username_password() {\r
  75         global $yourls_user_passwords;\r
  76         if( isset( $yourls_user_passwords[ $_REQUEST['username'] ] ) && $yourls_user_passwords[ $_REQUEST['username'] ] == $_REQUEST['password'] ) {\r
  77                 yourls_set_user( $_REQUEST['username'] );\r
  78                 return true;\r
  79         }\r
  80         return false;\r
  81 }\r
  82 \r
  83 // Check auth against encrypted COOKIE data. Sets user if applicable, returns bool\r
  84 function yourls_check_auth_cookie() {\r
  85         global $yourls_user_passwords;\r
  86         foreach( $yourls_user_passwords as $valid_user => $valid_password ) {\r
  87                 if( \r
  88                         yourls_salt($valid_user) == $_COOKIE['yourls_username']\r
  89                         && yourls_salt($valid_password) == $_COOKIE['yourls_password'] \r
  90                 ) {\r
  91                         yourls_set_user( $valid_user );\r
  92                         return true;\r
  93                 }\r
  94         }\r
  95         return false;\r
  96 }\r
  97 \r
  98 // Check auth against signature and timestamp. Sets user if applicable, returns bool\r
  99 function yourls_check_signature_timestamp() {\r
 100         // Timestamp in PHP : time()\r
 101         // Timestamp in JS: parseInt(new Date().getTime() / 1000)\r
 102         global $yourls_user_passwords;\r
 103         foreach( $yourls_user_passwords as $valid_user => $valid_password ) {\r
 104                 if (\r
 105                         (\r
 106                                 md5( $_REQUEST['timestamp'].yourls_auth_signature( $valid_user ) ) == $_REQUEST['signature']\r
 107                                 or\r
 108                                 md5( yourls_auth_signature( $valid_user ).$_REQUEST['timestamp'] ) == $_REQUEST['signature']\r
 109                         )\r
 110                         &&\r
 111                         yourls_check_timestamp( $_REQUEST['timestamp'] )\r
 112                         ) {\r
 113                         yourls_set_user( $valid_user );\r
 114                         return true;\r
 115                 }\r
 116         }\r
 117         return false;\r
 118 }\r
 119 \r
 120 // Check auth against signature. Sets user if applicable, returns bool\r
 121 function yourls_check_signature() {\r
 122         global $yourls_user_passwords;\r
 123         foreach( $yourls_user_passwords as $valid_user => $valid_password ) {\r
 124                 if ( yourls_auth_signature( $valid_user ) == $_REQUEST['signature'] ) {\r
 125                         yourls_set_user( $valid_user );\r
 126                         return true;\r
 127                 }\r
 128         }\r
 129         return false;\r
 130 }\r
 131 \r
 132 // Generate secret signature hash\r
 133 function yourls_auth_signature( $username = false ) {\r
 134         if( !$username && defined('YOURLS_USER') ) {\r
 135                 $username = YOURLS_USER;\r
 136         }\r
 137         return ( $username ? substr( yourls_salt( $username ), 0, 10 ) : 'Cannot generate auth signature: no username' );\r
 138 }\r
 139 \r
 140 // Check a timestamp is from the past and not too old\r
 141 function yourls_check_timestamp( $time ) {\r
 142         $now = time();\r
 143         return ( $now >= $time && ceil( $now - $time ) < YOURLS_NONCE_LIFE );\r
 144 }\r
 145 \r
 146 // Store new cookie. No $user will delete the cookie.\r
 147 function yourls_store_cookie( $user = null ) {\r
 148         if( !$user ) {\r
 149                 $pass = null;\r
 150                 $time = time() - 3600;\r
 151         } else {\r
 152                 global $yourls_user_passwords;\r
 153                 if( isset($yourls_user_passwords[$user]) ) {\r
 154                         $pass = $yourls_user_passwords[$user];\r
 155                 } else {\r
 156                         die('Stealing cookies?'); // This should never happen\r
 157                 }\r
 158                 $time = time() + YOURLS_COOKIE_LIFE;\r
 159         }\r
 160         \r
 161         $domain   = yourls_apply_filter( 'setcookie_domain',   parse_url( YOURLS_SITE, 1 ) );\r
 162         $secure   = yourls_apply_filter( 'setcookie_secure',   yourls_is_ssl() );\r
 163         $httponly = yourls_apply_filter( 'setcookie_httponly', true );\r
 164                 \r
 165         if ( !headers_sent() ) {\r
 166                 // Set httponly if the php version is >= 5.2.0\r
 167                 if( version_compare( phpversion(), '5.2.0', 'ge' ) ) {\r
 168                         setcookie('yourls_username', yourls_salt( $user ), $time, '/', $domain, $secure, $httponly );\r
 169                         setcookie('yourls_password', yourls_salt( $pass ), $time, '/', $domain, $secure, $httponly );\r
 170                 } else {\r
 171                         setcookie('yourls_username', yourls_salt( $user ), $time, '/', $domain, $secure );\r
 172                         setcookie('yourls_password', yourls_salt( $pass ), $time, '/', $domain, $secure );\r
 173                 }\r
 174         }\r
 175 }\r
 176 \r
 177 // Set user name\r
 178 function yourls_set_user( $user ) {\r
 179         if( !defined('YOURLS_USER') )\r
 180                 define('YOURLS_USER', $user);\r
 181 }\r