1/*
2 * Copyright (C) 2008 Apple Inc. All Rights Reserved.
3 *
4 * Redistribution and use in source and binary forms, with or without
5 * modification, are permitted provided that the following conditions
6 * are met:
7 * 1. Redistributions of source code must retain the above copyright
8 *    notice, this list of conditions and the following disclaimer.
9 * 2. Redistributions in binary form must reproduce the above copyright
10 *    notice, this list of conditions and the following disclaimer in the
11 *    documentation and/or other materials provided with the distribution.
12 *
13 * THIS SOFTWARE IS PROVIDED BY APPLE INC. ``AS IS'' AND ANY
14 * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
15 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
16 * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL APPLE INC. OR
17 * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
18 * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
19 * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
20 * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY
21 * OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
22 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
23 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
24 */
25
26#include "config.h"
27#include "StorageThread.h"
28
29#include "StorageAreaSync.h"
30#include <wtf/AutodrainedPool.h>
31#include <wtf/HashSet.h>
32#include <wtf/MainThread.h>
33
34namespace WebCore {
35
36static HashSet<StorageThread*>& activeStorageThreads()
37{
38    ASSERT(isMainThread());
39    DEPRECATED_DEFINE_STATIC_LOCAL(HashSet<StorageThread*>, threads, ());
40    return threads;
41}
42
43StorageThread::StorageThread()
44    : m_threadID(0)
45{
46    ASSERT(isMainThread());
47}
48
49StorageThread::~StorageThread()
50{
51    ASSERT(isMainThread());
52    ASSERT(!m_threadID);
53}
54
55bool StorageThread::start()
56{
57    ASSERT(isMainThread());
58    if (!m_threadID)
59        m_threadID = createThread(StorageThread::threadEntryPointCallback, this, "WebCore: LocalStorage");
60    activeStorageThreads().add(this);
61    return m_threadID;
62}
63
64void StorageThread::threadEntryPointCallback(void* thread)
65{
66    static_cast<StorageThread*>(thread)->threadEntryPoint();
67}
68
69void StorageThread::threadEntryPoint()
70{
71    ASSERT(!isMainThread());
72
73    while (auto function = m_queue.waitForMessage()) {
74        AutodrainedPool pool;
75        (*function)();
76    }
77}
78
79void StorageThread::dispatch(const Function<void ()>& function)
80{
81    ASSERT(isMainThread());
82    ASSERT(!m_queue.killed() && m_threadID);
83    m_queue.append(std::make_unique<Function<void ()>>(function));
84}
85
86void StorageThread::terminate()
87{
88    ASSERT(isMainThread());
89    ASSERT(!m_queue.killed() && m_threadID);
90    activeStorageThreads().remove(this);
91    // Even in weird, exceptional cases, don't wait on a nonexistent thread to terminate.
92    if (!m_threadID)
93        return;
94
95    m_queue.append(std::make_unique<Function<void ()>>(bind(&StorageThread::performTerminate, this)));
96    waitForThreadCompletion(m_threadID);
97    ASSERT(m_queue.killed());
98    m_threadID = 0;
99}
100
101void StorageThread::performTerminate()
102{
103    ASSERT(!isMainThread());
104    m_queue.kill();
105}
106
107void StorageThread::releaseFastMallocFreeMemoryInAllThreads()
108{
109    HashSet<StorageThread*>& threads = activeStorageThreads();
110
111    for (HashSet<StorageThread*>::iterator it = threads.begin(), end = threads.end(); it != end; ++it)
112        (*it)->dispatch(bind(WTF::releaseFastMallocFreeMemory));
113}
114
115}
116