15821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Copyright (c) 2012 The Chromium Authors. All rights reserved. 25821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Use of this source code is governed by a BSD-style license that can be 35821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// found in the LICENSE file. 45821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 55821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "courgette/memory_allocator.h" 65821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 75821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <map> 85821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 91320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci#include "base/files/file_util.h" 105e3f23d412006dc4db4e659864679f29341e113fTorne (Richard Coles)#include "base/strings/stringprintf.h" 115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 122a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#if defined(OS_WIN) 135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 14a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles)namespace { 155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 16a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles)// The file is created in the %TEMP% folder. 17a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles)// NOTE: Since the file will be used as backing for a memory allocation, 18a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles)// it will never be so big that size_t cannot represent its size. 19a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles)base::File CreateTempFile() { 202a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::FilePath path; 21a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles) if (!base::CreateTemporaryFile(&path)) 22a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) return base::File(); 235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 24a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) int flags = base::File::FLAG_OPEN_ALWAYS | base::File::FLAG_READ | 25a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) base::File::FLAG_WRITE | base::File::FLAG_DELETE_ON_CLOSE | 26a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) base::File::FLAG_TEMPORARY; 27a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) return base::File(path, flags); 285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 30a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles)} // namespace 315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 32a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles)namespace courgette { 335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// FileMapping 355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)FileMapping::FileMapping() : mapping_(NULL), view_(NULL) { 375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)FileMapping::~FileMapping() { 405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) Close(); 415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)bool FileMapping::InitializeView(size_t size) { 445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DCHECK(view_ == NULL); 455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DCHECK(mapping_ != NULL); 465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) view_ = ::MapViewOfFile(mapping_, FILE_MAP_WRITE, 0, 0, size); 475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!view_) { 485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) Close(); 495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return false; 505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return true; 525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)bool FileMapping::Create(HANDLE file, size_t size) { 555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DCHECK(file != INVALID_HANDLE_VALUE); 565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DCHECK(!valid()); 575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) mapping_ = ::CreateFileMapping(file, NULL, PAGE_READWRITE, 0, 0, NULL); 585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!mapping_) 595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return false; 605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return InitializeView(size); 625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void FileMapping::Close() { 655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (view_) 665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ::UnmapViewOfFile(view_); 675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (mapping_) 685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ::CloseHandle(mapping_); 695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) mapping_ = NULL; 705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) view_ = NULL; 715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)bool FileMapping::valid() const { 745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return view_ != NULL; 755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void* FileMapping::view() const { 785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return view_; 795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// TempMapping 825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)TempMapping::TempMapping() { 845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)TempMapping::~TempMapping() { 875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)bool TempMapping::Initialize(size_t size) { 90a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) file_ = CreateTempFile(); 91a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) if (!file_.IsValid()) 92a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) return false; 93a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) 945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // TODO(tommi): The assumption here is that the alignment of pointers (this) 955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // is as strict or stricter than the alignment of the element type. This is 965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // not always true, e.g. __m128 has 16-byte alignment. 975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) size += sizeof(this); 98a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) if (!file_.SetLength(size) || 99a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) !mapping_.Create(file_.GetPlatformFile(), size)) { 1005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) file_.Close(); 1015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return false; 1025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) TempMapping** write = reinterpret_cast<TempMapping**>(mapping_.view()); 1055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) write[0] = this; 1065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return true; 1085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void* TempMapping::memory() const { 1115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) uint8* mem = reinterpret_cast<uint8*>(mapping_.view()); 1125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // The 'this' pointer is written at the start of mapping_.view(), so 1135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // go past it. (See Initialize()). 1145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (mem) 1155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) mem += sizeof(this); 1165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DCHECK(mem); 1175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return mem; 1185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)bool TempMapping::valid() const { 1215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return mapping_.valid(); 1225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// static 1255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)TempMapping* TempMapping::GetMappingFromPtr(void* mem) { 1265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) TempMapping* ret = NULL; 1275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (mem) { 1285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ret = reinterpret_cast<TempMapping**>(mem)[-1]; 1295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DCHECK(ret); 1315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return ret; 1325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} // namespace courgette 1355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#endif // OS_WIN 137