mirror of
https://github.com/googleprojectzero/Jackalope
synced 2026-06-08 14:27:44 +00:00
Ivan Fratric
131 lines
2.9 KiB
C++
131 lines
2.9 KiB
C++
/*
|
|
Copyright 2020 Google LLC
|
|
|
|
Licensed under the Apache License, Version 2.0 (the "License");
|
|
you may not use this file except in compliance with the License.
|
|
You may obtain a copy of the License at
|
|
|
|
https://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
Unless required by applicable law or agreed to in writing, software
|
|
distributed under the License is distributed on an "AS IS" BASIS,
|
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
See the License for the specific language governing permissions and
|
|
limitations under the License.
|
|
*/
|
|
|
|
#include "common.h"
|
|
#include "shm.h"
|
|
|
|
SharedMemory::SharedMemory() {
|
|
size = 0;
|
|
shm = NULL;
|
|
}
|
|
|
|
SharedMemory::SharedMemory(char* name, size_t size) {
|
|
Open(name, size);
|
|
}
|
|
|
|
SharedMemory::~SharedMemory() {
|
|
Close();
|
|
}
|
|
|
|
|
|
#if defined(WIN32) || defined(_WIN32) || defined(__WIN32)
|
|
|
|
void SharedMemory::Open(char* name, size_t size) {
|
|
shm_handle = CreateFileMapping(
|
|
INVALID_HANDLE_VALUE,
|
|
NULL,
|
|
PAGE_READWRITE,
|
|
0,
|
|
(DWORD)size,
|
|
name);
|
|
|
|
if (shm_handle == NULL) {
|
|
FATAL("CreateFileMapping failed, %x", GetLastError());
|
|
}
|
|
|
|
shm = (unsigned char*)MapViewOfFile(
|
|
shm_handle, // handle to map object
|
|
FILE_MAP_ALL_ACCESS, // read/write permission
|
|
0,
|
|
0,
|
|
size
|
|
);
|
|
|
|
if (!shm) {
|
|
FATAL("MapViewOfFile failed");
|
|
}
|
|
}
|
|
|
|
void SharedMemory::Close() {
|
|
UnmapViewOfFile(shm);
|
|
CloseHandle(shm_handle);
|
|
}
|
|
|
|
#else
|
|
|
|
#include <unistd.h>
|
|
#include <sys/mman.h>
|
|
#include <fcntl.h>
|
|
|
|
void SharedMemory::Open(char* name, size_t size) {
|
|
#if defined(__ANDROID__) && !defined(ANDROID_VM)
|
|
FATAL("Shared memory is not implemented on Android");
|
|
#else
|
|
int res;
|
|
|
|
this->size = size;
|
|
|
|
// get shared memory file descriptor (NOT a file)
|
|
#ifdef ANDROID_VM
|
|
size_t name_size = strlen(name) + strlen("/dev/shm");
|
|
this->name = (char*)malloc(name_size + 1);
|
|
strcpy(this->name, "/dev/shm");
|
|
strcat(this->name, name);
|
|
fd = open(this->name, O_RDWR | O_CREAT, S_IRUSR | S_IWUSR);
|
|
#else
|
|
size_t name_size = strlen(name);
|
|
this->name = (char*)malloc(name_size + 1);
|
|
strcpy(this->name, name);
|
|
fd = shm_open(this->name, O_RDWR | O_CREAT, S_IRUSR | S_IWUSR);
|
|
#endif
|
|
if (fd == -1)
|
|
{
|
|
FATAL("Error creating shared memory");
|
|
}
|
|
|
|
// extend shared memory object as by default it's initialized with size 0
|
|
res = ftruncate(fd, size);
|
|
if (res == -1)
|
|
{
|
|
FATAL("Error creating shared memory");
|
|
}
|
|
|
|
// map shared memory to process address space
|
|
shm = (unsigned char*)mmap(NULL, size, PROT_WRITE, MAP_SHARED, fd, 0);
|
|
if (shm == MAP_FAILED)
|
|
{
|
|
FATAL("Error creating shared memory");
|
|
}
|
|
#endif
|
|
}
|
|
|
|
void SharedMemory::Close() {
|
|
#if defined(__ANDROID__) && !defined(ANDROID_VM)
|
|
FATAL("Shared memory is not implemented on Android");
|
|
#else
|
|
munmap(shm, size);
|
|
#ifdef ANDROID_VM
|
|
unlink(name);
|
|
#else
|
|
shm_unlink(name);
|
|
#endif
|
|
close(fd);
|
|
free(name);
|
|
#endif
|
|
}
|
|
|
|
#endif
|