OLD | NEW |
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #ifndef REMOTING_HOST_POLICY_HACK_NAT_POLICY_H_ | 5 #ifndef REMOTING_HOST_POLICY_HACK_NAT_POLICY_H_ |
6 #define REMOTING_HOST_POLICY_HACK_NAT_POLICY_H_ | 6 #define REMOTING_HOST_POLICY_HACK_NAT_POLICY_H_ |
7 | 7 |
8 #include "base/callback.h" | 8 #include "base/callback.h" |
9 #include "base/memory/weak_ptr.h" | 9 #include "base/memory/weak_ptr.h" |
10 | 10 |
11 namespace base { | 11 namespace base { |
12 class DictionaryValue; | 12 class DictionaryValue; |
13 class MessageLoopProxy; | 13 class SingleThreadTaskRunner; |
14 class TimeDelta; | 14 class TimeDelta; |
15 class WaitableEvent; | 15 class WaitableEvent; |
16 } // namespace base | 16 } // namespace base |
17 | 17 |
18 namespace remoting { | 18 namespace remoting { |
19 namespace policy_hack { | 19 namespace policy_hack { |
20 | 20 |
21 // Watches for changes to the managed remote access host NAT policies. | 21 // Watches for changes to the managed remote access host NAT policies. |
22 // If StartWatching() has been called, then before this object can be deleted, | 22 // If StartWatching() has been called, then before this object can be deleted, |
23 // StopWatching() have completed (the provided |done| event must be signaled). | 23 // StopWatching() have completed (the provided |done| event must be signaled). |
24 class NatPolicy { | 24 class NatPolicy { |
25 public: | 25 public: |
26 // Called with the current status of whether or not NAT traversal is enabled. | 26 // Called with the current status of whether or not NAT traversal is enabled. |
27 typedef base::Callback<void(bool)> NatEnabledCallback; | 27 typedef base::Callback<void(bool)> NatEnabledCallback; |
28 | 28 |
29 explicit NatPolicy(base::MessageLoopProxy* message_loop_proxy); | 29 explicit NatPolicy(scoped_refptr<base::SingleThreadTaskRunner> task_runner); |
30 virtual ~NatPolicy(); | 30 virtual ~NatPolicy(); |
31 | 31 |
32 // This guarantees that the |nat_enabled_cb| is called at least once with | 32 // This guarantees that the |nat_enabled_cb| is called at least once with |
33 // the current policy. After that, |nat_enabled_cb| will be called whenever | 33 // the current policy. After that, |nat_enabled_cb| will be called whenever |
34 // a change to the nat policy is detected. | 34 // a change to the nat policy is detected. |
35 virtual void StartWatching(const NatEnabledCallback& nat_enabled_cb); | 35 virtual void StartWatching(const NatEnabledCallback& nat_enabled_cb); |
36 | 36 |
37 // Should be called after StartWatching() before the object is deleted. Calls | 37 // Should be called after StartWatching() before the object is deleted. Calls |
38 // just wait for |done| to be signaled before deleting the object. | 38 // just wait for |done| to be signaled before deleting the object. |
39 virtual void StopWatching(base::WaitableEvent* done); | 39 virtual void StopWatching(base::WaitableEvent* done); |
40 | 40 |
41 // Implemented by each platform. This message loop should be an IO message | 41 // Implemented by each platform. This message loop should be an IO message |
42 // loop. | 42 // loop. |
43 static NatPolicy* Create(base::MessageLoopProxy* message_loop_proxy); | 43 static NatPolicy* Create( |
| 44 scoped_refptr<base::SingleThreadTaskRunner> task_runner); |
44 | 45 |
45 protected: | 46 protected: |
46 virtual void StartWatchingInternal() = 0; | 47 virtual void StartWatchingInternal() = 0; |
47 virtual void StopWatchingInternal() = 0; | 48 virtual void StopWatchingInternal() = 0; |
48 virtual void Reload() = 0; | 49 virtual void Reload() = 0; |
49 | 50 |
50 // Used to check if the class is on the right thread. | 51 // Used to check if the class is on the right thread. |
51 bool OnPolicyThread() const; | 52 bool OnPolicyThread() const; |
52 | 53 |
53 // Takes the policy dictionary from the OS specific store and extracts the | 54 // Takes the policy dictionary from the OS specific store and extracts the |
54 // NAT traversal setting. | 55 // NAT traversal setting. |
55 void UpdateNatPolicy(base::DictionaryValue* new_policy); | 56 void UpdateNatPolicy(base::DictionaryValue* new_policy); |
56 | 57 |
57 // Used for time-based reloads in case something goes wrong with the | 58 // Used for time-based reloads in case something goes wrong with the |
58 // notification system. | 59 // notification system. |
59 void ScheduleFallbackReloadTask(); | 60 void ScheduleFallbackReloadTask(); |
60 void ScheduleReloadTask(const base::TimeDelta& delay); | 61 void ScheduleReloadTask(const base::TimeDelta& delay); |
61 | 62 |
62 static const char kNatPolicyName[]; | 63 static const char kNatPolicyName[]; |
63 | 64 |
64 private: | 65 private: |
65 scoped_refptr<base::MessageLoopProxy> message_loop_proxy_; | 66 scoped_refptr<base::SingleThreadTaskRunner> task_runner_; |
66 | 67 |
67 NatEnabledCallback nat_enabled_cb_; | 68 NatEnabledCallback nat_enabled_cb_; |
68 bool current_nat_enabled_state_; | 69 bool current_nat_enabled_state_; |
69 bool first_state_published_; | 70 bool first_state_published_; |
70 | 71 |
71 // Allows us to cancel any inflight FileWatcher events or scheduled reloads. | 72 // Allows us to cancel any inflight FileWatcher events or scheduled reloads. |
72 base::WeakPtrFactory<NatPolicy> weak_factory_; | 73 base::WeakPtrFactory<NatPolicy> weak_factory_; |
73 }; | 74 }; |
74 | 75 |
75 } // namespace policy_hack | 76 } // namespace policy_hack |
76 } // namespace remoting | 77 } // namespace remoting |
77 | 78 |
78 #endif // REMOTING_HOST_POLICY_HACK_NAT_POLICY_H_ | 79 #endif // REMOTING_HOST_POLICY_HACK_NAT_POLICY_H_ |
OLD | NEW |