blob: 8589779995d4091cd1b206d23d559329106ceb21 [file] [log] [blame]
// Copyright (c) 2012 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#ifndef BASE_SHARED_MEMORY_H_
#define BASE_SHARED_MEMORY_H_
#include "build/build_config.h"
#include <string>
#if defined(OS_POSIX) && !defined(__LB_SHELL__)
#include <stdio.h>
#include <sys/types.h>
#include <semaphore.h>
#endif
#include "base/base_export.h"
#include "base/basictypes.h"
#include "base/process.h"
#if defined(__LB_SHELL__)
#include <stdlib.h>
#elif defined(OS_STARBOARD)
#include "starboard/memory.h"
#endif
#if defined(__LB_SHELL__) || defined(OS_STARBOARD)
#include "base/memory/ref_counted.h"
#include "base/synchronization/lock.h"
#endif
#if defined(OS_POSIX) && !defined(__LB_SHELL__)
#include "base/file_descriptor_posix.h"
#endif
class FilePath;
namespace base {
// SharedMemoryHandle is a platform specific type which represents
// the underlying OS handle to a shared memory segment.
#if defined(OS_WIN)
typedef HANDLE SharedMemoryHandle;
typedef HANDLE SharedMemoryLock;
#elif defined(OS_POSIX) && !defined(__LB_SHELL__)
// A SharedMemoryId is sufficient to identify a given shared memory segment on a
// system, but insufficient to map it.
typedef FileDescriptor SharedMemoryHandle;
typedef ino_t SharedMemoryId;
// On POSIX, the lock is implemented as a lockf() on the mapped file,
// so no additional member (or definition of SharedMemoryLock) is
// needed.
#elif defined(__LB_SHELL__) || defined(OS_STARBOARD)
// For LB_SHELL, and Starboard, we will only ever have one process, and so we
// use standard heap memory for this.
class RefCountedMem : public base::RefCountedThreadSafe<RefCountedMem> {
public:
RefCountedMem(size_t size) {
size_ = size;
#if defined(OS_STARBOARD)
memory_ = SbMemoryAllocate(size);
#else
memory_ = malloc(size);
#endif
}
~RefCountedMem() {
#if defined(OS_STARBOARD)
SbMemoryDeallocate(memory_);
#else
free(memory_);
#endif
}
void* GetMemory() const { return memory_; }
size_t GetSize() const { return size_; }
Lock& lock() { return lock_; }
private:
size_t size_;
void* memory_;
Lock lock_;
};
typedef scoped_refptr<RefCountedMem> SharedMemoryHandle;
#endif
// Options for creating a shared memory object.
struct SharedMemoryCreateOptions {
SharedMemoryCreateOptions() : name(NULL), size(0), open_existing(false),
executable(false) {}
// If NULL, the object is anonymous. This pointer is owned by the caller
// and must live through the call to Create().
const std::string* name;
// Size of the shared memory object to be created.
// When opening an existing object, this has no effect.
size_t size;
// If true, and the shared memory already exists, Create() will open the
// existing shared memory and ignore the size parameter. If false,
// shared memory must not exist. This flag is meaningless unless name is
// non-NULL.
bool open_existing;
// If true, mappings might need to be made executable later.
bool executable;
};
// Platform abstraction for shared memory. Provides a C++ wrapper
// around the OS primitive for a memory mapped file.
class BASE_EXPORT SharedMemory {
public:
SharedMemory();
#if defined(OS_WIN)
// Similar to the default constructor, except that this allows for
// calling Lock() to acquire the named mutex before either Create or Open
// are called on Windows.
explicit SharedMemory(const std::wstring& name);
#endif
// Create a new SharedMemory object from an existing, open
// shared memory file.
SharedMemory(SharedMemoryHandle handle, bool read_only);
// Create a new SharedMemory object from an existing, open
// shared memory file that was created by a remote process and not shared
// to the current process.
SharedMemory(SharedMemoryHandle handle, bool read_only,
ProcessHandle process);
// Closes any open files.
~SharedMemory();
// Return true iff the given handle is valid (i.e. not the distingished
// invalid value; NULL for a HANDLE and -1 for a file descriptor)
static bool IsHandleValid(const SharedMemoryHandle& handle);
// Returns invalid handle (see comment above for exact definition).
static SharedMemoryHandle NULLHandle();
// Closes a shared memory handle.
static void CloseHandle(const SharedMemoryHandle& handle);
// Creates a shared memory object as described by the options struct.
// Returns true on success and false on failure.
bool Create(const SharedMemoryCreateOptions& options);
// Creates and maps an anonymous shared memory segment of size size.
// Returns true on success and false on failure.
bool CreateAndMapAnonymous(size_t size);
// Creates an anonymous shared memory segment of size size.
// Returns true on success and false on failure.
bool CreateAnonymous(size_t size) {
SharedMemoryCreateOptions options;
options.size = size;
return Create(options);
}
// Creates or opens a shared memory segment based on a name.
// If open_existing is true, and the shared memory already exists,
// opens the existing shared memory and ignores the size parameter.
// If open_existing is false, shared memory must not exist.
// size is the size of the block to be created.
// Returns true on success, false on failure.
bool CreateNamed(const std::string& name, bool open_existing, size_t size) {
SharedMemoryCreateOptions options;
options.name = &name;
options.open_existing = open_existing;
options.size = size;
return Create(options);
}
// Deletes resources associated with a shared memory segment based on name.
// Not all platforms require this call.
bool Delete(const std::string& name);
// Opens a shared memory segment based on a name.
// If read_only is true, opens for read-only access.
// Returns true on success, false on failure.
bool Open(const std::string& name, bool read_only);
// Maps the shared memory into the caller's address space.
// Returns true on success, false otherwise. The memory address
// is accessed via the memory() accessor. The mapped address is guaranteed to
// have an alignment of at least MAP_MINIMUM_ALIGNMENT.
bool Map(size_t bytes);
enum { MAP_MINIMUM_ALIGNMENT = 32 };
// Unmaps the shared memory from the caller's address space.
// Returns true if successful; returns false on error or if the
// memory is not mapped.
bool Unmap();
// Get the size of the shared memory backing file.
// Note: This size is only available to the creator of the
// shared memory, and not to those that opened shared memory
// created externally.
// Returns 0 if not created or unknown.
// Deprecated method, please keep track of the size yourself if you created
// it.
// http://crbug.com/60821
size_t created_size() const { return created_size_; }
// Gets a pointer to the opened memory space if it has been
// Mapped via Map(). Returns NULL if it is not mapped.
void *memory() const { return memory_; }
// Returns the underlying OS handle for this segment.
// Use of this handle for anything other than an opaque
// identifier is not portable.
SharedMemoryHandle handle() const;
#if defined(OS_POSIX) && !defined(OS_NACL) && !defined(__LB_SHELL__)
// Returns a unique identifier for this shared memory segment. Inode numbers
// are technically only unique to a single filesystem. However, we always
// allocate shared memory backing files from the same directory, so will end
// up on the same filesystem.
SharedMemoryId id() const { return inode_; }
#endif
// Closes the open shared memory segment.
// It is safe to call Close repeatedly.
void Close();
// Shares the shared memory to another process. Attempts
// to create a platform-specific new_handle which can be
// used in a remote process to access the shared memory
// file. new_handle is an ouput parameter to receive
// the handle for use in the remote process.
// Returns true on success, false otherwise.
bool ShareToProcess(ProcessHandle process,
SharedMemoryHandle* new_handle) {
return ShareToProcessCommon(process, new_handle, false);
}
// Logically equivalent to:
// bool ok = ShareToProcess(process, new_handle);
// Close();
// return ok;
// Note that the memory is unmapped by calling this method, regardless of the
// return value.
bool GiveToProcess(ProcessHandle process,
SharedMemoryHandle* new_handle) {
return ShareToProcessCommon(process, new_handle, true);
}
// Locks the shared memory.
//
// WARNING: on POSIX the memory locking primitive only works across
// processes, not across threads. The Lock method is not currently
// used in inner loops, so we protect against multiple threads in a
// critical section using a class global lock.
void Lock();
#if defined(OS_WIN)
// A Lock() implementation with a timeout that also allows setting
// security attributes on the mutex. sec_attr may be NULL.
// Returns true if the Lock() has been acquired, false if the timeout was
// reached.
bool Lock(uint32 timeout_ms, SECURITY_ATTRIBUTES* sec_attr);
#endif
// Releases the shared memory lock.
void Unlock();
private:
#if defined(OS_POSIX) && !defined(OS_NACL) && !defined(__LB_SHELL__)
bool PrepareMapFile(FILE *fp);
bool FilePathForMemoryName(const std::string& mem_name, FilePath* path);
void LockOrUnlockCommon(int function);
#endif
bool ShareToProcessCommon(ProcessHandle process,
SharedMemoryHandle* new_handle,
bool close_self);
#if defined(OS_WIN)
std::wstring name_;
HANDLE mapped_file_;
#elif defined(OS_POSIX) && !defined(__LB_SHELL__)
int mapped_file_;
size_t mapped_size_;
ino_t inode_;
#elif defined(__LB_SHELL__)
scoped_refptr<RefCountedMem> handle_;
#endif
void* memory_;
bool read_only_;
size_t created_size_;
#if !defined(OS_POSIX) && !defined(OS_STARBOARD)
SharedMemoryLock lock_;
#endif
DISALLOW_COPY_AND_ASSIGN(SharedMemory);
};
// A helper class that acquires the shared memory lock while
// the SharedMemoryAutoLock is in scope.
class SharedMemoryAutoLock {
public:
explicit SharedMemoryAutoLock(SharedMemory* shared_memory)
: shared_memory_(shared_memory) {
shared_memory_->Lock();
}
~SharedMemoryAutoLock() {
shared_memory_->Unlock();
}
private:
SharedMemory* shared_memory_;
DISALLOW_COPY_AND_ASSIGN(SharedMemoryAutoLock);
};
} // namespace base
#endif // BASE_SHARED_MEMORY_H_