Chromium Code Reviews| Index: chrome/common/net/gaia/oauth2_token_mint_fetcher.cc |
| =================================================================== |
| --- chrome/common/net/gaia/oauth2_token_mint_fetcher.cc (revision 0) |
| +++ chrome/common/net/gaia/oauth2_token_mint_fetcher.cc (revision 0) |
| @@ -0,0 +1,203 @@ |
| +// Copyright (c) 2011 The Chromium Authors. All rights reserved. |
| +// Use of this source code is governed by a BSD-style license that can be |
| +// found in the LICENSE file. |
| + |
| +#include "chrome/common/net/gaia/oauth2_token_mint_fetcher.h" |
| + |
| +#include <algorithm> |
| +#include <string> |
| + |
| +#include "base/json/json_reader.h" |
| +#include "base/string_util.h" |
| +#include "base/stringprintf.h" |
| +#include "base/values.h" |
| +#include "chrome/common/net/gaia/gaia_urls.h" |
| +#include "chrome/common/net/gaia/google_service_auth_error.h" |
| +#include "net/base/escape.h" |
| +#include "net/base/load_flags.h" |
| +#include "net/http/http_status_code.h" |
| +#include "net/url_request/url_request_context_getter.h" |
| +#include "net/url_request/url_request_status.h" |
| + |
| +using content::URLFetcher; |
| +using content::URLFetcherDelegate; |
| +using net::ResponseCookies; |
| +using net::URLRequestContextGetter; |
| +using net::URLRequestStatus; |
| + |
| +namespace { |
| +static const char kAuthorizationHeaderFormat[] = |
| + "Authorization: Bearer %s"; |
| +static const char kOAuth2IssueTokenBodyFormat[] = |
| + "force=true" |
| + "&response_type=token" |
| + "&scope=%s" |
| + "&client_id=%s" |
| + "&origin=%s"; |
| +static const char kAccessTokenKey[] = "token"; |
| + |
| +static bool GetStringFromDictionary(const DictionaryValue* dict, |
|
asargent_no_longer_on_chrome
2012/03/01 00:53:36
nit: I think you can get rid of this function and
Munjal (Google)
2012/03/01 18:38:52
Done.
|
| + const std::string& key, |
| + std::string* value) { |
| + Value* json_value; |
| + if (!dict->Get(key, &json_value)) |
| + return false; |
| + if (json_value->GetType() != base::Value::TYPE_STRING) |
| + return false; |
| + |
| + StringValue* json_str_value = static_cast<StringValue*>(json_value); |
| + json_str_value->GetAsString(value); |
| + return true; |
| +} |
| + |
| +static GoogleServiceAuthError CreateAuthError(URLRequestStatus status) { |
| + CHECK(!status.is_success()); |
| + if (status.status() == URLRequestStatus::CANCELED) { |
| + return GoogleServiceAuthError(GoogleServiceAuthError::REQUEST_CANCELED); |
| + } else { |
| + DLOG(WARNING) << "Could not reach Google Accounts servers: errno " |
| + << status.error(); |
| + return GoogleServiceAuthError::FromConnectionError(status.error()); |
| + } |
| +} |
| + |
| +static URLFetcher* CreateFetcher(URLRequestContextGetter* getter, |
| + const GURL& url, |
| + const std::string& headers, |
| + const std::string& body, |
| + URLFetcherDelegate* delegate) { |
| + bool empty_body = body.empty(); |
| + URLFetcher* result = URLFetcher::Create( |
| + 0, url, |
| + empty_body ? URLFetcher::GET : URLFetcher::POST, |
| + delegate); |
| + |
| + result->SetRequestContext(getter); |
| + result->SetLoadFlags(net::LOAD_DO_NOT_SEND_COOKIES | |
| + net::LOAD_DO_NOT_SAVE_COOKIES); |
| + |
| + if (!empty_body) |
| + result->SetUploadData("application/x-www-form-urlencoded", body); |
| + if (!headers.empty()) |
| + result->SetExtraRequestHeaders(headers); |
| + |
| + return result; |
| +} |
| +} // namespace |
| + |
| +OAuth2TokenMintFetcher::OAuth2TokenMintFetcher( |
| + OAuth2TokenMintConsumer* consumer, |
| + URLRequestContextGetter* getter, |
| + const std::string& source) |
| + : consumer_(consumer), |
| + getter_(getter), |
| + source_(source), |
| + state_(INITIAL) { } |
| + |
| +OAuth2TokenMintFetcher::~OAuth2TokenMintFetcher() { } |
| + |
| +void OAuth2TokenMintFetcher::CancelRequest() { |
| + fetcher_.reset(); |
| +} |
| + |
| +void OAuth2TokenMintFetcher::Start(const std::string& oauth_login_access_token, |
| + const std::string& client_id, |
| + const std::vector<std::string>& scopes, |
| + const std::string& origin) { |
| + oauth_login_access_token_ = oauth_login_access_token; |
| + client_id_ = client_id; |
| + scopes_ = scopes; |
| + origin_ = origin; |
| + StartMintToken(); |
| +} |
| + |
| +void OAuth2TokenMintFetcher::StartMintToken() { |
| + CHECK_EQ(INITIAL, state_); |
| + state_ = MINT_TOKEN_STARTED; |
| + fetcher_.reset(CreateFetcher( |
| + getter_, |
| + MakeMintTokenUrl(), |
| + MakeMintTokenHeader(oauth_login_access_token_), |
| + MakeMintTokenBody(client_id_, scopes_, origin_), |
| + this)); |
| + fetcher_->Start(); // OnURLFetchComplete will be called. |
| +} |
| + |
| +void OAuth2TokenMintFetcher::EndMintToken(const URLFetcher* source) { |
| + CHECK_EQ(MINT_TOKEN_STARTED, state_); |
| + state_ = MINT_TOKEN_DONE; |
| + |
| + URLRequestStatus status = source->GetStatus(); |
| + if (!status.is_success()) { |
| + OnMintTokenFailure(CreateAuthError(status)); |
| + return; |
| + } |
| + |
| + if (source->GetResponseCode() != net::HTTP_OK) { |
| + OnMintTokenFailure(GoogleServiceAuthError( |
| + GoogleServiceAuthError::INVALID_GAIA_CREDENTIALS)); |
| + return; |
| + } |
| + |
| + // The request was successfully fetched and it returned OK. |
| + // Parse out the access token. |
| + std::string access_token; |
| + ParseMintTokenResponse(source, &access_token); |
| + OnMintTokenSuccess(access_token); |
| +} |
| + |
| +void OAuth2TokenMintFetcher::OnMintTokenSuccess( |
| + const std::string& access_token) { |
| + consumer_->OnMintTokenSuccess(access_token); |
| +} |
| + |
| +void OAuth2TokenMintFetcher::OnMintTokenFailure(GoogleServiceAuthError error) { |
| + state_ = ERROR_STATE; |
| + consumer_->OnMintTokenFailure(error); |
| +} |
| + |
| +void OAuth2TokenMintFetcher::OnURLFetchComplete(const URLFetcher* source) { |
| + CHECK(source); |
| + CHECK_EQ(MINT_TOKEN_STARTED, state_); |
| + EndMintToken(source); |
| +} |
| + |
| +// static |
| +GURL OAuth2TokenMintFetcher::MakeMintTokenUrl() { |
| + return GURL(GaiaUrls::GetInstance()->oauth2_issue_token_url()); |
| +} |
| + |
| +// static |
| +std::string OAuth2TokenMintFetcher::MakeMintTokenHeader( |
| + const std::string& access_token) { |
| + return StringPrintf(kAuthorizationHeaderFormat, access_token); |
| +} |
| + |
| +// static |
| +std::string OAuth2TokenMintFetcher::MakeMintTokenBody( |
| + const std::string& client_id, |
| + const std::vector<std::string>& scopes, |
| + const std::string& origin) { |
| + return StringPrintf( |
| + kOAuth2IssueTokenBodyFormat, |
| + net::EscapeUrlEncodedData(JoinString(scopes, ','), true).c_str(), |
| + net::EscapeUrlEncodedData(client_id, true).c_str(), |
| + net::EscapeUrlEncodedData(origin, true).c_str()); |
| +} |
| + |
| +// static |
| +bool OAuth2TokenMintFetcher::ParseMintTokenResponse( |
| + const URLFetcher* source, |
| + std::string* access_token) { |
| + CHECK(source); |
| + CHECK(access_token); |
| + std::string data; |
| + source->GetResponseAsString(&data); |
| + base::JSONReader reader; |
| + scoped_ptr<base::Value> value(reader.Read(data, false)); |
| + if (!value.get() || value->GetType() != base::Value::TYPE_DICTIONARY) |
| + return false; |
| + |
| + DictionaryValue* dict = static_cast<DictionaryValue*>(value.get()); |
| + return GetStringFromDictionary(dict, kAccessTokenKey, access_token); |
| +} |