OLD | NEW |
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 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 | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #ifndef REMOTING_HOST_ELEVATED_CONTROLLER_WIN_H_ | 5 #ifndef REMOTING_HOST_ELEVATED_CONTROLLER_WIN_H_ |
6 #define REMOTING_HOST_ELEVATED_CONTROLLER_WIN_H_ | 6 #define REMOTING_HOST_ELEVATED_CONTROLLER_WIN_H_ |
7 | 7 |
8 #include <atlbase.h> | 8 #include <atlbase.h> |
9 #include <atlcom.h> | 9 #include <atlcom.h> |
10 #include <atlctl.h> | 10 #include <atlctl.h> |
11 | 11 |
12 #include "remoting/base/scoped_sc_handle_win.h" | 12 #include "remoting/base/scoped_sc_handle_win.h" |
13 | 13 |
14 // MIDL-generated declarations. | 14 // MIDL-generated declarations. |
15 #include "remoting/host/elevated_controller.h" | 15 #include "remoting/host/elevated_controller.h" |
16 | 16 |
17 namespace remoting { | 17 namespace remoting { |
18 | 18 |
19 class ATL_NO_VTABLE ElevatedControllerWin | 19 class ATL_NO_VTABLE ElevatedControllerWin |
20 : public ATL::CComObjectRootEx<ATL::CComSingleThreadModel>, | 20 : public ATL::CComObjectRootEx<ATL::CComSingleThreadModel>, |
21 public ATL::CComCoClass<ElevatedControllerWin, &CLSID_ElevatedController>, | 21 public ATL::CComCoClass<ElevatedControllerWin, &CLSID_ElevatedController>, |
22 public ATL::IDispatchImpl<IDaemonControl, &IID_IDaemonControl, | 22 public ATL::IDispatchImpl<IDaemonControl2, &IID_IDaemonControl2, |
23 &LIBID_ChromotingElevatedControllerLib, 1, 0> { | 23 &LIBID_ChromotingElevatedControllerLib, 1, 1> { |
24 public: | 24 public: |
25 ElevatedControllerWin(); | 25 ElevatedControllerWin(); |
26 | 26 |
27 HRESULT FinalConstruct(); | 27 HRESULT FinalConstruct(); |
28 void FinalRelease(); | 28 void FinalRelease(); |
29 | 29 |
30 // IDaemonControl implementation. | 30 // IDaemonControl implementation. |
31 STDMETHOD(GetConfig)(BSTR* config_out); | 31 STDMETHOD(GetConfig)(BSTR* config_out); |
32 STDMETHOD(GetVersion)(BSTR* version_out); | 32 STDMETHOD(GetVersion)(BSTR* version_out); |
33 STDMETHOD(SetConfig)(BSTR config); | 33 STDMETHOD(SetConfig)(BSTR config); |
34 STDMETHOD(SetOwnerWindow)(LONG_PTR owner_window); | 34 STDMETHOD(SetOwnerWindow)(LONG_PTR owner_window); |
35 STDMETHOD(StartDaemon)(); | 35 STDMETHOD(StartDaemon)(); |
36 STDMETHOD(StopDaemon)(); | 36 STDMETHOD(StopDaemon)(); |
37 STDMETHOD(UpdateConfig)(BSTR config); | 37 STDMETHOD(UpdateConfig)(BSTR config); |
38 | 38 |
| 39 // IDaemonControl2 implementation. |
| 40 STDMETHOD(GetUsageStatsConsent)(BOOL* allowed, BOOL* set_by_policy); |
| 41 STDMETHOD(SetUsageStatsConsent)(BOOL allowed); |
| 42 |
39 DECLARE_NO_REGISTRY() | 43 DECLARE_NO_REGISTRY() |
40 | 44 |
41 private: | 45 private: |
42 HRESULT OpenService(ScopedScHandle* service_out); | 46 HRESULT OpenService(ScopedScHandle* service_out); |
43 | 47 |
44 BEGIN_COM_MAP(ElevatedControllerWin) | 48 BEGIN_COM_MAP(ElevatedControllerWin) |
45 COM_INTERFACE_ENTRY(IDaemonControl) | 49 COM_INTERFACE_ENTRY(IDaemonControl) |
| 50 COM_INTERFACE_ENTRY(IDaemonControl2) |
46 COM_INTERFACE_ENTRY(IDispatch) | 51 COM_INTERFACE_ENTRY(IDispatch) |
47 END_COM_MAP() | 52 END_COM_MAP() |
48 | 53 |
49 // Handle of the owner window (if any) for any UI to be shown. | 54 // Handle of the owner window (if any) for any UI to be shown. |
50 HWND owner_window_; | 55 HWND owner_window_; |
51 | 56 |
52 DECLARE_PROTECT_FINAL_CONSTRUCT() | 57 DECLARE_PROTECT_FINAL_CONSTRUCT() |
53 }; | 58 }; |
54 | 59 |
55 OBJECT_ENTRY_AUTO(CLSID_ElevatedController, ElevatedControllerWin) | 60 OBJECT_ENTRY_AUTO(CLSID_ElevatedController, ElevatedControllerWin) |
56 | 61 |
57 } // namespace remoting | 62 } // namespace remoting |
58 | 63 |
59 #endif // REMOTING_HOST_ELEVATED_CONTROLLER_WIN_H_ | 64 #endif // REMOTING_HOST_ELEVATED_CONTROLLER_WIN_H_ |
OLD | NEW |