1 /*
2  * Copyright (c) 2022 Huawei Device Co., Ltd.
3  * Licensed under the Apache License, Version 2.0 (the "License");
4  * you may not use this file except in compliance with the License.
5  * You may obtain a copy of the License at
6  *
7  *     http://www.apache.org/licenses/LICENSE-2.0
8  *
9  * Unless required by applicable law or agreed to in writing, software
10  * distributed under the License is distributed on an "AS IS" BASIS,
11  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12  * See the License for the specific language governing permissions and
13  * limitations under the License.
14  */
15 
16 #include "ecmascript/platform/map.h"
17 
18 #include <io.h>
19 #include <windows.h>
20 
21 #ifdef ERROR
22 #undef ERROR
23 #endif
24 
25 #ifdef VOID
26 #undef VOID
27 #endif
28 
29 #ifdef CONST
30 #undef CONST
31 #endif
32 
33 #include "ecmascript/log_wrapper.h"
34 #include "ecmascript/mem/mem.h"
35 
36 namespace panda::ecmascript {
37 static constexpr int INSUFFICIENT_CONTINUOUS_MEM = 1455;
38 
PageMap(size_t size, int prot, size_t alignment, void *addr, [[maybe_unused]] int flags)39 MemMap PageMap(size_t size, int prot, size_t alignment, void *addr, [[maybe_unused]] int flags)
40 {
41     ASSERT(size == AlignUp(size, PageSize()));
42     ASSERT(alignment == AlignUp(alignment, PageSize()));
43     size_t allocSize = size + alignment;
44     void *result = VirtualAlloc(addr, allocSize, MEM_COMMIT, prot);
45     if (result == nullptr) {
46         int errCode = GetLastError();
47         if (errCode == INSUFFICIENT_CONTINUOUS_MEM) {
48             LOG_NO_TAG(ERROR) << "[ArkRuntime Log]Failed to request a continuous segment of " << size
49                               << " virtual memory. Please clean up other heavy processes or restart the computer.";
50         }
51         LOG_ECMA(FATAL) << "PageMap "<< size << ", prot:" << prot << " fail, the error code is: " << errCode;
52     }
53     if (alignment != 0) {
54         auto alignResult = AlignUp(reinterpret_cast<uintptr_t>(result), alignment);
55         return MemMap(result, reinterpret_cast<void *>(alignResult), size);
56     }
57     return MemMap(result, result, size);
58 }
59 
PageUnmap(MemMap it)60 void PageUnmap(MemMap it)
61 {
62     if (!VirtualFree(it.GetOriginAddr(), 0, MEM_RELEASE)) {
63         int errCode = GetLastError();
64         LOG_ECMA(ERROR) << "PageUnmap failed, error code is " << errCode;
65     }
66 }
67 
MachineCodePageMap(size_t size, int prot, size_t alignment)68 MemMap MachineCodePageMap(size_t size, int prot, size_t alignment)
69 {
70     MemMap memMap = PageMap(size, prot, alignment);
71     PageTag(memMap.GetMem(), memMap.GetSize(), PageTagType::MACHINE_CODE);
72     return memMap;
73 }
74 
MachineCodePageUnmap(MemMap it)75 void MachineCodePageUnmap(MemMap it)
76 {
77     PageClearTag(it.GetMem(), it.GetSize());
78     if (!PageProtect(it.GetMem(), it.GetSize(), PAGE_PROT_NONE)) {
79         return;
80     }
81     PageUnmap(it);
82 }
83 
PageRelease([[maybe_unused]] void *mem, [[maybe_unused]] size_t size)84 void PageRelease([[maybe_unused]] void *mem, [[maybe_unused]] size_t size)
85 {
86 }
87 
PagePreRead([[maybe_unused]] void *mem, [[maybe_unused]] size_t size)88 void PagePreRead([[maybe_unused]] void *mem, [[maybe_unused]] size_t size)
89 {
90 }
91 
PageTag([[maybe_unused]] void *mem, [[maybe_unused]] size_t size, [[maybe_unused]] PageTagType type, [[maybe_unused]] const std::string &spaceName, [[maybe_unused]] const uint32_t threadId)92 void PageTag([[maybe_unused]] void *mem, [[maybe_unused]] size_t size, [[maybe_unused]] PageTagType type,
93              [[maybe_unused]] const std::string &spaceName, [[maybe_unused]] const uint32_t threadId)
94 {
95 }
96 
PageClearTag([[maybe_unused]] void *mem, [[maybe_unused]] size_t size)97 void PageClearTag([[maybe_unused]] void *mem, [[maybe_unused]] size_t size)
98 {
99 }
100 
PageProtect(void *mem, size_t size, int prot)101 bool PageProtect(void *mem, size_t size, int prot)
102 {
103     [[maybe_unused]] DWORD oldProtect;
104     if (!VirtualProtect(mem, size, prot, &oldProtect)) {
105         int errCode = GetLastError();
106         LOG_ECMA(ERROR) << "PageProtect mem = " << mem << ", size = " << size <<
107             ", change to " << prot << " failed, error code is " << errCode;
108         return false;
109     }
110     return true;
111 }
112 
PageSize()113 size_t PageSize()
114 {
115     SYSTEM_INFO info;
116     GetSystemInfo(&info);
117     return info.dwPageSize;
118 }
119 }  // namespace panda::ecmascript
120