blob: c4dce87d03eaefb741db2238665685f17de49011 (
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
135
136
137
138
|
// 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_WTS_SESSION_PROCESS_LAUNCHER_H_
#define REMOTING_HOST_WIN_WTS_SESSION_PROCESS_LAUNCHER_H_
#include <windows.h>
#include "base/basictypes.h"
#include "base/file_path.h"
#include "base/compiler_specific.h"
#include "base/memory/ref_counted.h"
#include "base/memory/scoped_ptr.h"
#include "base/process.h"
#include "base/time.h"
#include "base/timer.h"
#include "base/win/scoped_handle.h"
#include "base/win/object_watcher.h"
#include "ipc/ipc_channel.h"
#include "remoting/base/stoppable.h"
#include "remoting/host/win/wts_console_observer.h"
namespace base {
class SingleThreadTaskRunner;
} // namespace base
namespace IPC {
class ChannelProxy;
class Message;
} // namespace IPC
namespace remoting {
// Session id that does not represent any session.
extern const uint32 kInvalidSessionId;
class SasInjector;
class WtsConsoleMonitor;
class WtsSessionProcessLauncher
: public Stoppable,
public base::win::ObjectWatcher::Delegate,
public IPC::Listener,
public WtsConsoleObserver {
public:
// Constructs a WtsSessionProcessLauncher object. |host_binary| is the name of
// the executable to be launched in the console session. All interaction with
// |monitor| should happen on |main_message_loop|. |ipc_message_loop| has
// to be an I/O message loop.
WtsSessionProcessLauncher(
const base::Closure& stopped_callback,
WtsConsoleMonitor* monitor,
const FilePath& host_binary,
scoped_refptr<base::SingleThreadTaskRunner> main_message_loop,
scoped_refptr<base::SingleThreadTaskRunner> ipc_message_loop);
virtual ~WtsSessionProcessLauncher();
// base::win::ObjectWatcher::Delegate implementation.
virtual void OnObjectSignaled(HANDLE object) OVERRIDE;
// IPC::Listener implementation.
virtual bool OnMessageReceived(const IPC::Message& message) OVERRIDE;
// WtsConsoleObserver implementation.
virtual void OnSessionAttached(uint32 session_id) OVERRIDE;
virtual void OnSessionDetached() OVERRIDE;
protected:
// Stoppable implementation.
virtual void DoStop() OVERRIDE;
private:
// Attempts to launch the host process in the current console session.
// Schedules next launch attempt if creation of the process fails for any
// reason.
void LaunchProcess();
// Sends the Secure Attention Sequence to the session represented by
// |session_token_|.
void OnSendSasToConsole();
// Name of the host executable.
FilePath host_binary_;
// Time of the last launch attempt.
base::Time launch_time_;
// Current backoff delay.
base::TimeDelta launch_backoff_;
// Timer used to schedule the next attempt to launch the process.
base::OneShotTimer<WtsSessionProcessLauncher> timer_;
// The main service message loop.
scoped_refptr<base::SingleThreadTaskRunner> main_message_loop_;
// Message loop used by the IPC channel.
scoped_refptr<base::SingleThreadTaskRunner> ipc_message_loop_;
// This pointer is used to unsubscribe from session attach and detach events.
WtsConsoleMonitor* monitor_;
// Impersonation token that has the SE_TCB_NAME privilege enabled.
base::win::ScopedHandle privileged_token_;
// The handle of the process injected into the console session.
base::Process process_;
// Used to determine when the launched process terminates.
base::win::ObjectWatcher process_watcher_;
// The token to be used to launch a process in a different session.
base::win::ScopedHandle session_token_;
// Defines the states the process launcher can be in.
enum State {
StateDetached,
StateStarting,
StateAttached,
};
// Current state of the process launcher.
State state_;
// The Chromoting IPC channel connecting the service to the per-session
// process.
scoped_ptr<IPC::ChannelProxy> chromoting_channel_;
scoped_ptr<SasInjector> sas_injector_;
DISALLOW_COPY_AND_ASSIGN(WtsSessionProcessLauncher);
};
} // namespace remoting
#endif // REMOTING_HOST_WIN_WTS_SESSION_PROCESS_LAUNCHER_H_
|
