1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
|
//===- Unix/MappedFile.cpp - Unix MappedFile Implementation -----*- C++ -*-===//
//
// The LLVM Compiler Infrastructure
//
// This file was developed by Reid Spencer and is distributed under the
// University of Illinois Open Source License. See LICENSE.TXT for details.
//
//===----------------------------------------------------------------------===//
//
// This file provides the generic Unix implementation of the MappedFile concept.
//
//===----------------------------------------------------------------------===//
//===----------------------------------------------------------------------===//
//=== WARNING: Implementation here must contain only generic UNIX code that
//=== is guaranteed to work on *all* UNIX variants.
//===----------------------------------------------------------------------===//
#include "Unix.h"
#include "llvm/System/Process.h"
#ifdef HAVE_FCNTL_H
#include <fcntl.h>
#endif
#ifdef HAVE_SYS_MMAN_H
#include <sys/mman.h>
#endif
#ifdef HAVE_SYS_STAT_H
#include <sys/stat.h>
#endif
namespace llvm {
using namespace sys;
struct sys::MappedFileInfo {
int FD;
off_t Size;
};
void MappedFile::initialize() {
if (!path_.exists())
throw std::string("Can't open file: ") + path_.toString();
int mode = 0;
if (options_ & READ_ACCESS)
if (options_ & WRITE_ACCESS)
mode = O_RDWR;
else
mode = O_RDONLY;
else if (options_ & WRITE_ACCESS)
mode = O_WRONLY;
int FD = ::open(path_.c_str(), mode);
if (FD < 0)
ThrowErrno(std::string("Can't open file: ") + path_.toString());
struct stat sbuf;
if(::fstat(FD, &sbuf) < 0) {
::close(FD);
ThrowErrno(std::string("Can't stat file: ") + path_.toString());
}
info_ = new MappedFileInfo;
info_->FD = FD;
info_->Size = sbuf.st_size;
}
void MappedFile::terminate() {
assert(info_ && "MappedFile not initialized");
::close(info_->FD);
delete info_;
info_ = 0;
}
void MappedFile::unmap() {
assert(info_ && "MappedFile not initialized");
if (isMapped()) {
if (options_ & WRITE_ACCESS)
::msync(base_, info_->Size, MS_SYNC);
::munmap(base_, info_->Size);
}
}
void* MappedFile::map() {
assert(info_ && "MappedFile not initialized");
if (!isMapped()) {
int prot = PROT_NONE;
int flags = 0;
#ifdef MAP_FILE
flags |= MAP_FILE;
#endif
if (options_ == 0) {
prot = PROT_READ;
flags = MAP_PRIVATE;
} else {
if (options_ & READ_ACCESS)
prot |= PROT_READ;
if (options_ & WRITE_ACCESS)
prot |= PROT_WRITE;
if (options_ & EXEC_ACCESS)
prot |= PROT_EXEC;
if (options_ & SHARED_MAPPING)
flags |= MAP_SHARED;
else
flags |= MAP_PRIVATE;
}
size_t map_size = ((info_->Size / Process::GetPageSize())+1) *
Process::GetPageSize();
base_ = ::mmap(0, map_size, prot, flags, info_->FD, 0);
if (base_ == MAP_FAILED)
ThrowErrno(std::string("Can't map file:") + path_.toString());
}
return base_;
}
size_t MappedFile::size() const {
assert(info_ && "MappedFile not initialized");
return info_->Size;
}
void MappedFile::size(size_t new_size) {
assert(info_ && "MappedFile not initialized");
// Take the mapping out of memory
this->unmap();
// Adjust the current size to a page boundary
size_t cur_size = ((info_->Size / Process::GetPageSize())+1) *
Process::GetPageSize();
// Adjust the new_size to a page boundary
new_size = ((new_size / Process::GetPageSize())+1) *
Process::GetPageSize();
// If the file needs to be extended
if (new_size > cur_size) {
// Ensure we can allocate at least the idodes necessary to handle the
// file size requested.
::lseek(info_->FD, new_size, SEEK_SET);
::write(info_->FD, "\0", 1);
}
// Seek to current end of file.
this->map();
}
}
|