Passed
Pull Request — master (#1410)
by
unknown
38:34 queued 03:31
created

DeviceCodeGrant::respondToAccessTokenRequest()   B

Complexity

Conditions 6
Paths 10

Size

Total Lines 52
Code Lines 25

Duplication

Lines 0
Ratio 0 %

Code Coverage

Tests 25
CRAP Score 6

Importance

Changes 1
Bugs 0 Features 0
Metric Value
cc 6
eloc 25
c 1
b 0
f 0
nc 10
nop 3
dl 0
loc 52
ccs 25
cts 25
cp 1
crap 6
rs 8.8977

How to fix   Long Method   

Long Method

Small methods make your code easier to understand, in particular if combined with a good name. Besides, if your method is small, finding a good name is usually much easier.

For example, if you find yourself adding comments to a method's body, this is usually a good sign to extract the commented part to a new method, and use the comment as a starting point when coming up with a good name for this new method.

Commonly applied refactorings include:

1
<?php
2
3
/**
4
 * OAuth 2.0 Device Code grant.
5
 *
6
 * @author      Andrew Millington <[email protected]>
7
 * @copyright   Copyright (c) Alex Bilbie
8
 * @license     http://mit-license.org/
9
 *
10
 * @link        https://github.com/thephpleague/oauth2-server
11
 */
12
13
declare(strict_types=1);
14
15
namespace League\OAuth2\Server\Grant;
16
17
use DateInterval;
18
use DateTimeImmutable;
19
use Error;
20
use Exception;
21
use League\OAuth2\Server\Entities\ClientEntityInterface;
22
use League\OAuth2\Server\Entities\DeviceCodeEntityInterface;
23
use League\OAuth2\Server\Entities\ScopeEntityInterface;
24
use League\OAuth2\Server\Exception\OAuthServerException;
25
use League\OAuth2\Server\Exception\UniqueTokenIdentifierConstraintViolationException;
26
use League\OAuth2\Server\Repositories\DeviceCodeRepositoryInterface;
27
use League\OAuth2\Server\Repositories\RefreshTokenRepositoryInterface;
28
use League\OAuth2\Server\RequestEvent;
29
use League\OAuth2\Server\ResponseTypes\DeviceCodeResponse;
30
use League\OAuth2\Server\ResponseTypes\ResponseTypeInterface;
31
use Psr\Http\Message\ServerRequestInterface;
32
use TypeError;
33
34
use function is_null;
35
use function random_int;
36
use function strlen;
37
use function time;
38
39
/**
40
 * Device Code grant class.
41
 */
42
class DeviceCodeGrant extends AbstractGrant
43
{
44
    protected DeviceCodeRepositoryInterface $deviceCodeRepository;
45
    private bool $includeVerificationUriComplete = false;
46
    private bool $intervalVisibility = false;
47
    private string $verificationUri;
48
49 17
    public function __construct(
50
        DeviceCodeRepositoryInterface $deviceCodeRepository,
51
        RefreshTokenRepositoryInterface $refreshTokenRepository,
52
        private DateInterval $deviceCodeTTL,
53
        string $verificationUri,
54
        private readonly int $defaultInterval = 5
55
    ) {
56 17
        $this->setDeviceCodeRepository($deviceCodeRepository);
57 17
        $this->setRefreshTokenRepository($refreshTokenRepository);
58
59 17
        $this->refreshTokenTTL = new DateInterval('P1M');
60
61 17
        $this->setVerificationUri($verificationUri);
62
    }
63
64
    /**
65
     * {@inheritdoc}
66
     */
67 2
    public function canRespondToDeviceAuthorizationRequest(ServerRequestInterface $request): bool
68
    {
69 2
        return true;
70
    }
71
72
    /**
73
     * {@inheritdoc}
74
     */
75 7
    public function respondToDeviceAuthorizationRequest(ServerRequestInterface $request): DeviceCodeResponse
76
    {
77 7
        $clientId = $this->getRequestParameter(
78 7
            'client_id',
79 7
            $request,
80 7
            $this->getServerParameter('PHP_AUTH_USER', $request)
81 7
        );
82
83 7
        if ($clientId === null) {
84 2
            throw OAuthServerException::invalidRequest('client_id');
85
        }
86
87 5
        $client = $this->getClientEntityOrFail($clientId, $request);
88
89 4
        $scopes = $this->validateScopes($this->getRequestParameter('scope', $request, $this->defaultScope));
90
91 4
        $deviceCodeEntity = $this->issueDeviceCode(
92 4
            $this->deviceCodeTTL,
93 4
            $client,
94 4
            $this->verificationUri,
95 4
            $scopes
96 4
        );
97
98 4
        $response = new DeviceCodeResponse();
99
100 4
        if ($this->includeVerificationUriComplete === true) {
101 1
            $response->includeVerificationUriComplete();
102
        }
103
104 4
        if ($this->intervalVisibility === true) {
105
            $response->includeInterval();
106 4
        }
107
108
        $response->setDeviceCodeEntity($deviceCodeEntity);
109
110
        return $response;
111
    }
112 3
113
    /**
114 3
     * {@inheritdoc}
115
     */
116 3
    public function completeDeviceAuthorizationRequest(string $deviceCode, string $userId, bool $userApproved): void
117
    {
118
        $deviceCode = $this->deviceCodeRepository->getDeviceCodeEntityByDeviceCode($deviceCode);
119
120 3
        if ($deviceCode instanceof DeviceCodeEntityInterface === false) {
121
            throw OAuthServerException::invalidRequest('device_code', 'Device code does not exist');
122
        }
123
124 3
        if ($userId === '') {
125 3
            throw OAuthServerException::invalidRequest('user_id', 'User ID is required');
126
        }
127 3
128
        $deviceCode->setUserIdentifier($userId);
129
        $deviceCode->setUserApproved($userApproved);
130
131
        $this->deviceCodeRepository->persistDeviceCode($deviceCode);
132
    }
133 7
134
    /**
135
     * {@inheritdoc}
136
     */
137
    public function respondToAccessTokenRequest(
138
        ServerRequestInterface $request,
139 7
        ResponseTypeInterface $responseType,
140 6
        DateInterval $accessTokenTTL
141 6
    ): ResponseTypeInterface {
142
        // Validate request
143 3
        $client = $this->validateClient($request);
144 3
        $scopes = $this->validateScopes($this->getRequestParameter('scope', $request, $this->defaultScope));
145
        $deviceCodeEntity = $this->validateDeviceCode($request, $client);
146
        $shouldSlowDown = false;
147 3
148 1
        if ($this->deviceCodePolledTooSoon($deviceCodeEntity) === true) {
149
            $deviceCodeEntity->setInterval($deviceCodeEntity->getInterval() + 5);
150
151 2
            $shouldSlowDown = true;
152 1
        }
153
154
        $deviceCodeEntity->setLastPolledAt(new DateTimeImmutable());
155
        $this->deviceCodeRepository->persistDeviceCode($deviceCodeEntity);
156 1
157
        if ($shouldSlowDown) {
158
            throw OAuthServerException::slowDown();
159 1
        }
160 1
161 1
        // If device code has no user associated, respond with pending
162
        if (is_null($deviceCodeEntity->getUserIdentifier())) {
163
            throw OAuthServerException::authorizationPending();
164 1
        }
165
166 1
        if ($deviceCodeEntity->getUserApproved() === false) {
167 1
            throw OAuthServerException::accessDenied();
168 1
        }
169
170
        // Finalize the requested scopes
171 1
        $finalizedScopes = $this->scopeRepository->finalizeScopes($scopes, $this->getIdentifier(), $client, $deviceCodeEntity->getUserIdentifier());
172
173 1
        // Issue and persist new access token
174
        $accessToken = $this->issueAccessToken($accessTokenTTL, $client, $deviceCodeEntity->getUserIdentifier(), $finalizedScopes);
175
        $this->getEmitter()->emit(new RequestEvent(RequestEvent::ACCESS_TOKEN_ISSUED, $request));
176
        $responseType->setAccessToken($accessToken);
177
178
        // Issue and persist new refresh token if given
179 6
        $refreshToken = $this->issueRefreshToken($accessToken);
180
181 6
        if ($refreshToken !== null) {
182
            $this->getEmitter()->emit(new RequestEvent(RequestEvent::REFRESH_TOKEN_ISSUED, $request));
183 6
            $responseType->setRefreshToken($refreshToken);
184 1
        }
185
186
        $this->deviceCodeRepository->revokeDeviceCode($deviceCodeEntity->getIdentifier());
187 5
188 5
        return $responseType;
189 5
    }
190
191 5
    /**
192
     * @throws OAuthServerException
193
     */
194
    protected function validateDeviceCode(ServerRequestInterface $request, ClientEntityInterface $client): DeviceCodeEntityInterface
195
    {
196
        $deviceCode = $this->getRequestParameter('device_code', $request);
197 5
198 1
        if (is_null($deviceCode)) {
199
            throw OAuthServerException::invalidRequest('device_code');
200
        }
201 4
202
        $deviceCodeEntity = $this->deviceCodeRepository->getDeviceCodeEntityByDeviceCode(
203
            $deviceCode
204
        );
205 4
206
        if ($deviceCodeEntity instanceof DeviceCodeEntityInterface === false) {
207
            $this->getEmitter()->emit(new RequestEvent(RequestEvent::USER_AUTHENTICATION_FAILED, $request));
208
209 4
            throw OAuthServerException::invalidGrant();
210 1
        }
211
212
        if (time() > $deviceCodeEntity->getExpiryDateTime()->getTimestamp()) {
213 3
            throw OAuthServerException::expiredToken('device_code');
214
        }
215
216 4
        if ($this->deviceCodeRepository->isDeviceCodeRevoked($deviceCode) === true) {
217
            throw OAuthServerException::invalidRequest('device_code', 'Device code has been revoked');
218 4
        }
219
220
        if ($deviceCodeEntity->getClient()->getIdentifier() !== $client->getIdentifier()) {
221
            throw OAuthServerException::invalidRequest('device_code', 'Device code was not issued to this client');
222
        }
223
224 17
        return $deviceCodeEntity;
225
    }
226 17
227
    private function deviceCodePolledTooSoon(DeviceCodeEntityInterface $deviceCodeEntity): bool
228
    {
229
        $lastPoll = $deviceCodeEntity->getLastPolledAt();
230
231
        return $lastPoll !== null && $lastPoll->getTimestamp() + $deviceCodeEntity->getInterval() > time();
232 8
    }
233
234 8
    /**
235
     * Set the verification uri
236
     */
237 17
    public function setVerificationUri(string $verificationUri): void
238
    {
239 17
        $this->verificationUri = $verificationUri;
240
    }
241
242
    /**
243
     * {@inheritdoc}
244
     */
245
    public function getIdentifier(): string
246
    {
247
        return 'urn:ietf:params:oauth:grant-type:device_code';
248
    }
249
250 4
    private function setDeviceCodeRepository(DeviceCodeRepositoryInterface $deviceCodeRepository): void
251
    {
252
        $this->deviceCodeRepository = $deviceCodeRepository;
253
    }
254
255
    /**
256 4
     * Issue a device code.
257
     *
258 4
     * @param ScopeEntityInterface[] $scopes
259 4
     *
260 4
     * @throws OAuthServerException
261 4
     * @throws UniqueTokenIdentifierConstraintViolationException
262
     */
263 4
    protected function issueDeviceCode(
264 1
        DateInterval $deviceCodeTTL,
265
        ClientEntityInterface $client,
266
        string $verificationUri,
267 4
        array $scopes = [],
268 4
    ): DeviceCodeEntityInterface {
269
        $maxGenerationAttempts = self::MAX_RANDOM_TOKEN_GENERATION_ATTEMPTS;
270
271 4
        $deviceCode = $this->deviceCodeRepository->getNewDeviceCode();
272 4
        $deviceCode->setExpiryDateTime((new DateTimeImmutable())->add($deviceCodeTTL));
273 4
        $deviceCode->setClient($client);
274
        $deviceCode->setVerificationUri($verificationUri);
275
        $deviceCode->setInterval($this->defaultInterval);
276 4
277
        foreach ($scopes as $scope) {
278 4
            $deviceCode->addScope($scope);
279
        }
280
281
        while ($maxGenerationAttempts-- > 0) {
282
            $deviceCode->setIdentifier($this->generateUniqueIdentifier());
283
            $deviceCode->setUserCode($this->generateUserCode());
284
285
            try {
286
                $this->deviceCodeRepository->persistDeviceCode($deviceCode);
287
288
                return $deviceCode;
289
            } catch (UniqueTokenIdentifierConstraintViolationException $e) {
290
                if ($maxGenerationAttempts === 0) {
291
                    throw $e;
292
                }
293
            }
294
        }
295 4
296
        // This should never be hit. It is here to work around a PHPStan false error
297
        return $deviceCode;
298 4
    }
299 4
300
    /**
301 4
     * Generate a new user code.
302 4
     *
303
     * @throws OAuthServerException
304
     */
305 4
    protected function generateUserCode(int $length = 8): string
306
    {
307
        try {
308
            $userCode = '';
309
            $userCodeCharacters = 'BCDFGHJKLMNPQRSTVWXZ';
310
311
            while (strlen($userCode) < $length) {
312
                $userCode .= $userCodeCharacters[random_int(0, 19)];
313
            }
314
315
            return $userCode;
316 1
            // @codeCoverageIgnoreStart
317
        } catch (TypeError | Error $e) {
318 1
            throw OAuthServerException::serverError('An unexpected error has occurred', $e);
319
        } catch (Exception $e) {
320
            // If you get this message, the CSPRNG failed hard.
321 4
            throw OAuthServerException::serverError('Could not generate a random string', $e);
322
        }
323 4
        // @codeCoverageIgnoreEnd
324
    }
325
326 1
    public function setIntervalVisibility(bool $intervalVisibility): void
327
    {
328 1
        $this->intervalVisibility = $intervalVisibility;
329
    }
330
331
    public function getIntervalVisibility(): bool
332
    {
333
        return $this->intervalVisibility;
334
    }
335
336
    public function setIncludeVerificationUriComplete(bool $includeVerificationUriComplete): void
337
    {
338
        $this->includeVerificationUriComplete = $includeVerificationUriComplete;
339
    }
340
}
341