src/Network/FKOAuthDataStore.php

   1 <?php
   2
   3 /**
   4  * @file src/Protocol/FKOAuthDataStore.php
   5  * OAuth server
   6  * Based on oauth2-php <http://code.google.com/p/oauth2-php/>
   7  *
   8  */
   9
  10 namespace Friendica\Network;
  11
  12 use Friendica\App;
  13 use Friendica\Core\Config;
  14 use Friendica\Core\System;
  15 use Friendica\Database\DBM;
  16 use dba;
  17 use OAuthDataStore;
  18
  19 define('REQUEST_TOKEN_DURATION', 300);
  20 define('ACCESS_TOKEN_DURATION', 31536000);
  21
  22 require_once 'include/dba.php';
  23
  24 require_once "library/OAuth1.php";
  25 require_once "library/oauth2-php/lib/OAuth2.inc";
  26
  27 /**
  28  * @brief OAuthDataStore class
  29  */
  30 class FKOAuthDataStore extends OAuthDataStore
  31 {
  32         /**
  33          * @return string
  34          */
  35         private static function genToken()
  36         {
  37                 return md5(base64_encode(pack('N6', mt_rand(), mt_rand(), mt_rand(), mt_rand(), mt_rand(), uniqid())));
  38         }
  39
  40         /**
  41          * @param string $consumer_key key
  42          * @return mixed
  43          */
  44         public function lookup_consumer($consumer_key)
  45         {
  46                 logger(__function__ . ":" . $consumer_key);
  47
  48                 $s = dba::select('clients', array('client_id', 'pw', 'redirect_uri'), array('client_id' => $consumer_key));
  49                 $r = dba::inArray($s);
  50
  51                 if (DBM::is_result($r)) {
  52                         return new \OAuthConsumer($r[0]['client_id'], $r[0]['pw'], $r[0]['redirect_uri']);
  53                 }
  54
  55                 return null;
  56         }
  57
  58         /**
  59          * @param string $consumer   consumer
  60          * @param string $token_type type
  61          * @param string $token      token
  62          * @return mixed
  63          */
  64         public function lookup_token($consumer, $token_type, $token)
  65         {
  66                 logger(__function__ . ":" . $consumer . ", " . $token_type . ", " . $token);
  67
  68                 $s = dba::select('tokens', array('id', 'secret', 'scope', 'expires', 'uid'), array('client_id' => $consumer->key, 'scope' => $token_type, 'id' => $token));
  69                 $r = dba::inArray($s);
  70
  71                 if (DBM::is_result($r)) {
  72                         $ot = new \OAuthToken($r[0]['id'], $r[0]['secret']);
  73                         $ot->scope = $r[0]['scope'];
  74                         $ot->expires = $r[0]['expires'];
  75                         $ot->uid = $r[0]['uid'];
  76                         return $ot;
  77                 }
  78
  79                 return null;
  80         }
  81
  82         /**
  83          * @param string $consumer  consumer
  84          * @param string $token     token
  85          * @param string $nonce     nonce
  86          * @param string $timestamp timestamp
  87          * @return mixed
  88          */
  89         public function lookup_nonce($consumer, $token, $nonce, $timestamp)
  90         {
  91                 $token = dba::selectFirst('tokens', ['id', 'secret'], ['client_id' => $consumer->key, 'id' => $nonce, 'expires' => $timestamp]);
  92                 if (DBM::is_result($token)) {
  93                         return new \OAuthToken($token['id'], $token['secret']);
  94                 }
  95
  96                 return null;
  97         }
  98
  99         /**
 100          * @param string $consumer consumer
 101          * @param string $callback optional, default null
 102          * @return mixed
 103          */
 104         public function new_request_token($consumer, $callback = null)
 105         {
 106                 logger(__function__ . ":" . $consumer . ", " . $callback);
 107                 $key = self::genToken();
 108                 $sec = self::genToken();
 109
 110                 if ($consumer->key) {
 111                         $k = $consumer->key;
 112                 } else {
 113                         $k = $consumer;
 114                 }
 115
 116                 $r = dba::insert(
 117                         'tokens',
 118                         array(
 119                                 'id' => $key,
 120                                 'secret' => $sec,
 121                                 'client_id' => $k,
 122                                 'scope' => 'request',
 123                                 'expires' => time() + REQUEST_TOKEN_DURATION)
 124                 );
 125
 126                 if (!$r) {
 127                         return null;
 128                 }
 129
 130                 return new \OAuthToken($key, $sec);
 131         }
 132
 133         /**
 134          * @param string $token    token
 135          * @param string $consumer consumer
 136          * @param string $verifier optional, defult null
 137          * @return object
 138          */
 139         public function new_access_token($token, $consumer, $verifier = null)
 140         {
 141                 logger(__function__ . ":" . $token . ", " . $consumer . ", " . $verifier);
 142
 143                 // return a new access token attached to this consumer
 144                 // for the user associated with this token if the request token
 145                 // is authorized
 146                 // should also invalidate the request token
 147
 148                 $ret = null;
 149
 150                 // get user for this verifier
 151                 $uverifier = Config::get("oauth", $verifier);
 152                 logger(__function__ . ":" . $verifier . "," . $uverifier);
 153
 154                 if (is_null($verifier) || ($uverifier !== false)) {
 155                         $key = self::genToken();
 156                         $sec = self::genToken();
 157                         $r = dba::insert(
 158                                 'tokens',
 159                                 array(
 160                                         'id' => $key,
 161                                         'secret' => $sec,
 162                                         'client_id' => $consumer->key,
 163                                         'scope' => 'access',
 164                                         'expires' => time() + ACCESS_TOKEN_DURATION,
 165                                         'uid' => $uverifier)
 166                         );
 167
 168                         if ($r) {
 169                                 $ret = new \OAuthToken($key, $sec);
 170                         }
 171                 }
 172
 173                 dba::delete('tokens', array('id' => $token->key));
 174
 175                 if (!is_null($ret) && !is_null($uverifier)) {
 176                         Config::delete("oauth", $verifier);
 177                 }
 178
 179                 return $ret;
 180         }
 181 }