Gardening: fix CLoop build.
[WebKit-https.git] / Source / JavaScriptCore / assembler / ProbeStack.cpp
1 /*
2  * Copyright (C) 2017 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 "ProbeStack.h"
28
29 #include <memory>
30
31 #if ENABLE(MASM_PROBE)
32
33 namespace JSC {
34 namespace Probe {
35
36 Page::Page(void* baseAddress)
37     : m_baseLogicalAddress(baseAddress)
38 {
39     memcpy(&m_buffer, baseAddress, s_pageSize);
40 }
41
42 void Page::flushWrites()
43 {
44     uintptr_t dirtyBits = m_dirtyBits;
45     size_t offset = 0;
46     while (dirtyBits) {
47         // Find start.
48         if (dirtyBits & 1) {
49             size_t startOffset = offset;
50             // Find end.
51             do {
52                 dirtyBits = dirtyBits >> 1;
53                 offset += s_chunkSize;
54             } while (dirtyBits & 1);
55
56             size_t size = offset - startOffset;
57             uint8_t* src = reinterpret_cast<uint8_t*>(&m_buffer) + startOffset;
58             uint8_t* dst = reinterpret_cast<uint8_t*>(m_baseLogicalAddress) + startOffset;
59             memcpy(dst, src, size);
60         }
61         dirtyBits = dirtyBits >> 1;
62         offset += s_chunkSize;
63     }
64     m_dirtyBits = 0;
65 }
66
67 Stack::Stack(Stack&& other)
68     : m_newStackPointer(other.m_newStackPointer)
69     , m_lowWatermark(other.m_lowWatermark)
70     , m_stackBounds(WTFMove(other.m_stackBounds))
71     , m_pages(WTFMove(other.m_pages))
72 {
73 #if !ASSERT_DISABLED
74     other.m_isValid = false;
75 #endif
76 }
77
78 bool Stack::hasWritesToFlush()
79 {
80     return std::any_of(m_pages.begin(), m_pages.end(), [] (auto& it) { return it.value->hasWritesToFlush(); });
81 }
82
83 void Stack::flushWrites()
84 {
85     for (auto it = m_pages.begin(); it != m_pages.end(); ++it)
86         it->value->flushWritesIfNeeded();
87 }
88
89 Page* Stack::ensurePageFor(void* address)
90 {
91     // Since the machine stack is always allocated in units of whole pages, asserting
92     // that the address is contained in the stack is sufficient to infer that its page
93     // is also contained in the stack.
94     RELEASE_ASSERT(m_stackBounds.contains(address));
95
96     // We may have gotten here because of a cache miss. So, look up the page first
97     // before allocating a new one,
98     void* baseAddress = Page::baseAddressFor(address);
99     auto it = m_pages.find(baseAddress);
100     if (LIKELY(it != m_pages.end()))
101         m_lastAccessedPage = it->value.get();
102     else {
103         std::unique_ptr<Page> page = std::make_unique<Page>(baseAddress);
104         auto result = m_pages.add(baseAddress, WTFMove(page));
105         m_lastAccessedPage = result.iterator->value.get();
106     }
107     m_lastAccessedPageBaseAddress = baseAddress;
108     return m_lastAccessedPage;
109 }
110
111 } // namespace Probe
112 } // namespace JSC
113
114 #endif // ENABLE(MASM_PROBE)