1 /*
2 * Copyright (c) 2023 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 "file_mapper.h"
17 #include <sys/mman.h>
18 #include "zip_file_reader.h"
19
20 namespace panda {
21 namespace ecmascript {
22 namespace {
23 long g_pageSize = 0;
24 const int32_t MAP_XPM = 0x40;
25 }
FileMapper()26 FileMapper::FileMapper()
27 {
28 if (g_pageSize <= 0) {
29 g_pageSize = sysconf(_SC_PAGESIZE);
30 }
31 }
32
~FileMapper()33 FileMapper::~FileMapper()
34 {
35 if (basePtr_ != nullptr && type_ == FileMapperType::SHARED_MMAP) {
36 munmap(basePtr_, baseLen_);
37 }
38 }
39
CreateFileMapper(const std::string & fileName,bool compress,int32_t fd,size_t offset,size_t len,FileMapperType type)40 bool FileMapper::CreateFileMapper(const std::string &fileName, bool compress,
41 int32_t fd, size_t offset, size_t len, FileMapperType type)
42 {
43 if (fd < 0 || len == 0 || type == FileMapperType::NORMAL_MEM) {
44 return false;
45 }
46 if (g_pageSize <= 0) {
47 return false;
48 }
49 if (dataLen_ > 0) {
50 return false;
51 }
52
53 size_t adjust = offset % static_cast<size_t>(g_pageSize);
54 size_t adjOffset = offset - adjust;
55 baseLen_ = len + adjust;
56 int32_t mmapFlag = MAP_PRIVATE | MAP_XPM;
57 if (type == FileMapperType::SHARED_MMAP) {
58 mmapFlag = MAP_SHARED;
59 }
60 basePtr_ = reinterpret_cast<uint8_t*>(mmap(nullptr, baseLen_, PROT_READ,
61 mmapFlag, fd, adjOffset));
62 if (basePtr_ == MAP_FAILED) {
63 baseLen_ = 0;
64 return false;
65 }
66
67 isCompressed = compress;
68 fileName_ = fileName;
69 offset_ = offset;
70 dataLen_ = len;
71 usePtr_ = reinterpret_cast<uint8_t *>(basePtr_) + adjust;
72 type_ = type;
73 return true;
74 }
75
CreateFileMapper(std::shared_ptr<ZipFileReader> fileReader,const std::string & fileName,size_t offset,size_t len,bool compress)76 bool FileMapper::CreateFileMapper(std::shared_ptr<ZipFileReader> fileReader, const std::string &fileName,
77 size_t offset, size_t len, bool compress)
78 {
79 if (!fileReader) {
80 return false;
81 }
82 if (!fileName_.empty()) {
83 return false;
84 }
85
86 dataPtr_ = std::make_unique<uint8_t[]>(len);
87 if (!fileReader->ReadBuffer(dataPtr_.get(), offset, len)) {
88 return false;
89 }
90 isCompressed = compress;
91 dataLen_ = len;
92 offset_ = offset;
93 fileName_ = fileName;
94
95 return true;
96 }
97
IsCompressed()98 bool FileMapper::IsCompressed()
99 {
100 return isCompressed;
101 }
102
GetDataPtr()103 uint8_t* FileMapper::GetDataPtr()
104 {
105 return dataPtr_ ? dataPtr_.get() : usePtr_;
106 }
107
GetDataLen()108 size_t FileMapper::GetDataLen()
109 {
110 return dataLen_;
111 }
112
GetFileName()113 std::string FileMapper::GetFileName()
114 {
115 return fileName_;
116 }
117
GetOffset()118 int32_t FileMapper::GetOffset()
119 {
120 return offset_;
121 }
122 } // namespace AbilityBase
123 } // namespace OHOS
124