mirror of
https://github.com/FairRootGroup/FairMQ.git
synced 2025-10-12 16:21:13 +00:00
93 lines
3.2 KiB
C++
93 lines
3.2 KiB
C++
/********************************************************************************
|
|
* Copyright (C) 2014 GSI Helmholtzzentrum fuer Schwerionenforschung GmbH *
|
|
* *
|
|
* This software is distributed under the terms of the *
|
|
* GNU Lesser General Public Licence (LGPL) version 3, *
|
|
* copied verbatim in the file "LICENSE" *
|
|
********************************************************************************/
|
|
|
|
#ifndef FAIR_MQ_ZMQ_UNMANAGEDREGION_H
|
|
#define FAIR_MQ_ZMQ_UNMANAGEDREGION_H
|
|
|
|
#include <fairmq/zeromq/Context.h>
|
|
#include <FairMQUnmanagedRegion.h>
|
|
#include <FairMQLogger.h>
|
|
|
|
#include <cstddef> // size_t
|
|
#include <string>
|
|
#include <utility> // move
|
|
|
|
#include <sys/mman.h> // mlock
|
|
|
|
namespace fair::mq::zmq
|
|
{
|
|
|
|
class UnmanagedRegion final : public fair::mq::UnmanagedRegion
|
|
{
|
|
friend class Socket;
|
|
friend class Message;
|
|
|
|
public:
|
|
UnmanagedRegion(Context& ctx,
|
|
size_t size,
|
|
int64_t userFlags,
|
|
RegionCallback callback,
|
|
RegionBulkCallback bulkCallback,
|
|
FairMQTransportFactory* factory,
|
|
fair::mq::RegionConfig cfg)
|
|
: fair::mq::UnmanagedRegion(factory)
|
|
, fCtx(ctx)
|
|
, fId(fCtx.RegionCount())
|
|
, fBuffer(malloc(size))
|
|
, fSize(size)
|
|
, fUserFlags(userFlags)
|
|
, fCallback(std::move(callback))
|
|
, fBulkCallback(std::move(bulkCallback))
|
|
{
|
|
if (cfg.lock) {
|
|
LOG(debug) << "Locking region " << fId << "...";
|
|
if (mlock(fBuffer, fSize) == -1) {
|
|
LOG(error) << "Could not lock region " << fId << ". Code: " << errno << ", reason: " << strerror(errno);
|
|
}
|
|
LOG(debug) << "Successfully locked region " << fId << ".";
|
|
}
|
|
if (cfg.zero) {
|
|
LOG(debug) << "Zeroing free memory of region " << fId << "...";
|
|
memset(fBuffer, 0x00, fSize);
|
|
LOG(debug) << "Successfully zeroed free memory of region " << fId << ".";
|
|
}
|
|
}
|
|
|
|
UnmanagedRegion(const UnmanagedRegion&) = delete;
|
|
UnmanagedRegion operator=(const UnmanagedRegion&) = delete;
|
|
|
|
virtual void* GetData() const override { return fBuffer; }
|
|
virtual size_t GetSize() const override { return fSize; }
|
|
uint16_t GetId() const override { return fId; }
|
|
int64_t GetUserFlags() const { return fUserFlags; }
|
|
void SetLinger(uint32_t /* linger */) override { LOG(debug) << "ZeroMQ UnmanagedRegion linger option not implemented. Acknowledgements are local."; }
|
|
uint32_t GetLinger() const override { LOG(debug) << "ZeroMQ UnmanagedRegion linger option not implemented. Acknowledgements are local."; return 0; }
|
|
|
|
Transport GetType() const override { return Transport::ZMQ; }
|
|
|
|
virtual ~UnmanagedRegion()
|
|
{
|
|
LOG(debug) << "destroying region " << fId;
|
|
fCtx.RemoveRegion(fId);
|
|
free(fBuffer);
|
|
}
|
|
|
|
private:
|
|
Context& fCtx;
|
|
uint16_t fId;
|
|
void* fBuffer;
|
|
size_t fSize;
|
|
int64_t fUserFlags;
|
|
RegionCallback fCallback;
|
|
RegionBulkCallback fBulkCallback;
|
|
};
|
|
|
|
} // namespace fair::mq::zmq
|
|
|
|
#endif /* FAIR_MQ_ZMQ_UNMANAGEDREGION_H */
|