FairMQ: shared memory unmanaged region updates:

- use same mapping for sender/receiver (created/remote) region. - rename FairMQRegion -> FairMQUnmanagedRegion.
2025-10-15 09:31:45 +00:00 · 2017-10-19 11:17:05 +02:00
parent f7d3a5e8ae
commit 0bb866ff36
24 changed files with 212 additions and 327 deletions
--- a/fairmq/shmem/FairMQUnmanagedRegionSHM.h
+++ b/fairmq/shmem/FairMQUnmanagedRegionSHM.h
@@ -0,0 +1,79 @@
+/********************************************************************************
+ *    Copyright (C) 2014 GSI Helmholtzzentrum fuer Schwerionenforschung GmbH    *
+ *                                                                              *
+ *              This software is distributed under the terms of the             *
+ *         GNU Lesser General Public Licence version 3 (LGPL) version 3,        *
+ *                  copied verbatim in the file "LICENSE"                       *
+ ********************************************************************************/
+
+#ifndef FAIRMQUNMANAGEDREGIONSHM_H_
+#define FAIRMQUNMANAGEDREGIONSHM_H_
+
+#include "FairMQUnmanagedRegion.h"
+#include "FairMQLogger.h"
+
+#include <boost/interprocess/shared_memory_object.hpp>
+#include <boost/interprocess/mapped_region.hpp>
+
+#include <cstddef> // size_t
+#include <atomic>
+#include <memory>
+#include <string>
+#include <unordered_map>
+
+struct RemoteRegion // todo: better name?
+{
+    RemoteRegion(std::string regionIdStr, uint64_t size)
+        : fRegionName(regionIdStr)
+        , fShmemObject(boost::interprocess::open_or_create, regionIdStr.c_str(), boost::interprocess::read_write)
+    {
+        if (size > 0)
+        {
+            fShmemObject.truncate(size);
+        }
+        fRegion = boost::interprocess::mapped_region(fShmemObject, boost::interprocess::read_write); // TODO: add HUGEPAGES flag here
+    }
+
+    RemoteRegion() = delete;
+
+    RemoteRegion(const RemoteRegion& rr) = default;
+    RemoteRegion(RemoteRegion&& rr) = default;
+
+    ~RemoteRegion()
+    {
+        if (boost::interprocess::shared_memory_object::remove(fRegionName.c_str()))
+        {
+            LOG(DEBUG) << "destroyed region " << fRegionName;
+        }
+    }
+
+    std::string fRegionName;
+    boost::interprocess::shared_memory_object fShmemObject;
+    boost::interprocess::mapped_region fRegion;
+};
+
+class FairMQUnmanagedRegionSHM : public FairMQUnmanagedRegion
+{
+    friend class FairMQSocketSHM;
+    friend class FairMQMessageSHM;
+
+  public:
+    FairMQUnmanagedRegionSHM(const size_t size);
+
+    virtual void* GetData() const override;
+    virtual size_t GetSize() const override;
+
+    static boost::interprocess::mapped_region* GetRemoteRegion(uint64_t regionId);
+
+    virtual ~FairMQUnmanagedRegionSHM();
+
+  private:
+    static std::atomic<bool> fInterrupted;
+    boost::interprocess::mapped_region* fRegion;
+    uint64_t fRegionId;
+    std::string fRegionIdStr;
+    bool fRemote;
+    static std::unordered_map<uint64_t, RemoteRegion> fRemoteRegionMap;
+};
+
+#endif /* FAIRMQUNMANAGEDREGIONSHM_H_ */