15821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Copyright (c) 2012 The Chromium Authors. All rights reserved. 25821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Use of this source code is governed by a BSD-style license that can be 35821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// found in the LICENSE file. 45821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 55821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#ifndef REMOTING_HOST_SETUP_WIN_AUTH_CODE_GETTER_H 65821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#define REMOTING_HOST_SETUP_WIN_AUTH_CODE_GETTER_H 75821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 85821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <ole2.h> 95821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <exdisp.h> 105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <string> 125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/callback.h" 145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/threading/non_thread_safe.h" 15eb525c5499e34cc9c4b825d6d9e75bb07cc06aceBen Murdoch#include "base/timer/timer.h" 165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/win/scoped_comptr.h" 175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace remoting { 195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// A class for getting an OAuth authorization code. 215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)class AuthCodeGetter : public base::NonThreadSafe { 225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) public: 235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) AuthCodeGetter(); 245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ~AuthCodeGetter(); 255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Starts a browser and navigates it to a URL that starts an Installed 275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Application OAuth flow. |on_auth_code| will be called with an 285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // authorization code, or an empty string on error. 295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void GetAuthCode(base::Callback<void(const std::string&)> on_auth_code); 305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) private: 325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Starts a timer used to poll the browser's URL. 335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void StartTimer(); 345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Called when that timer fires. 355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void OnTimer(); 365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Returns whether to stop polling the browser's URL. If true, then 375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // |auth_code| is an authorization code, or the empty string on an error. 385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) bool TestBrowserUrl(std::string* auth_code); 395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Kills the browser. 405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void KillBrowser(); 415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // The authorization code callback. 435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::Callback<void(const std::string&)> on_auth_code_; 445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // The browser through which the user requests an authorization code. 455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::win::ScopedComPtr<IWebBrowser2, &IID_IWebBrowser2> browser_; 465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // A timer used to poll the browser's URL. 475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::OneShotTimer<AuthCodeGetter> timer_; 485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // The interval at which the timer fires. 495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::TimeDelta timer_interval_; 505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DISALLOW_COPY_AND_ASSIGN(AuthCodeGetter); 525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}; 535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} // namespace remoting 555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#endif // REMOTING_HOST_SETUP_WIN_AUTH_CODE_GETTER_H 57