mirror of
https://github.com/slsdetectorgroup/slsDetectorPackage.git
synced 2025-04-20 02:40:03 +02:00
Refactoring of SharedMemory.h (#418)
Function names match Detector.h Removed double print due to LOG then throw file descriptor not kept as a member variable
This commit is contained in:
parent
66900da476
commit
e55e18d5e9
@ -26,8 +26,8 @@ void freeSharedMemory(int detectorIndex, int moduleIndex) {
|
|||||||
// single module
|
// single module
|
||||||
if (moduleIndex >= 0) {
|
if (moduleIndex >= 0) {
|
||||||
SharedMemory<sharedModule> moduleShm(detectorIndex, moduleIndex);
|
SharedMemory<sharedModule> moduleShm(detectorIndex, moduleIndex);
|
||||||
if (moduleShm.IsExisting()) {
|
if (moduleShm.exists()) {
|
||||||
moduleShm.RemoveSharedMemory();
|
moduleShm.removeSharedMemory();
|
||||||
}
|
}
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
@ -36,21 +36,21 @@ void freeSharedMemory(int detectorIndex, int moduleIndex) {
|
|||||||
SharedMemory<sharedDetector> detectorShm(detectorIndex, -1);
|
SharedMemory<sharedDetector> detectorShm(detectorIndex, -1);
|
||||||
int numDetectors = 0;
|
int numDetectors = 0;
|
||||||
|
|
||||||
if (detectorShm.IsExisting()) {
|
if (detectorShm.exists()) {
|
||||||
detectorShm.OpenSharedMemory();
|
detectorShm.openSharedMemory();
|
||||||
numDetectors = detectorShm()->numberOfModules;
|
numDetectors = detectorShm()->numberOfModules;
|
||||||
detectorShm.RemoveSharedMemory();
|
detectorShm.removeSharedMemory();
|
||||||
}
|
}
|
||||||
|
|
||||||
for (int i = 0; i < numDetectors; ++i) {
|
for (int i = 0; i < numDetectors; ++i) {
|
||||||
SharedMemory<sharedModule> moduleShm(detectorIndex, i);
|
SharedMemory<sharedModule> moduleShm(detectorIndex, i);
|
||||||
moduleShm.RemoveSharedMemory();
|
moduleShm.removeSharedMemory();
|
||||||
}
|
}
|
||||||
|
|
||||||
// Ctb configuration
|
// Ctb configuration
|
||||||
SharedMemory<CtbConfig> ctbShm(detectorIndex, -1, CtbConfig::shm_tag());
|
SharedMemory<CtbConfig> ctbShm(detectorIndex, -1, CtbConfig::shm_tag());
|
||||||
if (ctbShm.IsExisting())
|
if (ctbShm.exists())
|
||||||
ctbShm.RemoveSharedMemory();
|
ctbShm.removeSharedMemory();
|
||||||
}
|
}
|
||||||
|
|
||||||
using defs = slsDetectorDefs;
|
using defs = slsDetectorDefs;
|
||||||
|
@ -46,8 +46,8 @@ void DetectorImpl::setupDetector(bool verify, bool update) {
|
|||||||
updateUserdetails();
|
updateUserdetails();
|
||||||
}
|
}
|
||||||
|
|
||||||
if (ctb_shm.IsExisting())
|
if (ctb_shm.exists())
|
||||||
ctb_shm.OpenSharedMemory();
|
ctb_shm.openSharedMemory();
|
||||||
}
|
}
|
||||||
|
|
||||||
void DetectorImpl::setAcquiringFlag(bool flag) { shm()->acquiringFlag = flag; }
|
void DetectorImpl::setAcquiringFlag(bool flag) { shm()->acquiringFlag = flag; }
|
||||||
@ -58,8 +58,8 @@ void DetectorImpl::freeSharedMemory(int detectorIndex, int detPos) {
|
|||||||
// single
|
// single
|
||||||
if (detPos >= 0) {
|
if (detPos >= 0) {
|
||||||
SharedMemory<sharedModule> moduleShm(detectorIndex, detPos);
|
SharedMemory<sharedModule> moduleShm(detectorIndex, detPos);
|
||||||
if (moduleShm.IsExisting()) {
|
if (moduleShm.exists()) {
|
||||||
moduleShm.RemoveSharedMemory();
|
moduleShm.removeSharedMemory();
|
||||||
}
|
}
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
@ -68,20 +68,20 @@ void DetectorImpl::freeSharedMemory(int detectorIndex, int detPos) {
|
|||||||
SharedMemory<sharedDetector> detectorShm(detectorIndex, -1);
|
SharedMemory<sharedDetector> detectorShm(detectorIndex, -1);
|
||||||
int numModules = 0;
|
int numModules = 0;
|
||||||
|
|
||||||
if (detectorShm.IsExisting()) {
|
if (detectorShm.exists()) {
|
||||||
detectorShm.OpenSharedMemory();
|
detectorShm.openSharedMemory();
|
||||||
numModules = detectorShm()->numberOfModules;
|
numModules = detectorShm()->numberOfModules;
|
||||||
detectorShm.RemoveSharedMemory();
|
detectorShm.removeSharedMemory();
|
||||||
}
|
}
|
||||||
|
|
||||||
for (int i = 0; i < numModules; ++i) {
|
for (int i = 0; i < numModules; ++i) {
|
||||||
SharedMemory<sharedModule> moduleShm(detectorIndex, i);
|
SharedMemory<sharedModule> moduleShm(detectorIndex, i);
|
||||||
moduleShm.RemoveSharedMemory();
|
moduleShm.removeSharedMemory();
|
||||||
}
|
}
|
||||||
|
|
||||||
SharedMemory<CtbConfig> ctbShm(detectorIndex, -1, CtbConfig::shm_tag());
|
SharedMemory<CtbConfig> ctbShm(detectorIndex, -1, CtbConfig::shm_tag());
|
||||||
if (ctbShm.IsExisting())
|
if (ctbShm.exists())
|
||||||
ctbShm.RemoveSharedMemory();
|
ctbShm.removeSharedMemory();
|
||||||
}
|
}
|
||||||
|
|
||||||
void DetectorImpl::freeSharedMemory() {
|
void DetectorImpl::freeSharedMemory() {
|
||||||
@ -92,11 +92,11 @@ void DetectorImpl::freeSharedMemory() {
|
|||||||
modules.clear();
|
modules.clear();
|
||||||
|
|
||||||
// clear detector shm
|
// clear detector shm
|
||||||
shm.RemoveSharedMemory();
|
shm.removeSharedMemory();
|
||||||
client_downstream = false;
|
client_downstream = false;
|
||||||
|
|
||||||
if (ctb_shm.IsExisting())
|
if (ctb_shm.exists())
|
||||||
ctb_shm.RemoveSharedMemory();
|
ctb_shm.removeSharedMemory();
|
||||||
}
|
}
|
||||||
|
|
||||||
std::string DetectorImpl::getUserDetails() {
|
std::string DetectorImpl::getUserDetails() {
|
||||||
@ -140,11 +140,11 @@ void DetectorImpl::setInitialChecks(const bool value) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
void DetectorImpl::initSharedMemory(bool verify) {
|
void DetectorImpl::initSharedMemory(bool verify) {
|
||||||
if (!shm.IsExisting()) {
|
if (!shm.exists()) {
|
||||||
shm.CreateSharedMemory();
|
shm.createSharedMemory();
|
||||||
initializeDetectorStructure();
|
initializeDetectorStructure();
|
||||||
} else {
|
} else {
|
||||||
shm.OpenSharedMemory();
|
shm.openSharedMemory();
|
||||||
if (verify && shm()->shmversion != DETECTOR_SHMVERSION) {
|
if (verify && shm()->shmversion != DETECTOR_SHMVERSION) {
|
||||||
LOG(logERROR) << "Detector shared memory (" << detectorIndex
|
LOG(logERROR) << "Detector shared memory (" << detectorIndex
|
||||||
<< ") version mismatch "
|
<< ") version mismatch "
|
||||||
@ -263,10 +263,10 @@ void DetectorImpl::setHostname(const std::vector<std::string> &name) {
|
|||||||
// if needed, CTB dac names are only on detector level
|
// if needed, CTB dac names are only on detector level
|
||||||
|
|
||||||
if (shm()->detType == defs::CHIPTESTBOARD) {
|
if (shm()->detType == defs::CHIPTESTBOARD) {
|
||||||
if (ctb_shm.IsExisting())
|
if (ctb_shm.exists())
|
||||||
ctb_shm.OpenSharedMemory();
|
ctb_shm.openSharedMemory();
|
||||||
else
|
else
|
||||||
ctb_shm.CreateSharedMemory();
|
ctb_shm.createSharedMemory();
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -33,11 +33,11 @@ Module::Module(detectorType type, int det_id, int module_index, bool verify)
|
|||||||
: moduleIndex(module_index), shm(det_id, module_index) {
|
: moduleIndex(module_index), shm(det_id, module_index) {
|
||||||
|
|
||||||
// ensure shared memory was not created before
|
// ensure shared memory was not created before
|
||||||
if (shm.IsExisting()) {
|
if (shm.exists()) {
|
||||||
LOG(logWARNING) << "This shared memory should have been "
|
LOG(logWARNING) << "This shared memory should have been "
|
||||||
"deleted before! "
|
"deleted before! "
|
||||||
<< shm.GetName() << ". Freeing it again";
|
<< shm.getName() << ". Freeing it again";
|
||||||
shm.RemoveSharedMemory();
|
shm.removeSharedMemory();
|
||||||
}
|
}
|
||||||
|
|
||||||
initSharedMemory(type, det_id, verify);
|
initSharedMemory(type, det_id, verify);
|
||||||
@ -55,8 +55,8 @@ Module::Module(int det_id, int module_index, bool verify)
|
|||||||
Module::~Module() = default;
|
Module::~Module() = default;
|
||||||
|
|
||||||
void Module::freeSharedMemory() {
|
void Module::freeSharedMemory() {
|
||||||
if (shm.IsExisting()) {
|
if (shm.exists()) {
|
||||||
shm.RemoveSharedMemory();
|
shm.removeSharedMemory();
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -3171,20 +3171,20 @@ Ret Module::sendToReceiver(int fnum, const Arg &args) {
|
|||||||
|
|
||||||
slsDetectorDefs::detectorType Module::getDetectorTypeFromShm(int det_id,
|
slsDetectorDefs::detectorType Module::getDetectorTypeFromShm(int det_id,
|
||||||
bool verify) {
|
bool verify) {
|
||||||
if (!shm.IsExisting()) {
|
if (!shm.exists()) {
|
||||||
throw SharedMemoryError("Shared memory " + shm.GetName() +
|
throw SharedMemoryError("Shared memory " + shm.getName() +
|
||||||
"does not exist.\n Corrupted Multi Shared "
|
"does not exist.\n Corrupted Multi Shared "
|
||||||
"memory. Please free shared memory.");
|
"memory. Please free shared memory.");
|
||||||
}
|
}
|
||||||
|
|
||||||
shm.OpenSharedMemory();
|
shm.openSharedMemory();
|
||||||
if (verify && shm()->shmversion != MODULE_SHMVERSION) {
|
if (verify && shm()->shmversion != MODULE_SHMVERSION) {
|
||||||
std::ostringstream ss;
|
std::ostringstream ss;
|
||||||
ss << "Single shared memory (" << det_id << "-" << moduleIndex
|
ss << "Single shared memory (" << det_id << "-" << moduleIndex
|
||||||
<< ":)version mismatch (expected 0x" << std::hex << MODULE_SHMVERSION
|
<< ":)version mismatch (expected 0x" << std::hex << MODULE_SHMVERSION
|
||||||
<< " but got 0x" << shm()->shmversion << ")" << std::dec
|
<< " but got 0x" << shm()->shmversion << ")" << std::dec
|
||||||
<< ". Clear Shared memory to continue.";
|
<< ". Clear Shared memory to continue.";
|
||||||
shm.UnmapSharedMemory();
|
shm.unmapSharedMemory();
|
||||||
throw SharedMemoryError(ss.str());
|
throw SharedMemoryError(ss.str());
|
||||||
}
|
}
|
||||||
return shm()->detType;
|
return shm()->detType;
|
||||||
@ -3192,11 +3192,11 @@ slsDetectorDefs::detectorType Module::getDetectorTypeFromShm(int det_id,
|
|||||||
|
|
||||||
void Module::initSharedMemory(detectorType type, int det_id, bool verify) {
|
void Module::initSharedMemory(detectorType type, int det_id, bool verify) {
|
||||||
shm = SharedMemory<sharedModule>(det_id, moduleIndex);
|
shm = SharedMemory<sharedModule>(det_id, moduleIndex);
|
||||||
if (!shm.IsExisting()) {
|
if (!shm.exists()) {
|
||||||
shm.CreateSharedMemory();
|
shm.createSharedMemory();
|
||||||
initializeModuleStructure(type);
|
initializeModuleStructure(type);
|
||||||
} else {
|
} else {
|
||||||
shm.OpenSharedMemory();
|
shm.openSharedMemory();
|
||||||
if (verify && shm()->shmversion != MODULE_SHMVERSION) {
|
if (verify && shm()->shmversion != MODULE_SHMVERSION) {
|
||||||
std::ostringstream ss;
|
std::ostringstream ss;
|
||||||
ss << "Single shared memory (" << det_id << "-" << moduleIndex
|
ss << "Single shared memory (" << det_id << "-" << moduleIndex
|
||||||
|
@ -13,7 +13,8 @@
|
|||||||
#include "sls/logger.h"
|
#include "sls/logger.h"
|
||||||
#include "sls/sls_detector_exceptions.h"
|
#include "sls/sls_detector_exceptions.h"
|
||||||
|
|
||||||
#include "stdlib.h"
|
// #include "stdlib.h"
|
||||||
|
#include <cstdlib>
|
||||||
#include <cerrno> // errno
|
#include <cerrno> // errno
|
||||||
#include <cstring> // strerror
|
#include <cstring> // strerror
|
||||||
#include <fcntl.h> // O_CREAT, O_TRUNC..
|
#include <fcntl.h> // O_CREAT, O_TRUNC..
|
||||||
@ -33,268 +34,178 @@
|
|||||||
namespace sls {
|
namespace sls {
|
||||||
|
|
||||||
template <typename T> class SharedMemory {
|
template <typename T> class SharedMemory {
|
||||||
|
static constexpr int NAME_MAX_LENGTH = 255;
|
||||||
|
std::string name;
|
||||||
|
T *shared_struct{};
|
||||||
|
|
||||||
public:
|
public:
|
||||||
//moduleid of -1 creates a detector only shared memory
|
// moduleid of -1 creates a detector only shared memory
|
||||||
SharedMemory(int detectorId, int moduleIndex, const std::string& tag = "") {
|
SharedMemory(int detectorId, int moduleIndex, const std::string &tag = "") {
|
||||||
name = ConstructSharedMemoryName(detectorId, moduleIndex, tag);
|
name = constructSharedMemoryName(detectorId, moduleIndex, tag);
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
// Disable copy, since we refer to a unique location
|
||||||
* Delete the copy constructor and copy assignment since we don't want two
|
|
||||||
* objects managing the same resource
|
|
||||||
*/
|
|
||||||
SharedMemory(const SharedMemory &) = delete;
|
SharedMemory(const SharedMemory &) = delete;
|
||||||
SharedMemory &operator=(const SharedMemory &other) = delete;
|
SharedMemory &operator=(const SharedMemory &other) = delete;
|
||||||
|
|
||||||
// Move constructor
|
|
||||||
SharedMemory(SharedMemory &&other)
|
SharedMemory(SharedMemory &&other)
|
||||||
: name(other.name), fd(other.fd), shmSize(other.shmSize),
|
: name(other.name), shared_struct(other.shared_struct) {
|
||||||
shared_struct(other.shared_struct) {
|
|
||||||
|
|
||||||
other.fd = -1;
|
|
||||||
other.shared_struct = nullptr;
|
other.shared_struct = nullptr;
|
||||||
other.shmSize = 0;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// Move assignment
|
|
||||||
SharedMemory &operator=(SharedMemory &&other) {
|
SharedMemory &operator=(SharedMemory &&other) {
|
||||||
name = other.name;
|
name = other.name;
|
||||||
if (fd) {
|
if (shared_struct != nullptr)
|
||||||
close(fd);
|
unmapSharedMemory();
|
||||||
}
|
|
||||||
fd = other.fd;
|
|
||||||
other.fd = -1;
|
|
||||||
|
|
||||||
if (shared_struct != nullptr) {
|
|
||||||
UnmapSharedMemory();
|
|
||||||
}
|
|
||||||
shared_struct = other.shared_struct;
|
shared_struct = other.shared_struct;
|
||||||
other.shared_struct = nullptr;
|
other.shared_struct = nullptr;
|
||||||
|
|
||||||
shmSize = other.shmSize;
|
|
||||||
other.shmSize = 0;
|
|
||||||
return *this;
|
return *this;
|
||||||
}
|
}
|
||||||
|
|
||||||
~SharedMemory() {
|
~SharedMemory() {
|
||||||
if (fd >= 0)
|
if (shared_struct)
|
||||||
close(fd);
|
unmapSharedMemory();
|
||||||
|
|
||||||
if (shared_struct) {
|
|
||||||
UnmapSharedMemory();
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
T *operator()() { return shared_struct; }
|
||||||
* Verify if it exists
|
const T *operator()() const { return shared_struct; }
|
||||||
* @return true if exists, else false
|
std::string getName() const { return name; }
|
||||||
*/
|
|
||||||
bool IsExisting() {
|
bool exists() {
|
||||||
bool ret = true;
|
|
||||||
int tempfd = shm_open(name.c_str(), O_RDWR, 0);
|
int tempfd = shm_open(name.c_str(), O_RDWR, 0);
|
||||||
if ((tempfd < 0) && (errno == ENOENT)) {
|
if ((tempfd < 0) && (errno == ENOENT)) {
|
||||||
ret = false;
|
return false;
|
||||||
}
|
}
|
||||||
close(tempfd);
|
close(tempfd);
|
||||||
return ret;
|
return true;
|
||||||
}
|
}
|
||||||
|
|
||||||
std::string GetName() const { return name; }
|
void createSharedMemory() {
|
||||||
size_t size() const { return shmSize; }
|
int fd = shm_open(name.c_str(), O_CREAT | O_TRUNC | O_EXCL | O_RDWR,
|
||||||
|
S_IRUSR | S_IWUSR);
|
||||||
/**
|
|
||||||
* Create Shared memory and call MapSharedMemory to map it to an address
|
|
||||||
* throws a SharedMemoryError exception on failure to create, ftruncate or
|
|
||||||
* map
|
|
||||||
*/
|
|
||||||
void CreateSharedMemory() {
|
|
||||||
fd = shm_open(name.c_str(), O_CREAT | O_TRUNC | O_EXCL | O_RDWR,
|
|
||||||
S_IRUSR | S_IWUSR);
|
|
||||||
if (fd < 0) {
|
if (fd < 0) {
|
||||||
std::string msg =
|
std::string msg =
|
||||||
"Create shared memory " + name + " failed: " + strerror(errno);
|
"Create shared memory " + name + " failed: " + strerror(errno);
|
||||||
LOG(logERROR) << msg;
|
|
||||||
throw SharedMemoryError(msg);
|
throw SharedMemoryError(msg);
|
||||||
}
|
}
|
||||||
|
|
||||||
if (ftruncate(fd, sizeof(T)) < 0) {
|
if (ftruncate(fd, sizeof(T)) < 0) {
|
||||||
std::string msg = "Create shared memory " + name +
|
std::string msg = "Create shared memory " + name +
|
||||||
" failed at ftruncate: " + strerror(errno);
|
" failed at ftruncate: " + strerror(errno);
|
||||||
LOG(logERROR) << msg;
|
|
||||||
close(fd);
|
close(fd);
|
||||||
RemoveSharedMemory();
|
removeSharedMemory();
|
||||||
throw SharedMemoryError(msg);
|
throw SharedMemoryError(msg);
|
||||||
}
|
}
|
||||||
shared_struct = MapSharedMemory();
|
shared_struct = mapSharedMemory(fd);
|
||||||
new (shared_struct) T{};
|
new (shared_struct) T{};
|
||||||
LOG(logINFO) << "Shared memory created " << name;
|
LOG(logINFO) << "Shared memory created " << name;
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
void openSharedMemory() {
|
||||||
* Open existing Shared memory and call MapSharedMemory to map it to an
|
int fd = shm_open(name.c_str(), O_RDWR, 0);
|
||||||
* address throws a SharedMemoryError exception on failure to open or map
|
|
||||||
*/
|
|
||||||
void OpenSharedMemory() {
|
|
||||||
fd = shm_open(name.c_str(), O_RDWR, 0);
|
|
||||||
if (fd < 0) {
|
if (fd < 0) {
|
||||||
std::string msg = "Open existing shared memory " + name +
|
std::string msg = "Open existing shared memory " + name +
|
||||||
" failed: " + strerror(errno);
|
" failed: " + strerror(errno);
|
||||||
LOG(logERROR) << msg;
|
|
||||||
throw SharedMemoryError(msg);
|
throw SharedMemoryError(msg);
|
||||||
}
|
}
|
||||||
|
checkSize(fd);
|
||||||
shared_struct = MapSharedMemory();
|
shared_struct = mapSharedMemory(fd);
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
void unmapSharedMemory() {
|
||||||
* Unmap shared memory from an address
|
|
||||||
* throws a SharedMemoryError exception on failure
|
|
||||||
*/
|
|
||||||
void UnmapSharedMemory() {
|
|
||||||
if (shared_struct != nullptr) {
|
if (shared_struct != nullptr) {
|
||||||
if (munmap(shared_struct, shmSize) < 0) {
|
if (munmap(shared_struct, sizeof(T)) < 0) {
|
||||||
std::string msg = "Unmapping shared memory " + name +
|
std::string msg = "Unmapping shared memory " + name +
|
||||||
" failed: " + strerror(errno);
|
" failed: " + strerror(errno);
|
||||||
LOG(logERROR) << msg;
|
|
||||||
close(fd);
|
|
||||||
throw SharedMemoryError(msg);
|
throw SharedMemoryError(msg);
|
||||||
}
|
}
|
||||||
shared_struct = nullptr;
|
shared_struct = nullptr;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
void removeSharedMemory() {
|
||||||
* Remove existing Shared memory
|
unmapSharedMemory();
|
||||||
*/
|
|
||||||
void RemoveSharedMemory() {
|
|
||||||
UnmapSharedMemory();
|
|
||||||
if (shm_unlink(name.c_str()) < 0) {
|
if (shm_unlink(name.c_str()) < 0) {
|
||||||
// silent exit if shm did not exist anyway
|
// silent exit if shm did not exist anyway
|
||||||
if (errno == ENOENT)
|
if (errno == ENOENT)
|
||||||
return;
|
return;
|
||||||
std::string msg =
|
std::string msg =
|
||||||
"Free Shared Memory " + name + " Failed: " + strerror(errno);
|
"Free Shared Memory " + name + " Failed: " + strerror(errno);
|
||||||
LOG(logERROR) << msg;
|
|
||||||
throw SharedMemoryError(msg);
|
throw SharedMemoryError(msg);
|
||||||
}
|
}
|
||||||
LOG(logINFO) << "Shared memory deleted " << name;
|
LOG(logINFO) << "Shared memory deleted " << name;
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
|
||||||
* Maximum length of name as from man pages
|
|
||||||
*/
|
|
||||||
static const int NAME_MAX_LENGTH = 255;
|
|
||||||
|
|
||||||
/**
|
|
||||||
*Using the call operator to access the pointer
|
|
||||||
*/
|
|
||||||
T *operator()() { return shared_struct; }
|
|
||||||
|
|
||||||
/**
|
|
||||||
*Using the call operator to access the pointer, const overload
|
|
||||||
*/
|
|
||||||
const T *operator()() const { return shared_struct; }
|
|
||||||
|
|
||||||
private:
|
private:
|
||||||
/**
|
std::string constructSharedMemoryName(int detectorId, int moduleIndex,
|
||||||
* Create Shared memory name
|
const std::string &tag) {
|
||||||
* throws exception if name created is longer than required 255(manpages)
|
|
||||||
* @param detectorId detector id
|
|
||||||
* @param moduleIndex module id, -1 if a detector shared memory
|
|
||||||
* @returns shared memory name
|
|
||||||
*/
|
|
||||||
std::string ConstructSharedMemoryName(int detectorId, int moduleIndex, const std::string& tag) {
|
|
||||||
|
|
||||||
// using environment path
|
// using environment variable
|
||||||
std::string sEnvPath;
|
std::string slsdetname;
|
||||||
char *envpath = getenv(SHM_ENV_NAME);
|
char *envpath = getenv(SHM_ENV_NAME);
|
||||||
if (envpath != nullptr) {
|
if (envpath != nullptr) {
|
||||||
sEnvPath.assign(envpath);
|
slsdetname = envpath;
|
||||||
sEnvPath.insert(0, "_");
|
slsdetname.insert(0, "_");
|
||||||
}
|
}
|
||||||
|
|
||||||
std::stringstream ss;
|
std::stringstream ss;
|
||||||
if (moduleIndex < 0){
|
if (moduleIndex < 0) {
|
||||||
ss << SHM_DETECTOR_PREFIX << detectorId << sEnvPath;
|
ss << SHM_DETECTOR_PREFIX << detectorId << slsdetname;
|
||||||
if (!tag.empty())
|
if (!tag.empty())
|
||||||
ss << "_" << tag;
|
ss << "_" << tag;
|
||||||
}
|
} else {
|
||||||
|
|
||||||
else
|
|
||||||
ss << SHM_DETECTOR_PREFIX << detectorId << SHM_MODULE_PREFIX
|
ss << SHM_DETECTOR_PREFIX << detectorId << SHM_MODULE_PREFIX
|
||||||
<< moduleIndex << sEnvPath;
|
<< moduleIndex << slsdetname;
|
||||||
|
}
|
||||||
|
|
||||||
std::string temp = ss.str();
|
std::string shm_name = ss.str();
|
||||||
if (temp.length() > NAME_MAX_LENGTH) {
|
if (shm_name.length() > NAME_MAX_LENGTH) {
|
||||||
std::string msg =
|
std::string msg =
|
||||||
"Shared memory initialization failed. " + temp + " has " +
|
"Shared memory initialization failed. " + shm_name + " has " +
|
||||||
std::to_string(temp.length()) + " characters. \n" +
|
std::to_string(shm_name.length()) + " characters. \n" +
|
||||||
"Maximum is " + std::to_string(NAME_MAX_LENGTH) +
|
"Maximum is " + std::to_string(NAME_MAX_LENGTH) +
|
||||||
". Change the environment variable " + SHM_ENV_NAME;
|
". Change the environment variable " + SHM_ENV_NAME;
|
||||||
LOG(logERROR) << msg;
|
|
||||||
throw SharedMemoryError(msg);
|
throw SharedMemoryError(msg);
|
||||||
}
|
}
|
||||||
return temp;
|
return shm_name;
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
// from the Linux manual:
|
||||||
* Map shared memory to an address
|
// After the mmap() call has returned, the file descriptor, fd, can
|
||||||
* throws a SharedMemoryException exception on failure
|
// be closed immediately without invalidating the mapping.
|
||||||
*/
|
T *mapSharedMemory(int fd) {
|
||||||
|
|
||||||
T *MapSharedMemory() {
|
|
||||||
void *addr =
|
void *addr =
|
||||||
mmap(nullptr, sizeof(T), PROT_READ | PROT_WRITE, MAP_SHARED, fd, 0);
|
mmap(nullptr, sizeof(T), PROT_READ | PROT_WRITE, MAP_SHARED, fd, 0);
|
||||||
|
close(fd);
|
||||||
if (addr == MAP_FAILED) {
|
if (addr == MAP_FAILED) {
|
||||||
std::string msg =
|
std::string msg =
|
||||||
"Mapping shared memory " + name + " failed: " + strerror(errno);
|
"Mapping shared memory " + name + " failed: " + strerror(errno);
|
||||||
LOG(logERROR) << msg;
|
|
||||||
close(fd);
|
|
||||||
throw SharedMemoryError(msg);
|
throw SharedMemoryError(msg);
|
||||||
}
|
}
|
||||||
shmSize = sizeof(T);
|
return static_cast<T *>(addr);
|
||||||
close(fd);
|
|
||||||
return (T *)addr;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
void checkSize(int fd) {
|
||||||
* Verify if existing shared memory size matches expected size
|
|
||||||
* @param expectedSize expected size of shared memory, replaced with smaller
|
|
||||||
* size if size does not match
|
|
||||||
* @return 0 for success, 1 for fail
|
|
||||||
*/
|
|
||||||
int VerifySizeMatch(size_t expectedSize) {
|
|
||||||
struct stat sb;
|
struct stat sb;
|
||||||
// could not fstat
|
|
||||||
if (fstat(fd, &sb) < 0) {
|
if (fstat(fd, &sb) < 0) {
|
||||||
std::string msg = "Could not verify existing shared memory " +
|
std::string msg = "Could not verify existing shared memory " +
|
||||||
name + " size match " +
|
name + " size match " +
|
||||||
"(could not fstat): " + strerror(errno);
|
"(could not fstat): " + strerror(errno);
|
||||||
LOG(logERROR) << msg;
|
|
||||||
close(fd);
|
close(fd);
|
||||||
throw SharedMemoryError(msg);
|
throw SharedMemoryError(msg);
|
||||||
}
|
}
|
||||||
|
|
||||||
// size does not match
|
auto actual_size = static_cast<size_t>(sb.st_size);
|
||||||
auto sz = static_cast<size_t>(sb.st_size);
|
auto expected_size = sizeof(T);
|
||||||
if (sz != expectedSize) {
|
if (actual_size != expected_size) {
|
||||||
std::string msg = "Existing shared memory " + name +
|
std::string msg =
|
||||||
" size does not match" + "Expected " +
|
"Existing shared memory " + name + " size does not match. " +
|
||||||
std::to_string(expectedSize) + ", found " +
|
"Expected " + std::to_string(expected_size) + ", found " +
|
||||||
std::to_string(sz);
|
std::to_string(actual_size) +
|
||||||
LOG(logERROR) << msg;
|
". Detector software mismatch? Try freeing shared memory.";
|
||||||
throw SharedMemoryError(msg);
|
throw SharedMemoryError(msg);
|
||||||
return 1;
|
|
||||||
}
|
}
|
||||||
return 0;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
std::string name;
|
|
||||||
int fd{-1};
|
|
||||||
size_t shmSize{0};
|
|
||||||
T *shared_struct{nullptr};
|
|
||||||
};
|
};
|
||||||
|
|
||||||
} // namespace sls
|
} // namespace sls
|
||||||
|
@ -32,8 +32,8 @@ TEST_CASE("Is shm fixed pattern shm compatible") {
|
|||||||
|
|
||||||
// Set shm version to 0
|
// Set shm version to 0
|
||||||
sls::SharedMemory<sls::sharedModule> shm(0, 0);
|
sls::SharedMemory<sls::sharedModule> shm(0, 0);
|
||||||
REQUIRE(shm.IsExisting() == true);
|
REQUIRE(shm.exists() == true);
|
||||||
shm.OpenSharedMemory();
|
shm.openSharedMemory();
|
||||||
shm()->shmversion = 0;
|
shm()->shmversion = 0;
|
||||||
|
|
||||||
// Should fail since version is set to 0
|
// Should fail since version is set to 0
|
||||||
|
@ -20,8 +20,8 @@ constexpr int shm_id = 10;
|
|||||||
TEST_CASE("Create SharedMemory read and write", "[detector]") {
|
TEST_CASE("Create SharedMemory read and write", "[detector]") {
|
||||||
|
|
||||||
SharedMemory<Data> shm(shm_id, -1);
|
SharedMemory<Data> shm(shm_id, -1);
|
||||||
shm.CreateSharedMemory();
|
shm.createSharedMemory();
|
||||||
CHECK(shm.GetName() == std::string("/slsDetectorPackage_detector_") +
|
CHECK(shm.getName() == std::string("/slsDetectorPackage_detector_") +
|
||||||
std::to_string(shm_id));
|
std::to_string(shm_id));
|
||||||
|
|
||||||
shm()->x = 3;
|
shm()->x = 3;
|
||||||
@ -32,25 +32,25 @@ TEST_CASE("Create SharedMemory read and write", "[detector]") {
|
|||||||
CHECK(shm()->y == 5.7);
|
CHECK(shm()->y == 5.7);
|
||||||
CHECK(std::string(shm()->mess) == "Some string");
|
CHECK(std::string(shm()->mess) == "Some string");
|
||||||
|
|
||||||
shm.UnmapSharedMemory();
|
shm.unmapSharedMemory();
|
||||||
shm.RemoveSharedMemory();
|
shm.removeSharedMemory();
|
||||||
|
|
||||||
CHECK(shm.IsExisting() == false);
|
CHECK(shm.exists() == false);
|
||||||
}
|
}
|
||||||
|
|
||||||
TEST_CASE("Open existing SharedMemory and read", "[detector]") {
|
TEST_CASE("Open existing SharedMemory and read", "[detector]") {
|
||||||
|
|
||||||
{
|
{
|
||||||
SharedMemory<double> shm(shm_id, -1);
|
SharedMemory<double> shm(shm_id, -1);
|
||||||
shm.CreateSharedMemory();
|
shm.createSharedMemory();
|
||||||
*shm() = 5.3;
|
*shm() = 5.3;
|
||||||
}
|
}
|
||||||
|
|
||||||
SharedMemory<double> shm2(shm_id, -1);
|
SharedMemory<double> shm2(shm_id, -1);
|
||||||
shm2.OpenSharedMemory();
|
shm2.openSharedMemory();
|
||||||
CHECK(*shm2() == 5.3);
|
CHECK(*shm2() == 5.3);
|
||||||
|
|
||||||
shm2.RemoveSharedMemory();
|
shm2.removeSharedMemory();
|
||||||
}
|
}
|
||||||
|
|
||||||
TEST_CASE("Creating a second shared memory with the same name throws",
|
TEST_CASE("Creating a second shared memory with the same name throws",
|
||||||
@ -59,24 +59,24 @@ TEST_CASE("Creating a second shared memory with the same name throws",
|
|||||||
SharedMemory<double> shm0(shm_id, -1);
|
SharedMemory<double> shm0(shm_id, -1);
|
||||||
SharedMemory<double> shm1(shm_id, -1);
|
SharedMemory<double> shm1(shm_id, -1);
|
||||||
|
|
||||||
shm0.CreateSharedMemory();
|
shm0.createSharedMemory();
|
||||||
CHECK_THROWS(shm1.CreateSharedMemory());
|
CHECK_THROWS(shm1.createSharedMemory());
|
||||||
shm0.RemoveSharedMemory();
|
shm0.removeSharedMemory();
|
||||||
}
|
}
|
||||||
|
|
||||||
TEST_CASE("Open two shared memories to the same place", "[detector]") {
|
TEST_CASE("Open two shared memories to the same place", "[detector]") {
|
||||||
|
|
||||||
// Create the first shared memory
|
// Create the first shared memory
|
||||||
SharedMemory<Data> shm(shm_id, -1);
|
SharedMemory<Data> shm(shm_id, -1);
|
||||||
shm.CreateSharedMemory();
|
shm.createSharedMemory();
|
||||||
shm()->x = 5;
|
shm()->x = 5;
|
||||||
CHECK(shm()->x == 5);
|
CHECK(shm()->x == 5);
|
||||||
|
|
||||||
// Open the second shared memory with the same name
|
// Open the second shared memory with the same name
|
||||||
SharedMemory<Data> shm2(shm_id, -1);
|
SharedMemory<Data> shm2(shm_id, -1);
|
||||||
shm2.OpenSharedMemory();
|
shm2.openSharedMemory();
|
||||||
CHECK(shm2()->x == 5);
|
CHECK(shm2()->x == 5);
|
||||||
CHECK(shm.GetName() == shm2.GetName());
|
CHECK(shm.getName() == shm2.getName());
|
||||||
|
|
||||||
// Check that they still point to the same place
|
// Check that they still point to the same place
|
||||||
shm2()->x = 7;
|
shm2()->x = 7;
|
||||||
@ -84,31 +84,28 @@ TEST_CASE("Open two shared memories to the same place", "[detector]") {
|
|||||||
|
|
||||||
// Remove only needs to be done once since they refer
|
// Remove only needs to be done once since they refer
|
||||||
// to the same memory
|
// to the same memory
|
||||||
shm2.RemoveSharedMemory();
|
shm2.removeSharedMemory();
|
||||||
CHECK(shm.IsExisting() == false);
|
CHECK(shm.exists() == false);
|
||||||
CHECK(shm2.IsExisting() == false);
|
CHECK(shm2.exists() == false);
|
||||||
}
|
}
|
||||||
|
|
||||||
TEST_CASE("Move SharedMemory", "[detector]") {
|
TEST_CASE("Move SharedMemory", "[detector]") {
|
||||||
|
|
||||||
SharedMemory<Data> shm(shm_id, -1);
|
SharedMemory<Data> shm(shm_id, -1);
|
||||||
CHECK(shm.GetName() == std::string("/slsDetectorPackage_detector_") +
|
CHECK(shm.getName() == std::string("/slsDetectorPackage_detector_") +
|
||||||
std::to_string(shm_id));
|
std::to_string(shm_id));
|
||||||
shm.CreateSharedMemory();
|
shm.createSharedMemory();
|
||||||
shm()->x = 9;
|
shm()->x = 9;
|
||||||
|
|
||||||
CHECK(shm.size() == sizeof(Data));
|
|
||||||
|
|
||||||
SharedMemory<Data> shm2(shm_id + 1, -1);
|
SharedMemory<Data> shm2(shm_id + 1, -1);
|
||||||
shm2 = std::move(shm); // shm is now a moved from object!
|
shm2 = std::move(shm); // shm is now a moved from object!
|
||||||
|
|
||||||
CHECK(shm2()->x == 9);
|
CHECK(shm2()->x == 9);
|
||||||
CHECK(shm() == nullptr);
|
CHECK(shm() == nullptr);
|
||||||
CHECK(shm.size() == 0);
|
CHECK(shm2.getName() == std::string("/slsDetectorPackage_detector_") +
|
||||||
|
|
||||||
CHECK(shm2.GetName() == std::string("/slsDetectorPackage_detector_") +
|
|
||||||
std::to_string(shm_id));
|
std::to_string(shm_id));
|
||||||
shm2.RemoveSharedMemory();
|
shm2.removeSharedMemory();
|
||||||
}
|
}
|
||||||
|
|
||||||
TEST_CASE("Create several shared memories", "[detector]") {
|
TEST_CASE("Create several shared memories", "[detector]") {
|
||||||
@ -117,27 +114,27 @@ TEST_CASE("Create several shared memories", "[detector]") {
|
|||||||
v.reserve(N);
|
v.reserve(N);
|
||||||
for (int i = 0; i != N; ++i) {
|
for (int i = 0; i != N; ++i) {
|
||||||
v.emplace_back(shm_id + i, -1);
|
v.emplace_back(shm_id + i, -1);
|
||||||
CHECK(v[i].IsExisting() == false);
|
CHECK(v[i].exists() == false);
|
||||||
v[i].CreateSharedMemory();
|
v[i].createSharedMemory();
|
||||||
*v[i]() = i;
|
*v[i]() = i;
|
||||||
CHECK(*v[i]() == i);
|
CHECK(*v[i]() == i);
|
||||||
}
|
}
|
||||||
|
|
||||||
for (int i = 0; i != N; ++i) {
|
for (int i = 0; i != N; ++i) {
|
||||||
CHECK(*v[i]() == i);
|
CHECK(*v[i]() == i);
|
||||||
CHECK(v[i].GetName() == std::string("/slsDetectorPackage_detector_") +
|
CHECK(v[i].getName() == std::string("/slsDetectorPackage_detector_") +
|
||||||
std::to_string(i + shm_id));
|
std::to_string(i + shm_id));
|
||||||
}
|
}
|
||||||
|
|
||||||
for (int i = 0; i != N; ++i) {
|
for (int i = 0; i != N; ++i) {
|
||||||
v[i].RemoveSharedMemory();
|
v[i].removeSharedMemory();
|
||||||
CHECK(v[i].IsExisting() == false);
|
CHECK(v[i].exists() == false);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
TEST_CASE("Create create a shared memory with a tag"){
|
TEST_CASE("Create create a shared memory with a tag"){
|
||||||
SharedMemory<int> shm(0, -1, "ctbdacs");
|
SharedMemory<int> shm(0, -1, "ctbdacs");
|
||||||
REQUIRE(shm.GetName() == "/slsDetectorPackage_detector_0_ctbdacs");
|
REQUIRE(shm.getName() == "/slsDetectorPackage_detector_0_ctbdacs");
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -152,7 +149,7 @@ TEST_CASE("Create create a shared memory with a tag when SLSDETNAME is set"){
|
|||||||
setenv(SHM_ENV_NAME, "myprefix", 1);
|
setenv(SHM_ENV_NAME, "myprefix", 1);
|
||||||
|
|
||||||
SharedMemory<int> shm(0, -1, "ctbdacs");
|
SharedMemory<int> shm(0, -1, "ctbdacs");
|
||||||
REQUIRE(shm.GetName() == "/slsDetectorPackage_detector_0_myprefix_ctbdacs");
|
REQUIRE(shm.getName() == "/slsDetectorPackage_detector_0_myprefix_ctbdacs");
|
||||||
|
|
||||||
// Clean up after us
|
// Clean up after us
|
||||||
if (old_slsdetname.empty())
|
if (old_slsdetname.empty())
|
||||||
@ -162,3 +159,17 @@ TEST_CASE("Create create a shared memory with a tag when SLSDETNAME is set"){
|
|||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
TEST_CASE("map int64 to int32 throws"){
|
||||||
|
SharedMemory<int32_t> shm(shm_id, -1);
|
||||||
|
shm.createSharedMemory();
|
||||||
|
*shm() = 7;
|
||||||
|
|
||||||
|
SharedMemory<int64_t> shm2(shm_id, -1);
|
||||||
|
REQUIRE_THROWS(shm2.openSharedMemory());
|
||||||
|
|
||||||
|
shm.removeSharedMemory();
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
}
|
Loading…
x
Reference in New Issue
Block a user