VisitedLinkStore.cpp [plain text]
#include "config.h"
#include "VisitedLinkStore.h"
#include "VisitedLinkStoreMessages.h"
#include "VisitedLinkTableControllerMessages.h"
#include "WebPageProxy.h"
#include "WebProcessMessages.h"
#include "WebProcessPool.h"
#include "WebProcessProxy.h"
namespace WebKit {
using namespace WebCore;
Ref<VisitedLinkStore> VisitedLinkStore::create()
{
return adoptRef(*new VisitedLinkStore);
}
VisitedLinkStore::~VisitedLinkStore()
{
RELEASE_ASSERT(m_processes.computesEmpty());
}
VisitedLinkStore::VisitedLinkStore()
: m_linkHashStore(*this)
{
}
void VisitedLinkStore::addProcess(WebProcessProxy& process)
{
ASSERT(!m_processes.contains(process));
if (!m_processes.add(process).isNewEntry)
return;
process.addMessageReceiver(Messages::VisitedLinkStore::messageReceiverName(), identifier(), *this);
if (m_linkHashStore.isEmpty())
return;
sendStoreHandleToProcess(process);
}
void VisitedLinkStore::removeProcess(WebProcessProxy& process)
{
ASSERT(m_processes.contains(process));
if (!m_processes.remove(process))
return;
process.removeMessageReceiver(Messages::VisitedLinkStore::messageReceiverName(), identifier());
}
void VisitedLinkStore::addVisitedLinkHash(SharedStringHash linkHash)
{
m_linkHashStore.scheduleAddition(linkHash);
}
bool VisitedLinkStore::containsVisitedLinkHash(WebCore::SharedStringHash linkHash)
{
return m_linkHashStore.contains(linkHash);
}
void VisitedLinkStore::removeVisitedLinkHash(WebCore::SharedStringHash linkHash)
{
m_linkHashStore.scheduleRemoval(linkHash);
}
void VisitedLinkStore::removeAll()
{
m_linkHashStore.clear();
for (auto& process : m_processes) {
ASSERT(process.processPool().processes().contains(&process));
process.send(Messages::VisitedLinkTableController::RemoveAllVisitedLinks(), identifier());
}
}
void VisitedLinkStore::addVisitedLinkHashFromPage(WebPageProxyIdentifier pageProxyID, SharedStringHash linkHash)
{
if (auto* webPageProxy = WebProcessProxy::webPage(pageProxyID)) {
if (!webPageProxy->addsVisitedLinks())
return;
}
addVisitedLinkHash(linkHash);
}
void VisitedLinkStore::sendStoreHandleToProcess(WebProcessProxy& process)
{
ASSERT(process.processPool().processes().contains(&process));
SharedMemory::Handle handle;
if (!m_linkHashStore.createSharedMemoryHandle(handle))
return;
#if OS(DARWIN) || OS(WINDOWS)
uint64_t dataSize = handle.size();
#else
uint64_t dataSize = 0;
#endif
process.send(Messages::VisitedLinkTableController::SetVisitedLinkTable(SharedMemory::IPCHandle { WTFMove(handle), dataSize }), identifier());
}
void VisitedLinkStore::didInvalidateSharedMemory()
{
for (auto& process : m_processes)
sendStoreHandleToProcess(process);
}
void VisitedLinkStore::didUpdateSharedStringHashes(const Vector<WebCore::SharedStringHash>& addedHashes, const Vector<WebCore::SharedStringHash>& removedHashes)
{
ASSERT(!addedHashes.isEmpty() || !removedHashes.isEmpty());
for (auto& process : m_processes) {
ASSERT(process.processPool().processes().contains(&process));
if (addedHashes.size() > 20 || !removedHashes.isEmpty())
process.send(Messages::VisitedLinkTableController::AllVisitedLinkStateChanged(), identifier());
else
process.send(Messages::VisitedLinkTableController::VisitedLinkStateChanged(addedHashes), identifier());
}
}
}