OLD | NEW |
---|---|
(Empty) | |
1 // Copyright 2016 The Chromium Authors. All rights reserved. | |
2 // Use of this source code is governed by a BSD-style license that can be | |
3 // found in the LICENSE file. | |
4 | |
5 #ifndef CONTENT_BROWSER_LOADER_REPLACING_RESOURCE_HANDLER_H_ | |
6 #define CONTENT_BROWSER_LOADER_REPLACING_RESOURCE_HANDLER_H_ | |
7 | |
8 #include <string> | |
9 #include <vector> | |
10 | |
11 #include "base/macros.h" | |
12 #include "base/memory/weak_ptr.h" | |
13 #include "content/browser/loader/layered_resource_handler.h" | |
14 #include "content/common/content_export.h" | |
15 | |
16 namespace net { | |
17 class URLRequest; | |
18 } | |
19 | |
20 namespace content { | |
21 class PluginService; | |
22 class ResourceDispatcherHostImpl; | |
23 struct WebPluginInfo; | |
24 | |
25 // ResourceHandler that initiates special handling of the response if needed, | |
26 // based on the response's MIME type (starts downloads, sends data to some | |
27 // plugin types via a special channel). | |
28 class CONTENT_EXPORT ReplacingResourceHandler : public LayeredResourceHandler { | |
mmenke
2016/06/09 18:09:29
I'm not a big fan of this name. ResourceHandlerRe
clamy
2016/06/21 16:14:15
Done.
| |
29 public: | |
30 // If ENABLE_PLUGINS is defined, |plugin_service| must not be NULL. | |
31 ReplacingResourceHandler(std::unique_ptr<ResourceHandler> next_handler, | |
32 ResourceDispatcherHostImpl* host, | |
33 PluginService* plugin_service, | |
34 net::URLRequest* request); | |
35 ~ReplacingResourceHandler() override; | |
36 | |
37 private: | |
38 enum class State { | |
mmenke
2016/06/09 18:09:29
I think these states need some documentation.
clamy
2016/06/21 16:14:15
Done.
| |
39 STARTING, | |
40 WAITING_FOR_NEW_HANDLER, | |
mmenke
2016/06/09 18:09:29
We only keep this state when we're waiting to get
clamy
2016/06/21 16:14:15
Done.
| |
41 WAITING_FOR_BUFFER_COPY, | |
mmenke
2016/06/09 18:09:29
This state is only needed because the MimeTypeReso
clamy
2016/06/21 16:14:15
Done.
| |
42 CHOICE_MADE, | |
mmenke
2016/06/09 18:09:29
We've also made the choice in WAITING_FOR_BUFFER_C
clamy
2016/06/21 16:14:15
I've called it DONE, since the InterceptingResourc
| |
43 }; | |
mmenke
2016/06/09 18:09:29
nit: Suggest a blank line here.
clamy
2016/06/21 16:14:15
Done.
| |
44 // ResourceHandler implementation: | |
45 bool OnResponseStarted(ResourceResponse* response, bool* defer) override; | |
46 bool OnWillRead(scoped_refptr<net::IOBuffer>* buf, | |
47 int* buf_size, | |
48 int min_size) override; | |
49 bool OnReadCompleted(int bytes_read, bool* defer) override; | |
50 | |
51 bool MakeHandlerChoice(bool* defer); | |
52 | |
53 // Determines whether a plugin will handle the current request, and if so, | |
54 // sets up the handler to direct the response to that plugin. Returns false | |
55 // if there is an error and the request should be cancelled and true | |
56 // otherwise. |defer| is set to true if plugin data is stale and needs to be | |
57 // refreshed before the request can be handled (in this case the function | |
58 // still returns true). If the request is directed to a plugin, | |
59 // |handled_by_plugin| is set to true. | |
60 bool SelectPluginHandler(bool* defer, bool* handled_by_plugin); | |
61 // Returns false if the request should be cancelled. | |
62 bool SelectNextHandler(bool* defer); | |
63 bool UseAlternateNextHandler(std::unique_ptr<ResourceHandler> handler, | |
64 const std::string& payload_for_old_handler); | |
65 | |
66 bool MustDownload(); | |
67 | |
68 // Called on the IO thread once the list of plugins has been loaded. | |
69 void OnPluginsLoaded(const std::vector<WebPluginInfo>& plugins); | |
70 | |
71 bool CopyReadBufferToNextHandler(); | |
72 | |
73 State state_; | |
74 | |
75 ResourceDispatcherHostImpl* host_; | |
76 #if defined(ENABLE_PLUGINS) | |
77 PluginService* plugin_service_; | |
78 #endif | |
79 | |
80 bool switched_handler_; | |
81 | |
82 bool must_download_; | |
83 bool must_download_is_set_; | |
84 | |
85 // Used to copy data to the new next ResourceHandler. | |
86 scoped_refptr<ResourceResponse> response_; | |
87 scoped_refptr<net::IOBuffer> read_buffer_; | |
88 int bytes_read_; | |
89 | |
90 base::WeakPtrFactory<ReplacingResourceHandler> weak_ptr_factory_; | |
91 | |
92 DISALLOW_COPY_AND_ASSIGN(ReplacingResourceHandler); | |
93 }; | |
94 | |
95 } // namespace content | |
96 | |
97 #endif // CONTENT_BROWSER_LOADER_REPLACING_RESOURCE_HANDLER_H_ | |
OLD | NEW |