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) 5a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles)#ifndef COMPONENTS_POLICY_CORE_COMMON_CLOUD_DEVICE_MANAGEMENT_SERVICE_H_ 6a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles)#define COMPONENTS_POLICY_CORE_COMMON_CLOUD_DEVICE_MANAGEMENT_SERVICE_H_ 75821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 85821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <deque> 95821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <map> 105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <string> 115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <vector> 125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/basictypes.h" 145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/callback.h" 155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/compiler_specific.h" 161320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci#include "base/memory/ref_counted.h" 175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/memory/weak_ptr.h" 18a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles)#include "components/policy/core/common/cloud/cloud_policy_constants.h" 19a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles)#include "components/policy/policy_export.h" 205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "net/url_request/url_fetcher_delegate.h" 21a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles)#include "policy/proto/device_management_backend.pb.h" 225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace net { 245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)class URLRequestContextGetter; 255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace policy { 285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)class DeviceManagementRequestJobImpl; 305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)class DeviceManagementService; 315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// DeviceManagementRequestJob describes a request to send to the device 335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// management service. Jobs are created by DeviceManagementService. They can be 345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// canceled by deleting the object. 35a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles)class POLICY_EXPORT DeviceManagementRequestJob { 365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) public: 375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Describes the job type. 385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) enum JobType { 395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) TYPE_AUTO_ENROLLMENT, 405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) TYPE_REGISTRATION, 41c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) TYPE_API_AUTH_CODE_FETCH, 425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) TYPE_POLICY_FETCH, 435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) TYPE_UNREGISTRATION, 44c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) TYPE_UPLOAD_CERTIFICATE, 45a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) TYPE_DEVICE_STATE_RETRIEVAL, 465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) }; 475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) typedef base::Callback< 49b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) void(DeviceManagementStatus, int, 505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const enterprise_management::DeviceManagementResponse&)> Callback; 515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 522a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) typedef base::Callback<void(DeviceManagementRequestJob*)> RetryCallback; 532a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual ~DeviceManagementRequestJob(); 555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Functions for configuring the job. These should only be called before 575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Start()ing the job, but never afterwards. 585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void SetGaiaToken(const std::string& gaia_token); 595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void SetOAuthToken(const std::string& oauth_token); 605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void SetUserAffiliation(UserAffiliation user_affiliation); 615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void SetDMToken(const std::string& dm_token); 625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void SetClientID(const std::string& client_id); 635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) enterprise_management::DeviceManagementRequest* GetRequest(); 645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 652a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // A job may automatically retry if it fails due to a temporary condition, or 662a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // due to proxy misconfigurations. If a |retry_callback| is set then it will 672a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // be invoked with the DeviceManagementRequestJob as an argument when that 682a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // happens, so that the job's owner can customize the retry request before 692a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // it's sent. 702a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) void SetRetryCallback(const RetryCallback& retry_callback); 712a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Starts the job. |callback| will be invoked on completion. 732a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) void Start(const Callback& callback); 745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) protected: 765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) typedef std::vector<std::pair<std::string, std::string> > ParameterMap; 775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 784e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) DeviceManagementRequestJob(JobType type, 794e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) const std::string& agent_parameter, 804e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) const std::string& platform_parameter); 815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Appends a parameter to |query_params|. 835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void AddParameter(const std::string& name, const std::string& value); 845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Fires the job, to be filled in by implementations. 865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual void Run() = 0; 875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ParameterMap query_params_; 895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) std::string gaia_token_; 905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) std::string dm_token_; 915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) enterprise_management::DeviceManagementRequest request_; 922a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) RetryCallback retry_callback_; 935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) Callback callback_; 955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) private: 975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DISALLOW_COPY_AND_ASSIGN(DeviceManagementRequestJob); 985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}; 995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// The device management service is responsible for everything related to 1015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// communication with the device management server. It creates the backends 1025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// objects that the device management policy provider and friends use to issue 1035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// requests. 104a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles)class POLICY_EXPORT DeviceManagementService : public net::URLFetcherDelegate { 1055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) public: 1064e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) // Obtains the parameters used to contact the server. 1074e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) // This allows creating the DeviceManagementService early and getting these 1084e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) // parameters later. Passing the parameters directly in the ctor isn't 1094e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) // possible because some aren't ready during startup. http://crbug.com/302798 110a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles) class POLICY_EXPORT Configuration { 1114e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) public: 1124e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) virtual ~Configuration() {} 1134e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) 1144e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) // Server at which to contact the service. 1154e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) virtual std::string GetServerUrl() = 0; 1164e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) 1174e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) // Agent reported in the "agent" query parameter. 1184e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) virtual std::string GetAgentParameter() = 0; 1194e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) 1204e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) // The platform reported in the "platform" query parameter. 1214e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) virtual std::string GetPlatformParameter() = 0; 1224e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) }; 1234e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) 124a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles) explicit DeviceManagementService(scoped_ptr<Configuration> configuration); 1255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual ~DeviceManagementService(); 1265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1272a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // The ID of URLFetchers created by the DeviceManagementService. This can be 1282a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // used by tests that use a TestURLFetcherFactory to get the pending fetchers 1292a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // created by the DeviceManagementService. 1302a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) static const int kURLFetcherID; 1312a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 1325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Creates a new device management request job. Ownership is transferred to 1335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // the caller. 1345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual DeviceManagementRequestJob* CreateJob( 135a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles) DeviceManagementRequestJob::JobType type, 1361320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci const scoped_refptr<net::URLRequestContextGetter>& request_context); 1375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Schedules a task to run |Initialize| after |delay_milliseconds| had passed. 1395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void ScheduleInitialization(int64 delay_milliseconds); 1405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 141a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles) // Makes the service stop all requests. 1425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void Shutdown(); 1435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 144a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles) // Gets the URL that the DMServer requests are sent to. 145a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles) std::string GetServerUrl(); 146a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles) 1475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) private: 1485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) typedef std::map<const net::URLFetcher*, 1495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DeviceManagementRequestJobImpl*> JobFetcherMap; 1505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) typedef std::deque<DeviceManagementRequestJobImpl*> JobQueue; 1515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) friend class DeviceManagementRequestJobImpl; 1535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // net::URLFetcherDelegate override. 1555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual void OnURLFetchComplete(const net::URLFetcher* source) OVERRIDE; 1565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 157a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles) // Starts processing any queued jobs. 1585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void Initialize(); 1595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Starts a job. 1612a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) void StartJob(DeviceManagementRequestJobImpl* job); 1625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Adds a job. Caller must make sure the job pointer stays valid until the job 1645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // completes or gets canceled via RemoveJob(). 1655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void AddJob(DeviceManagementRequestJobImpl* job); 1665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Removes a job. The job will be removed and won't receive a completion 1685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // callback. 1695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void RemoveJob(DeviceManagementRequestJobImpl* job); 1705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1714e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) // A Configuration implementation that is used to obtain various parameters 1724e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) // used to talk to the device management server. 1734e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) scoped_ptr<Configuration> configuration_; 1744e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) 1755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // The jobs we currently have in flight. 1765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) JobFetcherMap pending_jobs_; 1775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Jobs that are registered, but not started yet. 1795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) JobQueue queued_jobs_; 1805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // If this service is initialized, incoming requests get fired instantly. 1825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // If it is not initialized, incoming requests are queued. 1835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) bool initialized_; 1845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Used to create tasks to run |Initialize| delayed on the UI thread. 1865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::WeakPtrFactory<DeviceManagementService> weak_ptr_factory_; 1875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DISALLOW_COPY_AND_ASSIGN(DeviceManagementService); 1895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}; 1905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} // namespace policy 1925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 193a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles)#endif // COMPONENTS_POLICY_CORE_COMMON_CLOUD_DEVICE_MANAGEMENT_SERVICE_H_ 194