1 //===- Win32/MappedFile.cpp - Win32 MappedFile Implementation ---*- C++ -*-===//
3 // The LLVM Compiler Infrastructure
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
8 //===----------------------------------------------------------------------===//
10 // This file provides the Win32 implementation of the MappedFile concept.
12 //===----------------------------------------------------------------------===//
14 //===----------------------------------------------------------------------===//
15 //=== WARNING: Implementation here must contain only Win32 code.
16 //===----------------------------------------------------------------------===//
19 #include "llvm/System/Process.h"
24 struct sys::MappedFileInfo {
30 bool MappedFile::initialize(std::string* ErrMsg) {
32 info_ = new MappedFileInfo;
33 info_->hFile = INVALID_HANDLE_VALUE;
34 info_->hMapping = NULL;
36 DWORD mode = options_ & WRITE_ACCESS ? GENERIC_WRITE : GENERIC_READ;
37 DWORD disposition = options_ & WRITE_ACCESS ? OPEN_ALWAYS : OPEN_EXISTING;
38 DWORD share = options_ & WRITE_ACCESS ? FILE_SHARE_WRITE : FILE_SHARE_READ;
39 share = options_ & SHARED_MAPPING ? share : 0;
40 info_->hFile = CreateFile(path_.c_str(), mode, share, NULL, disposition,
41 FILE_ATTRIBUTE_NORMAL, NULL);
42 if (info_->hFile == INVALID_HANDLE_VALUE) {
45 return MakeErrMsg(ErrMsg,
46 std::string("Can't open file: ") + path_.toString());
50 if (!GetFileSizeEx(info_->hFile, &size) ||
51 (info_->size = size_t(size.QuadPart), info_->size != size.QuadPart)) {
52 CloseHandle(info_->hFile);
55 return MakeErrMsg(ErrMsg,
56 std::string("Can't get size of file: ") + path_.toString());
62 void MappedFile::terminate() {
64 if (info_->hFile != INVALID_HANDLE_VALUE)
65 CloseHandle(info_->hFile);
70 void MappedFile::unmap() {
71 assert(info_ && "MappedFile not initialized");
73 UnmapViewOfFile(base_);
76 if (info_->hMapping != INVALID_HANDLE_VALUE) {
77 CloseHandle(info_->hMapping);
78 info_->hMapping = NULL;
82 void* MappedFile::map(std::string* ErrMsg) {
84 DWORD prot = PAGE_READONLY;
85 if (options_ & EXEC_ACCESS)
87 else if (options_ & WRITE_ACCESS)
88 prot = PAGE_READWRITE;
89 info_->hMapping = CreateFileMapping(info_->hFile, NULL, prot, 0, 0, NULL);
90 if (info_->hMapping == NULL) {
91 MakeErrMsg(ErrMsg, std::string("Can't map file: ") + path_.toString());
95 prot = (options_ & WRITE_ACCESS) ? FILE_MAP_WRITE : FILE_MAP_READ;
96 base_ = MapViewOfFileEx(info_->hMapping, prot, 0, 0, 0, NULL);
98 CloseHandle(info_->hMapping);
99 info_->hMapping = NULL;
100 MakeErrMsg(ErrMsg, std::string("Can't map file: ") + path_.toString());
107 size_t MappedFile::size() const {
108 assert(info_ && "MappedFile not initialized");
112 bool MappedFile::size(size_t new_size, std::string* ErrMsg) {
113 assert(info_ && "MappedFile not initialized");
115 // Take the mapping out of memory.
118 // Adjust the new_size to a page boundary.
119 size_t pagesizem1 = Process::GetPageSize() - 1;
120 new_size = (new_size + pagesizem1) & ~pagesizem1;
122 // If the file needs to be extended, do so.
123 if (new_size > info_->size) {
125 eof.QuadPart = new_size;
126 if (!SetFilePointerEx(info_->hFile, eof, NULL, FILE_BEGIN))
127 return MakeErrMsg(ErrMsg,
128 std::string("Can't set end of file: ") + path_.toString());
129 if (!SetEndOfFile(info_->hFile))
130 return MakeErrMsg(ErrMsg,
131 std::string("Can't set end of file: ") + path_.toString());
132 info_->size = new_size;