1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. 2 // Use of this source code is governed by a BSD-style license that can be 3 // found in the LICENSE file. 4 5 #include "chrome/browser/ui/auto_login_prompter.h" 6 7 #include "base/bind.h" 8 #include "base/command_line.h" 9 #include "base/logging.h" 10 #include "base/prefs/pref_service.h" 11 #include "chrome/browser/google/google_url_tracker.h" 12 #include "chrome/browser/infobars/infobar_service.h" 13 #include "chrome/browser/profiles/profile.h" 14 #include "chrome/browser/signin/profile_oauth2_token_service.h" 15 #include "chrome/browser/signin/profile_oauth2_token_service_factory.h" 16 #include "chrome/browser/signin/signin_manager.h" 17 #include "chrome/browser/signin/signin_manager_factory.h" 18 #include "chrome/browser/tab_contents/tab_util.h" 19 #include "chrome/common/chrome_switches.h" 20 #include "chrome/common/pref_names.h" 21 #include "components/auto_login_parser/auto_login_parser.h" 22 #include "content/public/browser/browser_thread.h" 23 #include "content/public/browser/web_contents.h" 24 #include "net/url_request/url_request.h" 25 #include "url/gurl.h" 26 27 using content::BrowserThread; 28 using content::WebContents; 29 30 namespace { 31 32 bool FetchUsernameThroughSigninManager(Profile* profile, std::string* output) { 33 // In an incognito window these services are not available. 34 ProfileOAuth2TokenService* token_service = 35 ProfileOAuth2TokenServiceFactory::GetForProfile(profile); 36 if (!token_service || !token_service->RefreshTokenIsAvailable()) 37 return false; 38 39 SigninManagerBase* signin_manager = 40 SigninManagerFactory::GetInstance()->GetForProfile(profile); 41 if (!signin_manager) 42 return false; 43 44 *output = signin_manager->GetAuthenticatedUsername(); 45 return true; 46 } 47 48 } // namespace 49 50 AutoLoginPrompter::AutoLoginPrompter(WebContents* web_contents, 51 const Params& params, 52 const GURL& url) 53 : WebContentsObserver(web_contents), 54 params_(params), 55 url_(url), 56 infobar_shown_(false) { 57 if (!web_contents->IsLoading()) { 58 // If the WebContents isn't loading a page, the load notification will never 59 // be triggered. Try adding the InfoBar now. 60 AddInfoBarToWebContents(); 61 } 62 } 63 64 AutoLoginPrompter::~AutoLoginPrompter() { 65 } 66 67 // static 68 void AutoLoginPrompter::ShowInfoBarIfPossible(net::URLRequest* request, 69 int child_id, 70 int route_id) { 71 if (!CommandLine::ForCurrentProcess()->HasSwitch(switches::kEnableAutologin)) 72 return; 73 74 // See if the response contains the X-Auto-Login header. If so, this was 75 // a request for a login page, and the server is allowing the browser to 76 // suggest auto-login, if available. 77 Params params; 78 // Currently we only accept GAIA credentials in Chrome. 79 if (!auto_login_parser::ParserHeaderInResponse( 80 request, auto_login_parser::ONLY_GOOGLE_COM, ¶ms.header)) 81 return; 82 83 BrowserThread::PostTask( 84 BrowserThread::UI, FROM_HERE, 85 base::Bind(&ShowInfoBarUIThread, 86 params, request->url(), child_id, route_id)); 87 } 88 89 90 // static 91 void AutoLoginPrompter::ShowInfoBarUIThread(Params params, 92 const GURL& url, 93 int child_id, 94 int route_id) { 95 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); 96 WebContents* web_contents = tab_util::GetWebContentsByID(child_id, route_id); 97 if (!web_contents) 98 return; 99 100 Profile* profile = 101 Profile::FromBrowserContext(web_contents->GetBrowserContext()); 102 103 if (!profile->GetPrefs()->GetBoolean(prefs::kAutologinEnabled)) 104 return; 105 106 #if !defined(OS_ANDROID) 107 // On Android, the username is fetched on the Java side from the 108 // AccountManager provided by the platform. 109 if (!FetchUsernameThroughSigninManager(profile, ¶ms.username)) 110 return; 111 #endif 112 113 // Make sure that |account|, if specified, matches the logged in user. 114 // However, |account| is usually empty. 115 if (!params.username.empty() && !params.header.account.empty() && 116 params.username != params.header.account) 117 return; 118 // We can't add the infobar just yet, since we need to wait for the tab to 119 // finish loading. If we don't, the info bar appears and then disappears 120 // immediately. Create an AutoLoginPrompter instance to listen for the 121 // relevant notifications; it will delete itself. 122 new AutoLoginPrompter(web_contents, params, url); 123 } 124 125 void AutoLoginPrompter::DidStopLoading( 126 content::RenderViewHost* render_view_host) { 127 AddInfoBarToWebContents(); 128 delete this; 129 } 130 131 void AutoLoginPrompter::WebContentsDestroyed(WebContents* web_contents) { 132 // The WebContents was destroyed before the navigation completed. 133 delete this; 134 } 135 136 void AutoLoginPrompter::AddInfoBarToWebContents() { 137 if (infobar_shown_) 138 return; 139 140 InfoBarService* infobar_service = 141 InfoBarService::FromWebContents(web_contents()); 142 // |infobar_service| is NULL for WebContents hosted in WebDialog. 143 if (infobar_service) { 144 AutoLoginInfoBarDelegate::Create(infobar_service, params_); 145 infobar_shown_ = true; 146 } 147 } 148