access_token.py 9.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215
  1. # -*- coding: utf-8 -*-
  2. """
  3. oauthlib.oauth1.rfc5849.endpoints.access_token
  4. ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
  5. This module is an implementation of the access token provider logic of
  6. OAuth 1.0 RFC 5849. It validates the correctness of access token requests,
  7. creates and persists tokens as well as create the proper response to be
  8. returned to the client.
  9. """
  10. import logging
  11. from oauthlib.common import urlencode
  12. from .. import errors
  13. from .base import BaseEndpoint
  14. log = logging.getLogger(__name__)
  15. class AccessTokenEndpoint(BaseEndpoint):
  16. """An endpoint responsible for providing OAuth 1 access tokens.
  17. Typical use is to instantiate with a request validator and invoke the
  18. ``create_access_token_response`` from a view function. The tuple returned
  19. has all information necessary (body, status, headers) to quickly form
  20. and return a proper response. See :doc:`/oauth1/validator` for details on which
  21. validator methods to implement for this endpoint.
  22. """
  23. def create_access_token(self, request, credentials):
  24. """Create and save a new access token.
  25. Similar to OAuth 2, indication of granted scopes will be included as a
  26. space separated list in ``oauth_authorized_realms``.
  27. :param request: OAuthlib request.
  28. :type request: oauthlib.common.Request
  29. :returns: The token as an urlencoded string.
  30. """
  31. request.realms = self.request_validator.get_realms(
  32. request.resource_owner_key, request)
  33. token = {
  34. 'oauth_token': self.token_generator(),
  35. 'oauth_token_secret': self.token_generator(),
  36. # Backport the authorized scopes indication used in OAuth2
  37. 'oauth_authorized_realms': ' '.join(request.realms)
  38. }
  39. token.update(credentials)
  40. self.request_validator.save_access_token(token, request)
  41. return urlencode(token.items())
  42. def create_access_token_response(self, uri, http_method='GET', body=None,
  43. headers=None, credentials=None):
  44. """Create an access token response, with a new request token if valid.
  45. :param uri: The full URI of the token request.
  46. :param http_method: A valid HTTP verb, i.e. GET, POST, PUT, HEAD, etc.
  47. :param body: The request body as a string.
  48. :param headers: The request headers as a dict.
  49. :param credentials: A list of extra credentials to include in the token.
  50. :returns: A tuple of 3 elements.
  51. 1. A dict of headers to set on the response.
  52. 2. The response body as a string.
  53. 3. The response status code as an integer.
  54. An example of a valid request::
  55. >>> from your_validator import your_validator
  56. >>> from oauthlib.oauth1 import AccessTokenEndpoint
  57. >>> endpoint = AccessTokenEndpoint(your_validator)
  58. >>> h, b, s = endpoint.create_access_token_response(
  59. ... 'https://your.provider/access_token?foo=bar',
  60. ... headers={
  61. ... 'Authorization': 'OAuth oauth_token=234lsdkf....'
  62. ... },
  63. ... credentials={
  64. ... 'my_specific': 'argument',
  65. ... })
  66. >>> h
  67. {'Content-Type': 'application/x-www-form-urlencoded'}
  68. >>> b
  69. 'oauth_token=lsdkfol23w54jlksdef&oauth_token_secret=qwe089234lkjsdf&oauth_authorized_realms=movies+pics&my_specific=argument'
  70. >>> s
  71. 200
  72. An response to invalid request would have a different body and status::
  73. >>> b
  74. 'error=invalid_request&description=missing+resource+owner+key'
  75. >>> s
  76. 400
  77. The same goes for an an unauthorized request:
  78. >>> b
  79. ''
  80. >>> s
  81. 401
  82. """
  83. resp_headers = {'Content-Type': 'application/x-www-form-urlencoded'}
  84. try:
  85. request = self._create_request(uri, http_method, body, headers)
  86. valid, processed_request = self.validate_access_token_request(
  87. request)
  88. if valid:
  89. token = self.create_access_token(request, credentials or {})
  90. self.request_validator.invalidate_request_token(
  91. request.client_key,
  92. request.resource_owner_key,
  93. request)
  94. return resp_headers, token, 200
  95. else:
  96. return {}, None, 401
  97. except errors.OAuth1Error as e:
  98. return resp_headers, e.urlencoded, e.status_code
  99. def validate_access_token_request(self, request):
  100. """Validate an access token request.
  101. :param request: OAuthlib request.
  102. :type request: oauthlib.common.Request
  103. :raises: OAuth1Error if the request is invalid.
  104. :returns: A tuple of 2 elements.
  105. 1. The validation result (True or False).
  106. 2. The request object.
  107. """
  108. self._check_transport_security(request)
  109. self._check_mandatory_parameters(request)
  110. if not request.resource_owner_key:
  111. raise errors.InvalidRequestError(
  112. description='Missing resource owner.')
  113. if not self.request_validator.check_request_token(
  114. request.resource_owner_key):
  115. raise errors.InvalidRequestError(
  116. description='Invalid resource owner key format.')
  117. if not request.verifier:
  118. raise errors.InvalidRequestError(
  119. description='Missing verifier.')
  120. if not self.request_validator.check_verifier(request.verifier):
  121. raise errors.InvalidRequestError(
  122. description='Invalid verifier format.')
  123. if not self.request_validator.validate_timestamp_and_nonce(
  124. request.client_key, request.timestamp, request.nonce, request,
  125. request_token=request.resource_owner_key):
  126. return False, request
  127. # The server SHOULD return a 401 (Unauthorized) status code when
  128. # receiving a request with invalid client credentials.
  129. # Note: This is postponed in order to avoid timing attacks, instead
  130. # a dummy client is assigned and used to maintain near constant
  131. # time request verification.
  132. #
  133. # Note that early exit would enable client enumeration
  134. valid_client = self.request_validator.validate_client_key(
  135. request.client_key, request)
  136. if not valid_client:
  137. request.client_key = self.request_validator.dummy_client
  138. # The server SHOULD return a 401 (Unauthorized) status code when
  139. # receiving a request with invalid or expired token.
  140. # Note: This is postponed in order to avoid timing attacks, instead
  141. # a dummy token is assigned and used to maintain near constant
  142. # time request verification.
  143. #
  144. # Note that early exit would enable resource owner enumeration
  145. valid_resource_owner = self.request_validator.validate_request_token(
  146. request.client_key, request.resource_owner_key, request)
  147. if not valid_resource_owner:
  148. request.resource_owner_key = self.request_validator.dummy_request_token
  149. # The server MUST verify (Section 3.2) the validity of the request,
  150. # ensure that the resource owner has authorized the provisioning of
  151. # token credentials to the client, and ensure that the temporary
  152. # credentials have not expired or been used before. The server MUST
  153. # also verify the verification code received from the client.
  154. # .. _`Section 3.2`: https://tools.ietf.org/html/rfc5849#section-3.2
  155. #
  156. # Note that early exit would enable resource owner authorization
  157. # verifier enumertion.
  158. valid_verifier = self.request_validator.validate_verifier(
  159. request.client_key,
  160. request.resource_owner_key,
  161. request.verifier,
  162. request)
  163. valid_signature = self._check_signature(request, is_token_request=True)
  164. # log the results to the validator_log
  165. # this lets us handle internal reporting and analysis
  166. request.validator_log['client'] = valid_client
  167. request.validator_log['resource_owner'] = valid_resource_owner
  168. request.validator_log['verifier'] = valid_verifier
  169. request.validator_log['signature'] = valid_signature
  170. # We delay checking validity until the very end, using dummy values for
  171. # calculations and fetching secrets/keys to ensure the flow of every
  172. # request remains almost identical regardless of whether valid values
  173. # have been supplied. This ensures near constant time execution and
  174. # prevents malicious users from guessing sensitive information
  175. v = all((valid_client, valid_resource_owner, valid_verifier,
  176. valid_signature))
  177. if not v:
  178. log.info("[Failure] request verification failed.")
  179. log.info("Valid client:, %s", valid_client)
  180. log.info("Valid token:, %s", valid_resource_owner)
  181. log.info("Valid verifier:, %s", valid_verifier)
  182. log.info("Valid signature:, %s", valid_signature)
  183. return v, request