1 // Copyright 2014 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_SERVICE_WORKER_SERVICE_WORKER_WRITE_TO_CACHE_JOB_H_ 6 #define CONTENT_BROWSER_SERVICE_WORKER_SERVICE_WORKER_WRITE_TO_CACHE_JOB_H_ 7 8 #include <string> 9 10 #include "base/memory/weak_ptr.h" 11 #include "content/browser/service_worker/service_worker_disk_cache.h" 12 #include "content/browser/service_worker/service_worker_version.h" 13 #include "content/common/content_export.h" 14 #include "content/common/service_worker/service_worker_status_code.h" 15 #include "content/common/service_worker/service_worker_types.h" 16 #include "content/public/common/resource_type.h" 17 #include "net/url_request/url_request.h" 18 #include "net/url_request/url_request_job.h" 19 20 namespace content { 21 22 class ServiceWorkerContextCore; 23 class ServiceWorkerResponseWriter; 24 class ServiceWorkerVersions; 25 26 // A URLRequestJob derivative used to cache the main script 27 // and its imports during the initial install of a new version. 28 // Another separate URLRequest is started which will perform 29 // a network fetch. The response produced for that separate 30 // request is written to the service worker script cache and piped 31 // to the consumer of the ServiceWorkerWriteToCacheJob for delivery 32 // to the renderer process housing the worker. 33 class CONTENT_EXPORT ServiceWorkerWriteToCacheJob 34 : public net::URLRequestJob, 35 public net::URLRequest::Delegate { 36 public: 37 ServiceWorkerWriteToCacheJob( 38 net::URLRequest* request, 39 net::NetworkDelegate* network_delegate, 40 ResourceType resource_type, 41 base::WeakPtr<ServiceWorkerContextCore> context, 42 ServiceWorkerVersion* version, 43 int extra_load_flags, 44 int64 response_id); 45 46 private: 47 FRIEND_TEST_ALL_PREFIXES(ServiceWorkerContextRequestHandlerTest, 48 UpdateBefore24Hours); 49 FRIEND_TEST_ALL_PREFIXES(ServiceWorkerContextRequestHandlerTest, 50 UpdateAfter24Hours); 51 52 virtual ~ServiceWorkerWriteToCacheJob(); 53 54 // net::URLRequestJob overrides 55 virtual void Start() OVERRIDE; 56 virtual void Kill() OVERRIDE; 57 virtual net::LoadState GetLoadState() const OVERRIDE; 58 virtual bool GetCharset(std::string* charset) OVERRIDE; 59 virtual bool GetMimeType(std::string* mime_type) const OVERRIDE; 60 virtual void GetResponseInfo(net::HttpResponseInfo* info) OVERRIDE; 61 virtual int GetResponseCode() const OVERRIDE; 62 virtual void SetExtraRequestHeaders( 63 const net::HttpRequestHeaders& headers) OVERRIDE; 64 virtual bool ReadRawData(net::IOBuffer* buf, 65 int buf_size, 66 int *bytes_read) OVERRIDE; 67 68 const net::HttpResponseInfo* http_info() const; 69 70 // Methods to drive the net request forward and 71 // write data to the disk cache. 72 void InitNetRequest(int extra_load_flags); 73 void StartNetRequest(); 74 net::URLRequestStatus ReadNetData( 75 net::IOBuffer* buf, 76 int buf_size, 77 int *bytes_read); 78 void WriteHeadersToCache(); 79 void OnWriteHeadersComplete(int result); 80 void WriteDataToCache(int bytes_to_write); 81 void OnWriteDataComplete(int result); 82 83 // net::URLRequest::Delegate overrides that observe the net request. 84 virtual void OnReceivedRedirect( 85 net::URLRequest* request, 86 const net::RedirectInfo& redirect_info, 87 bool* defer_redirect) OVERRIDE; 88 virtual void OnAuthRequired( 89 net::URLRequest* request, 90 net::AuthChallengeInfo* auth_info) OVERRIDE; 91 virtual void OnCertificateRequested( 92 net::URLRequest* request, 93 net::SSLCertRequestInfo* cert_request_info) OVERRIDE; 94 virtual void OnSSLCertificateError( 95 net::URLRequest* request, 96 const net::SSLInfo& ssl_info, 97 bool fatal) OVERRIDE; 98 virtual void OnBeforeNetworkStart( 99 net::URLRequest* request, 100 bool* defer) OVERRIDE; 101 virtual void OnResponseStarted(net::URLRequest* request) OVERRIDE; 102 virtual void OnReadCompleted( 103 net::URLRequest* request, 104 int bytes_read) OVERRIDE; 105 106 void AsyncNotifyDoneHelper(const net::URLRequestStatus& status); 107 108 ResourceType resource_type_; // Differentiate main script and imports 109 scoped_refptr<net::IOBuffer> io_buffer_; 110 scoped_refptr<HttpResponseInfoIOBuffer> info_buffer_; 111 base::WeakPtr<ServiceWorkerContextCore> context_; 112 GURL url_; 113 int64 response_id_; 114 scoped_ptr<net::URLRequest> net_request_; 115 scoped_ptr<net::HttpResponseInfo> http_info_; 116 scoped_ptr<ServiceWorkerResponseWriter> writer_; 117 scoped_refptr<ServiceWorkerVersion> version_; 118 bool has_been_killed_; 119 bool did_notify_started_; 120 bool did_notify_finished_; 121 base::WeakPtrFactory<ServiceWorkerWriteToCacheJob> weak_factory_; 122 123 DISALLOW_COPY_AND_ASSIGN(ServiceWorkerWriteToCacheJob); 124 }; 125 126 } // namespace content 127 128 #endif // CONTENT_BROWSER_SERVICE_WORKER_SERVICE_WORKER_WRITE_TO_CACHE_JOB_H_ 129