mirror of
https://github.com/mozilla/gecko-dev.git
synced 2024-10-21 17:25:36 +00:00
a89b67e772
Backed out changeset a8518ea4b594 (bug 1479960)
Backed out changeset 7172762c4b87 (bug 1536697)
Backed out changeset 2ea5ccb8f3a1 (bug 1426526)
Backed out changeset d892a888fe9c (bug 1426526)
Backed out changeset 2c4d12bdfec3 (bug 1479960)
Backed out changeset 8a322064cf6d (bug 1479960)
Backed out changeset 47d387b6cd4a (bug 1479960)
Backed out changeset 8332565a6943 (bug 1479960)
Backed out changeset 9d7f1835f96f (bug 1479960)
Backed out changeset 0aa8af4965c5 (bug 1479960)
Backed out changeset 036809330a51 (bug 1479960)
Backed out changeset 39e18373e3d3 (bug 1479960)
Backed out changeset 6c2b995a9d30 (bug 1479960)
Backed out changeset 3c2b31744645 (bug 1534780)
Backed out changeset 26bb00a94d5d (bug 1534780)
115 lines
3.0 KiB
C++
115 lines
3.0 KiB
C++
/* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
|
|
/* vim: set ts=8 sts=2 et sw=2 tw=80: */
|
|
/* This Source Code Form is subject to the terms of the Mozilla Public
|
|
* License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
|
|
|
|
#include "MemMapSnapshot.h"
|
|
|
|
#include "base/eintr_wrapper.h"
|
|
#include "base/file_util.h"
|
|
#include "mozilla/ResultExtensions.h"
|
|
#include "mozilla/ScopeExit.h"
|
|
#include "mozilla/ipc/FileDescriptorUtils.h"
|
|
#include "nsIFile.h"
|
|
|
|
#ifdef XP_UNIX
|
|
# include <sys/stat.h>
|
|
#endif
|
|
|
|
namespace mozilla {
|
|
|
|
using loader::AutoMemMap;
|
|
|
|
namespace ipc {
|
|
|
|
Result<Ok, nsresult> MemMapSnapshot::Init(size_t aSize) {
|
|
MOZ_ASSERT(!mInitialized);
|
|
|
|
MOZ_TRY(Create(aSize));
|
|
|
|
mInitialized = true;
|
|
return Ok();
|
|
}
|
|
|
|
Result<Ok, nsresult> MemMapSnapshot::Finalize(AutoMemMap& aMem) {
|
|
MOZ_ASSERT(mInitialized);
|
|
|
|
MOZ_TRY(Freeze(aMem));
|
|
|
|
mInitialized = false;
|
|
return Ok();
|
|
}
|
|
|
|
#if defined(XP_WIN)
|
|
|
|
Result<Ok, nsresult> MemMapSnapshot::Create(size_t aSize) {
|
|
HANDLE handle = CreateFileMapping(INVALID_HANDLE_VALUE, nullptr,
|
|
PAGE_READWRITE, 0, DWORD(aSize), nullptr);
|
|
|
|
if (!handle) {
|
|
return Err(NS_ERROR_FAILURE);
|
|
}
|
|
|
|
mFile.emplace(handle);
|
|
return mMem.initWithHandle(mFile.ref(), aSize, PR_PROT_READWRITE);
|
|
}
|
|
|
|
Result<Ok, nsresult> MemMapSnapshot::Freeze(AutoMemMap& aMem) {
|
|
auto orig = mFile.ref().ClonePlatformHandle();
|
|
mFile.reset();
|
|
|
|
HANDLE handle;
|
|
if (!::DuplicateHandle(
|
|
GetCurrentProcess(), orig.release(), GetCurrentProcess(), &handle,
|
|
GENERIC_READ | FILE_MAP_READ, false, DUPLICATE_CLOSE_SOURCE)) {
|
|
return Err(NS_ERROR_FAILURE);
|
|
}
|
|
|
|
return aMem.initWithHandle(FileDescriptor(handle), mMem.size());
|
|
}
|
|
|
|
#elif defined(XP_UNIX)
|
|
|
|
Result<Ok, nsresult> MemMapSnapshot::Create(size_t aSize) {
|
|
FilePath path;
|
|
ScopedCloseFile fd(file_util::CreateAndOpenTemporaryShmemFile(&path));
|
|
if (!fd) {
|
|
return Err(NS_ERROR_FAILURE);
|
|
}
|
|
|
|
if (HANDLE_EINTR(ftruncate(fileno(fd), aSize)) != 0) {
|
|
return Err(NS_ERROR_FAILURE);
|
|
}
|
|
|
|
MOZ_TRY(mMem.init(FILEToFileDescriptor(fd), PR_PROT_READWRITE));
|
|
|
|
mPath.Assign(path.value().data(), path.value().length());
|
|
return Ok();
|
|
}
|
|
|
|
Result<Ok, nsresult> MemMapSnapshot::Freeze(AutoMemMap& aMem) {
|
|
// Delete the shm file after we're done here, whether we succeed or not. The
|
|
// open file descriptor will keep it alive until all remaining references
|
|
// are closed, at which point it will be automatically freed.
|
|
auto cleanup = MakeScopeExit([&]() { PR_Delete(mPath.get()); });
|
|
|
|
// Make the shm file readonly. This doesn't make a difference in practice,
|
|
// since we open and share a read-only file descriptor, and then delete the
|
|
// file. But it doesn't hurt, either.
|
|
chmod(mPath.get(), 0400);
|
|
|
|
nsCOMPtr<nsIFile> file;
|
|
MOZ_TRY(NS_NewNativeLocalFile(mPath, /* followLinks = */ false,
|
|
getter_AddRefs(file)));
|
|
|
|
return aMem.init(file);
|
|
}
|
|
|
|
#else
|
|
# error "Unsupported build configuration"
|
|
#endif
|
|
|
|
} // namespace ipc
|
|
} // namespace mozilla
|