blob: 4dac8bf86718753ddacf3ea7ca258445200a5b22 [file] [log] [blame]
// Copyright (c) 2013 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 "nacl_io/mount_node_http.h"
#include <assert.h>
#include <errno.h>
#include <stdio.h>
#include <string.h>
#include <ppapi/c/pp_errors.h>
#include "nacl_io/mount_http.h"
#include "nacl_io/osinttypes.h"
#if defined(WIN32)
#define snprintf _snprintf
#endif
namespace nacl_io {
namespace {
// If we're attempting to read a partial request, but the server returns a full
// request, we need to read all of the data up to the start of our partial
// request into a dummy buffer. This is the maximum size of that buffer.
const size_t MAX_READ_BUFFER_SIZE = 64 * 1024;
const int32_t STATUSCODE_OK = 200;
const int32_t STATUSCODE_PARTIAL_CONTENT = 206;
const int32_t STATUSCODE_FORBIDDEN = 403;
const int32_t STATUSCODE_NOT_FOUND = 404;
StringMap_t ParseHeaders(const char* headers, int32_t headers_length) {
enum State {
FINDING_KEY,
SKIPPING_WHITESPACE,
FINDING_VALUE,
};
StringMap_t result;
std::string key;
std::string value;
State state = FINDING_KEY;
const char* start = headers;
for (int i = 0; i < headers_length; ++i) {
switch (state) {
case FINDING_KEY:
if (headers[i] == ':') {
// Found key.
key.assign(start, &headers[i] - start);
key = NormalizeHeaderKey(key);
state = SKIPPING_WHITESPACE;
}
break;
case SKIPPING_WHITESPACE:
if (headers[i] == ' ') {
// Found whitespace, keep going...
break;
}
// Found a non-whitespace, mark this as the start of the value.
start = &headers[i];
state = FINDING_VALUE;
// Fallthrough to start processing value without incrementing i.
case FINDING_VALUE:
if (headers[i] == '\n') {
// Found value.
value.assign(start, &headers[i] - start);
result[key] = value;
start = &headers[i + 1];
state = FINDING_KEY;
}
break;
}
}
return result;
}
bool ParseContentLength(const StringMap_t& headers, size_t* content_length) {
StringMap_t::const_iterator iter = headers.find("Content-Length");
if (iter == headers.end())
return false;
*content_length = strtoul(iter->second.c_str(), NULL, 10);
return true;
}
bool ParseContentRange(const StringMap_t& headers,
size_t* read_start,
size_t* read_end,
size_t* entity_length) {
StringMap_t::const_iterator iter = headers.find("Content-Range");
if (iter == headers.end())
return false;
// The key should look like "bytes ##-##/##" or "bytes ##-##/*". The last
// value is the entity length, which can potentially be * (i.e. unknown).
int read_start_int;
int read_end_int;
int entity_length_int;
int result = sscanf(iter->second.c_str(),
"bytes %" SCNuS "-%" SCNuS "/%" SCNuS,
&read_start_int,
&read_end_int,
&entity_length_int);
// The Content-Range header specifies an inclusive range: e.g. the first ten
// bytes is "bytes 0-9/*". Convert it to a half-open range by incrementing
// read_end.
if (result == 2) {
*read_start = read_start_int;
*read_end = read_end_int + 1;
*entity_length = 0;
return true;
} else if (result == 3) {
*read_start = read_start_int;
*read_end = read_end_int + 1;
*entity_length = entity_length_int;
return true;
}
return false;
}
// Maps an HTTP |status_code| onto the appropriate errno code.
int HTTPStatusCodeToErrno(int status_code) {
switch (status_code) {
case STATUSCODE_OK:
case STATUSCODE_PARTIAL_CONTENT:
return 0;
case STATUSCODE_FORBIDDEN:
return EACCES;
case STATUSCODE_NOT_FOUND:
return ENOENT;
}
if (status_code >= 400 && status_code < 500)
return EINVAL;
return EIO;
}
} // namespace
void MountNodeHttp::SetCachedSize(off_t size) {
has_cached_size_ = true;
stat_.st_size = size;
}
Error MountNodeHttp::FSync() { return ENOSYS; }
Error MountNodeHttp::GetDents(size_t offs,
struct dirent* pdir,
size_t count,
int* out_bytes) {
*out_bytes = 0;
return ENOSYS;
}
Error MountNodeHttp::GetStat(struct stat* stat) {
AUTO_LOCK(node_lock_);
// Assume we need to 'HEAD' if we do not know the size, otherwise, assume
// that the information is constant. We can add a timeout if needed.
MountHttp* mount = static_cast<MountHttp*>(mount_);
if (stat_.st_size == 0 || !mount->cache_stat_) {
StringMap_t headers;
PP_Resource loader;
PP_Resource request;
PP_Resource response;
int32_t statuscode;
StringMap_t response_headers;
Error error = OpenUrl("HEAD",
&headers,
&loader,
&request,
&response,
&statuscode,
&response_headers);
if (error)
return error;
ScopedResource scoped_loader(mount_->ppapi(), loader);
ScopedResource scoped_request(mount_->ppapi(), request);
ScopedResource scoped_response(mount_->ppapi(), response);
size_t entity_length;
if (ParseContentLength(response_headers, &entity_length)) {
SetCachedSize(static_cast<off_t>(entity_length));
} else if (cache_content_ && !has_cached_size_) {
error = DownloadToCache();
// TODO(binji): this error should not be dropped, but it requires a bit
// of a refactor of the tests. See crbug.com/245431
// if (error)
// return error;
} else {
// Don't use SetCachedSize here -- it is actually unknown.
stat_.st_size = 0;
}
stat_.st_atime = 0; // TODO(binji): Use "Last-Modified".
stat_.st_mtime = 0;
stat_.st_ctime = 0;
}
// Fill the stat structure if provided
if (stat)
memcpy(stat, &stat_, sizeof(stat_));
return 0;
}
Error MountNodeHttp::Read(size_t offs,
void* buf,
size_t count,
int* out_bytes) {
*out_bytes = 0;
AUTO_LOCK(node_lock_);
if (cache_content_) {
if (cached_data_.empty()) {
Error error = DownloadToCache();
if (error)
return error;
}
return ReadPartialFromCache(offs, buf, count, out_bytes);
}
return DownloadPartial(offs, buf, count, out_bytes);
}
Error MountNodeHttp::FTruncate(off_t size) { return ENOSYS; }
Error MountNodeHttp::Write(size_t offs,
const void* buf,
size_t count,
int* out_bytes) {
// TODO(binji): support POST?
*out_bytes = 0;
return ENOSYS;
}
Error MountNodeHttp::GetSize(size_t* out_size) {
*out_size = 0;
// TODO(binji): This value should be cached properly; i.e. obey the caching
// headers returned by the server.
AUTO_LOCK(node_lock_);
if (!has_cached_size_) {
// Even if DownloadToCache fails, the best result we can return is what
// was written to stat_.st_size.
if (cache_content_) {
Error error = DownloadToCache();
if (error)
return error;
}
}
*out_size = stat_.st_size;
return 0;
}
MountNodeHttp::MountNodeHttp(Mount* mount,
const std::string& url,
bool cache_content)
: MountNode(mount),
url_(url),
cache_content_(cache_content),
has_cached_size_(false) {}
Error MountNodeHttp::OpenUrl(const char* method,
StringMap_t* request_headers,
PP_Resource* out_loader,
PP_Resource* out_request,
PP_Resource* out_response,
int32_t* out_statuscode,
StringMap_t* out_response_headers) {
// Assume lock_ is already held.
PepperInterface* ppapi = mount_->ppapi();
MountHttp* mount_http = static_cast<MountHttp*>(mount_);
ScopedResource request(
ppapi, mount_http->MakeUrlRequestInfo(url_, method, request_headers));
if (!request.pp_resource())
return EINVAL;
URLLoaderInterface* loader_interface = ppapi->GetURLLoaderInterface();
URLResponseInfoInterface* response_interface =
ppapi->GetURLResponseInfoInterface();
VarInterface* var_interface = ppapi->GetVarInterface();
ScopedResource loader(ppapi, loader_interface->Create(ppapi->GetInstance()));
if (!loader.pp_resource())
return EINVAL;
int32_t result = loader_interface->Open(
loader.pp_resource(), request.pp_resource(), PP_BlockUntilComplete());
if (result != PP_OK)
return PPErrorToErrno(result);
ScopedResource response(
ppapi, loader_interface->GetResponseInfo(loader.pp_resource()));
if (!response.pp_resource())
return EINVAL;
// Get response statuscode.
PP_Var statuscode = response_interface->GetProperty(
response.pp_resource(), PP_URLRESPONSEPROPERTY_STATUSCODE);
if (statuscode.type != PP_VARTYPE_INT32)
return EINVAL;
*out_statuscode = statuscode.value.as_int;
// Only accept OK or Partial Content.
Error error = HTTPStatusCodeToErrno(*out_statuscode);
if (error)
return error;
// Get response headers.
PP_Var response_headers_var = response_interface->GetProperty(
response.pp_resource(), PP_URLRESPONSEPROPERTY_HEADERS);
uint32_t response_headers_length;
const char* response_headers_str =
var_interface->VarToUtf8(response_headers_var, &response_headers_length);
*out_loader = loader.Release();
*out_request = request.Release();
*out_response = response.Release();
*out_response_headers =
ParseHeaders(response_headers_str, response_headers_length);
return 0;
}
Error MountNodeHttp::DownloadToCache() {
StringMap_t headers;
PP_Resource loader;
PP_Resource request;
PP_Resource response;
int32_t statuscode;
StringMap_t response_headers;
Error error = OpenUrl("GET",
&headers,
&loader,
&request,
&response,
&statuscode,
&response_headers);
if (error)
return error;
PepperInterface* ppapi = mount_->ppapi();
ScopedResource scoped_loader(ppapi, loader);
ScopedResource scoped_request(ppapi, request);
ScopedResource scoped_response(ppapi, response);
size_t content_length = 0;
if (ParseContentLength(response_headers, &content_length)) {
cached_data_.resize(content_length);
int real_size;
error = DownloadToBuffer(
loader, cached_data_.data(), content_length, &real_size);
if (error)
return error;
SetCachedSize(real_size);
cached_data_.resize(real_size);
return 0;
}
// We don't know how big the file is. Read in chunks.
cached_data_.resize(MAX_READ_BUFFER_SIZE);
int total_bytes_read = 0;
int bytes_to_read = MAX_READ_BUFFER_SIZE;
while (true) {
char* buf = cached_data_.data() + total_bytes_read;
int bytes_read;
error = DownloadToBuffer(loader, buf, bytes_to_read, &bytes_read);
if (error)
return error;
total_bytes_read += bytes_read;
if (bytes_read < bytes_to_read) {
SetCachedSize(total_bytes_read);
cached_data_.resize(total_bytes_read);
return 0;
}
cached_data_.resize(total_bytes_read + bytes_to_read);
}
}
Error MountNodeHttp::ReadPartialFromCache(size_t offs,
void* buf,
int count,
int* out_bytes) {
*out_bytes = 0;
if (offs > cached_data_.size())
return EINVAL;
count = std::min(count, static_cast<int>(cached_data_.size() - offs));
memcpy(buf, &cached_data_.data()[offs], count);
*out_bytes = count;
return 0;
}
Error MountNodeHttp::DownloadPartial(size_t offs,
void* buf,
size_t count,
int* out_bytes) {
*out_bytes = 0;
StringMap_t headers;
char buffer[100];
// Range request is inclusive: 0-99 returns 100 bytes.
snprintf(&buffer[0],
sizeof(buffer),
"bytes=%" PRIuS "-%" PRIuS,
offs,
offs + count - 1);
headers["Range"] = buffer;
PP_Resource loader;
PP_Resource request;
PP_Resource response;
int32_t statuscode;
StringMap_t response_headers;
Error error = OpenUrl("GET",
&headers,
&loader,
&request,
&response,
&statuscode,
&response_headers);
if (error)
return error;
PepperInterface* ppapi = mount_->ppapi();
ScopedResource scoped_loader(ppapi, loader);
ScopedResource scoped_request(ppapi, request);
ScopedResource scoped_response(ppapi, response);
size_t read_start = 0;
if (statuscode == STATUSCODE_OK) {
// No partial result, read everything starting from the part we care about.
size_t content_length;
if (ParseContentLength(response_headers, &content_length)) {
if (offs >= content_length)
return EINVAL;
// Clamp count, if trying to read past the end of the file.
if (offs + count > content_length) {
count = content_length - offs;
}
}
} else if (statuscode == STATUSCODE_PARTIAL_CONTENT) {
// Determine from the headers where we are reading.
size_t read_end;
size_t entity_length;
if (ParseContentRange(
response_headers, &read_start, &read_end, &entity_length)) {
if (read_start > offs || read_start > read_end) {
// If this error occurs, the server is returning bogus values.
return EINVAL;
}
// Clamp count, if trying to read past the end of the file.
count = std::min(read_end - read_start, count);
} else {
// Partial Content without Content-Range. Assume that the server gave us
// exactly what we asked for. This can happen even when the server
// returns 200 -- the cache may return 206 in this case, but not modify
// the headers.
read_start = offs;
}
}
if (read_start < offs) {
// We aren't yet at the location where we want to start reading. Read into
// our dummy buffer until then.
size_t bytes_to_read = offs - read_start;
if (buffer_.size() < bytes_to_read)
buffer_.resize(std::min(bytes_to_read, MAX_READ_BUFFER_SIZE));
while (bytes_to_read > 0) {
int32_t bytes_read;
Error error =
DownloadToBuffer(loader, buffer_.data(), buffer_.size(), &bytes_read);
if (error)
return error;
bytes_to_read -= bytes_read;
}
}
return DownloadToBuffer(loader, buf, count, out_bytes);
}
Error MountNodeHttp::DownloadToBuffer(PP_Resource loader,
void* buf,
int count,
int* out_bytes) {
*out_bytes = 0;
PepperInterface* ppapi = mount_->ppapi();
URLLoaderInterface* loader_interface = ppapi->GetURLLoaderInterface();
char* out_buffer = static_cast<char*>(buf);
int bytes_to_read = count;
while (bytes_to_read > 0) {
int bytes_read = loader_interface->ReadResponseBody(
loader, out_buffer, bytes_to_read, PP_BlockUntilComplete());
if (bytes_read == 0) {
// This is not an error -- it may just be that we were trying to read
// more data than exists.
*out_bytes = count - bytes_to_read;
return 0;
}
if (bytes_read < 0)
return PPErrorToErrno(bytes_read);
assert(bytes_read <= bytes_to_read);
bytes_to_read -= bytes_read;
out_buffer += bytes_read;
}
*out_bytes = count;
return 0;
}
} // namespace nacl_io