blob: 0b05cbb9dd5899b3a2187d19efab598ced7a4c73 [file] [log] [blame]
// Copyright (c) 2011 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.
#include "net/url_request/url_request_file_dir_job.h"
#include "base/bind.h"
#include "base/compiler_specific.h"
#include "base/file_util.h"
#include "base/message_loop.h"
#include "base/sys_string_conversions.h"
#include "base/utf_string_conversions.h"
#include "base/time.h"
#include "googleurl/src/gurl.h"
#include "net/base/io_buffer.h"
#include "net/base/net_errors.h"
#include "net/base/net_util.h"
#include "net/url_request/url_request.h"
#if defined(OS_POSIX)
#include <sys/stat.h>
#endif
namespace net {
URLRequestFileDirJob::URLRequestFileDirJob(URLRequest* request,
const FilePath& dir_path)
: URLRequestJob(request),
ALLOW_THIS_IN_INITIALIZER_LIST(lister_(dir_path, this)),
dir_path_(dir_path),
canceled_(false),
list_complete_(false),
wrote_header_(false),
read_pending_(false),
read_buffer_length_(0),
ALLOW_THIS_IN_INITIALIZER_LIST(weak_factory_(this)) {
}
void URLRequestFileDirJob::StartAsync() {
lister_.Start();
NotifyHeadersComplete();
}
void URLRequestFileDirJob::Start() {
// Start reading asynchronously so that all error reporting and data
// callbacks happen as they would for network requests.
MessageLoop::current()->PostTask(
FROM_HERE,
base::Bind(&URLRequestFileDirJob::StartAsync,
weak_factory_.GetWeakPtr()));
}
void URLRequestFileDirJob::Kill() {
if (canceled_)
return;
canceled_ = true;
if (!list_complete_)
lister_.Cancel();
URLRequestJob::Kill();
weak_factory_.InvalidateWeakPtrs();
}
bool URLRequestFileDirJob::ReadRawData(IOBuffer* buf, int buf_size,
int *bytes_read) {
DCHECK(bytes_read);
*bytes_read = 0;
if (is_done())
return true;
if (FillReadBuffer(buf->data(), buf_size, bytes_read))
return true;
// We are waiting for more data
read_pending_ = true;
read_buffer_ = buf;
read_buffer_length_ = buf_size;
SetStatus(URLRequestStatus(URLRequestStatus::IO_PENDING, 0));
return false;
}
bool URLRequestFileDirJob::GetMimeType(std::string* mime_type) const {
*mime_type = "text/html";
return true;
}
bool URLRequestFileDirJob::GetCharset(std::string* charset) {
// All the filenames are converted to UTF-8 before being added.
*charset = "utf-8";
return true;
}
void URLRequestFileDirJob::OnListFile(
const DirectoryLister::DirectoryListerData& data) {
// We wait to write out the header until we get the first file, so that we
// can catch errors from DirectoryLister and show an error page.
if (!wrote_header_) {
#if defined(OS_WIN)
const string16& title = dir_path_.value();
#elif defined(OS_POSIX)
// TODO(jungshik): Add SysNativeMBToUTF16 to sys_string_conversions.
// On Mac, need to add NFKC->NFC conversion either here or in file_path.
// On Linux, the file system encoding is not defined, but we assume that
// SysNativeMBToWide takes care of it at least for now. We can try something
// more sophisticated if necessary later.
const string16& title = WideToUTF16(
base::SysNativeMBToWide(dir_path_.value()));
#endif
data_.append(GetDirectoryListingHeader(title));
wrote_header_ = true;
}
#if defined(OS_WIN)
int64 size = (static_cast<unsigned __int64>(data.info.nFileSizeHigh) << 32) |
data.info.nFileSizeLow;
// Note that we should not convert ftLastWriteTime to the local time because
// ICU's datetime formatting APIs expect time in UTC and take into account
// the timezone before formatting.
data_.append(GetDirectoryListingEntry(
data.info.cFileName, std::string(),
(data.info.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) ? true : false,
size,
base::Time::FromFileTime(data.info.ftLastWriteTime)));
#elif defined(OS_POSIX)
// TOOD(jungshik): The same issue as for the directory name.
data_.append(GetDirectoryListingEntry(
WideToUTF16(base::SysNativeMBToWide(data.info.filename)),
data.info.filename,
S_ISDIR(data.info.stat.st_mode),
data.info.stat.st_size,
base::Time::FromTimeT(data.info.stat.st_mtime)));
#endif
// TODO(darin): coalesce more?
CompleteRead();
}
void URLRequestFileDirJob::OnListDone(int error) {
DCHECK(!canceled_);
if (error != OK) {
read_pending_ = false;
NotifyDone(URLRequestStatus(URLRequestStatus::FAILED, error));
} else {
list_complete_ = true;
CompleteRead();
}
}
URLRequestFileDirJob::~URLRequestFileDirJob() {}
void URLRequestFileDirJob::CompleteRead() {
if (read_pending_) {
int bytes_read;
if (FillReadBuffer(read_buffer_->data(), read_buffer_length_,
&bytes_read)) {
// We completed the read, so reset the read buffer.
read_pending_ = false;
read_buffer_ = NULL;
read_buffer_length_ = 0;
SetStatus(URLRequestStatus());
NotifyReadComplete(bytes_read);
} else {
NOTREACHED();
// TODO: Better error code.
NotifyDone(URLRequestStatus(URLRequestStatus::FAILED, 0));
}
}
}
bool URLRequestFileDirJob::FillReadBuffer(char *buf, int buf_size,
int *bytes_read) {
DCHECK(bytes_read);
*bytes_read = 0;
int count = std::min(buf_size, static_cast<int>(data_.size()));
if (count) {
memcpy(buf, &data_[0], count);
data_.erase(0, count);
*bytes_read = count;
return true;
} else if (list_complete_) {
// EOF
return true;
}
return false;
}
} // namespace net