1 // Copyright (c) 2011 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 NET_URL_REQUEST_URL_REQUEST_FILE_DIR_JOB_H_ 6 #define NET_URL_REQUEST_URL_REQUEST_FILE_DIR_JOB_H_ 7 8 #include <string> 9 10 #include "base/files/file_path.h" 11 #include "base/memory/weak_ptr.h" 12 #include "net/base/directory_lister.h" 13 #include "net/url_request/url_request_job.h" 14 15 namespace net { 16 17 class URLRequestFileDirJob 18 : public URLRequestJob, 19 public DirectoryLister::DirectoryListerDelegate { 20 public: 21 URLRequestFileDirJob(URLRequest* request, 22 NetworkDelegate* network_delegate, 23 const base::FilePath& dir_path); 24 25 bool list_complete() const { return list_complete_; } 26 27 virtual void StartAsync(); 28 29 // Overridden from URLRequestJob: 30 virtual void Start() OVERRIDE; 31 virtual void Kill() OVERRIDE; 32 virtual bool ReadRawData(IOBuffer* buf, 33 int buf_size, 34 int* bytes_read) OVERRIDE; 35 virtual bool GetMimeType(std::string* mime_type) const OVERRIDE; 36 virtual bool GetCharset(std::string* charset) OVERRIDE; 37 38 // Overridden from DirectoryLister::DirectoryListerDelegate: 39 virtual void OnListFile( 40 const DirectoryLister::DirectoryListerData& data) OVERRIDE; 41 virtual void OnListDone(int error) OVERRIDE; 42 43 private: 44 virtual ~URLRequestFileDirJob(); 45 46 void CloseLister(); 47 48 // When we have data and a read has been pending, this function 49 // will fill the response buffer and notify the request 50 // appropriately. 51 void CompleteRead(); 52 53 // Fills a buffer with the output. 54 bool FillReadBuffer(char *buf, int buf_size, int *bytes_read); 55 56 DirectoryLister lister_; 57 base::FilePath dir_path_; 58 std::string data_; 59 bool canceled_; 60 61 // Indicates whether we have the complete list of the dir 62 bool list_complete_; 63 64 // Indicates whether we have written the HTML header 65 bool wrote_header_; 66 67 // To simulate Async IO, we hold onto the Reader's buffer while 68 // we wait for IO to complete. When done, we fill the buffer 69 // manually. 70 bool read_pending_; 71 scoped_refptr<IOBuffer> read_buffer_; 72 int read_buffer_length_; 73 base::WeakPtrFactory<URLRequestFileDirJob> weak_factory_; 74 75 DISALLOW_COPY_AND_ASSIGN(URLRequestFileDirJob); 76 }; 77 78 } // namespace net 79 80 #endif // NET_URL_REQUEST_URL_REQUEST_FILE_DIR_JOB_H_ 81