net/url_request/url_request_file_dir_job.cc

   1 // Copyright (c) 2012 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 #include "net/url_request/url_request_file_dir_job.h"
   6
   7 #include "base/bind.h"
   8 #include "base/compiler_specific.h"
   9 #include "base/message_loop.h"
  10 #include "base/sys_string_conversions.h"
  11 #include "base/utf_string_conversions.h"
  12 #include "base/time.h"
  13 #include "googleurl/src/gurl.h"
  14 #include "net/base/io_buffer.h"
  15 #include "net/base/net_errors.h"
  16 #include "net/base/net_util.h"
  17 #include "net/url_request/url_request_status.h"
  18
  19 #if defined(OS_POSIX)
  20 #include <sys/stat.h>
  21 #endif
  22
  23 namespace net {
  24
  25 URLRequestFileDirJob::URLRequestFileDirJob(URLRequest* request,
  26                                            NetworkDelegate* network_delegate,
  27                                            const FilePath& dir_path)
  28     : URLRequestJob(request, network_delegate),
  29       ALLOW_THIS_IN_INITIALIZER_LIST(lister_(dir_path, this)),
  30       dir_path_(dir_path),
  31       canceled_(false),
  32       list_complete_(false),
  33       wrote_header_(false),
  34       read_pending_(false),
  35       read_buffer_length_(0),
  36       ALLOW_THIS_IN_INITIALIZER_LIST(weak_factory_(this)) {
  37 }
  38
  39 void URLRequestFileDirJob::StartAsync() {
  40   lister_.Start();
  41
  42   NotifyHeadersComplete();
  43 }
  44
  45 void URLRequestFileDirJob::Start() {
  46   // Start reading asynchronously so that all error reporting and data
  47   // callbacks happen as they would for network requests.
  48   MessageLoop::current()->PostTask(
  49       FROM_HERE,
  50       base::Bind(&URLRequestFileDirJob::StartAsync,
  51                  weak_factory_.GetWeakPtr()));
  52 }
  53
  54 void URLRequestFileDirJob::Kill() {
  55   if (canceled_)
  56     return;
  57
  58   canceled_ = true;
  59
  60   if (!list_complete_)
  61     lister_.Cancel();
  62
  63   URLRequestJob::Kill();
  64
  65   weak_factory_.InvalidateWeakPtrs();
  66 }
  67
  68 bool URLRequestFileDirJob::ReadRawData(IOBuffer* buf, int buf_size,
  69                                        int* bytes_read) {
  70   DCHECK(bytes_read);
  71   *bytes_read = 0;
  72
  73   if (is_done())
  74     return true;
  75
  76   if (FillReadBuffer(buf->data(), buf_size, bytes_read))
  77     return true;
  78
  79   // We are waiting for more data
  80   read_pending_ = true;
  81   read_buffer_ = buf;
  82   read_buffer_length_ = buf_size;
  83   SetStatus(URLRequestStatus(URLRequestStatus::IO_PENDING, 0));
  84   return false;
  85 }
  86
  87 bool URLRequestFileDirJob::GetMimeType(std::string* mime_type) const {
  88   *mime_type = "text/html";
  89   return true;
  90 }
  91
  92 bool URLRequestFileDirJob::GetCharset(std::string* charset) {
  93   // All the filenames are converted to UTF-8 before being added.
  94   *charset = "utf-8";
  95   return true;
  96 }
  97
  98 void URLRequestFileDirJob::OnListFile(
  99     const DirectoryLister::DirectoryListerData& data) {
 100   // We wait to write out the header until we get the first file, so that we
 101   // can catch errors from DirectoryLister and show an error page.
 102   if (!wrote_header_) {
 103 #if defined(OS_WIN)
 104     const string16& title = dir_path_.value();
 105 #elif defined(OS_POSIX)
 106     // TODO(jungshik): Add SysNativeMBToUTF16 to sys_string_conversions.
 107     // On Mac, need to add NFKC->NFC conversion either here or in file_path.
 108     // On Linux, the file system encoding is not defined, but we assume that
 109     // SysNativeMBToWide takes care of it at least for now. We can try something
 110     // more sophisticated if necessary later.
 111     const string16& title = WideToUTF16(
 112         base::SysNativeMBToWide(dir_path_.value()));
 113 #endif
 114     data_.append(GetDirectoryListingHeader(title));
 115     wrote_header_ = true;
 116   }
 117
 118 #if defined(OS_WIN)
 119   int64 size = (static_cast<unsigned __int64>(data.info.nFileSizeHigh) << 32) |
 120       data.info.nFileSizeLow;
 121
 122   // Note that we should not convert ftLastWriteTime to the local time because
 123   // ICU's datetime formatting APIs expect time in UTC and take into account
 124   // the timezone before formatting.
 125   data_.append(GetDirectoryListingEntry(
 126       data.info.cFileName,
 127       std::string(),
 128       ((data.info.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) != 0),
 129       size,
 130       base::Time::FromFileTime(data.info.ftLastWriteTime)));
 131 #elif defined(OS_POSIX)
 132   // TOOD(jungshik): The same issue as for the directory name.
 133   data_.append(GetDirectoryListingEntry(
 134       WideToUTF16(base::SysNativeMBToWide(data.info.filename)),
 135       data.info.filename,
 136       S_ISDIR(data.info.stat.st_mode),
 137       data.info.stat.st_size,
 138       base::Time::FromTimeT(data.info.stat.st_mtime)));
 139 #endif
 140
 141   // TODO(darin): coalesce more?
 142   CompleteRead();
 143 }
 144
 145 void URLRequestFileDirJob::OnListDone(int error) {
 146   DCHECK(!canceled_);
 147   if (error != OK) {
 148     read_pending_ = false;
 149     NotifyDone(URLRequestStatus(URLRequestStatus::FAILED, error));
 150   } else {
 151     list_complete_ = true;
 152     CompleteRead();
 153   }
 154 }
 155
 156 URLRequestFileDirJob::~URLRequestFileDirJob() {}
 157
 158 void URLRequestFileDirJob::CompleteRead() {
 159   if (read_pending_) {
 160     int bytes_read;
 161     if (FillReadBuffer(read_buffer_->data(), read_buffer_length_,
 162                        &bytes_read)) {
 163       // We completed the read, so reset the read buffer.
 164       read_pending_ = false;
 165       read_buffer_ = NULL;
 166       read_buffer_length_ = 0;
 167
 168       SetStatus(URLRequestStatus());
 169       NotifyReadComplete(bytes_read);
 170     } else {
 171       NOTREACHED();
 172       // TODO: Better error code.
 173       NotifyDone(URLRequestStatus(URLRequestStatus::FAILED, 0));
 174     }
 175   }
 176 }
 177
 178 bool URLRequestFileDirJob::FillReadBuffer(char* buf, int buf_size,
 179                                           int* bytes_read) {
 180   DCHECK(bytes_read);
 181
 182   *bytes_read = 0;
 183
 184   int count = std::min(buf_size, static_cast<int>(data_.size()));
 185   if (count) {
 186     memcpy(buf, &data_[0], count);
 187     data_.erase(0, count);
 188     *bytes_read = count;
 189     return true;
 190   } else if (list_complete_) {
 191     // EOF
 192     return true;
 193   }
 194   return false;
 195 }
 196
 197 }  // namespace net