components/visitedlink/renderer/visitedlink_slave.cc

   1 // Copyright (c) 2006-2008 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 "components/visitedlink/renderer/visitedlink_slave.h"
   6
   7 #include "base/logging.h"
   8 #include "base/memory/shared_memory.h"
   9 #include "components/visitedlink/common/visitedlink_messages.h"
  10 #include "third_party/WebKit/public/web/WebView.h"
  11
  12 using WebKit::WebView;
  13
  14 namespace visitedlink {
  15
  16 VisitedLinkSlave::VisitedLinkSlave() : shared_memory_(NULL) {}
  17
  18 VisitedLinkSlave::~VisitedLinkSlave() {
  19   FreeTable();
  20 }
  21
  22 bool VisitedLinkSlave::OnControlMessageReceived(const IPC::Message& message) {
  23   bool handled = true;
  24   IPC_BEGIN_MESSAGE_MAP(VisitedLinkSlave, message)
  25     IPC_MESSAGE_HANDLER(ChromeViewMsg_VisitedLink_NewTable,
  26                         OnUpdateVisitedLinks)
  27     IPC_MESSAGE_HANDLER(ChromeViewMsg_VisitedLink_Add, OnAddVisitedLinks)
  28     IPC_MESSAGE_HANDLER(ChromeViewMsg_VisitedLink_Reset, OnResetVisitedLinks)
  29     IPC_MESSAGE_UNHANDLED(handled = false)
  30   IPC_END_MESSAGE_MAP()
  31   return handled;
  32 }
  33
  34 // This function's job is to initialize the table with the given
  35 // shared memory handle. This memory is mapped into the process.
  36 void VisitedLinkSlave::OnUpdateVisitedLinks(base::SharedMemoryHandle table) {
  37   DCHECK(base::SharedMemory::IsHandleValid(table)) << "Bad table handle";
  38   // since this function may be called again to change the table, we may need
  39   // to free old objects
  40   FreeTable();
  41   DCHECK(shared_memory_ == NULL && hash_table_ == NULL);
  42
  43   // create the shared memory object
  44   shared_memory_ = new base::SharedMemory(table, true);
  45   if (!shared_memory_)
  46     return;
  47
  48   // map the header into our process so we can see how long the rest is,
  49   // and set the salt
  50   if (!shared_memory_->Map(sizeof(SharedHeader)))
  51     return;
  52   SharedHeader* header =
  53     static_cast<SharedHeader*>(shared_memory_->memory());
  54   DCHECK(header);
  55   int32 table_len = header->length;
  56   memcpy(salt_, header->salt, sizeof(salt_));
  57   shared_memory_->Unmap();
  58
  59   // now do the whole table because we know the length
  60   if (!shared_memory_->Map(sizeof(SharedHeader) +
  61                           table_len * sizeof(Fingerprint))) {
  62     shared_memory_->Close();
  63     return;
  64   }
  65
  66   // commit the data
  67   DCHECK(shared_memory_->memory());
  68   hash_table_ = reinterpret_cast<Fingerprint*>(
  69       static_cast<char*>(shared_memory_->memory()) + sizeof(SharedHeader));
  70   table_length_ = table_len;
  71 }
  72
  73 void VisitedLinkSlave::OnAddVisitedLinks(
  74     const VisitedLinkSlave::Fingerprints& fingerprints) {
  75   for (size_t i = 0; i < fingerprints.size(); ++i)
  76     WebView::updateVisitedLinkState(fingerprints[i]);
  77 }
  78
  79 void VisitedLinkSlave::OnResetVisitedLinks() {
  80   WebView::resetVisitedLinkState();
  81 }
  82
  83 void VisitedLinkSlave::FreeTable() {
  84   if (shared_memory_) {
  85     delete shared_memory_;
  86     shared_memory_ = NULL;
  87   }
  88   hash_table_ = NULL;
  89   table_length_ = 0;
  90 }
  91
  92 }  // namespace visitedlink