blob: 58a8e564eac2c6602f45284d4bf55630be4bff0f (
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
121
122
123
124
125
126
127
128
129
130
131
132
133
134
|
// 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 <stdint.h>
#include <list>
#include "base/macros.h"
#include "base/memory/ref_counted.h"
#include "base/memory/singleton.h"
#include "base/memory/weak_ptr.h"
#include "base/synchronization/waitable_event.h"
#include "remoting/host/win/wts_terminal_monitor.h"
namespace base {
class CommandLine;
class SingleThreadTaskRunner;
} // namespace base
namespace remoting {
class AutoThreadTaskRunner;
class DaemonProcess;
class WtsTerminalObserver;
class HostService : public WtsTerminalMonitor {
public:
static HostService* GetInstance();
// This function parses the command line and selects the action routine.
bool InitWithCommandLine(const base::CommandLine* command_line);
// Invoke the choosen action routine.
int Run();
// WtsTerminalMonitor implementation
bool AddWtsTerminalObserver(const std::string& terminal_id,
WtsTerminalObserver* observer) override;
void RemoveWtsTerminalObserver(
WtsTerminalObserver* observer) override;
private:
HostService();
~HostService() override;
// Notifies the service of changes in session state.
void OnSessionChange(uint32_t event, uint32_t session_id);
// Creates the process launcher.
void CreateLauncher(scoped_refptr<AutoThreadTaskRunner> task_runner);
// This function handshakes with the service control manager and starts
// the service.
int RunAsService();
// Runs the service on the service thread. A separate routine is used to make
// sure all local objects are destoyed by the time |stopped_event_| is
// signalled.
void RunAsServiceImpl();
// This function starts the service in interactive mode (i.e. as a plain
// console application).
int RunInConsole();
// Stops and deletes |daemon_process_|.
void StopDaemonProcess();
// Handles WM_WTSSESSION_CHANGE messages.
bool HandleMessage(UINT message,
WPARAM wparam,
LPARAM lparam,
LRESULT* result);
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[]);
struct RegisteredObserver {
// Unique identifier of the terminal to observe.
std::string terminal_id;
// Specifies ID of the attached session or |kInvalidSession| if no session
// is attached to the WTS terminal.
uint32_t session_id;
// Points to the observer receiving notifications about the WTS terminal
// identified by |terminal_id|.
WtsTerminalObserver* observer;
};
// The list of observers receiving session notifications.
std::list<RegisteredObserver> observers_;
scoped_ptr<DaemonProcess> daemon_process_;
// Service message loop. |main_task_runner_| must be valid as long as the
// Control+C or service notification handler is registered.
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_;
base::WeakPtr<HostService> weak_ptr_;
// Used to post session change notifications and control events.
base::WeakPtrFactory<HostService> weak_factory_;
// Singleton.
friend struct base::DefaultSingletonTraits<HostService>;
DISALLOW_COPY_AND_ASSIGN(HostService);
};
} // namespace remoting
#endif // REMOTING_HOST_WIN_HOST_SERVICE_H_
|
