| Index: remoting/host/curtaining_host_observer.cc
|
| diff --git a/remoting/host/curtaining_host_observer.cc b/remoting/host/curtaining_host_observer.cc
|
| index 6eacabe6c08be853ac9d92e798d8a63e347ab5ac..31bb41154f9e80a00add82bfe4976aa44a3e2cf3 100644
|
| --- a/remoting/host/curtaining_host_observer.cc
|
| +++ b/remoting/host/curtaining_host_observer.cc
|
| @@ -6,18 +6,19 @@
|
|
|
| #include "base/logging.h"
|
| #include "remoting/host/curtain_mode.h"
|
| -#include "remoting/host/chromoting_host.h"
|
| +#include "remoting/host/host_status_monitor.h"
|
|
|
| namespace remoting {
|
|
|
| CurtainingHostObserver::CurtainingHostObserver(
|
| - CurtainMode *curtain, scoped_refptr<ChromotingHost> host)
|
| - : curtain_(curtain), host_(host) {
|
| - host_->AddStatusObserver(this);
|
| + CurtainMode *curtain, base::WeakPtr<HostStatusMonitor> monitor)
|
| + : curtain_(curtain), monitor_(monitor) {
|
| + monitor_->AddStatusObserver(this);
|
| }
|
|
|
| CurtainingHostObserver::~CurtainingHostObserver() {
|
| - host_->RemoveStatusObserver(this);
|
| + if (monitor_)
|
| + monitor_->RemoveStatusObserver(this);
|
| curtain_->SetActivated(false);
|
| }
|
|
|
|
|