blob: 9533f02d779b46d58f579b57f1e63f6980653668 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
|
// Copyright (c) 2012 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.
#ifndef REMOTING_HOST_WIN_HOST_SERVICE_H_
#define REMOTING_HOST_WIN_HOST_SERVICE_H_
#include <windows.h>
#include "base/memory/ref_counted.h"
#include "base/memory/singleton.h"
#include "base/observer_list.h"
#include "base/synchronization/waitable_event.h"
#include "remoting/host/win/wts_console_monitor.h"
class CommandLine;
class MessageLoop;
namespace base {
class SingleThreadTaskRunner;
} // namespace base
namespace remoting {
#if defined(REMOTING_MULTI_PROCESS)
class DaemonProcess;
#endif // defined(REMOTING_MULTI_PROCESS)
class Stoppable;
class WtsConsoleObserver;
#if !defined(REMOTING_MULTI_PROCESS)
class WtsSessionProcessLauncher;
#endif // !defined(REMOTING_MULTI_PROCESS)
class HostService : public WtsConsoleMonitor {
public:
static HostService* GetInstance();
// This function parses the command line and selects the action routine.
bool InitWithCommandLine(const CommandLine* command_line);
// Invoke the choosen action routine.
int Run();
// WtsConsoleMonitor implementation
virtual void AddWtsConsoleObserver(WtsConsoleObserver* observer) OVERRIDE;
virtual void RemoveWtsConsoleObserver(
WtsConsoleObserver* observer) OVERRIDE;
private:
HostService();
~HostService();
void OnChildStopped();
// Notifies the service of changes in session state.
void OnSessionChange();
// This is a common entry point to the main service loop called by both
// RunAsService() and RunInConsole().
void RunMessageLoop(MessageLoop* message_loop);
// Runs the binary specified by the command line, elevated.
int Elevate();
// This function handshakes with the service control manager and starts
// the service.
int RunAsService();
// This function starts the service in interactive mode (i.e. as a plain
// console application).
int RunInConsole();
static BOOL WINAPI ConsoleControlHandler(DWORD event);
// The control handler of the service.
static DWORD WINAPI ServiceControlHandler(DWORD control,
DWORD event_type,
LPVOID event_data,
LPVOID context);
// The main service entry point.
static VOID WINAPI ServiceMain(DWORD argc, WCHAR* argv[]);
static LRESULT CALLBACK SessionChangeNotificationProc(HWND hwnd,
UINT message,
WPARAM wparam,
LPARAM lparam);
// Current physical console session id.
uint32 console_session_id_;
// The list of observers receiving notifications about any session attached
// to the physical console.
ObserverList<WtsConsoleObserver> console_observers_;
scoped_ptr<Stoppable> child_;
// Service message loop.
scoped_refptr<base::SingleThreadTaskRunner> main_task_runner_;
// The action routine to be executed.
int (HostService::*run_routine_)();
// The service status handle.
SERVICE_STATUS_HANDLE service_status_handle_;
// A waitable event that is used to wait until the service is stopped.
base::WaitableEvent stopped_event_;
// Singleton.
friend struct DefaultSingletonTraits<HostService>;
DISALLOW_COPY_AND_ASSIGN(HostService);
};
} // namespace remoting
#endif // REMOTING_HOST_WIN_HOST_SERVICE_H_
|
