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); |
} |