OLD | NEW |
1 // Copyright (c) 2010 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2010 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 CHROME_BROWSER_NET_PASSIVE_LOG_COLLECTOR_H_ | 5 #ifndef CHROME_BROWSER_NET_PASSIVE_LOG_COLLECTOR_H_ |
6 #define CHROME_BROWSER_NET_PASSIVE_LOG_COLLECTOR_H_ | 6 #define CHROME_BROWSER_NET_PASSIVE_LOG_COLLECTOR_H_ |
7 | 7 |
8 #include <vector> | 8 #include <vector> |
9 | 9 |
10 #include "base/hash_tables.h" | 10 #include "base/hash_tables.h" |
11 #include "chrome/browser/net/chrome_net_log.h" | 11 #include "chrome/browser/net/chrome_net_log.h" |
12 #include "net/base/net_log.h" | 12 #include "net/base/net_log.h" |
13 | 13 |
14 class PassiveLogCollector : public ChromeNetLog::Observer { | 14 class PassiveLogCollector : public ChromeNetLog::Observer { |
15 public: | 15 public: |
16 struct RequestInfo { | 16 struct RequestInfo { |
17 RequestInfo() : num_entries_truncated(0) {} | 17 RequestInfo() : num_entries_truncated(0) {} |
18 std::string url; | 18 std::string url; |
19 std::vector<net::NetLog::Entry> entries; | 19 net::CapturingNetLog::EntryList entries; |
20 size_t num_entries_truncated; | 20 size_t num_entries_truncated; |
21 }; | 21 }; |
22 | 22 |
23 typedef std::vector<RequestInfo> RequestInfoList; | 23 typedef std::vector<RequestInfo> RequestInfoList; |
24 | 24 |
25 // This class stores and manages the passively logged information for | 25 // This class stores and manages the passively logged information for |
26 // URLRequests/SocketStreams/ConnectJobs. | 26 // URLRequests/SocketStreams/ConnectJobs. |
27 class RequestTrackerBase { | 27 class RequestTrackerBase { |
28 public: | 28 public: |
29 explicit RequestTrackerBase(size_t max_graveyard_size); | 29 explicit RequestTrackerBase(size_t max_graveyard_size); |
30 | 30 |
31 void OnAddEntry(const net::NetLog::Entry& entry); | 31 void OnAddEntry(const net::CapturingNetLog::Entry& entry); |
32 | 32 |
33 RequestInfoList GetLiveRequests() const; | 33 RequestInfoList GetLiveRequests() const; |
34 void ClearRecentlyDeceased(); | 34 void ClearRecentlyDeceased(); |
35 RequestInfoList GetRecentlyDeceased() const; | 35 RequestInfoList GetRecentlyDeceased() const; |
36 void SetUnbounded(bool unbounded); | 36 void SetUnbounded(bool unbounded); |
37 | 37 |
38 bool IsUnbounded() const { return is_unbounded_; } | 38 bool IsUnbounded() const { return is_unbounded_; } |
39 | 39 |
40 void Clear(); | 40 void Clear(); |
41 | 41 |
42 const RequestInfo* GetRequestInfoFromGraveyard(int id) const; | 42 const RequestInfo* GetRequestInfoFromGraveyard(int id) const; |
43 | 43 |
44 protected: | 44 protected: |
45 enum Action { | 45 enum Action { |
46 ACTION_NONE, | 46 ACTION_NONE, |
47 ACTION_DELETE, | 47 ACTION_DELETE, |
48 ACTION_MOVE_TO_GRAVEYARD, | 48 ACTION_MOVE_TO_GRAVEYARD, |
49 }; | 49 }; |
50 | 50 |
51 // Updates |out_info| with the information from |entry|. Returns an action | 51 // Updates |out_info| with the information from |entry|. Returns an action |
52 // to perform for this map entry on completion. | 52 // to perform for this map entry on completion. |
53 virtual Action DoAddEntry(const net::NetLog::Entry& entry, | 53 virtual Action DoAddEntry(const net::CapturingNetLog::Entry& entry, |
54 RequestInfo* out_info) = 0; | 54 RequestInfo* out_info) = 0; |
55 | 55 |
56 bool is_unbounded() const { return is_unbounded_; } | 56 bool is_unbounded() const { return is_unbounded_; } |
57 | 57 |
58 private: | 58 private: |
59 typedef base::hash_map<int, RequestInfo> SourceIDToInfoMap; | 59 typedef base::hash_map<int, RequestInfo> SourceIDToInfoMap; |
60 | 60 |
61 bool HandleNotificationOfConnectJobID(const net::NetLog::Entry& entry, | |
62 RequestInfo* live_entry); | |
63 | |
64 void RemoveFromLiveRequests(int source_id); | 61 void RemoveFromLiveRequests(int source_id); |
65 void InsertIntoGraveyard(const RequestInfo& info); | 62 void InsertIntoGraveyard(const RequestInfo& info); |
66 | 63 |
67 SourceIDToInfoMap live_requests_; | 64 SourceIDToInfoMap live_requests_; |
68 size_t max_graveyard_size_; | 65 size_t max_graveyard_size_; |
69 size_t next_graveyard_index_; | 66 size_t next_graveyard_index_; |
70 RequestInfoList graveyard_; | 67 RequestInfoList graveyard_; |
71 bool is_unbounded_; | 68 bool is_unbounded_; |
72 | 69 |
73 DISALLOW_COPY_AND_ASSIGN(RequestTrackerBase); | 70 DISALLOW_COPY_AND_ASSIGN(RequestTrackerBase); |
74 }; | 71 }; |
75 | 72 |
76 // Specialization of RequestTrackerBase for handling ConnectJobs. | 73 // Specialization of RequestTrackerBase for handling ConnectJobs. |
77 class ConnectJobTracker : public RequestTrackerBase { | 74 class ConnectJobTracker : public RequestTrackerBase { |
78 public: | 75 public: |
79 static const size_t kMaxGraveyardSize; | 76 static const size_t kMaxGraveyardSize; |
80 | 77 |
81 ConnectJobTracker(); | 78 ConnectJobTracker(); |
82 | 79 |
83 protected: | 80 protected: |
84 virtual Action DoAddEntry(const net::NetLog::Entry& entry, | 81 virtual Action DoAddEntry(const net::CapturingNetLog::Entry& entry, |
85 RequestInfo* out_info); | 82 RequestInfo* out_info); |
86 private: | 83 private: |
87 DISALLOW_COPY_AND_ASSIGN(ConnectJobTracker); | 84 DISALLOW_COPY_AND_ASSIGN(ConnectJobTracker); |
88 }; | 85 }; |
89 | 86 |
90 // Specialization of RequestTrackerBase for handling URLRequest/SocketStream. | 87 // Specialization of RequestTrackerBase for handling URLRequest/SocketStream. |
91 class RequestTracker : public RequestTrackerBase { | 88 class RequestTracker : public RequestTrackerBase { |
92 public: | 89 public: |
93 static const size_t kMaxGraveyardSize; | 90 static const size_t kMaxGraveyardSize; |
94 static const size_t kMaxGraveyardURLSize; | 91 static const size_t kMaxGraveyardURLSize; |
95 | 92 |
96 explicit RequestTracker(ConnectJobTracker* connect_job_tracker); | 93 explicit RequestTracker(ConnectJobTracker* connect_job_tracker); |
97 | 94 |
98 protected: | 95 protected: |
99 virtual Action DoAddEntry(const net::NetLog::Entry& entry, | 96 virtual Action DoAddEntry(const net::CapturingNetLog::Entry& entry, |
100 RequestInfo* out_info); | 97 RequestInfo* out_info); |
101 | 98 |
102 private: | 99 private: |
103 // Searches through |connect_job_tracker_| for information on the | 100 // Searches through |connect_job_tracker_| for information on the |
104 // ConnectJob specified in |entry|, and appends it to |live_entry|. | 101 // ConnectJob specified in |entry|, and appends it to |live_entry|. |
105 void AddConnectJobInfo(const net::NetLog::Entry& entry, | 102 void AddConnectJobInfo(const net::CapturingNetLog::Entry& entry, |
106 RequestInfo* live_entry); | 103 RequestInfo* live_entry); |
107 | 104 |
108 ConnectJobTracker* connect_job_tracker_; | 105 ConnectJobTracker* connect_job_tracker_; |
109 | 106 |
110 DISALLOW_COPY_AND_ASSIGN(RequestTracker); | 107 DISALLOW_COPY_AND_ASSIGN(RequestTracker); |
111 }; | 108 }; |
112 | 109 |
113 // Tracks the log entries for the last seen SOURCE_INIT_PROXY_RESOLVER. | 110 // Tracks the log entries for the last seen SOURCE_INIT_PROXY_RESOLVER. |
114 class InitProxyResolverTracker { | 111 class InitProxyResolverTracker { |
115 public: | 112 public: |
116 InitProxyResolverTracker(); | 113 InitProxyResolverTracker(); |
117 | 114 |
118 void OnAddEntry(const net::NetLog::Entry& entry); | 115 void OnAddEntry(const net::CapturingNetLog::Entry& entry); |
119 | 116 |
120 const std::vector<net::NetLog::Entry>& entries() const { | 117 const net::CapturingNetLog::EntryList& entries() const { |
121 return entries_; | 118 return entries_; |
122 } | 119 } |
123 | 120 |
124 private: | 121 private: |
125 std::vector<net::NetLog::Entry> entries_; | 122 net::CapturingNetLog::EntryList entries_; |
126 DISALLOW_COPY_AND_ASSIGN(InitProxyResolverTracker); | 123 DISALLOW_COPY_AND_ASSIGN(InitProxyResolverTracker); |
127 }; | 124 }; |
128 | 125 |
129 PassiveLogCollector(); | 126 PassiveLogCollector(); |
130 ~PassiveLogCollector(); | 127 ~PassiveLogCollector(); |
131 | 128 |
132 // Observer implementation: | 129 // Observer implementation: |
133 virtual void OnAddEntry(const net::NetLog::Entry& entry); | 130 virtual void OnAddEntry(net::NetLog::EventType type, |
| 131 const base::TimeTicks& time, |
| 132 const net::NetLog::Source& source, |
| 133 net::NetLog::EventPhase phase, |
| 134 net::NetLog::EventParameters* extra_parameters); |
134 | 135 |
135 // Clears all of the passively logged data. | 136 // Clears all of the passively logged data. |
136 void Clear(); | 137 void Clear(); |
137 | 138 |
138 RequestTracker* url_request_tracker() { | 139 RequestTracker* url_request_tracker() { |
139 return &url_request_tracker_; | 140 return &url_request_tracker_; |
140 } | 141 } |
141 | 142 |
142 RequestTracker* socket_stream_tracker() { | 143 RequestTracker* socket_stream_tracker() { |
143 return &socket_stream_tracker_; | 144 return &socket_stream_tracker_; |
144 } | 145 } |
145 | 146 |
146 InitProxyResolverTracker* init_proxy_resolver_tracker() { | 147 InitProxyResolverTracker* init_proxy_resolver_tracker() { |
147 return &init_proxy_resolver_tracker_; | 148 return &init_proxy_resolver_tracker_; |
148 } | 149 } |
149 | 150 |
150 private: | 151 private: |
151 ConnectJobTracker connect_job_tracker_; | 152 ConnectJobTracker connect_job_tracker_; |
152 RequestTracker url_request_tracker_; | 153 RequestTracker url_request_tracker_; |
153 RequestTracker socket_stream_tracker_; | 154 RequestTracker socket_stream_tracker_; |
154 InitProxyResolverTracker init_proxy_resolver_tracker_; | 155 InitProxyResolverTracker init_proxy_resolver_tracker_; |
155 | 156 |
156 DISALLOW_COPY_AND_ASSIGN(PassiveLogCollector); | 157 DISALLOW_COPY_AND_ASSIGN(PassiveLogCollector); |
157 }; | 158 }; |
158 | 159 |
159 #endif // CHROME_BROWSER_NET_PASSIVE_LOG_COLLECTOR_H_ | 160 #endif // CHROME_BROWSER_NET_PASSIVE_LOG_COLLECTOR_H_ |
OLD | NEW |