// Copyright (c) 2012 The Chromium Authors. All rights reserved. // Use of this source code is governed by a BSD-style license that can be // found in the LICENSE file. #ifndef CONTENT_BROWSER_LOADER_THROTTLING_RESOURCE_HANDLER_H_ #define CONTENT_BROWSER_LOADER_THROTTLING_RESOURCE_HANDLER_H_ #include "base/memory/ref_counted.h" #include "base/memory/scoped_vector.h" #include "content/browser/loader/layered_resource_handler.h" #include "content/public/browser/resource_controller.h" #include "googleurl/src/gurl.h" namespace content { class ResourceThrottle; struct ResourceResponse; // Used to apply a list of ResourceThrottle instances to an URLRequest. class ThrottlingResourceHandler : public LayeredResourceHandler, public ResourceController { public: // Takes ownership of the ResourceThrottle instances. ThrottlingResourceHandler(scoped_ptr next_handler, int child_id, int request_id, ScopedVector throttles); virtual ~ThrottlingResourceHandler(); // LayeredResourceHandler overrides: virtual bool OnRequestRedirected(int request_id, const GURL& url, ResourceResponse* response, bool* defer) OVERRIDE; virtual bool OnResponseStarted(int request_id, ResourceResponse* response, bool* defer) OVERRIDE; virtual bool OnWillStart(int request_id, const GURL& url, bool* defer) OVERRIDE; // ResourceThrottleController implementation: virtual void Cancel() OVERRIDE; virtual void CancelAndIgnore() OVERRIDE; virtual void CancelWithError(int error_code) OVERRIDE; virtual void Resume() OVERRIDE; private: void ResumeStart(); void ResumeRedirect(); void ResumeResponse(); enum DeferredStage { DEFERRED_NONE, DEFERRED_START, DEFERRED_REDIRECT, DEFERRED_RESPONSE }; DeferredStage deferred_stage_; int request_id_; ScopedVector throttles_; size_t index_; GURL deferred_url_; scoped_refptr deferred_response_; bool cancelled_by_resource_throttle_; }; } // namespace content #endif // CONTENT_BROWSER_LOADER_THROTTLING_RESOURCE_HANDLER_H_