blob: 35a2100dcb4e41f7e7c115e7561914d2a3f9f9e8 [file] [log] [blame]
// Copyright (c) 2006-2008 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 "chrome/renderer/visitedlink_slave.h"
#include "base/logging.h"
#include "base/shared_memory.h"
#include "chrome/common/render_messages.h"
#include "third_party/WebKit/Source/WebKit/chromium/public/WebView.h"
using WebKit::WebView;
VisitedLinkSlave::VisitedLinkSlave() : shared_memory_(NULL) {
}
VisitedLinkSlave::~VisitedLinkSlave() {
FreeTable();
}
bool VisitedLinkSlave::OnControlMessageReceived(const IPC::Message& message) {
bool handled = true;
IPC_BEGIN_MESSAGE_MAP(VisitedLinkSlave, message)
IPC_MESSAGE_HANDLER(ChromeViewMsg_VisitedLink_NewTable,
OnUpdateVisitedLinks)
IPC_MESSAGE_HANDLER(ChromeViewMsg_VisitedLink_Add, OnAddVisitedLinks)
IPC_MESSAGE_HANDLER(ChromeViewMsg_VisitedLink_Reset, OnResetVisitedLinks)
IPC_MESSAGE_UNHANDLED(handled = false)
IPC_END_MESSAGE_MAP()
return handled;
}
// This function's job is to initialize the table with the given
// shared memory handle. This memory is mapped into the process.
void VisitedLinkSlave::OnUpdateVisitedLinks(base::SharedMemoryHandle table) {
DCHECK(base::SharedMemory::IsHandleValid(table)) << "Bad table handle";
// since this function may be called again to change the table, we may need
// to free old objects
FreeTable();
DCHECK(shared_memory_ == NULL && hash_table_ == NULL);
// create the shared memory object
shared_memory_ = new base::SharedMemory(table, true);
if (!shared_memory_)
return;
// map the header into our process so we can see how long the rest is,
// and set the salt
if (!shared_memory_->Map(sizeof(SharedHeader)))
return;
SharedHeader* header =
static_cast<SharedHeader*>(shared_memory_->memory());
DCHECK(header);
int32 table_len = header->length;
memcpy(salt_, header->salt, sizeof(salt_));
shared_memory_->Unmap();
// now do the whole table because we know the length
if (!shared_memory_->Map(sizeof(SharedHeader) +
table_len * sizeof(Fingerprint))) {
shared_memory_->Close();
return;
}
// commit the data
DCHECK(shared_memory_->memory());
hash_table_ = reinterpret_cast<Fingerprint*>(
static_cast<char*>(shared_memory_->memory()) + sizeof(SharedHeader));
table_length_ = table_len;
}
void VisitedLinkSlave::OnAddVisitedLinks(
const VisitedLinkSlave::Fingerprints& fingerprints) {
for (size_t i = 0; i < fingerprints.size(); ++i)
WebView::updateVisitedLinkState(fingerprints[i]);
}
void VisitedLinkSlave::OnResetVisitedLinks() {
WebView::resetVisitedLinkState();
}
void VisitedLinkSlave::FreeTable() {
if (shared_memory_) {
delete shared_memory_;
shared_memory_ = NULL;
}
hash_table_ = NULL;
table_length_ = 0;
}